mirror of https://gitlab.com/ita1024/waf.git
151 lines
3.5 KiB
Python
151 lines
3.5 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
# Tool to embed file into objects
|
|
|
|
__author__ = __maintainer__ = "Jérôme Carretero <cJ-waf@zougloub.eu>"
|
|
__copyright__ = "Jérôme Carretero, 2014"
|
|
|
|
"""
|
|
|
|
This tool allows to embed file contents in object files (.o).
|
|
It is not exactly portable, and the file contents are reachable
|
|
using various non-portable fashions.
|
|
The goal here is to provide a functional interface to the embedding
|
|
of file data in objects.
|
|
See the ``playground/embedded_resources`` example for an example.
|
|
|
|
Usage::
|
|
|
|
bld(
|
|
name='pipeline',
|
|
# ^ Reference this in use="..." for things using the generated code
|
|
features='file_to_object',
|
|
source='some.file',
|
|
# ^ Name of the file to embed in binary section.
|
|
)
|
|
|
|
Known issues:
|
|
|
|
- Currently only handles elf files with GNU ld.
|
|
|
|
- Destination is named like source, with extension renamed to .o
|
|
eg. some.file -> some.o
|
|
|
|
"""
|
|
|
|
import os, binascii
|
|
|
|
from waflib import Task, Utils, TaskGen, Errors
|
|
|
|
|
|
def filename_c_escape(x):
|
|
return x.replace("\\", "\\\\")
|
|
|
|
class file_to_object_s(Task.Task):
|
|
color = 'CYAN'
|
|
dep_vars = ('DEST_CPU', 'DEST_BINFMT')
|
|
|
|
def run(self):
|
|
name = []
|
|
for i, x in enumerate(self.inputs[0].name):
|
|
if x.isalnum():
|
|
name.append(x)
|
|
else:
|
|
name.append('_')
|
|
file = self.inputs[0].abspath()
|
|
size = os.path.getsize(file)
|
|
if self.env.DEST_CPU in ('x86_64', 'ia', 'aarch64'):
|
|
unit = 'quad'
|
|
align = 8
|
|
elif self.env.DEST_CPU in ('x86','arm', 'thumb', 'm68k'):
|
|
unit = 'long'
|
|
align = 4
|
|
else:
|
|
raise Errors.WafError("Unsupported DEST_CPU, please report bug!")
|
|
|
|
file = filename_c_escape(file)
|
|
name = "_binary_" + "".join(name)
|
|
rodata = ".section .rodata"
|
|
if self.env.DEST_BINFMT == "mac-o":
|
|
name = "_" + name
|
|
rodata = ".section __TEXT,__const"
|
|
|
|
with open(self.outputs[0].abspath(), 'w') as f:
|
|
f.write(\
|
|
"""
|
|
.global %(name)s_start
|
|
.global %(name)s_end
|
|
.global %(name)s_size
|
|
%(rodata)s
|
|
%(name)s_start:
|
|
.incbin "%(file)s"
|
|
%(name)s_end:
|
|
.align %(align)d
|
|
%(name)s_size:
|
|
.%(unit)s 0x%(size)x
|
|
""" % locals())
|
|
|
|
class file_to_object_c(Task.Task):
|
|
color = 'CYAN'
|
|
def run(self):
|
|
name = []
|
|
for i, x in enumerate(self.inputs[0].name):
|
|
if x.isalnum():
|
|
name.append(x)
|
|
else:
|
|
name.append('_')
|
|
file = self.inputs[0].abspath()
|
|
size = os.path.getsize(file)
|
|
|
|
name = "_binary_" + "".join(name)
|
|
|
|
data = []
|
|
data = self.inputs[0].read()
|
|
data = binascii.hexlify(data)
|
|
data = [ ("0x%s" % (data[i:i+2])) for i in range(0, len(data), 2) ]
|
|
data = ",\n ".join(data)
|
|
|
|
with open(self.outputs[0].abspath(), 'w') as f:
|
|
f.write(\
|
|
"""
|
|
char const %(name)s[] = {
|
|
%(data)s
|
|
};
|
|
unsigned long %(name)s_size = %(size)dL;
|
|
char const * %(name)s_start = %(name)s;
|
|
char const * %(name)s_end = &%(name)s[%(size)d];
|
|
""" % locals())
|
|
with open(self.outputs[0].abspath(), 'w') as f:
|
|
f.write(\
|
|
"""
|
|
unsigned long %(name)s_size = %(size)dL;
|
|
char const %(name)s_start[] = {
|
|
%(data)s
|
|
};
|
|
char const %(name)s_end[] = {
|
|
};
|
|
""" % locals())
|
|
|
|
@TaskGen.feature('file_to_object')
|
|
@TaskGen.before_method('process_source')
|
|
def tg_file_to_object(self):
|
|
bld = self.bld
|
|
src = self.to_nodes(self.source)
|
|
assert len(src) == 1
|
|
src = src[0]
|
|
if bld.env.F2O_METHOD == ["asm"]:
|
|
tgt = src.parent.find_or_declare(src.name + '.f2o.s')
|
|
task = self.create_task('file_to_object_s',
|
|
src, tgt, cwd=src.parent.abspath())
|
|
self.source = [tgt]
|
|
else:
|
|
tgt = src.parent.find_or_declare(src.name + '.f2o.c')
|
|
task = self.create_task('file_to_object_c',
|
|
src, tgt, cwd=src.parent.abspath())
|
|
self.source = [tgt]
|
|
|
|
def configure(conf):
|
|
conf.load('gas')
|
|
conf.env.F2O_METHOD = ["asm"]
|
|
|