Getting our feet wet with Python's C API
by Luca Di Sera
Approaching the shore
The Pythons’C API is something that I was always interested in. Nonetheless, I never found an exercise for which it would have made sense to try my hand at it. The toy exercise I’m currently working on, that expands on the last assignment of the work of genius that is the From Nand To Tetris Course, provided me with a decent option to go at it.
I will write about that exercise in a future post. For this one, instead, I’d like to provide a resource for some of the basics related to writing Python’s extension modules.
I must say, to my disdain, that to actually start and write something with the C API was a little bit more difficult than I initially thought. The first impression leaves you with some question as there is a lot to learn. I could not find that many good resources on it and the best resource there is, the Python/C API Reference Manual and the Extending and Embedding the Python Interpreter Tutorial, while good, are not QT level good regarding documentation. Nonetheless, with a bit of effort, combined with the CPython source code, it is more than enough to start writing something. In the end, it isn’t as complicated as it may seem.
To have some code reference, I will use the first thing I build in this last few days, a static size array of references that is type-checked at runtime ( similar to a C array of pointers in some way but with some notable differences ).
This is an example of what our extension module will let us do:
>>> import lds_array
>>> a = lds_array.array(4, int, 3, 5, 6, 7)
>>> b = lds_array.array(3, str, "aaa", "nnn", "ffff")
>>> print(a)
[3, 5, 6, 7]
>>> print(a * 5)
[3, 5, 6, 7, 3, 5, 6, 7, 3, 5, 6, 7, 3, 5, 6, 7, 3, 5, 6, 7]
>>> print(b + lds_array.array(2, str, "abc", "bcs"))
[aaa, nnn, ffff, abc, bcs]
>>> for x in b:
... print(x * 5)
...
aaaaaaaaaaaaaaa
nnnnnnnnnnnnnnn
ffffffffffffffffffff
>>> a[3]
7
>>> a[3] = 56
>>> a[3]
56
Before we start looking at the C API, I’d like to say a few things. First, I don’t particularly feel suitable to teach anyone anything about the C API. I used it for only a few days ( as this was the time I allotted for exercising with it ), and probably am not even at the level that I can write something decent without the risk of leaking memory or riddling it with bugs.
For these reasons, I can’t say that the code I will use in this post is a good example of the API usage nor that is correctly idiomatic in the way it is written. So please take everything with a grain of salt.
Furthermore, we will only look at using C code in Python and not at calling Python from C.
The goal of this post is to try and streamline the very first step of approaching the C API, the step I did this last few days, in the hope of helping someone that is beginning this journey that may have a few hurdles at the start.
With this behind us, let’s go on.
Some basics
Before starting with the code, I’d like to talk about some concepts and idioms that are present in the C API. If you are like me, some of this will look too extraneous to remember without some code. In that case, I suggest to skim this part and return to it, if needed, while reading the rest of the post.
Accessing the C API
Most of what we need to access the C API is stored in the header file Python.h that usually comes with your python installation. There are a few “catches” with this header file that are of note.
- It brings with it some standard headers, precisely:
- stdio.h
- string.h
- errno.h
- stdlib.h, if this is not present malloc, realloc and free are defined by the Python.h header file.
- This header file can define some pre-processor definitions that change the way in which standard header files behave. As such, it is important to #include it before any standard header.
All Python.h user-visible names, be it of function, macros or structure, have the prefix Py or PY and it is strongly advised to never use this prefixes in our code to avoid any kind of confusion.
Another useful header is structmember.h which contains what we need to map our C structure representation of objects to Python’s object attributes. It brings stddef.h with it to provide offsetof that we need when we define our attributes.
Reference Counting
Python manages the lifetime of its objects trough Reference Counting ( boasting an optional cycle detector on top that we won’t use in this exercise ). While all the intricacies of this are hidden from a Python user, while working with the C API we have to manually manage references. Failure to do so would cause memory leaks in the program.
While this may seem like a complicated matter ( and well it is, worse than managing C heap-allocated memory in my opinion ), the tools we use to accomplish this task are small and simple. Py_INCREF, Py_DECREF and Py_XDECREF are the three macros that Python.h provides us to control the refcount of PyObjects.
Those three macros take a PyObject* as their only argument. As you may imagine, Py_INCREF increments the count by one while Py_DECREF decrements it by one.
Furthermore, if Py_DECREF decrements a refcount to 0, it frees the object memory, not directly but by calling the object destructor trough a function pointer which is stored in the tp_deallocator member of its PyTypeObject. Py_XDECREF works exactly like Py_DECREF but, while the latter requires its argument to be non-NULL, correctly handles a NULL-argument.
There exists an x macro for increasing too, Py_XINCREF
We have yet another ( and last ) macro at our disposal in Py_CLEAR, which decrements a refcount to 0, frees the object and sets its argument to NULL ( while the other decrement macros don’t ). Py_CLEAR like Py_XDECREF handles NULL arguments ( by doing nothing ).
Now, while we have to manually manage the reference count of objects, it is not always the case to do so. Python itself owns each and every one of its objects. What we are able to own, is a reference to one of the objects. This reference is the one we are going to actually manage, deciding the object lifetime at the same time.
In Python’s ownership model, we can identify three types of references, each one with its own management tactic.
Firstly, we have new references, for example, the one returned by a PyObject-building function; e.g Py_BuildValue or PyTuple_New. When we get hold of a new reference, as already said this usually happens when we construct a new object, we become its owner. As the owner of a new reference it is our job to dispose of it, by Py_DECREFing it to 0 or to, otherwise, pass it to someone who will do it for us, essentially giving up our ownership in the process.
If we won’t take care of our new reference this way, we will end with a memory leak.
For example:
static PyObject* square(long n) {
PyObject* num = NULL;
PyObject* res = NULL;
num = PyLong_FromLong(n); // Here we will get a new reference to a python int object we created from the original argument
res = PyNumber_Multiply(num, num); // Again we get a new reference that we have to manage
Py_DECREF(num); // As we don't need it anymore we do our job and decrease the reference, deleting it
return res; // Instead of managing res ourselves, we pass the ownership to the caller, which will need to eventually dispose of it or pass it along, as a new reference
}
We have forgone any error checking here for the sake of the example readability but we can incur in a core dump as both new reference-returning function we call may return NULL on failure which is not supported by Py_DECREF ( we could use Py_XDECREF, but it would be wrong as we would have a dangling error which, as we will see later, should be propagated ). It may be tempting to simply write something along the line of:
static PyObject* square(long n) {
return PyNumber_Multiply(PyLong_FromLong(n), PyLong_FromLong(n));
But in this case, and every case where a new reference-returning function return value is used as a temporary object, we will leak both of the PyLong_FromLong created references as the PyNumber_Multiply function will only borrow them without taking on their ownership ( and as such, Py_DECREF them to zero ).
The second type we can encounter are stolen references. Stolen references are usually encountered when compositing a reference we own with another object that manages it, e.g any kind of container. When we deal with stolen references the obligation of managing the reference is left to the “thief”.
For example:
static PyObject* longToUnaryTuple(long n) {
PyObject* res = NULL;
res = PyTuple_New(1); // We build a new tuple and own its reference
PyTuple_SetItem(res, 0, PyLong_FromLong(n)); // The PyLong_FromLong new reference gets stolen by SetItem and isn't our responsability anymore
return res; // We pass ownership of the tuple reference to the caller
}
There isn’t much more to stolen references. If we know which functions steal a reference we are good to go. The one thing we have to be careful about is not mindlessly dabbling with a reference that was stolen:
static PyObject* longToUnaryTuple(long n) {
PyObject* res = NULL;
PyObject* num = PyLong_FromLong(n);
res = PyTuple_New(1);
PyTuple_SetItem(res, 0, num);
Py_DECREF(num) // We don't really own the reference anymore and are potentially disrupting the tuple internals
return res;
}
We should not decrement a reference refcount that we don’t own without first incrementing it ( that we do if we need it to be surely available for a certain time or scope ). Usually, we should have nothing to do with a reference that was stolen as it isn’t our concern anymore in any way.
Getting to the last type we encounter borrowed references. While they are pretty essential in their contract, we can use a borrowed reference but we aren’t its owner, they have some tricky parts. The fact is that we have a reference that is actually managed by someone else that we need to use ( contrary to new references that we own and stolen references which aren’t ours anymore and we should not manage ). They may, for example, be invalidated by code that dabbles with the original owner.
For example:
static void buggyTupleAccess(PyObject* o) {
if (PyTuple_Check(o)) {
PyObject* last = PyTuple_GetItem(o, PyTuple_Size(o)-1); // Here we get a borrowed reference to the last item in the tuple
PyTuple_SetItem(o, PyTuple_Size(o)-1, PyLong_FromLong(10000)); // Here we set a new item as the last element of the tuple
// do something with last
}
}
The problem with this snippet is that when we modify the last element of the list, setting a new one, the list, that manages the reference we borrowed, may as well ( and actually will ) Py_DECREF it, potentially freeing the object in the process. When we access last later in the code, we have no guarantee that it wasn’t invalidated.
The trickiest part is that these type of bugs are not always obvious. Python can reuse memory addresses that were previously freed, meaning that we won’t necessarily get bogus values in the process. Or it may be that sometimes last may have a refcount greater than 1, making it seem like the code is working and then randomly crash when this is not true anymore.
Sometimes we may work with special values, like the integers from -5 to 255 which Python always keeps in memory, that makes the code work.
For all these reasons, when using borrowed references, we should increment the refcount while in scope, and decrement it when we are done:
static void buggyTupleAccess(PyObject* o) {
if (PyTuple_Check(o)) {
PyObject* last = PyTuple_GetItem(o, PyTuple_Size(o)-1);
Py_INCREF(last); // increase when the scope we need it in starts
PyTuple_SetItem(o, PyTuple_Size(o)-1, PyLong_FromLong(10000));
// do something with last
Py_DECREF(last) // Decrease the reference, potentially freeing it, as we have no need for it anymore
}
}
{ By the way, another thing we can do here is to get the item trough PySequence_GetItem, from the abstract sequence protocol, which will give us a new reference we only have to decref ).
Simple enough but can get tricky at time. Furthermore, we have to make sure that the reference gets decreased in each code path.
Exception handling
In the C API, exceptions are raised as a two-step process.
First, we have to actually set an exception. The current exception details are stored globally, per thread. You can see the members that store the exception in cpython/pystate.h under the thread state structure. To do this we can use a series of macros defined in cpython/error.c.
The most important one is probably PyObject* PyErr_Format(PyObject* exception, const char* format, …), which sets a given exception and a formatted message. The C API provides the same standard exception objects that Pyhon does as global PyObject*s.
As a second step, we have to return an error indicator value from the current function. For a pointer-returning function, we should return NULL. For int-returning functions, we should return -1 ( and 0 on success ) { An exception to this is the PyArg_ family of functions which return 1 for success and 0 for failure }.
For example:
static PyObject* tuple_get(PyObject* o, Py_ssize_t index) {
if (!PyTuple_CheckExact(o)) {
PyErr_Format(PyExc_TypeError, "Unsupported operand type for %s: '%s'", __FUNCTION__, Py_TYPE(o)->tp_name);
return NULL;
}
.....
}
A failure to follow the two-step process will result in an error. If we forget to set an exception we will get a SystemError:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
SystemError: error return without exception set
Setting an exception but failing to return a value that signals it, will give us an error at runtime. Usually, we should check for most return values and propagate an exception when set ( Without overwriting it so that we know where and what initially occurred ).
Other than using the built-in exception, we can build custom module-level exceptions trough PyErr_NewException and PyErr_NewExceptionWithDoc. For example:
static PyObject* CustomException; // We need to declare a static PyObject* that points to our exception
...
PyMODINIT_FUNC
PyInit_custom()
{
PyObject* module; // This is our module object. We skip how to build it in this example
...
CustomException = PyErr_NewException("custom.CustomException", NULL, NULL); // We create the exception object and point to it
Py_INCREF(CustomException);
PyModule_AddObject(module, "CustomException", CustomException); // We need to add the exception to the module's objects
...
}
PyObject and PyTypeObject
At the heart of all Python’s object resides the PyObject structure:
typedef struct _object {
_PyObject_HEAD_EXTRA
Py_ssize_t ob_refcnt;
struct _typeobject *ob_type;
} PyObject;
A PyObject, as you can see, simply contains the object reference count and a pointer to its type ( which we will look at in a moment ). There are a few interesting things to know about Python’s object ( and a few more interesting things you can read in CPython source that we normally should not need to know ).
All objects are allocated on the heap ( there are some exceptions regarding type objects ). We always work with PyObject* and each Python’s object can be cast to a PyObject*.
The most interesting part about the PyObject structure is the pointer to a type object, this is where all the action actually resides. This structure contains all the pieces of information needed to know how the type actually operates.
We will look at some of the fields later while looking at the array implementation as it is easier to understand through an example.
There are a few more things we could talk about but those were the most important concepts. Many things are just unclear without a more consistent example and as such we will explore them with the array example.
Building an array
Now we finally get to do some practice. As said before, this was the very first thing I build and is riddled with bad choices. I’ve decided to use this as an example as I think it may better present some questions and errors that someone who has never touched the C API may encounter. Furthermore, it gives me a lot of great chances to think and discuss how things could have been done differently.
The code
#include <Python.h>
#include <structmember.h>
static PyTypeObject arrayType;
typedef struct {
PyObject_HEAD
PyObject** data;
Py_ssize_t size;
PyTypeObject* storedType;
} array;
static PyMemberDef array_members[] = {
{"size", T_PYSSIZET, offsetof(array, size), READONLY, "Describe how many elements are stored in the array"},
{NULL}
};
//-- Array Utilities --//
#define ARR_AS_ARRAY(obj) ((array*)(obj))
#define ARR_EXACT_CHECK(obj) (Py_TYPE((obj)) == &arrayType)
#define ARR_CHECK(obj) (PyObject_Check((obj), &arrayType))
#define ARR_SIZE(arr) ((ARR_AS_ARRAY(arr))->size)
#define ARR_STORED_TYPE(arr) (((ARR_AS_ARRAY(arr))->storedType))
#define ARR_CHECK_TYPE(arr, obj) (ARR_STORED_TYPE((arr)) == Py_TYPE((obj)))
#define ARR_GET(arr, i) (*(((ARR_AS_ARRAY(arr)->data)) + (i)))
#define ARR_ASSIGN(arr, i, obj) ((ARR_GET((arr), (i))) = (obj))
static PyObject* newArray(PyTypeObject* type, PyTypeObject* storedType, Py_ssize_t size) {
array* self;
self = (array*)type->tp_alloc(type, 0);
if (self == NULL) {
return NULL;
}
self->data = (PyObject**)PyMem_Calloc(sizeof(PyObject*), size);
if (self->data == NULL) {
Py_DECREF(self);
return NULL;
}
self->size = size;
self->storedType = storedType;
Py_INCREF(self->storedType);
return (PyObject*)self;
}
// Array Utilities End //
//-- Array Type Methods --//
static void array_dealloc(array* self) {
for (Py_ssize_t index = 0; index < self->size; ++index) {
Py_XDECREF(ARR_GET(self, index));
}
PyMem_Free((void*)self->data);
Py_DECREF(ARR_STORED_TYPE(self));
}
static PyObject* array_new(PyTypeObject* type, PyObject* args, PyObject* kwds) {
if (PyTuple_Size(args) < 2) {
PyErr_Format(PyExc_TypeError, "%s() takes at least 2 arguments (%i given)", __FUNCTION__, PyTuple_Size(args));
return NULL;
}
Py_ssize_t size = PyLong_AsSsize_t(PyTuple_GET_ITEM(args, 0));
PyObject* storedType = PyTuple_GET_ITEM(args, 1);
if (PyErr_Occurred() || !PyType_Check(storedType)) {
PyErr_Format(PyExc_TypeError, "Unsupported operand type(s) for %s: '%s' and '%s'", __FUNCTION__, Py_TYPE(PyTuple_GET_ITEM(args, 0))->tp_name, Py_TYPE(PyTuple_GET_ITEM(args, 1))->tp_name);
return NULL;
}
if (size <= 0) {
PyErr_Format(PyExc_ValueError, "The array size must be a positive integer greater than 0");
return NULL;
}
if (size > (PY_SSIZE_T_MAX / sizeof(PyObject*))) {
return PyErr_NoMemory();
}
return newArray(type, (PyTypeObject*)storedType, size);
}
static int array_init(array* self, PyObject* args, PyObject* kwds) {
static const Py_ssize_t MIN_ARGUMENTS = 2;
Py_ssize_t argsSize = PyTuple_Size(args);
if (argsSize > (ARR_SIZE(self) + MIN_ARGUMENTS)) {
PyErr_Format(PyExc_TypeError, "%s() takes at most %i arguments for an array of size %i (%i given)", __FUNCTION__, ARR_SIZE(self) + MIN_ARGUMENTS, ARR_SIZE(self), argsSize);
return -1;
}
for (Py_ssize_t index = 2; index < argsSize; ++index) {
PyObject* tmp = PyTuple_GET_ITEM(args, index);
if (!ARR_CHECK_TYPE(self, tmp)) {
PyErr_Format(PyExc_TypeError, "Unsupported operand type(s) for %s for array of type '%s': '%s'", __FUNCTION__, ARR_STORED_TYPE(self)->tp_name, Py_TYPE(tmp)->tp_name);
return -1;
}
Py_INCREF(tmp);
ARR_ASSIGN(self, index-MIN_ARGUMENTS, tmp);
}
return 0;
}
static PyObject* array_str(PyObject* o) {
PyObject* openingCharacter = PyUnicode_FromString("[");
PyObject* closeningCharacter = PyUnicode_FromString("]");
PyObject* separator = PyUnicode_FromString(", ");
PyObject* stringRepresentations = PyTuple_New(ARR_SIZE(o));
for (Py_ssize_t index = 0; index < ARR_SIZE(o); ++index) {
PyTuple_SET_ITEM(stringRepresentations, index, PyObject_Str(ARR_GET(o, index)));
}
PyObject* elementsString = PyUnicode_Join(separator, stringRepresentations);
PyObject* openedString = PyUnicode_Concat(openingCharacter, elementsString);
PyObject* completeString = PyUnicode_Concat(openedString, closeningCharacter);
Py_DECREF(openingCharacter);
Py_DECREF(closeningCharacter);
Py_DECREF(separator);
Py_DECREF(stringRepresentations);
Py_DECREF(elementsString);
Py_DECREF(openedString);
return completeString;
}
// Array Type Methods End //
//-- Sequence Protocol --//
static Py_ssize_t array_sq_length(PyObject* o) {
return ARR_SIZE(o);
}
static PyObject* array_sq_item(PyObject* o, Py_ssize_t i) {
PyObject* item = NULL;
if (i < 0 || i >= ARR_SIZE(o)) {
PyErr_Format(PyExc_IndexError, "array index out of range");
return NULL;
}
item = ARR_GET(o, i);
if (item == NULL) {
PyErr_Format(PyExc_IndexError, "Accessing uninitialized object at index %i", i);
return NULL;
}
Py_INCREF(item);
return item;
}
static int array_sq_ass_item(PyObject* o, Py_ssize_t i, PyObject* v) {
if (!ARR_CHECK_TYPE(o, v)) {
PyErr_Format(PyExc_TypeError, "Unsupported operand type(s) for %s for array of type '%s': '%s'", __FUNCTION__, ARR_STORED_TYPE(o)->tp_name, Py_TYPE(v)->tp_name);
return -1;
}
if (i < 0 || i >= ARR_SIZE(o)) {
PyErr_Format(PyExc_IndexError, "array index out of range");
return -1;
}
Py_XDECREF(ARR_GET(o, i));
ARR_ASSIGN(o, i, v);
Py_INCREF(v);
return 0;
}
static PyObject* array_sq_concat(PyObject* o1, PyObject* o2) {
if (!ARR_EXACT_CHECK(o2)) {
PyErr_Format(PyExc_TypeError, "can only concatenate array ( not '%s' ) to array", Py_TYPE(o2)->tp_name);
return NULL;
}
if (ARR_STORED_TYPE(o1) != ARR_STORED_TYPE(o2)) {
PyErr_Format(PyExc_TypeError, "can only concatenate array of type '%s' ( not '%s' ) to array of type '%s'", ARR_STORED_TYPE(o1)->tp_name, ARR_STORED_TYPE(o2)->tp_name, ARR_STORED_TYPE(o1)->tp_name);
return NULL;
}
PyObject* arr = newArray(Py_TYPE(o1), ARR_STORED_TYPE(o1), ARR_SIZE(o1) + ARR_SIZE(o2));
if (arr == NULL) {
return NULL;
}
for (Py_ssize_t index = 0; index < ARR_SIZE(o1); ++index) {
array_sq_ass_item(arr, index, ARR_GET(o1, index));
if (PyErr_Occurred()) {
Py_DECREF(arr);
return NULL;
}
}
for (Py_ssize_t index = 0; index < ARR_SIZE(o2); ++index) {
array_sq_ass_item(arr, index + ARR_SIZE(o1), ARR_GET(o2, index));
if (PyErr_Occurred()) {
Py_DECREF(arr);
return NULL;
}
}
return arr;
}
static PyObject* array_sq_repeat(PyObject* o, Py_ssize_t count) {
if (count <= 0) {
PyErr_Format(PyExc_ValueError, "Can't multiply array by non-positive non-greater than 0 int");
}
PyObject* arr = newArray(Py_TYPE(o), ARR_STORED_TYPE(o), ARR_SIZE(o) * count);
if (arr == NULL) {
return NULL;
}
for (Py_ssize_t repetition = 0; repetition < count; ++repetition) {
for (Py_ssize_t index = 0; index < ARR_SIZE(o); ++index) {
array_sq_ass_item(arr, index + (ARR_SIZE(o) * repetition), ARR_GET(o, index));
if (PyErr_Occurred()) {
Py_DECREF(arr);
return NULL;
}
}
}
return arr;
}
static PySequenceMethods array_sq_methods = {
.sq_length = array_sq_length,
.sq_item = array_sq_item,
.sq_ass_item = array_sq_ass_item,
.sq_concat = array_sq_concat,
.sq_repeat = array_sq_repeat
};
// Sequence Protocol End //
//-- Array Structures and Type --//
static PyTypeObject arrayType = {
PyVarObject_HEAD_INIT(NULL, 0)
.tp_name = "lds_array.array",
.tp_doc = "A c-like array structure",
.tp_basicsize = sizeof(array),
.tp_itemsize = 0,
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
.tp_new = array_new,
.tp_init = array_init,
.tp_dealloc = array_dealloc,
.tp_members = array_members,
.tp_as_sequence = &array_sq_methods,
.tp_str = array_str
};
// Array Structures and Type End //
//-- Module --//
static PyModuleDef arraymodule = {
PyModuleDef_HEAD_INIT,
.m_name = "lds_array",
.m_doc = "C-like array module",
.m_size = -1
};
PyMODINIT_FUNC
PyInit_lds_array() {
PyObject* m;
if (PyType_Ready(&arrayType) < 0) {
return NULL;
}
m = PyModule_Create(&arraymodule);
if (m == NULL) {
return NULL;
}
Py_INCREF(&arrayType);
PyModule_AddObject(m, "array", (PyObject*)&arrayType);
return m;
}
// Module End //
It has a few things going on. Some things may be a bit difficult to untangle at first glance if you’ve never used the C API but the example should be simple enough to be used as a kick-start.
The module object and initialization
While it is the last thing in the array code, the simplest unit we can start from to untangle the code is the module object.
static PyModuleDef arraymodule = {
PyModuleDef_HEAD_INIT,
.m_name = "lds_array",
.m_doc = "C-like array module",
.m_size = -1
};
PyMODINIT_FUNC
PyInit_lds_array() {
PyObject* m;
if (PyType_Ready(&arrayType) < 0) {
return NULL;
}
m = PyModule_Create(&arraymodule);
if (m == NULL) {
return NULL;
}
Py_INCREF(&arrayType);
PyModule_AddObject(m, "array", (PyObject*)&arrayType);
return m;
}
When we add an extension through the C API we are adding a new module. To do this we have a two-stage process where we define the module and a function to initialize it. PyModuleDef is a structure that defines all the fields that a module object has. We will see that there is a similar structure for each type of entity we can add.
While the module array module we provide is really basic, PyModuleDef has a few members we can use:
Type | Member | Use |
---|---|---|
PyModuleDef_Base | m_base | This is always initialized to PyModuleDef_HEAD_INIT which adds some needed members |
const char* | m_name | The name for the module |
const char* | m_doc | The docstring for the module. As for all docstrings members, we can use a PyDoc_STRVAR variable |
Py_ssize_t | m_size | A module state may be kept in a module-specific memory. m_size defines the size of that memory. If set to -1 the module has, instead, global state and does not support sub-interpreters. A non-negative size is required for multi-phase initialization |
PyMethodDef* | m_methods | Points to a PyMethodDef structure that defines all the module-level functions |
PyModuleDef_Slot* | m_slots | Slot definitions for multi-phase initialization. Must be NULL when single-phase initialization is used |
traverseproc | m_traverse | The traversal function neede for GC traversal. May be NULL in case the GC is not needed |
inquiry | m_clear | The clear function for the GC. May be NULL in case the GC is not needed |
freefunc | m_free | A function called to deallocate the module. May be NULL if it isn’t needed |
While some members are pretty straightforward, like m_name and m_size, a few interesting things are introduced in this table. The most important one for modules, as it is specific to them, is the difference between multi-phase and single-phase initialization, which we will look at in a bit.
Going top to bottom, we have three things to talk about before initialization:
Docstrings:
Everyone who used Python will know what docstrings are. From a C API point of view, we can relate a docstring to a module, function or anything else thanks to the doc member we can find in many definition structures.
While we can simply pass a string literal or a C-string to it, the C API provides us with the PyDoc_STRVAR helper macro. Interesting enough, this does not seem to be documented in the API reference, but we can look at the code to see that it simply declares a static char[] variable containing the string we pass to it.
PyDoc_STRVAR(custom_doc, "A custom docstring");
/*
which expands to
static char custom_doc[] = "A custom docstring";
or to
static char custom_doc[] = "";
if WITH_DOC_STRING is not defined in pyconfig.h
*/
The PyMethodDef structure:
The PyMethodDef structure is used to define a python method.
It only has a few members:
Type | Member | Use |
---|---|---|
const char* | ml_name | The name of the method |
PyCFunction | ml_meth | Pointer the function implementation |
int | ml_flags | Bits-flag indicating how Python should construct the call for this method. See below for more pieces of information |
const char* | ml_doc | Docstring for the method |
There are two interesting bits here, the PyCFunction type and ml_flags. PyCFunction is a typedefed function pointer to a function that returns a PyObject* and accepts two PyObject* as parameters. This is the basic signature for C Functions that are callable by python.
The first PyObject*, usually called self, is actually the self object ( the same you would get in a Python instance method ). For a module-level function this is the module instance itself. The second parameter, usually called args, is a PyTuple object that contains all the arguments of the function. Again, you can see the parallelism with Python.
Extending on this is PyCFunctionWithKeywords, which adds a third PyObject* parameter that, as you probably guessed, is a PyDictionary containing the named arguments passed to the function. We will look at how to deal with both the args and kwargs arguments later.
The ml_meth field has to be a PyCFunction pointer or a pointer to another function that is castable, and casted, to it. Python bases the way in which the arguments are passed to this function, when called, on the ml_flags parameter. There are a few flags we can use:
Flag | Meaning |
---|---|
METH_VARARGS | This is the standard calling convention. It passes a self and args argument and expects the method to be of the PyCFucntion type |
METH_KEYWORDS | This is for PyCFunctionWithKeywords types. It passes self and args as METH_VARARGS but adds a third dictionary argument on top of them |
METH_NOARGS | This is for a function that expects no arguments. The ml_meth should still be of the PyCFunction type but the second parameter will always be passed as NULL |
METH_O | This is a commodity flag for PyCFunctions that expects a single PyObject as argument. Instead of passing a tuple the second argument is passed directly as the object that would be the sole element of the tuple. |
Of all these flags only METH_VARARGS and METH_KEYWORDS can be combined together. There are two more flags, called binding flags, that can be combined with any of the previously described flags but are exclusive between themselves.
Flag | Meaning |
---|---|
METH_CLASS | The self parameter will be the type object of the instance instead of the instance itself. This is used to create class methods |
METH_STATIC | The self parameter will be NULL. Used to create static methods |
An example to better wrap your head around this ( with some spoilers on arguments parsing ):
static PyObject* simplePow(PyObject* self, PyObject* args) { // PyCFunction
PyObject* base = NULL;
PyObject* exponent = NULL;
PyObject* result = NULL;
if (!PyArg_UnpackTuple(args, __FUNCTION__, 2, 2, &base, &exponent)) { // This is one of the ways to parse the tuple arguments
return NULL;
}
Py_INCREF(base); Py_INCREF(exponent); // We got borrowed references from the tuple. We increment the refcount for good measure.
if (!PyNumber_CHECK(base) || !PyNumber_CHECK(exponent)) { // We check if the passed object implements the Number Protocol otherwise we raise an exception
PyErr_Format(PyExc_TypeError, "Unsupported operand type(s) for %s: '%s' and '%s'", __FUNCTION__, Py_TYPE(base)->tp_name, Py_TYPE(exponent)->tp_name);
Py_DECREF(base); Py_DECREF(exponent); // Remember to decrement the reference count in each path!
return NULL;
}
result = PyNumber(base, exponent, Py_None); // We should probably check if any error happened here but for this example we won't
Py_DECREF(base); Py_DECREF(exponent); // We have to decrement the refcount of the borrowed references
return result; // result is a new reference that we are passing to the caller to handle
}
// Check https://pythonextensionpatterns.readthedocs.io/en/latest/canonical_function.html for another way we could structure this ( and many other ) function ( This pattern is used throughout some of the CPython source code too ).
static PyMethodDef custom_methods[] = {
{ "simplePow", simplePow, METH_VARARGS, "" }, // Our method entry
{ NULL, NULL, 0, NULL} // A sentinel to know when we are at the end
}
static struct PyModuleDef custom_module = {
PyModuleDef_HEAD_INIT,
"custom",
NULL,
-1,
custom_methods
};
PyMODINIT_FUNC
PyInit_spam(void)
{
return PyModule_Create(&custom_module);
}
m_traverse, m_clear and GC Support
As said before, on top of the reference counting system Python ships with a optional-use garbage collector to deal, specifically, with cycles. To quote the API reference:
Python’s support for detecting and collecting garbage which involves circular references requires support from object types which are “containers” for other objects which may also be containers. Types which do not store references to other objects, or which only store references to atomic types (such as numbers or strings), do not need to provide any explicit support for garbage collection.
The array structure we are building should actually implement GC support. I must say that in the few days I studied the API I never really implemented GC support nor looked into it much. While I can’t give you any personal experience “insight”, I can at least provide you with pointers to how such support is added and where to look for informations on it.
Implementing GC support doesn’t seem to require too much. First, we have to provide the traverse function, we will see later that the same GC support functions appear for object types too, which has a traverseproc signature.
If needed we have to provide a tp_clear function, which is of the inquiry type. Going by the reference, the discriminator factor for needing a tp_clear function is the fact that the container we are garbage-collecting is mutable ( in which case we need one ) or not. Nonetheless, it seems a bit tricky to provide one and I advise you to read the reference entry on it if interested.
The other things we have to ensure are that, quoting the reference:
- The memory for the object must be allocated using PyObject_GC_New() or PyObject_GC_NewVar().
- Once all the fields which may contain references to other containers are initialized, it must call PyObject_GC_Track() { which we will later untrack }.
Furthermore, to actually enable the GC for the object, we need to add the Py_TPFLAGS_HAVE_GC flag to its type.
To start working with the GC the best resource seems to be the GC-support tutorial in the Extending and Embedding the Python Interpreter guide.
Single-phase initialization and Multi-phase initialization
With those out of the way, the last thing we are interested in is the difference between the two types of module initialization. I’ve found the PEP 489 to be a really cool read on this argument. The reference advises reading PEP 3121 for more details on the sense of the m_size field.
Like the names may imply, the two strategies differ in the number of steps taken to initialize a module and the way in which those steps are carried.
With single-phase initialization, the module is created, populated and returned directly by the initialization function, uniting the creation and initialization process and generating a singleton module.
As we can read from PEP 489, this process is different from how Python Modules are built and is specific to C’s extension modules.
Furthermore, as the initialization function receives no context, the initialization process lacks access to some pieces of information and brings along some difficult to resolve problems, like the support for sub-interpreters.
On a practical note, this is what our array is doing and how single-phase initialization is supported in the C API:
PyMODINIT_FUNC
PyInit_lds_array() {
PyObject* m;
if (PyType_Ready(&arrayType) < 0) {
return NULL;
}
m = PyModule_Create(&arraymodule);
if (m == NULL) {
return NULL;
}
Py_INCREF(&arrayType);
PyModule_AddObject(m, "array", (PyObject*)&arrayType);
return m;
}
This is a way of initializing the module, where we create a module instance from its definition and then populate it through the use of supporting functions like PyModule_AddObject, that uses single-phase initialization.
The code is pretty self-explicative so I won’t explain it.
Multi-phase initialization is a bit more interesting. Let’s start by going practical. To request Multi-phase initialization, the Init function should return a PyModuleDef instance pointer for which the m_slots field is correctly set and non-empty.
This should be done trough the use of the PyModuleDef_Init function:
PyMODINIT_FUNC
PyInit_lds_array(void)
{
return PyModuleDef_Init(&arraymodule);
}
This will give our module def to Python which will call the constructing functions provided through the m_slots field. The structure connects our m_slots field should be an array of PyModuleDef_slots which has two fields:
Type | Member | Use |
---|---|---|
int | slot | Defines the type of this slot. See below |
void* | value | The function for this slot whose meaning depends on the slot field |
The slot field has two possible types.
Py_mod_create and Py_mod_exec.
The first points to a function of signature:
PyObject* f(PyObject *spec, PyModuleDef *def)
which should create the module object and return it or set an error if it is not possible. This is the equivalent step to new in the module initialization. The second parameter will receive the PyModuleDef we return from the init function. The spec parameter is a bit more interesting. There is a lot to understand about the ModuleSpec object and I will point you to PEP 451 for more informations on it.
If a creation slot is not provided, Python will create the module through the PyModule_New function. I could not really find an example of a creation slot in the source code. It seems idiomatic to leave this part to Python.
The second type points to a function with signature:
int f(PyObject* module)
that should actually execute the module ( which is equivalent to actually evaluating the code in a module ). We should add classes, functions and so on to our module in this function.
More than one execution can be provided. In this latter case they are executed sequentially in the order they appear in the PyModuleDef_slot array.
If we wanted to rewrite our example code to support Multi-phase initialization, we could try something like the following:
static int array_mod_exec(PyObject* module) {
if (PyType_Ready(&arrayType) < 0) {
return -1;
}
Py_INCREF(&arrayType);
if (PyModule_AddObject(module, "array", (PyObject *) &arrayType) < 0) {
// While I thought that we would need to decref &arrayType, every snippet of this code that I found in the CPython source doesn't
// I'm still not sure why
return -1;
}
return 0;
}
static PyModuleDef_Slot[] arrayslots { // Our slots array
{Py_mod_exec, array_mod_exec}, // An exec function. We don't provide a create function and let Python do it for us
{0, NULL} // a sentinel
}
static PyModuleDef arraymodule = {
PyModuleDef_HEAD_INIT,
.m_name = "lds_array",
.m_doc = "C-like array module",
.m_size = -1,
.m_slots = arrayslots // We added our slots to the module def
};
PyMODINIT_FUNC
PyInit_lds_array() {
return PyModuleDef_Init(&arraymodule); // We now return a PyModuleDef instance instead of the complete module
}
This is actually pretty similar to what we had before, we mostly moved some things around.
If you’ve read PEP 489, there seems to be quite a few advantages to it. I think this should be the preferred initialization for our extension modules.
The array object
Moving on we should look at our array. First thing first, this is how we have defined the array:
typedef struct {
PyObject_HEAD
PyObject** data;
Py_ssize_t size;
PyTypeObject* storedType;
} array;
It doesn’t have many fields. PyObject_HEAD is boilerplate code that defines the initial segment of a PyObject. We always need it at the head of our custom objects.
data Is where our references will be stored. We will allocate this memory in the new method of our array.
size tells us how big the array is. This is non-changing value that is decided on construction. Py_ssize_t is a signed number type for which
sizeof(size_t) == sizeof(Py_ssize_t)
holds.
storedType is a pointer to a Python Type ( which is a PyObject too ) we will use to provide some type-checks of inserted elements.
One interesting thing is that, later, I’ve found out that Python provides a second type of PyObject specifically for objects that have a notion of length ( like containers ): the PyVarObject; that adds a field to store the number of items in the object, ob_size.
I think we may have used it for the array object but I haven’t tried it.
To make an object a var object we have to use PyObject_VAR_HEAD instead of PyObject_HEAD and ensure that the last field of the struct is an array of length one where Python will malloc enough space for ob_size elements. Lastly, we have to make some changes to the type of the object that we will see soon enough. You can see an example of this here.
static PyMemberDef array_members[] = {
{"size", T_PYSSIZET, offsetof(array, size), READONLY, "Describe how many elements are stored in the array"},
{NULL}
};
PyMemberDef describes the attribute of a type as a C struct member. It has a few fields:
Type | Member | Use |
---|---|---|
const char* | name | The name of the attribute |
int | type | The type of the attribute. See below |
Py_ssize_t | offset | The offset in bytes that the member is located in the C struct. We use offsetof for this |
int | flags | Flags that defines the readability and writability of the attribute |
const char* | doc | The docstring for the attribute |
Flags can either be 0 for Read-Write access or READONLY going by the reference. We can see from the source code that there exist a restricted version of both read and write. They are used in just a few places in the code but I could not find what they exactly represent or if we should care about them.
We have a few possible types:
Macro | CType |
---|---|
T_SHORT | short |
T_INT | int |
T_LONG | long |
T_FLOAT | float |
T_DOUBLE | double |
T_STRING | const char* |
T_OBJECT | PyObject* |
T_OBJECT_EX | PyObject* |
T_CHAR | char |
T_BYTE | char |
T_UBYTE | unsigned char |
T_UINT | unsigned int |
T_USHORT | unsigned short |
T_ULONG | unsigned long |
T_BOOL | char |
T_LONGLONG | long long |
T_ULONGLONG | unsigned long long |
T_PYSSIZET | Py_ssize_t |
The difference between T_OBJECT and T_OBJECT_EX is that the first returns None if the member is NULL while the latter raises an AttributeError. The reference advises us to use the EX type because it handles the del statement more correctly.
This NULL-terminated array of PyMemberDef structures is plugged into the array type to define the object attributes. There is no real reason to give read access to the size attribute as we support Python’s len trough the sequence protocol. Nonetheless, I was trying out attributes.
Going forward we finally get to the core of the action of a PyObject: its type.
static PyTypeObject arrayType = {
PyVarObject_HEAD_INIT(NULL, 0)
.tp_name = "lds_array.array",
.tp_doc = "A c-like array structure",
.tp_basicsize = sizeof(array),
.tp_itemsize = 0,
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
.tp_new = array_new,
.tp_init = array_init,
.tp_dealloc = array_dealloc,
.tp_members = array_members,
.tp_as_sequence = &array_sq_methods,
.tp_str = array_str
};
Our array has a pretty bare type, but we have a lot of possibilities to customize our types.
Instead of rewriting them here I advise you to look at the reference page. We will look specifically at our example.
Some of those are already familiar like (tp_)name and (tp_)doc.
tp_basicsize and tp_itemsize work in tandem to define how much space the object will need. tp_basicsize is the size of the actual structure. tp_itemsize should be 0 for all objects that aren’t variable in size. For varObjects it represents the size of a single stored item. Together with the ob_size field it defines how much memory the stored elements occupy.
tp_flags is used for various reasons. Usually, it should have at least Py_TPFLAGS_DEFAULT set. We have a few choices here:
Flag | Use |
---|---|
PY_TPFLAGS_HEAPTYPE | This has to be set to indicate that the type object itself is heap allocated |
Py_TPFLAGS_BASETYPE | This has to be set to enable the type to be subtyped |
Py_TPFLAGS_READY | This gets set by PyType_Ready when the type has been fully initialized |
Py_TPFLAGS_READYING | This gets set by PyType_Ready when the type is being initialized |
Py_TPFLAGS_HAVE_GC | This has to be set if the object supports garbage collection |
Py_TPFLAGS_DEFAULT | |
Py_TPFLAGS_LONG_SUBCLASS | |
Py_TPFLAGS_LIST_SUBCLASS | |
Py_TPFLAGS_TUPLE_SUBCLASS | |
Py_TPFLAGS_BYTES_SUBCLASS | see below |
Py_TPFLAGS_UNICODE_SUBCLASS | |
Py_TPFLAGS_DICT_SUBCLASS | |
Py_TPFLAGS_BASE_EXC_SUBCLASS | |
Py_TPFLAGS_TYPE_SUBCLASS | |
Py_TPFLAGS_HAVE_FINALIZE | Set when the tp_finalize field is non-empty. See PEP442 for more informations |
The subclass flags should be set when we inherit from one of the base types. They do not do anything special but they are used by the type-specific *check macros to see if an object is a kind of type.
tp_new, tp_init, tp_dealloc are, respectively, used to construct the instance, initialize the instance and destroy the instance internals. You are probably familiar with their use as they’re exposed in Python as __new__, __init__ and __del__.
tp_str is, again, familiar as it is exposed __str__ and is used to support str().
tp_as_sequence is an interesting field. It is used to support the sequence abstract protocol. There are quite a few protocols. They are not that much different from a Java-like interface in concept.
With the sequence protocol, we can support indexing operations and expressions like len. As with many, but not all, other protocols, we simply have to put in the type a reference to a specific structure that points to the requested implementations. For the sequence protocol this structure is PySequenceMethods.
Again, there are a few things we can implement for this structure:
sq_length has signature lenfunc and is used by PySequence_size and PyObject_Size which are both equivalent to Python’s len. It should return the sequence length.
sq_length is used to handle negative indexes in sq_item and sq_ass_item. If a negative index is passed sq_length is called and its result is added to the negative index. If sq_length is missing the index is passed as is.
sq_concat is a binaryfunc that is called by PySequence_Concat and by the + operator if the nb_slot ( __add__ ) is missing. It should return a new sequence containing all elements from the left argument followed by all elements of the right argument. There is an in-place version of concat in sq_inplace_concat that modifies the first argument instead of creating a new sequence.
sq_repeat is an ssizeargfunc that is called by PySequence_Repeat and by the * operator if the nb_multiply (__mul__ ) slot is missing. It should return a new sequence with the elements of the original sequence repeated n times. There is an in-place version of repeat in sq_inplace_repeat that modifies the sequence instead of creating a new one.
sq_item is an ssizeargfunc that is called by PySequence_GetItem and PyObject_GetItem which are equivalent to Python’s index operator. It should return a new reference to the item at the requested index.
sq_ass_item is an ssizeobjargproc that is called by PySequence_SetItem that is called by PyObject_SetItem and PyObject_DelItem if the mp_subscript slot from the mapping protocol is missing. It is equivalent to assigning an item at a certain index in a sequence.
sq_contains is an objobjproc that is called by PySequence_Contains. It is equivalent to the in keyword. If it is not present PySequence_Contains simply travels the sequence linearly to find if the item is contained.
There are two more fields, sq_was_slice and sq_was_ass_slice, but I could not found how and if they are used.
The implementation
The last thing we have to look at is the implementation of all those methods we talked about. As said before the implementation I did was pretty awkward for this first exercise. This is true both from a C API point of view and from a more general coding-style, logic, structure and elegance point of view.
Starting from new:
static PyObject* newArray(PyTypeObject* type, PyTypeObject* storedType, Py_ssize_t size) {
array* self;
self = (array*)type->tp_alloc(type, 0);
if (self == NULL) {
return NULL;
}
self->data = (PyObject**)PyMem_Calloc(sizeof(PyObject*), size);
if (self->data == NULL) {
Py_DECREF(self);
return NULL;
}
self->size = size;
self->storedType = storedType;
Py_INCREF(self->storedType);
return (PyObject*)self;
}
static PyObject* array_new(PyTypeObject* type, PyObject* args, PyObject* kwds) {
if (PyTuple_Size(args) < 2) {
PyErr_Format(PyExc_TypeError, "%s() takes at least 2 arguments (%i given)", __FUNCTION__, PyTuple_Size(args));
return NULL;
}
Py_ssize_t size = PyLong_AsSsize_t(PyTuple_GET_ITEM(args, 0));
PyObject* storedType = PyTuple_GET_ITEM(args, 1);
if (PyErr_Occurred() || !PyType_Check(storedType)) {
PyErr_Format(PyExc_TypeError, "Unsupported operand type(s) for %s: '%s' and '%s'", __FUNCTION__, Py_TYPE(PyTuple_GET_ITEM(args, 0))->tp_name, Py_TYPE(PyTuple_GET_ITEM(args, 1))->tp_name);
return NULL;
}
if (size <= 0) {
PyErr_Format(PyExc_ValueError, "The array size must be a positive integer greater than 0");
return NULL;
}
if (size > (PY_SSIZE_T_MAX / sizeof(PyObject*))) {
return PyErr_NoMemory();
}
return newArray(type, (PyTypeObject*)storedType, size);
}
A new function has type newfunc and is expected to allocate a new instance and do the bare minimum initialization of its members. In the case of array, we allocate as much space for n pointers and set them to zero.
To construct an array, I decided to require at least two arguments, a size and a type, and at most 2 + size arguments. The optional arguments are used to initialize the array values. While the first two arguments are needed to build the instance, the initialization of the values is deferred to the __init__ function.
This is seen at the start of array_new, where we instantly raise an exception if there are less than two arguments. As a quick tangent, this is a good place to see how to parse arguments.
As you can see, we are using PyTuple* macros and functions to access the args argument. As said before we are guaranteed that args is a tuple containing unnamed positional arguments ( or the first and only argument in the case of a method with the METH_O flag set ) and kwds is a dictionary containing the named arguments and their values.
While we don’t use them here, for reasons I will explain in a moment, we have a series of helper functions to unpack the arguments.
PyArg_ParseTuple and PyArg_ParseTupleAndKeywords are the bread and butter of argument parsing.
The former takes the args tuple, a format string and a variable number of arguments that are used to store the parsed arguments, similar to printf. The latter takes, additionally, the kwds object and a keyword list as its second and fourth argument, after the args tuple and then after the format string. A keyword list is a NULL-terminated array of char* that represents the named arguments’ names.
The format string for those functions has quite a few options.
Both of those functions have a correspandant function that accepts a va_list instead of a variable number of arguments, PyArg_VaParse and PyArg_VaParseTupleAndKeywords.
We then have PyArg_UnpackTuple which doesn’t take a format string. It takes, instead, a tuple object, again it will usually be our args parameter, a const char* used as the name for error reporting, two Py_ssize_t min and max, and a variable number of arguments to store the values in. The tuple size should be at least min and no more than max. All parsed arguments have to be stored in PyObject*s.
All of these functions return an int value that is true if it succeeds and false if there was an error.
Here are some examples:
int size;
PyObject* type = NULL;
PyArg_ParseTuple(args, "iO", &size, &type); // We expect a Python Integer and a PyObject
const char* kwds_list[] = { "size", "type", NULL }
PyObject* o1 = NULL;
PyObject* o2 = NULL;
int size;
PyObject* type = NULL;
// We expect two PyObject followed by two keyword argument, which are optional, one int and one PyObject.
PyArg_ParseTupleAndKeyWords(args, kwds, "OO|$iO", kwds_list, &o1, &o2, &size, &type);
PyObject* first = NULL;
PyObject* second = NULL;
// We expecta at least one argument and at most two
PyArg_UnpackTuple(args, "test", 1, 2, &first, &second);
For array_new I’ve not used them as it was a little difficult working with a variable number of argument. For format string parser we should’ve had to build the string dynamically as an error is returned if the args tuple length does not match the format we are giving or the input is not exhausted.
We can’t use PyArg_UnpackTuple with a min of 2 and a max of 2 as it returns an error when the size of the tuple is smaller than the min or bigger than the max. We could do something like using a max that is the theoretical limit that an array can hold based on the size of a PyObject, but we would need to provide as many PyObject to store the arguments as there are arguments in the tuple, which isn’t particularly feasible since we don’t even know how many there are yet.
If we really wanted to use them we could probably do something like slicing the tuple but we would have to deal with a new reference. Furthermore, I think doing this by hand is clearer.
In the end, while those helper functions are really really handy ( and they provide some form of error checking that we may otherwise do by hand ), args and kwds are still PyObjects that we can handle manually.
While PyTuple_GET_ITEM returns a borrowed reference, and we don’t have to manage it, we should probably expand the function to increase the borrowed references and decrease them again before exiting.
PyErr_Occurred checks if an exception is currently set. We are using it to check if anything happened in PyLong_AsSsize_t. Beware that the PyTuple_GET_ITEM macro does no error or range checking contrary to its sister function PyTuple_GetItem.
PyType_Check tells us if an object is of type “type” and is part of the family of *_Check macros.
One thing we are doing here is actually hiding the real exception in case any occurred. I’m not sure this is a good choice, and before I said that it was suggested not to do so, but I think it may be ok here as the exception that should come from Py_AsSsize_t should be because of a wrong argument, making our message, possibly, more explicit.
I must say that I’m not sure if we should actually check for a NULL argument here. While PyLong_AsSsize_t will raise an exception if the object is NULL, PyType_Check will try to do its thing and fail badly. It might actually be a good idea if we want to be on the secure side of things.
After some more error checking we pass the ball to new_array.
array* self;
self = (array*)type->tp_alloc(type, 0);
if (self == NULL) {
return NULL;
}
This is something that you will see in most __new__ functions. We use the type’s alloc function to create the instance and we check if it was actually created.
An alloc function initializes the memory for the instance itself. We can provide a custom one, but we will usually fall on the standard Python’s allocation strategy.
With our instance in hand, we simply have to initialize the needed memory for our array. We do this through Python’s memory interface functions. The API provides our known malloc, free, etc… It does even provide a new for object allocation.
I’m not sure we actually have to increase the reference count of a type object or if it is safe to do so. I could not find a definitive answer.
Moving on, we have to complete the instance initialization with our __init__.
static int array_init(array* self, PyObject* args, PyObject* kwds) {
static const Py_ssize_t MIN_ARGUMENTS = 2;
Py_ssize_t argsSize = PyTuple_Size(args);
if (argsSize > (ARR_SIZE(self) + MIN_ARGUMENTS)) {
PyErr_Format(PyExc_TypeError, "%s() takes at most %i arguments for an array of size %i (%i given)", __FUNCTION__, ARR_SIZE(self) + MIN_ARGUMENTS, ARR_SIZE(self), argsSize);
return -1;
}
for (Py_ssize_t index = 2; index < argsSize; ++index) {
PyObject* tmp = PyTuple_GET_ITEM(args, index);
if (!ARR_CHECK_TYPE(self, tmp)) {
PyErr_Format(PyExc_TypeError, "Unsupported operand type(s) for %s for array of type '%s': '%s'", __FUNCTION__, ARR_STORED_TYPE(self)->tp_name, Py_TYPE(tmp)->tp_name);
return -1;
}
Py_INCREF(tmp);
ARR_ASSIGN(self, index-MIN_ARGUMENTS, tmp);
}
return 0;
}
You should probably know what is going on by now as we are using things we have already seen. As you can see, this time we return a negative number to signify errors as expected from int-returning functions.
The deallocator is pretty simple too:
static void array_dealloc(array* self) {
for (Py_ssize_t index = 0; index < self->size; ++index) {
Py_XDECREF(ARR_GET(self, index));
}
PyMem_Free((void*)self->data);
Py_DECREF(ARR_STORED_TYPE(self));
}
Most of the code should be understandable by now. It is pretty simple. There may be errors here and there that I haven’t seen but I hope this is not the case. There are surely things that can be done in different ways, and maybe should. I particularly dislike the __str__ implementation I wrote which was a testing one that never got refactored ( and it should probably do some error checking by the way ).
Some afterwords
Unfortunately, I used all the allotted time for this post and could not cover some interesting things that I learnt. Initially, I wanted to produce a more tutorial-like post but I completely got lost in my ramblings and wrote more than I could handle on the time I had. I still hope this can be used as a beginner resource for starting out with the C API.
I don’t particularly like working with Python as a language. Nonetheless, working the C API was a really enjoyable experience that shed some lights on how some things work internally in Python and gave me some better tools to appreciate it and its usage.
I’d really like to have the chance, later on, to write some interesting extension modules and to find a use-case for the C API.