868c2d3a42369543ca6d83c514d36b405ad006f7
[people/andreif/gpxe.git] / contrib / errcode / build_errcodedb.py
1 #!/usr/bin/env python
2 # Copyright (C) 2008 Stefan Hajnoczi <stefanha@gmail.com>.
3 #
4 # This program is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU General Public License as
6 # published by the Free Software Foundation; either version 2 of the
7 # License, or any later version.
8 #
9 # This program is distributed in the hope that it will be useful, but
10 # WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 # General Public License for more details.
13 #
14 # You should have received a copy of the GNU General Public License
15 # along with this program; if not, write to the Free Software
16 # Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17 import sys
18 import re
19
20 pxenv_status_files = ('../../src/include/errno.h', )
21 errfile_files = ('../../src/include/gpxe/errfile.h',
22             '../../src/arch/i386/include/bits/errfile.h')
23 posix_errno_files = ('../../src/include/errno.h', )
24
25 PXENV_STATUS_RE = re.compile(r'^#define\s+(PXENV_STATUS_[^\s]+)\s+(.+)$')
26 ERRFILE_RE = re.compile(r'^#define\s+(ERRFILE_[^\s]+)\s+(.+)$')
27 POSIX_ERRNO_RE = re.compile(r'^#define\s+(E[A-Z]+)\s+.*(0x[0-9a-f]+).*$')
28
29 def err(msg):
30     sys.stderr.write('%s: %s\n' % (sys.argv[0], msg))
31     sys.exit(1)
32
33 def to_pxenv_status(errno):
34     return errno & 0xff
35
36 def to_errfile(errno):
37     return (errno >> 13) & 0x7ff
38
39 def to_posix_errno(errno):
40     return (errno >> 24) & 0x7f
41
42 def load_header_file(filename, regexp):
43     defines = {}
44     for line in open(filename, 'r'):
45         m = regexp.match(line)
46         if m:
47             key, val = m.groups()
48             defines[key] = val
49     return defines
50
51 def evaluate(defines, expr):
52     pyexpr = ''
53     for token in expr.split():
54         if token in '()':
55             pass
56         elif token.startswith('/*') or token.startswith('//'):
57             break
58         elif token.startswith('0x') or token == '|':
59             pyexpr += token
60         else:
61             if token in defines:
62                 pyexpr += '0x%x' % defines[token]
63             else:
64                 return -1
65     if not re.match(r'^[0-9a-zA-Z_|]+$', pyexpr):
66         err('invalid expression')
67     return eval(pyexpr)
68
69 def build(filenames, regexp, selector):
70     unevaluated = {}
71     for filename in filenames:
72         unevaluated.update(load_header_file(filename, regexp))
73
74     evaluated = {}
75     changed = True
76     while changed:
77         changed = False
78         for key in list(unevaluated.keys()):
79             val = evaluate(evaluated, unevaluated[key])
80             if val != -1:
81                 del unevaluated[key]
82                 evaluated[key] = val
83                 changed = True
84     if unevaluated:
85         err('unable to evaluate all #defines')
86
87     lookup = {}
88     for key, val in evaluated.iteritems():
89         lookup[selector(val)] = key
90     return lookup
91
92 print 'pxenv_status =', repr(build(pxenv_status_files, PXENV_STATUS_RE, to_pxenv_status))
93 print 'errfile =', repr(build(errfile_files, ERRFILE_RE, to_errfile))
94 print 'posix_errno =', repr(build(posix_errno_files, POSIX_ERRNO_RE, to_posix_errno))