gen-lockedsig-cache 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. #!/usr/bin/env python3
  2. #
  3. # SPDX-License-Identifier: GPL-2.0-only
  4. #
  5. import os
  6. import sys
  7. import glob
  8. import shutil
  9. import errno
  10. def mkdir(d):
  11. try:
  12. os.makedirs(d)
  13. except OSError as e:
  14. if e.errno != errno.EEXIST:
  15. raise e
  16. if len(sys.argv) < 5:
  17. print("Incorrect number of arguments specified")
  18. print("syntax: gen-lockedsig-cache <locked-sigs.inc> <input-cachedir> <output-cachedir> <nativelsbstring> [filterfile]")
  19. sys.exit(1)
  20. filterlist = []
  21. if len(sys.argv) > 5:
  22. print('Reading filter file %s' % sys.argv[5])
  23. with open(sys.argv[5]) as f:
  24. for l in f.readlines():
  25. if ":" in l:
  26. filterlist.append(l.rstrip())
  27. print('Reading %s' % sys.argv[1])
  28. sigs = []
  29. with open(sys.argv[1]) as f:
  30. for l in f.readlines():
  31. if ":" in l:
  32. task, sig = l.split()[0].rsplit(':', 1)
  33. if filterlist and not task in filterlist:
  34. print('Filtering out %s' % task)
  35. else:
  36. sigs.append(sig)
  37. print('Gathering file list')
  38. files = set()
  39. for s in sigs:
  40. p = sys.argv[2] + "/" + s[:2] + "/*" + s + "*"
  41. files |= set(glob.glob(p))
  42. p = sys.argv[2] + "/%s/" % sys.argv[4] + s[:2] + "/*" + s + "*"
  43. files |= set(glob.glob(p))
  44. print('Processing files')
  45. for f in files:
  46. sys.stdout.write('Processing %s... ' % f)
  47. _, ext = os.path.splitext(f)
  48. if not ext in ['.tgz', '.siginfo', '.sig']:
  49. # Most likely a temp file, skip it
  50. print('skipping')
  51. continue
  52. dst = os.path.join(sys.argv[3], os.path.relpath(f, sys.argv[2]))
  53. destdir = os.path.dirname(dst)
  54. mkdir(destdir)
  55. src = os.path.realpath(f)
  56. if os.path.exists(dst):
  57. os.remove(dst)
  58. if (os.stat(src).st_dev == os.stat(destdir).st_dev):
  59. print('linking')
  60. try:
  61. os.link(src, dst)
  62. except OSError as e:
  63. print('hard linking failed, copying')
  64. shutil.copyfile(src, dst)
  65. else:
  66. print('copying')
  67. shutil.copyfile(src, dst)
  68. print('Done!')