1 #!/usr/bin/env python 2 3 """ 4 Check boot jars. 5 6 Usage: check_boot_jars.py <package_whitelist_file> <jar1> <jar2> ... 7 """ 8 import logging 9 import os.path 10 import re 11 import subprocess 12 import sys 13 14 15 # The compiled whitelist RE. 16 whitelist_re = None 17 18 19 def LoadWhitelist(filename): 20 """ Load and compile whitelist regular expressions from filename. 21 """ 22 lines = [] 23 with open(filename, 'r') as f: 24 for line in f: 25 line = line.strip() 26 if not line or line.startswith('#'): 27 continue 28 lines.append(line) 29 combined_re = r'^(%s)$' % '|'.join(lines) 30 global whitelist_re 31 try: 32 whitelist_re = re.compile(combined_re) 33 except re.error: 34 logging.exception( 35 'Cannot compile package whitelist regular expression: %r', 36 combined_re) 37 whitelist_re = None 38 return False 39 return True 40 41 42 def CheckJar(jar): 43 """Check a jar file. 44 """ 45 # Get the list of files inside the jar file. 46 p = subprocess.Popen(args='jar tf %s' % jar, 47 stdout=subprocess.PIPE, shell=True) 48 stdout, _ = p.communicate() 49 if p.returncode != 0: 50 return False 51 items = stdout.split() 52 for f in items: 53 if f.endswith('.class'): 54 package_name = os.path.dirname(f) 55 package_name = package_name.replace('/', '.') 56 # Skip class without a package name 57 if package_name and not whitelist_re.match(package_name): 58 print >> sys.stderr, ('Error: %s: unknown package name of class file %s' 59 % (jar, f)) 60 return False 61 return True 62 63 64 def main(argv): 65 if len(argv) < 2: 66 print __doc__ 67 sys.exit(1) 68 69 if not LoadWhitelist(argv[0]): 70 sys.exit(1) 71 72 passed = True 73 for jar in argv[1:]: 74 if not CheckJar(jar): 75 passed = False 76 if not passed: 77 return 1 78 79 return 0 80 81 82 if __name__ == '__main__': 83 main(sys.argv[1:]) 84
