1#!@TOOLS_PYTHON@ -Es
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# Copyright 2018 OmniOS Community Edition (OmniOSce) Association.
26#
27
28#
29# Check for valid link-editor mapfile comment blocks in source files.
30#
31
32import sys, os, io, getopt, fnmatch
33
34sys.path.insert(1, os.path.join(os.path.dirname(__file__), "..", "lib",
35                                "python%d.%d" % sys.version_info[:2]))
36
37# Allow running from the source tree, using the modules in the source tree
38sys.path.insert(2, os.path.join(os.path.dirname(__file__), '..'))
39
40from onbld.Checks.Mapfile import mapfilechk
41
42class ExceptionList(object):
43	def __init__(self):
44		self.dirs = []
45		self.files = []
46		self.extensions = []
47
48	def load(self, exfile):
49		fh = None
50		try:
51			fh = open(exfile, 'r')
52		except IOError as e:
53			sys.stderr.write('Failed to open exception list: '
54					 '%s: %s\n' % (e.filename, e.strerror))
55			sys.exit(2)
56
57		for line in fh:
58			line = line.strip()
59
60			if line.strip().endswith('/'):
61				self.dirs.append(line[0:-1])
62			elif line.startswith('*.'):
63				self.extensions.append(line)
64			else:
65				self.files.append(line)
66
67		fh.close()
68
69	def match(self, filename):
70		if os.path.isdir(filename):
71			return filename in self.dirs
72		else:
73			if filename in self.files:
74				return True
75
76			for pat in self.extensions:
77				if fnmatch.fnmatch(filename, pat):
78					return True
79
80	def __contains__(self, elt):
81		return self.match(elt)
82
83def usage():
84	progname = os.path.split(sys.argv[0])[1]
85	sys.stderr.write('''Usage: %s [-v] [-x exceptions] paths...
86        -v		report on all files, not just those with errors.
87        -x exceptions	load an exceptions file
88''' % progname)
89	sys.exit(2)
90
91
92def check(filename, opts):
93	try:
94		with io.open(filename, encoding='utf-8',
95		    errors='replace') as fh:
96			return mapfilechk(fh, verbose=opts['verbose'],
97			       output=sys.stdout)
98	except IOError as e:
99		sys.stderr.write("failed to open '%s': %s\n" %
100				 (e.filename, e.strerror))
101		return 1
102
103def walker(opts, dirname, fnames):
104	for f in fnames:
105		path = os.path.join(dirname, f)
106
107		if not os.path.isdir(path):
108			if not path in opts['exclude']:
109				opts['status'] |= check(path, opts)
110		else:
111			if path in opts['exclude']:
112				fnames.remove(f)
113
114def walkpath(path, opts):
115	if os.path.isdir(path):
116		os.path.walk(path, walker, opts)
117	else:
118		if not path in opts['exclude']:
119			opts['status'] |= check(path, opts)
120
121def main(args):
122	options = {
123		'status': 0,
124		'verbose': False,
125		'exclude': ExceptionList()
126	}
127
128	try:
129		opts, args = getopt.getopt(sys.argv[1:], 'avx:')
130	except getopt.GetoptError:
131		usage()
132		sys.exit(2)
133
134	for opt, arg in opts:
135		if opt == '-v':
136			options['verbose'] = True
137		elif opt == '-x':
138			options['exclude'].load(arg)
139
140	for path in args:
141		walkpath(path, options)
142
143	return options['status']
144
145if __name__ == '__main__':
146	sys.exit(main(sys.argv[1:]))
147