summaryrefslogtreecommitdiff
path: root/tools/file2carray.py
blob: 7150364bf1ca1db913a898eb227fc2f0e4051367 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#!/usr/bin/env python3
import argparse
import random
import os
import sys
import time
import subprocess
from pathlib import Path
from multiprocessing import Pool
from weakref import finalize


def print_carray(f, payload):
    while len(payload) > 0:
        f.write('\n    ')
        f.write(', '.join('0x{:02x}'.format(x) for x in payload[0:16]))
        f.write(',')
        payload = payload[16:]
    f.write('\n')


def main():
    parser = argparse.ArgumentParser(description='Convert binary files to C array format')
    parser.add_argument('files', nargs='+', help='Binary files to convert')
    args = parser.parse_args()

    files = args.files
    for fin_name in files:
        if not os.path.isfile(fin_name):
            print(f"File {fin_name} does not exist")
            continue

        with open(fin_name, 'rb') as fin:
            contents = fin.read()
            fout_name = fin_name + '.h'
            with open(fout_name, 'w') as fout:
                print(f"Converting {fin_name} to {fout_name}")
                fout.write(f'enum {{ BINDATA_LEN = {len(contents)} }};\n')
                fout.write(f'const size_t bindata_len = BINDATA_LEN;\n')
                fout.write(f'const uint8_t bindata[] __attribute__((aligned(16))) = {{')
                print_carray(fout, contents)
                fout.write('};\n')


if __name__ == '__main__':
    sys.exit(main())