blob: f41d4fec0395c91ec16d60997cd256d409ac1df6 [file] [log] [blame]
Tim Peters9ea17ac2001-02-02 05:57:15 +00001/*
2 * C Extension module to test Python interpreter C APIs.
3 *
4 * The 'test_*' functions exported by this module are run as part of the
5 * standard Python regression test, via Lib/test/test_capi.py.
6 */
7
Neal Norwitz8866e0a2007-10-27 04:01:17 +00008#define PY_SSIZE_T_CLEAN
9
Tim Peters9ea17ac2001-02-02 05:57:15 +000010#include "Python.h"
Thomas Wouters89f507f2006-12-13 04:49:30 +000011#include <float.h>
12#include "structmember.h"
Tim Peters9ea17ac2001-02-02 05:57:15 +000013
Mark Hammond8d98d2c2003-04-19 15:41:53 +000014#ifdef WITH_THREAD
15#include "pythread.h"
16#endif /* WITH_THREAD */
Tim Peters9ea17ac2001-02-02 05:57:15 +000017static PyObject *TestError; /* set to exception object in init */
18
Tim Peters91621db2001-06-12 20:10:01 +000019/* Raise TestError with test_name + ": " + msg, and return NULL. */
20
21static PyObject *
22raiseTestError(const char* test_name, const char* msg)
23{
24 char buf[2048];
25
26 if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
27 PyErr_SetString(TestError, "internal error msg too large");
28 else {
Tim Peters885d4572001-11-28 20:27:42 +000029 PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
Tim Peters91621db2001-06-12 20:10:01 +000030 PyErr_SetString(TestError, buf);
31 }
32 return NULL;
33}
34
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +000035/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
Tim Peters9ea17ac2001-02-02 05:57:15 +000036
37 The ones derived from autoconf on the UNIX-like OSes can be relied
38 upon (in the absence of sloppy cross-compiling), but the Windows
39 platforms have these hardcoded. Better safe than sorry.
40*/
41static PyObject*
Thomas Wouters89f507f2006-12-13 04:49:30 +000042sizeof_error(const char* fatname, const char* typname,
Tim Peters9ea17ac2001-02-02 05:57:15 +000043 int expected, int got)
44{
45 char buf[1024];
Tim Petersbaefd9e2003-01-28 20:37:45 +000046 PyOS_snprintf(buf, sizeof(buf),
Tim Peters885d4572001-11-28 20:27:42 +000047 "%.200s #define == %d but sizeof(%.200s) == %d",
Thomas Wouters89f507f2006-12-13 04:49:30 +000048 fatname, expected, typname, got);
Tim Peters9ea17ac2001-02-02 05:57:15 +000049 PyErr_SetString(TestError, buf);
50 return (PyObject*)NULL;
51}
52
53static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000054test_config(PyObject *self)
Tim Peters9ea17ac2001-02-02 05:57:15 +000055{
Tim Peters9ea17ac2001-02-02 05:57:15 +000056#define CHECK_SIZEOF(FATNAME, TYPE) \
57 if (FATNAME != sizeof(TYPE)) \
58 return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
59
Tim Peters208efe52001-06-26 22:40:47 +000060 CHECK_SIZEOF(SIZEOF_SHORT, short);
Tim Peters9ea17ac2001-02-02 05:57:15 +000061 CHECK_SIZEOF(SIZEOF_INT, int);
62 CHECK_SIZEOF(SIZEOF_LONG, long);
63 CHECK_SIZEOF(SIZEOF_VOID_P, void*);
64 CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
65#ifdef HAVE_LONG_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +000066 CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
Tim Peters9ea17ac2001-02-02 05:57:15 +000067#endif
68
69#undef CHECK_SIZEOF
70
71 Py_INCREF(Py_None);
72 return Py_None;
73}
74
Tim Peters5c4d5bf2001-02-12 22:13:26 +000075static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000076test_list_api(PyObject *self)
Tim Peters5c4d5bf2001-02-12 22:13:26 +000077{
78 PyObject* list;
79 int i;
Tim Peters5c4d5bf2001-02-12 22:13:26 +000080
81 /* SF bug 132008: PyList_Reverse segfaults */
82#define NLIST 30
83 list = PyList_New(NLIST);
84 if (list == (PyObject*)NULL)
85 return (PyObject*)NULL;
86 /* list = range(NLIST) */
87 for (i = 0; i < NLIST; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +000088 PyObject* anint = PyLong_FromLong(i);
Tim Peters5c4d5bf2001-02-12 22:13:26 +000089 if (anint == (PyObject*)NULL) {
90 Py_DECREF(list);
91 return (PyObject*)NULL;
92 }
93 PyList_SET_ITEM(list, i, anint);
94 }
95 /* list.reverse(), via PyList_Reverse() */
96 i = PyList_Reverse(list); /* should not blow up! */
97 if (i != 0) {
98 Py_DECREF(list);
99 return (PyObject*)NULL;
100 }
101 /* Check that list == range(29, -1, -1) now */
102 for (i = 0; i < NLIST; ++i) {
103 PyObject* anint = PyList_GET_ITEM(list, i);
Christian Heimes217cfd12007-12-02 14:31:20 +0000104 if (PyLong_AS_LONG(anint) != NLIST-1-i) {
Tim Peters5c4d5bf2001-02-12 22:13:26 +0000105 PyErr_SetString(TestError,
106 "test_list_api: reverse screwed up");
107 Py_DECREF(list);
108 return (PyObject*)NULL;
109 }
110 }
111 Py_DECREF(list);
112#undef NLIST
113
114 Py_INCREF(Py_None);
115 return Py_None;
116}
117
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000118static int
119test_dict_inner(int count)
120{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000121 Py_ssize_t pos = 0, iterations = 0;
122 int i;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000123 PyObject *dict = PyDict_New();
124 PyObject *v, *k;
125
126 if (dict == NULL)
127 return -1;
128
129 for (i = 0; i < count; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +0000130 v = PyLong_FromLong(i);
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000131 PyDict_SetItem(dict, v, v);
132 Py_DECREF(v);
133 }
134
135 while (PyDict_Next(dict, &pos, &k, &v)) {
136 PyObject *o;
137 iterations++;
138
Christian Heimes217cfd12007-12-02 14:31:20 +0000139 i = PyLong_AS_LONG(v) + 1;
140 o = PyLong_FromLong(i);
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000141 if (o == NULL)
142 return -1;
143 if (PyDict_SetItem(dict, k, o) < 0) {
144 Py_DECREF(o);
145 return -1;
146 }
147 Py_DECREF(o);
148 }
149
150 Py_DECREF(dict);
151
152 if (iterations != count) {
153 PyErr_SetString(
154 TestError,
155 "test_dict_iteration: dict iteration went wrong ");
156 return -1;
157 } else {
158 return 0;
159 }
160}
161
162static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +0000163test_dict_iteration(PyObject* self)
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000164{
165 int i;
166
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000167 for (i = 0; i < 200; i++) {
168 if (test_dict_inner(i) < 0) {
169 return NULL;
170 }
171 }
172
173 Py_INCREF(Py_None);
174 return Py_None;
175}
176
Tim Peters91621db2001-06-12 20:10:01 +0000177
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000178/* Issue #4701: Check that PyObject_Hash implicitly calls
179 * PyType_Ready if it hasn't already been called
180 */
181static PyTypeObject _HashInheritanceTester_Type = {
Hirokazu Yamamoto8ebab5d2008-12-31 06:05:46 +0000182 PyVarObject_HEAD_INIT(NULL, 0)
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000183 "hashinheritancetester", /* Name of this type */
184 sizeof(PyObject), /* Basic object size */
185 0, /* Item size for varobject */
186 (destructor)PyObject_Del, /* tp_dealloc */
187 0, /* tp_print */
188 0, /* tp_getattr */
189 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +0000190 0, /* tp_reserved */
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000191 0, /* tp_repr */
192 0, /* tp_as_number */
193 0, /* tp_as_sequence */
194 0, /* tp_as_mapping */
195 0, /* tp_hash */
196 0, /* tp_call */
197 0, /* tp_str */
198 PyObject_GenericGetAttr, /* tp_getattro */
199 0, /* tp_setattro */
200 0, /* tp_as_buffer */
201 Py_TPFLAGS_DEFAULT, /* tp_flags */
202 0, /* tp_doc */
203 0, /* tp_traverse */
204 0, /* tp_clear */
205 0, /* tp_richcompare */
206 0, /* tp_weaklistoffset */
207 0, /* tp_iter */
208 0, /* tp_iternext */
209 0, /* tp_methods */
210 0, /* tp_members */
211 0, /* tp_getset */
212 0, /* tp_base */
213 0, /* tp_dict */
214 0, /* tp_descr_get */
215 0, /* tp_descr_set */
216 0, /* tp_dictoffset */
217 0, /* tp_init */
218 0, /* tp_alloc */
219 PyType_GenericNew, /* tp_new */
220};
221
222static PyObject*
223test_lazy_hash_inheritance(PyObject* self)
224{
225 PyTypeObject *type;
226 PyObject *obj;
227 long hash;
228
229 type = &_HashInheritanceTester_Type;
Benjamin Petersonf1726372009-05-05 21:11:54 +0000230
Benjamin Petersonb43d3252009-05-05 23:14:15 +0000231 if (type->tp_dict != NULL)
232 /* The type has already been initialized. This probably means
233 -R is being used. */
234 Py_RETURN_NONE;
Benjamin Petersonf1726372009-05-05 21:11:54 +0000235
236
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000237 obj = PyObject_New(PyObject, type);
238 if (obj == NULL) {
239 PyErr_Clear();
240 PyErr_SetString(
241 TestError,
242 "test_lazy_hash_inheritance: failed to create object");
243 return NULL;
244 }
245
246 if (type->tp_dict != NULL) {
247 PyErr_SetString(
248 TestError,
249 "test_lazy_hash_inheritance: type initialised too soon");
250 Py_DECREF(obj);
251 return NULL;
252 }
253
254 hash = PyObject_Hash(obj);
255 if ((hash == -1) && PyErr_Occurred()) {
256 PyErr_Clear();
257 PyErr_SetString(
258 TestError,
259 "test_lazy_hash_inheritance: could not hash object");
260 Py_DECREF(obj);
261 return NULL;
262 }
263
264 if (type->tp_dict == NULL) {
265 PyErr_SetString(
266 TestError,
267 "test_lazy_hash_inheritance: type not initialised by hash()");
268 Py_DECREF(obj);
269 return NULL;
270 }
271
272 if (type->tp_hash != PyType_Type.tp_hash) {
273 PyErr_SetString(
274 TestError,
275 "test_lazy_hash_inheritance: unexpected hash function");
276 Py_DECREF(obj);
277 return NULL;
278 }
279
Benjamin Petersonb43d3252009-05-05 23:14:15 +0000280 Py_DECREF(obj);
Benjamin Petersonf1726372009-05-05 21:11:54 +0000281
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000282 Py_RETURN_NONE;
283}
284
285
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000286/* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
Tim Petersff70d3c2001-06-14 01:11:03 +0000287 PyLong_{As, From}{Unsigned,}LongLong().
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000288
289 Note that the meat of the test is contained in testcapi_long.h.
290 This is revolting, but delicate code duplication is worse: "almost
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000291 exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000292 dependence on type names makes it impossible to use a parameterized
293 function. A giant macro would be even worse than this. A C++ template
294 would be perfect.
295
296 The "report an error" functions are deliberately not part of the #include
297 file: if the test fails, you can set a breakpoint in the appropriate
298 error function directly, and crawl back from there in the debugger.
299*/
300
301#define UNBIND(X) Py_DECREF(X); (X) = NULL
302
303static PyObject *
304raise_test_long_error(const char* msg)
305{
306 return raiseTestError("test_long_api", msg);
307}
308
309#define TESTNAME test_long_api_inner
310#define TYPENAME long
311#define F_S_TO_PY PyLong_FromLong
312#define F_PY_TO_S PyLong_AsLong
313#define F_U_TO_PY PyLong_FromUnsignedLong
314#define F_PY_TO_U PyLong_AsUnsignedLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000315
316#include "testcapi_long.h"
317
318static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000319test_long_api(PyObject* self)
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000320{
Tim Peters83c9edc2001-06-16 08:10:13 +0000321 return TESTNAME(raise_test_long_error);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000322}
323
324#undef TESTNAME
325#undef TYPENAME
326#undef F_S_TO_PY
327#undef F_PY_TO_S
328#undef F_U_TO_PY
329#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000330
331#ifdef HAVE_LONG_LONG
Tim Peters91621db2001-06-12 20:10:01 +0000332
333static PyObject *
Tim Petersd1a7da62001-06-13 00:35:57 +0000334raise_test_longlong_error(const char* msg)
335{
336 return raiseTestError("test_longlong_api", msg);
337}
338
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000339#define TESTNAME test_longlong_api_inner
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000340#define TYPENAME PY_LONG_LONG
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000341#define F_S_TO_PY PyLong_FromLongLong
342#define F_PY_TO_S PyLong_AsLongLong
343#define F_U_TO_PY PyLong_FromUnsignedLongLong
344#define F_PY_TO_U PyLong_AsUnsignedLongLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000345
346#include "testcapi_long.h"
Tim Petersd1a7da62001-06-13 00:35:57 +0000347
348static PyObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000349test_longlong_api(PyObject* self, PyObject *args)
Tim Peters91621db2001-06-12 20:10:01 +0000350{
Tim Peters83c9edc2001-06-16 08:10:13 +0000351 return TESTNAME(raise_test_longlong_error);
Tim Peters91621db2001-06-12 20:10:01 +0000352}
353
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000354#undef TESTNAME
355#undef TYPENAME
356#undef F_S_TO_PY
357#undef F_PY_TO_S
358#undef F_U_TO_PY
359#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000360
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000361/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
Tim Petersd38b1c72001-09-30 05:09:37 +0000362 for both long and int arguments. The test may leak a little memory if
363 it fails.
364*/
365static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000366test_L_code(PyObject *self)
Tim Petersd38b1c72001-09-30 05:09:37 +0000367{
368 PyObject *tuple, *num;
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000369 PY_LONG_LONG value;
Tim Petersd38b1c72001-09-30 05:09:37 +0000370
Tim Petersd38b1c72001-09-30 05:09:37 +0000371 tuple = PyTuple_New(1);
372 if (tuple == NULL)
373 return NULL;
374
375 num = PyLong_FromLong(42);
376 if (num == NULL)
377 return NULL;
378
379 PyTuple_SET_ITEM(tuple, 0, num);
380
381 value = -1;
382 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
383 return NULL;
384 if (value != 42)
385 return raiseTestError("test_L_code",
386 "L code returned wrong value for long 42");
387
388 Py_DECREF(num);
Christian Heimes217cfd12007-12-02 14:31:20 +0000389 num = PyLong_FromLong(42);
Tim Petersd38b1c72001-09-30 05:09:37 +0000390 if (num == NULL)
391 return NULL;
392
393 PyTuple_SET_ITEM(tuple, 0, num);
394
395 value = -1;
396 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
397 return NULL;
398 if (value != 42)
399 return raiseTestError("test_L_code",
400 "L code returned wrong value for int 42");
401
402 Py_DECREF(tuple);
403 Py_INCREF(Py_None);
404 return Py_None;
405}
406
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000407#endif /* ifdef HAVE_LONG_LONG */
408
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000409/* Test tuple argument processing */
410static PyObject *
411getargs_tuple(PyObject *self, PyObject *args)
412{
413 int a, b, c;
414 if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
415 return NULL;
416 return Py_BuildValue("iii", a, b, c);
417}
418
Christian Heimes380f7f22008-02-28 11:19:05 +0000419/* test PyArg_ParseTupleAndKeywords */
420static PyObject *getargs_keywords(PyObject *self, PyObject *args, PyObject *kwargs)
421{
422 static char *keywords[] = {"arg1","arg2","arg3","arg4","arg5", NULL};
423 static char *fmt="(ii)i|(i(ii))(iii)i";
424 int int_args[10]={-1, -1, -1, -1, -1, -1, -1, -1, -1, -1};
425
426 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
427 &int_args[0], &int_args[1], &int_args[2], &int_args[3], &int_args[4],
428 &int_args[5], &int_args[6], &int_args[7], &int_args[8], &int_args[9]))
429 return NULL;
430 return Py_BuildValue("iiiiiiiiii",
431 int_args[0], int_args[1], int_args[2], int_args[3], int_args[4],
432 int_args[5], int_args[6], int_args[7], int_args[8], int_args[9]);
433}
434
Thomas Heller3457e4b2003-04-24 16:14:27 +0000435/* Functions to call PyArg_ParseTuple with integer format codes,
436 and return the result.
437*/
Thomas Hellera4ea6032003-04-17 18:55:45 +0000438static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000439getargs_b(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000440{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000441 unsigned char value;
442 if (!PyArg_ParseTuple(args, "b", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000443 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000444 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000445}
446
Thomas Heller3457e4b2003-04-24 16:14:27 +0000447static PyObject *
448getargs_B(PyObject *self, PyObject *args)
449{
450 unsigned char value;
451 if (!PyArg_ParseTuple(args, "B", &value))
452 return NULL;
453 return PyLong_FromUnsignedLong((unsigned long)value);
454}
455
456static PyObject *
457getargs_H(PyObject *self, PyObject *args)
458{
459 unsigned short value;
460 if (!PyArg_ParseTuple(args, "H", &value))
461 return NULL;
462 return PyLong_FromUnsignedLong((unsigned long)value);
463}
464
465static PyObject *
466getargs_I(PyObject *self, PyObject *args)
467{
468 unsigned int value;
469 if (!PyArg_ParseTuple(args, "I", &value))
470 return NULL;
471 return PyLong_FromUnsignedLong((unsigned long)value);
472}
473
474static PyObject *
475getargs_k(PyObject *self, PyObject *args)
476{
477 unsigned long value;
478 if (!PyArg_ParseTuple(args, "k", &value))
479 return NULL;
480 return PyLong_FromUnsignedLong(value);
481}
482
483static PyObject *
484getargs_i(PyObject *self, PyObject *args)
485{
486 int value;
487 if (!PyArg_ParseTuple(args, "i", &value))
488 return NULL;
489 return PyLong_FromLong((long)value);
490}
491
Thomas Hellera4ea6032003-04-17 18:55:45 +0000492static PyObject *
493getargs_l(PyObject *self, PyObject *args)
494{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000495 long value;
496 if (!PyArg_ParseTuple(args, "l", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000497 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000498 return PyLong_FromLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000499}
500
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000501static PyObject *
502getargs_n(PyObject *self, PyObject *args)
503{
504 Py_ssize_t value;
505 if (!PyArg_ParseTuple(args, "n", &value))
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000506 return NULL;
507 return PyLong_FromSsize_t(value);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000508}
509
Thomas Hellera4ea6032003-04-17 18:55:45 +0000510#ifdef HAVE_LONG_LONG
Thomas Hellera4ea6032003-04-17 18:55:45 +0000511static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000512getargs_L(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000513{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000514 PY_LONG_LONG value;
515 if (!PyArg_ParseTuple(args, "L", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000516 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000517 return PyLong_FromLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000518}
519
Thomas Hellera4ea6032003-04-17 18:55:45 +0000520static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000521getargs_K(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000522{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000523 unsigned PY_LONG_LONG value;
524 if (!PyArg_ParseTuple(args, "K", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000525 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000526 return PyLong_FromUnsignedLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000527}
528#endif
529
530/* This function not only tests the 'k' getargs code, but also the
Christian Heimes217cfd12007-12-02 14:31:20 +0000531 PyLong_AsUnsignedLongMask() and PyLong_AsUnsignedLongMask() functions. */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000532static PyObject *
533test_k_code(PyObject *self)
534{
535 PyObject *tuple, *num;
536 unsigned long value;
537
538 tuple = PyTuple_New(1);
539 if (tuple == NULL)
540 return NULL;
541
Neal Norwitz699cbb72003-04-22 01:28:57 +0000542 /* a number larger than ULONG_MAX even on 64-bit platforms */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000543 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
544 if (num == NULL)
545 return NULL;
546
Christian Heimes217cfd12007-12-02 14:31:20 +0000547 value = PyLong_AsUnsignedLongMask(num);
Neal Norwitz699cbb72003-04-22 01:28:57 +0000548 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000549 return raiseTestError("test_k_code",
Christian Heimes217cfd12007-12-02 14:31:20 +0000550 "PyLong_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +0000551
552 PyTuple_SET_ITEM(tuple, 0, num);
553
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000554 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000555 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
556 return NULL;
Neal Norwitz699cbb72003-04-22 01:28:57 +0000557 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000558 return raiseTestError("test_k_code",
559 "k code returned wrong value for long 0xFFF...FFF");
560
561 Py_DECREF(num);
562 num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
563 if (num == NULL)
564 return NULL;
565
Christian Heimes217cfd12007-12-02 14:31:20 +0000566 value = PyLong_AsUnsignedLongMask(num);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000567 if (value != (unsigned long)-0x42)
568 return raiseTestError("test_k_code",
Christian Heimes217cfd12007-12-02 14:31:20 +0000569 "PyLong_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +0000570
571 PyTuple_SET_ITEM(tuple, 0, num);
572
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000573 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000574 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
575 return NULL;
576 if (value != (unsigned long)-0x42)
577 return raiseTestError("test_k_code",
578 "k code returned wrong value for long -0xFFF..000042");
579
580 Py_DECREF(tuple);
581 Py_INCREF(Py_None);
582 return Py_None;
583}
584
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000585
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +0000586/* Test the s and z codes for PyArg_ParseTuple.
587*/
588static PyObject *
589test_s_code(PyObject *self)
590{
591 /* Unicode strings should be accepted */
592 PyObject *tuple, *obj;
593 char *value;
594
595 tuple = PyTuple_New(1);
596 if (tuple == NULL)
597 return NULL;
598
599 obj = PyUnicode_Decode("t\xeate", strlen("t\xeate"),
600 "latin-1", NULL);
601 if (obj == NULL)
602 return NULL;
603
604 PyTuple_SET_ITEM(tuple, 0, obj);
605
606 /* These two blocks used to raise a TypeError:
607 * "argument must be string without null bytes, not str"
608 */
609 if (PyArg_ParseTuple(tuple, "s:test_s_code1", &value) < 0)
610 return NULL;
611
612 if (PyArg_ParseTuple(tuple, "z:test_s_code2", &value) < 0)
613 return NULL;
614
Alexandre Vassalottib645bc72008-05-15 22:06:59 +0000615 Py_DECREF(tuple);
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +0000616 Py_RETURN_NONE;
617}
618
Mark Dickinsonf08173b2009-12-03 10:59:46 +0000619static PyObject *
620test_bug_7414(PyObject *self)
621{
622 /* Issue #7414: for PyArg_ParseTupleAndKeywords, 'C' code wasn't being
623 skipped properly in skipitem() */
624 int a = 0, b = 0, result;
625 char *kwlist[] = {"a", "b", NULL};
626 PyObject *tuple = NULL, *dict = NULL, *b_str;
627
628 tuple = PyTuple_New(0);
629 if (tuple == NULL)
630 goto failure;
631 dict = PyDict_New();
632 if (dict == NULL)
633 goto failure;
634 b_str = PyUnicode_FromString("b");
635 if (b_str == NULL)
636 goto failure;
637 result = PyDict_SetItemString(dict, "b", b_str);
638 Py_DECREF(b_str);
639 if (result < 0)
640 goto failure;
641
642 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|CC",
643 kwlist, &a, &b);
644 if (!result)
645 goto failure;
646
647 if (a != 0)
648 return raiseTestError("test_bug_7414",
649 "C format code not skipped properly");
650 if (b != 'b')
651 return raiseTestError("test_bug_7414",
652 "C format code returned wrong value");
653
654 Py_DECREF(dict);
655 Py_DECREF(tuple);
656 Py_RETURN_NONE;
657
658 failure:
659 Py_XDECREF(dict);
660 Py_XDECREF(tuple);
661 return NULL;
662}
663
664
Benjamin Peterson92035012008-12-27 16:00:54 +0000665static volatile int x;
666
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000667/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
668 of an error.
669*/
670static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000671test_u_code(PyObject *self)
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000672{
673 PyObject *tuple, *obj;
674 Py_UNICODE *value;
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000675 Py_ssize_t len;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000676
Benjamin Peterson206e3072008-10-19 14:07:49 +0000677 /* issue4122: Undefined reference to _Py_ascii_whitespace on Windows */
678 /* Just use the macro and check that it compiles */
Christian Heimes727c2c52008-11-12 09:04:04 +0000679 x = Py_UNICODE_ISSPACE(25);
Benjamin Peterson206e3072008-10-19 14:07:49 +0000680
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000681 tuple = PyTuple_New(1);
682 if (tuple == NULL)
683 return NULL;
684
685 obj = PyUnicode_Decode("test", strlen("test"),
686 "ascii", NULL);
687 if (obj == NULL)
688 return NULL;
689
690 PyTuple_SET_ITEM(tuple, 0, obj);
691
692 value = 0;
693 if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
694 return NULL;
695 if (value != PyUnicode_AS_UNICODE(obj))
696 return raiseTestError("test_u_code",
697 "u code returned wrong value for u'test'");
698 value = 0;
699 if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
700 return NULL;
701 if (value != PyUnicode_AS_UNICODE(obj) ||
702 len != PyUnicode_GET_SIZE(obj))
703 return raiseTestError("test_u_code",
704 "u# code returned wrong values for u'test'");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000705
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000706 Py_DECREF(tuple);
707 Py_INCREF(Py_None);
708 return Py_None;
709}
710
Guido van Rossumfb67be22007-08-29 18:38:11 +0000711/* Test Z and Z# codes for PyArg_ParseTuple */
712static PyObject *
713test_Z_code(PyObject *self)
714{
715 PyObject *tuple, *obj;
716 Py_UNICODE *value1, *value2;
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000717 Py_ssize_t len1, len2;
Guido van Rossumfb67be22007-08-29 18:38:11 +0000718
719 tuple = PyTuple_New(2);
720 if (tuple == NULL)
721 return NULL;
722
723 obj = PyUnicode_FromString("test");
724 PyTuple_SET_ITEM(tuple, 0, obj);
725 Py_INCREF(Py_None);
726 PyTuple_SET_ITEM(tuple, 1, Py_None);
727
728 /* swap values on purpose */
729 value1 = NULL;
730 value2 = PyUnicode_AS_UNICODE(obj);
731
732 /* Test Z for both values */
733 if (PyArg_ParseTuple(tuple, "ZZ:test_Z_code", &value1, &value2) < 0)
734 return NULL;
735 if (value1 != PyUnicode_AS_UNICODE(obj))
736 return raiseTestError("test_Z_code",
737 "Z code returned wrong value for 'test'");
738 if (value2 != NULL)
739 return raiseTestError("test_Z_code",
740 "Z code returned wrong value for None");
741
742 value1 = NULL;
743 value2 = PyUnicode_AS_UNICODE(obj);
744 len1 = -1;
745 len2 = -1;
746
747 /* Test Z# for both values */
748 if (PyArg_ParseTuple(tuple, "Z#Z#:test_Z_code", &value1, &len1,
749 &value2, &len2) < 0)
750 return NULL;
751 if (value1 != PyUnicode_AS_UNICODE(obj) ||
752 len1 != PyUnicode_GET_SIZE(obj))
753 return raiseTestError("test_Z_code",
754 "Z# code returned wrong values for 'test'");
755 if (value2 != NULL ||
756 len2 != 0)
757 return raiseTestError("test_Z_code",
758 "Z# code returned wrong values for None'");
759
760 Py_DECREF(tuple);
761 Py_RETURN_NONE;
762}
763
Thomas Wouters477c8d52006-05-27 19:21:47 +0000764static PyObject *
Mark Dickinson081dfee2009-03-18 14:47:41 +0000765test_widechar(PyObject *self)
766{
767#if defined(SIZEOF_WCHAR_T) && (SIZEOF_WCHAR_T == 4)
768 const wchar_t wtext[2] = {(wchar_t)0x10ABCDu};
769 size_t wtextlen = 1;
770#else
771 const wchar_t wtext[3] = {(wchar_t)0xDBEAu, (wchar_t)0xDFCDu};
772 size_t wtextlen = 2;
773#endif
774 PyObject *wide, *utf8;
775
776 wide = PyUnicode_FromWideChar(wtext, wtextlen);
777 if (wide == NULL)
778 return NULL;
779
780 utf8 = PyUnicode_FromString("\xf4\x8a\xaf\x8d");
781 if (utf8 == NULL) {
782 Py_DECREF(wide);
783 return NULL;
784 }
785
786 if (PyUnicode_GET_SIZE(wide) != PyUnicode_GET_SIZE(utf8)) {
787 Py_DECREF(wide);
788 Py_DECREF(utf8);
789 return raiseTestError("test_widechar",
790 "wide string and utf8 string "
791 "have different length");
792 }
793 if (PyUnicode_Compare(wide, utf8)) {
794 Py_DECREF(wide);
795 Py_DECREF(utf8);
796 if (PyErr_Occurred())
797 return NULL;
798 return raiseTestError("test_widechar",
799 "wide string and utf8 string "
800 "are different");
801 }
802
803 Py_DECREF(wide);
804 Py_DECREF(utf8);
805 Py_RETURN_NONE;
806}
807
808static PyObject *
Benjamin Peterson92035012008-12-27 16:00:54 +0000809test_empty_argparse(PyObject *self)
810{
811 /* Test that formats can begin with '|'. See issue #4720. */
812 PyObject *tuple, *dict = NULL;
813 static char *kwlist[] = {NULL};
814 int result;
815 tuple = PyTuple_New(0);
816 if (!tuple)
817 return NULL;
818 if ((result = PyArg_ParseTuple(tuple, "|:test_empty_argparse")) < 0)
819 goto done;
820 dict = PyDict_New();
821 if (!dict)
822 goto done;
823 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|:test_empty_argparse", kwlist);
824 done:
825 Py_DECREF(tuple);
826 Py_XDECREF(dict);
827 if (result < 0)
828 return NULL;
829 else {
830 Py_RETURN_NONE;
831 }
832}
833
834static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +0000835codec_incrementalencoder(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000836{
837 const char *encoding, *errors = NULL;
838 if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
839 &encoding, &errors))
840 return NULL;
841 return PyCodec_IncrementalEncoder(encoding, errors);
842}
843
Thomas Wouters477c8d52006-05-27 19:21:47 +0000844static PyObject *
845codec_incrementaldecoder(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000846{
847 const char *encoding, *errors = NULL;
848 if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
849 &encoding, &errors))
850 return NULL;
851 return PyCodec_IncrementalDecoder(encoding, errors);
852}
853
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000854
Tim Peters5b8132f2003-01-31 15:52:05 +0000855/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
Tim Petersbaefd9e2003-01-28 20:37:45 +0000856static PyObject *
857test_long_numbits(PyObject *self)
858{
Tim Peters5b8132f2003-01-31 15:52:05 +0000859 struct triple {
Tim Petersbaefd9e2003-01-28 20:37:45 +0000860 long input;
Tim Peters5b8132f2003-01-31 15:52:05 +0000861 size_t nbits;
862 int sign;
863 } testcases[] = {{0, 0, 0},
864 {1L, 1, 1},
865 {-1L, 1, -1},
866 {2L, 2, 1},
867 {-2L, 2, -1},
868 {3L, 2, 1},
869 {-3L, 2, -1},
870 {4L, 3, 1},
871 {-4L, 3, -1},
872 {0x7fffL, 15, 1}, /* one Python long digit */
873 {-0x7fffL, 15, -1},
874 {0xffffL, 16, 1},
875 {-0xffffL, 16, -1},
876 {0xfffffffL, 28, 1},
877 {-0xfffffffL, 28, -1}};
Tim Petersbaefd9e2003-01-28 20:37:45 +0000878 int i;
879
Tim Peters5b8132f2003-01-31 15:52:05 +0000880 for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
881 PyObject *plong = PyLong_FromLong(testcases[i].input);
Tim Petersbaefd9e2003-01-28 20:37:45 +0000882 size_t nbits = _PyLong_NumBits(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +0000883 int sign = _PyLong_Sign(plong);
Tim Petersbaefd9e2003-01-28 20:37:45 +0000884
885 Py_DECREF(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +0000886 if (nbits != testcases[i].nbits)
Tim Petersbaefd9e2003-01-28 20:37:45 +0000887 return raiseTestError("test_long_numbits",
Tim Peters5b8132f2003-01-31 15:52:05 +0000888 "wrong result for _PyLong_NumBits");
889 if (sign != testcases[i].sign)
890 return raiseTestError("test_long_numbits",
891 "wrong result for _PyLong_Sign");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000892 }
893 Py_INCREF(Py_None);
894 return Py_None;
895}
896
Thomas Heller519a0422007-11-15 20:48:54 +0000897/* Example passing NULLs to PyObject_Str(NULL). */
Thomas Woutersa9773292006-04-21 09:43:23 +0000898
899static PyObject *
900test_null_strings(PyObject *self)
901{
Thomas Heller519a0422007-11-15 20:48:54 +0000902 PyObject *o1 = PyObject_Str(NULL), *o2 = PyObject_Str(NULL);
Thomas Woutersa9773292006-04-21 09:43:23 +0000903 PyObject *tuple = PyTuple_Pack(2, o1, o2);
904 Py_XDECREF(o1);
905 Py_XDECREF(o2);
906 return tuple;
907}
908
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000909static PyObject *
910raise_exception(PyObject *self, PyObject *args)
911{
912 PyObject *exc;
913 PyObject *exc_args, *v;
914 int num_args, i;
915
916 if (!PyArg_ParseTuple(args, "Oi:raise_exception",
917 &exc, &num_args))
918 return NULL;
919
920 exc_args = PyTuple_New(num_args);
921 if (exc_args == NULL)
922 return NULL;
923 for (i = 0; i < num_args; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +0000924 v = PyLong_FromLong(i);
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000925 if (v == NULL) {
926 Py_DECREF(exc_args);
927 return NULL;
928 }
929 PyTuple_SET_ITEM(exc_args, i, v);
930 }
931 PyErr_SetObject(exc, exc_args);
Michael W. Hudsonf02bcee2003-08-15 13:03:30 +0000932 Py_DECREF(exc_args);
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000933 return NULL;
934}
Tim Peters91621db2001-06-12 20:10:01 +0000935
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000936#ifdef WITH_THREAD
937
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000938/* test_thread_state spawns a thread of its own, and that thread releases
939 * `thread_done` when it's finished. The driver code has to know when the
940 * thread finishes, because the thread uses a PyObject (the callable) that
941 * may go away when the driver finishes. The former lack of this explicit
942 * synchronization caused rare segfaults, so rare that they were seen only
943 * on a Mac buildbot (although they were possible on any box).
944 */
945static PyThread_type_lock thread_done = NULL;
946
Benjamin Petersona786b022008-08-25 21:05:21 +0000947static int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000948_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000949{
950 PyObject *rc;
Benjamin Petersona786b022008-08-25 21:05:21 +0000951 int success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000952 PyGILState_STATE s = PyGILState_Ensure();
Thomas Wouters89f507f2006-12-13 04:49:30 +0000953 rc = PyObject_CallFunction((PyObject *)callable, "");
Benjamin Petersona786b022008-08-25 21:05:21 +0000954 success = (rc != NULL);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000955 Py_XDECREF(rc);
956 PyGILState_Release(s);
Benjamin Petersona786b022008-08-25 21:05:21 +0000957 return success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000958}
959
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000960/* Same thing, but releases `thread_done` when it returns. This variant
961 * should be called only from threads spawned by test_thread_state().
962 */
963static void
964_make_call_from_thread(void *callable)
965{
966 _make_call(callable);
967 PyThread_release_lock(thread_done);
968}
969
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000970static PyObject *
971test_thread_state(PyObject *self, PyObject *args)
972{
973 PyObject *fn;
Benjamin Petersona786b022008-08-25 21:05:21 +0000974 int success = 1;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000975
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000976 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
977 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000978
Benjamin Petersona786b022008-08-25 21:05:21 +0000979 if (!PyCallable_Check(fn)) {
980 PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
981 fn->ob_type->tp_name);
982 return NULL;
983 }
984
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000985 /* Ensure Python is set up for threading */
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000986 PyEval_InitThreads();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000987 thread_done = PyThread_allocate_lock();
988 if (thread_done == NULL)
989 return PyErr_NoMemory();
990 PyThread_acquire_lock(thread_done, 1);
991
992 /* Start a new thread with our callback. */
993 PyThread_start_new_thread(_make_call_from_thread, fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000994 /* Make the callback with the thread lock held by this thread */
Benjamin Petersona786b022008-08-25 21:05:21 +0000995 success &= _make_call(fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000996 /* Do it all again, but this time with the thread-lock released */
997 Py_BEGIN_ALLOW_THREADS
Benjamin Petersona786b022008-08-25 21:05:21 +0000998 success &= _make_call(fn);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000999 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001000 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001001
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001002 /* And once more with and without a thread
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001003 XXX - should use a lock and work out exactly what we are trying
1004 to test <wink>
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001005 */
1006 Py_BEGIN_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001007 PyThread_start_new_thread(_make_call_from_thread, fn);
Benjamin Petersona786b022008-08-25 21:05:21 +00001008 success &= _make_call(fn);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001009 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001010 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001011
Thomas Wouters89f507f2006-12-13 04:49:30 +00001012 /* Release lock we acquired above. This is required on HP-UX. */
1013 PyThread_release_lock(thread_done);
1014
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001015 PyThread_free_lock(thread_done);
Benjamin Petersona786b022008-08-25 21:05:21 +00001016 if (!success)
1017 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001018 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001019}
Benjamin Petersona54c9092009-01-13 02:11:23 +00001020
1021/* test Py_AddPendingCalls using threads */
1022static int _pending_callback(void *arg)
1023{
1024 /* we assume the argument is callable object to which we own a reference */
1025 PyObject *callable = (PyObject *)arg;
1026 PyObject *r = PyObject_CallObject(callable, NULL);
1027 Py_DECREF(callable);
1028 Py_XDECREF(r);
1029 return r != NULL ? 0 : -1;
1030}
1031
1032/* The following requests n callbacks to _pending_callback. It can be
1033 * run from any python thread.
1034 */
1035PyObject *pending_threadfunc(PyObject *self, PyObject *arg)
1036{
1037 PyObject *callable;
1038 int r;
1039 if (PyArg_ParseTuple(arg, "O", &callable) == 0)
1040 return NULL;
1041
1042 /* create the reference for the callbackwhile we hold the lock */
1043 Py_INCREF(callable);
1044
1045 Py_BEGIN_ALLOW_THREADS
1046 r = Py_AddPendingCall(&_pending_callback, callable);
1047 Py_END_ALLOW_THREADS
1048
1049 if (r<0) {
1050 Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
1051 Py_INCREF(Py_False);
1052 return Py_False;
1053 }
1054 Py_INCREF(Py_True);
1055 return Py_True;
1056}
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001057#endif
1058
Neal Norwitzb0d26332007-08-25 00:49:05 +00001059/* Some tests of PyUnicode_FromFormat(). This needs more tests. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001060static PyObject *
1061test_string_from_format(PyObject *self, PyObject *args)
1062{
1063 PyObject *result;
1064 char *msg;
1065
1066#define CHECK_1_FORMAT(FORMAT, TYPE) \
Neal Norwitzb0d26332007-08-25 00:49:05 +00001067 result = PyUnicode_FromFormat(FORMAT, (TYPE)1); \
Thomas Wouters477c8d52006-05-27 19:21:47 +00001068 if (result == NULL) \
1069 return NULL; \
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00001070 if (strcmp(_PyUnicode_AsString(result), "1")) { \
Thomas Wouters477c8d52006-05-27 19:21:47 +00001071 msg = FORMAT " failed at 1"; \
1072 goto Fail; \
1073 } \
1074 Py_DECREF(result)
1075
1076 CHECK_1_FORMAT("%d", int);
1077 CHECK_1_FORMAT("%ld", long);
1078 /* The z width modifier was added in Python 2.5. */
1079 CHECK_1_FORMAT("%zd", Py_ssize_t);
1080
1081 /* The u type code was added in Python 2.5. */
1082 CHECK_1_FORMAT("%u", unsigned int);
1083 CHECK_1_FORMAT("%lu", unsigned long);
1084 CHECK_1_FORMAT("%zu", size_t);
1085
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001086 /* "%lld" and "%llu" support added in Python 2.7. */
1087#ifdef HAVE_LONG_LONG
1088 CHECK_1_FORMAT("%llu", unsigned PY_LONG_LONG);
1089 CHECK_1_FORMAT("%lld", PY_LONG_LONG);
1090#endif
1091
Thomas Wouters477c8d52006-05-27 19:21:47 +00001092 Py_RETURN_NONE;
1093
1094 Fail:
1095 Py_XDECREF(result);
1096 return raiseTestError("test_string_from_format", msg);
1097
1098#undef CHECK_1_FORMAT
1099}
1100
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001101/* This is here to provide a docstring for test_descr. */
1102static PyObject *
1103test_with_docstring(PyObject *self)
1104{
1105 Py_RETURN_NONE;
1106}
1107
Mark Dickinson725bfd82009-05-03 20:33:40 +00001108/* Test PyOS_string_to_double. */
1109static PyObject *
1110test_string_to_double(PyObject *self) {
1111 double result;
1112 char *msg;
1113
1114#define CHECK_STRING(STR, expected) \
1115 result = PyOS_string_to_double(STR, NULL, NULL); \
1116 if (result == -1.0 && PyErr_Occurred()) \
1117 return NULL; \
1118 if (result != expected) { \
1119 msg = "conversion of " STR " to float failed"; \
1120 goto fail; \
1121 }
1122
1123#define CHECK_INVALID(STR) \
1124 result = PyOS_string_to_double(STR, NULL, NULL); \
1125 if (result == -1.0 && PyErr_Occurred()) { \
1126 if (PyErr_ExceptionMatches(PyExc_ValueError)) \
1127 PyErr_Clear(); \
1128 else \
1129 return NULL; \
1130 } \
1131 else { \
1132 msg = "conversion of " STR " didn't raise ValueError"; \
1133 goto fail; \
1134 }
1135
1136 CHECK_STRING("0.1", 0.1);
1137 CHECK_STRING("1.234", 1.234);
1138 CHECK_STRING("-1.35", -1.35);
1139 CHECK_STRING(".1e01", 1.0);
1140 CHECK_STRING("2.e-2", 0.02);
1141
1142 CHECK_INVALID(" 0.1");
1143 CHECK_INVALID("\t\n-3");
1144 CHECK_INVALID(".123 ");
1145 CHECK_INVALID("3\n");
1146 CHECK_INVALID("123abc");
1147
1148 Py_RETURN_NONE;
1149 fail:
1150 return raiseTestError("test_string_to_double", msg);
1151#undef CHECK_STRING
1152#undef CHECK_INVALID
1153}
1154
1155
Benjamin Petersonb173f782009-05-05 22:31:58 +00001156/* Coverage testing of capsule objects. */
1157
1158static const char *capsule_name = "capsule name";
1159static char *capsule_pointer = "capsule pointer";
1160static char *capsule_context = "capsule context";
1161static const char *capsule_error = NULL;
1162static int
1163capsule_destructor_call_count = 0;
1164
1165static void
1166capsule_destructor(PyObject *o) {
1167 capsule_destructor_call_count++;
1168 if (PyCapsule_GetContext(o) != capsule_context) {
1169 capsule_error = "context did not match in destructor!";
1170 } else if (PyCapsule_GetDestructor(o) != capsule_destructor) {
1171 capsule_error = "destructor did not match in destructor! (woah!)";
1172 } else if (PyCapsule_GetName(o) != capsule_name) {
1173 capsule_error = "name did not match in destructor!";
1174 } else if (PyCapsule_GetPointer(o, capsule_name) != capsule_pointer) {
1175 capsule_error = "pointer did not match in destructor!";
1176 }
1177}
1178
1179typedef struct {
1180 char *name;
1181 char *module;
1182 char *attribute;
1183} known_capsule;
1184
1185static PyObject *
1186test_capsule(PyObject *self, PyObject *args)
1187{
1188 PyObject *object;
1189 const char *error = NULL;
1190 void *pointer;
1191 void *pointer2;
1192 known_capsule known_capsules[] = {
1193 #define KNOWN_CAPSULE(module, name) { module "." name, module, name }
1194 KNOWN_CAPSULE("_socket", "CAPI"),
1195 KNOWN_CAPSULE("_curses", "_C_API"),
1196 KNOWN_CAPSULE("datetime", "datetime_CAPI"),
1197 { NULL, NULL },
1198 };
1199 known_capsule *known = &known_capsules[0];
1200
1201#define FAIL(x) { error = (x); goto exit; }
1202
1203#define CHECK_DESTRUCTOR \
1204 if (capsule_error) { \
1205 FAIL(capsule_error); \
1206 } \
1207 else if (!capsule_destructor_call_count) { \
1208 FAIL("destructor not called!"); \
1209 } \
1210 capsule_destructor_call_count = 0; \
1211
1212 object = PyCapsule_New(capsule_pointer, capsule_name, capsule_destructor);
1213 PyCapsule_SetContext(object, capsule_context);
1214 capsule_destructor(object);
1215 CHECK_DESTRUCTOR;
1216 Py_DECREF(object);
1217 CHECK_DESTRUCTOR;
1218
1219 object = PyCapsule_New(known, "ignored", NULL);
1220 PyCapsule_SetPointer(object, capsule_pointer);
1221 PyCapsule_SetName(object, capsule_name);
1222 PyCapsule_SetDestructor(object, capsule_destructor);
1223 PyCapsule_SetContext(object, capsule_context);
1224 capsule_destructor(object);
1225 CHECK_DESTRUCTOR;
1226 /* intentionally access using the wrong name */
1227 pointer2 = PyCapsule_GetPointer(object, "the wrong name");
1228 if (!PyErr_Occurred()) {
1229 FAIL("PyCapsule_GetPointer should have failed but did not!");
1230 }
1231 PyErr_Clear();
1232 if (pointer2) {
1233 if (pointer2 == capsule_pointer) {
1234 FAIL("PyCapsule_GetPointer should not have"
1235 " returned the internal pointer!");
1236 } else {
1237 FAIL("PyCapsule_GetPointer should have "
1238 "returned NULL pointer but did not!");
1239 }
1240 }
1241 PyCapsule_SetDestructor(object, NULL);
1242 Py_DECREF(object);
1243 if (capsule_destructor_call_count) {
1244 FAIL("destructor called when it should not have been!");
1245 }
1246
1247 for (known = &known_capsules[0]; known->module != NULL; known++) {
1248 /* yeah, ordinarily I wouldn't do this either,
1249 but it's fine for this test harness.
1250 */
1251 static char buffer[256];
1252#undef FAIL
1253#define FAIL(x) \
1254 { \
1255 sprintf(buffer, "%s module: \"%s\" attribute: \"%s\"", \
1256 x, known->module, known->attribute); \
1257 error = buffer; \
1258 goto exit; \
1259 } \
1260
1261 PyObject *module = PyImport_ImportModule(known->module);
1262 if (module) {
1263 pointer = PyCapsule_Import(known->name, 0);
1264 if (!pointer) {
1265 Py_DECREF(module);
1266 FAIL("PyCapsule_GetPointer returned NULL unexpectedly!");
1267 }
1268 object = PyObject_GetAttrString(module, known->attribute);
1269 if (!object) {
1270 Py_DECREF(module);
1271 return NULL;
1272 }
1273 pointer2 = PyCapsule_GetPointer(object,
1274 "weebles wobble but they don't fall down");
1275 if (!PyErr_Occurred()) {
1276 Py_DECREF(object);
1277 Py_DECREF(module);
1278 FAIL("PyCapsule_GetPointer should have failed but did not!");
1279 }
1280 PyErr_Clear();
1281 if (pointer2) {
1282 Py_DECREF(module);
1283 Py_DECREF(object);
1284 if (pointer2 == pointer) {
1285 FAIL("PyCapsule_GetPointer should not have"
1286 " returned its internal pointer!");
1287 } else {
1288 FAIL("PyCapsule_GetPointer should have"
1289 " returned NULL pointer but did not!");
1290 }
1291 }
1292 Py_DECREF(object);
1293 Py_DECREF(module);
1294 }
1295 }
1296
1297 exit:
1298 if (error) {
1299 return raiseTestError("test_capsule", error);
1300 }
1301 Py_RETURN_NONE;
1302#undef FAIL
1303}
1304
Guido van Rossumddefaf32007-01-14 03:31:43 +00001305#ifdef HAVE_GETTIMEOFDAY
1306/* Profiling of integer performance */
Martin v. Löwis1c951552008-06-13 07:48:19 +00001307static void print_delta(int test, struct timeval *s, struct timeval *e)
Guido van Rossumddefaf32007-01-14 03:31:43 +00001308{
1309 e->tv_sec -= s->tv_sec;
1310 e->tv_usec -= s->tv_usec;
1311 if (e->tv_usec < 0) {
1312 e->tv_sec -=1;
1313 e->tv_usec += 1000000;
1314 }
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001315 printf("Test %d: %d.%06ds\n", test, (int)e->tv_sec, (int)e->tv_usec);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001316}
1317
1318static PyObject *
1319profile_int(PyObject *self, PyObject* args)
1320{
1321 int i, k;
1322 struct timeval start, stop;
1323 PyObject *single, **multiple, *op1, *result;
1324
1325 /* Test 1: Allocate and immediately deallocate
1326 many small integers */
1327 gettimeofday(&start, NULL);
1328 for(k=0; k < 20000; k++)
1329 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001330 single = PyLong_FromLong(i);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001331 Py_DECREF(single);
1332 }
1333 gettimeofday(&stop, NULL);
1334 print_delta(1, &start, &stop);
1335
1336 /* Test 2: Allocate and immediately deallocate
1337 many large integers */
1338 gettimeofday(&start, NULL);
1339 for(k=0; k < 20000; k++)
1340 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001341 single = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001342 Py_DECREF(single);
1343 }
1344 gettimeofday(&stop, NULL);
1345 print_delta(2, &start, &stop);
1346
1347 /* Test 3: Allocate a few integers, then release
1348 them all simultaneously. */
1349 multiple = malloc(sizeof(PyObject*) * 1000);
1350 gettimeofday(&start, NULL);
1351 for(k=0; k < 20000; k++) {
1352 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001353 multiple[i] = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001354 }
1355 for(i=0; i < 1000; i++) {
1356 Py_DECREF(multiple[i]);
1357 }
1358 }
1359 gettimeofday(&stop, NULL);
1360 print_delta(3, &start, &stop);
1361
1362 /* Test 4: Allocate many integers, then release
1363 them all simultaneously. */
1364 multiple = malloc(sizeof(PyObject*) * 1000000);
1365 gettimeofday(&start, NULL);
1366 for(k=0; k < 20; k++) {
1367 for(i=0; i < 1000000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001368 multiple[i] = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001369 }
1370 for(i=0; i < 1000000; i++) {
1371 Py_DECREF(multiple[i]);
1372 }
1373 }
1374 gettimeofday(&stop, NULL);
1375 print_delta(4, &start, &stop);
1376
1377 /* Test 5: Allocate many integers < 32000 */
1378 multiple = malloc(sizeof(PyObject*) * 1000000);
1379 gettimeofday(&start, NULL);
1380 for(k=0; k < 10; k++) {
1381 for(i=0; i < 1000000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001382 multiple[i] = PyLong_FromLong(i+1000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001383 }
1384 for(i=0; i < 1000000; i++) {
1385 Py_DECREF(multiple[i]);
1386 }
1387 }
1388 gettimeofday(&stop, NULL);
1389 print_delta(5, &start, &stop);
1390
1391 /* Test 6: Perform small int addition */
Christian Heimes217cfd12007-12-02 14:31:20 +00001392 op1 = PyLong_FromLong(1);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001393 gettimeofday(&start, NULL);
1394 for(i=0; i < 10000000; i++) {
1395 result = PyNumber_Add(op1, op1);
1396 Py_DECREF(result);
1397 }
1398 gettimeofday(&stop, NULL);
1399 Py_DECREF(op1);
1400 print_delta(6, &start, &stop);
1401
1402 /* Test 7: Perform medium int addition */
Christian Heimes217cfd12007-12-02 14:31:20 +00001403 op1 = PyLong_FromLong(1000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001404 gettimeofday(&start, NULL);
1405 for(i=0; i < 10000000; i++) {
1406 result = PyNumber_Add(op1, op1);
1407 Py_DECREF(result);
1408 }
1409 gettimeofday(&stop, NULL);
1410 Py_DECREF(op1);
1411 print_delta(7, &start, &stop);
1412
1413 Py_INCREF(Py_None);
1414 return Py_None;
1415}
1416#endif
1417
Christian Heimes81ee3ef2008-05-04 22:42:01 +00001418/* To test the format of tracebacks as printed out. */
1419static PyObject *
1420traceback_print(PyObject *self, PyObject *args)
1421{
1422 PyObject *file;
1423 PyObject *traceback;
1424 int result;
1425
1426 if (!PyArg_ParseTuple(args, "OO:traceback_print",
1427 &traceback, &file))
1428 return NULL;
1429
1430 result = PyTraceBack_Print(traceback, file);
1431 if (result < 0)
1432 return NULL;
1433 Py_RETURN_NONE;
1434}
1435
Benjamin Petersone6528212008-07-15 15:32:09 +00001436/* To test the format of exceptions as printed out. */
1437static PyObject *
1438exception_print(PyObject *self, PyObject *args)
1439{
1440 PyObject *value;
1441 PyObject *tb;
1442
1443 if (!PyArg_ParseTuple(args, "O:exception_print",
1444 &value))
1445 return NULL;
Benjamin Peterson6784eb72008-08-23 20:32:27 +00001446 if (!PyExceptionInstance_Check(value)) {
1447 PyErr_Format(PyExc_TypeError, "an exception instance is required");
1448 return NULL;
1449 }
Benjamin Petersone6528212008-07-15 15:32:09 +00001450
1451 tb = PyException_GetTraceback(value);
1452 PyErr_Display((PyObject *) Py_TYPE(value), value, tb);
1453 Py_XDECREF(tb);
1454
1455 Py_RETURN_NONE;
1456}
1457
1458
1459
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001460
1461/* reliably raise a MemoryError */
1462static PyObject *
1463raise_memoryerror(PyObject *self)
1464{
1465 PyErr_NoMemory();
1466 return NULL;
1467}
1468
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00001469/* Issue 6012 */
1470static PyObject *str1, *str2;
1471static int
1472failing_converter(PyObject *obj, void *arg)
1473{
1474 /* Clone str1, then let the conversion fail. */
1475 assert(str1);
1476 str2 = str1;
1477 Py_INCREF(str2);
1478 return 0;
1479}
1480static PyObject*
1481argparsing(PyObject *o, PyObject *args)
1482{
1483 PyObject *res;
1484 str1 = str2 = NULL;
1485 if (!PyArg_ParseTuple(args, "O&O&",
1486 PyUnicode_FSConverter, &str1,
1487 failing_converter, &str2)) {
1488 if (!str2)
1489 /* argument converter not called? */
1490 return NULL;
1491 /* Should be 1 */
1492 res = PyLong_FromLong(Py_REFCNT(str2));
1493 Py_DECREF(str2);
1494 PyErr_Clear();
1495 return res;
1496 }
1497 Py_RETURN_NONE;
1498}
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001499
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00001500/* To test that the result of PyCode_NewEmpty has the right members. */
1501static PyObject *
1502code_newempty(PyObject *self, PyObject *args)
1503{
1504 const char *filename;
1505 const char *funcname;
1506 int firstlineno;
1507
1508 if (!PyArg_ParseTuple(args, "ssi:code_newempty",
1509 &filename, &funcname, &firstlineno))
1510 return NULL;
1511
1512 return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
1513}
1514
Tim Peters9ea17ac2001-02-02 05:57:15 +00001515static PyMethodDef TestMethods[] = {
Fred Drakeacee69f2002-04-01 14:28:58 +00001516 {"raise_exception", raise_exception, METH_VARARGS},
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001517 {"raise_memoryerror", (PyCFunction)raise_memoryerror, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001518 {"test_config", (PyCFunction)test_config, METH_NOARGS},
1519 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
1520 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
Nick Coghlanf1f2f682008-12-30 07:29:12 +00001521 {"test_lazy_hash_inheritance", (PyCFunction)test_lazy_hash_inheritance,METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001522 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
Tim Petersbaefd9e2003-01-28 20:37:45 +00001523 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
Thomas Hellera4ea6032003-04-17 18:55:45 +00001524 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Benjamin Peterson92035012008-12-27 16:00:54 +00001525 {"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
Mark Dickinsonf08173b2009-12-03 10:59:46 +00001526 {"test_bug_7414", (PyCFunction)test_bug_7414, METH_NOARGS},
Thomas Woutersa9773292006-04-21 09:43:23 +00001527 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
Thomas Wouters477c8d52006-05-27 19:21:47 +00001528 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001529 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
1530 PyDoc_STR("This is a pretty normal docstring.")},
Benjamin Petersonb173f782009-05-05 22:31:58 +00001531 {"test_string_to_double", (PyCFunction)test_string_to_double, METH_NOARGS},
1532 {"test_capsule", (PyCFunction)test_capsule, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001533 {"getargs_tuple", getargs_tuple, METH_VARARGS},
Christian Heimes380f7f22008-02-28 11:19:05 +00001534 {"getargs_keywords", (PyCFunction)getargs_keywords,
1535 METH_VARARGS|METH_KEYWORDS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001536 {"getargs_b", getargs_b, METH_VARARGS},
1537 {"getargs_B", getargs_B, METH_VARARGS},
1538 {"getargs_H", getargs_H, METH_VARARGS},
1539 {"getargs_I", getargs_I, METH_VARARGS},
1540 {"getargs_k", getargs_k, METH_VARARGS},
1541 {"getargs_i", getargs_i, METH_VARARGS},
1542 {"getargs_l", getargs_l, METH_VARARGS},
1543 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001544#ifdef HAVE_LONG_LONG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001545 {"getargs_L", getargs_L, METH_VARARGS},
1546 {"getargs_K", getargs_K, METH_VARARGS},
1547 {"test_longlong_api", test_longlong_api, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001548 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001549 {"codec_incrementalencoder",
1550 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
1551 {"codec_incrementaldecoder",
1552 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001553#endif
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +00001554 {"test_s_code", (PyCFunction)test_s_code, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001555 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
Guido van Rossumfb67be22007-08-29 18:38:11 +00001556 {"test_Z_code", (PyCFunction)test_Z_code, METH_NOARGS},
Mark Dickinson081dfee2009-03-18 14:47:41 +00001557 {"test_widechar", (PyCFunction)test_widechar, METH_NOARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001558#ifdef WITH_THREAD
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001559 {"_test_thread_state", test_thread_state, METH_VARARGS},
Benjamin Petersona54c9092009-01-13 02:11:23 +00001560 {"_pending_threadfunc", pending_threadfunc, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001561#endif
Guido van Rossumddefaf32007-01-14 03:31:43 +00001562#ifdef HAVE_GETTIMEOFDAY
1563 {"profile_int", profile_int, METH_NOARGS},
1564#endif
Christian Heimes81ee3ef2008-05-04 22:42:01 +00001565 {"traceback_print", traceback_print, METH_VARARGS},
Benjamin Petersone6528212008-07-15 15:32:09 +00001566 {"exception_print", exception_print, METH_VARARGS},
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00001567 {"argparsing", argparsing, METH_VARARGS},
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00001568 {"code_newempty", code_newempty, METH_VARARGS},
Tim Peters9ea17ac2001-02-02 05:57:15 +00001569 {NULL, NULL} /* sentinel */
1570};
1571
Thomas Hellera4ea6032003-04-17 18:55:45 +00001572#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
1573
Thomas Wouters89f507f2006-12-13 04:49:30 +00001574typedef struct {
Christian Heimes1af737c2008-01-23 08:24:23 +00001575 char bool_member;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001576 char byte_member;
1577 unsigned char ubyte_member;
1578 short short_member;
1579 unsigned short ushort_member;
1580 int int_member;
1581 unsigned int uint_member;
1582 long long_member;
1583 unsigned long ulong_member;
1584 float float_member;
1585 double double_member;
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001586#ifdef HAVE_LONG_LONG
1587 PY_LONG_LONG longlong_member;
1588 unsigned PY_LONG_LONG ulonglong_member;
1589#endif
Thomas Wouters89f507f2006-12-13 04:49:30 +00001590} all_structmembers;
1591
1592typedef struct {
1593 PyObject_HEAD
1594 all_structmembers structmembers;
1595} test_structmembers;
1596
1597static struct PyMemberDef test_members[] = {
Christian Heimes1af737c2008-01-23 08:24:23 +00001598 {"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
Thomas Wouters89f507f2006-12-13 04:49:30 +00001599 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
1600 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
1601 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
1602 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
1603 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
1604 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
1605 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
1606 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
1607 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
1608 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001609#ifdef HAVE_LONG_LONG
1610 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
1611 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
1612#endif
Thomas Wouters89f507f2006-12-13 04:49:30 +00001613 {NULL}
1614};
1615
1616
Christian Heimes1af737c2008-01-23 08:24:23 +00001617static PyObject *
1618test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
1619{
1620 static char *keywords[] = {
1621 "T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
1622 "T_INT", "T_UINT", "T_LONG", "T_ULONG",
1623 "T_FLOAT", "T_DOUBLE",
1624#ifdef HAVE_LONG_LONG
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001625 "T_LONGLONG", "T_ULONGLONG",
Christian Heimes1af737c2008-01-23 08:24:23 +00001626#endif
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001627 NULL};
Christian Heimes1af737c2008-01-23 08:24:23 +00001628 static char *fmt = "|bbBhHiIlkfd"
1629#ifdef HAVE_LONG_LONG
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001630 "LK"
Christian Heimes1af737c2008-01-23 08:24:23 +00001631#endif
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001632 ;
Christian Heimes1af737c2008-01-23 08:24:23 +00001633 test_structmembers *ob;
1634 ob = PyObject_New(test_structmembers, type);
1635 if (ob == NULL)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001636 return NULL;
1637 memset(&ob->structmembers, 0, sizeof(all_structmembers));
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001638 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
Christian Heimes1af737c2008-01-23 08:24:23 +00001639 &ob->structmembers.bool_member,
1640 &ob->structmembers.byte_member,
1641 &ob->structmembers.ubyte_member,
1642 &ob->structmembers.short_member,
1643 &ob->structmembers.ushort_member,
1644 &ob->structmembers.int_member,
1645 &ob->structmembers.uint_member,
1646 &ob->structmembers.long_member,
1647 &ob->structmembers.ulong_member,
1648 &ob->structmembers.float_member,
1649 &ob->structmembers.double_member
1650#ifdef HAVE_LONG_LONG
1651 , &ob->structmembers.longlong_member,
1652 &ob->structmembers.ulonglong_member
1653#endif
1654 )) {
Thomas Wouters89f507f2006-12-13 04:49:30 +00001655 Py_DECREF(ob);
1656 return NULL;
Christian Heimes1af737c2008-01-23 08:24:23 +00001657 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00001658 return (PyObject *)ob;
1659}
1660
Christian Heimes1af737c2008-01-23 08:24:23 +00001661static void
1662test_structmembers_free(PyObject *ob)
1663{
Thomas Wouters89f507f2006-12-13 04:49:30 +00001664 PyObject_FREE(ob);
1665}
1666
1667static PyTypeObject test_structmembersType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001668 PyVarObject_HEAD_INIT(NULL, 0)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001669 "test_structmembersType",
1670 sizeof(test_structmembers), /* tp_basicsize */
1671 0, /* tp_itemsize */
1672 test_structmembers_free, /* destructor tp_dealloc */
1673 0, /* tp_print */
1674 0, /* tp_getattr */
1675 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +00001676 0, /* tp_reserved */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001677 0, /* tp_repr */
1678 0, /* tp_as_number */
1679 0, /* tp_as_sequence */
1680 0, /* tp_as_mapping */
1681 0, /* tp_hash */
1682 0, /* tp_call */
1683 0, /* tp_str */
Christian Heimes1af737c2008-01-23 08:24:23 +00001684 PyObject_GenericGetAttr, /* tp_getattro */
1685 PyObject_GenericSetAttr, /* tp_setattro */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001686 0, /* tp_as_buffer */
1687 0, /* tp_flags */
1688 "Type containing all structmember types",
1689 0, /* traverseproc tp_traverse */
1690 0, /* tp_clear */
1691 0, /* tp_richcompare */
1692 0, /* tp_weaklistoffset */
1693 0, /* tp_iter */
1694 0, /* tp_iternext */
1695 0, /* tp_methods */
Christian Heimes1af737c2008-01-23 08:24:23 +00001696 test_members, /* tp_members */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001697 0,
1698 0,
1699 0,
1700 0,
1701 0,
1702 0,
1703 0,
1704 0,
Christian Heimes1af737c2008-01-23 08:24:23 +00001705 test_structmembers_new, /* tp_new */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001706};
1707
1708
Martin v. Löwis1a214512008-06-11 05:26:20 +00001709
1710static struct PyModuleDef _testcapimodule = {
1711 PyModuleDef_HEAD_INIT,
1712 "_testcapi",
1713 NULL,
1714 -1,
1715 TestMethods,
1716 NULL,
1717 NULL,
1718 NULL,
1719 NULL
1720};
1721
Mark Hammond62b1ab12002-07-23 06:31:15 +00001722PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001723PyInit__testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +00001724{
Fred Drakeacee69f2002-04-01 14:28:58 +00001725 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001726
Martin v. Löwis1a214512008-06-11 05:26:20 +00001727 m = PyModule_Create(&_testcapimodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001728 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00001729 return NULL;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001730
Hirokazu Yamamoto8ebab5d2008-12-31 06:05:46 +00001731 Py_TYPE(&_HashInheritanceTester_Type)=&PyType_Type;
1732
Christian Heimes90aa7642007-12-19 02:45:37 +00001733 Py_TYPE(&test_structmembersType)=&PyType_Type;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001734 Py_INCREF(&test_structmembersType);
1735 PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
1736
Christian Heimes217cfd12007-12-02 14:31:20 +00001737 PyModule_AddObject(m, "CHAR_MAX", PyLong_FromLong(CHAR_MAX));
1738 PyModule_AddObject(m, "CHAR_MIN", PyLong_FromLong(CHAR_MIN));
1739 PyModule_AddObject(m, "UCHAR_MAX", PyLong_FromLong(UCHAR_MAX));
1740 PyModule_AddObject(m, "SHRT_MAX", PyLong_FromLong(SHRT_MAX));
1741 PyModule_AddObject(m, "SHRT_MIN", PyLong_FromLong(SHRT_MIN));
1742 PyModule_AddObject(m, "USHRT_MAX", PyLong_FromLong(USHRT_MAX));
Thomas Wouters89f507f2006-12-13 04:49:30 +00001743 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
1744 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001745 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
Christian Heimes217cfd12007-12-02 14:31:20 +00001746 PyModule_AddObject(m, "LONG_MAX", PyLong_FromLong(LONG_MAX));
1747 PyModule_AddObject(m, "LONG_MIN", PyLong_FromLong(LONG_MIN));
Thomas Wouters89f507f2006-12-13 04:49:30 +00001748 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
1749 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
1750 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
1751 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
1752 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001753 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
1754 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
1755 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
Christian Heimes217cfd12007-12-02 14:31:20 +00001756 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyLong_FromSsize_t(PY_SSIZE_T_MAX));
1757 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyLong_FromSsize_t(PY_SSIZE_T_MIN));
Robert Schuppeniesfbe94c52008-07-14 10:13:31 +00001758 PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyLong_FromSsize_t(sizeof(PyGC_Head)));
Benjamin Peterson9b6df6a2008-10-16 23:56:29 +00001759 Py_INCREF(&PyInstanceMethod_Type);
1760 PyModule_AddObject(m, "instancemethod", (PyObject *)&PyInstanceMethod_Type);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001761
Tim Petersd66595f2001-02-04 03:09:53 +00001762 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
Fred Drakeacee69f2002-04-01 14:28:58 +00001763 Py_INCREF(TestError);
1764 PyModule_AddObject(m, "error", TestError);
Martin v. Löwis1a214512008-06-11 05:26:20 +00001765 return m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001766}