3rd-party/linux-kernel-module-cheat
1
0
Fork 0
mirror of https://github.com/cirosantilli/linux-kernel-module-cheat.git synced 2026-01-23 02:05:57 +01:00
Code Issues Releases Wiki Activity

embedding python move from python-cheat

Browse Source
This commit is contained in:
Ciro Santilli 六四事件 法轮功
2020-02-24 00:00:01 +00:00
parent b468b5bc95
commit 7c93912413
9 changed files with 997 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
userland/libs/python_embed/Makefile Normal file
View File

@@ -0,0 +1,34 @@
.POSIX:
CC = gcc
CXX = g++
CFLGS = -std=c99
CCFLGS = -ggdb3 -O0 -pedantic-errors -Wall -Wextra
CXXFLGS = -std=c++11
IN_EXT = .c
IN_CXX_EXT = .cpp
OUT_EXT = .out
OUTS = $(addsuffix $(OUT_EXT), $(basename $(wildcard *$(IN_EXT))) $(basename $(wildcard *$(IN_CXX_EXT))))
.PHONY: all clean test
all: $(OUTS)
%$(OUT_EXT): %$(IN_EXT)
$(CC) $(CFLGS) $(CCFLGS) $$(pkg-config --cflags python3) -o '$@' '$<' $$(pkg-config --libs python3)
%$(OUT_EXT): %$(IN_CXX_EXT)
$(CXX) $(CXXFLGS) $(CCFLGS) $$(pkg-config --cflags python3) -o '$@' '$<' $$(pkg-config --libs python3)
clean:
rm -rf *'$(OUT_EXT)' __pycache__ *.pyc
test: all
@\
for t in *"$(OUT_EXT)"; do\
if ! ./"$$t"; then\
echo "ASSERT FAILED: $$t";\
exit 1;\
fi;\
done;\

1
userland/libs/python_embed/README.adoc Normal file
View File

@@ -0,0 +1 @@
https://cirosantilli.com/linux-kernel-module-cheat#embedding-python-in-another-application

24
userland/libs/python_embed/eval.c Normal file
View File

@@ -0,0 +1,24 @@
/* https://cirosantilli.com/linux-kernel-module-cheat#embedding-python-in-another-application
*
* Adapted from: https://docs.python.org/3.7/extending/embedding.html#very-high-level-embedding
*/
#define PY_SSIZE_T_CLEAN
#include <Python.h>
int main(int argc, char *argv[]) {
(void)argc;
wchar_t *program = Py_DecodeLocale(argv[0], NULL);
if (program == NULL) {
fprintf(stderr, "Fatal error: cannot decode argv[0]\n");
exit(1);
}
Py_SetProgramName(program);
Py_Initialize();
PyRun_SimpleString(argv[1]);
if (Py_FinalizeEx() < 0) {
exit(120);
}
PyMem_RawFree(program);
return 0;
}

439
userland/libs/python_embed/pure.c Normal file
View File

@@ -0,0 +1,439 @@
/* Adapted from:
*
* * https://docs.python.org/3.7/extending/embedding.html#pure-embedding
* * https://docs.python.org/3/extending/newtypes_tutorial.html
*
* Full integration: pass function arguments and get results back.
* Called a "pure embedding" by Python.
*
* Also shows how to extend Python with C through the embedding,
* providing a C implementation of a Python functionality.
*/
#define PY_SSIZE_T_CLEAN
#include <Python.h>
#include "structmember.h"
/* Define a function in C to be accessible from the Python code.
*
* ....
* def my_native_method(i, j):
* return i * j
* ....
*/
static PyObject*
my_native_module_my_native_method(PyObject *self, PyObject *args) {
(void)self;
int i, j;
if(!PyArg_ParseTuple(args, "ii", &i, &j))
return NULL;
/* "i" stands for integer. */
return Py_BuildValue("i", i * j);
}
static PyMethodDef my_native_methods[] = {
{
"my_native_method",
my_native_module_my_native_method,
METH_VARARGS,
"always returns the integer 2, boring"
},
/* NULL terminator sentinel denotes the end of the defined methods. */
{NULL}
};
/* Define a class in C to be accessible from the Python code.
*
* AKA how to write 20 lines of Python in 200 lines of C.
*
* ....
* class MyNativeClass:
*
* def __init__(self, first='', last='', number=0):
* self.first = first
* self.last = last
* self.number = number
*
* def name(self):
* return '{} {}'.format(self.first, self.last)
*
* class MyDerivedNativeClass(MyNativeClass):
*
* def __init__(self, first='', last='', number=0, first2='', last2='', number2=0):
* super().__init__(first, last, number, number)
* self.first2 = first2
* self.last2 = last2
* self.number2 = number2
*
* def name(self):
* return '{} {} {} {} 2'.format(self.first, self.last, self.first2, self.last2)
* ....
*/
typedef struct {
PyObject_HEAD
PyObject *first;
PyObject *last;
int number;
} my_native_module_MyNativeClass;
static void
my_native_module_MyNativeClass_dealloc(my_native_module_MyNativeClass *self) {
Py_XDECREF(self->first);
Py_XDECREF(self->last);
Py_TYPE(self)->tp_free((PyObject *) self);
}
static int
my_native_module_MyNativeClass_traverse(my_native_module_MyNativeClass *self, visitproc visit, void *arg)
{
Py_VISIT(self->first);
Py_VISIT(self->last);
return 0;
}
static int
my_native_module_MyNativeClass_clear(my_native_module_MyNativeClass *self)
{
Py_CLEAR(self->first);
Py_CLEAR(self->last);
return 0;
}
static PyObject *
my_native_module_MyNativeClass_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
(void)args;
(void)kwds;
my_native_module_MyNativeClass *self;
self = (my_native_module_MyNativeClass *) type->tp_alloc(type, 0);
if (self != NULL) {
self->first = PyUnicode_FromString("");
if (self->first == NULL) {
Py_DECREF(self);
return NULL;
}
self->last = PyUnicode_FromString("");
if (self->last == NULL) {
Py_DECREF(self);
return NULL;
}
self->number = 0;
}
return (PyObject *) self;
}
static int
my_native_module_MyNativeClass_init(my_native_module_MyNativeClass *self, PyObject *args, PyObject *kwds)
{
static char *kwlist[] = {"first", "last", "number", NULL};
PyObject *first = NULL, *last = NULL, *tmp;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOi", kwlist,
&first, &last, &self->number))
return -1;
if (first) {
tmp = self->first;
Py_INCREF(first);
self->first = first;
Py_XDECREF(tmp);
}
if (last) {
tmp = self->last;
Py_INCREF(last);
self->last = last;
Py_XDECREF(tmp);
}
return 0;
}
static PyMemberDef my_native_module_MyNativeClass_members[] = {
{"first", T_OBJECT_EX, offsetof(my_native_module_MyNativeClass, first), 0,
"first name"},
{"last", T_OBJECT_EX, offsetof(my_native_module_MyNativeClass, last), 0,
"last name"},
{"number", T_INT, offsetof(my_native_module_MyNativeClass, number), 0,
"custom number"},
{NULL}
};
static PyObject *
my_native_module_MyNativeClass_name(my_native_module_MyNativeClass *self, PyObject *Py_UNUSED(ignored))
{
if (self->first == NULL) {
PyErr_SetString(PyExc_AttributeError, "first");
return NULL;
}
if (self->last == NULL) {
PyErr_SetString(PyExc_AttributeError, "last");
return NULL;
}
return PyUnicode_FromFormat("%S %S", self->first, self->last);
}
static PyMethodDef my_native_module_MyNativeClass_methods[] = {
{
"name",
(PyCFunction)my_native_module_MyNativeClass_name,
METH_NOARGS,
"Return the name, combining the first and last name"
},
{NULL}
};
static PyTypeObject my_native_module_MyNativeClassType = {
PyVarObject_HEAD_INIT(NULL, 0)
.tp_name = "my_native_module.MyNativeClass",
.tp_doc = "My native class",
.tp_basicsize = sizeof(my_native_module_MyNativeClass),
.tp_itemsize = 0,
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC,
.tp_new = my_native_module_MyNativeClass_new,
.tp_init = (initproc) my_native_module_MyNativeClass_init,
.tp_dealloc = (destructor) my_native_module_MyNativeClass_dealloc,
.tp_traverse = (traverseproc) my_native_module_MyNativeClass_traverse,
.tp_clear = (inquiry) my_native_module_MyNativeClass_clear,
.tp_members = my_native_module_MyNativeClass_members,
.tp_methods = my_native_module_MyNativeClass_methods,
};
typedef struct {
my_native_module_MyNativeClass base;
PyObject *first2;
PyObject *last2;
int number2;
} my_native_module_MyDerivedNativeClass;
static void
my_native_module_MyDerivedNativeClass_dealloc(my_native_module_MyDerivedNativeClass *self) {
Py_XDECREF(self->first2);
Py_XDECREF(self->last2);
Py_TYPE(self)->tp_free((PyObject *) self);
}
static int
my_native_module_MyDerivedNativeClass_traverse(my_native_module_MyDerivedNativeClass *self, visitproc visit, void *arg)
{
Py_VISIT(self->first2);
Py_VISIT(self->last2);
return 0;
}
static int
my_native_module_MyDerivedNativeClass_clear(my_native_module_MyDerivedNativeClass *self)
{
Py_CLEAR(self->first2);
Py_CLEAR(self->last2);
return 0;
}
static PyObject *
my_native_module_MyDerivedNativeClass_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
(void)args;
(void)kwds;
my_native_module_MyDerivedNativeClass *self;
self = (my_native_module_MyDerivedNativeClass *) type->tp_alloc(type, 0);
if (self != NULL) {
self->first2 = PyUnicode_FromString("");
if (self->first2 == NULL) {
Py_DECREF(self);
return NULL;
}
self->last2 = PyUnicode_FromString("");
if (self->last2 == NULL) {
Py_DECREF(self);
return NULL;
}
self->number2 = 0;
}
return (PyObject *) self;
}
static int
my_native_module_MyDerivedNativeClass_init(my_native_module_MyDerivedNativeClass *self, PyObject *args, PyObject *kwds)
{
static char *kwlist[] = {"first", "last", "number", "first2", "last2", "number2", NULL};
PyObject *first = NULL, *last = NULL, *first2 = NULL, *last2 = NULL, *tmp;
int number;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOiOOi", kwlist,
&first, &last, &number, &first2, &last2, &self->number2))
return -1;
/* Call parent class constructor. TODO learn how to remove the
* "*2" args that are consumed by this constructor.. */
if (my_native_module_MyNativeClassType.tp_init((PyObject *) self, args, kwds) < 0)
return -1;
if (first2) {
tmp = self->first2;
Py_INCREF(first2);
self->first2 = first2;
Py_XDECREF(tmp);
}
if (last2) {
tmp = self->last2;
Py_INCREF(last2);
self->last2 = last2;
Py_XDECREF(tmp);
}
return 0;
}
static PyMemberDef my_native_module_MyDerivedNativeClass_members[] = {
{"first2", T_OBJECT_EX, offsetof(my_native_module_MyDerivedNativeClass, first2), 0,
"first2 name2"},
{"last2", T_OBJECT_EX, offsetof(my_native_module_MyDerivedNativeClass, last2), 0,
"last2 name2"},
{"number2", T_INT, offsetof(my_native_module_MyDerivedNativeClass, number2), 0,
"custom number2"},
{NULL}
};
static PyObject *
my_native_module_MyDerivedNativeClass_name2(my_native_module_MyDerivedNativeClass *self, PyObject *Py_UNUSED(ignored))
{
if (self->base.first == NULL) {
PyErr_SetString(PyExc_AttributeError, "first");
return NULL;
}
if (self->base.last == NULL) {
PyErr_SetString(PyExc_AttributeError, "last");
return NULL;
}
if (self->first2 == NULL) {
PyErr_SetString(PyExc_AttributeError, "first2");
return NULL;
}
if (self->last2 == NULL) {
PyErr_SetString(PyExc_AttributeError, "last2");
return NULL;
}
return PyUnicode_FromFormat("%S %S %S %S", self->base.first, self->base.last, self->first2, self->last2);
}
static PyMethodDef my_native_module_MyDerivedNativeClass_methods[] = {
{
"name2",
(PyCFunction)my_native_module_MyDerivedNativeClass_name2,
METH_NOARGS,
"Return the name2, combining the first2 and last2 name2"
},
{NULL}
};
static PyTypeObject my_native_module_MyDerivedNativeClassType = {
PyVarObject_HEAD_INIT(NULL, 0)
.tp_name = "my_native_module.MyDerivedNativeClass",
.tp_doc = "My native class",
.tp_basicsize = sizeof(my_native_module_MyDerivedNativeClass),
.tp_itemsize = 0,
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC,
.tp_new = my_native_module_MyDerivedNativeClass_new,
.tp_init = (initproc) my_native_module_MyDerivedNativeClass_init,
.tp_dealloc = (destructor) my_native_module_MyDerivedNativeClass_dealloc,
.tp_traverse = (traverseproc) my_native_module_MyDerivedNativeClass_traverse,
.tp_clear = (inquiry) my_native_module_MyDerivedNativeClass_clear,
.tp_members = my_native_module_MyDerivedNativeClass_members,
.tp_methods = my_native_module_MyDerivedNativeClass_methods,
};
static PyModuleDef my_native_module = {
PyModuleDef_HEAD_INIT,
.m_name = "my_native_module",
.m_size = -1,
.m_methods = my_native_methods,
};
PyMODINIT_FUNC
my_native_module_init_func(void) {
PyObject *m;
/* Create the module */
m = PyModule_Create(&my_native_module);
if (m == NULL)
return NULL;
/* Create MyNativeClass */
if (PyType_Ready(&my_native_module_MyNativeClassType) < 0)
return NULL;
Py_INCREF(&my_native_module_MyNativeClassType);
if (PyModule_AddObject(m, "MyNativeClass", (PyObject *) &my_native_module_MyNativeClassType) < 0) {
Py_DECREF(&my_native_module_MyNativeClassType);
Py_DECREF(m);
return NULL;
}
/* Create MyDerivedNativeClass */
/* This is the line that actually specifies the base class in Python. */
my_native_module_MyDerivedNativeClassType.tp_base = &my_native_module_MyNativeClassType;
if (PyType_Ready(&my_native_module_MyDerivedNativeClassType) < 0)
return NULL;
Py_INCREF(&my_native_module_MyDerivedNativeClassType);
if (PyModule_AddObject(m, "MyDerivedNativeClass", (PyObject *) &my_native_module_MyDerivedNativeClassType) < 0) {
Py_DECREF(&my_native_module_MyDerivedNativeClassType);
Py_DECREF(m);
return NULL;
}
return m;
}
int
main(int argc, char *argv[])
{
PyObject *pName, *pModule, *pFunc;
PyObject *pArgs, *pValue;
int i;
if (argc < 3) {
fprintf(stderr,"Usage: ./pure.out pythonfile funcname [args]\n");
return 1;
}
PyImport_AppendInittab("my_native_module", &my_native_module_init_func);
Py_Initialize();
pName = PyUnicode_DecodeFSDefault(argv[1]);
pModule = PyImport_Import(pName);
Py_DECREF(pName);
if (pModule != NULL) {
pFunc = PyObject_GetAttrString(pModule, argv[2]);
/* pFunc is a new reference */
if (pFunc && PyCallable_Check(pFunc)) {
pArgs = PyTuple_New(argc - 3);
for (i = 0; i < argc - 3; ++i) {
pValue = PyLong_FromLong(atoi(argv[i + 3]));
if (!pValue) {
Py_DECREF(pArgs);
Py_DECREF(pModule);
fprintf(stderr, "Cannot convert argument\n");
return 1;
}
/* pValue reference stolen here: */
PyTuple_SetItem(pArgs, i, pValue);
}
pValue = PyObject_CallObject(pFunc, pArgs);
Py_DECREF(pArgs);
if (pValue != NULL) {
printf("%ld\n", PyLong_AsLong(pValue));
Py_DECREF(pValue);
} else {
Py_DECREF(pFunc);
Py_DECREF(pModule);
PyErr_Print();
fprintf(stderr,"Call failed\n");
return 1;
}
} else {
if (PyErr_Occurred())
PyErr_Print();
fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]);
}
Py_XDECREF(pFunc);
Py_DECREF(pModule);
} else {
PyErr_Print();
fprintf(stderr, "Failed to load \"%s\"\n", argv[1]);
return 1;
}
if (Py_FinalizeEx() < 0) {
return 120;
}
return 0;
}

29
userland/libs/python_embed/pure.py Normal file
View File

@@ -0,0 +1,29 @@
#!/usr/bin/env python3
# This is provided by the C file.
# But since we are embedding Python, we don't even need to create a separate
# .so file: it is provided directly through the Python invocation!
import my_native_module
def test_native_method(a, b):
return my_native_module.my_native_method(a, b)
def test_native_class():
# Positional arguments.
my_native_object = my_native_module.MyNativeClass('ab', 'cd', 13)
assert my_native_object.name() == 'ab cd'
# Named arguments.
my_native_object = my_native_module.MyNativeClass(first='ef', last='gh', number=13)
assert my_native_object.name() == 'ef gh'
# Default values and set property.
my_native_object = my_native_module.MyNativeClass()
my_native_object.first = 'ih'
assert my_native_object.name() == 'ih '
# TODO see TODO in pure.c.
#my_derived_native_object = my_native_module.MyDerivedNativeClass('ab', 'cd', 13, 'ef', 'gh', 21)
#print(my_derived_native_object.name2())
return 13

430
userland/libs/python_embed/pure_cpp.cpp Normal file
View File

@@ -0,0 +1,430 @@
/* C++ version, the main goal is to show how to interface with C++ classes. */
#define PY_SSIZE_T_CLEAN
#include <Python.h>
#include "structmember.h"
/* Define a function in C to be accessible from the Python code.
*
* ....
* def my_native_method(i, j):
* return i * j
* ....
*/
static PyObject*
my_native_module_my_native_method(PyObject *self, PyObject *args) {
(void)self;
int i, j;
if(!PyArg_ParseTuple(args, "ii", &i, &j))
return NULL;
/* "i" stands for integer. */
return Py_BuildValue("i", i * j);
}
static PyMethodDef my_native_methods[] = {
{
"my_native_method",
my_native_module_my_native_method,
METH_VARARGS,
"always returns the integer 2, boring"
},
/* NULL terminator sentinel denotes the end of the defined methods. */
{NULL}
};
/* Define a class in C to be accessible from the Python code.
*
* AKA how to write 20 lines of Python in 200 lines of C.
*
* ....
* class MyNativeClass:
*
* def __init__(self, first='', last='', number=0):
* self.first = first
* self.last = last
* self.number = number
*
* def name(self):
* return '{} {}'.format(self.first, self.last)
*
* class MyDerivedNativeClass(MyNativeClass):
*
* def __init__(self, first='', last='', number=0, first2='', last2='', number2=0):
* super().__init__(first, last, number, number)
* self.first2 = first2
* self.last2 = last2
* self.number2 = number2
*
* def name(self):
* return '{} {} {} {} 2'.format(self.first, self.last, self.first2, self.last2)
* ....
*/
typedef struct {
PyObject_HEAD
PyObject *first;
PyObject *last;
int number;
} my_native_module_MyNativeClass;
static void
my_native_module_MyNativeClass_dealloc(my_native_module_MyNativeClass *self) {
Py_XDECREF(self->first);
Py_XDECREF(self->last);
Py_TYPE(self)->tp_free((PyObject *) self);
}
static int
my_native_module_MyNativeClass_traverse(my_native_module_MyNativeClass *self, visitproc visit, void *arg)
{
Py_VISIT(self->first);
Py_VISIT(self->last);
return 0;
}
static int
my_native_module_MyNativeClass_clear(my_native_module_MyNativeClass *self)
{
Py_CLEAR(self->first);
Py_CLEAR(self->last);
return 0;
}
static PyObject *
my_native_module_MyNativeClass_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
(void)args;
(void)kwds;
my_native_module_MyNativeClass *self;
self = (my_native_module_MyNativeClass *) type->tp_alloc(type, 0);
if (self != NULL) {
self->first = PyUnicode_FromString("");
if (self->first == NULL) {
Py_DECREF(self);
return NULL;
}
self->last = PyUnicode_FromString("");
if (self->last == NULL) {
Py_DECREF(self);
return NULL;
}
self->number = 0;
}
return (PyObject *) self;
}
static int
my_native_module_MyNativeClass_init(my_native_module_MyNativeClass *self, PyObject *args, PyObject *kwds)
{
static const char *kwlist[] = {"first", "last", "number", NULL};
PyObject *first = NULL, *last = NULL, *tmp;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOi", (char**)kwlist,
&first, &last, &self->number))
return -1;
if (first) {
tmp = self->first;
Py_INCREF(first);
self->first = first;
Py_XDECREF(tmp);
}
if (last) {
tmp = self->last;
Py_INCREF(last);
self->last = last;
Py_XDECREF(tmp);
}
return 0;
}
static PyMemberDef my_native_module_MyNativeClass_members[] = {
{"first", T_OBJECT_EX, offsetof(my_native_module_MyNativeClass, first), 0,
"first name"},
{"last", T_OBJECT_EX, offsetof(my_native_module_MyNativeClass, last), 0,
"last name"},
{"number", T_INT, offsetof(my_native_module_MyNativeClass, number), 0,
"custom number"},
{NULL}
};
static PyObject *
my_native_module_MyNativeClass_name(my_native_module_MyNativeClass *self, PyObject *Py_UNUSED(ignored))
{
if (self->first == NULL) {
PyErr_SetString(PyExc_AttributeError, "first");
return NULL;
}
if (self->last == NULL) {
PyErr_SetString(PyExc_AttributeError, "last");
return NULL;
}
return PyUnicode_FromFormat("%S %S", self->first, self->last);
}
static PyMethodDef my_native_module_MyNativeClass_methods[] = {
{
"name",
(PyCFunction)my_native_module_MyNativeClass_name,
METH_NOARGS,
"Return the name, combining the first and last name"
},
{NULL}
};
static PyTypeObject my_native_module_MyNativeClassType = {
PyVarObject_HEAD_INIT(NULL, 0)
};
typedef struct {
my_native_module_MyNativeClass base;
PyObject *first2;
PyObject *last2;
int number2;
} my_native_module_MyDerivedNativeClass;
static void
my_native_module_MyDerivedNativeClass_dealloc(my_native_module_MyDerivedNativeClass *self) {
Py_XDECREF(self->first2);
Py_XDECREF(self->last2);
Py_TYPE(self)->tp_free((PyObject *) self);
}
static int
my_native_module_MyDerivedNativeClass_traverse(my_native_module_MyDerivedNativeClass *self, visitproc visit, void *arg)
{
Py_VISIT(self->first2);
Py_VISIT(self->last2);
return 0;
}
static int
my_native_module_MyDerivedNativeClass_clear(my_native_module_MyDerivedNativeClass *self)
{
Py_CLEAR(self->first2);
Py_CLEAR(self->last2);
return 0;
}
static PyObject *
my_native_module_MyDerivedNativeClass_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
(void)args;
(void)kwds;
my_native_module_MyDerivedNativeClass *self;
self = (my_native_module_MyDerivedNativeClass *) type->tp_alloc(type, 0);
if (self != NULL) {
self->first2 = PyUnicode_FromString("");
if (self->first2 == NULL) {
Py_DECREF(self);
return NULL;
}
self->last2 = PyUnicode_FromString("");
if (self->last2 == NULL) {
Py_DECREF(self);
return NULL;
}
self->number2 = 0;
}
return (PyObject *) self;
}
static int
my_native_module_MyDerivedNativeClass_init(my_native_module_MyDerivedNativeClass *self, PyObject *args, PyObject *kwds)
{
static const char *kwlist[] = {"first", "last", "number", "first2", "last2", "number2", NULL};
PyObject *first = NULL, *last = NULL, *first2 = NULL, *last2 = NULL, *tmp;
int number;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OOiOOi", (char**)kwlist,
&first, &last, &number, &first2, &last2, &self->number2))
return -1;
/* Call parent class constructor. TODO learn how to remove the
* "*2" args that are consumed by this constructor.. */
if (my_native_module_MyNativeClassType.tp_init((PyObject *) self, args, kwds) < 0)
return -1;
if (first2) {
tmp = self->first2;
Py_INCREF(first2);
self->first2 = first2;
Py_XDECREF(tmp);
}
if (last2) {
tmp = self->last2;
Py_INCREF(last2);
self->last2 = last2;
Py_XDECREF(tmp);
}
return 0;
}
static PyMemberDef my_native_module_MyDerivedNativeClass_members[] = {
{"first2", T_OBJECT_EX, offsetof(my_native_module_MyDerivedNativeClass, first2), 0,
"first2 name2"},
{"last2", T_OBJECT_EX, offsetof(my_native_module_MyDerivedNativeClass, last2), 0,
"last2 name2"},
{"number2", T_INT, offsetof(my_native_module_MyDerivedNativeClass, number2), 0,
"custom number2"},
{NULL}
};
static PyObject *
my_native_module_MyDerivedNativeClass_name2(my_native_module_MyDerivedNativeClass *self, PyObject *Py_UNUSED(ignored))
{
if (self->base.first == NULL) {
PyErr_SetString(PyExc_AttributeError, "first");
return NULL;
}
if (self->base.last == NULL) {
PyErr_SetString(PyExc_AttributeError, "last");
return NULL;
}
if (self->first2 == NULL) {
PyErr_SetString(PyExc_AttributeError, "first2");
return NULL;
}
if (self->last2 == NULL) {
PyErr_SetString(PyExc_AttributeError, "last2");
return NULL;
}
return PyUnicode_FromFormat("%S %S %S %S", self->base.first, self->base.last, self->first2, self->last2);
}
static PyMethodDef my_native_module_MyDerivedNativeClass_methods[] = {
{
"name2",
(PyCFunction)my_native_module_MyDerivedNativeClass_name2,
METH_NOARGS,
"Return the name2, combining the first2 and last2 name2"
},
{NULL}
};
static PyTypeObject my_native_module_MyDerivedNativeClassType = {
PyVarObject_HEAD_INIT(NULL, 0)
};
static PyModuleDef my_native_module = {
PyModuleDef_HEAD_INIT,
"my_native_module",
"My native module",
-1,
my_native_methods,
};
PyMODINIT_FUNC
my_native_module_init_func(void) {
PyObject *m;
/* Create the module */
m = PyModule_Create(&my_native_module);
if (m == NULL)
return NULL;
/* Create MyNativeClass */
if (PyType_Ready(&my_native_module_MyNativeClassType) < 0)
return NULL;
Py_INCREF(&my_native_module_MyNativeClassType);
if (PyModule_AddObject(m, "MyNativeClass", (PyObject *) &my_native_module_MyNativeClassType) < 0) {
Py_DECREF(&my_native_module_MyNativeClassType);
Py_DECREF(m);
return NULL;
}
/* Create MyDerivedNativeClass */
my_native_module_MyDerivedNativeClassType.tp_base = &my_native_module_MyNativeClassType;
if (PyType_Ready(&my_native_module_MyDerivedNativeClassType) < 0)
return NULL;
Py_INCREF(&my_native_module_MyDerivedNativeClassType);
if (PyModule_AddObject(m, "MyDerivedNativeClass", (PyObject *) &my_native_module_MyDerivedNativeClassType) < 0) {
Py_DECREF(&my_native_module_MyDerivedNativeClassType);
Py_DECREF(m);
return NULL;
}
return m;
}
int
main(int argc, char *argv[])
{
PyObject *pName, *pModule, *pFunc;
PyObject *pArgs, *pValue;
int i;
if (argc < 3) {
fprintf(stderr,"Usage: call pythonfile funcname [args]\n");
return 1;
}
my_native_module_MyNativeClassType.tp_name = "my_native_module.MyNativeClass";
my_native_module_MyNativeClassType.tp_doc = "My native class";
my_native_module_MyNativeClassType.tp_basicsize = sizeof(my_native_module_MyNativeClass);
my_native_module_MyNativeClassType.tp_itemsize = 0;
my_native_module_MyNativeClassType.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC;
my_native_module_MyNativeClassType.tp_new = my_native_module_MyNativeClass_new;
my_native_module_MyNativeClassType.tp_init = (initproc) my_native_module_MyNativeClass_init;
my_native_module_MyNativeClassType.tp_dealloc = (destructor) my_native_module_MyNativeClass_dealloc;
my_native_module_MyNativeClassType.tp_traverse = (traverseproc) my_native_module_MyNativeClass_traverse;
my_native_module_MyNativeClassType.tp_clear = (inquiry) my_native_module_MyNativeClass_clear;
my_native_module_MyNativeClassType.tp_members = my_native_module_MyNativeClass_members;
my_native_module_MyNativeClassType.tp_methods = my_native_module_MyNativeClass_methods;
my_native_module_MyDerivedNativeClassType.tp_name = "my_native_module.MyDerivedNativeClass";
my_native_module_MyDerivedNativeClassType.tp_doc = "My native class";
my_native_module_MyDerivedNativeClassType.tp_basicsize = sizeof(my_native_module_MyDerivedNativeClass);
my_native_module_MyDerivedNativeClassType.tp_itemsize = 0;
my_native_module_MyDerivedNativeClassType.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC;
my_native_module_MyDerivedNativeClassType.tp_new = my_native_module_MyDerivedNativeClass_new;
my_native_module_MyDerivedNativeClassType.tp_init = (initproc) my_native_module_MyDerivedNativeClass_init;
my_native_module_MyDerivedNativeClassType.tp_dealloc = (destructor) my_native_module_MyDerivedNativeClass_dealloc;
my_native_module_MyDerivedNativeClassType.tp_traverse = (traverseproc) my_native_module_MyDerivedNativeClass_traverse;
my_native_module_MyDerivedNativeClassType.tp_clear = (inquiry) my_native_module_MyDerivedNativeClass_clear;
my_native_module_MyDerivedNativeClassType.tp_members = my_native_module_MyDerivedNativeClass_members;
my_native_module_MyDerivedNativeClassType.tp_methods = my_native_module_MyDerivedNativeClass_methods;
PyImport_AppendInittab("my_native_module", &my_native_module_init_func);
Py_Initialize();
pName = PyUnicode_DecodeFSDefault(argv[1]);
pModule = PyImport_Import(pName);
Py_DECREF(pName);
if (pModule != NULL) {
pFunc = PyObject_GetAttrString(pModule, argv[2]);
if (pFunc && PyCallable_Check(pFunc)) {
pArgs = PyTuple_New(argc - 3);
for (i = 0; i < argc - 3; ++i) {
pValue = PyLong_FromLong(atoi(argv[i + 3]));
if (!pValue) {
Py_DECREF(pArgs);
Py_DECREF(pModule);
fprintf(stderr, "Cannot convert argument\n");
return 1;
}
PyTuple_SetItem(pArgs, i, pValue);
}
pValue = PyObject_CallObject(pFunc, pArgs);
Py_DECREF(pArgs);
if (pValue != NULL) {
printf("%ld\n", PyLong_AsLong(pValue));
Py_DECREF(pValue);
} else {
Py_DECREF(pFunc);
Py_DECREF(pModule);
PyErr_Print();
fprintf(stderr,"Call failed\n");
return 1;
}
} else {
if (PyErr_Occurred())
PyErr_Print();
fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]);
}
Py_XDECREF(pFunc);
Py_DECREF(pModule);
} else {
PyErr_Print();
fprintf(stderr, "Failed to load \"%s\"\n", argv[1]);
return 1;
}
if (Py_FinalizeEx() < 0) {
return 120;
}
return 0;
}

7
userland/libs/python_embed/test Executable file
View File

@@ -0,0 +1,7 @@
#!/usr/bin/env bash
set -eu
[ "$(./eval.out 'print(2 ** 3)')" = 8 ]
[ "$(PYTHONPATH="${PYTHONPATH:-}:." ./pure.out pure test_native_method 2 3)" = 6 ]
[ "$(PYTHONPATH="${PYTHONPATH:-}:." ./pure.out pure test_native_class)" = 13 ]
[ "$(PYTHONPATH="${PYTHONPATH:-}:." ./pure_cpp.out pure test_native_method 2 3)" = 6 ]
[ "$(PYTHONPATH="${PYTHONPATH:-}:." ./pure_cpp.out pure test_native_class)" = 13 ]