123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219 |
- //////////////////// MainFunction ////////////////////
- #ifdef __FreeBSD__
- #include <floatingpoint.h>
- #endif
- #if PY_MAJOR_VERSION < 3
- int %(main_method)s(int argc, char** argv) {
- #elif defined(WIN32) || defined(MS_WINDOWS)
- int %(wmain_method)s(int argc, wchar_t **argv) {
- #else
- static int __Pyx_main(int argc, wchar_t **argv) {
- #endif
- /* 754 requires that FP exceptions run in "no stop" mode by default,
- * and until C vendors implement C99's ways to control FP exceptions,
- * Python requires non-stop mode. Alas, some platforms enable FP
- * exceptions by default. Here we disable them.
- */
- #ifdef __FreeBSD__
- fp_except_t m;
- m = fpgetmask();
- fpsetmask(m & ~FP_X_OFL);
- #endif
- if (argc && argv)
- Py_SetProgramName(argv[0]);
- Py_Initialize();
- if (argc && argv)
- PySys_SetArgv(argc, argv);
- { /* init module '%(module_name)s' as '__main__' */
- PyObject* m = NULL;
- %(module_is_main)s = 1;
- #if PY_MAJOR_VERSION < 3
- init%(module_name)s();
- #elif CYTHON_PEP489_MULTI_PHASE_INIT
- m = PyInit_%(module_name)s();
- if (!PyModule_Check(m)) {
- PyModuleDef *mdef = (PyModuleDef *) m;
- PyObject *modname = PyUnicode_FromString("__main__");
- m = NULL;
- if (modname) {
- // FIXME: not currently calling PyModule_FromDefAndSpec() here because we do not have a module spec!
- // FIXME: not currently setting __file__, __path__, __spec__, ...
- m = PyModule_NewObject(modname);
- Py_DECREF(modname);
- if (m) PyModule_ExecDef(m, mdef);
- }
- }
- #else
- m = PyInit_%(module_name)s();
- #endif
- if (PyErr_Occurred()) {
- PyErr_Print(); /* This exits with the right code if SystemExit. */
- #if PY_MAJOR_VERSION < 3
- if (Py_FlushLine()) PyErr_Clear();
- #endif
- return 1;
- }
- Py_XDECREF(m);
- }
- #if PY_VERSION_HEX < 0x03060000
- Py_Finalize();
- #else
- if (Py_FinalizeEx() < 0)
- return 2;
- #endif
- return 0;
- }
- #if PY_MAJOR_VERSION >= 3 && !defined(WIN32) && !defined(MS_WINDOWS)
- #include <locale.h>
- static wchar_t*
- __Pyx_char2wchar(char* arg)
- {
- wchar_t *res;
- #ifdef HAVE_BROKEN_MBSTOWCS
- /* Some platforms have a broken implementation of
- * mbstowcs which does not count the characters that
- * would result from conversion. Use an upper bound.
- */
- size_t argsize = strlen(arg);
- #else
- size_t argsize = mbstowcs(NULL, arg, 0);
- #endif
- size_t count;
- unsigned char *in;
- wchar_t *out;
- #ifdef HAVE_MBRTOWC
- mbstate_t mbs;
- #endif
- if (argsize != (size_t)-1) {
- res = (wchar_t *)malloc((argsize+1)*sizeof(wchar_t));
- if (!res)
- goto oom;
- count = mbstowcs(res, arg, argsize+1);
- if (count != (size_t)-1) {
- wchar_t *tmp;
- /* Only use the result if it contains no
- surrogate characters. */
- for (tmp = res; *tmp != 0 &&
- (*tmp < 0xd800 || *tmp > 0xdfff); tmp++)
- ;
- if (*tmp == 0)
- return res;
- }
- free(res);
- }
- /* Conversion failed. Fall back to escaping with surrogateescape. */
- #ifdef HAVE_MBRTOWC
- /* Try conversion with mbrtwoc (C99), and escape non-decodable bytes. */
- /* Overallocate; as multi-byte characters are in the argument, the
- actual output could use less memory. */
- argsize = strlen(arg) + 1;
- res = (wchar_t *)malloc(argsize*sizeof(wchar_t));
- if (!res) goto oom;
- in = (unsigned char*)arg;
- out = res;
- memset(&mbs, 0, sizeof mbs);
- while (argsize) {
- size_t converted = mbrtowc(out, (char*)in, argsize, &mbs);
- if (converted == 0)
- /* Reached end of string; null char stored. */
- break;
- if (converted == (size_t)-2) {
- /* Incomplete character. This should never happen,
- since we provide everything that we have -
- unless there is a bug in the C library, or I
- misunderstood how mbrtowc works. */
- fprintf(stderr, "unexpected mbrtowc result -2\\n");
- free(res);
- return NULL;
- }
- if (converted == (size_t)-1) {
- /* Conversion error. Escape as UTF-8b, and start over
- in the initial shift state. */
- *out++ = 0xdc00 + *in++;
- argsize--;
- memset(&mbs, 0, sizeof mbs);
- continue;
- }
- if (*out >= 0xd800 && *out <= 0xdfff) {
- /* Surrogate character. Escape the original
- byte sequence with surrogateescape. */
- argsize -= converted;
- while (converted--)
- *out++ = 0xdc00 + *in++;
- continue;
- }
- /* successfully converted some bytes */
- in += converted;
- argsize -= converted;
- out++;
- }
- #else
- /* Cannot use C locale for escaping; manually escape as if charset
- is ASCII (i.e. escape all bytes > 128. This will still roundtrip
- correctly in the locale's charset, which must be an ASCII superset. */
- res = (wchar_t *)malloc((strlen(arg)+1)*sizeof(wchar_t));
- if (!res) goto oom;
- in = (unsigned char*)arg;
- out = res;
- while(*in)
- if(*in < 128)
- *out++ = *in++;
- else
- *out++ = 0xdc00 + *in++;
- *out = 0;
- #endif
- return res;
- oom:
- fprintf(stderr, "out of memory\\n");
- return NULL;
- }
- int
- %(main_method)s(int argc, char **argv)
- {
- if (!argc) {
- return __Pyx_main(0, NULL);
- }
- else {
- int i, res;
- wchar_t **argv_copy = (wchar_t **)malloc(sizeof(wchar_t*)*argc);
- /* We need a second copy, as Python might modify the first one. */
- wchar_t **argv_copy2 = (wchar_t **)malloc(sizeof(wchar_t*)*argc);
- char *oldloc = strdup(setlocale(LC_ALL, NULL));
- if (!argv_copy || !argv_copy2 || !oldloc) {
- fprintf(stderr, "out of memory\\n");
- free(argv_copy);
- free(argv_copy2);
- free(oldloc);
- return 1;
- }
- res = 0;
- setlocale(LC_ALL, "");
- for (i = 0; i < argc; i++) {
- argv_copy2[i] = argv_copy[i] = __Pyx_char2wchar(argv[i]);
- if (!argv_copy[i]) res = 1; /* failure, but continue to simplify cleanup */
- }
- setlocale(LC_ALL, oldloc);
- free(oldloc);
- if (res == 0)
- res = __Pyx_main(argc, argv_copy);
- for (i = 0; i < argc; i++) {
- #if PY_VERSION_HEX < 0x03050000
- free(argv_copy2[i]);
- #else
- PyMem_RawFree(argv_copy2[i]);
- #endif
- }
- free(argv_copy);
- free(argv_copy2);
- return res;
- }
- }
- #endif
|