1 #!/usr/bin/env python 2 """ 3 Create a C file for embedding one or more Cython source files. 4 Requires Cython 0.11.2 (or perhaps newer). 5 6 See Demos/freeze/README.txt for more details. 7 """ 8 9 import optparse 10 from os.path import splitext, basename 11 12 usage= '%prog [-o outfile] [-p] module [module ...]' 13 description = 'Create a C file for embedding Cython modules.' 14 p = optparse.OptionParser(usage=usage, description=description) 15 p.add_option('-o', '--output', metavar='FILE', 16 help='write output to FILE instead of standard output') 17 p.add_option('-p', '--pymain', action='store_true', default=False, 18 help='do not automatically run the first module as __main__') 19 20 options, args = p.parse_args() 21 22 if len(args) < 1: 23 p.print_help() 24 p.exit(1) 25 26 if options.output: 27 import sys 28 old_stdout = sys.stdout 29 sys.stdout = open(options.output, 'w') 30 31 modules = [basename(splitext(x)[0]).replace('.', '_') for x in args] 32 33 print """\ 34 #include <Python.h> 35 #include <locale.h> 36 #include <stdio.h> 37 #include <stdlib.h> 38 39 #ifdef __FreeBSD__ 40 #include <floatingpoint.h> 41 #endif 42 43 #if PY_MAJOR_VERSION < 3 44 # define MODINIT(name) init ## name 45 #else 46 # define MODINIT(name) PyInit_ ## name 47 #endif 48 """ 49 50 for name in modules: 51 print "PyMODINIT_FUNC MODINIT(%s) (void);" % name 52 53 print """ 54 static struct _inittab inittab[] = {""" 55 56 for name in modules: 57 print ' {"%(name)s", MODINIT(%(name)s)},' % {'name' : name} 58 59 print """ {NULL, NULL} 60 }; 61 """, 62 63 if not options.pymain: 64 print "\nextern int __pyx_module_is_main_%s;" % modules[0] 65 66 print """ 67 #if PY_MAJOR_VERSION < 3 68 int main(int argc, char** argv) { 69 #elif defined(WIN32) || defined(MS_WINDOWS) 70 int wmain(int argc, wchar_t **argv) { 71 #else 72 static int python_main(int argc, wchar_t **argv) { 73 #endif 74 """, 75 if not options.pymain: 76 print """\ 77 PyObject *m = NULL; 78 int r = 0; 79 """, 80 print """\ 81 /* 754 requires that FP exceptions run in "no stop" mode by default, 82 * and until C vendors implement C99's ways to control FP exceptions, 83 * Python requires non-stop mode. Alas, some platforms enable FP 84 * exceptions by default. Here we disable them. 85 */ 86 #ifdef __FreeBSD__ 87 fp_except_t m; 88 89 m = fpgetmask(); 90 fpsetmask(m & ~FP_X_OFL); 91 #endif 92 if (PyImport_ExtendInittab(inittab)) { 93 fprintf(stderr, "No memory\\n"); 94 exit(1); 95 } 96 """, 97 if options.pymain: 98 print """\ 99 return Py_Main(argc, argv); 100 } 101 """ 102 else: 103 print """\ 104 Py_SetProgramName(argv[0]); 105 Py_Initialize(); 106 PySys_SetArgv(argc, argv); 107 __pyx_module_is_main_%(main)s = 1; 108 m = PyImport_ImportModule(inittab[0].name); 109 if (!m) { 110 r = 1; 111 PyErr_Print(); /* This exits with the right code if SystemExit. */ 112 #if PY_MAJOR_VERSION < 3 113 if (Py_FlushLine()) 114 PyErr_Clear(); 115 #endif 116 } 117 Py_XDECREF(m); 118 Py_Finalize(); 119 return r; 120 } 121 """ % {'main' : modules[0]}, 122 123 print r""" 124 #if PY_MAJOR_VERSION >= 3 && !defined(WIN32) && !defined(MS_WINDOWS) 125 static wchar_t* 126 char2wchar(char* arg) 127 { 128 wchar_t *res; 129 #ifdef HAVE_BROKEN_MBSTOWCS 130 /* Some platforms have a broken implementation of 131 * mbstowcs which does not count the characters that 132 * would result from conversion. Use an upper bound. 133 */ 134 size_t argsize = strlen(arg); 135 #else 136 size_t argsize = mbstowcs(NULL, arg, 0); 137 #endif 138 size_t count; 139 unsigned char *in; 140 wchar_t *out; 141 #ifdef HAVE_MBRTOWC 142 mbstate_t mbs; 143 #endif 144 if (argsize != (size_t)-1) { 145 res = (wchar_t *)malloc((argsize+1)*sizeof(wchar_t)); 146 if (!res) 147 goto oom; 148 count = mbstowcs(res, arg, argsize+1); 149 if (count != (size_t)-1) { 150 wchar_t *tmp; 151 /* Only use the result if it contains no 152 surrogate characters. */ 153 for (tmp = res; *tmp != 0 && 154 (*tmp < 0xd800 || *tmp > 0xdfff); tmp++) 155 ; 156 if (*tmp == 0) 157 return res; 158 } 159 free(res); 160 } 161 /* Conversion failed. Fall back to escaping with surrogateescape. */ 162 #ifdef HAVE_MBRTOWC 163 /* Try conversion with mbrtwoc (C99), and escape non-decodable bytes. */ 164 165 /* Overallocate; as multi-byte characters are in the argument, the 166 actual output could use less memory. */ 167 argsize = strlen(arg) + 1; 168 res = malloc(argsize*sizeof(wchar_t)); 169 if (!res) goto oom; 170 in = (unsigned char*)arg; 171 out = res; 172 memset(&mbs, 0, sizeof mbs); 173 while (argsize) { 174 size_t converted = mbrtowc(out, (char*)in, argsize, &mbs); 175 if (converted == 0) 176 /* Reached end of string; null char stored. */ 177 break; 178 if (converted == (size_t)-2) { 179 /* Incomplete character. This should never happen, 180 since we provide everything that we have - 181 unless there is a bug in the C library, or I 182 misunderstood how mbrtowc works. */ 183 fprintf(stderr, "unexpected mbrtowc result -2\n"); 184 return NULL; 185 } 186 if (converted == (size_t)-1) { 187 /* Conversion error. Escape as UTF-8b, and start over 188 in the initial shift state. */ 189 *out++ = 0xdc00 + *in++; 190 argsize--; 191 memset(&mbs, 0, sizeof mbs); 192 continue; 193 } 194 if (*out >= 0xd800 && *out <= 0xdfff) { 195 /* Surrogate character. Escape the original 196 byte sequence with surrogateescape. */ 197 argsize -= converted; 198 while (converted--) 199 *out++ = 0xdc00 + *in++; 200 continue; 201 } 202 /* successfully converted some bytes */ 203 in += converted; 204 argsize -= converted; 205 out++; 206 } 207 #else 208 /* Cannot use C locale for escaping; manually escape as if charset 209 is ASCII (i.e. escape all bytes > 128. This will still roundtrip 210 correctly in the locale's charset, which must be an ASCII superset. */ 211 res = malloc((strlen(arg)+1)*sizeof(wchar_t)); 212 if (!res) goto oom; 213 in = (unsigned char*)arg; 214 out = res; 215 while(*in) 216 if(*in < 128) 217 *out++ = *in++; 218 else 219 *out++ = 0xdc00 + *in++; 220 *out = 0; 221 #endif 222 return res; 223 oom: 224 fprintf(stderr, "out of memory\n"); 225 return NULL; 226 } 227 228 int 229 main(int argc, char **argv) 230 { 231 wchar_t **argv_copy = (wchar_t **)malloc(sizeof(wchar_t*)*argc); 232 /* We need a second copies, as Python might modify the first one. */ 233 wchar_t **argv_copy2 = (wchar_t **)malloc(sizeof(wchar_t*)*argc); 234 int i, res; 235 char *oldloc; 236 if (!argv_copy || !argv_copy2) { 237 fprintf(stderr, "out of memory\n"); 238 return 1; 239 } 240 oldloc = strdup(setlocale(LC_ALL, NULL)); 241 setlocale(LC_ALL, ""); 242 for (i = 0; i < argc; i++) { 243 argv_copy2[i] = argv_copy[i] = char2wchar(argv[i]); 244 if (!argv_copy[i]) 245 return 1; 246 } 247 setlocale(LC_ALL, oldloc); 248 free(oldloc); 249 res = python_main(argc, argv_copy); 250 for (i = 0; i < argc; i++) { 251 free(argv_copy2[i]); 252 } 253 free(argv_copy); 254 free(argv_copy2); 255 return res; 256 } 257 #endif""" 258
