1
2/* This module is compiled using limited API from Python 3.5,
3 * making sure that it works as expected.
4 *
5 * See the xxlimited module for an extension module template.
6 */
7
8#define Py_LIMITED_API 0x03050000
9
10#include "Python.h"
11
12/* Xxo objects */
13
14static PyObject *ErrorObject;
15
16typedef struct {
17 PyObject_HEAD
18 PyObject *x_attr; /* Attributes dictionary */
19} XxoObject;
20
21static PyObject *Xxo_Type;
22
23#define XxoObject_Check(v) Py_IS_TYPE(v, Xxo_Type)
24
25static XxoObject *
26newXxoObject(PyObject *arg)
27{
28 XxoObject *self;
29 self = PyObject_GC_New(XxoObject, (PyTypeObject*)Xxo_Type);
30 if (self == NULL)
31 return NULL;
32 self->x_attr = NULL;
33 return self;
34}
35
36/* Xxo methods */
37
38static int
39Xxo_traverse(XxoObject *self, visitproc visit, void *arg)
40{
41 Py_VISIT(Py_TYPE(self));
42 Py_VISIT(self->x_attr);
43 return 0;
44}
45
46static int
47Xxo_clear(XxoObject *self)
48{
49 Py_CLEAR(self->x_attr);
50 return 0;
51}
52
53static void
54Xxo_finalize(XxoObject *self)
55{
56 Py_CLEAR(self->x_attr);
57}
58
59static PyObject *
60Xxo_demo(XxoObject *self, PyObject *args)
61{
62 PyObject *o = NULL;
63 if (!PyArg_ParseTuple(args, "|O:demo", &o))
64 return NULL;
65 /* Test availability of fast type checks */
66 if (o != NULL && PyUnicode_Check(o)) {
67 Py_INCREF(o);
68 return o;
69 }
70 Py_INCREF(Py_None);
71 return Py_None;
72}
73
74static PyMethodDef Xxo_methods[] = {
75 {"demo", (PyCFunction)Xxo_demo, METH_VARARGS,
76 PyDoc_STR("demo() -> None")},
77 {NULL, NULL} /* sentinel */
78};
79
80static PyObject *
81Xxo_getattro(XxoObject *self, PyObject *name)
82{
83 if (self->x_attr != NULL) {
84 PyObject *v = PyDict_GetItemWithError(self->x_attr, name);
85 if (v != NULL) {
86 Py_INCREF(v);
87 return v;
88 }
89 else if (PyErr_Occurred()) {
90 return NULL;
91 }
92 }
93 return PyObject_GenericGetAttr((PyObject *)self, name);
94}
95
96static int
97Xxo_setattr(XxoObject *self, const char *name, PyObject *v)
98{
99 if (self->x_attr == NULL) {
100 self->x_attr = PyDict_New();
101 if (self->x_attr == NULL)
102 return -1;
103 }
104 if (v == NULL) {
105 int rv = PyDict_DelItemString(self->x_attr, name);
106 if (rv < 0 && PyErr_ExceptionMatches(PyExc_KeyError))
107 PyErr_SetString(PyExc_AttributeError,
108 "delete non-existing Xxo attribute");
109 return rv;
110 }
111 else
112 return PyDict_SetItemString(self->x_attr, name, v);
113}
114
115static PyType_Slot Xxo_Type_slots[] = {
116 {Py_tp_doc, "The Xxo type"},
117 {Py_tp_traverse, Xxo_traverse},
118 {Py_tp_clear, Xxo_clear},
119 {Py_tp_finalize, Xxo_finalize},
120 {Py_tp_getattro, Xxo_getattro},
121 {Py_tp_setattr, Xxo_setattr},
122 {Py_tp_methods, Xxo_methods},
123 {0, 0},
124};
125
126static PyType_Spec Xxo_Type_spec = {
127 "xxlimited.Xxo",
128 sizeof(XxoObject),
129 0,
130 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,
131 Xxo_Type_slots
132};
133
134/* --------------------------------------------------------------------- */
135
136/* Function of two integers returning integer */
137
138PyDoc_STRVAR(xx_foo_doc,
139"foo(i,j)\n\
140\n\
141Return the sum of i and j.");
142
143static PyObject *
144xx_foo(PyObject *self, PyObject *args)
145{
146 long i, j;
147 long res;
148 if (!PyArg_ParseTuple(args, "ll:foo", &i, &j))
149 return NULL;
150 res = i+j; /* XXX Do something here */
151 return PyLong_FromLong(res);
152}
153
154
155/* Function of no arguments returning new Xxo object */
156
157static PyObject *
158xx_new(PyObject *self, PyObject *args)
159{
160 XxoObject *rv;
161
162 if (!PyArg_ParseTuple(args, ":new"))
163 return NULL;
164 rv = newXxoObject(args);
165 if (rv == NULL)
166 return NULL;
167 return (PyObject *)rv;
168}
169
170/* Test bad format character */
171
172static PyObject *
173xx_roj(PyObject *self, PyObject *args)
174{
175 PyObject *a;
176 long b;
177 if (!PyArg_ParseTuple(args, "O#:roj", &a, &b))
178 return NULL;
179 Py_INCREF(Py_None);
180 return Py_None;
181}
182
183
184/* ---------- */
185
186static PyType_Slot Str_Type_slots[] = {
187 {Py_tp_base, NULL}, /* filled out in module init function */
188 {0, 0},
189};
190
191static PyType_Spec Str_Type_spec = {
192 "xxlimited.Str",
193 0,
194 0,
195 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
196 Str_Type_slots
197};
198
199/* ---------- */
200
201static PyObject *
202null_richcompare(PyObject *self, PyObject *other, int op)
203{
204 Py_RETURN_NOTIMPLEMENTED;
205}
206
207static PyType_Slot Null_Type_slots[] = {
208 {Py_tp_base, NULL}, /* filled out in module init */
209 {Py_tp_new, NULL},
210 {Py_tp_richcompare, null_richcompare},
211 {0, 0}
212};
213
214static PyType_Spec Null_Type_spec = {
215 "xxlimited.Null",
216 0, /* basicsize */
217 0, /* itemsize */
218 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
219 Null_Type_slots
220};
221
222/* ---------- */
223
224/* List of functions defined in the module */
225
226static PyMethodDef xx_methods[] = {
227 {"roj", xx_roj, METH_VARARGS,
228 PyDoc_STR("roj(a,b) -> None")},
229 {"foo", xx_foo, METH_VARARGS,
230 xx_foo_doc},
231 {"new", xx_new, METH_VARARGS,
232 PyDoc_STR("new() -> new Xx object")},
233 {NULL, NULL} /* sentinel */
234};
235
236PyDoc_STRVAR(module_doc,
237"This is a module for testing limited API from Python 3.5.");
238
239static int
240xx_modexec(PyObject *m)
241{
242 PyObject *o;
243
244 /* Due to cross platform compiler issues the slots must be filled
245 * here. It's required for portability to Windows without requiring
246 * C++. */
247 Null_Type_slots[0].pfunc = &PyBaseObject_Type;
248 Null_Type_slots[1].pfunc = PyType_GenericNew;
249 Str_Type_slots[0].pfunc = &PyUnicode_Type;
250
251 Xxo_Type = PyType_FromSpec(&Xxo_Type_spec);
252 if (Xxo_Type == NULL)
253 goto fail;
254
255 /* Add some symbolic constants to the module */
256 if (ErrorObject == NULL) {
257 ErrorObject = PyErr_NewException("xxlimited.error", NULL, NULL);
258 if (ErrorObject == NULL)
259 goto fail;
260 }
261 Py_INCREF(ErrorObject);
262 PyModule_AddObject(m, "error", ErrorObject);
263
264 /* Add Xxo */
265 o = PyType_FromSpec(&Xxo_Type_spec);
266 if (o == NULL)
267 goto fail;
268 PyModule_AddObject(m, "Xxo", o);
269
270 /* Add Str */
271 o = PyType_FromSpec(&Str_Type_spec);
272 if (o == NULL)
273 goto fail;
274 PyModule_AddObject(m, "Str", o);
275
276 /* Add Null */
277 o = PyType_FromSpec(&Null_Type_spec);
278 if (o == NULL)
279 goto fail;
280 PyModule_AddObject(m, "Null", o);
281 return 0;
282 fail:
283 Py_XDECREF(m);
284 return -1;
285}
286
287
288static PyModuleDef_Slot xx_slots[] = {
289 {Py_mod_exec, xx_modexec},
290 {0, NULL}
291};
292
293static struct PyModuleDef xxmodule = {
294 PyModuleDef_HEAD_INIT,
295 "xxlimited_35",
296 module_doc,
297 0,
298 xx_methods,
299 xx_slots,
300 NULL,
301 NULL,
302 NULL
303};
304
305/* Export function for the module (*must* be called PyInit_xx) */
306
307PyMODINIT_FUNC
308PyInit_xxlimited_35(void)
309{
310 return PyModuleDef_Init(&xxmodule);
311}
312