makeqstrdata.py 2.07 KB
Newer Older
1
2
import argparse
import re
3
from htmlentitydefs import codepoint2name
4
5
6
7
8
9
10
11
12
13

# this must match the equivalent function in qstr.c
def compute_hash(qstr):
    hash = 0
    for char in qstr:
        hash += ord(char)
    return hash & 0xffff

def do_work(infiles):
    # read the qstrs in from the input files
14
    qstrs = {}
15
16
17
18
19
20
21
22
23
24
25
26
    for infile in infiles:
        with open(infile, 'rt') as f:
            line_number = 0
            for line in f:
                line_number += 1
                line = line.strip()

                # ignore blank lines and comments
                if len(line) == 0 or line.startswith('//'):
                    continue

                # verify line is of the correct form
27
                match = re.match(r'Q\((.+)\)$', line)
28
29
30
31
32
33
                if not match:
                    print('({}:{}) bad qstr format, got {}'.format(infile, line_number, line))
                    return False

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

                # don't add duplicates
37
                if ident in qstrs:
38
39
40
                    continue

                # add the qstr to the list
41
                qstrs[ident] = qstr
42
43
44

    # process the qstrs, printing out the generated C header file
    print('// This file was automatically generated by makeqstrdata.py')
45
    print('')
46
    for ident, qstr in qstrs.items():
47
48
        qhash = compute_hash(qstr)
        qlen = len(qstr)
49
        print('Q({}, (const byte*)"\\x{:02x}\\x{:02x}\\x{:02x}\\x{:02x}" "{}")'.format(ident, qhash & 0xff, (qhash >> 8) & 0xff, qlen & 0xff, (qlen >> 8) & 0xff, qstr))
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

    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:
        print('exiting with error code')
        exit(1)

if __name__ == "__main__":
    main()