File: bin_to_c.py

package info (click to toggle)
optee-os 4.7.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 31,560 kB
  • sloc: ansic: 441,914; asm: 12,903; python: 3,719; makefile: 1,676; sh: 238
file content (74 lines) | stat: -rwxr-xr-x 2,051 bytes parent folder | download | duplicates (2)
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#!/usr/bin/env python3
# SPDX-License-Identifier: BSD-2-Clause
#
# Copyright (c) 2018, Linaro Limited
#

import argparse
import array
import os
import re
import sys


def get_args():

    parser = argparse.ArgumentParser(description='Converts a binary file '
                                     'into C source file defining binary '
                                     'data as a constant byte array.')

    parser.add_argument('--bin', required=True,
                        help='Path to the input binary file')

    parser.add_argument('--vname', required=True,
                        help='Variable name for the generated table in '
                        'the output C source file.')

    parser.add_argument('--out', required=True,
                        help='Path for the generated C file')

    parser.add_argument('--text', required=False, action='store_true',
                        help='Treat input as a text file')

    return parser.parse_args()


def main():

    args = get_args()

    with open(args.bin, 'rb') as indata:
        bytes = indata.read()
        if args.text:
            bytes += b'\0'
        size = len(bytes)

    f = open(args.out, 'w')
    f.write('/* Generated from ' + args.bin + ' by ' +
            os.path.basename(__file__) + ' */\n\n')
    f.write('#include <compiler.h>\n')
    f.write('#include <stdint.h>\n')
    if args.text:
        f.write('__extension__ const char ' + args.vname + '[] = {\n')
    else:
        f.write('__extension__ const uint8_t ' + args.vname + '[] ' +
                ' __aligned(__alignof__(uint64_t)) = {\n')
    i = 0
    while i < size:
        if i % 8 == 0:
            f.write('\t\t')
        if args.text and i != size - 1 and bytes[i] == b'\0':
            print('Error: null byte encountered in text file')
            sys.exit(1)
        f.write(hex(bytes[i]) + ',')
        i = i + 1
        if i % 8 == 0 or i == size:
            f.write('\n')
        else:
            f.write(' ')
    f.write('};\n')
    f.close()


if __name__ == "__main__":
    main()