makeqstrdata.py 2.92 KB
Newer Older
1
2
from __future__ import print_function

3
4
import argparse
import re
5
import sys
6
7
8
9
10
11
12

# codepoint2name is different in Python 2 to Python 3
import platform
if platform.python_version_tuple()[0] == '2':
    from htmlentitydefs import codepoint2name
elif platform.python_version_tuple()[0] == '3':
    from html.entities import codepoint2name
13
codepoint2name[ord('-')] = 'hyphen';
14

15
# add some custom names to map characters that aren't in HTML
16
17
18
codepoint2name[ord(' ')] = 'space'
codepoint2name[ord('\'')] = 'squot'
codepoint2name[ord(',')] = 'comma'
19
codepoint2name[ord('.')] = 'dot'
20
21
codepoint2name[ord(':')] = 'colon'
codepoint2name[ord('/')] = 'slash'
Damien George's avatar
Damien George committed
22
codepoint2name[ord('%')] = 'percent'
23
codepoint2name[ord('#')] = 'hash'
24
25
26
27
codepoint2name[ord('(')] = 'paren_open'
codepoint2name[ord(')')] = 'paren_close'
codepoint2name[ord('[')] = 'bracket_open'
codepoint2name[ord(']')] = 'bracket_close'
28
29
codepoint2name[ord('{')] = 'brace_open'
codepoint2name[ord('}')] = 'brace_close'
30
codepoint2name[ord('*')] = 'star'
31
codepoint2name[ord('!')] = 'bang'
32

33
34
# this must match the equivalent function in qstr.c
def compute_hash(qstr):
35
    hash = 5381
36
    for char in qstr:
37
        hash = (hash * 33) ^ ord(char)
38
39
    # Make sure that valid hash is never zero, zero means "hash not computed"
    return (hash & 0xffff) or 1
40
41
42

def do_work(infiles):
    # read the qstrs in from the input files
43
    qstrs = {}
44
45
46
    for infile in infiles:
        with open(infile, 'rt') as f:
            for line in f:
47
48
49
                # is this a QSTR line?
                match = re.match(r'^Q\((.+)\)$', line.strip())
                if not match:
50
                    continue
51
52
53

                # get the qstr value
                qstr = match.group(1)
54
                ident = re.sub(r'[^A-Za-z0-9_]', lambda s: "_" + codepoint2name[ord(s.group(0))] + "_", qstr)
55
56

                # don't add duplicates
57
                if ident in qstrs:
58
59
                    continue

60
                # add the qstr to the list, with order number to retain original order in file
61
                qstrs[ident] = (len(qstrs), ident, qstr)
62
63
64

    # process the qstrs, printing out the generated C header file
    print('// This file was automatically generated by makeqstrdata.py')
65
    print('')
66
    for order, ident, qstr in sorted(qstrs.values(), key=lambda x: x[0]):
67
68
        qhash = compute_hash(qstr)
        qlen = len(qstr)
69
70
        qdata = qstr.replace('"', '\\"')
        print('Q(%s, (const byte*)"\\x%02x\\x%02x\\x%02x\\x%02x" "%s")' % (ident, qhash & 0xff, (qhash >> 8) & 0xff, qlen & 0xff, (qlen >> 8) & 0xff, qdata))
71
72
73
74
75
76
77
78
79
80

    return True

def main():
    arg_parser = argparse.ArgumentParser(description='Process raw qstr file and output qstr data with length, hash and data bytes')
    arg_parser.add_argument('files', nargs='+', help='input file(s)')
    args = arg_parser.parse_args()

    result = do_work(args.files)
    if not result:
81
        print('exiting with error code', file=sys.stderr)
82
83
84
85
        exit(1)

if __name__ == "__main__":
    main()