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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
# ========================== begin_copyright_notice ============================
#
# Copyright (C) 2017-2021 Intel Corporation
#
# SPDX-License-Identifier: MIT
#
# =========================== end_copyright_notice =============================
import binascii
import os
import re
import sys
def PrintHelp():
sys.stdout.write('Usage: {0} <input file> <output file> <symbol name> <attribute>\n'.format(os.path.basename(__file__)))
sys.stdout.write('\n')
sys.stdout.write(' <input file> - Path to input file which will be embedded.\n')
sys.stdout.write(' <output file> - Path to output .cpp file which embedded data\n')
sys.stdout.write(' will be written to.\n')
sys.stdout.write(' <symbol name> - Base name of symbol which identifies embedded data.\n')
sys.stdout.write(' <attribute> - "visibility" to add visibility attribute, "no_attr" to add no attribute\n')
class ResEmb:
def __init__(self, inFile, outFile, symName, attrOpt):
self.__symRe = re.compile('^[a-zA-Z_][a-zA-Z0-9_]*$')
self.__lineSize = 20
self.__chunkSize = 131072
self.__symName = symName
self.__attrOpt = attrOpt
self.__openedFiles = list()
self.__attr = ""
self.__validate()
try:
self.__inFile = open(inFile, 'rb')
# Note: open can throw OSError or IOError on Windows
except EnvironmentError as ex:
sys.stderr.write('ERROR: Cannot open input file "{0}".\n {1}.\n'.format(inFile, ex.strerror))
raise RuntimeError("failed to open input file")
self.__openedFiles.append(self.__inFile)
try:
self.__outFile = open(outFile, 'w')
except EnvironmentError as ex:
sys.stderr.write('ERROR: Cannot create/open output file "{0}".\n {1}.\n'.format(outFile, ex.strerror))
raise RuntimeError("failed to open output file")
self.__openedFiles.append(self.__outFile)
def __del__(self):
for openedFile in self.__openedFiles:
openedFile.close()
def __validate(self):
symName = self.__symName
if not self.__symRe.match(symName) or symName.endswith('_size'):
sys.stderr.write('ERROR: Invalid symbol name "{0}".\n'.format(symName))
raise ValueError
attrOpt = self.__attrOpt
if attrOpt == "visibility":
self.__attr = '__attribute__((visibility("default")))'
elif attrOpt == "no_attr":
self.__attr = ""
else:
sys.stderr.write('ERROR: Invalid attribute argument: "{0}".\n'.format(attrOpt))
raise ValueError
def run(self):
self.__outFile.write('// This file is auto generated by resource_embedder, DO NOT EDIT\n\n')
self.__outFile.write('unsigned char {0} {1}[] = {{'.format(self.__attr, self.__symName))
chunkSize = self.__chunkSize
lineSize = self.__lineSize
embeddedSize = 0;
readBytes = self.__inFile.read(chunkSize)
while len(readBytes) > 0:
readSize = len(readBytes)
hexBytes = binascii.hexlify(readBytes)
if embeddedSize > 0:
self.__outFile.write(',')
self.__outFile.write(','.join((('\n 0x' if (embeddedSize + i) % lineSize == 0 else ' 0x') + hexBytes[2*i:2*i+2].decode("utf-8")) for i in range(readSize)))
embeddedSize += readSize
readBytes = self.__inFile.read(chunkSize)
self.__outFile.write('\n };\n\n');
self.__outFile.write('unsigned int {0} {1}_size = {2};\n\n'.format(self.__attr, self.__symName, embeddedSize))
if __name__ == "__main__":
numArgs = 5
if len(sys.argv) < numArgs:
PrintHelp()
exit(0)
for arg in sys.argv:
if arg == '-h' or arg == '--help':
PrintHelp()
exit(0)
if len(sys.argv) > numArgs:
sys.stderr.write('WARNING: Number of arguments is greater than number of supported arguments.\n')
sys.stderr.write(' All additional arguments will be ignored.\n')
try:
RE = ResEmb(sys.argv[1], sys.argv[2], sys.argv[3].strip(),
sys.argv[4].strip())
RE.run()
except Exception as ex:
sys.stderr.write('ERROR: Embeded script error.\n {0}.\n'.format(repr(ex)))
exit(1)
|
