80 lines
2.2 KiB
C
80 lines
2.2 KiB
C
__attribute__((constructor))
|
||
void preload_init() {
|
||
Py_Initialize();
|
||
PyRun_SimpleString("print('hello from c'); import extra.dsp.hook");
|
||
}
|
||
|
||
#define _GNU_SOURCE // Must be defined before any includes for RTLD_NEXT
|
||
#include <stdio.h>
|
||
#include <dlfcn.h>
|
||
#include <Python.h> // Include Python header
|
||
//#include <sys/ioctl.h>
|
||
|
||
// Define the original ioctl function pointer
|
||
static int (*real_ioctl)(int fd, unsigned long request, void *arg) = NULL;
|
||
|
||
// Our custom ioctl hook
|
||
int ioctl(int fd, unsigned long request, void *arg) {
|
||
// Initialize the real ioctl function pointer on first call
|
||
if (!real_ioctl) {
|
||
real_ioctl = dlsym(RTLD_NEXT, "ioctl");
|
||
if (!real_ioctl) {
|
||
fprintf(stderr, "Error: Could not find real ioctl\n");
|
||
return -1;
|
||
}
|
||
}
|
||
|
||
// Log the call
|
||
//printf("Hooked ioctl: tid=%d fd=%d, request=0x%lx, arg=%p\n", gettid(), fd, request, arg);
|
||
|
||
// Call a Python function from extra.dsp.hook
|
||
PyObject *pName, *pModule, *pFunc, *pArgs, *pValue;
|
||
PyGILState_STATE gstate;
|
||
|
||
// Ensure the GIL is held (required for Python calls in multi-threaded apps)
|
||
//gstate = PyGILState_Ensure();
|
||
|
||
// Import the module
|
||
pName = PyUnicode_FromString("extra.dsp.hook");
|
||
pModule = PyImport_Import(pName);
|
||
Py_DECREF(pName);
|
||
|
||
// Call the original ioctl
|
||
int ret = real_ioctl(fd, request, arg);
|
||
|
||
if (pModule != NULL) {
|
||
// Get the function (assume it’s called "handle_ioctl")
|
||
pFunc = PyObject_GetAttrString(pModule, "handle_ioctl");
|
||
|
||
if (pFunc && PyCallable_Check(pFunc)) {
|
||
// Create arguments tuple (fd, request, arg, ret)
|
||
pArgs = PyTuple_Pack(4,
|
||
PyLong_FromLong(fd),
|
||
PyLong_FromUnsignedLong(request),
|
||
PyLong_FromVoidPtr(arg),
|
||
PyLong_FromLong(ret));
|
||
pValue = PyObject_CallObject(pFunc, pArgs);
|
||
Py_DECREF(pArgs);
|
||
|
||
if (pValue != NULL) {
|
||
Py_DECREF(pValue);
|
||
} else {
|
||
PyErr_Print(); // Print Python error if call fails
|
||
}
|
||
Py_DECREF(pFunc);
|
||
} else {
|
||
if (PyErr_Occurred()) PyErr_Print();
|
||
fprintf(stderr, "Cannot find function 'handle_ioctl'\n");
|
||
}
|
||
Py_DECREF(pModule);
|
||
} else {
|
||
PyErr_Print();
|
||
fprintf(stderr, "Failed to load 'extra.dsp.hook'\n");
|
||
}
|
||
|
||
// Release the GIL
|
||
//PyGILState_Release(gstate);
|
||
return ret;
|
||
}
|
||
|