blob: 70da707d4370aafdbb630ad187a66a8d4f2cee91 [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"
Benjamin Peterson91d58bd2009-12-13 21:30:54 +000013#include "datetime.h"
Tim Peters9ea17ac2001-02-02 05:57:15 +000014
Mark Hammond8d98d2c2003-04-19 15:41:53 +000015#ifdef WITH_THREAD
16#include "pythread.h"
17#endif /* WITH_THREAD */
Tim Peters9ea17ac2001-02-02 05:57:15 +000018static PyObject *TestError; /* set to exception object in init */
19
Tim Peters91621db2001-06-12 20:10:01 +000020/* Raise TestError with test_name + ": " + msg, and return NULL. */
21
22static PyObject *
23raiseTestError(const char* test_name, const char* msg)
24{
25 char buf[2048];
26
27 if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
28 PyErr_SetString(TestError, "internal error msg too large");
29 else {
Tim Peters885d4572001-11-28 20:27:42 +000030 PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
Tim Peters91621db2001-06-12 20:10:01 +000031 PyErr_SetString(TestError, buf);
32 }
33 return NULL;
34}
35
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +000036/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
Tim Peters9ea17ac2001-02-02 05:57:15 +000037
38 The ones derived from autoconf on the UNIX-like OSes can be relied
39 upon (in the absence of sloppy cross-compiling), but the Windows
40 platforms have these hardcoded. Better safe than sorry.
41*/
42static PyObject*
Thomas Wouters89f507f2006-12-13 04:49:30 +000043sizeof_error(const char* fatname, const char* typname,
Tim Peters9ea17ac2001-02-02 05:57:15 +000044 int expected, int got)
45{
46 char buf[1024];
Tim Petersbaefd9e2003-01-28 20:37:45 +000047 PyOS_snprintf(buf, sizeof(buf),
Tim Peters885d4572001-11-28 20:27:42 +000048 "%.200s #define == %d but sizeof(%.200s) == %d",
Thomas Wouters89f507f2006-12-13 04:49:30 +000049 fatname, expected, typname, got);
Tim Peters9ea17ac2001-02-02 05:57:15 +000050 PyErr_SetString(TestError, buf);
51 return (PyObject*)NULL;
52}
53
54static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000055test_config(PyObject *self)
Tim Peters9ea17ac2001-02-02 05:57:15 +000056{
Tim Peters9ea17ac2001-02-02 05:57:15 +000057#define CHECK_SIZEOF(FATNAME, TYPE) \
58 if (FATNAME != sizeof(TYPE)) \
59 return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
60
Tim Peters208efe52001-06-26 22:40:47 +000061 CHECK_SIZEOF(SIZEOF_SHORT, short);
Tim Peters9ea17ac2001-02-02 05:57:15 +000062 CHECK_SIZEOF(SIZEOF_INT, int);
63 CHECK_SIZEOF(SIZEOF_LONG, long);
64 CHECK_SIZEOF(SIZEOF_VOID_P, void*);
65 CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
66#ifdef HAVE_LONG_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +000067 CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
Tim Peters9ea17ac2001-02-02 05:57:15 +000068#endif
69
70#undef CHECK_SIZEOF
71
72 Py_INCREF(Py_None);
73 return Py_None;
74}
75
Tim Peters5c4d5bf2001-02-12 22:13:26 +000076static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000077test_list_api(PyObject *self)
Tim Peters5c4d5bf2001-02-12 22:13:26 +000078{
79 PyObject* list;
80 int i;
Tim Peters5c4d5bf2001-02-12 22:13:26 +000081
82 /* SF bug 132008: PyList_Reverse segfaults */
83#define NLIST 30
84 list = PyList_New(NLIST);
85 if (list == (PyObject*)NULL)
86 return (PyObject*)NULL;
87 /* list = range(NLIST) */
88 for (i = 0; i < NLIST; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +000089 PyObject* anint = PyLong_FromLong(i);
Tim Peters5c4d5bf2001-02-12 22:13:26 +000090 if (anint == (PyObject*)NULL) {
91 Py_DECREF(list);
92 return (PyObject*)NULL;
93 }
94 PyList_SET_ITEM(list, i, anint);
95 }
96 /* list.reverse(), via PyList_Reverse() */
97 i = PyList_Reverse(list); /* should not blow up! */
98 if (i != 0) {
99 Py_DECREF(list);
100 return (PyObject*)NULL;
101 }
102 /* Check that list == range(29, -1, -1) now */
103 for (i = 0; i < NLIST; ++i) {
104 PyObject* anint = PyList_GET_ITEM(list, i);
Christian Heimes217cfd12007-12-02 14:31:20 +0000105 if (PyLong_AS_LONG(anint) != NLIST-1-i) {
Tim Peters5c4d5bf2001-02-12 22:13:26 +0000106 PyErr_SetString(TestError,
107 "test_list_api: reverse screwed up");
108 Py_DECREF(list);
109 return (PyObject*)NULL;
110 }
111 }
112 Py_DECREF(list);
113#undef NLIST
114
115 Py_INCREF(Py_None);
116 return Py_None;
117}
118
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000119static int
120test_dict_inner(int count)
121{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000122 Py_ssize_t pos = 0, iterations = 0;
123 int i;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000124 PyObject *dict = PyDict_New();
125 PyObject *v, *k;
126
127 if (dict == NULL)
128 return -1;
129
130 for (i = 0; i < count; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +0000131 v = PyLong_FromLong(i);
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000132 PyDict_SetItem(dict, v, v);
133 Py_DECREF(v);
134 }
135
136 while (PyDict_Next(dict, &pos, &k, &v)) {
137 PyObject *o;
138 iterations++;
139
Christian Heimes217cfd12007-12-02 14:31:20 +0000140 i = PyLong_AS_LONG(v) + 1;
141 o = PyLong_FromLong(i);
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000142 if (o == NULL)
143 return -1;
144 if (PyDict_SetItem(dict, k, o) < 0) {
145 Py_DECREF(o);
146 return -1;
147 }
148 Py_DECREF(o);
149 }
150
151 Py_DECREF(dict);
152
153 if (iterations != count) {
154 PyErr_SetString(
155 TestError,
156 "test_dict_iteration: dict iteration went wrong ");
157 return -1;
158 } else {
159 return 0;
160 }
161}
162
163static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +0000164test_dict_iteration(PyObject* self)
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000165{
166 int i;
167
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000168 for (i = 0; i < 200; i++) {
169 if (test_dict_inner(i) < 0) {
170 return NULL;
171 }
172 }
173
174 Py_INCREF(Py_None);
175 return Py_None;
176}
177
Tim Peters91621db2001-06-12 20:10:01 +0000178
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000179/* Issue #4701: Check that PyObject_Hash implicitly calls
180 * PyType_Ready if it hasn't already been called
181 */
182static PyTypeObject _HashInheritanceTester_Type = {
Hirokazu Yamamoto8ebab5d2008-12-31 06:05:46 +0000183 PyVarObject_HEAD_INIT(NULL, 0)
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000184 "hashinheritancetester", /* Name of this type */
185 sizeof(PyObject), /* Basic object size */
186 0, /* Item size for varobject */
187 (destructor)PyObject_Del, /* tp_dealloc */
188 0, /* tp_print */
189 0, /* tp_getattr */
190 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +0000191 0, /* tp_reserved */
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000192 0, /* tp_repr */
193 0, /* tp_as_number */
194 0, /* tp_as_sequence */
195 0, /* tp_as_mapping */
196 0, /* tp_hash */
197 0, /* tp_call */
198 0, /* tp_str */
199 PyObject_GenericGetAttr, /* tp_getattro */
200 0, /* tp_setattro */
201 0, /* tp_as_buffer */
202 Py_TPFLAGS_DEFAULT, /* tp_flags */
203 0, /* tp_doc */
204 0, /* tp_traverse */
205 0, /* tp_clear */
206 0, /* tp_richcompare */
207 0, /* tp_weaklistoffset */
208 0, /* tp_iter */
209 0, /* tp_iternext */
210 0, /* tp_methods */
211 0, /* tp_members */
212 0, /* tp_getset */
213 0, /* tp_base */
214 0, /* tp_dict */
215 0, /* tp_descr_get */
216 0, /* tp_descr_set */
217 0, /* tp_dictoffset */
218 0, /* tp_init */
219 0, /* tp_alloc */
220 PyType_GenericNew, /* tp_new */
221};
222
223static PyObject*
224test_lazy_hash_inheritance(PyObject* self)
225{
226 PyTypeObject *type;
227 PyObject *obj;
228 long hash;
229
230 type = &_HashInheritanceTester_Type;
Benjamin Petersonf1726372009-05-05 21:11:54 +0000231
Benjamin Petersonb43d3252009-05-05 23:14:15 +0000232 if (type->tp_dict != NULL)
233 /* The type has already been initialized. This probably means
234 -R is being used. */
235 Py_RETURN_NONE;
Benjamin Petersonf1726372009-05-05 21:11:54 +0000236
237
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000238 obj = PyObject_New(PyObject, type);
239 if (obj == NULL) {
240 PyErr_Clear();
241 PyErr_SetString(
242 TestError,
243 "test_lazy_hash_inheritance: failed to create object");
244 return NULL;
245 }
246
247 if (type->tp_dict != NULL) {
248 PyErr_SetString(
249 TestError,
250 "test_lazy_hash_inheritance: type initialised too soon");
251 Py_DECREF(obj);
252 return NULL;
253 }
254
255 hash = PyObject_Hash(obj);
256 if ((hash == -1) && PyErr_Occurred()) {
257 PyErr_Clear();
258 PyErr_SetString(
259 TestError,
260 "test_lazy_hash_inheritance: could not hash object");
261 Py_DECREF(obj);
262 return NULL;
263 }
264
265 if (type->tp_dict == NULL) {
266 PyErr_SetString(
267 TestError,
268 "test_lazy_hash_inheritance: type not initialised by hash()");
269 Py_DECREF(obj);
270 return NULL;
271 }
272
273 if (type->tp_hash != PyType_Type.tp_hash) {
274 PyErr_SetString(
275 TestError,
276 "test_lazy_hash_inheritance: unexpected hash function");
277 Py_DECREF(obj);
278 return NULL;
279 }
280
Benjamin Petersonb43d3252009-05-05 23:14:15 +0000281 Py_DECREF(obj);
Benjamin Petersonf1726372009-05-05 21:11:54 +0000282
Nick Coghlanf1f2f682008-12-30 07:29:12 +0000283 Py_RETURN_NONE;
284}
285
286
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000287/* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
Tim Petersff70d3c2001-06-14 01:11:03 +0000288 PyLong_{As, From}{Unsigned,}LongLong().
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000289
290 Note that the meat of the test is contained in testcapi_long.h.
291 This is revolting, but delicate code duplication is worse: "almost
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000292 exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000293 dependence on type names makes it impossible to use a parameterized
294 function. A giant macro would be even worse than this. A C++ template
295 would be perfect.
296
297 The "report an error" functions are deliberately not part of the #include
298 file: if the test fails, you can set a breakpoint in the appropriate
299 error function directly, and crawl back from there in the debugger.
300*/
301
302#define UNBIND(X) Py_DECREF(X); (X) = NULL
303
304static PyObject *
305raise_test_long_error(const char* msg)
306{
307 return raiseTestError("test_long_api", msg);
308}
309
310#define TESTNAME test_long_api_inner
311#define TYPENAME long
312#define F_S_TO_PY PyLong_FromLong
313#define F_PY_TO_S PyLong_AsLong
314#define F_U_TO_PY PyLong_FromUnsignedLong
315#define F_PY_TO_U PyLong_AsUnsignedLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000316
317#include "testcapi_long.h"
318
319static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000320test_long_api(PyObject* self)
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000321{
Tim Peters83c9edc2001-06-16 08:10:13 +0000322 return TESTNAME(raise_test_long_error);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000323}
324
325#undef TESTNAME
326#undef TYPENAME
327#undef F_S_TO_PY
328#undef F_PY_TO_S
329#undef F_U_TO_PY
330#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000331
332#ifdef HAVE_LONG_LONG
Tim Peters91621db2001-06-12 20:10:01 +0000333
334static PyObject *
Tim Petersd1a7da62001-06-13 00:35:57 +0000335raise_test_longlong_error(const char* msg)
336{
337 return raiseTestError("test_longlong_api", msg);
338}
339
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000340#define TESTNAME test_longlong_api_inner
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000341#define TYPENAME PY_LONG_LONG
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000342#define F_S_TO_PY PyLong_FromLongLong
343#define F_PY_TO_S PyLong_AsLongLong
344#define F_U_TO_PY PyLong_FromUnsignedLongLong
345#define F_PY_TO_U PyLong_AsUnsignedLongLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000346
347#include "testcapi_long.h"
Tim Petersd1a7da62001-06-13 00:35:57 +0000348
349static PyObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000350test_longlong_api(PyObject* self, PyObject *args)
Tim Peters91621db2001-06-12 20:10:01 +0000351{
Tim Peters83c9edc2001-06-16 08:10:13 +0000352 return TESTNAME(raise_test_longlong_error);
Tim Peters91621db2001-06-12 20:10:01 +0000353}
354
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000355#undef TESTNAME
356#undef TYPENAME
357#undef F_S_TO_PY
358#undef F_PY_TO_S
359#undef F_U_TO_PY
360#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000361
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000362/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
Tim Petersd38b1c72001-09-30 05:09:37 +0000363 for both long and int arguments. The test may leak a little memory if
364 it fails.
365*/
366static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000367test_L_code(PyObject *self)
Tim Petersd38b1c72001-09-30 05:09:37 +0000368{
369 PyObject *tuple, *num;
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000370 PY_LONG_LONG value;
Tim Petersd38b1c72001-09-30 05:09:37 +0000371
Tim Petersd38b1c72001-09-30 05:09:37 +0000372 tuple = PyTuple_New(1);
373 if (tuple == NULL)
374 return NULL;
375
376 num = PyLong_FromLong(42);
377 if (num == NULL)
378 return NULL;
379
380 PyTuple_SET_ITEM(tuple, 0, num);
381
382 value = -1;
383 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
384 return NULL;
385 if (value != 42)
386 return raiseTestError("test_L_code",
387 "L code returned wrong value for long 42");
388
389 Py_DECREF(num);
Christian Heimes217cfd12007-12-02 14:31:20 +0000390 num = PyLong_FromLong(42);
Tim Petersd38b1c72001-09-30 05:09:37 +0000391 if (num == NULL)
392 return NULL;
393
394 PyTuple_SET_ITEM(tuple, 0, num);
395
396 value = -1;
397 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
398 return NULL;
399 if (value != 42)
400 return raiseTestError("test_L_code",
401 "L code returned wrong value for int 42");
402
403 Py_DECREF(tuple);
404 Py_INCREF(Py_None);
405 return Py_None;
406}
407
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000408#endif /* ifdef HAVE_LONG_LONG */
409
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000410/* Test tuple argument processing */
411static PyObject *
412getargs_tuple(PyObject *self, PyObject *args)
413{
414 int a, b, c;
415 if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
416 return NULL;
417 return Py_BuildValue("iii", a, b, c);
418}
419
Christian Heimes380f7f22008-02-28 11:19:05 +0000420/* test PyArg_ParseTupleAndKeywords */
421static PyObject *getargs_keywords(PyObject *self, PyObject *args, PyObject *kwargs)
422{
423 static char *keywords[] = {"arg1","arg2","arg3","arg4","arg5", NULL};
424 static char *fmt="(ii)i|(i(ii))(iii)i";
425 int int_args[10]={-1, -1, -1, -1, -1, -1, -1, -1, -1, -1};
426
427 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
428 &int_args[0], &int_args[1], &int_args[2], &int_args[3], &int_args[4],
429 &int_args[5], &int_args[6], &int_args[7], &int_args[8], &int_args[9]))
430 return NULL;
431 return Py_BuildValue("iiiiiiiiii",
432 int_args[0], int_args[1], int_args[2], int_args[3], int_args[4],
433 int_args[5], int_args[6], int_args[7], int_args[8], int_args[9]);
434}
435
Thomas Heller3457e4b2003-04-24 16:14:27 +0000436/* Functions to call PyArg_ParseTuple with integer format codes,
437 and return the result.
438*/
Thomas Hellera4ea6032003-04-17 18:55:45 +0000439static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000440getargs_b(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000441{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000442 unsigned char value;
443 if (!PyArg_ParseTuple(args, "b", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000444 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000445 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000446}
447
Thomas Heller3457e4b2003-04-24 16:14:27 +0000448static PyObject *
449getargs_B(PyObject *self, PyObject *args)
450{
451 unsigned char value;
452 if (!PyArg_ParseTuple(args, "B", &value))
453 return NULL;
454 return PyLong_FromUnsignedLong((unsigned long)value);
455}
456
457static PyObject *
Mark Dickinson1554b182009-12-20 16:03:30 +0000458getargs_h(PyObject *self, PyObject *args)
459{
460 short value;
461 if (!PyArg_ParseTuple(args, "h", &value))
462 return NULL;
463 return PyLong_FromLong((long)value);
464}
465
466static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000467getargs_H(PyObject *self, PyObject *args)
468{
469 unsigned short value;
470 if (!PyArg_ParseTuple(args, "H", &value))
471 return NULL;
472 return PyLong_FromUnsignedLong((unsigned long)value);
473}
474
475static PyObject *
476getargs_I(PyObject *self, PyObject *args)
477{
478 unsigned int value;
479 if (!PyArg_ParseTuple(args, "I", &value))
480 return NULL;
481 return PyLong_FromUnsignedLong((unsigned long)value);
482}
483
484static PyObject *
485getargs_k(PyObject *self, PyObject *args)
486{
487 unsigned long value;
488 if (!PyArg_ParseTuple(args, "k", &value))
489 return NULL;
490 return PyLong_FromUnsignedLong(value);
491}
492
493static PyObject *
494getargs_i(PyObject *self, PyObject *args)
495{
496 int value;
497 if (!PyArg_ParseTuple(args, "i", &value))
498 return NULL;
499 return PyLong_FromLong((long)value);
500}
501
Thomas Hellera4ea6032003-04-17 18:55:45 +0000502static PyObject *
503getargs_l(PyObject *self, PyObject *args)
504{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000505 long value;
506 if (!PyArg_ParseTuple(args, "l", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000507 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000508 return PyLong_FromLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000509}
510
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000511static PyObject *
512getargs_n(PyObject *self, PyObject *args)
513{
514 Py_ssize_t value;
515 if (!PyArg_ParseTuple(args, "n", &value))
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000516 return NULL;
517 return PyLong_FromSsize_t(value);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000518}
519
Thomas Hellera4ea6032003-04-17 18:55:45 +0000520#ifdef HAVE_LONG_LONG
Thomas Hellera4ea6032003-04-17 18:55:45 +0000521static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000522getargs_L(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000523{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000524 PY_LONG_LONG value;
525 if (!PyArg_ParseTuple(args, "L", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000526 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000527 return PyLong_FromLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000528}
529
Thomas Hellera4ea6032003-04-17 18:55:45 +0000530static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000531getargs_K(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000532{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000533 unsigned PY_LONG_LONG value;
534 if (!PyArg_ParseTuple(args, "K", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000535 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000536 return PyLong_FromUnsignedLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000537}
538#endif
539
540/* This function not only tests the 'k' getargs code, but also the
Christian Heimes217cfd12007-12-02 14:31:20 +0000541 PyLong_AsUnsignedLongMask() and PyLong_AsUnsignedLongMask() functions. */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000542static PyObject *
543test_k_code(PyObject *self)
544{
545 PyObject *tuple, *num;
546 unsigned long value;
547
548 tuple = PyTuple_New(1);
549 if (tuple == NULL)
550 return NULL;
551
Neal Norwitz699cbb72003-04-22 01:28:57 +0000552 /* a number larger than ULONG_MAX even on 64-bit platforms */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000553 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
554 if (num == NULL)
555 return NULL;
556
Christian Heimes217cfd12007-12-02 14:31:20 +0000557 value = PyLong_AsUnsignedLongMask(num);
Neal Norwitz699cbb72003-04-22 01:28:57 +0000558 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000559 return raiseTestError("test_k_code",
Christian Heimes217cfd12007-12-02 14:31:20 +0000560 "PyLong_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +0000561
562 PyTuple_SET_ITEM(tuple, 0, num);
563
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000564 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000565 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
566 return NULL;
Neal Norwitz699cbb72003-04-22 01:28:57 +0000567 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000568 return raiseTestError("test_k_code",
569 "k code returned wrong value for long 0xFFF...FFF");
570
571 Py_DECREF(num);
572 num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
573 if (num == NULL)
574 return NULL;
575
Christian Heimes217cfd12007-12-02 14:31:20 +0000576 value = PyLong_AsUnsignedLongMask(num);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000577 if (value != (unsigned long)-0x42)
578 return raiseTestError("test_k_code",
Christian Heimes217cfd12007-12-02 14:31:20 +0000579 "PyLong_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +0000580
581 PyTuple_SET_ITEM(tuple, 0, num);
582
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000583 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000584 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
585 return NULL;
586 if (value != (unsigned long)-0x42)
587 return raiseTestError("test_k_code",
588 "k code returned wrong value for long -0xFFF..000042");
589
590 Py_DECREF(tuple);
591 Py_INCREF(Py_None);
592 return Py_None;
593}
594
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000595
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +0000596/* Test the s and z codes for PyArg_ParseTuple.
597*/
598static PyObject *
599test_s_code(PyObject *self)
600{
601 /* Unicode strings should be accepted */
602 PyObject *tuple, *obj;
603 char *value;
604
605 tuple = PyTuple_New(1);
606 if (tuple == NULL)
607 return NULL;
608
609 obj = PyUnicode_Decode("t\xeate", strlen("t\xeate"),
610 "latin-1", NULL);
611 if (obj == NULL)
612 return NULL;
613
614 PyTuple_SET_ITEM(tuple, 0, obj);
615
616 /* These two blocks used to raise a TypeError:
617 * "argument must be string without null bytes, not str"
618 */
619 if (PyArg_ParseTuple(tuple, "s:test_s_code1", &value) < 0)
620 return NULL;
621
622 if (PyArg_ParseTuple(tuple, "z:test_s_code2", &value) < 0)
623 return NULL;
624
Alexandre Vassalottib645bc72008-05-15 22:06:59 +0000625 Py_DECREF(tuple);
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +0000626 Py_RETURN_NONE;
627}
628
Mark Dickinsonf08173b2009-12-03 10:59:46 +0000629static PyObject *
630test_bug_7414(PyObject *self)
631{
632 /* Issue #7414: for PyArg_ParseTupleAndKeywords, 'C' code wasn't being
633 skipped properly in skipitem() */
634 int a = 0, b = 0, result;
635 char *kwlist[] = {"a", "b", NULL};
636 PyObject *tuple = NULL, *dict = NULL, *b_str;
637
638 tuple = PyTuple_New(0);
639 if (tuple == NULL)
640 goto failure;
641 dict = PyDict_New();
642 if (dict == NULL)
643 goto failure;
644 b_str = PyUnicode_FromString("b");
645 if (b_str == NULL)
646 goto failure;
647 result = PyDict_SetItemString(dict, "b", b_str);
648 Py_DECREF(b_str);
649 if (result < 0)
650 goto failure;
651
652 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|CC",
653 kwlist, &a, &b);
654 if (!result)
655 goto failure;
656
657 if (a != 0)
658 return raiseTestError("test_bug_7414",
659 "C format code not skipped properly");
660 if (b != 'b')
661 return raiseTestError("test_bug_7414",
662 "C format code returned wrong value");
663
664 Py_DECREF(dict);
665 Py_DECREF(tuple);
666 Py_RETURN_NONE;
667
668 failure:
669 Py_XDECREF(dict);
670 Py_XDECREF(tuple);
671 return NULL;
672}
673
674
Benjamin Peterson92035012008-12-27 16:00:54 +0000675static volatile int x;
676
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000677/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
678 of an error.
679*/
680static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000681test_u_code(PyObject *self)
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000682{
683 PyObject *tuple, *obj;
684 Py_UNICODE *value;
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000685 Py_ssize_t len;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000686
Benjamin Peterson206e3072008-10-19 14:07:49 +0000687 /* issue4122: Undefined reference to _Py_ascii_whitespace on Windows */
688 /* Just use the macro and check that it compiles */
Christian Heimes727c2c52008-11-12 09:04:04 +0000689 x = Py_UNICODE_ISSPACE(25);
Benjamin Peterson206e3072008-10-19 14:07:49 +0000690
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000691 tuple = PyTuple_New(1);
692 if (tuple == NULL)
693 return NULL;
694
695 obj = PyUnicode_Decode("test", strlen("test"),
696 "ascii", NULL);
697 if (obj == NULL)
698 return NULL;
699
700 PyTuple_SET_ITEM(tuple, 0, obj);
701
702 value = 0;
703 if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
704 return NULL;
705 if (value != PyUnicode_AS_UNICODE(obj))
706 return raiseTestError("test_u_code",
707 "u code returned wrong value for u'test'");
708 value = 0;
709 if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
710 return NULL;
711 if (value != PyUnicode_AS_UNICODE(obj) ||
712 len != PyUnicode_GET_SIZE(obj))
713 return raiseTestError("test_u_code",
714 "u# code returned wrong values for u'test'");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000715
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000716 Py_DECREF(tuple);
717 Py_INCREF(Py_None);
718 return Py_None;
719}
720
Guido van Rossumfb67be22007-08-29 18:38:11 +0000721/* Test Z and Z# codes for PyArg_ParseTuple */
722static PyObject *
723test_Z_code(PyObject *self)
724{
725 PyObject *tuple, *obj;
726 Py_UNICODE *value1, *value2;
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000727 Py_ssize_t len1, len2;
Guido van Rossumfb67be22007-08-29 18:38:11 +0000728
729 tuple = PyTuple_New(2);
730 if (tuple == NULL)
731 return NULL;
732
733 obj = PyUnicode_FromString("test");
734 PyTuple_SET_ITEM(tuple, 0, obj);
735 Py_INCREF(Py_None);
736 PyTuple_SET_ITEM(tuple, 1, Py_None);
737
738 /* swap values on purpose */
739 value1 = NULL;
740 value2 = PyUnicode_AS_UNICODE(obj);
741
742 /* Test Z for both values */
743 if (PyArg_ParseTuple(tuple, "ZZ:test_Z_code", &value1, &value2) < 0)
744 return NULL;
745 if (value1 != PyUnicode_AS_UNICODE(obj))
746 return raiseTestError("test_Z_code",
747 "Z code returned wrong value for 'test'");
748 if (value2 != NULL)
749 return raiseTestError("test_Z_code",
750 "Z code returned wrong value for None");
751
752 value1 = NULL;
753 value2 = PyUnicode_AS_UNICODE(obj);
754 len1 = -1;
755 len2 = -1;
756
757 /* Test Z# for both values */
758 if (PyArg_ParseTuple(tuple, "Z#Z#:test_Z_code", &value1, &len1,
759 &value2, &len2) < 0)
760 return NULL;
761 if (value1 != PyUnicode_AS_UNICODE(obj) ||
762 len1 != PyUnicode_GET_SIZE(obj))
763 return raiseTestError("test_Z_code",
764 "Z# code returned wrong values for 'test'");
765 if (value2 != NULL ||
766 len2 != 0)
767 return raiseTestError("test_Z_code",
768 "Z# code returned wrong values for None'");
769
770 Py_DECREF(tuple);
771 Py_RETURN_NONE;
772}
773
Thomas Wouters477c8d52006-05-27 19:21:47 +0000774static PyObject *
Mark Dickinson081dfee2009-03-18 14:47:41 +0000775test_widechar(PyObject *self)
776{
777#if defined(SIZEOF_WCHAR_T) && (SIZEOF_WCHAR_T == 4)
778 const wchar_t wtext[2] = {(wchar_t)0x10ABCDu};
779 size_t wtextlen = 1;
780#else
781 const wchar_t wtext[3] = {(wchar_t)0xDBEAu, (wchar_t)0xDFCDu};
782 size_t wtextlen = 2;
783#endif
784 PyObject *wide, *utf8;
785
786 wide = PyUnicode_FromWideChar(wtext, wtextlen);
787 if (wide == NULL)
788 return NULL;
789
790 utf8 = PyUnicode_FromString("\xf4\x8a\xaf\x8d");
791 if (utf8 == NULL) {
792 Py_DECREF(wide);
793 return NULL;
794 }
795
796 if (PyUnicode_GET_SIZE(wide) != PyUnicode_GET_SIZE(utf8)) {
797 Py_DECREF(wide);
798 Py_DECREF(utf8);
799 return raiseTestError("test_widechar",
800 "wide string and utf8 string "
801 "have different length");
802 }
803 if (PyUnicode_Compare(wide, utf8)) {
804 Py_DECREF(wide);
805 Py_DECREF(utf8);
806 if (PyErr_Occurred())
807 return NULL;
808 return raiseTestError("test_widechar",
809 "wide string and utf8 string "
810 "are different");
811 }
812
813 Py_DECREF(wide);
814 Py_DECREF(utf8);
815 Py_RETURN_NONE;
816}
817
818static PyObject *
Benjamin Peterson92035012008-12-27 16:00:54 +0000819test_empty_argparse(PyObject *self)
820{
821 /* Test that formats can begin with '|'. See issue #4720. */
822 PyObject *tuple, *dict = NULL;
823 static char *kwlist[] = {NULL};
824 int result;
825 tuple = PyTuple_New(0);
826 if (!tuple)
827 return NULL;
828 if ((result = PyArg_ParseTuple(tuple, "|:test_empty_argparse")) < 0)
829 goto done;
830 dict = PyDict_New();
831 if (!dict)
832 goto done;
833 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|:test_empty_argparse", kwlist);
834 done:
835 Py_DECREF(tuple);
836 Py_XDECREF(dict);
837 if (result < 0)
838 return NULL;
839 else {
840 Py_RETURN_NONE;
841 }
842}
843
844static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +0000845codec_incrementalencoder(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_incrementalencoder",
849 &encoding, &errors))
850 return NULL;
851 return PyCodec_IncrementalEncoder(encoding, errors);
852}
853
Thomas Wouters477c8d52006-05-27 19:21:47 +0000854static PyObject *
855codec_incrementaldecoder(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000856{
857 const char *encoding, *errors = NULL;
858 if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
859 &encoding, &errors))
860 return NULL;
861 return PyCodec_IncrementalDecoder(encoding, errors);
862}
863
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000864
Tim Peters5b8132f2003-01-31 15:52:05 +0000865/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
Tim Petersbaefd9e2003-01-28 20:37:45 +0000866static PyObject *
867test_long_numbits(PyObject *self)
868{
Tim Peters5b8132f2003-01-31 15:52:05 +0000869 struct triple {
Tim Petersbaefd9e2003-01-28 20:37:45 +0000870 long input;
Tim Peters5b8132f2003-01-31 15:52:05 +0000871 size_t nbits;
872 int sign;
873 } testcases[] = {{0, 0, 0},
874 {1L, 1, 1},
875 {-1L, 1, -1},
876 {2L, 2, 1},
877 {-2L, 2, -1},
878 {3L, 2, 1},
879 {-3L, 2, -1},
880 {4L, 3, 1},
881 {-4L, 3, -1},
882 {0x7fffL, 15, 1}, /* one Python long digit */
883 {-0x7fffL, 15, -1},
884 {0xffffL, 16, 1},
885 {-0xffffL, 16, -1},
886 {0xfffffffL, 28, 1},
887 {-0xfffffffL, 28, -1}};
Tim Petersbaefd9e2003-01-28 20:37:45 +0000888 int i;
889
Tim Peters5b8132f2003-01-31 15:52:05 +0000890 for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
891 PyObject *plong = PyLong_FromLong(testcases[i].input);
Tim Petersbaefd9e2003-01-28 20:37:45 +0000892 size_t nbits = _PyLong_NumBits(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +0000893 int sign = _PyLong_Sign(plong);
Tim Petersbaefd9e2003-01-28 20:37:45 +0000894
895 Py_DECREF(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +0000896 if (nbits != testcases[i].nbits)
Tim Petersbaefd9e2003-01-28 20:37:45 +0000897 return raiseTestError("test_long_numbits",
Tim Peters5b8132f2003-01-31 15:52:05 +0000898 "wrong result for _PyLong_NumBits");
899 if (sign != testcases[i].sign)
900 return raiseTestError("test_long_numbits",
901 "wrong result for _PyLong_Sign");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000902 }
903 Py_INCREF(Py_None);
904 return Py_None;
905}
906
Thomas Heller519a0422007-11-15 20:48:54 +0000907/* Example passing NULLs to PyObject_Str(NULL). */
Thomas Woutersa9773292006-04-21 09:43:23 +0000908
909static PyObject *
910test_null_strings(PyObject *self)
911{
Thomas Heller519a0422007-11-15 20:48:54 +0000912 PyObject *o1 = PyObject_Str(NULL), *o2 = PyObject_Str(NULL);
Thomas Woutersa9773292006-04-21 09:43:23 +0000913 PyObject *tuple = PyTuple_Pack(2, o1, o2);
914 Py_XDECREF(o1);
915 Py_XDECREF(o2);
916 return tuple;
917}
918
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000919static PyObject *
920raise_exception(PyObject *self, PyObject *args)
921{
922 PyObject *exc;
923 PyObject *exc_args, *v;
924 int num_args, i;
925
926 if (!PyArg_ParseTuple(args, "Oi:raise_exception",
927 &exc, &num_args))
928 return NULL;
929
930 exc_args = PyTuple_New(num_args);
931 if (exc_args == NULL)
932 return NULL;
933 for (i = 0; i < num_args; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +0000934 v = PyLong_FromLong(i);
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000935 if (v == NULL) {
936 Py_DECREF(exc_args);
937 return NULL;
938 }
939 PyTuple_SET_ITEM(exc_args, i, v);
940 }
941 PyErr_SetObject(exc, exc_args);
Michael W. Hudsonf02bcee2003-08-15 13:03:30 +0000942 Py_DECREF(exc_args);
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000943 return NULL;
944}
Tim Peters91621db2001-06-12 20:10:01 +0000945
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000946#ifdef WITH_THREAD
947
Benjamin Peterson91d58bd2009-12-13 21:30:54 +0000948static PyObject *
949test_datetime_capi(PyObject *self, PyObject *args) {
950 if (PyDateTimeAPI) {
951 PyErr_SetString(PyExc_AssertionError,
952 "PyDateTime_CAPI somehow initialized");
953 return NULL;
954 }
955 PyDateTime_IMPORT;
956 if (PyDateTimeAPI)
957 Py_RETURN_NONE;
958 else
959 return NULL;
960}
961
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000962/* test_thread_state spawns a thread of its own, and that thread releases
963 * `thread_done` when it's finished. The driver code has to know when the
964 * thread finishes, because the thread uses a PyObject (the callable) that
965 * may go away when the driver finishes. The former lack of this explicit
966 * synchronization caused rare segfaults, so rare that they were seen only
967 * on a Mac buildbot (although they were possible on any box).
968 */
969static PyThread_type_lock thread_done = NULL;
970
Benjamin Petersona786b022008-08-25 21:05:21 +0000971static int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000972_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000973{
974 PyObject *rc;
Benjamin Petersona786b022008-08-25 21:05:21 +0000975 int success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000976 PyGILState_STATE s = PyGILState_Ensure();
Thomas Wouters89f507f2006-12-13 04:49:30 +0000977 rc = PyObject_CallFunction((PyObject *)callable, "");
Benjamin Petersona786b022008-08-25 21:05:21 +0000978 success = (rc != NULL);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000979 Py_XDECREF(rc);
980 PyGILState_Release(s);
Benjamin Petersona786b022008-08-25 21:05:21 +0000981 return success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000982}
983
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000984/* Same thing, but releases `thread_done` when it returns. This variant
985 * should be called only from threads spawned by test_thread_state().
986 */
987static void
988_make_call_from_thread(void *callable)
989{
990 _make_call(callable);
991 PyThread_release_lock(thread_done);
992}
993
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000994static PyObject *
995test_thread_state(PyObject *self, PyObject *args)
996{
997 PyObject *fn;
Benjamin Petersona786b022008-08-25 21:05:21 +0000998 int success = 1;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000999
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001000 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
1001 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001002
Benjamin Petersona786b022008-08-25 21:05:21 +00001003 if (!PyCallable_Check(fn)) {
1004 PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
1005 fn->ob_type->tp_name);
1006 return NULL;
1007 }
1008
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001009 /* Ensure Python is set up for threading */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001010 PyEval_InitThreads();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001011 thread_done = PyThread_allocate_lock();
1012 if (thread_done == NULL)
1013 return PyErr_NoMemory();
1014 PyThread_acquire_lock(thread_done, 1);
1015
1016 /* Start a new thread with our callback. */
1017 PyThread_start_new_thread(_make_call_from_thread, fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001018 /* Make the callback with the thread lock held by this thread */
Benjamin Petersona786b022008-08-25 21:05:21 +00001019 success &= _make_call(fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001020 /* Do it all again, but this time with the thread-lock released */
1021 Py_BEGIN_ALLOW_THREADS
Benjamin Petersona786b022008-08-25 21:05:21 +00001022 success &= _make_call(fn);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001023 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001024 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001025
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001026 /* And once more with and without a thread
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001027 XXX - should use a lock and work out exactly what we are trying
1028 to test <wink>
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001029 */
1030 Py_BEGIN_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001031 PyThread_start_new_thread(_make_call_from_thread, fn);
Benjamin Petersona786b022008-08-25 21:05:21 +00001032 success &= _make_call(fn);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001033 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001034 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001035
Thomas Wouters89f507f2006-12-13 04:49:30 +00001036 /* Release lock we acquired above. This is required on HP-UX. */
1037 PyThread_release_lock(thread_done);
1038
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001039 PyThread_free_lock(thread_done);
Benjamin Petersona786b022008-08-25 21:05:21 +00001040 if (!success)
1041 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001042 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001043}
Benjamin Petersona54c9092009-01-13 02:11:23 +00001044
1045/* test Py_AddPendingCalls using threads */
1046static int _pending_callback(void *arg)
1047{
1048 /* we assume the argument is callable object to which we own a reference */
1049 PyObject *callable = (PyObject *)arg;
1050 PyObject *r = PyObject_CallObject(callable, NULL);
1051 Py_DECREF(callable);
1052 Py_XDECREF(r);
1053 return r != NULL ? 0 : -1;
1054}
1055
1056/* The following requests n callbacks to _pending_callback. It can be
1057 * run from any python thread.
1058 */
1059PyObject *pending_threadfunc(PyObject *self, PyObject *arg)
1060{
1061 PyObject *callable;
1062 int r;
1063 if (PyArg_ParseTuple(arg, "O", &callable) == 0)
1064 return NULL;
1065
1066 /* create the reference for the callbackwhile we hold the lock */
1067 Py_INCREF(callable);
1068
1069 Py_BEGIN_ALLOW_THREADS
1070 r = Py_AddPendingCall(&_pending_callback, callable);
1071 Py_END_ALLOW_THREADS
1072
1073 if (r<0) {
1074 Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
1075 Py_INCREF(Py_False);
1076 return Py_False;
1077 }
1078 Py_INCREF(Py_True);
1079 return Py_True;
1080}
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001081#endif
1082
Neal Norwitzb0d26332007-08-25 00:49:05 +00001083/* Some tests of PyUnicode_FromFormat(). This needs more tests. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001084static PyObject *
1085test_string_from_format(PyObject *self, PyObject *args)
1086{
1087 PyObject *result;
1088 char *msg;
1089
1090#define CHECK_1_FORMAT(FORMAT, TYPE) \
Neal Norwitzb0d26332007-08-25 00:49:05 +00001091 result = PyUnicode_FromFormat(FORMAT, (TYPE)1); \
Thomas Wouters477c8d52006-05-27 19:21:47 +00001092 if (result == NULL) \
1093 return NULL; \
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00001094 if (strcmp(_PyUnicode_AsString(result), "1")) { \
Thomas Wouters477c8d52006-05-27 19:21:47 +00001095 msg = FORMAT " failed at 1"; \
1096 goto Fail; \
1097 } \
1098 Py_DECREF(result)
1099
1100 CHECK_1_FORMAT("%d", int);
1101 CHECK_1_FORMAT("%ld", long);
1102 /* The z width modifier was added in Python 2.5. */
1103 CHECK_1_FORMAT("%zd", Py_ssize_t);
1104
1105 /* The u type code was added in Python 2.5. */
1106 CHECK_1_FORMAT("%u", unsigned int);
1107 CHECK_1_FORMAT("%lu", unsigned long);
1108 CHECK_1_FORMAT("%zu", size_t);
1109
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001110 /* "%lld" and "%llu" support added in Python 2.7. */
1111#ifdef HAVE_LONG_LONG
1112 CHECK_1_FORMAT("%llu", unsigned PY_LONG_LONG);
1113 CHECK_1_FORMAT("%lld", PY_LONG_LONG);
1114#endif
1115
Thomas Wouters477c8d52006-05-27 19:21:47 +00001116 Py_RETURN_NONE;
1117
1118 Fail:
1119 Py_XDECREF(result);
1120 return raiseTestError("test_string_from_format", msg);
1121
1122#undef CHECK_1_FORMAT
1123}
1124
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001125/* This is here to provide a docstring for test_descr. */
1126static PyObject *
1127test_with_docstring(PyObject *self)
1128{
1129 Py_RETURN_NONE;
1130}
1131
Mark Dickinson725bfd82009-05-03 20:33:40 +00001132/* Test PyOS_string_to_double. */
1133static PyObject *
1134test_string_to_double(PyObject *self) {
1135 double result;
1136 char *msg;
1137
1138#define CHECK_STRING(STR, expected) \
1139 result = PyOS_string_to_double(STR, NULL, NULL); \
1140 if (result == -1.0 && PyErr_Occurred()) \
1141 return NULL; \
1142 if (result != expected) { \
1143 msg = "conversion of " STR " to float failed"; \
1144 goto fail; \
1145 }
1146
1147#define CHECK_INVALID(STR) \
1148 result = PyOS_string_to_double(STR, NULL, NULL); \
1149 if (result == -1.0 && PyErr_Occurred()) { \
1150 if (PyErr_ExceptionMatches(PyExc_ValueError)) \
1151 PyErr_Clear(); \
1152 else \
1153 return NULL; \
1154 } \
1155 else { \
1156 msg = "conversion of " STR " didn't raise ValueError"; \
1157 goto fail; \
1158 }
1159
1160 CHECK_STRING("0.1", 0.1);
1161 CHECK_STRING("1.234", 1.234);
1162 CHECK_STRING("-1.35", -1.35);
1163 CHECK_STRING(".1e01", 1.0);
1164 CHECK_STRING("2.e-2", 0.02);
1165
1166 CHECK_INVALID(" 0.1");
1167 CHECK_INVALID("\t\n-3");
1168 CHECK_INVALID(".123 ");
1169 CHECK_INVALID("3\n");
1170 CHECK_INVALID("123abc");
1171
1172 Py_RETURN_NONE;
1173 fail:
1174 return raiseTestError("test_string_to_double", msg);
1175#undef CHECK_STRING
1176#undef CHECK_INVALID
1177}
1178
1179
Benjamin Petersonb173f782009-05-05 22:31:58 +00001180/* Coverage testing of capsule objects. */
1181
1182static const char *capsule_name = "capsule name";
1183static char *capsule_pointer = "capsule pointer";
1184static char *capsule_context = "capsule context";
1185static const char *capsule_error = NULL;
1186static int
1187capsule_destructor_call_count = 0;
1188
1189static void
1190capsule_destructor(PyObject *o) {
1191 capsule_destructor_call_count++;
1192 if (PyCapsule_GetContext(o) != capsule_context) {
1193 capsule_error = "context did not match in destructor!";
1194 } else if (PyCapsule_GetDestructor(o) != capsule_destructor) {
1195 capsule_error = "destructor did not match in destructor! (woah!)";
1196 } else if (PyCapsule_GetName(o) != capsule_name) {
1197 capsule_error = "name did not match in destructor!";
1198 } else if (PyCapsule_GetPointer(o, capsule_name) != capsule_pointer) {
1199 capsule_error = "pointer did not match in destructor!";
1200 }
1201}
1202
1203typedef struct {
1204 char *name;
1205 char *module;
1206 char *attribute;
1207} known_capsule;
1208
1209static PyObject *
1210test_capsule(PyObject *self, PyObject *args)
1211{
1212 PyObject *object;
1213 const char *error = NULL;
1214 void *pointer;
1215 void *pointer2;
1216 known_capsule known_capsules[] = {
1217 #define KNOWN_CAPSULE(module, name) { module "." name, module, name }
1218 KNOWN_CAPSULE("_socket", "CAPI"),
1219 KNOWN_CAPSULE("_curses", "_C_API"),
1220 KNOWN_CAPSULE("datetime", "datetime_CAPI"),
1221 { NULL, NULL },
1222 };
1223 known_capsule *known = &known_capsules[0];
1224
1225#define FAIL(x) { error = (x); goto exit; }
1226
1227#define CHECK_DESTRUCTOR \
1228 if (capsule_error) { \
1229 FAIL(capsule_error); \
1230 } \
1231 else if (!capsule_destructor_call_count) { \
1232 FAIL("destructor not called!"); \
1233 } \
1234 capsule_destructor_call_count = 0; \
1235
1236 object = PyCapsule_New(capsule_pointer, capsule_name, capsule_destructor);
1237 PyCapsule_SetContext(object, capsule_context);
1238 capsule_destructor(object);
1239 CHECK_DESTRUCTOR;
1240 Py_DECREF(object);
1241 CHECK_DESTRUCTOR;
1242
1243 object = PyCapsule_New(known, "ignored", NULL);
1244 PyCapsule_SetPointer(object, capsule_pointer);
1245 PyCapsule_SetName(object, capsule_name);
1246 PyCapsule_SetDestructor(object, capsule_destructor);
1247 PyCapsule_SetContext(object, capsule_context);
1248 capsule_destructor(object);
1249 CHECK_DESTRUCTOR;
1250 /* intentionally access using the wrong name */
1251 pointer2 = PyCapsule_GetPointer(object, "the wrong name");
1252 if (!PyErr_Occurred()) {
1253 FAIL("PyCapsule_GetPointer should have failed but did not!");
1254 }
1255 PyErr_Clear();
1256 if (pointer2) {
1257 if (pointer2 == capsule_pointer) {
1258 FAIL("PyCapsule_GetPointer should not have"
1259 " returned the internal pointer!");
1260 } else {
1261 FAIL("PyCapsule_GetPointer should have "
1262 "returned NULL pointer but did not!");
1263 }
1264 }
1265 PyCapsule_SetDestructor(object, NULL);
1266 Py_DECREF(object);
1267 if (capsule_destructor_call_count) {
1268 FAIL("destructor called when it should not have been!");
1269 }
1270
1271 for (known = &known_capsules[0]; known->module != NULL; known++) {
1272 /* yeah, ordinarily I wouldn't do this either,
1273 but it's fine for this test harness.
1274 */
1275 static char buffer[256];
1276#undef FAIL
1277#define FAIL(x) \
1278 { \
1279 sprintf(buffer, "%s module: \"%s\" attribute: \"%s\"", \
1280 x, known->module, known->attribute); \
1281 error = buffer; \
1282 goto exit; \
1283 } \
1284
1285 PyObject *module = PyImport_ImportModule(known->module);
1286 if (module) {
1287 pointer = PyCapsule_Import(known->name, 0);
1288 if (!pointer) {
1289 Py_DECREF(module);
1290 FAIL("PyCapsule_GetPointer returned NULL unexpectedly!");
1291 }
1292 object = PyObject_GetAttrString(module, known->attribute);
1293 if (!object) {
1294 Py_DECREF(module);
1295 return NULL;
1296 }
1297 pointer2 = PyCapsule_GetPointer(object,
1298 "weebles wobble but they don't fall down");
1299 if (!PyErr_Occurred()) {
1300 Py_DECREF(object);
1301 Py_DECREF(module);
1302 FAIL("PyCapsule_GetPointer should have failed but did not!");
1303 }
1304 PyErr_Clear();
1305 if (pointer2) {
1306 Py_DECREF(module);
1307 Py_DECREF(object);
1308 if (pointer2 == pointer) {
1309 FAIL("PyCapsule_GetPointer should not have"
1310 " returned its internal pointer!");
1311 } else {
1312 FAIL("PyCapsule_GetPointer should have"
1313 " returned NULL pointer but did not!");
1314 }
1315 }
1316 Py_DECREF(object);
1317 Py_DECREF(module);
1318 }
1319 }
1320
1321 exit:
1322 if (error) {
1323 return raiseTestError("test_capsule", error);
1324 }
Mark Dickinson6d138f12009-12-13 20:03:21 +00001325 /* 13/12/2009: something is causing test_capi to fail occasionally on
1326 the Solaris buildbot, with the output:
1327
1328 internal test_L_code
1329 internal test_Z_code
1330 internal test_bug_7414
1331 internal test_capsule
1332 XXX undetected error
1333 internaltest test_capi crashed -- <class 'ImportError'>: No module named datetime
1334
1335 It seems possible that test_capsule is raising an exception but
1336 failing to return NULL. Do a PyErr_Occurred check to find out.
1337 */
1338 if (PyErr_Occurred())
1339 return NULL;
Benjamin Petersonb173f782009-05-05 22:31:58 +00001340 Py_RETURN_NONE;
1341#undef FAIL
1342}
1343
Guido van Rossumddefaf32007-01-14 03:31:43 +00001344#ifdef HAVE_GETTIMEOFDAY
1345/* Profiling of integer performance */
Martin v. Löwis1c951552008-06-13 07:48:19 +00001346static void print_delta(int test, struct timeval *s, struct timeval *e)
Guido van Rossumddefaf32007-01-14 03:31:43 +00001347{
1348 e->tv_sec -= s->tv_sec;
1349 e->tv_usec -= s->tv_usec;
1350 if (e->tv_usec < 0) {
1351 e->tv_sec -=1;
1352 e->tv_usec += 1000000;
1353 }
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001354 printf("Test %d: %d.%06ds\n", test, (int)e->tv_sec, (int)e->tv_usec);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001355}
1356
1357static PyObject *
1358profile_int(PyObject *self, PyObject* args)
1359{
1360 int i, k;
1361 struct timeval start, stop;
1362 PyObject *single, **multiple, *op1, *result;
1363
1364 /* Test 1: Allocate and immediately deallocate
1365 many small integers */
1366 gettimeofday(&start, NULL);
1367 for(k=0; k < 20000; k++)
1368 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001369 single = PyLong_FromLong(i);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001370 Py_DECREF(single);
1371 }
1372 gettimeofday(&stop, NULL);
1373 print_delta(1, &start, &stop);
1374
1375 /* Test 2: Allocate and immediately deallocate
1376 many large integers */
1377 gettimeofday(&start, NULL);
1378 for(k=0; k < 20000; k++)
1379 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001380 single = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001381 Py_DECREF(single);
1382 }
1383 gettimeofday(&stop, NULL);
1384 print_delta(2, &start, &stop);
1385
1386 /* Test 3: Allocate a few integers, then release
1387 them all simultaneously. */
1388 multiple = malloc(sizeof(PyObject*) * 1000);
1389 gettimeofday(&start, NULL);
1390 for(k=0; k < 20000; k++) {
1391 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001392 multiple[i] = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001393 }
1394 for(i=0; i < 1000; i++) {
1395 Py_DECREF(multiple[i]);
1396 }
1397 }
1398 gettimeofday(&stop, NULL);
1399 print_delta(3, &start, &stop);
1400
1401 /* Test 4: Allocate many integers, then release
1402 them all simultaneously. */
1403 multiple = malloc(sizeof(PyObject*) * 1000000);
1404 gettimeofday(&start, NULL);
1405 for(k=0; k < 20; k++) {
1406 for(i=0; i < 1000000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001407 multiple[i] = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001408 }
1409 for(i=0; i < 1000000; i++) {
1410 Py_DECREF(multiple[i]);
1411 }
1412 }
1413 gettimeofday(&stop, NULL);
1414 print_delta(4, &start, &stop);
1415
1416 /* Test 5: Allocate many integers < 32000 */
1417 multiple = malloc(sizeof(PyObject*) * 1000000);
1418 gettimeofday(&start, NULL);
1419 for(k=0; k < 10; k++) {
1420 for(i=0; i < 1000000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001421 multiple[i] = PyLong_FromLong(i+1000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001422 }
1423 for(i=0; i < 1000000; i++) {
1424 Py_DECREF(multiple[i]);
1425 }
1426 }
1427 gettimeofday(&stop, NULL);
1428 print_delta(5, &start, &stop);
1429
1430 /* Test 6: Perform small int addition */
Christian Heimes217cfd12007-12-02 14:31:20 +00001431 op1 = PyLong_FromLong(1);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001432 gettimeofday(&start, NULL);
1433 for(i=0; i < 10000000; i++) {
1434 result = PyNumber_Add(op1, op1);
1435 Py_DECREF(result);
1436 }
1437 gettimeofday(&stop, NULL);
1438 Py_DECREF(op1);
1439 print_delta(6, &start, &stop);
1440
1441 /* Test 7: Perform medium int addition */
Christian Heimes217cfd12007-12-02 14:31:20 +00001442 op1 = PyLong_FromLong(1000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001443 gettimeofday(&start, NULL);
1444 for(i=0; i < 10000000; i++) {
1445 result = PyNumber_Add(op1, op1);
1446 Py_DECREF(result);
1447 }
1448 gettimeofday(&stop, NULL);
1449 Py_DECREF(op1);
1450 print_delta(7, &start, &stop);
1451
1452 Py_INCREF(Py_None);
1453 return Py_None;
1454}
1455#endif
1456
Christian Heimes81ee3ef2008-05-04 22:42:01 +00001457/* To test the format of tracebacks as printed out. */
1458static PyObject *
1459traceback_print(PyObject *self, PyObject *args)
1460{
1461 PyObject *file;
1462 PyObject *traceback;
1463 int result;
1464
1465 if (!PyArg_ParseTuple(args, "OO:traceback_print",
1466 &traceback, &file))
1467 return NULL;
1468
1469 result = PyTraceBack_Print(traceback, file);
1470 if (result < 0)
1471 return NULL;
1472 Py_RETURN_NONE;
1473}
1474
Benjamin Petersone6528212008-07-15 15:32:09 +00001475/* To test the format of exceptions as printed out. */
1476static PyObject *
1477exception_print(PyObject *self, PyObject *args)
1478{
1479 PyObject *value;
1480 PyObject *tb;
1481
1482 if (!PyArg_ParseTuple(args, "O:exception_print",
1483 &value))
1484 return NULL;
Benjamin Peterson6784eb72008-08-23 20:32:27 +00001485 if (!PyExceptionInstance_Check(value)) {
1486 PyErr_Format(PyExc_TypeError, "an exception instance is required");
1487 return NULL;
1488 }
Benjamin Petersone6528212008-07-15 15:32:09 +00001489
1490 tb = PyException_GetTraceback(value);
1491 PyErr_Display((PyObject *) Py_TYPE(value), value, tb);
1492 Py_XDECREF(tb);
1493
1494 Py_RETURN_NONE;
1495}
1496
1497
1498
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001499
1500/* reliably raise a MemoryError */
1501static PyObject *
1502raise_memoryerror(PyObject *self)
1503{
1504 PyErr_NoMemory();
1505 return NULL;
1506}
1507
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00001508/* Issue 6012 */
1509static PyObject *str1, *str2;
1510static int
1511failing_converter(PyObject *obj, void *arg)
1512{
1513 /* Clone str1, then let the conversion fail. */
1514 assert(str1);
1515 str2 = str1;
1516 Py_INCREF(str2);
1517 return 0;
1518}
1519static PyObject*
1520argparsing(PyObject *o, PyObject *args)
1521{
1522 PyObject *res;
1523 str1 = str2 = NULL;
1524 if (!PyArg_ParseTuple(args, "O&O&",
1525 PyUnicode_FSConverter, &str1,
1526 failing_converter, &str2)) {
1527 if (!str2)
1528 /* argument converter not called? */
1529 return NULL;
1530 /* Should be 1 */
1531 res = PyLong_FromLong(Py_REFCNT(str2));
1532 Py_DECREF(str2);
1533 PyErr_Clear();
1534 return res;
1535 }
1536 Py_RETURN_NONE;
1537}
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001538
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00001539/* To test that the result of PyCode_NewEmpty has the right members. */
1540static PyObject *
1541code_newempty(PyObject *self, PyObject *args)
1542{
1543 const char *filename;
1544 const char *funcname;
1545 int firstlineno;
1546
1547 if (!PyArg_ParseTuple(args, "ssi:code_newempty",
1548 &filename, &funcname, &firstlineno))
1549 return NULL;
1550
1551 return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
1552}
1553
Tim Peters9ea17ac2001-02-02 05:57:15 +00001554static PyMethodDef TestMethods[] = {
Fred Drakeacee69f2002-04-01 14:28:58 +00001555 {"raise_exception", raise_exception, METH_VARARGS},
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001556 {"raise_memoryerror", (PyCFunction)raise_memoryerror, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001557 {"test_config", (PyCFunction)test_config, METH_NOARGS},
Benjamin Peterson91d58bd2009-12-13 21:30:54 +00001558 {"test_datetime_capi", test_datetime_capi, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001559 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
1560 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
Nick Coghlanf1f2f682008-12-30 07:29:12 +00001561 {"test_lazy_hash_inheritance", (PyCFunction)test_lazy_hash_inheritance,METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001562 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
Tim Petersbaefd9e2003-01-28 20:37:45 +00001563 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
Thomas Hellera4ea6032003-04-17 18:55:45 +00001564 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Benjamin Peterson92035012008-12-27 16:00:54 +00001565 {"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
Mark Dickinsonf08173b2009-12-03 10:59:46 +00001566 {"test_bug_7414", (PyCFunction)test_bug_7414, METH_NOARGS},
Thomas Woutersa9773292006-04-21 09:43:23 +00001567 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
Thomas Wouters477c8d52006-05-27 19:21:47 +00001568 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001569 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
1570 PyDoc_STR("This is a pretty normal docstring.")},
Benjamin Petersonb173f782009-05-05 22:31:58 +00001571 {"test_string_to_double", (PyCFunction)test_string_to_double, METH_NOARGS},
1572 {"test_capsule", (PyCFunction)test_capsule, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001573 {"getargs_tuple", getargs_tuple, METH_VARARGS},
Christian Heimes380f7f22008-02-28 11:19:05 +00001574 {"getargs_keywords", (PyCFunction)getargs_keywords,
1575 METH_VARARGS|METH_KEYWORDS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001576 {"getargs_b", getargs_b, METH_VARARGS},
1577 {"getargs_B", getargs_B, METH_VARARGS},
Mark Dickinson1554b182009-12-20 16:03:30 +00001578 {"getargs_h", getargs_h, METH_VARARGS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001579 {"getargs_H", getargs_H, METH_VARARGS},
1580 {"getargs_I", getargs_I, METH_VARARGS},
1581 {"getargs_k", getargs_k, METH_VARARGS},
1582 {"getargs_i", getargs_i, METH_VARARGS},
1583 {"getargs_l", getargs_l, METH_VARARGS},
1584 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001585#ifdef HAVE_LONG_LONG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001586 {"getargs_L", getargs_L, METH_VARARGS},
1587 {"getargs_K", getargs_K, METH_VARARGS},
1588 {"test_longlong_api", test_longlong_api, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001589 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001590 {"codec_incrementalencoder",
1591 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
1592 {"codec_incrementaldecoder",
1593 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001594#endif
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +00001595 {"test_s_code", (PyCFunction)test_s_code, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001596 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
Guido van Rossumfb67be22007-08-29 18:38:11 +00001597 {"test_Z_code", (PyCFunction)test_Z_code, METH_NOARGS},
Mark Dickinson081dfee2009-03-18 14:47:41 +00001598 {"test_widechar", (PyCFunction)test_widechar, METH_NOARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001599#ifdef WITH_THREAD
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001600 {"_test_thread_state", test_thread_state, METH_VARARGS},
Benjamin Petersona54c9092009-01-13 02:11:23 +00001601 {"_pending_threadfunc", pending_threadfunc, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001602#endif
Guido van Rossumddefaf32007-01-14 03:31:43 +00001603#ifdef HAVE_GETTIMEOFDAY
1604 {"profile_int", profile_int, METH_NOARGS},
1605#endif
Christian Heimes81ee3ef2008-05-04 22:42:01 +00001606 {"traceback_print", traceback_print, METH_VARARGS},
Benjamin Petersone6528212008-07-15 15:32:09 +00001607 {"exception_print", exception_print, METH_VARARGS},
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00001608 {"argparsing", argparsing, METH_VARARGS},
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00001609 {"code_newempty", code_newempty, METH_VARARGS},
Tim Peters9ea17ac2001-02-02 05:57:15 +00001610 {NULL, NULL} /* sentinel */
1611};
1612
Thomas Hellera4ea6032003-04-17 18:55:45 +00001613#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
1614
Thomas Wouters89f507f2006-12-13 04:49:30 +00001615typedef struct {
Christian Heimes1af737c2008-01-23 08:24:23 +00001616 char bool_member;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001617 char byte_member;
1618 unsigned char ubyte_member;
1619 short short_member;
1620 unsigned short ushort_member;
1621 int int_member;
1622 unsigned int uint_member;
1623 long long_member;
1624 unsigned long ulong_member;
1625 float float_member;
1626 double double_member;
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001627#ifdef HAVE_LONG_LONG
1628 PY_LONG_LONG longlong_member;
1629 unsigned PY_LONG_LONG ulonglong_member;
1630#endif
Thomas Wouters89f507f2006-12-13 04:49:30 +00001631} all_structmembers;
1632
1633typedef struct {
1634 PyObject_HEAD
1635 all_structmembers structmembers;
1636} test_structmembers;
1637
1638static struct PyMemberDef test_members[] = {
Christian Heimes1af737c2008-01-23 08:24:23 +00001639 {"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
Thomas Wouters89f507f2006-12-13 04:49:30 +00001640 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
1641 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
1642 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
1643 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
1644 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
1645 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
1646 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
1647 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
1648 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
1649 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001650#ifdef HAVE_LONG_LONG
1651 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
1652 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
1653#endif
Thomas Wouters89f507f2006-12-13 04:49:30 +00001654 {NULL}
1655};
1656
1657
Christian Heimes1af737c2008-01-23 08:24:23 +00001658static PyObject *
1659test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
1660{
1661 static char *keywords[] = {
1662 "T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
1663 "T_INT", "T_UINT", "T_LONG", "T_ULONG",
1664 "T_FLOAT", "T_DOUBLE",
1665#ifdef HAVE_LONG_LONG
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001666 "T_LONGLONG", "T_ULONGLONG",
Christian Heimes1af737c2008-01-23 08:24:23 +00001667#endif
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001668 NULL};
Christian Heimes1af737c2008-01-23 08:24:23 +00001669 static char *fmt = "|bbBhHiIlkfd"
1670#ifdef HAVE_LONG_LONG
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001671 "LK"
Christian Heimes1af737c2008-01-23 08:24:23 +00001672#endif
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001673 ;
Christian Heimes1af737c2008-01-23 08:24:23 +00001674 test_structmembers *ob;
1675 ob = PyObject_New(test_structmembers, type);
1676 if (ob == NULL)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001677 return NULL;
1678 memset(&ob->structmembers, 0, sizeof(all_structmembers));
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001679 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
Christian Heimes1af737c2008-01-23 08:24:23 +00001680 &ob->structmembers.bool_member,
1681 &ob->structmembers.byte_member,
1682 &ob->structmembers.ubyte_member,
1683 &ob->structmembers.short_member,
1684 &ob->structmembers.ushort_member,
1685 &ob->structmembers.int_member,
1686 &ob->structmembers.uint_member,
1687 &ob->structmembers.long_member,
1688 &ob->structmembers.ulong_member,
1689 &ob->structmembers.float_member,
1690 &ob->structmembers.double_member
1691#ifdef HAVE_LONG_LONG
1692 , &ob->structmembers.longlong_member,
1693 &ob->structmembers.ulonglong_member
1694#endif
1695 )) {
Thomas Wouters89f507f2006-12-13 04:49:30 +00001696 Py_DECREF(ob);
1697 return NULL;
Christian Heimes1af737c2008-01-23 08:24:23 +00001698 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00001699 return (PyObject *)ob;
1700}
1701
Christian Heimes1af737c2008-01-23 08:24:23 +00001702static void
1703test_structmembers_free(PyObject *ob)
1704{
Thomas Wouters89f507f2006-12-13 04:49:30 +00001705 PyObject_FREE(ob);
1706}
1707
1708static PyTypeObject test_structmembersType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001709 PyVarObject_HEAD_INIT(NULL, 0)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001710 "test_structmembersType",
1711 sizeof(test_structmembers), /* tp_basicsize */
1712 0, /* tp_itemsize */
1713 test_structmembers_free, /* destructor tp_dealloc */
1714 0, /* tp_print */
1715 0, /* tp_getattr */
1716 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +00001717 0, /* tp_reserved */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001718 0, /* tp_repr */
1719 0, /* tp_as_number */
1720 0, /* tp_as_sequence */
1721 0, /* tp_as_mapping */
1722 0, /* tp_hash */
1723 0, /* tp_call */
1724 0, /* tp_str */
Christian Heimes1af737c2008-01-23 08:24:23 +00001725 PyObject_GenericGetAttr, /* tp_getattro */
1726 PyObject_GenericSetAttr, /* tp_setattro */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001727 0, /* tp_as_buffer */
1728 0, /* tp_flags */
1729 "Type containing all structmember types",
1730 0, /* traverseproc tp_traverse */
1731 0, /* tp_clear */
1732 0, /* tp_richcompare */
1733 0, /* tp_weaklistoffset */
1734 0, /* tp_iter */
1735 0, /* tp_iternext */
1736 0, /* tp_methods */
Christian Heimes1af737c2008-01-23 08:24:23 +00001737 test_members, /* tp_members */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001738 0,
1739 0,
1740 0,
1741 0,
1742 0,
1743 0,
1744 0,
1745 0,
Christian Heimes1af737c2008-01-23 08:24:23 +00001746 test_structmembers_new, /* tp_new */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001747};
1748
1749
Martin v. Löwis1a214512008-06-11 05:26:20 +00001750
1751static struct PyModuleDef _testcapimodule = {
1752 PyModuleDef_HEAD_INIT,
1753 "_testcapi",
1754 NULL,
1755 -1,
1756 TestMethods,
1757 NULL,
1758 NULL,
1759 NULL,
1760 NULL
1761};
1762
Mark Hammond62b1ab12002-07-23 06:31:15 +00001763PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001764PyInit__testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +00001765{
Fred Drakeacee69f2002-04-01 14:28:58 +00001766 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001767
Martin v. Löwis1a214512008-06-11 05:26:20 +00001768 m = PyModule_Create(&_testcapimodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001769 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00001770 return NULL;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001771
Hirokazu Yamamoto8ebab5d2008-12-31 06:05:46 +00001772 Py_TYPE(&_HashInheritanceTester_Type)=&PyType_Type;
1773
Christian Heimes90aa7642007-12-19 02:45:37 +00001774 Py_TYPE(&test_structmembersType)=&PyType_Type;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001775 Py_INCREF(&test_structmembersType);
1776 PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
1777
Christian Heimes217cfd12007-12-02 14:31:20 +00001778 PyModule_AddObject(m, "CHAR_MAX", PyLong_FromLong(CHAR_MAX));
1779 PyModule_AddObject(m, "CHAR_MIN", PyLong_FromLong(CHAR_MIN));
1780 PyModule_AddObject(m, "UCHAR_MAX", PyLong_FromLong(UCHAR_MAX));
1781 PyModule_AddObject(m, "SHRT_MAX", PyLong_FromLong(SHRT_MAX));
1782 PyModule_AddObject(m, "SHRT_MIN", PyLong_FromLong(SHRT_MIN));
1783 PyModule_AddObject(m, "USHRT_MAX", PyLong_FromLong(USHRT_MAX));
Thomas Wouters89f507f2006-12-13 04:49:30 +00001784 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
1785 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001786 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
Christian Heimes217cfd12007-12-02 14:31:20 +00001787 PyModule_AddObject(m, "LONG_MAX", PyLong_FromLong(LONG_MAX));
1788 PyModule_AddObject(m, "LONG_MIN", PyLong_FromLong(LONG_MIN));
Thomas Wouters89f507f2006-12-13 04:49:30 +00001789 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
1790 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
1791 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
1792 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
1793 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001794 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
1795 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
1796 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
Christian Heimes217cfd12007-12-02 14:31:20 +00001797 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyLong_FromSsize_t(PY_SSIZE_T_MAX));
1798 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyLong_FromSsize_t(PY_SSIZE_T_MIN));
Robert Schuppeniesfbe94c52008-07-14 10:13:31 +00001799 PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyLong_FromSsize_t(sizeof(PyGC_Head)));
Benjamin Peterson9b6df6a2008-10-16 23:56:29 +00001800 Py_INCREF(&PyInstanceMethod_Type);
1801 PyModule_AddObject(m, "instancemethod", (PyObject *)&PyInstanceMethod_Type);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001802
Tim Petersd66595f2001-02-04 03:09:53 +00001803 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
Fred Drakeacee69f2002-04-01 14:28:58 +00001804 Py_INCREF(TestError);
1805 PyModule_AddObject(m, "error", TestError);
Martin v. Löwis1a214512008-06-11 05:26:20 +00001806 return m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001807}