Tools/freeze/makefreeze.py

   1 import marshal
   2 import bkfile
   3
   4
   5 # Write a file containing frozen code for the modules in the dictionary.
   6
   7 header = """
   8 #include "Python.h"
   9
  10 static struct _frozen _PyImport_FrozenModules[] = {
  11 """
  12 trailer = """\
  13     {0, 0, 0} /* sentinel */
  14 };
  15 """
  16
  17 # if __debug__ == 0 (i.e. -O option given), set Py_OptimizeFlag in frozen app.
  18 default_entry_point = """
  19 int
  20 main(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
  32 def 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
  80 def 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: `s`[1:-1], str.split('"')))))