blob: 60de586d0bb95469f1ce08a5f30ecc203624ada3 [file] [log] [blame]
Fred Drakec37b65e2001-11-28 07:26:15 +00001\chapter{Extending Python with C or \Cpp \label{intro}}
Fred Drakecc8f44b2001-08-20 19:30:29 +00002
3
4It is quite easy to add new built-in modules to Python, if you know
5how to program in C. Such \dfn{extension modules} can do two things
6that can't be done directly in Python: they can implement new built-in
7object types, and they can call C library functions and system calls.
8
9To support extensions, the Python API (Application Programmers
10Interface) defines a set of functions, macros and variables that
11provide access to most aspects of the Python run-time system. The
12Python API is incorporated in a C source file by including the header
13\code{"Python.h"}.
14
15The compilation of an extension module depends on its intended use as
16well as on your system setup; details are given in later chapters.
17
18
19\section{A Simple Example
20 \label{simpleExample}}
21
22Let's create an extension module called \samp{spam} (the favorite food
23of Monty Python fans...) and let's say we want to create a Python
24interface to the C library function \cfunction{system()}.\footnote{An
25interface for this function already exists in the standard module
26\module{os} --- it was chosen as a simple and straightfoward example.}
27This function takes a null-terminated character string as argument and
28returns an integer. We want this function to be callable from Python
29as follows:
30
31\begin{verbatim}
32>>> import spam
33>>> status = spam.system("ls -l")
34\end{verbatim}
35
36Begin by creating a file \file{spammodule.c}. (Historically, if a
37module is called \samp{spam}, the C file containing its implementation
38is called \file{spammodule.c}; if the module name is very long, like
39\samp{spammify}, the module name can be just \file{spammify.c}.)
40
41The first line of our file can be:
42
43\begin{verbatim}
44#include <Python.h>
45\end{verbatim}
46
47which pulls in the Python API (you can add a comment describing the
48purpose of the module and a copyright notice if you like).
Fred Drake396ca572001-09-06 16:30:30 +000049Since Python may define some pre-processor definitions which affect
50the standard headers on some systems, you must include \file{Python.h}
51before any standard headers are included.
Fred Drakecc8f44b2001-08-20 19:30:29 +000052
Fred Drake396ca572001-09-06 16:30:30 +000053All user-visible symbols defined by \file{Python.h} have a prefix of
Fred Drakecc8f44b2001-08-20 19:30:29 +000054\samp{Py} or \samp{PY}, except those defined in standard header files.
55For convenience, and since they are used extensively by the Python
56interpreter, \code{"Python.h"} includes a few standard header files:
57\code{<stdio.h>}, \code{<string.h>}, \code{<errno.h>}, and
58\code{<stdlib.h>}. If the latter header file does not exist on your
59system, it declares the functions \cfunction{malloc()},
60\cfunction{free()} and \cfunction{realloc()} directly.
61
62The next thing we add to our module file is the C function that will
63be called when the Python expression \samp{spam.system(\var{string})}
64is evaluated (we'll see shortly how it ends up being called):
65
66\begin{verbatim}
67static PyObject *
Fred Drake723f94b2002-06-22 01:42:00 +000068spam_system(PyObject *self, PyObject *args)
Fred Drakecc8f44b2001-08-20 19:30:29 +000069{
70 char *command;
71 int sts;
72
73 if (!PyArg_ParseTuple(args, "s", &command))
74 return NULL;
75 sts = system(command);
76 return Py_BuildValue("i", sts);
77}
78\end{verbatim}
79
80There is a straightforward translation from the argument list in
81Python (for example, the single expression \code{"ls -l"}) to the
82arguments passed to the C function. The C function always has two
83arguments, conventionally named \var{self} and \var{args}.
84
85The \var{self} argument is only used when the C function implements a
86built-in method, not a function. In the example, \var{self} will
87always be a \NULL{} pointer, since we are defining a function, not a
88method. (This is done so that the interpreter doesn't have to
89understand two different types of C functions.)
90
91The \var{args} argument will be a pointer to a Python tuple object
92containing the arguments. Each item of the tuple corresponds to an
93argument in the call's argument list. The arguments are Python
94objects --- in order to do anything with them in our C function we have
95to convert them to C values. The function \cfunction{PyArg_ParseTuple()}
96in the Python API checks the argument types and converts them to C
97values. It uses a template string to determine the required types of
98the arguments as well as the types of the C variables into which to
99store the converted values. More about this later.
100
101\cfunction{PyArg_ParseTuple()} returns true (nonzero) if all arguments have
102the right type and its components have been stored in the variables
103whose addresses are passed. It returns false (zero) if an invalid
104argument list was passed. In the latter case it also raises an
105appropriate exception so the calling function can return
106\NULL{} immediately (as we saw in the example).
107
108
109\section{Intermezzo: Errors and Exceptions
110 \label{errors}}
111
112An important convention throughout the Python interpreter is the
113following: when a function fails, it should set an exception condition
114and return an error value (usually a \NULL{} pointer). Exceptions
115are stored in a static global variable inside the interpreter; if this
116variable is \NULL{} no exception has occurred. A second global
117variable stores the ``associated value'' of the exception (the second
118argument to \keyword{raise}). A third variable contains the stack
119traceback in case the error originated in Python code. These three
120variables are the C equivalents of the Python variables
121\code{sys.exc_type}, \code{sys.exc_value} and \code{sys.exc_traceback} (see
122the section on module \module{sys} in the
123\citetitle[../lib/lib.html]{Python Library Reference}). It is
124important to know about them to understand how errors are passed
125around.
126
127The Python API defines a number of functions to set various types of
128exceptions.
129
130The most common one is \cfunction{PyErr_SetString()}. Its arguments
131are an exception object and a C string. The exception object is
132usually a predefined object like \cdata{PyExc_ZeroDivisionError}. The
133C string indicates the cause of the error and is converted to a
134Python string object and stored as the ``associated value'' of the
135exception.
136
137Another useful function is \cfunction{PyErr_SetFromErrno()}, which only
138takes an exception argument and constructs the associated value by
139inspection of the global variable \cdata{errno}. The most
140general function is \cfunction{PyErr_SetObject()}, which takes two object
141arguments, the exception and its associated value. You don't need to
142\cfunction{Py_INCREF()} the objects passed to any of these functions.
143
144You can test non-destructively whether an exception has been set with
145\cfunction{PyErr_Occurred()}. This returns the current exception object,
146or \NULL{} if no exception has occurred. You normally don't need
147to call \cfunction{PyErr_Occurred()} to see whether an error occurred in a
148function call, since you should be able to tell from the return value.
149
150When a function \var{f} that calls another function \var{g} detects
151that the latter fails, \var{f} should itself return an error value
152(usually \NULL{} or \code{-1}). It should \emph{not} call one of the
153\cfunction{PyErr_*()} functions --- one has already been called by \var{g}.
154\var{f}'s caller is then supposed to also return an error indication
155to \emph{its} caller, again \emph{without} calling \cfunction{PyErr_*()},
156and so on --- the most detailed cause of the error was already
157reported by the function that first detected it. Once the error
158reaches the Python interpreter's main loop, this aborts the currently
159executing Python code and tries to find an exception handler specified
160by the Python programmer.
161
162(There are situations where a module can actually give a more detailed
163error message by calling another \cfunction{PyErr_*()} function, and in
164such cases it is fine to do so. As a general rule, however, this is
165not necessary, and can cause information about the cause of the error
166to be lost: most operations can fail for a variety of reasons.)
167
168To ignore an exception set by a function call that failed, the exception
169condition must be cleared explicitly by calling \cfunction{PyErr_Clear()}.
170The only time C code should call \cfunction{PyErr_Clear()} is if it doesn't
171want to pass the error on to the interpreter but wants to handle it
172completely by itself (possibly by trying something else, or pretending
173nothing went wrong).
174
175Every failing \cfunction{malloc()} call must be turned into an
176exception --- the direct caller of \cfunction{malloc()} (or
177\cfunction{realloc()}) must call \cfunction{PyErr_NoMemory()} and
178return a failure indicator itself. All the object-creating functions
179(for example, \cfunction{PyInt_FromLong()}) already do this, so this
180note is only relevant to those who call \cfunction{malloc()} directly.
181
182Also note that, with the important exception of
183\cfunction{PyArg_ParseTuple()} and friends, functions that return an
184integer status usually return a positive value or zero for success and
185\code{-1} for failure, like \UNIX{} system calls.
186
187Finally, be careful to clean up garbage (by making
188\cfunction{Py_XDECREF()} or \cfunction{Py_DECREF()} calls for objects
189you have already created) when you return an error indicator!
190
191The choice of which exception to raise is entirely yours. There are
192predeclared C objects corresponding to all built-in Python exceptions,
193such as \cdata{PyExc_ZeroDivisionError}, which you can use directly.
194Of course, you should choose exceptions wisely --- don't use
195\cdata{PyExc_TypeError} to mean that a file couldn't be opened (that
196should probably be \cdata{PyExc_IOError}). If something's wrong with
197the argument list, the \cfunction{PyArg_ParseTuple()} function usually
198raises \cdata{PyExc_TypeError}. If you have an argument whose value
199must be in a particular range or must satisfy other conditions,
200\cdata{PyExc_ValueError} is appropriate.
201
202You can also define a new exception that is unique to your module.
203For this, you usually declare a static object variable at the
204beginning of your file:
205
206\begin{verbatim}
207static PyObject *SpamError;
208\end{verbatim}
209
210and initialize it in your module's initialization function
211(\cfunction{initspam()}) with an exception object (leaving out
212the error checking for now):
213
214\begin{verbatim}
215void
Fred Drakeef6373a2001-11-17 06:50:42 +0000216initspam(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +0000217{
Fred Drake63e40a52002-04-12 19:08:31 +0000218 PyObject *m;
Fred Drakecc8f44b2001-08-20 19:30:29 +0000219
220 m = Py_InitModule("spam", SpamMethods);
Fred Drake63e40a52002-04-12 19:08:31 +0000221
Fred Drakecc8f44b2001-08-20 19:30:29 +0000222 SpamError = PyErr_NewException("spam.error", NULL, NULL);
Fred Drake63e40a52002-04-12 19:08:31 +0000223 Py_INCREF(SpamError);
224 PyModule_AddObject(m, "error", SpamError);
Fred Drakecc8f44b2001-08-20 19:30:29 +0000225}
226\end{verbatim}
227
228Note that the Python name for the exception object is
229\exception{spam.error}. The \cfunction{PyErr_NewException()} function
230may create a class with the base class being \exception{Exception}
231(unless another class is passed in instead of \NULL), described in the
232\citetitle[../lib/lib.html]{Python Library Reference} under ``Built-in
233Exceptions.''
234
235Note also that the \cdata{SpamError} variable retains a reference to
236the newly created exception class; this is intentional! Since the
237exception could be removed from the module by external code, an owned
238reference to the class is needed to ensure that it will not be
239discarded, causing \cdata{SpamError} to become a dangling pointer.
240Should it become a dangling pointer, C code which raises the exception
241could cause a core dump or other unintended side effects.
242
243
244\section{Back to the Example
245 \label{backToExample}}
246
247Going back to our example function, you should now be able to
248understand this statement:
249
250\begin{verbatim}
251 if (!PyArg_ParseTuple(args, "s", &command))
252 return NULL;
253\end{verbatim}
254
255It returns \NULL{} (the error indicator for functions returning
256object pointers) if an error is detected in the argument list, relying
257on the exception set by \cfunction{PyArg_ParseTuple()}. Otherwise the
258string value of the argument has been copied to the local variable
259\cdata{command}. This is a pointer assignment and you are not supposed
260to modify the string to which it points (so in Standard C, the variable
261\cdata{command} should properly be declared as \samp{const char
262*command}).
263
264The next statement is a call to the \UNIX{} function
265\cfunction{system()}, passing it the string we just got from
266\cfunction{PyArg_ParseTuple()}:
267
268\begin{verbatim}
269 sts = system(command);
270\end{verbatim}
271
272Our \function{spam.system()} function must return the value of
273\cdata{sts} as a Python object. This is done using the function
274\cfunction{Py_BuildValue()}, which is something like the inverse of
275\cfunction{PyArg_ParseTuple()}: it takes a format string and an
276arbitrary number of C values, and returns a new Python object.
277More info on \cfunction{Py_BuildValue()} is given later.
278
279\begin{verbatim}
280 return Py_BuildValue("i", sts);
281\end{verbatim}
282
283In this case, it will return an integer object. (Yes, even integers
284are objects on the heap in Python!)
285
286If you have a C function that returns no useful argument (a function
287returning \ctype{void}), the corresponding Python function must return
288\code{None}. You need this idiom to do so:
289
290\begin{verbatim}
291 Py_INCREF(Py_None);
292 return Py_None;
293\end{verbatim}
294
295\cdata{Py_None} is the C name for the special Python object
296\code{None}. It is a genuine Python object rather than a \NULL{}
297pointer, which means ``error'' in most contexts, as we have seen.
298
299
300\section{The Module's Method Table and Initialization Function
301 \label{methodTable}}
302
303I promised to show how \cfunction{spam_system()} is called from Python
304programs. First, we need to list its name and address in a ``method
305table'':
306
307\begin{verbatim}
308static PyMethodDef SpamMethods[] = {
309 ...
Fred Drakeef6373a2001-11-17 06:50:42 +0000310 {"system", spam_system, METH_VARARGS,
311 "Execute a shell command."},
Fred Drakecc8f44b2001-08-20 19:30:29 +0000312 ...
Fred Drakeef6373a2001-11-17 06:50:42 +0000313 {NULL, NULL, 0, NULL} /* Sentinel */
Fred Drakecc8f44b2001-08-20 19:30:29 +0000314};
315\end{verbatim}
316
317Note the third entry (\samp{METH_VARARGS}). This is a flag telling
318the interpreter the calling convention to be used for the C
319function. It should normally always be \samp{METH_VARARGS} or
320\samp{METH_VARARGS | METH_KEYWORDS}; a value of \code{0} means that an
321obsolete variant of \cfunction{PyArg_ParseTuple()} is used.
322
323When using only \samp{METH_VARARGS}, the function should expect
324the Python-level parameters to be passed in as a tuple acceptable for
325parsing via \cfunction{PyArg_ParseTuple()}; more information on this
326function is provided below.
327
328The \constant{METH_KEYWORDS} bit may be set in the third field if
329keyword arguments should be passed to the function. In this case, the
330C function should accept a third \samp{PyObject *} parameter which
331will be a dictionary of keywords. Use
332\cfunction{PyArg_ParseTupleAndKeywords()} to parse the arguments to
333such a function.
334
335The method table must be passed to the interpreter in the module's
336initialization function. The initialization function must be named
337\cfunction{init\var{name}()}, where \var{name} is the name of the
338module, and should be the only non-\keyword{static} item defined in
339the module file:
340
341\begin{verbatim}
342void
Fred Drakeef6373a2001-11-17 06:50:42 +0000343initspam(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +0000344{
345 (void) Py_InitModule("spam", SpamMethods);
346}
347\end{verbatim}
348
349Note that for \Cpp, this method must be declared \code{extern "C"}.
350
351When the Python program imports module \module{spam} for the first
352time, \cfunction{initspam()} is called. (See below for comments about
353embedding Python.) It calls
354\cfunction{Py_InitModule()}, which creates a ``module object'' (which
355is inserted in the dictionary \code{sys.modules} under the key
356\code{"spam"}), and inserts built-in function objects into the newly
357created module based upon the table (an array of \ctype{PyMethodDef}
358structures) that was passed as its second argument.
359\cfunction{Py_InitModule()} returns a pointer to the module object
360that it creates (which is unused here). It aborts with a fatal error
361if the module could not be initialized satisfactorily, so the caller
362doesn't need to check for errors.
363
364When embedding Python, the \cfunction{initspam()} function is not
365called automatically unless there's an entry in the
366\cdata{_PyImport_Inittab} table. The easiest way to handle this is to
367statically initialize your statically-linked modules by directly
368calling \cfunction{initspam()} after the call to
369\cfunction{Py_Initialize()} or \cfunction{PyMac_Initialize()}:
370
371\begin{verbatim}
Fred Drake723f94b2002-06-22 01:42:00 +0000372int
373main(int argc, char *argv[])
Fred Drakecc8f44b2001-08-20 19:30:29 +0000374{
375 /* Pass argv[0] to the Python interpreter */
376 Py_SetProgramName(argv[0]);
377
378 /* Initialize the Python interpreter. Required. */
379 Py_Initialize();
380
381 /* Add a static module */
382 initspam();
383\end{verbatim}
384
385An example may be found in the file \file{Demo/embed/demo.c} in the
386Python source distribution.
387
Fred Drake0aa811c2001-10-20 04:24:09 +0000388\note{Removing entries from \code{sys.modules} or importing
Fred Drakecc8f44b2001-08-20 19:30:29 +0000389compiled modules into multiple interpreters within a process (or
390following a \cfunction{fork()} without an intervening
391\cfunction{exec()}) can create problems for some extension modules.
392Extension module authors should exercise caution when initializing
393internal data structures.
394Note also that the \function{reload()} function can be used with
395extension modules, and will call the module initialization function
396(\cfunction{initspam()} in the example), but will not load the module
397again if it was loaded from a dynamically loadable object file
Fred Drake0aa811c2001-10-20 04:24:09 +0000398(\file{.so} on \UNIX, \file{.dll} on Windows).}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000399
400A more substantial example module is included in the Python source
401distribution as \file{Modules/xxmodule.c}. This file may be used as a
402template or simply read as an example. The \program{modulator.py}
403script included in the source distribution or Windows install provides
404a simple graphical user interface for declaring the functions and
405objects which a module should implement, and can generate a template
406which can be filled in. The script lives in the
407\file{Tools/modulator/} directory; see the \file{README} file there
408for more information.
409
410
411\section{Compilation and Linkage
412 \label{compilation}}
413
414There are two more things to do before you can use your new extension:
415compiling and linking it with the Python system. If you use dynamic
Fred Drake4e765552002-05-16 13:48:14 +0000416loading, the details may depend on the style of dynamic loading your
417system uses; see the chapters about building extension modules
418(chapter \ref{building}) and additional information that pertains only
419to building on Windows (chapter \ref{building-on-windows}) for more
420information about this.
421% XXX Add information about Mac OS
Fred Drakecc8f44b2001-08-20 19:30:29 +0000422
423If you can't use dynamic loading, or if you want to make your module a
424permanent part of the Python interpreter, you will have to change the
425configuration setup and rebuild the interpreter. Luckily, this is
Fred Drake4e765552002-05-16 13:48:14 +0000426very simple on \UNIX: just place your file (\file{spammodule.c} for
427example) in the \file{Modules/} directory of an unpacked source
428distribution, add a line to the file \file{Modules/Setup.local}
429describing your file:
Fred Drakecc8f44b2001-08-20 19:30:29 +0000430
431\begin{verbatim}
432spam spammodule.o
433\end{verbatim}
434
435and rebuild the interpreter by running \program{make} in the toplevel
436directory. You can also run \program{make} in the \file{Modules/}
437subdirectory, but then you must first rebuild \file{Makefile}
438there by running `\program{make} Makefile'. (This is necessary each
439time you change the \file{Setup} file.)
440
441If your module requires additional libraries to link with, these can
442be listed on the line in the configuration file as well, for instance:
443
444\begin{verbatim}
445spam spammodule.o -lX11
446\end{verbatim}
447
448\section{Calling Python Functions from C
449 \label{callingPython}}
450
451So far we have concentrated on making C functions callable from
452Python. The reverse is also useful: calling Python functions from C.
453This is especially the case for libraries that support so-called
454``callback'' functions. If a C interface makes use of callbacks, the
455equivalent Python often needs to provide a callback mechanism to the
456Python programmer; the implementation will require calling the Python
457callback functions from a C callback. Other uses are also imaginable.
458
459Fortunately, the Python interpreter is easily called recursively, and
460there is a standard interface to call a Python function. (I won't
461dwell on how to call the Python parser with a particular string as
462input --- if you're interested, have a look at the implementation of
463the \programopt{-c} command line option in \file{Python/pythonmain.c}
464from the Python source code.)
465
466Calling a Python function is easy. First, the Python program must
467somehow pass you the Python function object. You should provide a
468function (or some other interface) to do this. When this function is
469called, save a pointer to the Python function object (be careful to
470\cfunction{Py_INCREF()} it!) in a global variable --- or wherever you
471see fit. For example, the following function might be part of a module
472definition:
473
474\begin{verbatim}
475static PyObject *my_callback = NULL;
476
477static PyObject *
Fred Drake723f94b2002-06-22 01:42:00 +0000478my_set_callback(PyObject *dummy, PyObject *args)
Fred Drakecc8f44b2001-08-20 19:30:29 +0000479{
480 PyObject *result = NULL;
481 PyObject *temp;
482
483 if (PyArg_ParseTuple(args, "O:set_callback", &temp)) {
484 if (!PyCallable_Check(temp)) {
485 PyErr_SetString(PyExc_TypeError, "parameter must be callable");
486 return NULL;
487 }
488 Py_XINCREF(temp); /* Add a reference to new callback */
489 Py_XDECREF(my_callback); /* Dispose of previous callback */
490 my_callback = temp; /* Remember new callback */
491 /* Boilerplate to return "None" */
492 Py_INCREF(Py_None);
493 result = Py_None;
494 }
495 return result;
496}
497\end{verbatim}
498
499This function must be registered with the interpreter using the
500\constant{METH_VARARGS} flag; this is described in section
501\ref{methodTable}, ``The Module's Method Table and Initialization
502Function.'' The \cfunction{PyArg_ParseTuple()} function and its
Fred Drakec37b65e2001-11-28 07:26:15 +0000503arguments are documented in section~\ref{parseTuple}, ``Extracting
Fred Drakecc8f44b2001-08-20 19:30:29 +0000504Parameters in Extension Functions.''
505
506The macros \cfunction{Py_XINCREF()} and \cfunction{Py_XDECREF()}
507increment/decrement the reference count of an object and are safe in
508the presence of \NULL{} pointers (but note that \var{temp} will not be
Fred Drakec37b65e2001-11-28 07:26:15 +0000509\NULL{} in this context). More info on them in
510section~\ref{refcounts}, ``Reference Counts.''
Fred Drakecc8f44b2001-08-20 19:30:29 +0000511
512Later, when it is time to call the function, you call the C function
Fred Drake99181ac2001-11-29 05:02:34 +0000513\cfunction{PyEval_CallObject()}.\ttindex{PyEval_CallObject()} This
514function has two arguments, both pointers to arbitrary Python objects:
515the Python function, and the argument list. The argument list must
516always be a tuple object, whose length is the number of arguments. To
517call the Python function with no arguments, pass an empty tuple; to
518call it with one argument, pass a singleton tuple.
519\cfunction{Py_BuildValue()} returns a tuple when its format string
520consists of zero or more format codes between parentheses. For
521example:
Fred Drakecc8f44b2001-08-20 19:30:29 +0000522
523\begin{verbatim}
524 int arg;
525 PyObject *arglist;
526 PyObject *result;
527 ...
528 arg = 123;
529 ...
530 /* Time to call the callback */
531 arglist = Py_BuildValue("(i)", arg);
532 result = PyEval_CallObject(my_callback, arglist);
533 Py_DECREF(arglist);
534\end{verbatim}
535
536\cfunction{PyEval_CallObject()} returns a Python object pointer: this is
537the return value of the Python function. \cfunction{PyEval_CallObject()} is
538``reference-count-neutral'' with respect to its arguments. In the
539example a new tuple was created to serve as the argument list, which
540is \cfunction{Py_DECREF()}-ed immediately after the call.
541
542The return value of \cfunction{PyEval_CallObject()} is ``new'': either it
543is a brand new object, or it is an existing object whose reference
544count has been incremented. So, unless you want to save it in a
545global variable, you should somehow \cfunction{Py_DECREF()} the result,
546even (especially!) if you are not interested in its value.
547
548Before you do this, however, it is important to check that the return
Fred Drakec37b65e2001-11-28 07:26:15 +0000549value isn't \NULL. If it is, the Python function terminated by
Fred Drakecc8f44b2001-08-20 19:30:29 +0000550raising an exception. If the C code that called
551\cfunction{PyEval_CallObject()} is called from Python, it should now
552return an error indication to its Python caller, so the interpreter
553can print a stack trace, or the calling Python code can handle the
554exception. If this is not possible or desirable, the exception should
555be cleared by calling \cfunction{PyErr_Clear()}. For example:
556
557\begin{verbatim}
558 if (result == NULL)
559 return NULL; /* Pass error back */
560 ...use result...
561 Py_DECREF(result);
562\end{verbatim}
563
564Depending on the desired interface to the Python callback function,
565you may also have to provide an argument list to
566\cfunction{PyEval_CallObject()}. In some cases the argument list is
567also provided by the Python program, through the same interface that
568specified the callback function. It can then be saved and used in the
569same manner as the function object. In other cases, you may have to
570construct a new tuple to pass as the argument list. The simplest way
571to do this is to call \cfunction{Py_BuildValue()}. For example, if
572you want to pass an integral event code, you might use the following
573code:
574
575\begin{verbatim}
576 PyObject *arglist;
577 ...
578 arglist = Py_BuildValue("(l)", eventcode);
579 result = PyEval_CallObject(my_callback, arglist);
580 Py_DECREF(arglist);
581 if (result == NULL)
582 return NULL; /* Pass error back */
583 /* Here maybe use the result */
584 Py_DECREF(result);
585\end{verbatim}
586
587Note the placement of \samp{Py_DECREF(arglist)} immediately after the
588call, before the error check! Also note that strictly spoken this
589code is not complete: \cfunction{Py_BuildValue()} may run out of
590memory, and this should be checked.
591
592
593\section{Extracting Parameters in Extension Functions
594 \label{parseTuple}}
595
Fred Drakee9fba912002-03-28 22:36:56 +0000596\ttindex{PyArg_ParseTuple()}
597
Fred Drakecc8f44b2001-08-20 19:30:29 +0000598The \cfunction{PyArg_ParseTuple()} function is declared as follows:
599
600\begin{verbatim}
601int PyArg_ParseTuple(PyObject *arg, char *format, ...);
602\end{verbatim}
603
604The \var{arg} argument must be a tuple object containing an argument
605list passed from Python to a C function. The \var{format} argument
Fred Drake68304cc2002-04-05 23:01:14 +0000606must be a format string, whose syntax is explained in
607``\ulink{Parsing arguments and building
608values}{../api/arg-parsing.html}'' in the
609\citetitle[../api/api.html]{Python/C API Reference Manual}. The
Fred Drakecc8f44b2001-08-20 19:30:29 +0000610remaining arguments must be addresses of variables whose type is
Fred Drake68304cc2002-04-05 23:01:14 +0000611determined by the format string.
Fred Drakecc8f44b2001-08-20 19:30:29 +0000612
613Note that while \cfunction{PyArg_ParseTuple()} checks that the Python
614arguments have the required types, it cannot check the validity of the
615addresses of C variables passed to the call: if you make mistakes
616there, your code will probably crash or at least overwrite random bits
617in memory. So be careful!
618
Fred Drakecc8f44b2001-08-20 19:30:29 +0000619Note that any Python object references which are provided to the
620caller are \emph{borrowed} references; do not decrement their
621reference count!
622
Fred Drakecc8f44b2001-08-20 19:30:29 +0000623Some example calls:
624
625\begin{verbatim}
626 int ok;
627 int i, j;
628 long k, l;
629 char *s;
630 int size;
631
632 ok = PyArg_ParseTuple(args, ""); /* No arguments */
633 /* Python call: f() */
634\end{verbatim}
635
636\begin{verbatim}
637 ok = PyArg_ParseTuple(args, "s", &s); /* A string */
638 /* Possible Python call: f('whoops!') */
639\end{verbatim}
640
641\begin{verbatim}
642 ok = PyArg_ParseTuple(args, "lls", &k, &l, &s); /* Two longs and a string */
643 /* Possible Python call: f(1, 2, 'three') */
644\end{verbatim}
645
646\begin{verbatim}
647 ok = PyArg_ParseTuple(args, "(ii)s#", &i, &j, &s, &size);
648 /* A pair of ints and a string, whose size is also returned */
649 /* Possible Python call: f((1, 2), 'three') */
650\end{verbatim}
651
652\begin{verbatim}
653 {
654 char *file;
655 char *mode = "r";
656 int bufsize = 0;
657 ok = PyArg_ParseTuple(args, "s|si", &file, &mode, &bufsize);
658 /* A string, and optionally another string and an integer */
659 /* Possible Python calls:
660 f('spam')
661 f('spam', 'w')
662 f('spam', 'wb', 100000) */
663 }
664\end{verbatim}
665
666\begin{verbatim}
667 {
668 int left, top, right, bottom, h, v;
669 ok = PyArg_ParseTuple(args, "((ii)(ii))(ii)",
670 &left, &top, &right, &bottom, &h, &v);
671 /* A rectangle and a point */
672 /* Possible Python call:
673 f(((0, 0), (400, 300)), (10, 10)) */
674 }
675\end{verbatim}
676
677\begin{verbatim}
678 {
679 Py_complex c;
680 ok = PyArg_ParseTuple(args, "D:myfunction", &c);
681 /* a complex, also providing a function name for errors */
682 /* Possible Python call: myfunction(1+2j) */
683 }
684\end{verbatim}
685
686
687\section{Keyword Parameters for Extension Functions
688 \label{parseTupleAndKeywords}}
689
Fred Drakee9fba912002-03-28 22:36:56 +0000690\ttindex{PyArg_ParseTupleAndKeywords()}
691
Fred Drakecc8f44b2001-08-20 19:30:29 +0000692The \cfunction{PyArg_ParseTupleAndKeywords()} function is declared as
693follows:
694
695\begin{verbatim}
696int PyArg_ParseTupleAndKeywords(PyObject *arg, PyObject *kwdict,
Fred Drake723f94b2002-06-22 01:42:00 +0000697 char *format, char *kwlist[], ...);
Fred Drakecc8f44b2001-08-20 19:30:29 +0000698\end{verbatim}
699
700The \var{arg} and \var{format} parameters are identical to those of the
701\cfunction{PyArg_ParseTuple()} function. The \var{kwdict} parameter
702is the dictionary of keywords received as the third parameter from the
Fred Drakec37b65e2001-11-28 07:26:15 +0000703Python runtime. The \var{kwlist} parameter is a \NULL-terminated
Fred Drakecc8f44b2001-08-20 19:30:29 +0000704list of strings which identify the parameters; the names are matched
705with the type information from \var{format} from left to right. On
706success, \cfunction{PyArg_ParseTupleAndKeywords()} returns true,
707otherwise it returns false and raises an appropriate exception.
708
Fred Drake0aa811c2001-10-20 04:24:09 +0000709\note{Nested tuples cannot be parsed when using keyword
Fred Drakecc8f44b2001-08-20 19:30:29 +0000710arguments! Keyword parameters passed in which are not present in the
Fred Drake0aa811c2001-10-20 04:24:09 +0000711\var{kwlist} will cause \exception{TypeError} to be raised.}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000712
713Here is an example module which uses keywords, based on an example by
714Geoff Philbrick (\email{philbrick@hks.com}):%
715\index{Philbrick, Geoff}
716
717\begin{verbatim}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000718#include "Python.h"
719
720static PyObject *
Fred Drake723f94b2002-06-22 01:42:00 +0000721keywdarg_parrot(PyObject *self, PyObject *args, PyObject *keywds)
Fred Drakecc8f44b2001-08-20 19:30:29 +0000722{
723 int voltage;
724 char *state = "a stiff";
725 char *action = "voom";
726 char *type = "Norwegian Blue";
727
728 static char *kwlist[] = {"voltage", "state", "action", "type", NULL};
729
730 if (!PyArg_ParseTupleAndKeywords(args, keywds, "i|sss", kwlist,
731 &voltage, &state, &action, &type))
732 return NULL;
733
734 printf("-- This parrot wouldn't %s if you put %i Volts through it.\n",
735 action, voltage);
736 printf("-- Lovely plumage, the %s -- It's %s!\n", type, state);
737
738 Py_INCREF(Py_None);
739
740 return Py_None;
741}
742
743static PyMethodDef keywdarg_methods[] = {
744 /* The cast of the function is necessary since PyCFunction values
745 * only take two PyObject* parameters, and keywdarg_parrot() takes
746 * three.
747 */
Fred Drake31f84832002-03-28 20:19:23 +0000748 {"parrot", (PyCFunction)keywdarg_parrot, METH_VARARGS | METH_KEYWORDS,
Fred Drakeef6373a2001-11-17 06:50:42 +0000749 "Print a lovely skit to standard output."},
750 {NULL, NULL, 0, NULL} /* sentinel */
Fred Drakecc8f44b2001-08-20 19:30:29 +0000751};
Fred Drake31f84832002-03-28 20:19:23 +0000752\end{verbatim}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000753
Fred Drake31f84832002-03-28 20:19:23 +0000754\begin{verbatim}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000755void
Fred Drakeef6373a2001-11-17 06:50:42 +0000756initkeywdarg(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +0000757{
758 /* Create the module and add the functions */
759 Py_InitModule("keywdarg", keywdarg_methods);
760}
761\end{verbatim}
762
763
764\section{Building Arbitrary Values
765 \label{buildValue}}
766
767This function is the counterpart to \cfunction{PyArg_ParseTuple()}. It is
768declared as follows:
769
770\begin{verbatim}
771PyObject *Py_BuildValue(char *format, ...);
772\end{verbatim}
773
774It recognizes a set of format units similar to the ones recognized by
775\cfunction{PyArg_ParseTuple()}, but the arguments (which are input to the
776function, not output) must not be pointers, just values. It returns a
777new Python object, suitable for returning from a C function called
778from Python.
779
780One difference with \cfunction{PyArg_ParseTuple()}: while the latter
781requires its first argument to be a tuple (since Python argument lists
782are always represented as tuples internally),
783\cfunction{Py_BuildValue()} does not always build a tuple. It builds
784a tuple only if its format string contains two or more format units.
785If the format string is empty, it returns \code{None}; if it contains
786exactly one format unit, it returns whatever object is described by
787that format unit. To force it to return a tuple of size 0 or one,
788parenthesize the format string.
789
Fred Drakecc8f44b2001-08-20 19:30:29 +0000790Examples (to the left the call, to the right the resulting Python value):
791
792\begin{verbatim}
793 Py_BuildValue("") None
794 Py_BuildValue("i", 123) 123
795 Py_BuildValue("iii", 123, 456, 789) (123, 456, 789)
796 Py_BuildValue("s", "hello") 'hello'
797 Py_BuildValue("ss", "hello", "world") ('hello', 'world')
798 Py_BuildValue("s#", "hello", 4) 'hell'
799 Py_BuildValue("()") ()
800 Py_BuildValue("(i)", 123) (123,)
801 Py_BuildValue("(ii)", 123, 456) (123, 456)
802 Py_BuildValue("(i,i)", 123, 456) (123, 456)
803 Py_BuildValue("[i,i]", 123, 456) [123, 456]
804 Py_BuildValue("{s:i,s:i}",
805 "abc", 123, "def", 456) {'abc': 123, 'def': 456}
806 Py_BuildValue("((ii)(ii)) (ii)",
807 1, 2, 3, 4, 5, 6) (((1, 2), (3, 4)), (5, 6))
808\end{verbatim}
809
810
811\section{Reference Counts
812 \label{refcounts}}
813
Fred Drakec37b65e2001-11-28 07:26:15 +0000814In languages like C or \Cpp, the programmer is responsible for
Fred Drakecc8f44b2001-08-20 19:30:29 +0000815dynamic allocation and deallocation of memory on the heap. In C,
816this is done using the functions \cfunction{malloc()} and
Fred Drakec37b65e2001-11-28 07:26:15 +0000817\cfunction{free()}. In \Cpp, the operators \keyword{new} and
Fred Drakecc8f44b2001-08-20 19:30:29 +0000818\keyword{delete} are used with essentially the same meaning; they are
819actually implemented using \cfunction{malloc()} and
820\cfunction{free()}, so we'll restrict the following discussion to the
821latter.
822
823Every block of memory allocated with \cfunction{malloc()} should
824eventually be returned to the pool of available memory by exactly one
825call to \cfunction{free()}. It is important to call
826\cfunction{free()} at the right time. If a block's address is
827forgotten but \cfunction{free()} is not called for it, the memory it
828occupies cannot be reused until the program terminates. This is
829called a \dfn{memory leak}. On the other hand, if a program calls
830\cfunction{free()} for a block and then continues to use the block, it
831creates a conflict with re-use of the block through another
832\cfunction{malloc()} call. This is called \dfn{using freed memory}.
833It has the same bad consequences as referencing uninitialized data ---
834core dumps, wrong results, mysterious crashes.
835
836Common causes of memory leaks are unusual paths through the code. For
837instance, a function may allocate a block of memory, do some
838calculation, and then free the block again. Now a change in the
839requirements for the function may add a test to the calculation that
840detects an error condition and can return prematurely from the
841function. It's easy to forget to free the allocated memory block when
842taking this premature exit, especially when it is added later to the
843code. Such leaks, once introduced, often go undetected for a long
844time: the error exit is taken only in a small fraction of all calls,
845and most modern machines have plenty of virtual memory, so the leak
846only becomes apparent in a long-running process that uses the leaking
847function frequently. Therefore, it's important to prevent leaks from
848happening by having a coding convention or strategy that minimizes
849this kind of errors.
850
851Since Python makes heavy use of \cfunction{malloc()} and
852\cfunction{free()}, it needs a strategy to avoid memory leaks as well
853as the use of freed memory. The chosen method is called
854\dfn{reference counting}. The principle is simple: every object
855contains a counter, which is incremented when a reference to the
856object is stored somewhere, and which is decremented when a reference
857to it is deleted. When the counter reaches zero, the last reference
858to the object has been deleted and the object is freed.
859
860An alternative strategy is called \dfn{automatic garbage collection}.
861(Sometimes, reference counting is also referred to as a garbage
862collection strategy, hence my use of ``automatic'' to distinguish the
863two.) The big advantage of automatic garbage collection is that the
864user doesn't need to call \cfunction{free()} explicitly. (Another claimed
865advantage is an improvement in speed or memory usage --- this is no
866hard fact however.) The disadvantage is that for C, there is no
867truly portable automatic garbage collector, while reference counting
868can be implemented portably (as long as the functions \cfunction{malloc()}
869and \cfunction{free()} are available --- which the C Standard guarantees).
870Maybe some day a sufficiently portable automatic garbage collector
871will be available for C. Until then, we'll have to live with
872reference counts.
873
Fred Drake024e6472001-12-07 17:30:40 +0000874While Python uses the traditional reference counting implementation,
875it also offers a cycle detector that works to detect reference
876cycles. This allows applications to not worry about creating direct
877or indirect circular references; these are the weakness of garbage
878collection implemented using only reference counting. Reference
879cycles consist of objects which contain (possibly indirect) references
Tim Peters874c4f02001-12-07 17:51:41 +0000880to themselves, so that each object in the cycle has a reference count
Fred Drake024e6472001-12-07 17:30:40 +0000881which is non-zero. Typical reference counting implementations are not
Tim Peters874c4f02001-12-07 17:51:41 +0000882able to reclaim the memory belonging to any objects in a reference
Fred Drake024e6472001-12-07 17:30:40 +0000883cycle, or referenced from the objects in the cycle, even though there
884are no further references to the cycle itself.
885
886The cycle detector is able to detect garbage cycles and can reclaim
887them so long as there are no finalizers implemented in Python
888(\method{__del__()} methods). When there are such finalizers, the
889detector exposes the cycles through the \ulink{\module{gc}
Guido van Rossum44b3f762001-12-07 17:57:56 +0000890module}{../lib/module-gc.html} (specifically, the \code{garbage}
891variable in that module). The \module{gc} module also exposes a way
892to run the detector (the \function{collect()} function), as well as
Fred Drake024e6472001-12-07 17:30:40 +0000893configuration interfaces and the ability to disable the detector at
894runtime. The cycle detector is considered an optional component;
Guido van Rossum44b3f762001-12-07 17:57:56 +0000895though it is included by default, it can be disabled at build time
Fred Drake024e6472001-12-07 17:30:40 +0000896using the \longprogramopt{without-cycle-gc} option to the
897\program{configure} script on \UNIX{} platforms (including Mac OS X)
898or by removing the definition of \code{WITH_CYCLE_GC} in the
899\file{pyconfig.h} header on other platforms. If the cycle detector is
900disabled in this way, the \module{gc} module will not be available.
901
902
Fred Drakecc8f44b2001-08-20 19:30:29 +0000903\subsection{Reference Counting in Python
904 \label{refcountsInPython}}
905
906There are two macros, \code{Py_INCREF(x)} and \code{Py_DECREF(x)},
907which handle the incrementing and decrementing of the reference count.
908\cfunction{Py_DECREF()} also frees the object when the count reaches zero.
909For flexibility, it doesn't call \cfunction{free()} directly --- rather, it
910makes a call through a function pointer in the object's \dfn{type
911object}. For this purpose (and others), every object also contains a
912pointer to its type object.
913
914The big question now remains: when to use \code{Py_INCREF(x)} and
915\code{Py_DECREF(x)}? Let's first introduce some terms. Nobody
916``owns'' an object; however, you can \dfn{own a reference} to an
917object. An object's reference count is now defined as the number of
918owned references to it. The owner of a reference is responsible for
919calling \cfunction{Py_DECREF()} when the reference is no longer
920needed. Ownership of a reference can be transferred. There are three
921ways to dispose of an owned reference: pass it on, store it, or call
922\cfunction{Py_DECREF()}. Forgetting to dispose of an owned reference
923creates a memory leak.
924
925It is also possible to \dfn{borrow}\footnote{The metaphor of
926``borrowing'' a reference is not completely correct: the owner still
927has a copy of the reference.} a reference to an object. The borrower
928of a reference should not call \cfunction{Py_DECREF()}. The borrower must
929not hold on to the object longer than the owner from which it was
930borrowed. Using a borrowed reference after the owner has disposed of
931it risks using freed memory and should be avoided
932completely.\footnote{Checking that the reference count is at least 1
933\strong{does not work} --- the reference count itself could be in
934freed memory and may thus be reused for another object!}
935
936The advantage of borrowing over owning a reference is that you don't
937need to take care of disposing of the reference on all possible paths
938through the code --- in other words, with a borrowed reference you
939don't run the risk of leaking when a premature exit is taken. The
940disadvantage of borrowing over leaking is that there are some subtle
941situations where in seemingly correct code a borrowed reference can be
942used after the owner from which it was borrowed has in fact disposed
943of it.
944
945A borrowed reference can be changed into an owned reference by calling
946\cfunction{Py_INCREF()}. This does not affect the status of the owner from
947which the reference was borrowed --- it creates a new owned reference,
948and gives full owner responsibilities (the new owner must
949dispose of the reference properly, as well as the previous owner).
950
951
952\subsection{Ownership Rules
953 \label{ownershipRules}}
954
955Whenever an object reference is passed into or out of a function, it
956is part of the function's interface specification whether ownership is
957transferred with the reference or not.
958
959Most functions that return a reference to an object pass on ownership
960with the reference. In particular, all functions whose function it is
961to create a new object, such as \cfunction{PyInt_FromLong()} and
Fred Drake92024d12001-11-29 07:16:19 +0000962\cfunction{Py_BuildValue()}, pass ownership to the receiver. Even if
963the object is not actually new, you still receive ownership of a new
964reference to that object. For instance, \cfunction{PyInt_FromLong()}
965maintains a cache of popular values and can return a reference to a
966cached item.
Fred Drakecc8f44b2001-08-20 19:30:29 +0000967
968Many functions that extract objects from other objects also transfer
969ownership with the reference, for instance
970\cfunction{PyObject_GetAttrString()}. The picture is less clear, here,
971however, since a few common routines are exceptions:
972\cfunction{PyTuple_GetItem()}, \cfunction{PyList_GetItem()},
973\cfunction{PyDict_GetItem()}, and \cfunction{PyDict_GetItemString()}
974all return references that you borrow from the tuple, list or
975dictionary.
976
977The function \cfunction{PyImport_AddModule()} also returns a borrowed
978reference, even though it may actually create the object it returns:
979this is possible because an owned reference to the object is stored in
980\code{sys.modules}.
981
982When you pass an object reference into another function, in general,
983the function borrows the reference from you --- if it needs to store
984it, it will use \cfunction{Py_INCREF()} to become an independent
985owner. There are exactly two important exceptions to this rule:
986\cfunction{PyTuple_SetItem()} and \cfunction{PyList_SetItem()}. These
987functions take over ownership of the item passed to them --- even if
988they fail! (Note that \cfunction{PyDict_SetItem()} and friends don't
989take over ownership --- they are ``normal.'')
990
991When a C function is called from Python, it borrows references to its
992arguments from the caller. The caller owns a reference to the object,
993so the borrowed reference's lifetime is guaranteed until the function
994returns. Only when such a borrowed reference must be stored or passed
995on, it must be turned into an owned reference by calling
996\cfunction{Py_INCREF()}.
997
998The object reference returned from a C function that is called from
999Python must be an owned reference --- ownership is tranferred from the
1000function to its caller.
1001
1002
1003\subsection{Thin Ice
1004 \label{thinIce}}
1005
1006There are a few situations where seemingly harmless use of a borrowed
1007reference can lead to problems. These all have to do with implicit
1008invocations of the interpreter, which can cause the owner of a
1009reference to dispose of it.
1010
1011The first and most important case to know about is using
1012\cfunction{Py_DECREF()} on an unrelated object while borrowing a
1013reference to a list item. For instance:
1014
1015\begin{verbatim}
Fred Drake723f94b2002-06-22 01:42:00 +00001016void
1017bug(PyObject *list)
1018{
Fred Drakecc8f44b2001-08-20 19:30:29 +00001019 PyObject *item = PyList_GetItem(list, 0);
1020
1021 PyList_SetItem(list, 1, PyInt_FromLong(0L));
1022 PyObject_Print(item, stdout, 0); /* BUG! */
1023}
1024\end{verbatim}
1025
1026This function first borrows a reference to \code{list[0]}, then
1027replaces \code{list[1]} with the value \code{0}, and finally prints
1028the borrowed reference. Looks harmless, right? But it's not!
1029
1030Let's follow the control flow into \cfunction{PyList_SetItem()}. The list
1031owns references to all its items, so when item 1 is replaced, it has
1032to dispose of the original item 1. Now let's suppose the original
1033item 1 was an instance of a user-defined class, and let's further
1034suppose that the class defined a \method{__del__()} method. If this
1035class instance has a reference count of 1, disposing of it will call
1036its \method{__del__()} method.
1037
1038Since it is written in Python, the \method{__del__()} method can execute
1039arbitrary Python code. Could it perhaps do something to invalidate
1040the reference to \code{item} in \cfunction{bug()}? You bet! Assuming
1041that the list passed into \cfunction{bug()} is accessible to the
1042\method{__del__()} method, it could execute a statement to the effect of
1043\samp{del list[0]}, and assuming this was the last reference to that
1044object, it would free the memory associated with it, thereby
1045invalidating \code{item}.
1046
1047The solution, once you know the source of the problem, is easy:
1048temporarily increment the reference count. The correct version of the
1049function reads:
1050
1051\begin{verbatim}
Fred Drake723f94b2002-06-22 01:42:00 +00001052void
1053no_bug(PyObject *list)
1054{
Fred Drakecc8f44b2001-08-20 19:30:29 +00001055 PyObject *item = PyList_GetItem(list, 0);
1056
1057 Py_INCREF(item);
1058 PyList_SetItem(list, 1, PyInt_FromLong(0L));
1059 PyObject_Print(item, stdout, 0);
1060 Py_DECREF(item);
1061}
1062\end{verbatim}
1063
1064This is a true story. An older version of Python contained variants
1065of this bug and someone spent a considerable amount of time in a C
1066debugger to figure out why his \method{__del__()} methods would fail...
1067
1068The second case of problems with a borrowed reference is a variant
1069involving threads. Normally, multiple threads in the Python
1070interpreter can't get in each other's way, because there is a global
1071lock protecting Python's entire object space. However, it is possible
1072to temporarily release this lock using the macro
Fred Drake375e3022002-04-09 21:09:42 +00001073\csimplemacro{Py_BEGIN_ALLOW_THREADS}, and to re-acquire it using
1074\csimplemacro{Py_END_ALLOW_THREADS}. This is common around blocking
1075I/O calls, to let other threads use the processor while waiting for
1076the I/O to complete. Obviously, the following function has the same
1077problem as the previous one:
Fred Drakecc8f44b2001-08-20 19:30:29 +00001078
1079\begin{verbatim}
Fred Drake723f94b2002-06-22 01:42:00 +00001080void
1081bug(PyObject *list)
1082{
Fred Drakecc8f44b2001-08-20 19:30:29 +00001083 PyObject *item = PyList_GetItem(list, 0);
1084 Py_BEGIN_ALLOW_THREADS
1085 ...some blocking I/O call...
1086 Py_END_ALLOW_THREADS
1087 PyObject_Print(item, stdout, 0); /* BUG! */
1088}
1089\end{verbatim}
1090
1091
1092\subsection{NULL Pointers
1093 \label{nullPointers}}
1094
1095In general, functions that take object references as arguments do not
1096expect you to pass them \NULL{} pointers, and will dump core (or
1097cause later core dumps) if you do so. Functions that return object
1098references generally return \NULL{} only to indicate that an
1099exception occurred. The reason for not testing for \NULL{}
1100arguments is that functions often pass the objects they receive on to
Fred Drakec37b65e2001-11-28 07:26:15 +00001101other function --- if each function were to test for \NULL,
Fred Drakecc8f44b2001-08-20 19:30:29 +00001102there would be a lot of redundant tests and the code would run more
1103slowly.
1104
1105It is better to test for \NULL{} only at the ``source:'' when a
1106pointer that may be \NULL{} is received, for example, from
1107\cfunction{malloc()} or from a function that may raise an exception.
1108
1109The macros \cfunction{Py_INCREF()} and \cfunction{Py_DECREF()}
1110do not check for \NULL{} pointers --- however, their variants
1111\cfunction{Py_XINCREF()} and \cfunction{Py_XDECREF()} do.
1112
1113The macros for checking for a particular object type
1114(\code{Py\var{type}_Check()}) don't check for \NULL{} pointers ---
1115again, there is much code that calls several of these in a row to test
1116an object against various different expected types, and this would
1117generate redundant tests. There are no variants with \NULL{}
1118checking.
1119
1120The C function calling mechanism guarantees that the argument list
1121passed to C functions (\code{args} in the examples) is never
1122\NULL{} --- in fact it guarantees that it is always a tuple.\footnote{
1123These guarantees don't hold when you use the ``old'' style
1124calling convention --- this is still found in much existing code.}
1125
1126It is a severe error to ever let a \NULL{} pointer ``escape'' to
1127the Python user.
1128
1129% Frank Stajano:
1130% A pedagogically buggy example, along the lines of the previous listing,
1131% would be helpful here -- showing in more concrete terms what sort of
1132% actions could cause the problem. I can't very well imagine it from the
1133% description.
1134
1135
Fred Drakec37b65e2001-11-28 07:26:15 +00001136\section{Writing Extensions in \Cpp
Fred Drakecc8f44b2001-08-20 19:30:29 +00001137 \label{cplusplus}}
1138
Fred Drakec37b65e2001-11-28 07:26:15 +00001139It is possible to write extension modules in \Cpp. Some restrictions
Fred Drakecc8f44b2001-08-20 19:30:29 +00001140apply. If the main program (the Python interpreter) is compiled and
1141linked by the C compiler, global or static objects with constructors
1142cannot be used. This is not a problem if the main program is linked
1143by the \Cpp{} compiler. Functions that will be called by the
1144Python interpreter (in particular, module initalization functions)
1145have to be declared using \code{extern "C"}.
1146It is unnecessary to enclose the Python header files in
1147\code{extern "C" \{...\}} --- they use this form already if the symbol
1148\samp{__cplusplus} is defined (all recent \Cpp{} compilers define this
1149symbol).
1150
1151
1152\section{Providing a C API for an Extension Module
1153 \label{using-cobjects}}
1154\sectionauthor{Konrad Hinsen}{hinsen@cnrs-orleans.fr}
1155
1156Many extension modules just provide new functions and types to be
1157used from Python, but sometimes the code in an extension module can
1158be useful for other extension modules. For example, an extension
1159module could implement a type ``collection'' which works like lists
1160without order. Just like the standard Python list type has a C API
1161which permits extension modules to create and manipulate lists, this
1162new collection type should have a set of C functions for direct
1163manipulation from other extension modules.
1164
1165At first sight this seems easy: just write the functions (without
1166declaring them \keyword{static}, of course), provide an appropriate
1167header file, and document the C API. And in fact this would work if
1168all extension modules were always linked statically with the Python
1169interpreter. When modules are used as shared libraries, however, the
1170symbols defined in one module may not be visible to another module.
1171The details of visibility depend on the operating system; some systems
1172use one global namespace for the Python interpreter and all extension
1173modules (Windows, for example), whereas others require an explicit
1174list of imported symbols at module link time (AIX is one example), or
1175offer a choice of different strategies (most Unices). And even if
1176symbols are globally visible, the module whose functions one wishes to
1177call might not have been loaded yet!
1178
1179Portability therefore requires not to make any assumptions about
1180symbol visibility. This means that all symbols in extension modules
1181should be declared \keyword{static}, except for the module's
1182initialization function, in order to avoid name clashes with other
1183extension modules (as discussed in section~\ref{methodTable}). And it
1184means that symbols that \emph{should} be accessible from other
1185extension modules must be exported in a different way.
1186
1187Python provides a special mechanism to pass C-level information
1188(pointers) from one extension module to another one: CObjects.
1189A CObject is a Python data type which stores a pointer (\ctype{void
1190*}). CObjects can only be created and accessed via their C API, but
1191they can be passed around like any other Python object. In particular,
1192they can be assigned to a name in an extension module's namespace.
1193Other extension modules can then import this module, retrieve the
1194value of this name, and then retrieve the pointer from the CObject.
1195
1196There are many ways in which CObjects can be used to export the C API
1197of an extension module. Each name could get its own CObject, or all C
1198API pointers could be stored in an array whose address is published in
1199a CObject. And the various tasks of storing and retrieving the pointers
1200can be distributed in different ways between the module providing the
1201code and the client modules.
1202
1203The following example demonstrates an approach that puts most of the
1204burden on the writer of the exporting module, which is appropriate
1205for commonly used library modules. It stores all C API pointers
1206(just one in the example!) in an array of \ctype{void} pointers which
1207becomes the value of a CObject. The header file corresponding to
1208the module provides a macro that takes care of importing the module
1209and retrieving its C API pointers; client modules only have to call
1210this macro before accessing the C API.
1211
1212The exporting module is a modification of the \module{spam} module from
1213section~\ref{simpleExample}. The function \function{spam.system()}
1214does not call the C library function \cfunction{system()} directly,
1215but a function \cfunction{PySpam_System()}, which would of course do
1216something more complicated in reality (such as adding ``spam'' to
1217every command). This function \cfunction{PySpam_System()} is also
1218exported to other extension modules.
1219
1220The function \cfunction{PySpam_System()} is a plain C function,
1221declared \keyword{static} like everything else:
1222
1223\begin{verbatim}
1224static int
Fred Drake723f94b2002-06-22 01:42:00 +00001225PySpam_System(char *command)
Fred Drakecc8f44b2001-08-20 19:30:29 +00001226{
1227 return system(command);
1228}
1229\end{verbatim}
1230
1231The function \cfunction{spam_system()} is modified in a trivial way:
1232
1233\begin{verbatim}
1234static PyObject *
Fred Drake723f94b2002-06-22 01:42:00 +00001235spam_system(PyObject *self, PyObject *args)
Fred Drakecc8f44b2001-08-20 19:30:29 +00001236{
1237 char *command;
1238 int sts;
1239
1240 if (!PyArg_ParseTuple(args, "s", &command))
1241 return NULL;
1242 sts = PySpam_System(command);
1243 return Py_BuildValue("i", sts);
1244}
1245\end{verbatim}
1246
1247In the beginning of the module, right after the line
1248
1249\begin{verbatim}
1250#include "Python.h"
1251\end{verbatim}
1252
1253two more lines must be added:
1254
1255\begin{verbatim}
1256#define SPAM_MODULE
1257#include "spammodule.h"
1258\end{verbatim}
1259
1260The \code{\#define} is used to tell the header file that it is being
1261included in the exporting module, not a client module. Finally,
1262the module's initialization function must take care of initializing
1263the C API pointer array:
1264
1265\begin{verbatim}
1266void
Fred Drakeef6373a2001-11-17 06:50:42 +00001267initspam(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +00001268{
1269 PyObject *m;
1270 static void *PySpam_API[PySpam_API_pointers];
1271 PyObject *c_api_object;
1272
1273 m = Py_InitModule("spam", SpamMethods);
1274
1275 /* Initialize the C API pointer array */
1276 PySpam_API[PySpam_System_NUM] = (void *)PySpam_System;
1277
1278 /* Create a CObject containing the API pointer array's address */
1279 c_api_object = PyCObject_FromVoidPtr((void *)PySpam_API, NULL);
1280
Fred Drake63e40a52002-04-12 19:08:31 +00001281 if (c_api_object != NULL)
1282 PyModule_AddObject(m, "_C_API", c_api_object);
Fred Drakecc8f44b2001-08-20 19:30:29 +00001283}
1284\end{verbatim}
1285
Fred Drakeef6373a2001-11-17 06:50:42 +00001286Note that \code{PySpam_API} is declared \keyword{static}; otherwise
1287the pointer array would disappear when \function{initspam()} terminates!
Fred Drakecc8f44b2001-08-20 19:30:29 +00001288
1289The bulk of the work is in the header file \file{spammodule.h},
1290which looks like this:
1291
1292\begin{verbatim}
1293#ifndef Py_SPAMMODULE_H
1294#define Py_SPAMMODULE_H
1295#ifdef __cplusplus
1296extern "C" {
1297#endif
1298
1299/* Header file for spammodule */
1300
1301/* C API functions */
1302#define PySpam_System_NUM 0
1303#define PySpam_System_RETURN int
1304#define PySpam_System_PROTO (char *command)
1305
1306/* Total number of C API pointers */
1307#define PySpam_API_pointers 1
1308
1309
1310#ifdef SPAM_MODULE
1311/* This section is used when compiling spammodule.c */
1312
1313static PySpam_System_RETURN PySpam_System PySpam_System_PROTO;
1314
1315#else
1316/* This section is used in modules that use spammodule's API */
1317
1318static void **PySpam_API;
1319
1320#define PySpam_System \
1321 (*(PySpam_System_RETURN (*)PySpam_System_PROTO) PySpam_API[PySpam_System_NUM])
1322
Fred Drake63e40a52002-04-12 19:08:31 +00001323/* Return -1 and set exception on error, 0 on success. */
1324static int
1325import_spam(void)
1326{
1327 PyObject *module = PyImport_ImportModule("spam");
1328
1329 if (module != NULL) {
1330 PyObject *c_api_object = PyObject_GetAttrString(module, "_C_API");
1331 if (c_api_object == NULL)
1332 return -1;
1333 if (PyCObject_Check(c_api_object))
1334 PySpam_API = (void **)PyCObject_AsVoidPtr(c_api_object);
1335 Py_DECREF(c_api_object);
1336 }
1337 return 0;
Fred Drakecc8f44b2001-08-20 19:30:29 +00001338}
1339
1340#endif
1341
1342#ifdef __cplusplus
1343}
1344#endif
1345
1346#endif /* !defined(Py_SPAMMODULE_H */
1347\end{verbatim}
1348
1349All that a client module must do in order to have access to the
1350function \cfunction{PySpam_System()} is to call the function (or
1351rather macro) \cfunction{import_spam()} in its initialization
1352function:
1353
1354\begin{verbatim}
1355void
Fred Drakeef6373a2001-11-17 06:50:42 +00001356initclient(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +00001357{
1358 PyObject *m;
1359
1360 Py_InitModule("client", ClientMethods);
Fred Drake63e40a52002-04-12 19:08:31 +00001361 if (import_spam() < 0)
1362 return;
1363 /* additional initialization can happen here */
Fred Drakecc8f44b2001-08-20 19:30:29 +00001364}
1365\end{verbatim}
1366
1367The main disadvantage of this approach is that the file
1368\file{spammodule.h} is rather complicated. However, the
1369basic structure is the same for each function that is
1370exported, so it has to be learned only once.
1371
1372Finally it should be mentioned that CObjects offer additional
1373functionality, which is especially useful for memory allocation and
1374deallocation of the pointer stored in a CObject. The details
1375are described in the \citetitle[../api/api.html]{Python/C API
Fred Drake63e40a52002-04-12 19:08:31 +00001376Reference Manual} in the section
1377``\ulink{CObjects}{../api/cObjects.html}'' and in the implementation
1378of CObjects (files \file{Include/cobject.h} and
Fred Drakecc8f44b2001-08-20 19:30:29 +00001379\file{Objects/cobject.c} in the Python source code distribution).