static PyObject *
run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
PyObject *locals, PyCompilerFlags *flags)
{
PyCodeObject *co;
PyObject *v;
long magic;
long PyImport_GetMagicNumber(void);
magic = PyMarshal_ReadLongFromFile(fp);
if (magic != PyImport_GetMagicNumber()) {
PyErr_SetString(PyExc_RuntimeError,
"Bad magic number in .pyc file");
return NULL;
}
(void) PyMarshal_ReadLongFromFile(fp);
v = PyMarshal_ReadLastObjectFromFile(fp);
fclose(fp);
if (v == NULL || !PyCode_Check(v)) {
Py_XDECREF(v);
PyErr_SetString(PyExc_RuntimeError,
"Bad code object in .pyc file");
return NULL;
}
co = (PyCodeObject *)v;
v = PyEval_EvalCode(co, globals, locals);
if (v && flags)
flags->cf_flags |= (co->co_flags & PyCF_MASK);
Py_DECREF(co);
return v;
}
static PyObject *
run_pyc_file(FILE *fp, const char *filename, PyObject *globals,
PyObject *locals, PyCompilerFlags *flags)
{
PyCodeObject *co;
PyObject *v;
long magic;
long PyImport_GetMagicNumber(void);
magic = PyMarshal_ReadLongFromFile(fp);
if (magic != PyImport_GetMagicNumber()) {
if (!PyErr_Occurred())
PyErr_SetString(PyExc_RuntimeError,
"Bad magic number in .pyc file");
goto error;
}
/* Skip the rest of the header. */
(void) PyMarshal_ReadLongFromFile(fp);
(void) PyMarshal_ReadLongFromFile(fp);
(void) PyMarshal_ReadLongFromFile(fp);
if (PyErr_Occurred()) {
goto error;
}
v = PyMarshal_ReadLastObjectFromFile(fp);
if (v == NULL || !PyCode_Check(v)) {
Py_XDECREF(v);
PyErr_SetString(PyExc_RuntimeError,
"Bad code object in .pyc file");
goto error;
}
fclose(fp);
co = (PyCodeObject *)v;
v = run_eval_code_obj(co, globals, locals);
if (v && flags)
flags->cf_flags |= (co->co_flags & PyCF_MASK);
Py_DECREF(co);
return v;
error:
fclose(fp);
return NULL;
}
/**
* This function loads a given encrypted .pyc file and returns decrypted code object.
*/
static PyObject *dedrop_decrypt(PyObject *dummy, PyObject *args)
{
// Compare this to "run_pyc_file" method from upstream
const char *path;
if (!PyArg_ParseTuple(args, "s", &path))
return NULL;
long magic;
// printf("%s ->\n", path);
FILE *f = fopen(path, "rb");
if (!f) {
printf(" failed to open, errno=%d\n", errno);
return NULL;
}
/* TODO: verify magic number */
// magic = PyMarshal_ReadLongFromFile(f);
magic = PyMarshal_ReadLongFromFile(f);
printf(" magic: 0x%08lx\n", magic);
/* Skip .py modification timestamp */
(void) PyMarshal_ReadLongFromFile(f);
/* NOTE: Skip "size parameter" added in Python 3.3 */
(void) PyMarshal_ReadLongFromFile(f);
// XXX
(void) PyMarshal_ReadLongFromFile(f);
/* Here we call function from dropbox binary, yeah! */
PyObject *obj = PyMarshal_ReadLastObjectFromFile(f);
fclose(f);
/* Ownership is passed to the caller, right? */
return obj;
}
static PyObject*
__import__compiled(FILE *f, char *name, char *path, char *as, PyObject *local, PyObject *global)
{
char *module_name = as ? as : name;
PyCodeObject *co;
PyObject *m;
if(name == NULL)
return NULL;
//比较文件的魔数
if(PyMarshal_ReadLongFromFile(f) != PYC_MAGIC)
{
PyErr_Format(PyExc_ImportError, "Bad magic number of %s", name);
return NULL;
}
//读掉时间信息
(void*)PyMarshal_ReadLongFromFile(f);
//创建PyCodeObject
co = (PyCodeObject*)PyMarshal_ReadLastObjectFromFile(f);
if(co == NULL)
{
PyErr_Format(PyExc_ImportError, "Cannot create code object from module %s", name);
return NULL;
}
if(!PyCode_Check(co))
{
PyErr_Format(PyExc_ImportError, "Non-code object in module %s", name);
Py_DECREF(co);
return NULL;
}
/*创建模块*/
m = create_module(name, (PyObject*)co);
if(m == NULL)
{
Ps_Log("create_module failed\n", Ps_LOG_WARING);
return NULL;
}
Py_DECREF(co);
/*将模块导入命名空间*/
if(local && PyDict_Check(local))
{
Py_INCREF(m);
#ifdef IMPORT_DEBUG
Ps_LogFormat("The module name is %s\n", Ps_LOG_NORMAL, module_name);
Ps_LogObject(local, Ps_LOG_WARING);
Ps_LogObject(m, Ps_LOG_WARING);
int ret =
#endif
PyDict_SetItemString(local, module_name, m);
#ifdef IMPORT_DEBUG
if(ret == 0)
Ps_LogFormat("ret is %d, Import module %s successfully\n", Ps_LOG_NORMAL, ret, module_name);
else
Ps_LogFormat("ret is %d, Import module %s failed\n", Ps_LOG_NORMAL, ret, module_name);
#endif
}
else
{
PyObject *info = PyString_FromFormat("Import module %s failed", name);
if(!info)
{
PyErr_SetString(PyExc_ImportError, "Import module failed");
}
else
PyErr_SetObject(PyExc_ImportError, info);
return NULL;
}
return m;
}
