#include "uwsgi_python.h"

extern struct uwsgi_server uwsgi;

struct uwsgi_python up;

#include <glob.h>

extern PyTypeObject uwsgi_InputType;

void uwsgi_opt_pythonpath(char *opt, char *value, void *foobar) {

int i;

glob_t g;

if (glob(value, GLOB_MARK, NULL, &g)) {

uwsgi_string_new_list(&up.python_path, value);

}

else {

for (i = 0; i < (int) g.gl_pathc; i++) {

uwsgi_string_new_list(&up.python_path, g.gl_pathv[i]);

}

}

}

void uwsgi_opt_pyshell(char *opt, char *value, void *foobar) {

uwsgi.honour_stdin = 1;

if (value) {

up.pyshell = value;

}

else {

up.pyshell = "";

}

if (!strcmp("pyshell-oneshot", opt)) {

up.pyshell_oneshot = 1;

}

}

void uwsgi_opt_pyrun(char *opt, char *value, void *foobar) {

uwsgi.honour_stdin = 1;

uwsgi.command_mode = 1;

up.pyrun = value;

}

void uwsgi_opt_pyver(char *opt, char *foo, void *bar) {

const char *version = Py_GetVersion();

const char *space = strchr(version, ' ');

if (space) {

fprintf(stdout, "%.*s\n", (int) (space-version), version);

}

else {

fprintf(stdout, "%s\n", version);

}

exit(0);

}

int uwsgi_python_init(void);

void uwsgi_python_preinit_apps(void);

static void uwsgi_early_python(char *opt, char *foo, void *bar) {

static int early_initialized = 0;

if (early_initialized) return;

early_initialized = 1;

uwsgi_python_init();

uwsgi_python_preinit_apps();

}

static void uwsgi_early_python_import(char *opt, char *module, void *bar) {

uwsgi_early_python(opt, NULL, NULL);

if (strchr(module, '/') || uwsgi_endswith(module, ".py")) {

uwsgi_pyimport_by_filename(uwsgi_pythonize(module), module);

}

else {

if (PyImport_ImportModule(module) == NULL) {

PyErr_Print();

}

}

}

void uwsgi_opt_ini_paste(char *opt, char *value, void *foobar) {

uwsgi_opt_load_ini(opt, value, NULL);

if (value[0] != '/') {

up.paste = uwsgi_concat4("config:", uwsgi.cwd, "/", value);

}

else {

up.paste = uwsgi_concat2("config:", value);

}

if (!strcmp("ini-paste-logged", opt)) {

up.paste_logger = 1;

}

}

struct uwsgi_option uwsgi_python_options[] = {

{"wsgi-file", required_argument, 0, "load .wsgi file", uwsgi_opt_set_str, &up.file_config, 0},

{"file", required_argument, 0, "load .wsgi file", uwsgi_opt_set_str, &up.file_config, 0},

{"eval", required_argument, 0, "eval python code", uwsgi_opt_set_str, &up.eval, 0},

{"module", required_argument,'w', "load a WSGI module", uwsgi_opt_set_str, &up.wsgi_config, 0},

{"wsgi", required_argument, 'w', "load a WSGI module", uwsgi_opt_set_str, &up.wsgi_config, 0},

{"callable", required_argument, 0, "set default WSGI callable name", uwsgi_opt_set_str, &up.callable, 0},

{"test", required_argument, 'J', "test a module import", uwsgi_opt_set_str, &up.test_module, 0},

{"home", required_argument, 'H', "set PYTHONHOME/virtualenv", uwsgi_opt_set_str, &up.home, 0},

{"virtualenv", required_argument, 'H', "set PYTHONHOME/virtualenv", uwsgi_opt_set_str, &up.home, 0},

{"venv", required_argument, 'H', "set PYTHONHOME/virtualenv", uwsgi_opt_set_str, &up.home, 0},

{"pyhome", required_argument, 'H', "set PYTHONHOME/virtualenv", uwsgi_opt_set_str, &up.home, 0},

{"py-programname", required_argument, 0, "set python program name", uwsgi_opt_set_str, &up.programname, 0},

{"py-program-name", required_argument, 0, "set python program name", uwsgi_opt_set_str, &up.programname, 0},

{"pythonpath", required_argument, 0, "add directory (or glob) to pythonpath", uwsgi_opt_pythonpath, NULL, 0},

{"python-path", required_argument, 0, "add directory (or glob) to pythonpath", uwsgi_opt_pythonpath, NULL, 0},

{"pp", required_argument, 0, "add directory (or glob) to pythonpath", uwsgi_opt_pythonpath, NULL, 0},

{"pymodule-alias", required_argument, 0, "add a python alias module", uwsgi_opt_add_string_list, &up.pymodule_alias, 0},

{"post-pymodule-alias", required_argument, 0, "add a python module alias after uwsgi module initialization", uwsgi_opt_add_string_list, &up.post_pymodule_alias, 0},

{"import", required_argument, 0, "import a python module", uwsgi_opt_add_string_list, &up.import_list, 0},

{"pyimport", required_argument, 0, "import a python module", uwsgi_opt_add_string_list, &up.import_list, 0},

{"py-import", required_argument, 0, "import a python module", uwsgi_opt_add_string_list, &up.import_list, 0},

{"python-import", required_argument, 0, "import a python module", uwsgi_opt_add_string_list, &up.import_list, 0},

{"shared-import", required_argument, 0, "import a python module in all of the processes", uwsgi_opt_add_string_list, &up.shared_import_list, 0},

{"shared-pyimport", required_argument, 0, "import a python module in all of the processes", uwsgi_opt_add_string_list, &up.shared_import_list, 0},

{"shared-py-import", required_argument, 0, "import a python module in all of the processes", uwsgi_opt_add_string_list, &up.shared_import_list, 0},

{"shared-python-import", required_argument, 0, "import a python module in all of the processes", uwsgi_opt_add_string_list, &up.shared_import_list, 0},

{"spooler-import", required_argument, 0, "import a python module in the spooler", uwsgi_opt_add_string_list, &up.spooler_import_list},

{"spooler-pyimport", required_argument, 0, "import a python module in the spooler", uwsgi_opt_add_string_list, &up.spooler_import_list},

{"spooler-py-import", required_argument, 0, "import a python module in the spooler", uwsgi_opt_add_string_list, &up.spooler_import_list},

{"spooler-python-import", required_argument, 0, "import a python module in the spooler", uwsgi_opt_add_string_list, &up.spooler_import_list},

{"pyargv", required_argument, 0, "manually set sys.argv", uwsgi_opt_set_str, &up.argv, 0},

{"optimize", required_argument, 'O', "set python optimization level", uwsgi_opt_set_int, &up.optimize, 0},

{"pecan", required_argument, 0, "load a pecan config file", uwsgi_opt_set_str, &up.pecan, 0},

{"paste", required_argument, 0, "load a paste.deploy config file", uwsgi_opt_set_str, &up.paste, 0},

{"paste-logger", no_argument, 0, "enable paste fileConfig logger", uwsgi_opt_true, &up.paste_logger, 0},

{"paste-name", required_argument, 0, "specify the name of the paste section", uwsgi_opt_set_str, &up.paste_name, 0},

{"web3", required_argument, 0, "load a web3 app", uwsgi_opt_set_str, &up.web3, 0},

{"pump", required_argument, 0, "load a pump app", uwsgi_opt_set_str, &up.pump, 0},

{"wsgi-lite", required_argument, 0, "load a wsgi-lite app", uwsgi_opt_set_str, &up.wsgi_lite, 0},

{"ini-paste", required_argument, 0, "load a paste.deploy config file containing uwsgi section", uwsgi_opt_ini_paste, NULL, UWSGI_OPT_IMMEDIATE},

{"ini-paste-logged", required_argument, 0, "load a paste.deploy config file containing uwsgi section (load loggers too)", uwsgi_opt_ini_paste, NULL, UWSGI_OPT_IMMEDIATE},

{"reload-os-env", no_argument, 0, "force reload of os.environ at each request", uwsgi_opt_true, &up.reload_os_env, 0},

#ifndef __CYGWIN__

{"no-site", no_argument, 0, "do not import site module", uwsgi_opt_true, &Py_NoSiteFlag, 0},

{"pyshell", optional_argument, 0, "run an interactive python shell in the uWSGI environment", uwsgi_opt_pyshell, NULL, 0},

{"pyshell-oneshot", optional_argument, 0, "run an interactive python shell in the uWSGI environment (one-shot variant)", uwsgi_opt_pyshell, NULL, 0},

{"python", required_argument, 0, "run a python script in the uWSGI environment", uwsgi_opt_pyrun, NULL, 0},

{"py", required_argument, 0, "run a python script in the uWSGI environment", uwsgi_opt_pyrun, NULL, 0},

{"pyrun", required_argument, 0, "run a python script in the uWSGI environment", uwsgi_opt_pyrun, NULL, 0},

{"py-tracebacker", required_argument, 0, "enable the uWSGI python tracebacker", uwsgi_opt_set_str, &up.tracebacker, UWSGI_OPT_THREADS|UWSGI_OPT_MASTER},

{"py-auto-reload", required_argument, 0, "monitor python modules mtime to trigger reload (use only in development)", uwsgi_opt_set_int, &up.auto_reload, UWSGI_OPT_THREADS|UWSGI_OPT_MASTER},

{"py-autoreload", required_argument, 0, "monitor python modules mtime to trigger reload (use only in development)", uwsgi_opt_set_int, &up.auto_reload, UWSGI_OPT_THREADS|UWSGI_OPT_MASTER},

{"python-auto-reload", required_argument, 0, "monitor python modules mtime to trigger reload (use only in development)", uwsgi_opt_set_int, &up.auto_reload, UWSGI_OPT_THREADS|UWSGI_OPT_MASTER},

{"python-autoreload", required_argument, 0, "monitor python modules mtime to trigger reload (use only in development)", uwsgi_opt_set_int, &up.auto_reload, UWSGI_OPT_THREADS|UWSGI_OPT_MASTER},

{"py-auto-reload-ignore", required_argument, 0, "ignore the specified module during auto-reload scan (can be specified multiple times)", uwsgi_opt_add_string_list, &up.auto_reload_ignore, UWSGI_OPT_THREADS|UWSGI_OPT_MASTER},

{"wsgi-env-behaviour", required_argument, 0, "set the strategy for allocating/deallocating the WSGI env", uwsgi_opt_set_str, &up.wsgi_env_behaviour, 0},

{"wsgi-env-behavior", required_argument, 0, "set the strategy for allocating/deallocating the WSGI env", uwsgi_opt_set_str, &up.wsgi_env_behaviour, 0},

{"start_response-nodelay", no_argument, 0, "send WSGI http headers as soon as possible (PEP violation)", uwsgi_opt_true, &up.start_response_nodelay, 0},

{"wsgi-strict", no_argument, 0, "try to be fully PEP compliant disabling optimizations", uwsgi_opt_true, &up.wsgi_strict, 0},

{"wsgi-accept-buffer", no_argument, 0, "accept CPython buffer-compliant objects as WSGI response in addition to string/bytes", uwsgi_opt_true, &up.wsgi_accept_buffer, 0},

{"wsgi-accept-buffers", no_argument, 0, "accept CPython buffer-compliant objects as WSGI response in addition to string/bytes", uwsgi_opt_true, &up.wsgi_accept_buffer, 0},

{"wsgi-disable-file-wrapper", no_argument, 0, "disable wsgi.file_wrapper feature", uwsgi_opt_true, &up.wsgi_disable_file_wrapper, 0},

{"python-version", no_argument, 0, "report python version", uwsgi_opt_pyver, NULL, UWSGI_OPT_IMMEDIATE},

{"python-raw", required_argument, 0, "load a python file for managing raw requests", uwsgi_opt_set_str, &up.raw, 0},

#if defined(PYTHREE) || defined(Py_TPFLAGS_HAVE_NEWBUFFER)

{"py-sharedarea", required_argument, 0, "create a sharedarea from a python bytearray object of the specified size", uwsgi_opt_add_string_list, &up.sharedarea, 0},

{"py-call-osafterfork", no_argument, 0, "enable child processes running cpython to trap OS signals", uwsgi_opt_true, &up.call_osafterfork, 0},

{"early-python", no_argument, 0, "load the python VM as soon as possible (useful for the fork server)", uwsgi_early_python, NULL, UWSGI_OPT_IMMEDIATE},

{"early-pyimport", required_argument, 0, "import a python module in the early phase", uwsgi_early_python_import, NULL, UWSGI_OPT_IMMEDIATE},

{"early-python-import", required_argument, 0, "import a python module in the early phase", uwsgi_early_python_import, NULL, UWSGI_OPT_IMMEDIATE},

{"early-pythonpath", required_argument, 0, "add directory (or glob) to pythonpath (immediate version)", uwsgi_opt_pythonpath, NULL, UWSGI_OPT_IMMEDIATE},

{"early-python-path", required_argument, 0, "add directory (or glob) to pythonpath (immediate version)", uwsgi_opt_pythonpath, NULL, UWSGI_OPT_IMMEDIATE},

{"python-worker-override", required_argument, 0, "override worker with the specified python script", uwsgi_opt_set_str, &up.worker_override, 0},

{"wsgi-manage-chunked-input", no_argument, 0, "manage chunked input via the wsgi.input_terminated extension", uwsgi_opt_true, &up.wsgi_manage_chunked_input, 0},

{"py-master-check-signals", no_argument, 0, "enable python signal handlers in master", uwsgi_opt_true, &up.master_check_signals, 0},

{0, 0, 0, 0, 0, 0, 0},

};

void uwsgi_python_pthread_prepare(void) {

if (!up.is_dynamically_loading_an_app)

pthread_mutex_lock(&up.lock_pyloaders);

}

void uwsgi_python_pthread_parent(void) {

if (!up.is_dynamically_loading_an_app)

pthread_mutex_unlock(&up.lock_pyloaders);

}

void uwsgi_python_pthread_child(void) {

if (!up.is_dynamically_loading_an_app)

pthread_mutex_init(&up.lock_pyloaders, NULL);

}

PyMethodDef uwsgi_spit_method[] = { {"uwsgi_spit", py_uwsgi_spit, METH_VARARGS, ""} };

PyMethodDef uwsgi_write_method[] = { {"uwsgi_write", py_uwsgi_write, METH_VARARGS, ""} };

int uwsgi_python_init() {

char *pyversion = strchr(Py_GetVersion(), '\n');

if (!pyversion) {

uwsgi_log_initial("Python version: %s\n", Py_GetVersion());

}

else {

uwsgi_log_initial("Python version: %.*s %s\n", pyversion-Py_GetVersion(), Py_GetVersion(), Py_GetCompiler()+1);

}

if (Py_IsInitialized()) {

uwsgi_log("--- Python VM already initialized ---\n");

PyGILState_Ensure();

goto ready;

}

if (up.home != NULL) {

if (!uwsgi_is_dir(up.home)) {

uwsgi_log("Python Home is not a directory: %s\n", up.home);

exit(1);

}

#ifdef PYTHREE

// check for PEP 405 virtualenv (starting from python 3.3)

char *pep405_env = uwsgi_concat2(up.home, "/pyvenv.cfg");

if (uwsgi_file_exists(pep405_env)) {

uwsgi_log("PEP 405 virtualenv detected: %s\n", up.home);

free(pep405_env);

goto pep405;

}

free(pep405_env);

// build the PYTHONHOME wchar path

wchar_t *wpyhome;

size_t len = strlen(up.home) + 1;

wpyhome = uwsgi_calloc(sizeof(wchar_t) * len );

if (!wpyhome) {

uwsgi_error("malloc()");

exit(1);

}

mbstowcs(wpyhome, up.home, len);

Py_SetPythonHome(wpyhome);

// do not free this memory !!!

//free(wpyhome);

pep405:

Py_SetPythonHome(up.home);

uwsgi_log("Set PythonHome to %s\n", up.home);

}

char *program_name = up.programname;

if (!program_name) {

program_name = uwsgi.binary_path;

}

#ifdef PYTHREE

if (!up.programname) {

if (up.home) {

program_name = uwsgi_concat2(up.home, "/bin/python");

}

}

wchar_t *pname = uwsgi_calloc(sizeof(wchar_t) * (strlen(program_name)+1));

mbstowcs(pname, program_name, strlen(program_name)+1);

Py_SetProgramName(pname);

Py_SetProgramName(program_name);

Py_OptimizeFlag = up.optimize;

Py_Initialize();

ready:

if (!uwsgi.has_threads) {

uwsgi_log_initial("*** Python threads support is disabled. You can enable it with --enable-threads ***\n");

}

up.wsgi_spitout = PyCFunction_New(uwsgi_spit_method, NULL);

up.wsgi_writeout = PyCFunction_New(uwsgi_write_method, NULL);

up.main_thread = PyThreadState_Get();

// by default set a fake GIL (little impact on performance)

up.gil_get = gil_fake_get;

up.gil_release = gil_fake_release;

up.swap_ts = simple_swap_ts;

up.reset_ts = simple_reset_ts;

#if defined(PYTHREE) || defined(Py_TPFLAGS_HAVE_NEWBUFFER)

struct uwsgi_string_list *usl = NULL;

uwsgi_foreach(usl, up.sharedarea) {

uint64_t len = uwsgi_n64(usl->value);

PyObject *obj = PyByteArray_FromStringAndSize(NULL, len);

char *storage = PyByteArray_AsString(obj);

Py_INCREF(obj);

struct uwsgi_sharedarea *sa = uwsgi_sharedarea_init_ptr(storage, len);

sa->obj = obj;

}

uwsgi_log_initial("Python main interpreter initialized at %p\n", up.main_thread);

return 1;

}

void uwsgi_python_reset_random_seed() {

PyObject *random_module, *random_dict, *random_seed;

// reinitialize the random seed (thanks Jonas Borgström)

random_module = PyImport_ImportModule("random");

if (random_module) {

random_dict = PyModule_GetDict(random_module);

if (random_dict) {

random_seed = PyDict_GetItemString(random_dict, "seed");

if (random_seed) {

PyObject *random_args = PyTuple_New(1);

// pass no args

PyTuple_SetItem(random_args, 0, Py_None);

PyEval_CallObject(random_seed, random_args);

if (PyErr_Occurred()) {

PyErr_Print();

}

}

}

}

}

void uwsgi_python_atexit() {

if (uwsgi.mywid == 0) goto realstuff;

// if hijacked do not run atexit hooks

if (uwsgi.workers[uwsgi.mywid].hijacked)

return;

// if busy do not run atexit hooks

if (uwsgi_worker_is_busy(uwsgi.mywid))

return;

// managing atexit in async mode is a real pain...skip it for now

if (uwsgi.async > 0)

return;

realstuff:

// this time we use this higher level function

// as this code can be executed in a signal handler

#ifdef __GNU_kFreeBSD__

return;

if (!Py_IsInitialized()) {

return;

}

// always call it

PyGILState_Ensure();

// no need to worry about freeing memory

PyObject *uwsgi_dict = get_uwsgi_pydict("uwsgi");

if (uwsgi_dict) {

PyObject *ae = PyDict_GetItemString(uwsgi_dict, "atexit");

if (ae) {

python_call(ae, PyTuple_New(0), 0, NULL);

}

}

// this part is a 1:1 copy of mod_wsgi 3.x

// it is required to fix some atexit bug with python 3

// and to shutdown useless threads complaints

PyObject *module = PyImport_ImportModule("atexit");

Py_XDECREF(module);

if (uwsgi.has_threads) {

if (!PyImport_AddModule("dummy_threading"))

PyErr_Clear();

}

// For the reasons explained in

// https://github.com/unbit/uwsgi/issues/1384, tearing down

// the interpreter can be very expensive.

if (uwsgi.skip_atexit_teardown)

return;

Py_Finalize();

}

void uwsgi_python_post_fork() {

if (uwsgi.i_am_a_spooler) {

UWSGI_GET_GIL

}

// reset python signal flags so child processes can trap signals

if (up.call_osafterfork) {

#ifdef HAS_NOT_PyOS_AfterFork_Child

PyOS_AfterFork();

PyOS_AfterFork_Child();

}

uwsgi_python_reset_random_seed();

// call the post_fork_hook

PyObject *uwsgi_dict = get_uwsgi_pydict("uwsgi");

if (uwsgi_dict) {

PyObject *pfh = PyDict_GetItemString(uwsgi_dict, "post_fork_hook");

if (pfh) {

python_call(pfh, PyTuple_New(0), 0, NULL);

}

}

PyErr_Clear();

if (uwsgi.mywid > 0) {

uwsgi_python_set_thread_name(0);

if (up.auto_reload) {

// spawn the reloader thread

pthread_t par_tid;

pthread_create(&par_tid, NULL, uwsgi_python_autoreloader_thread, up.main_thread->interp);

}

if (up.tracebacker) {

// spawn the tracebacker thread

pthread_t ptb_tid;

pthread_create(&ptb_tid, NULL, uwsgi_python_tracebacker_thread, NULL);

}

}

}

PyObject *uwsgi_pyimport_by_filename(char *name, char *filename) {

FILE *pyfile;

struct _node *py_file_node = NULL;

PyObject *py_compiled_node, *py_file_module;

int is_a_package = 0;

struct stat pystat;

char *real_filename = filename;

if (!uwsgi_check_scheme(filename)) {

pyfile = fopen(filename, "r");

if (!pyfile) {

uwsgi_log("failed to open python file %s\n", filename);

return NULL;

}

if (fstat(fileno(pyfile), &pystat)) {

fclose(pyfile);

uwsgi_error("fstat()");

return NULL;

}

if (S_ISDIR(pystat.st_mode)) {

is_a_package = 1;

fclose(pyfile);

real_filename = uwsgi_concat2(filename, "/__init__.py");

pyfile = fopen(real_filename, "r");

if (!pyfile) {

uwsgi_error_open(real_filename);

free(real_filename);

return NULL;

}

}

py_file_node = PyParser_SimpleParseFile(pyfile, real_filename, Py_file_input);

if (!py_file_node) {

PyErr_Print();

uwsgi_log("failed to parse file %s\n", real_filename);

if (is_a_package)

free(real_filename);

fclose(pyfile);

return NULL;

}

fclose(pyfile);

}

else {

size_t pycontent_size = 0;

char *pycontent = uwsgi_open_and_read(filename, &pycontent_size, 1, NULL);

if (pycontent) {

py_file_node = PyParser_SimpleParseString(pycontent, Py_file_input);

if (!py_file_node) {

PyErr_Print();

uwsgi_log("failed to parse url %s\n", real_filename);

return NULL;

}

}

}

py_compiled_node = (PyObject *) PyNode_Compile(py_file_node, real_filename);

if (!py_compiled_node) {

PyErr_Print();

uwsgi_log("failed to compile python file %s\n", real_filename);

return NULL;

}

if (is_a_package) {

py_file_module = PyImport_AddModule(name);

if (py_file_module) {

PyModule_AddObject(py_file_module, "__path__", Py_BuildValue("[O]", PyString_FromString(filename)));

}

free(real_filename);

}

py_file_module = PyImport_ExecCodeModule(name, py_compiled_node);

if (!py_file_module) {

PyErr_Print();

return NULL;

}

Py_DECREF(py_compiled_node);

return py_file_module;

}

void init_uwsgi_vars() {

PyObject *pysys, *pysys_dict, *pypath;

PyObject *modules = PyImport_GetModuleDict();

PyObject *tmp_module;

/* add cwd to pythonpath */

pysys = PyImport_ImportModule("sys");

if (!pysys) {

PyErr_Print();

exit(1);

}

pysys_dict = PyModule_GetDict(pysys);

#ifdef PYTHREE

// In python3, stdout / stderr was changed to be buffered (a bug according

// to many):

// - https://bugs.python.org/issue13597

// - https://bugs.python.org/issue13601

// We'll fix this by manually restore the unbuffered behaviour.

// In the case of a tty, this fix breaks readline support in interactive

// debuggers so we'll only do this in the non-tty case.

if (!Py_FdIsInteractive(stdin, NULL)) {

#ifdef HAS_NO_ERRORS_IN_PyFile_FromFd

PyObject *new_stdprint = PyFile_FromFd(2, NULL, "w", _IOLBF, NULL, NULL, 0);

PyObject *new_stdprint = PyFile_FromFd(2, NULL, "w", _IOLBF, NULL, NULL, NULL, 0);

PyDict_SetItemString(pysys_dict, "stdout", new_stdprint);

PyDict_SetItemString(pysys_dict, "__stdout__", new_stdprint);

PyDict_SetItemString(pysys_dict, "stderr", new_stdprint);

PyDict_SetItemString(pysys_dict, "__stderr__", new_stdprint);

}

pypath = PyDict_GetItemString(pysys_dict, "path");

if (!pypath) {

PyErr_Print();

exit(1);

}

if (PyList_Insert(pypath, 0, UWSGI_PYFROMSTRING(".")) != 0) {

PyErr_Print();

}

struct uwsgi_string_list *uppp = up.python_path;

while(uppp) {

if (PyList_Insert(pypath, 0, UWSGI_PYFROMSTRING(uppp->value)) != 0) {

PyErr_Print();

}

else {

uwsgi_log("added %s to pythonpath.\n", uppp->value);

}

uppp = uppp->next;

}

struct uwsgi_string_list *uppma = up.pymodule_alias;

while(uppma) {

// split key=value

char *value = strchr(uppma->value, '=');

if (!value) {

uwsgi_log("invalid pymodule-alias syntax\n");

goto next;

}

value[0] = 0;

if (!strchr(value + 1, '/')) {

// this is a standard pymodule

tmp_module = PyImport_ImportModule(value + 1);

if (!tmp_module) {

PyErr_Print();

exit(1);

}

PyDict_SetItemString(modules, uppma->value, tmp_module);

}

else {

// this is a filepath that need to be mapped

tmp_module = uwsgi_pyimport_by_filename(uppma->value, value + 1);

if (!tmp_module) {

PyErr_Print();

exit(1);

}

}

uwsgi_log("mapped virtual pymodule \"%s\" to real pymodule \"%s\"\n", uppma->value, value + 1);

// reset original value

value[0] = '=';

next:

uppma = uppma->next;

}

}

PyDoc_STRVAR(uwsgi_py_doc, "uWSGI api module.");

#ifdef PYTHREE

static PyModuleDef uwsgi_module3 = {

PyModuleDef_HEAD_INIT,

"uwsgi",

uwsgi_py_doc,

-1,

NULL,

};

PyObject *init_uwsgi3(void) {

return PyModule_Create(&uwsgi_module3);

}

void init_uwsgi_embedded_module() {

PyObject *new_uwsgi_module, *zero;

int i;

if (PyType_Ready(&uwsgi_InputType) < 0) {

PyErr_Print();

uwsgi_log("could not initialize the uwsgi python module\n");

exit(1);

}

/* initialize for stats */

up.workers_tuple = PyTuple_New(uwsgi.numproc);

for (i = 0; i < uwsgi.numproc; i++) {

zero = PyDict_New();

Py_INCREF(zero);

PyTuple_SetItem(up.workers_tuple, i, zero);

}

#ifdef PYTHREE

PyImport_AppendInittab("uwsgi", init_uwsgi3);

new_uwsgi_module = PyImport_AddModule("uwsgi");

new_uwsgi_module = Py_InitModule3("uwsgi", NULL, uwsgi_py_doc);

if (new_uwsgi_module == NULL) {

uwsgi_log("could not initialize the uwsgi python module\n");

exit(1);

}

Py_INCREF((PyObject *) &uwsgi_InputType);

up.embedded_dict = PyModule_GetDict(new_uwsgi_module);

if (!up.embedded_dict) {

uwsgi_log("could not get uwsgi module __dict__\n");

exit(1);

}

// just for safety

Py_INCREF(up.embedded_dict);

if (PyDict_SetItemString(up.embedded_dict, "version", PyString_FromString(UWSGI_VERSION))) {

PyErr_Print();

exit(1);

}

PyObject *uwsgi_py_version_info = PyTuple_New(5);

PyTuple_SetItem(uwsgi_py_version_info, 0, PyInt_FromLong(UWSGI_VERSION_BASE));

PyTuple_SetItem(uwsgi_py_version_info, 1, PyInt_FromLong(UWSGI_VERSION_MAJOR));

PyTuple_SetItem(uwsgi_py_version_info, 2, PyInt_FromLong(UWSGI_VERSION_MINOR));

PyTuple_SetItem(uwsgi_py_version_info, 3, PyInt_FromLong(UWSGI_VERSION_REVISION));

PyTuple_SetItem(uwsgi_py_version_info, 4, PyString_FromString(UWSGI_VERSION_CUSTOM));

if (PyDict_SetItemString(up.embedded_dict, "version_info", uwsgi_py_version_info)) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "hostname", PyString_FromStringAndSize(uwsgi.hostname, uwsgi.hostname_len))) {

PyErr_Print();

exit(1);

}

if (uwsgi.mode) {

if (PyDict_SetItemString(up.embedded_dict, "mode", PyString_FromString(uwsgi.mode))) {

PyErr_Print();

exit(1);

}

}

if (uwsgi.pidfile) {

if (PyDict_SetItemString(up.embedded_dict, "pidfile", PyString_FromString(uwsgi.pidfile))) {

PyErr_Print();

exit(1);

}

}

if (uwsgi.spoolers) {

int sc = 0;

struct uwsgi_spooler *uspool = uwsgi.spoolers;

while(uspool) { sc++; uspool = uspool->next;}

PyObject *py_spooler_tuple = PyTuple_New(sc);

uspool = uwsgi.spoolers;

sc = 0;

while(uspool) {

PyTuple_SetItem(py_spooler_tuple, sc, PyString_FromString(uspool->dir));

sc++;

uspool = uspool->next;

}

if (PyDict_SetItemString(up.embedded_dict, "spoolers", py_spooler_tuple)) {

PyErr_Print();

exit(1);

}

}

if (PyDict_SetItemString(up.embedded_dict, "SPOOL_RETRY", PyInt_FromLong(-1))) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "SPOOL_OK", PyInt_FromLong(-2))) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "SPOOL_IGNORE", PyInt_FromLong(0))) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "numproc", PyInt_FromLong(uwsgi.numproc))) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "has_threads", PyInt_FromLong(uwsgi.has_threads))) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "cores", PyInt_FromLong(uwsgi.cores))) {

PyErr_Print();

exit(1);

}

if (uwsgi.loop) {

if (PyDict_SetItemString(up.embedded_dict, "loop", PyString_FromString(uwsgi.loop))) {

PyErr_Print();

exit(1);

}

}

else {

PyDict_SetItemString(up.embedded_dict, "loop", Py_None);

}

PyObject *py_opt_dict = PyDict_New();

for (i = 0; i < uwsgi.exported_opts_cnt; i++) {

if (PyDict_Contains(py_opt_dict, PyString_FromString(uwsgi.exported_opts[i]->key))) {

PyObject *py_opt_item = PyDict_GetItemString(py_opt_dict, uwsgi.exported_opts[i]->key);

if (PyList_Check(py_opt_item)) {

if (uwsgi.exported_opts[i]->value == NULL) {

PyList_Append(py_opt_item, Py_True);

}

else {

PyList_Append(py_opt_item, PyString_FromString(uwsgi.exported_opts[i]->value));

}

}

else {

PyObject *py_opt_list = PyList_New(0);

PyList_Append(py_opt_list, py_opt_item);

if (uwsgi.exported_opts[i]->value == NULL) {

PyList_Append(py_opt_list, Py_True);

}

else {

PyList_Append(py_opt_list, PyString_FromString(uwsgi.exported_opts[i]->value));

}

PyDict_SetItemString(py_opt_dict, uwsgi.exported_opts[i]->key, py_opt_list);

}

}

else {

if (uwsgi.exported_opts[i]->value == NULL) {

PyDict_SetItemString(py_opt_dict, uwsgi.exported_opts[i]->key, Py_True);

}

else {

PyDict_SetItemString(py_opt_dict, uwsgi.exported_opts[i]->key, PyString_FromString(uwsgi.exported_opts[i]->value));

}

}

}

if (PyDict_SetItemString(up.embedded_dict, "opt", py_opt_dict)) {

PyErr_Print();

exit(1);

}

PyObject *py_sockets_list = PyList_New(0);

struct uwsgi_socket *uwsgi_sock = uwsgi.sockets;

while(uwsgi_sock) {

if (uwsgi_sock->bound) {

PyList_Append(py_sockets_list, PyInt_FromLong(uwsgi_sock->fd));

}

uwsgi_sock = uwsgi_sock->next;

}

if (PyDict_SetItemString(up.embedded_dict, "sockets", py_sockets_list)) {

PyErr_Print();

exit(1);

}

PyObject *py_magic_table = PyDict_New();

uint8_t mtk;

for (i = 0; i <= 0xff; i++) {

// a bit of magic :P

mtk = i;

if (uwsgi.magic_table[i]) {

if (uwsgi.magic_table[i][0] != 0) {

PyDict_SetItem(py_magic_table, PyString_FromStringAndSize((char *) &mtk, 1), PyString_FromString(uwsgi.magic_table[i]));

}

}

}

if (PyDict_SetItemString(up.embedded_dict, "magic_table", py_magic_table)) {

PyErr_Print();

exit(1);

}

#ifdef UNBIT

if (PyDict_SetItemString(up.embedded_dict, "unbit", Py_True)) {

if (PyDict_SetItemString(up.embedded_dict, "unbit", Py_None)) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "buffer_size", PyInt_FromLong(uwsgi.buffer_size))) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "started_on", PyInt_FromLong(uwsgi.start_tv.tv_sec))) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "start_response", up.wsgi_spitout)) {

PyErr_Print();

exit(1);

}

if (PyDict_SetItemString(up.embedded_dict, "applications", Py_None)) {

PyErr_Print();

exit(1);

}

if (uwsgi.is_a_reload) {

if (PyDict_SetItemString(up.embedded_dict, "is_a_reload", Py_True)) {

PyErr_Print();

exit(1);

}

}

else {

if (PyDict_SetItemString(up.embedded_dict, "is_a_reload", Py_False)) {

PyErr_Print();

exit(1);

}

}

init_uwsgi_module_advanced(new_uwsgi_module);

if (uwsgi.spoolers) {

init_uwsgi_module_spooler(new_uwsgi_module);

}

if (uwsgi.sharedareas) {

init_uwsgi_module_sharedarea(new_uwsgi_module);

}

init_uwsgi_module_cache(new_uwsgi_module);

if (uwsgi.queue_size > 0) {

init_uwsgi_module_queue(new_uwsgi_module);

}

if (uwsgi.snmp) {

init_uwsgi_module_snmp(new_uwsgi_module);

}

if (up.extension) {

up.extension();

}

}

int uwsgi_python_magic(char *mountpoint, char *lazy) {

char *qc = strchr(lazy, ':');

if (qc) {

qc[0] = 0;

up.callable = qc + 1;

}

if (!strcmp(lazy + strlen(lazy) - 3, ".py")) {

up.file_config = lazy;

return 1;

}

else if (!strcmp(lazy + strlen(lazy) - 5, ".wsgi")) {

up.file_config = lazy;

return 1;

}

else if (qc && strchr(lazy, '.')) {

up.wsgi_config = lazy;

return 1;

}

// reset lazy

if (qc) {

qc[0] = ':';

}

return 0;

}

int uwsgi_python_mount_app(char *mountpoint, char *app) {

int id;

if (strchr(app, ':') || uwsgi_endswith(app, ".py") || uwsgi_endswith(app, ".wsgi")) {

uwsgi.wsgi_req->appid = mountpoint;

uwsgi.wsgi_req->appid_len = strlen(mountpoint);

// lazy ?

if (uwsgi.mywid > 0) UWSGI_GET_GIL

if (uwsgi.single_interpreter) {

id = init_uwsgi_app(LOADER_MOUNT, app, uwsgi.wsgi_req, up.main_thread, PYTHON_APP_TYPE_WSGI);

}

else {

id = init_uwsgi_app(LOADER_MOUNT, app, uwsgi.wsgi_req, NULL, PYTHON_APP_TYPE_WSGI);

}