mirror of
https://github.com/rizinorg/cutter.git
synced 2024-12-21 20:36:09 +00:00
104 lines
2.4 KiB
C++
104 lines
2.4 KiB
C++
#ifdef CUTTER_ENABLE_PYTHON
|
|
|
|
#include <Python.h>
|
|
#include <marshal.h>
|
|
|
|
#include "QtResImporter.h"
|
|
|
|
#include <QFile>
|
|
#include <QDebug>
|
|
|
|
int QtResExists(const char *name, QFile &file)
|
|
{
|
|
QString fname = QString::asprintf(":/python/%s.py", name);
|
|
file.setFileName(fname);
|
|
if (file.exists())
|
|
return 1;
|
|
fname.append('c');
|
|
file.setFileName(fname);
|
|
if (file.exists())
|
|
return 2;
|
|
return 0;
|
|
}
|
|
|
|
PyObject *QtResGetCode(const char *name)
|
|
{
|
|
QFile moduleFile;
|
|
bool isBytecode = false;
|
|
|
|
switch (QtResExists(name, moduleFile)) {
|
|
case 0:
|
|
return nullptr;
|
|
case 2:
|
|
isBytecode = true;
|
|
}
|
|
|
|
moduleFile.open(QIODevice::ReadOnly);
|
|
QByteArray data = moduleFile.readAll();
|
|
moduleFile.close();
|
|
|
|
PyObject *codeObject;
|
|
if (isBytecode) {
|
|
codeObject = PyMarshal_ReadObjectFromString(data.constData() + 12,
|
|
data.size() - 12);
|
|
} else {
|
|
codeObject = Py_CompileString(data.constData(),
|
|
moduleFile.fileName().toLocal8Bit().constData(),
|
|
Py_file_input);
|
|
}
|
|
if (!codeObject) {
|
|
qWarning() << "Couldn't unmarshal/compile " << moduleFile.fileName();
|
|
}
|
|
return codeObject;
|
|
}
|
|
|
|
PyObject *QtResImport(const char *name)
|
|
{
|
|
PyObject *codeObject = QtResGetCode(name);
|
|
if (!codeObject)
|
|
return nullptr;
|
|
PyObject *module = PyImport_ExecCodeModule(name, codeObject);
|
|
Py_DECREF(codeObject);
|
|
return module;
|
|
}
|
|
|
|
PyObject *qtres_exists(PyObject *self, PyObject *args)
|
|
{
|
|
Q_UNUSED(self)
|
|
char *name;
|
|
QFile resFile;
|
|
if (!PyArg_ParseTuple(args, "s", &name))
|
|
return NULL;
|
|
return PyBool_FromLong(QtResExists(name, resFile));
|
|
}
|
|
|
|
PyObject *qtres_get_code(PyObject *self, PyObject *args)
|
|
{
|
|
Q_UNUSED(self)
|
|
char *name;
|
|
if (!PyArg_ParseTuple(args, "s", &name))
|
|
return NULL;
|
|
PyObject *ret = QtResGetCode(name);
|
|
if (ret)
|
|
return ret;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
PyMethodDef QtResMethods[] = {
|
|
{ "exists", qtres_exists, METH_VARARGS, NULL },
|
|
{ "get_code", qtres_get_code, METH_VARARGS, NULL },
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
PyModuleDef QtResModule = {
|
|
PyModuleDef_HEAD_INIT, "_qtres", NULL, -1, QtResMethods,
|
|
NULL, NULL, NULL, NULL
|
|
};
|
|
|
|
PyObject *PyInit_qtres()
|
|
{
|
|
return PyModule_Create(&QtResModule);
|
|
}
|
|
|
|
#endif // CUTTER_ENABLE_PYTHON
|