e559c6294c39aef2819637125520a0721b14a95d
[openwrt/staging/chunkeey.git] / scripts / spdxcheck.py
1 #!/usr/bin/env python
2 # SPDX-License-Identifier: GPL-2.0
3 # Copyright Thomas Gleixner <tglx@linutronix.de>
4
5 from argparse import ArgumentParser
6 from ply import lex, yacc
7 import locale
8 import traceback
9 import sys
10 import git
11 import re
12 import os
13
14 class ParserException(Exception):
15 def __init__(self, tok, txt):
16 self.tok = tok
17 self.txt = txt
18
19 class SPDXException(Exception):
20 def __init__(self, el, txt):
21 self.el = el
22 self.txt = txt
23
24 class SPDXdata(object):
25 def __init__(self):
26 self.license_files = 0
27 self.exception_files = 0
28 self.licenses = [ ]
29 self.exceptions = { }
30
31 # Read the spdx data from the LICENSES directory
32 def read_spdxdata(repo):
33
34 # The subdirectories of LICENSES in the kernel source
35 license_dirs = [ "preferred", "other", "exceptions" ]
36 lictree = repo.head.commit.tree['LICENSES']
37
38 spdx = SPDXdata()
39
40 for d in license_dirs:
41 for el in lictree[d].traverse():
42 if not os.path.isfile(el.path):
43 continue
44
45 exception = None
46 for l in open(el.path).readlines():
47 if l.startswith('Valid-License-Identifier:'):
48 lid = l.split(':')[1].strip().upper()
49 if lid in spdx.licenses:
50 raise SPDXException(el, 'Duplicate License Identifier: %s' %lid)
51 else:
52 spdx.licenses.append(lid)
53
54 elif l.startswith('SPDX-Exception-Identifier:'):
55 exception = l.split(':')[1].strip().upper()
56 spdx.exceptions[exception] = []
57
58 elif l.startswith('SPDX-Licenses:'):
59 for lic in l.split(':')[1].upper().strip().replace(' ', '').replace('\t', '').split(','):
60 if not lic in spdx.licenses:
61 raise SPDXException(None, 'Exception %s missing license %s' %(ex, lic))
62 spdx.exceptions[exception].append(lic)
63
64 elif l.startswith("License-Text:"):
65 if exception:
66 if not len(spdx.exceptions[exception]):
67 raise SPDXException(el, 'Exception %s is missing SPDX-Licenses' %excid)
68 spdx.exception_files += 1
69 else:
70 spdx.license_files += 1
71 break
72 return spdx
73
74 class id_parser(object):
75
76 reserved = [ 'AND', 'OR', 'WITH' ]
77 tokens = [ 'LPAR', 'RPAR', 'ID', 'EXC' ] + reserved
78
79 precedence = ( ('nonassoc', 'AND', 'OR'), )
80
81 t_ignore = ' \t'
82
83 def __init__(self, spdx):
84 self.spdx = spdx
85 self.lasttok = None
86 self.lastid = None
87 self.lexer = lex.lex(module = self, reflags = re.UNICODE)
88 # Initialize the parser. No debug file and no parser rules stored on disk
89 # The rules are small enough to be generated on the fly
90 self.parser = yacc.yacc(module = self, write_tables = False, debug = False)
91 self.lines_checked = 0
92 self.checked = 0
93 self.spdx_valid = 0
94 self.spdx_errors = 0
95 self.curline = 0
96 self.deepest = 0
97
98 # Validate License and Exception IDs
99 def validate(self, tok):
100 id = tok.value.upper()
101 if tok.type == 'ID':
102 if not id in self.spdx.licenses:
103 raise ParserException(tok, 'Invalid License ID')
104 self.lastid = id
105 elif tok.type == 'EXC':
106 if id not in self.spdx.exceptions:
107 raise ParserException(tok, 'Invalid Exception ID')
108 if self.lastid not in self.spdx.exceptions[id]:
109 raise ParserException(tok, 'Exception not valid for license %s' %self.lastid)
110 self.lastid = None
111 elif tok.type != 'WITH':
112 self.lastid = None
113
114 # Lexer functions
115 def t_RPAR(self, tok):
116 r'\)'
117 self.lasttok = tok.type
118 return tok
119
120 def t_LPAR(self, tok):
121 r'\('
122 self.lasttok = tok.type
123 return tok
124
125 def t_ID(self, tok):
126 r'[A-Za-z.0-9\-+]+'
127
128 if self.lasttok == 'EXC':
129 print(tok)
130 raise ParserException(tok, 'Missing parentheses')
131
132 tok.value = tok.value.strip()
133 val = tok.value.upper()
134
135 if val in self.reserved:
136 tok.type = val
137 elif self.lasttok == 'WITH':
138 tok.type = 'EXC'
139
140 self.lasttok = tok.type
141 self.validate(tok)
142 return tok
143
144 def t_error(self, tok):
145 raise ParserException(tok, 'Invalid token')
146
147 def p_expr(self, p):
148 '''expr : ID
149 | ID WITH EXC
150 | expr AND expr
151 | expr OR expr
152 | LPAR expr RPAR'''
153 pass
154
155 def p_error(self, p):
156 if not p:
157 raise ParserException(None, 'Unfinished license expression')
158 else:
159 raise ParserException(p, 'Syntax error')
160
161 def parse(self, expr):
162 self.lasttok = None
163 self.lastid = None
164 self.parser.parse(expr, lexer = self.lexer)
165
166 def parse_lines(self, fd, maxlines, fname):
167 self.checked += 1
168 self.curline = 0
169 try:
170 for line in fd:
171 line = line.decode(locale.getpreferredencoding(False), errors='ignore')
172 self.curline += 1
173 if self.curline > maxlines:
174 break
175 self.lines_checked += 1
176 if line.find("SPDX-License-Identifier:") < 0:
177 continue
178 expr = line.split(':')[1].replace('*/', '').strip()
179 self.parse(expr)
180 self.spdx_valid += 1
181 #
182 # Should we check for more SPDX ids in the same file and
183 # complain if there are any?
184 #
185 break
186
187 except ParserException as pe:
188 if pe.tok:
189 col = line.find(expr) + pe.tok.lexpos
190 tok = pe.tok.value
191 sys.stdout.write('%s: %d:%d %s: %s\n' %(fname, self.curline, col, pe.txt, tok))
192 else:
193 sys.stdout.write('%s: %d:0 %s\n' %(fname, self.curline, col, pe.txt))
194 self.spdx_errors += 1
195
196 def scan_git_tree(tree):
197 for el in tree.traverse():
198 # Exclude stuff which would make pointless noise
199 # FIXME: Put this somewhere more sensible
200 if el.path.startswith("LICENSES"):
201 continue
202 if el.path.find("license-rules.rst") >= 0:
203 continue
204 if not os.path.isfile(el.path):
205 continue
206 with open(el.path, 'rb') as fd:
207 parser.parse_lines(fd, args.maxlines, el.path)
208
209 def scan_git_subtree(tree, path):
210 for p in path.strip('/').split('/'):
211 tree = tree[p]
212 scan_git_tree(tree)
213
214 if __name__ == '__main__':
215
216 ap = ArgumentParser(description='SPDX expression checker')
217 ap.add_argument('path', nargs='*', help='Check path or file. If not given full git tree scan. For stdin use "-"')
218 ap.add_argument('-m', '--maxlines', type=int, default=15,
219 help='Maximum number of lines to scan in a file. Default 15')
220 ap.add_argument('-v', '--verbose', action='store_true', help='Verbose statistics output')
221 args = ap.parse_args()
222
223 # Sanity check path arguments
224 if '-' in args.path and len(args.path) > 1:
225 sys.stderr.write('stdin input "-" must be the only path argument\n')
226 sys.exit(1)
227
228 try:
229 # Use git to get the valid license expressions
230 repo = git.Repo(os.getcwd())
231 assert not repo.bare
232
233 # Initialize SPDX data
234 spdx = read_spdxdata(repo)
235
236 # Initilize the parser
237 parser = id_parser(spdx)
238
239 except SPDXException as se:
240 if se.el:
241 sys.stderr.write('%s: %s\n' %(se.el.path, se.txt))
242 else:
243 sys.stderr.write('%s\n' %se.txt)
244 sys.exit(1)
245
246 except Exception as ex:
247 sys.stderr.write('FAIL: %s\n' %ex)
248 sys.stderr.write('%s\n' %traceback.format_exc())
249 sys.exit(1)
250
251 try:
252 if len(args.path) and args.path[0] == '-':
253 stdin = os.fdopen(sys.stdin.fileno(), 'rb')
254 parser.parse_lines(stdin, args.maxlines, '-')
255 else:
256 if args.path:
257 for p in args.path:
258 if os.path.isfile(p):
259 parser.parse_lines(open(p, 'rb'), args.maxlines, p)
260 elif os.path.isdir(p):
261 scan_git_subtree(repo.head.reference.commit.tree, p)
262 else:
263 sys.stderr.write('path %s does not exist\n' %p)
264 sys.exit(1)
265 else:
266 # Full git tree scan
267 scan_git_tree(repo.head.commit.tree)
268
269 if args.verbose:
270 sys.stderr.write('\n')
271 sys.stderr.write('License files: %12d\n' %spdx.license_files)
272 sys.stderr.write('Exception files: %12d\n' %spdx.exception_files)
273 sys.stderr.write('License IDs %12d\n' %len(spdx.licenses))
274 sys.stderr.write('Exception IDs %12d\n' %len(spdx.exceptions))
275 sys.stderr.write('\n')
276 sys.stderr.write('Files checked: %12d\n' %parser.checked)
277 sys.stderr.write('Lines checked: %12d\n' %parser.lines_checked)
278 sys.stderr.write('Files with SPDX: %12d\n' %parser.spdx_valid)
279 sys.stderr.write('Files with errors: %12d\n' %parser.spdx_errors)
280
281 sys.exit(0)
282
283 except Exception as ex:
284 sys.stderr.write('FAIL: %s\n' %ex)
285 sys.stderr.write('%s\n' %traceback.format_exc())
286 sys.exit(1)