|
#include <Python.h>
|
|
#include <stdbool.h>
|
|
|
|
PyObject *_pModule = NULL;
|
|
bool python_initialized = false;
|
|
|
|
PyObject *py_construct() {
|
|
if (!python_initialized) {
|
|
Py_Initialize();
|
|
python_initialized = true;
|
|
PyObject *sysPath = PySys_GetObject("path");
|
|
PyList_Append(sysPath, PyUnicode_FromString("."));
|
|
// Py_DECREF(sysPath);
|
|
PyObject *pName = PyUnicode_DecodeFSDefault("pgscript");
|
|
_pModule = PyImport_Import(pName);
|
|
Py_DECREF(pName);
|
|
|
|
python_initialized = true;
|
|
}
|
|
return _pModule;
|
|
}
|
|
|
|
void py_destruct() {
|
|
if (!python_initialized)
|
|
return;
|
|
Py_DECREF(_pModule);
|
|
Py_Finalize();
|
|
python_initialized = false;
|
|
}
|
|
|
|
int py_route(int downstream, int upstream) {
|
|
PyObject *pModule = py_construct();
|
|
long upstream_fd = 0;
|
|
if (pModule != NULL) {
|
|
PyObject *pFunc = PyObject_GetAttrString(pModule, "route");
|
|
if (PyCallable_Check(pFunc)) {
|
|
PyObject *pArgs = PyTuple_Pack(2, PyLong_FromLong(downstream),
|
|
PyLong_FromLong(upstream));
|
|
PyObject *pValue = PyObject_CallObject(pFunc, pArgs);
|
|
Py_DECREF(pArgs);
|
|
if (pValue != NULL) {
|
|
upstream_fd = PyLong_AsLong(pValue);
|
|
Py_DECREF(pValue);
|
|
} else {
|
|
PyErr_Print();
|
|
fprintf(stderr, "Call failed\n");
|
|
}
|
|
} else {
|
|
PyErr_Print();
|
|
fprintf(stderr, "Cannot find function 'route'\n");
|
|
}
|
|
|
|
Py_XDECREF(pFunc);
|
|
} else {
|
|
PyErr_Print();
|
|
fprintf(stderr, "Failed to load 'script'\n");
|
|
}
|
|
|
|
return (int)upstream_fd;
|
|
}
|