blob: b6354837b40c54f4ac1d495faf5168ed6cf867a1 [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
Mark Dickinson309aa2d2009-12-21 12:37:06 +0000362/* Test the PyLong_AsLongAndOverflow API. General conversion to PY_LONG
363 is tested by test_long_api_inner. This test will concentrate on proper
364 handling of overflow.
365*/
366
367static PyObject *
368test_long_and_overflow(PyObject *self)
369{
370 PyObject *num, *one, *temp;
371 long value;
372 int overflow;
373
374 /* Test that overflow is set properly for a large value. */
375 /* num is a number larger than LONG_MAX even on 64-bit platforms */
376 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
377 if (num == NULL)
378 return NULL;
379 overflow = 1234;
380 value = PyLong_AsLongAndOverflow(num, &overflow);
381 Py_DECREF(num);
382 if (value == -1 && PyErr_Occurred())
383 return NULL;
384 if (value != -1)
385 return raiseTestError("test_long_and_overflow",
386 "return value was not set to -1");
387 if (overflow != 1)
388 return raiseTestError("test_long_and_overflow",
389 "overflow was not set to 1");
390
391 /* Same again, with num = LONG_MAX + 1 */
392 num = PyLong_FromLong(LONG_MAX);
393 if (num == NULL)
394 return NULL;
395 one = PyLong_FromLong(1L);
396 if (one == NULL) {
397 Py_DECREF(num);
398 return NULL;
399 }
400 temp = PyNumber_Add(num, one);
401 Py_DECREF(one);
402 Py_DECREF(num);
403 num = temp;
404 if (num == NULL)
405 return NULL;
406 overflow = 0;
407 value = PyLong_AsLongAndOverflow(num, &overflow);
408 Py_DECREF(num);
409 if (value == -1 && PyErr_Occurred())
410 return NULL;
411 if (value != -1)
412 return raiseTestError("test_long_and_overflow",
413 "return value was not set to -1");
414 if (overflow != 1)
415 return raiseTestError("test_long_and_overflow",
416 "overflow was not set to 1");
417
418 /* Test that overflow is set properly for a large negative value. */
419 /* num is a number smaller than LONG_MIN even on 64-bit platforms */
420 num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
421 if (num == NULL)
422 return NULL;
423 overflow = 1234;
424 value = PyLong_AsLongAndOverflow(num, &overflow);
425 Py_DECREF(num);
426 if (value == -1 && PyErr_Occurred())
427 return NULL;
428 if (value != -1)
429 return raiseTestError("test_long_and_overflow",
430 "return value was not set to -1");
431 if (overflow != -1)
432 return raiseTestError("test_long_and_overflow",
433 "overflow was not set to -1");
434
435 /* Same again, with num = LONG_MIN - 1 */
436 num = PyLong_FromLong(LONG_MIN);
437 if (num == NULL)
438 return NULL;
439 one = PyLong_FromLong(1L);
440 if (one == NULL) {
441 Py_DECREF(num);
442 return NULL;
443 }
444 temp = PyNumber_Subtract(num, one);
445 Py_DECREF(one);
446 Py_DECREF(num);
447 num = temp;
448 if (num == NULL)
449 return NULL;
450 overflow = 0;
451 value = PyLong_AsLongAndOverflow(num, &overflow);
452 Py_DECREF(num);
453 if (value == -1 && PyErr_Occurred())
454 return NULL;
455 if (value != -1)
456 return raiseTestError("test_long_and_overflow",
457 "return value was not set to -1");
458 if (overflow != -1)
459 return raiseTestError("test_long_and_overflow",
460 "overflow was not set to -1");
461
462 /* Test that overflow is cleared properly for small values. */
463 num = PyLong_FromString("FF", NULL, 16);
464 if (num == NULL)
465 return NULL;
466 overflow = 1234;
467 value = PyLong_AsLongAndOverflow(num, &overflow);
468 Py_DECREF(num);
469 if (value == -1 && PyErr_Occurred())
470 return NULL;
471 if (value != 0xFF)
472 return raiseTestError("test_long_and_overflow",
473 "expected return value 0xFF");
474 if (overflow != 0)
475 return raiseTestError("test_long_and_overflow",
476 "overflow was not cleared");
477
478 num = PyLong_FromString("-FF", NULL, 16);
479 if (num == NULL)
480 return NULL;
481 overflow = 0;
482 value = PyLong_AsLongAndOverflow(num, &overflow);
483 Py_DECREF(num);
484 if (value == -1 && PyErr_Occurred())
485 return NULL;
486 if (value != -0xFF)
487 return raiseTestError("test_long_and_overflow",
488 "expected return value 0xFF");
489 if (overflow != 0)
490 return raiseTestError("test_long_and_overflow",
491 "overflow was set incorrectly");
492
493 num = PyLong_FromLong(LONG_MAX);
494 if (num == NULL)
495 return NULL;
496 overflow = 1234;
497 value = PyLong_AsLongAndOverflow(num, &overflow);
498 Py_DECREF(num);
499 if (value == -1 && PyErr_Occurred())
500 return NULL;
501 if (value != LONG_MAX)
502 return raiseTestError("test_long_and_overflow",
503 "expected return value LONG_MAX");
504 if (overflow != 0)
505 return raiseTestError("test_long_and_overflow",
506 "overflow was not cleared");
507
508 num = PyLong_FromLong(LONG_MIN);
509 if (num == NULL)
510 return NULL;
511 overflow = 0;
512 value = PyLong_AsLongAndOverflow(num, &overflow);
513 Py_DECREF(num);
514 if (value == -1 && PyErr_Occurred())
515 return NULL;
516 if (value != LONG_MIN)
517 return raiseTestError("test_long_and_overflow",
518 "expected return value LONG_MIN");
519 if (overflow != 0)
520 return raiseTestError("test_long_and_overflow",
521 "overflow was not cleared");
522
523 Py_INCREF(Py_None);
524 return Py_None;
525}
526
Mark Dickinson93f562c2010-01-30 10:30:15 +0000527/* Test the PyLong_AsLongLongAndOverflow API. General conversion to
528 PY_LONG_LONG is tested by test_long_api_inner. This test will
529 concentrate on proper handling of overflow.
530*/
531
532static PyObject *
533test_long_long_and_overflow(PyObject *self)
534{
535 PyObject *num, *one, *temp;
536 PY_LONG_LONG value;
537 int overflow;
538
539 /* Test that overflow is set properly for a large value. */
540 /* num is a number larger than PY_LLONG_MAX on a typical machine. */
541 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
542 if (num == NULL)
543 return NULL;
544 overflow = 1234;
545 value = PyLong_AsLongLongAndOverflow(num, &overflow);
546 Py_DECREF(num);
547 if (value == -1 && PyErr_Occurred())
548 return NULL;
549 if (value != -1)
550 return raiseTestError("test_long_long_and_overflow",
551 "return value was not set to -1");
552 if (overflow != 1)
553 return raiseTestError("test_long_long_and_overflow",
554 "overflow was not set to 1");
555
556 /* Same again, with num = PY_LLONG_MAX + 1 */
557 num = PyLong_FromLongLong(PY_LLONG_MAX);
558 if (num == NULL)
559 return NULL;
560 one = PyLong_FromLong(1L);
561 if (one == NULL) {
562 Py_DECREF(num);
563 return NULL;
564 }
565 temp = PyNumber_Add(num, one);
566 Py_DECREF(one);
567 Py_DECREF(num);
568 num = temp;
569 if (num == NULL)
570 return NULL;
571 overflow = 0;
572 value = PyLong_AsLongLongAndOverflow(num, &overflow);
573 Py_DECREF(num);
574 if (value == -1 && PyErr_Occurred())
575 return NULL;
576 if (value != -1)
577 return raiseTestError("test_long_long_and_overflow",
578 "return value was not set to -1");
579 if (overflow != 1)
580 return raiseTestError("test_long_long_and_overflow",
581 "overflow was not set to 1");
582
583 /* Test that overflow is set properly for a large negative value. */
584 /* num is a number smaller than PY_LLONG_MIN on a typical platform */
585 num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
586 if (num == NULL)
587 return NULL;
588 overflow = 1234;
589 value = PyLong_AsLongLongAndOverflow(num, &overflow);
590 Py_DECREF(num);
591 if (value == -1 && PyErr_Occurred())
592 return NULL;
593 if (value != -1)
594 return raiseTestError("test_long_long_and_overflow",
595 "return value was not set to -1");
596 if (overflow != -1)
597 return raiseTestError("test_long_long_and_overflow",
598 "overflow was not set to -1");
599
600 /* Same again, with num = PY_LLONG_MIN - 1 */
601 num = PyLong_FromLongLong(PY_LLONG_MIN);
602 if (num == NULL)
603 return NULL;
604 one = PyLong_FromLong(1L);
605 if (one == NULL) {
606 Py_DECREF(num);
607 return NULL;
608 }
609 temp = PyNumber_Subtract(num, one);
610 Py_DECREF(one);
611 Py_DECREF(num);
612 num = temp;
613 if (num == NULL)
614 return NULL;
615 overflow = 0;
616 value = PyLong_AsLongLongAndOverflow(num, &overflow);
617 Py_DECREF(num);
618 if (value == -1 && PyErr_Occurred())
619 return NULL;
620 if (value != -1)
621 return raiseTestError("test_long_long_and_overflow",
622 "return value was not set to -1");
623 if (overflow != -1)
624 return raiseTestError("test_long_long_and_overflow",
625 "overflow was not set to -1");
626
627 /* Test that overflow is cleared properly for small values. */
628 num = PyLong_FromString("FF", NULL, 16);
629 if (num == NULL)
630 return NULL;
631 overflow = 1234;
632 value = PyLong_AsLongLongAndOverflow(num, &overflow);
633 Py_DECREF(num);
634 if (value == -1 && PyErr_Occurred())
635 return NULL;
636 if (value != 0xFF)
637 return raiseTestError("test_long_long_and_overflow",
638 "expected return value 0xFF");
639 if (overflow != 0)
640 return raiseTestError("test_long_long_and_overflow",
641 "overflow was not cleared");
642
643 num = PyLong_FromString("-FF", NULL, 16);
644 if (num == NULL)
645 return NULL;
646 overflow = 0;
647 value = PyLong_AsLongLongAndOverflow(num, &overflow);
648 Py_DECREF(num);
649 if (value == -1 && PyErr_Occurred())
650 return NULL;
651 if (value != -0xFF)
652 return raiseTestError("test_long_long_and_overflow",
653 "expected return value 0xFF");
654 if (overflow != 0)
655 return raiseTestError("test_long_long_and_overflow",
656 "overflow was set incorrectly");
657
658 num = PyLong_FromLongLong(PY_LLONG_MAX);
659 if (num == NULL)
660 return NULL;
661 overflow = 1234;
662 value = PyLong_AsLongLongAndOverflow(num, &overflow);
663 Py_DECREF(num);
664 if (value == -1 && PyErr_Occurred())
665 return NULL;
666 if (value != PY_LLONG_MAX)
667 return raiseTestError("test_long_long_and_overflow",
668 "expected return value PY_LLONG_MAX");
669 if (overflow != 0)
670 return raiseTestError("test_long_long_and_overflow",
671 "overflow was not cleared");
672
673 num = PyLong_FromLongLong(PY_LLONG_MIN);
674 if (num == NULL)
675 return NULL;
676 overflow = 0;
677 value = PyLong_AsLongLongAndOverflow(num, &overflow);
678 Py_DECREF(num);
679 if (value == -1 && PyErr_Occurred())
680 return NULL;
681 if (value != PY_LLONG_MIN)
682 return raiseTestError("test_long_long_and_overflow",
683 "expected return value PY_LLONG_MIN");
684 if (overflow != 0)
685 return raiseTestError("test_long_long_and_overflow",
686 "overflow was not cleared");
687
688 Py_INCREF(Py_None);
689 return Py_None;
690}
691
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000692/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
Tim Petersd38b1c72001-09-30 05:09:37 +0000693 for both long and int arguments. The test may leak a little memory if
694 it fails.
695*/
696static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000697test_L_code(PyObject *self)
Tim Petersd38b1c72001-09-30 05:09:37 +0000698{
699 PyObject *tuple, *num;
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000700 PY_LONG_LONG value;
Tim Petersd38b1c72001-09-30 05:09:37 +0000701
Tim Petersd38b1c72001-09-30 05:09:37 +0000702 tuple = PyTuple_New(1);
703 if (tuple == NULL)
704 return NULL;
705
706 num = PyLong_FromLong(42);
707 if (num == NULL)
708 return NULL;
709
710 PyTuple_SET_ITEM(tuple, 0, num);
711
712 value = -1;
713 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
714 return NULL;
715 if (value != 42)
716 return raiseTestError("test_L_code",
717 "L code returned wrong value for long 42");
718
719 Py_DECREF(num);
Christian Heimes217cfd12007-12-02 14:31:20 +0000720 num = PyLong_FromLong(42);
Tim Petersd38b1c72001-09-30 05:09:37 +0000721 if (num == NULL)
722 return NULL;
723
724 PyTuple_SET_ITEM(tuple, 0, num);
725
726 value = -1;
727 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
728 return NULL;
729 if (value != 42)
730 return raiseTestError("test_L_code",
731 "L code returned wrong value for int 42");
732
733 Py_DECREF(tuple);
734 Py_INCREF(Py_None);
735 return Py_None;
736}
737
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000738#endif /* ifdef HAVE_LONG_LONG */
739
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000740/* Test tuple argument processing */
741static PyObject *
742getargs_tuple(PyObject *self, PyObject *args)
743{
744 int a, b, c;
745 if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
746 return NULL;
747 return Py_BuildValue("iii", a, b, c);
748}
749
Christian Heimes380f7f22008-02-28 11:19:05 +0000750/* test PyArg_ParseTupleAndKeywords */
751static PyObject *getargs_keywords(PyObject *self, PyObject *args, PyObject *kwargs)
752{
753 static char *keywords[] = {"arg1","arg2","arg3","arg4","arg5", NULL};
754 static char *fmt="(ii)i|(i(ii))(iii)i";
755 int int_args[10]={-1, -1, -1, -1, -1, -1, -1, -1, -1, -1};
756
757 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
758 &int_args[0], &int_args[1], &int_args[2], &int_args[3], &int_args[4],
759 &int_args[5], &int_args[6], &int_args[7], &int_args[8], &int_args[9]))
760 return NULL;
761 return Py_BuildValue("iiiiiiiiii",
762 int_args[0], int_args[1], int_args[2], int_args[3], int_args[4],
763 int_args[5], int_args[6], int_args[7], int_args[8], int_args[9]);
764}
765
Thomas Heller3457e4b2003-04-24 16:14:27 +0000766/* Functions to call PyArg_ParseTuple with integer format codes,
767 and return the result.
768*/
Thomas Hellera4ea6032003-04-17 18:55:45 +0000769static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000770getargs_b(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000771{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000772 unsigned char value;
773 if (!PyArg_ParseTuple(args, "b", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000774 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000775 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000776}
777
Thomas Heller3457e4b2003-04-24 16:14:27 +0000778static PyObject *
779getargs_B(PyObject *self, PyObject *args)
780{
781 unsigned char value;
782 if (!PyArg_ParseTuple(args, "B", &value))
783 return NULL;
784 return PyLong_FromUnsignedLong((unsigned long)value);
785}
786
787static PyObject *
Mark Dickinson1554b182009-12-20 16:03:30 +0000788getargs_h(PyObject *self, PyObject *args)
789{
790 short value;
791 if (!PyArg_ParseTuple(args, "h", &value))
792 return NULL;
793 return PyLong_FromLong((long)value);
794}
795
796static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000797getargs_H(PyObject *self, PyObject *args)
798{
799 unsigned short value;
800 if (!PyArg_ParseTuple(args, "H", &value))
801 return NULL;
802 return PyLong_FromUnsignedLong((unsigned long)value);
803}
804
805static PyObject *
806getargs_I(PyObject *self, PyObject *args)
807{
808 unsigned int value;
809 if (!PyArg_ParseTuple(args, "I", &value))
810 return NULL;
811 return PyLong_FromUnsignedLong((unsigned long)value);
812}
813
814static PyObject *
815getargs_k(PyObject *self, PyObject *args)
816{
817 unsigned long value;
818 if (!PyArg_ParseTuple(args, "k", &value))
819 return NULL;
820 return PyLong_FromUnsignedLong(value);
821}
822
823static PyObject *
824getargs_i(PyObject *self, PyObject *args)
825{
826 int value;
827 if (!PyArg_ParseTuple(args, "i", &value))
828 return NULL;
829 return PyLong_FromLong((long)value);
830}
831
Thomas Hellera4ea6032003-04-17 18:55:45 +0000832static PyObject *
833getargs_l(PyObject *self, PyObject *args)
834{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000835 long value;
836 if (!PyArg_ParseTuple(args, "l", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000837 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000838 return PyLong_FromLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000839}
840
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000841static PyObject *
842getargs_n(PyObject *self, PyObject *args)
843{
844 Py_ssize_t value;
845 if (!PyArg_ParseTuple(args, "n", &value))
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000846 return NULL;
847 return PyLong_FromSsize_t(value);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000848}
849
Thomas Hellera4ea6032003-04-17 18:55:45 +0000850#ifdef HAVE_LONG_LONG
Thomas Hellera4ea6032003-04-17 18:55:45 +0000851static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000852getargs_L(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000853{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000854 PY_LONG_LONG value;
855 if (!PyArg_ParseTuple(args, "L", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000856 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000857 return PyLong_FromLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000858}
859
Thomas Hellera4ea6032003-04-17 18:55:45 +0000860static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000861getargs_K(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000862{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000863 unsigned PY_LONG_LONG value;
864 if (!PyArg_ParseTuple(args, "K", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000865 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000866 return PyLong_FromUnsignedLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000867}
868#endif
869
870/* This function not only tests the 'k' getargs code, but also the
Christian Heimes217cfd12007-12-02 14:31:20 +0000871 PyLong_AsUnsignedLongMask() and PyLong_AsUnsignedLongMask() functions. */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000872static PyObject *
873test_k_code(PyObject *self)
874{
875 PyObject *tuple, *num;
876 unsigned long value;
877
878 tuple = PyTuple_New(1);
879 if (tuple == NULL)
880 return NULL;
881
Neal Norwitz699cbb72003-04-22 01:28:57 +0000882 /* a number larger than ULONG_MAX even on 64-bit platforms */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000883 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
884 if (num == NULL)
885 return NULL;
886
Christian Heimes217cfd12007-12-02 14:31:20 +0000887 value = PyLong_AsUnsignedLongMask(num);
Neal Norwitz699cbb72003-04-22 01:28:57 +0000888 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000889 return raiseTestError("test_k_code",
Christian Heimes217cfd12007-12-02 14:31:20 +0000890 "PyLong_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +0000891
892 PyTuple_SET_ITEM(tuple, 0, num);
893
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000894 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000895 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
896 return NULL;
Neal Norwitz699cbb72003-04-22 01:28:57 +0000897 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000898 return raiseTestError("test_k_code",
899 "k code returned wrong value for long 0xFFF...FFF");
900
901 Py_DECREF(num);
902 num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
903 if (num == NULL)
904 return NULL;
905
Christian Heimes217cfd12007-12-02 14:31:20 +0000906 value = PyLong_AsUnsignedLongMask(num);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000907 if (value != (unsigned long)-0x42)
908 return raiseTestError("test_k_code",
Christian Heimes217cfd12007-12-02 14:31:20 +0000909 "PyLong_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +0000910
911 PyTuple_SET_ITEM(tuple, 0, num);
912
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000913 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000914 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
915 return NULL;
916 if (value != (unsigned long)-0x42)
917 return raiseTestError("test_k_code",
918 "k code returned wrong value for long -0xFFF..000042");
919
920 Py_DECREF(tuple);
921 Py_INCREF(Py_None);
922 return Py_None;
923}
924
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000925
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +0000926/* Test the s and z codes for PyArg_ParseTuple.
927*/
928static PyObject *
929test_s_code(PyObject *self)
930{
931 /* Unicode strings should be accepted */
932 PyObject *tuple, *obj;
933 char *value;
934
935 tuple = PyTuple_New(1);
936 if (tuple == NULL)
937 return NULL;
938
939 obj = PyUnicode_Decode("t\xeate", strlen("t\xeate"),
940 "latin-1", NULL);
941 if (obj == NULL)
942 return NULL;
943
944 PyTuple_SET_ITEM(tuple, 0, obj);
945
946 /* These two blocks used to raise a TypeError:
947 * "argument must be string without null bytes, not str"
948 */
949 if (PyArg_ParseTuple(tuple, "s:test_s_code1", &value) < 0)
950 return NULL;
951
952 if (PyArg_ParseTuple(tuple, "z:test_s_code2", &value) < 0)
953 return NULL;
954
Alexandre Vassalottib645bc72008-05-15 22:06:59 +0000955 Py_DECREF(tuple);
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +0000956 Py_RETURN_NONE;
957}
958
Mark Dickinsonf08173b2009-12-03 10:59:46 +0000959static PyObject *
960test_bug_7414(PyObject *self)
961{
962 /* Issue #7414: for PyArg_ParseTupleAndKeywords, 'C' code wasn't being
963 skipped properly in skipitem() */
964 int a = 0, b = 0, result;
965 char *kwlist[] = {"a", "b", NULL};
966 PyObject *tuple = NULL, *dict = NULL, *b_str;
967
968 tuple = PyTuple_New(0);
969 if (tuple == NULL)
970 goto failure;
971 dict = PyDict_New();
972 if (dict == NULL)
973 goto failure;
974 b_str = PyUnicode_FromString("b");
975 if (b_str == NULL)
976 goto failure;
977 result = PyDict_SetItemString(dict, "b", b_str);
978 Py_DECREF(b_str);
979 if (result < 0)
980 goto failure;
981
982 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|CC",
983 kwlist, &a, &b);
984 if (!result)
985 goto failure;
986
987 if (a != 0)
988 return raiseTestError("test_bug_7414",
989 "C format code not skipped properly");
990 if (b != 'b')
991 return raiseTestError("test_bug_7414",
992 "C format code returned wrong value");
993
994 Py_DECREF(dict);
995 Py_DECREF(tuple);
996 Py_RETURN_NONE;
997
998 failure:
999 Py_XDECREF(dict);
1000 Py_XDECREF(tuple);
1001 return NULL;
1002}
1003
1004
Benjamin Peterson92035012008-12-27 16:00:54 +00001005static volatile int x;
1006
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001007/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
1008 of an error.
1009*/
1010static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +00001011test_u_code(PyObject *self)
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001012{
1013 PyObject *tuple, *obj;
1014 Py_UNICODE *value;
Neal Norwitz8866e0a2007-10-27 04:01:17 +00001015 Py_ssize_t len;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001016
Benjamin Peterson206e3072008-10-19 14:07:49 +00001017 /* issue4122: Undefined reference to _Py_ascii_whitespace on Windows */
1018 /* Just use the macro and check that it compiles */
Christian Heimes727c2c52008-11-12 09:04:04 +00001019 x = Py_UNICODE_ISSPACE(25);
Benjamin Peterson206e3072008-10-19 14:07:49 +00001020
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001021 tuple = PyTuple_New(1);
1022 if (tuple == NULL)
1023 return NULL;
1024
1025 obj = PyUnicode_Decode("test", strlen("test"),
1026 "ascii", NULL);
1027 if (obj == NULL)
1028 return NULL;
1029
1030 PyTuple_SET_ITEM(tuple, 0, obj);
1031
1032 value = 0;
1033 if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
1034 return NULL;
1035 if (value != PyUnicode_AS_UNICODE(obj))
1036 return raiseTestError("test_u_code",
1037 "u code returned wrong value for u'test'");
1038 value = 0;
1039 if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
1040 return NULL;
1041 if (value != PyUnicode_AS_UNICODE(obj) ||
1042 len != PyUnicode_GET_SIZE(obj))
1043 return raiseTestError("test_u_code",
1044 "u# code returned wrong values for u'test'");
Tim Petersbaefd9e2003-01-28 20:37:45 +00001045
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001046 Py_DECREF(tuple);
1047 Py_INCREF(Py_None);
1048 return Py_None;
1049}
1050
Guido van Rossumfb67be22007-08-29 18:38:11 +00001051/* Test Z and Z# codes for PyArg_ParseTuple */
1052static PyObject *
1053test_Z_code(PyObject *self)
1054{
1055 PyObject *tuple, *obj;
1056 Py_UNICODE *value1, *value2;
Neal Norwitz8866e0a2007-10-27 04:01:17 +00001057 Py_ssize_t len1, len2;
Guido van Rossumfb67be22007-08-29 18:38:11 +00001058
1059 tuple = PyTuple_New(2);
1060 if (tuple == NULL)
1061 return NULL;
1062
1063 obj = PyUnicode_FromString("test");
1064 PyTuple_SET_ITEM(tuple, 0, obj);
1065 Py_INCREF(Py_None);
1066 PyTuple_SET_ITEM(tuple, 1, Py_None);
1067
1068 /* swap values on purpose */
1069 value1 = NULL;
1070 value2 = PyUnicode_AS_UNICODE(obj);
1071
1072 /* Test Z for both values */
1073 if (PyArg_ParseTuple(tuple, "ZZ:test_Z_code", &value1, &value2) < 0)
1074 return NULL;
1075 if (value1 != PyUnicode_AS_UNICODE(obj))
1076 return raiseTestError("test_Z_code",
1077 "Z code returned wrong value for 'test'");
1078 if (value2 != NULL)
1079 return raiseTestError("test_Z_code",
1080 "Z code returned wrong value for None");
1081
1082 value1 = NULL;
1083 value2 = PyUnicode_AS_UNICODE(obj);
1084 len1 = -1;
1085 len2 = -1;
1086
1087 /* Test Z# for both values */
1088 if (PyArg_ParseTuple(tuple, "Z#Z#:test_Z_code", &value1, &len1,
1089 &value2, &len2) < 0)
1090 return NULL;
1091 if (value1 != PyUnicode_AS_UNICODE(obj) ||
1092 len1 != PyUnicode_GET_SIZE(obj))
1093 return raiseTestError("test_Z_code",
1094 "Z# code returned wrong values for 'test'");
1095 if (value2 != NULL ||
1096 len2 != 0)
1097 return raiseTestError("test_Z_code",
1098 "Z# code returned wrong values for None'");
1099
1100 Py_DECREF(tuple);
1101 Py_RETURN_NONE;
1102}
1103
Thomas Wouters477c8d52006-05-27 19:21:47 +00001104static PyObject *
Mark Dickinson081dfee2009-03-18 14:47:41 +00001105test_widechar(PyObject *self)
1106{
1107#if defined(SIZEOF_WCHAR_T) && (SIZEOF_WCHAR_T == 4)
1108 const wchar_t wtext[2] = {(wchar_t)0x10ABCDu};
1109 size_t wtextlen = 1;
1110#else
1111 const wchar_t wtext[3] = {(wchar_t)0xDBEAu, (wchar_t)0xDFCDu};
1112 size_t wtextlen = 2;
1113#endif
1114 PyObject *wide, *utf8;
1115
1116 wide = PyUnicode_FromWideChar(wtext, wtextlen);
1117 if (wide == NULL)
1118 return NULL;
1119
1120 utf8 = PyUnicode_FromString("\xf4\x8a\xaf\x8d");
1121 if (utf8 == NULL) {
1122 Py_DECREF(wide);
1123 return NULL;
1124 }
1125
1126 if (PyUnicode_GET_SIZE(wide) != PyUnicode_GET_SIZE(utf8)) {
1127 Py_DECREF(wide);
1128 Py_DECREF(utf8);
1129 return raiseTestError("test_widechar",
1130 "wide string and utf8 string "
1131 "have different length");
1132 }
1133 if (PyUnicode_Compare(wide, utf8)) {
1134 Py_DECREF(wide);
1135 Py_DECREF(utf8);
1136 if (PyErr_Occurred())
1137 return NULL;
1138 return raiseTestError("test_widechar",
1139 "wide string and utf8 string "
1140 "are different");
1141 }
1142
1143 Py_DECREF(wide);
1144 Py_DECREF(utf8);
1145 Py_RETURN_NONE;
1146}
1147
1148static PyObject *
Benjamin Peterson92035012008-12-27 16:00:54 +00001149test_empty_argparse(PyObject *self)
1150{
1151 /* Test that formats can begin with '|'. See issue #4720. */
1152 PyObject *tuple, *dict = NULL;
1153 static char *kwlist[] = {NULL};
1154 int result;
1155 tuple = PyTuple_New(0);
1156 if (!tuple)
1157 return NULL;
1158 if ((result = PyArg_ParseTuple(tuple, "|:test_empty_argparse")) < 0)
1159 goto done;
1160 dict = PyDict_New();
1161 if (!dict)
1162 goto done;
1163 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|:test_empty_argparse", kwlist);
1164 done:
1165 Py_DECREF(tuple);
1166 Py_XDECREF(dict);
1167 if (result < 0)
1168 return NULL;
1169 else {
1170 Py_RETURN_NONE;
1171 }
1172}
1173
1174static PyObject *
Thomas Wouters477c8d52006-05-27 19:21:47 +00001175codec_incrementalencoder(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001176{
1177 const char *encoding, *errors = NULL;
1178 if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
1179 &encoding, &errors))
1180 return NULL;
1181 return PyCodec_IncrementalEncoder(encoding, errors);
1182}
1183
Thomas Wouters477c8d52006-05-27 19:21:47 +00001184static PyObject *
1185codec_incrementaldecoder(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001186{
1187 const char *encoding, *errors = NULL;
1188 if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
1189 &encoding, &errors))
1190 return NULL;
1191 return PyCodec_IncrementalDecoder(encoding, errors);
1192}
1193
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001194
Tim Peters5b8132f2003-01-31 15:52:05 +00001195/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
Tim Petersbaefd9e2003-01-28 20:37:45 +00001196static PyObject *
1197test_long_numbits(PyObject *self)
1198{
Tim Peters5b8132f2003-01-31 15:52:05 +00001199 struct triple {
Tim Petersbaefd9e2003-01-28 20:37:45 +00001200 long input;
Tim Peters5b8132f2003-01-31 15:52:05 +00001201 size_t nbits;
1202 int sign;
1203 } testcases[] = {{0, 0, 0},
1204 {1L, 1, 1},
1205 {-1L, 1, -1},
1206 {2L, 2, 1},
1207 {-2L, 2, -1},
1208 {3L, 2, 1},
1209 {-3L, 2, -1},
1210 {4L, 3, 1},
1211 {-4L, 3, -1},
1212 {0x7fffL, 15, 1}, /* one Python long digit */
1213 {-0x7fffL, 15, -1},
1214 {0xffffL, 16, 1},
1215 {-0xffffL, 16, -1},
1216 {0xfffffffL, 28, 1},
1217 {-0xfffffffL, 28, -1}};
Tim Petersbaefd9e2003-01-28 20:37:45 +00001218 int i;
1219
Tim Peters5b8132f2003-01-31 15:52:05 +00001220 for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
1221 PyObject *plong = PyLong_FromLong(testcases[i].input);
Tim Petersbaefd9e2003-01-28 20:37:45 +00001222 size_t nbits = _PyLong_NumBits(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +00001223 int sign = _PyLong_Sign(plong);
Tim Petersbaefd9e2003-01-28 20:37:45 +00001224
1225 Py_DECREF(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +00001226 if (nbits != testcases[i].nbits)
Tim Petersbaefd9e2003-01-28 20:37:45 +00001227 return raiseTestError("test_long_numbits",
Tim Peters5b8132f2003-01-31 15:52:05 +00001228 "wrong result for _PyLong_NumBits");
1229 if (sign != testcases[i].sign)
1230 return raiseTestError("test_long_numbits",
1231 "wrong result for _PyLong_Sign");
Tim Petersbaefd9e2003-01-28 20:37:45 +00001232 }
1233 Py_INCREF(Py_None);
1234 return Py_None;
1235}
1236
Thomas Heller519a0422007-11-15 20:48:54 +00001237/* Example passing NULLs to PyObject_Str(NULL). */
Thomas Woutersa9773292006-04-21 09:43:23 +00001238
1239static PyObject *
1240test_null_strings(PyObject *self)
1241{
Thomas Heller519a0422007-11-15 20:48:54 +00001242 PyObject *o1 = PyObject_Str(NULL), *o2 = PyObject_Str(NULL);
Thomas Woutersa9773292006-04-21 09:43:23 +00001243 PyObject *tuple = PyTuple_Pack(2, o1, o2);
1244 Py_XDECREF(o1);
1245 Py_XDECREF(o2);
1246 return tuple;
1247}
1248
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001249static PyObject *
1250raise_exception(PyObject *self, PyObject *args)
1251{
1252 PyObject *exc;
1253 PyObject *exc_args, *v;
1254 int num_args, i;
1255
1256 if (!PyArg_ParseTuple(args, "Oi:raise_exception",
1257 &exc, &num_args))
1258 return NULL;
1259
1260 exc_args = PyTuple_New(num_args);
1261 if (exc_args == NULL)
1262 return NULL;
1263 for (i = 0; i < num_args; ++i) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001264 v = PyLong_FromLong(i);
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001265 if (v == NULL) {
1266 Py_DECREF(exc_args);
1267 return NULL;
1268 }
1269 PyTuple_SET_ITEM(exc_args, i, v);
1270 }
1271 PyErr_SetObject(exc, exc_args);
Michael W. Hudsonf02bcee2003-08-15 13:03:30 +00001272 Py_DECREF(exc_args);
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001273 return NULL;
1274}
Tim Peters91621db2001-06-12 20:10:01 +00001275
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001276#ifdef WITH_THREAD
1277
Benjamin Peterson91d58bd2009-12-13 21:30:54 +00001278static PyObject *
1279test_datetime_capi(PyObject *self, PyObject *args) {
1280 if (PyDateTimeAPI) {
1281 PyErr_SetString(PyExc_AssertionError,
1282 "PyDateTime_CAPI somehow initialized");
1283 return NULL;
1284 }
1285 PyDateTime_IMPORT;
1286 if (PyDateTimeAPI)
1287 Py_RETURN_NONE;
1288 else
1289 return NULL;
1290}
1291
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001292/* test_thread_state spawns a thread of its own, and that thread releases
1293 * `thread_done` when it's finished. The driver code has to know when the
1294 * thread finishes, because the thread uses a PyObject (the callable) that
1295 * may go away when the driver finishes. The former lack of this explicit
1296 * synchronization caused rare segfaults, so rare that they were seen only
1297 * on a Mac buildbot (although they were possible on any box).
1298 */
1299static PyThread_type_lock thread_done = NULL;
1300
Benjamin Petersona786b022008-08-25 21:05:21 +00001301static int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001302_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001303{
1304 PyObject *rc;
Benjamin Petersona786b022008-08-25 21:05:21 +00001305 int success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001306 PyGILState_STATE s = PyGILState_Ensure();
Thomas Wouters89f507f2006-12-13 04:49:30 +00001307 rc = PyObject_CallFunction((PyObject *)callable, "");
Benjamin Petersona786b022008-08-25 21:05:21 +00001308 success = (rc != NULL);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001309 Py_XDECREF(rc);
1310 PyGILState_Release(s);
Benjamin Petersona786b022008-08-25 21:05:21 +00001311 return success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001312}
1313
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001314/* Same thing, but releases `thread_done` when it returns. This variant
1315 * should be called only from threads spawned by test_thread_state().
1316 */
1317static void
1318_make_call_from_thread(void *callable)
1319{
1320 _make_call(callable);
1321 PyThread_release_lock(thread_done);
1322}
1323
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001324static PyObject *
1325test_thread_state(PyObject *self, PyObject *args)
1326{
1327 PyObject *fn;
Benjamin Petersona786b022008-08-25 21:05:21 +00001328 int success = 1;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001329
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001330 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
1331 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001332
Benjamin Petersona786b022008-08-25 21:05:21 +00001333 if (!PyCallable_Check(fn)) {
1334 PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
1335 fn->ob_type->tp_name);
1336 return NULL;
1337 }
1338
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001339 /* Ensure Python is set up for threading */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001340 PyEval_InitThreads();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001341 thread_done = PyThread_allocate_lock();
1342 if (thread_done == NULL)
1343 return PyErr_NoMemory();
1344 PyThread_acquire_lock(thread_done, 1);
1345
1346 /* Start a new thread with our callback. */
1347 PyThread_start_new_thread(_make_call_from_thread, fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001348 /* Make the callback with the thread lock held by this thread */
Benjamin Petersona786b022008-08-25 21:05:21 +00001349 success &= _make_call(fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001350 /* Do it all again, but this time with the thread-lock released */
1351 Py_BEGIN_ALLOW_THREADS
Benjamin Petersona786b022008-08-25 21:05:21 +00001352 success &= _make_call(fn);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001353 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001354 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001355
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001356 /* And once more with and without a thread
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001357 XXX - should use a lock and work out exactly what we are trying
1358 to test <wink>
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001359 */
1360 Py_BEGIN_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001361 PyThread_start_new_thread(_make_call_from_thread, fn);
Benjamin Petersona786b022008-08-25 21:05:21 +00001362 success &= _make_call(fn);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001363 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001364 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001365
Thomas Wouters89f507f2006-12-13 04:49:30 +00001366 /* Release lock we acquired above. This is required on HP-UX. */
1367 PyThread_release_lock(thread_done);
1368
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001369 PyThread_free_lock(thread_done);
Benjamin Petersona786b022008-08-25 21:05:21 +00001370 if (!success)
1371 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001372 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001373}
Benjamin Petersona54c9092009-01-13 02:11:23 +00001374
1375/* test Py_AddPendingCalls using threads */
1376static int _pending_callback(void *arg)
1377{
1378 /* we assume the argument is callable object to which we own a reference */
1379 PyObject *callable = (PyObject *)arg;
1380 PyObject *r = PyObject_CallObject(callable, NULL);
1381 Py_DECREF(callable);
1382 Py_XDECREF(r);
1383 return r != NULL ? 0 : -1;
1384}
1385
1386/* The following requests n callbacks to _pending_callback. It can be
1387 * run from any python thread.
1388 */
1389PyObject *pending_threadfunc(PyObject *self, PyObject *arg)
1390{
1391 PyObject *callable;
1392 int r;
1393 if (PyArg_ParseTuple(arg, "O", &callable) == 0)
1394 return NULL;
1395
1396 /* create the reference for the callbackwhile we hold the lock */
1397 Py_INCREF(callable);
1398
1399 Py_BEGIN_ALLOW_THREADS
1400 r = Py_AddPendingCall(&_pending_callback, callable);
1401 Py_END_ALLOW_THREADS
1402
1403 if (r<0) {
1404 Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
1405 Py_INCREF(Py_False);
1406 return Py_False;
1407 }
1408 Py_INCREF(Py_True);
1409 return Py_True;
1410}
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001411#endif
1412
Neal Norwitzb0d26332007-08-25 00:49:05 +00001413/* Some tests of PyUnicode_FromFormat(). This needs more tests. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00001414static PyObject *
1415test_string_from_format(PyObject *self, PyObject *args)
1416{
1417 PyObject *result;
1418 char *msg;
1419
1420#define CHECK_1_FORMAT(FORMAT, TYPE) \
Neal Norwitzb0d26332007-08-25 00:49:05 +00001421 result = PyUnicode_FromFormat(FORMAT, (TYPE)1); \
Thomas Wouters477c8d52006-05-27 19:21:47 +00001422 if (result == NULL) \
1423 return NULL; \
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00001424 if (strcmp(_PyUnicode_AsString(result), "1")) { \
Thomas Wouters477c8d52006-05-27 19:21:47 +00001425 msg = FORMAT " failed at 1"; \
1426 goto Fail; \
1427 } \
1428 Py_DECREF(result)
1429
1430 CHECK_1_FORMAT("%d", int);
1431 CHECK_1_FORMAT("%ld", long);
1432 /* The z width modifier was added in Python 2.5. */
1433 CHECK_1_FORMAT("%zd", Py_ssize_t);
1434
1435 /* The u type code was added in Python 2.5. */
1436 CHECK_1_FORMAT("%u", unsigned int);
1437 CHECK_1_FORMAT("%lu", unsigned long);
1438 CHECK_1_FORMAT("%zu", size_t);
1439
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001440 /* "%lld" and "%llu" support added in Python 2.7. */
1441#ifdef HAVE_LONG_LONG
1442 CHECK_1_FORMAT("%llu", unsigned PY_LONG_LONG);
1443 CHECK_1_FORMAT("%lld", PY_LONG_LONG);
1444#endif
1445
Thomas Wouters477c8d52006-05-27 19:21:47 +00001446 Py_RETURN_NONE;
1447
1448 Fail:
1449 Py_XDECREF(result);
1450 return raiseTestError("test_string_from_format", msg);
1451
1452#undef CHECK_1_FORMAT
1453}
1454
Benjamin Peterson8667a9b2010-01-09 21:45:28 +00001455
1456static PyObject *
1457test_unicode_compare_with_ascii(PyObject *self) {
1458 PyObject *py_s = PyUnicode_FromStringAndSize("str\0", 4);
1459 int result;
1460 if (py_s == NULL)
1461 return NULL;
1462 result = PyUnicode_CompareWithASCIIString(py_s, "str");
1463 Py_DECREF(py_s);
1464 if (!result) {
1465 PyErr_SetString(TestError, "Python string ending in NULL "
1466 "should not compare equal to c string.");
1467 return NULL;
1468 }
1469 Py_RETURN_NONE;
1470};
1471
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001472/* This is here to provide a docstring for test_descr. */
1473static PyObject *
1474test_with_docstring(PyObject *self)
1475{
1476 Py_RETURN_NONE;
1477}
1478
Mark Dickinson725bfd82009-05-03 20:33:40 +00001479/* Test PyOS_string_to_double. */
1480static PyObject *
1481test_string_to_double(PyObject *self) {
1482 double result;
1483 char *msg;
1484
1485#define CHECK_STRING(STR, expected) \
1486 result = PyOS_string_to_double(STR, NULL, NULL); \
1487 if (result == -1.0 && PyErr_Occurred()) \
1488 return NULL; \
1489 if (result != expected) { \
1490 msg = "conversion of " STR " to float failed"; \
1491 goto fail; \
1492 }
1493
1494#define CHECK_INVALID(STR) \
1495 result = PyOS_string_to_double(STR, NULL, NULL); \
1496 if (result == -1.0 && PyErr_Occurred()) { \
1497 if (PyErr_ExceptionMatches(PyExc_ValueError)) \
1498 PyErr_Clear(); \
1499 else \
1500 return NULL; \
1501 } \
1502 else { \
1503 msg = "conversion of " STR " didn't raise ValueError"; \
1504 goto fail; \
1505 }
1506
1507 CHECK_STRING("0.1", 0.1);
1508 CHECK_STRING("1.234", 1.234);
1509 CHECK_STRING("-1.35", -1.35);
1510 CHECK_STRING(".1e01", 1.0);
1511 CHECK_STRING("2.e-2", 0.02);
1512
1513 CHECK_INVALID(" 0.1");
1514 CHECK_INVALID("\t\n-3");
1515 CHECK_INVALID(".123 ");
1516 CHECK_INVALID("3\n");
1517 CHECK_INVALID("123abc");
1518
1519 Py_RETURN_NONE;
1520 fail:
1521 return raiseTestError("test_string_to_double", msg);
1522#undef CHECK_STRING
1523#undef CHECK_INVALID
1524}
1525
1526
Benjamin Petersonb173f782009-05-05 22:31:58 +00001527/* Coverage testing of capsule objects. */
1528
1529static const char *capsule_name = "capsule name";
1530static char *capsule_pointer = "capsule pointer";
1531static char *capsule_context = "capsule context";
1532static const char *capsule_error = NULL;
1533static int
1534capsule_destructor_call_count = 0;
1535
1536static void
1537capsule_destructor(PyObject *o) {
1538 capsule_destructor_call_count++;
1539 if (PyCapsule_GetContext(o) != capsule_context) {
1540 capsule_error = "context did not match in destructor!";
1541 } else if (PyCapsule_GetDestructor(o) != capsule_destructor) {
1542 capsule_error = "destructor did not match in destructor! (woah!)";
1543 } else if (PyCapsule_GetName(o) != capsule_name) {
1544 capsule_error = "name did not match in destructor!";
1545 } else if (PyCapsule_GetPointer(o, capsule_name) != capsule_pointer) {
1546 capsule_error = "pointer did not match in destructor!";
1547 }
1548}
1549
1550typedef struct {
1551 char *name;
1552 char *module;
1553 char *attribute;
1554} known_capsule;
1555
1556static PyObject *
1557test_capsule(PyObject *self, PyObject *args)
1558{
1559 PyObject *object;
1560 const char *error = NULL;
1561 void *pointer;
1562 void *pointer2;
1563 known_capsule known_capsules[] = {
1564 #define KNOWN_CAPSULE(module, name) { module "." name, module, name }
1565 KNOWN_CAPSULE("_socket", "CAPI"),
1566 KNOWN_CAPSULE("_curses", "_C_API"),
1567 KNOWN_CAPSULE("datetime", "datetime_CAPI"),
1568 { NULL, NULL },
1569 };
1570 known_capsule *known = &known_capsules[0];
1571
1572#define FAIL(x) { error = (x); goto exit; }
1573
1574#define CHECK_DESTRUCTOR \
1575 if (capsule_error) { \
1576 FAIL(capsule_error); \
1577 } \
1578 else if (!capsule_destructor_call_count) { \
1579 FAIL("destructor not called!"); \
1580 } \
1581 capsule_destructor_call_count = 0; \
1582
1583 object = PyCapsule_New(capsule_pointer, capsule_name, capsule_destructor);
1584 PyCapsule_SetContext(object, capsule_context);
1585 capsule_destructor(object);
1586 CHECK_DESTRUCTOR;
1587 Py_DECREF(object);
1588 CHECK_DESTRUCTOR;
1589
1590 object = PyCapsule_New(known, "ignored", NULL);
1591 PyCapsule_SetPointer(object, capsule_pointer);
1592 PyCapsule_SetName(object, capsule_name);
1593 PyCapsule_SetDestructor(object, capsule_destructor);
1594 PyCapsule_SetContext(object, capsule_context);
1595 capsule_destructor(object);
1596 CHECK_DESTRUCTOR;
1597 /* intentionally access using the wrong name */
1598 pointer2 = PyCapsule_GetPointer(object, "the wrong name");
1599 if (!PyErr_Occurred()) {
1600 FAIL("PyCapsule_GetPointer should have failed but did not!");
1601 }
1602 PyErr_Clear();
1603 if (pointer2) {
1604 if (pointer2 == capsule_pointer) {
1605 FAIL("PyCapsule_GetPointer should not have"
1606 " returned the internal pointer!");
1607 } else {
1608 FAIL("PyCapsule_GetPointer should have "
1609 "returned NULL pointer but did not!");
1610 }
1611 }
1612 PyCapsule_SetDestructor(object, NULL);
1613 Py_DECREF(object);
1614 if (capsule_destructor_call_count) {
1615 FAIL("destructor called when it should not have been!");
1616 }
1617
1618 for (known = &known_capsules[0]; known->module != NULL; known++) {
1619 /* yeah, ordinarily I wouldn't do this either,
1620 but it's fine for this test harness.
1621 */
1622 static char buffer[256];
1623#undef FAIL
1624#define FAIL(x) \
1625 { \
1626 sprintf(buffer, "%s module: \"%s\" attribute: \"%s\"", \
1627 x, known->module, known->attribute); \
1628 error = buffer; \
1629 goto exit; \
1630 } \
1631
1632 PyObject *module = PyImport_ImportModule(known->module);
1633 if (module) {
1634 pointer = PyCapsule_Import(known->name, 0);
1635 if (!pointer) {
1636 Py_DECREF(module);
1637 FAIL("PyCapsule_GetPointer returned NULL unexpectedly!");
1638 }
1639 object = PyObject_GetAttrString(module, known->attribute);
1640 if (!object) {
1641 Py_DECREF(module);
1642 return NULL;
1643 }
1644 pointer2 = PyCapsule_GetPointer(object,
1645 "weebles wobble but they don't fall down");
1646 if (!PyErr_Occurred()) {
1647 Py_DECREF(object);
1648 Py_DECREF(module);
1649 FAIL("PyCapsule_GetPointer should have failed but did not!");
1650 }
1651 PyErr_Clear();
1652 if (pointer2) {
1653 Py_DECREF(module);
1654 Py_DECREF(object);
1655 if (pointer2 == pointer) {
1656 FAIL("PyCapsule_GetPointer should not have"
1657 " returned its internal pointer!");
1658 } else {
1659 FAIL("PyCapsule_GetPointer should have"
1660 " returned NULL pointer but did not!");
1661 }
1662 }
1663 Py_DECREF(object);
1664 Py_DECREF(module);
1665 }
1666 }
1667
1668 exit:
1669 if (error) {
1670 return raiseTestError("test_capsule", error);
1671 }
Mark Dickinson6d138f12009-12-13 20:03:21 +00001672 /* 13/12/2009: something is causing test_capi to fail occasionally on
1673 the Solaris buildbot, with the output:
1674
1675 internal test_L_code
1676 internal test_Z_code
1677 internal test_bug_7414
1678 internal test_capsule
1679 XXX undetected error
1680 internaltest test_capi crashed -- <class 'ImportError'>: No module named datetime
1681
1682 It seems possible that test_capsule is raising an exception but
1683 failing to return NULL. Do a PyErr_Occurred check to find out.
1684 */
1685 if (PyErr_Occurred())
1686 return NULL;
Benjamin Petersonb173f782009-05-05 22:31:58 +00001687 Py_RETURN_NONE;
1688#undef FAIL
1689}
1690
Guido van Rossumddefaf32007-01-14 03:31:43 +00001691#ifdef HAVE_GETTIMEOFDAY
1692/* Profiling of integer performance */
Martin v. Löwis1c951552008-06-13 07:48:19 +00001693static void print_delta(int test, struct timeval *s, struct timeval *e)
Guido van Rossumddefaf32007-01-14 03:31:43 +00001694{
1695 e->tv_sec -= s->tv_sec;
1696 e->tv_usec -= s->tv_usec;
1697 if (e->tv_usec < 0) {
1698 e->tv_sec -=1;
1699 e->tv_usec += 1000000;
1700 }
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001701 printf("Test %d: %d.%06ds\n", test, (int)e->tv_sec, (int)e->tv_usec);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001702}
1703
1704static PyObject *
1705profile_int(PyObject *self, PyObject* args)
1706{
1707 int i, k;
1708 struct timeval start, stop;
1709 PyObject *single, **multiple, *op1, *result;
1710
1711 /* Test 1: Allocate and immediately deallocate
1712 many small integers */
1713 gettimeofday(&start, NULL);
1714 for(k=0; k < 20000; k++)
1715 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001716 single = PyLong_FromLong(i);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001717 Py_DECREF(single);
1718 }
1719 gettimeofday(&stop, NULL);
1720 print_delta(1, &start, &stop);
1721
1722 /* Test 2: Allocate and immediately deallocate
1723 many large integers */
1724 gettimeofday(&start, NULL);
1725 for(k=0; k < 20000; k++)
1726 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001727 single = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001728 Py_DECREF(single);
1729 }
1730 gettimeofday(&stop, NULL);
1731 print_delta(2, &start, &stop);
1732
1733 /* Test 3: Allocate a few integers, then release
1734 them all simultaneously. */
1735 multiple = malloc(sizeof(PyObject*) * 1000);
1736 gettimeofday(&start, NULL);
1737 for(k=0; k < 20000; k++) {
1738 for(i=0; i < 1000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001739 multiple[i] = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001740 }
1741 for(i=0; i < 1000; i++) {
1742 Py_DECREF(multiple[i]);
1743 }
1744 }
1745 gettimeofday(&stop, NULL);
1746 print_delta(3, &start, &stop);
1747
1748 /* Test 4: Allocate many integers, then release
1749 them all simultaneously. */
1750 multiple = malloc(sizeof(PyObject*) * 1000000);
1751 gettimeofday(&start, NULL);
1752 for(k=0; k < 20; k++) {
1753 for(i=0; i < 1000000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001754 multiple[i] = PyLong_FromLong(i+1000000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001755 }
1756 for(i=0; i < 1000000; i++) {
1757 Py_DECREF(multiple[i]);
1758 }
1759 }
1760 gettimeofday(&stop, NULL);
1761 print_delta(4, &start, &stop);
1762
1763 /* Test 5: Allocate many integers < 32000 */
1764 multiple = malloc(sizeof(PyObject*) * 1000000);
1765 gettimeofday(&start, NULL);
1766 for(k=0; k < 10; k++) {
1767 for(i=0; i < 1000000; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00001768 multiple[i] = PyLong_FromLong(i+1000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001769 }
1770 for(i=0; i < 1000000; i++) {
1771 Py_DECREF(multiple[i]);
1772 }
1773 }
1774 gettimeofday(&stop, NULL);
1775 print_delta(5, &start, &stop);
1776
1777 /* Test 6: Perform small int addition */
Christian Heimes217cfd12007-12-02 14:31:20 +00001778 op1 = PyLong_FromLong(1);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001779 gettimeofday(&start, NULL);
1780 for(i=0; i < 10000000; i++) {
1781 result = PyNumber_Add(op1, op1);
1782 Py_DECREF(result);
1783 }
1784 gettimeofday(&stop, NULL);
1785 Py_DECREF(op1);
1786 print_delta(6, &start, &stop);
1787
1788 /* Test 7: Perform medium int addition */
Christian Heimes217cfd12007-12-02 14:31:20 +00001789 op1 = PyLong_FromLong(1000);
Guido van Rossumddefaf32007-01-14 03:31:43 +00001790 gettimeofday(&start, NULL);
1791 for(i=0; i < 10000000; i++) {
1792 result = PyNumber_Add(op1, op1);
1793 Py_DECREF(result);
1794 }
1795 gettimeofday(&stop, NULL);
1796 Py_DECREF(op1);
1797 print_delta(7, &start, &stop);
1798
1799 Py_INCREF(Py_None);
1800 return Py_None;
1801}
1802#endif
1803
Christian Heimes81ee3ef2008-05-04 22:42:01 +00001804/* To test the format of tracebacks as printed out. */
1805static PyObject *
1806traceback_print(PyObject *self, PyObject *args)
1807{
1808 PyObject *file;
1809 PyObject *traceback;
1810 int result;
1811
1812 if (!PyArg_ParseTuple(args, "OO:traceback_print",
1813 &traceback, &file))
1814 return NULL;
1815
1816 result = PyTraceBack_Print(traceback, file);
1817 if (result < 0)
1818 return NULL;
1819 Py_RETURN_NONE;
1820}
1821
Benjamin Petersone6528212008-07-15 15:32:09 +00001822/* To test the format of exceptions as printed out. */
1823static PyObject *
1824exception_print(PyObject *self, PyObject *args)
1825{
1826 PyObject *value;
1827 PyObject *tb;
1828
1829 if (!PyArg_ParseTuple(args, "O:exception_print",
1830 &value))
1831 return NULL;
Benjamin Peterson6784eb72008-08-23 20:32:27 +00001832 if (!PyExceptionInstance_Check(value)) {
1833 PyErr_Format(PyExc_TypeError, "an exception instance is required");
1834 return NULL;
1835 }
Benjamin Petersone6528212008-07-15 15:32:09 +00001836
1837 tb = PyException_GetTraceback(value);
1838 PyErr_Display((PyObject *) Py_TYPE(value), value, tb);
1839 Py_XDECREF(tb);
1840
1841 Py_RETURN_NONE;
1842}
1843
1844
1845
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001846
1847/* reliably raise a MemoryError */
1848static PyObject *
1849raise_memoryerror(PyObject *self)
1850{
1851 PyErr_NoMemory();
1852 return NULL;
1853}
1854
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00001855/* Issue 6012 */
1856static PyObject *str1, *str2;
1857static int
1858failing_converter(PyObject *obj, void *arg)
1859{
1860 /* Clone str1, then let the conversion fail. */
1861 assert(str1);
1862 str2 = str1;
1863 Py_INCREF(str2);
1864 return 0;
1865}
1866static PyObject*
1867argparsing(PyObject *o, PyObject *args)
1868{
1869 PyObject *res;
1870 str1 = str2 = NULL;
1871 if (!PyArg_ParseTuple(args, "O&O&",
1872 PyUnicode_FSConverter, &str1,
1873 failing_converter, &str2)) {
1874 if (!str2)
1875 /* argument converter not called? */
1876 return NULL;
1877 /* Should be 1 */
1878 res = PyLong_FromLong(Py_REFCNT(str2));
1879 Py_DECREF(str2);
1880 PyErr_Clear();
1881 return res;
1882 }
1883 Py_RETURN_NONE;
1884}
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001885
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00001886/* To test that the result of PyCode_NewEmpty has the right members. */
1887static PyObject *
1888code_newempty(PyObject *self, PyObject *args)
1889{
1890 const char *filename;
1891 const char *funcname;
1892 int firstlineno;
1893
1894 if (!PyArg_ParseTuple(args, "ssi:code_newempty",
1895 &filename, &funcname, &firstlineno))
1896 return NULL;
1897
1898 return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
1899}
1900
Georg Brandl1e28a272009-12-28 08:41:01 +00001901/* Test PyErr_NewExceptionWithDoc (also exercise PyErr_NewException).
1902 Run via Lib/test/test_exceptions.py */
1903static PyObject *
1904make_exception_with_doc(PyObject *self, PyObject *args, PyObject *kwargs)
1905{
1906 const char *name;
1907 const char *doc = NULL;
1908 PyObject *base = NULL;
1909 PyObject *dict = NULL;
1910
1911 static char *kwlist[] = {"name", "doc", "base", "dict", NULL};
1912
1913 if (!PyArg_ParseTupleAndKeywords(args, kwargs,
1914 "s|sOO:make_exception_with_doc", kwlist,
1915 &name, &doc, &base, &dict))
1916 return NULL;
1917
1918 return PyErr_NewExceptionWithDoc(name, doc, base, dict);
1919}
1920
Tim Peters9ea17ac2001-02-02 05:57:15 +00001921static PyMethodDef TestMethods[] = {
Fred Drakeacee69f2002-04-01 14:28:58 +00001922 {"raise_exception", raise_exception, METH_VARARGS},
Benjamin Peterson0067bd62008-08-16 16:11:03 +00001923 {"raise_memoryerror", (PyCFunction)raise_memoryerror, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001924 {"test_config", (PyCFunction)test_config, METH_NOARGS},
Benjamin Peterson91d58bd2009-12-13 21:30:54 +00001925 {"test_datetime_capi", test_datetime_capi, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001926 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
1927 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
Nick Coghlanf1f2f682008-12-30 07:29:12 +00001928 {"test_lazy_hash_inheritance", (PyCFunction)test_lazy_hash_inheritance,METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001929 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
Mark Dickinson309aa2d2009-12-21 12:37:06 +00001930 {"test_long_and_overflow", (PyCFunction)test_long_and_overflow,
1931 METH_NOARGS},
Tim Petersbaefd9e2003-01-28 20:37:45 +00001932 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
Thomas Hellera4ea6032003-04-17 18:55:45 +00001933 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Benjamin Peterson92035012008-12-27 16:00:54 +00001934 {"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
Mark Dickinsonf08173b2009-12-03 10:59:46 +00001935 {"test_bug_7414", (PyCFunction)test_bug_7414, METH_NOARGS},
Thomas Woutersa9773292006-04-21 09:43:23 +00001936 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
Thomas Wouters477c8d52006-05-27 19:21:47 +00001937 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001938 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
1939 PyDoc_STR("This is a pretty normal docstring.")},
Benjamin Petersonb173f782009-05-05 22:31:58 +00001940 {"test_string_to_double", (PyCFunction)test_string_to_double, METH_NOARGS},
Benjamin Peterson8667a9b2010-01-09 21:45:28 +00001941 {"test_unicode_compare_with_ascii", (PyCFunction)test_unicode_compare_with_ascii, METH_NOARGS},
Benjamin Petersonb173f782009-05-05 22:31:58 +00001942 {"test_capsule", (PyCFunction)test_capsule, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001943 {"getargs_tuple", getargs_tuple, METH_VARARGS},
Christian Heimes380f7f22008-02-28 11:19:05 +00001944 {"getargs_keywords", (PyCFunction)getargs_keywords,
1945 METH_VARARGS|METH_KEYWORDS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001946 {"getargs_b", getargs_b, METH_VARARGS},
1947 {"getargs_B", getargs_B, METH_VARARGS},
Mark Dickinson1554b182009-12-20 16:03:30 +00001948 {"getargs_h", getargs_h, METH_VARARGS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001949 {"getargs_H", getargs_H, METH_VARARGS},
1950 {"getargs_I", getargs_I, METH_VARARGS},
1951 {"getargs_k", getargs_k, METH_VARARGS},
1952 {"getargs_i", getargs_i, METH_VARARGS},
1953 {"getargs_l", getargs_l, METH_VARARGS},
1954 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001955#ifdef HAVE_LONG_LONG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001956 {"getargs_L", getargs_L, METH_VARARGS},
1957 {"getargs_K", getargs_K, METH_VARARGS},
1958 {"test_longlong_api", test_longlong_api, METH_NOARGS},
Mark Dickinson93f562c2010-01-30 10:30:15 +00001959 {"test_long_long_and_overflow",
1960 (PyCFunction)test_long_long_and_overflow, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001961 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001962 {"codec_incrementalencoder",
1963 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
1964 {"codec_incrementaldecoder",
1965 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001966#endif
Amaury Forgeot d'Arc07404592008-05-12 13:19:07 +00001967 {"test_s_code", (PyCFunction)test_s_code, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001968 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
Guido van Rossumfb67be22007-08-29 18:38:11 +00001969 {"test_Z_code", (PyCFunction)test_Z_code, METH_NOARGS},
Mark Dickinson081dfee2009-03-18 14:47:41 +00001970 {"test_widechar", (PyCFunction)test_widechar, METH_NOARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001971#ifdef WITH_THREAD
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001972 {"_test_thread_state", test_thread_state, METH_VARARGS},
Benjamin Petersona54c9092009-01-13 02:11:23 +00001973 {"_pending_threadfunc", pending_threadfunc, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001974#endif
Guido van Rossumddefaf32007-01-14 03:31:43 +00001975#ifdef HAVE_GETTIMEOFDAY
1976 {"profile_int", profile_int, METH_NOARGS},
1977#endif
Christian Heimes81ee3ef2008-05-04 22:42:01 +00001978 {"traceback_print", traceback_print, METH_VARARGS},
Benjamin Petersone6528212008-07-15 15:32:09 +00001979 {"exception_print", exception_print, METH_VARARGS},
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00001980 {"argparsing", argparsing, METH_VARARGS},
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00001981 {"code_newempty", code_newempty, METH_VARARGS},
Georg Brandl1e28a272009-12-28 08:41:01 +00001982 {"make_exception_with_doc", (PyCFunction)make_exception_with_doc,
1983 METH_VARARGS | METH_KEYWORDS},
Tim Peters9ea17ac2001-02-02 05:57:15 +00001984 {NULL, NULL} /* sentinel */
1985};
1986
Thomas Hellera4ea6032003-04-17 18:55:45 +00001987#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
1988
Thomas Wouters89f507f2006-12-13 04:49:30 +00001989typedef struct {
Christian Heimes1af737c2008-01-23 08:24:23 +00001990 char bool_member;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001991 char byte_member;
1992 unsigned char ubyte_member;
1993 short short_member;
1994 unsigned short ushort_member;
1995 int int_member;
1996 unsigned int uint_member;
1997 long long_member;
1998 unsigned long ulong_member;
1999 float float_member;
2000 double double_member;
Guido van Rossumcd16bf62007-06-13 18:07:49 +00002001#ifdef HAVE_LONG_LONG
2002 PY_LONG_LONG longlong_member;
2003 unsigned PY_LONG_LONG ulonglong_member;
2004#endif
Thomas Wouters89f507f2006-12-13 04:49:30 +00002005} all_structmembers;
2006
2007typedef struct {
2008 PyObject_HEAD
2009 all_structmembers structmembers;
2010} test_structmembers;
2011
2012static struct PyMemberDef test_members[] = {
Christian Heimes1af737c2008-01-23 08:24:23 +00002013 {"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
Thomas Wouters89f507f2006-12-13 04:49:30 +00002014 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
2015 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
2016 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
2017 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
2018 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
2019 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
2020 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
2021 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
2022 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
2023 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
Guido van Rossumcd16bf62007-06-13 18:07:49 +00002024#ifdef HAVE_LONG_LONG
2025 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
2026 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
2027#endif
Thomas Wouters89f507f2006-12-13 04:49:30 +00002028 {NULL}
2029};
2030
2031
Christian Heimes1af737c2008-01-23 08:24:23 +00002032static PyObject *
2033test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
2034{
2035 static char *keywords[] = {
2036 "T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
2037 "T_INT", "T_UINT", "T_LONG", "T_ULONG",
2038 "T_FLOAT", "T_DOUBLE",
2039#ifdef HAVE_LONG_LONG
Guido van Rossumcd16bf62007-06-13 18:07:49 +00002040 "T_LONGLONG", "T_ULONGLONG",
Christian Heimes1af737c2008-01-23 08:24:23 +00002041#endif
Guido van Rossumcd16bf62007-06-13 18:07:49 +00002042 NULL};
Christian Heimes1af737c2008-01-23 08:24:23 +00002043 static char *fmt = "|bbBhHiIlkfd"
2044#ifdef HAVE_LONG_LONG
Guido van Rossumcd16bf62007-06-13 18:07:49 +00002045 "LK"
Christian Heimes1af737c2008-01-23 08:24:23 +00002046#endif
Guido van Rossumcd16bf62007-06-13 18:07:49 +00002047 ;
Christian Heimes1af737c2008-01-23 08:24:23 +00002048 test_structmembers *ob;
2049 ob = PyObject_New(test_structmembers, type);
2050 if (ob == NULL)
Thomas Wouters89f507f2006-12-13 04:49:30 +00002051 return NULL;
2052 memset(&ob->structmembers, 0, sizeof(all_structmembers));
Guido van Rossumcd16bf62007-06-13 18:07:49 +00002053 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
Christian Heimes1af737c2008-01-23 08:24:23 +00002054 &ob->structmembers.bool_member,
2055 &ob->structmembers.byte_member,
2056 &ob->structmembers.ubyte_member,
2057 &ob->structmembers.short_member,
2058 &ob->structmembers.ushort_member,
2059 &ob->structmembers.int_member,
2060 &ob->structmembers.uint_member,
2061 &ob->structmembers.long_member,
2062 &ob->structmembers.ulong_member,
2063 &ob->structmembers.float_member,
2064 &ob->structmembers.double_member
2065#ifdef HAVE_LONG_LONG
2066 , &ob->structmembers.longlong_member,
2067 &ob->structmembers.ulonglong_member
2068#endif
2069 )) {
Thomas Wouters89f507f2006-12-13 04:49:30 +00002070 Py_DECREF(ob);
2071 return NULL;
Christian Heimes1af737c2008-01-23 08:24:23 +00002072 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00002073 return (PyObject *)ob;
2074}
2075
Christian Heimes1af737c2008-01-23 08:24:23 +00002076static void
2077test_structmembers_free(PyObject *ob)
2078{
Thomas Wouters89f507f2006-12-13 04:49:30 +00002079 PyObject_FREE(ob);
2080}
2081
2082static PyTypeObject test_structmembersType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00002083 PyVarObject_HEAD_INIT(NULL, 0)
Thomas Wouters89f507f2006-12-13 04:49:30 +00002084 "test_structmembersType",
2085 sizeof(test_structmembers), /* tp_basicsize */
2086 0, /* tp_itemsize */
2087 test_structmembers_free, /* destructor tp_dealloc */
2088 0, /* tp_print */
2089 0, /* tp_getattr */
2090 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +00002091 0, /* tp_reserved */
Thomas Wouters89f507f2006-12-13 04:49:30 +00002092 0, /* tp_repr */
2093 0, /* tp_as_number */
2094 0, /* tp_as_sequence */
2095 0, /* tp_as_mapping */
2096 0, /* tp_hash */
2097 0, /* tp_call */
2098 0, /* tp_str */
Christian Heimes1af737c2008-01-23 08:24:23 +00002099 PyObject_GenericGetAttr, /* tp_getattro */
2100 PyObject_GenericSetAttr, /* tp_setattro */
Thomas Wouters89f507f2006-12-13 04:49:30 +00002101 0, /* tp_as_buffer */
2102 0, /* tp_flags */
2103 "Type containing all structmember types",
2104 0, /* traverseproc tp_traverse */
2105 0, /* tp_clear */
2106 0, /* tp_richcompare */
2107 0, /* tp_weaklistoffset */
2108 0, /* tp_iter */
2109 0, /* tp_iternext */
2110 0, /* tp_methods */
Christian Heimes1af737c2008-01-23 08:24:23 +00002111 test_members, /* tp_members */
Thomas Wouters89f507f2006-12-13 04:49:30 +00002112 0,
2113 0,
2114 0,
2115 0,
2116 0,
2117 0,
2118 0,
2119 0,
Christian Heimes1af737c2008-01-23 08:24:23 +00002120 test_structmembers_new, /* tp_new */
Thomas Wouters89f507f2006-12-13 04:49:30 +00002121};
2122
2123
Martin v. Löwis1a214512008-06-11 05:26:20 +00002124
2125static struct PyModuleDef _testcapimodule = {
2126 PyModuleDef_HEAD_INIT,
2127 "_testcapi",
2128 NULL,
2129 -1,
2130 TestMethods,
2131 NULL,
2132 NULL,
2133 NULL,
2134 NULL
2135};
2136
Mark Hammond62b1ab12002-07-23 06:31:15 +00002137PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00002138PyInit__testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +00002139{
Fred Drakeacee69f2002-04-01 14:28:58 +00002140 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00002141
Martin v. Löwis1a214512008-06-11 05:26:20 +00002142 m = PyModule_Create(&_testcapimodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00002143 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +00002144 return NULL;
Tim Peters9ea17ac2001-02-02 05:57:15 +00002145
Hirokazu Yamamoto8ebab5d2008-12-31 06:05:46 +00002146 Py_TYPE(&_HashInheritanceTester_Type)=&PyType_Type;
2147
Christian Heimes90aa7642007-12-19 02:45:37 +00002148 Py_TYPE(&test_structmembersType)=&PyType_Type;
Thomas Wouters89f507f2006-12-13 04:49:30 +00002149 Py_INCREF(&test_structmembersType);
2150 PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
2151
Christian Heimes217cfd12007-12-02 14:31:20 +00002152 PyModule_AddObject(m, "CHAR_MAX", PyLong_FromLong(CHAR_MAX));
2153 PyModule_AddObject(m, "CHAR_MIN", PyLong_FromLong(CHAR_MIN));
2154 PyModule_AddObject(m, "UCHAR_MAX", PyLong_FromLong(UCHAR_MAX));
2155 PyModule_AddObject(m, "SHRT_MAX", PyLong_FromLong(SHRT_MAX));
2156 PyModule_AddObject(m, "SHRT_MIN", PyLong_FromLong(SHRT_MIN));
2157 PyModule_AddObject(m, "USHRT_MAX", PyLong_FromLong(USHRT_MAX));
Thomas Wouters89f507f2006-12-13 04:49:30 +00002158 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
2159 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00002160 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
Christian Heimes217cfd12007-12-02 14:31:20 +00002161 PyModule_AddObject(m, "LONG_MAX", PyLong_FromLong(LONG_MAX));
2162 PyModule_AddObject(m, "LONG_MIN", PyLong_FromLong(LONG_MIN));
Thomas Wouters89f507f2006-12-13 04:49:30 +00002163 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
2164 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
2165 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
2166 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
2167 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
Guido van Rossumcd16bf62007-06-13 18:07:49 +00002168 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
2169 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
2170 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
Christian Heimes217cfd12007-12-02 14:31:20 +00002171 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyLong_FromSsize_t(PY_SSIZE_T_MAX));
2172 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyLong_FromSsize_t(PY_SSIZE_T_MIN));
Robert Schuppeniesfbe94c52008-07-14 10:13:31 +00002173 PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyLong_FromSsize_t(sizeof(PyGC_Head)));
Benjamin Peterson9b6df6a2008-10-16 23:56:29 +00002174 Py_INCREF(&PyInstanceMethod_Type);
2175 PyModule_AddObject(m, "instancemethod", (PyObject *)&PyInstanceMethod_Type);
Thomas Hellera4ea6032003-04-17 18:55:45 +00002176
Tim Petersd66595f2001-02-04 03:09:53 +00002177 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
Fred Drakeacee69f2002-04-01 14:28:58 +00002178 Py_INCREF(TestError);
2179 PyModule_AddObject(m, "error", TestError);
Martin v. Löwis1a214512008-06-11 05:26:20 +00002180 return m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00002181}