/Tools/freeze/makefreeze.py

 1import marshal
 2import bkfile
 3
 4
 5# Write a file containing frozen code for the modules in the dictionary.
 6
 7header = """
 8#include "Python.h"
 9
10static struct _frozen _PyImport_FrozenModules[] = {
11"""
12trailer = """\
13    {0, 0, 0} /* sentinel */
14};
15"""
16
17# if __debug__ == 0 (i.e. -O option given), set Py_OptimizeFlag in frozen app.
18default_entry_point = """
19int
20main(int argc, char **argv)
21{
22        extern int Py_FrozenMain(int, char **);
23""" + ((not __debug__ and """
24        Py_OptimizeFlag++;
25""") or "")  + """
26        PyImport_FrozenModules = _PyImport_FrozenModules;
27        return Py_FrozenMain(argc, argv);
28}
29
30"""
31
32def makefreeze(base, dict, debug=0, entry_point=None, fail_import=()):
33    if entry_point is None: entry_point = default_entry_point
34    done = []
35    files = []
36    mods = dict.keys()
37    mods.sort()
38    for mod in mods:
39        m = dict[mod]
40        mangled = "__".join(mod.split("."))
41        if m.__code__:
42            file = 'M_' + mangled + '.c'
43            outfp = bkfile.open(base + file, 'w')
44            files.append(file)
45            if debug:
46                print "freezing", mod, "..."
47            str = marshal.dumps(m.__code__)
48            size = len(str)
49            if m.__path__:
50                # Indicate package by negative size
51                size = -size
52            done.append((mod, mangled, size))
53            writecode(outfp, mangled, str)
54            outfp.close()
55    if debug:
56        print "generating table of frozen modules"
57    outfp = bkfile.open(base + 'frozen.c', 'w')
58    for mod, mangled, size in done:
59        outfp.write('extern unsigned char M_%s[];\n' % mangled)
60    outfp.write(header)
61    for mod, mangled, size in done:
62        outfp.write('\t{"%s", M_%s, %d},\n' % (mod, mangled, size))
63    outfp.write('\n')
64    # The following modules have a NULL code pointer, indicating
65    # that the prozen program should not search for them on the host
66    # system. Importing them will *always* raise an ImportError.
67    # The zero value size is never used.
68    for mod in fail_import:
69        outfp.write('\t{"%s", NULL, 0},\n' % (mod,))
70    outfp.write(trailer)
71    outfp.write(entry_point)
72    outfp.close()
73    return files
74
75
76
77# Write a C initializer for a module containing the frozen python code.
78# The array is called M_<mod>.
79
80def writecode(outfp, mod, str):
81    outfp.write('unsigned char M_%s[] = {' % mod)
82    for i in range(0, len(str), 16):
83        outfp.write('\n\t')
84        for c in str[i:i+16]:
85            outfp.write('%d,' % ord(c))
86    outfp.write('\n};\n')
87
88## def writecode(outfp, mod, str):
89##     outfp.write('unsigned char M_%s[%d] = "%s";\n' % (mod, len(str),
90##     '\\"'.join(map(lambda s: repr(s)[1:-1], str.split('"')))))