1#!@PYTHON@ 2# 3# CDDL HEADER START 4# 5# The contents of this file are subject to the terms of the 6# Common Development and Distribution License (the "License"). 7# You may not use this file except in compliance with the License. 8# 9# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 10# or http://www.opensolaris.org/os/licensing. 11# See the License for the specific language governing permissions 12# and limitations under the License. 13# 14# When distributing Covered Code, include this CDDL HEADER in each 15# file and include the License file at usr/src/OPENSOLARIS.LICENSE. 16# If applicable, add the following below this CDDL HEADER, with the 17# fields enclosed by brackets "[]" replaced with your own identifying 18# information: Portions Copyright [yyyy] [name of copyright owner] 19# 20# CDDL HEADER END 21# 22 23# 24# Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved. 25# 26 27# 28# Check for valid link-editor mapfile comment blocks in source files. 29# 30 31import sys, os, getopt, fnmatch 32 33sys.path.insert(1, os.path.join(os.path.dirname(__file__), "..", "lib", 34 "python%d.%d" % sys.version_info[:2])) 35 36# Allow running from the source tree, using the modules in the source tree 37sys.path.insert(2, os.path.join(os.path.dirname(__file__), '..')) 38 39from onbld.Checks.Mapfile import mapfilechk 40 41class ExceptionList(object): 42 def __init__(self): 43 self.dirs = [] 44 self.files = [] 45 self.extensions = [] 46 47 def load(self, exfile): 48 fh = None 49 try: 50 fh = open(exfile, 'r') 51 except IOError, e: 52 sys.stderr.write('Failed to open exception list: ' 53 '%s: %s\n' % (e.filename, e.strerror)) 54 sys.exit(2) 55 56 for line in fh: 57 line = line.strip() 58 59 if line.strip().endswith('/'): 60 self.dirs.append(line[0:-1]) 61 elif line.startswith('*.'): 62 self.extensions.append(line) 63 else: 64 self.files.append(line) 65 66 fh.close() 67 68 def match(self, filename): 69 if os.path.isdir(filename): 70 return filename in self.dirs 71 else: 72 if filename in self.files: 73 return True 74 75 for pat in self.extensions: 76 if fnmatch.fnmatch(filename, pat): 77 return True 78 79 def __contains__(self, elt): 80 return self.match(elt) 81 82def usage(): 83 progname = os.path.split(sys.argv[0])[1] 84 sys.stderr.write('''Usage: %s [-v] [-x exceptions] paths... 85 -v report on all files, not just those with errors. 86 -x exceptions load an exceptions file 87''' % progname) 88 sys.exit(2) 89 90 91def check(filename, opts): 92 try: 93 fh = open(filename, 'r') 94 except IOError, e: 95 sys.stderr.write("failed to open '%s': %s\n" % 96 (e.filename, e.strerror)) 97 return 1 98 else: 99 return mapfilechk(fh, verbose=opts['verbose'], 100 output=sys.stdout) 101 102def walker(opts, dirname, fnames): 103 for f in fnames: 104 path = os.path.join(dirname, f) 105 106 if not os.path.isdir(path): 107 if not path in opts['exclude']: 108 opts['status'] |= check(path, opts) 109 else: 110 if path in opts['exclude']: 111 fnames.remove(f) 112 113def walkpath(path, opts): 114 if os.path.isdir(path): 115 os.path.walk(path, walker, opts) 116 else: 117 if not path in opts['exclude']: 118 opts['status'] |= check(path, opts) 119 120def main(args): 121 options = { 122 'status': 0, 123 'verbose': False, 124 'exclude': ExceptionList() 125 } 126 127 try: 128 opts, args = getopt.getopt(sys.argv[1:], 'avx:') 129 except getopt.GetoptError: 130 usage() 131 sys.exit(2) 132 133 for opt, arg in opts: 134 if opt == '-v': 135 options['verbose'] = True 136 elif opt == '-x': 137 options['exclude'].load(arg) 138 139 for path in args: 140 walkpath(path, options) 141 142 return options['status'] 143 144if __name__ == '__main__': 145 sys.exit(main(sys.argv[1:])) 146