blob: 95a09202121d26ad2d2f40185e63b3e82a1d8a01 [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
8#include "Python.h"
Martin v. Löwis0347a9a2006-10-27 07:06:52 +00009#include <float.h>
Martin v. Löwisb5bc5372006-10-27 06:16:31 +000010#include "structmember.h"
Benjamin Peterson001e4a62009-12-13 21:27:53 +000011#include "datetime.h"
Serhiy Storchaka4068b012015-02-06 08:56:33 +020012#include "marshal.h"
Tim Peters9ea17ac2001-02-02 05:57:15 +000013
Mark Hammond8d98d2c2003-04-19 15:41:53 +000014#ifdef WITH_THREAD
15#include "pythread.h"
16#endif /* WITH_THREAD */
Antoine Pitrouc83ea132010-05-09 14:46:46 +000017static PyObject *TestError; /* set to exception object in init */
Tim Peters9ea17ac2001-02-02 05:57:15 +000018
Tim Peters91621db2001-06-12 20:10:01 +000019/* Raise TestError with test_name + ": " + msg, and return NULL. */
20
21static PyObject *
22raiseTestError(const char* test_name, const char* msg)
23{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000024 char buf[2048];
Tim Peters91621db2001-06-12 20:10:01 +000025
Antoine Pitrouc83ea132010-05-09 14:46:46 +000026 if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
27 PyErr_SetString(TestError, "internal error msg too large");
28 else {
29 PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
30 PyErr_SetString(TestError, buf);
31 }
32 return NULL;
Tim Peters91621db2001-06-12 20:10:01 +000033}
34
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +000035/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
Tim Peters9ea17ac2001-02-02 05:57:15 +000036
37 The ones derived from autoconf on the UNIX-like OSes can be relied
38 upon (in the absence of sloppy cross-compiling), but the Windows
39 platforms have these hardcoded. Better safe than sorry.
40*/
41static PyObject*
Martin v. Löwisb5bc5372006-10-27 06:16:31 +000042sizeof_error(const char* fatname, const char* typname,
Antoine Pitrouc83ea132010-05-09 14:46:46 +000043 int expected, int got)
Tim Peters9ea17ac2001-02-02 05:57:15 +000044{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000045 char buf[1024];
46 PyOS_snprintf(buf, sizeof(buf),
47 "%.200s #define == %d but sizeof(%.200s) == %d",
48 fatname, expected, typname, got);
49 PyErr_SetString(TestError, buf);
50 return (PyObject*)NULL;
Tim Peters9ea17ac2001-02-02 05:57:15 +000051}
52
53static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000054test_config(PyObject *self)
Tim Peters9ea17ac2001-02-02 05:57:15 +000055{
Tim Peters9ea17ac2001-02-02 05:57:15 +000056#define CHECK_SIZEOF(FATNAME, TYPE) \
Antoine Pitrouc83ea132010-05-09 14:46:46 +000057 if (FATNAME != sizeof(TYPE)) \
58 return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
Tim Peters9ea17ac2001-02-02 05:57:15 +000059
Antoine Pitrouc83ea132010-05-09 14:46:46 +000060 CHECK_SIZEOF(SIZEOF_SHORT, short);
61 CHECK_SIZEOF(SIZEOF_INT, int);
62 CHECK_SIZEOF(SIZEOF_LONG, long);
63 CHECK_SIZEOF(SIZEOF_VOID_P, void*);
64 CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
Tim Peters9ea17ac2001-02-02 05:57:15 +000065#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +000066 CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
Tim Peters9ea17ac2001-02-02 05:57:15 +000067#endif
68
69#undef CHECK_SIZEOF
70
Antoine Pitrouc83ea132010-05-09 14:46:46 +000071 Py_INCREF(Py_None);
72 return Py_None;
Tim Peters9ea17ac2001-02-02 05:57:15 +000073}
74
Tim Peters5c4d5bf2001-02-12 22:13:26 +000075static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000076test_list_api(PyObject *self)
Tim Peters5c4d5bf2001-02-12 22:13:26 +000077{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000078 PyObject* list;
79 int i;
Tim Peters5c4d5bf2001-02-12 22:13:26 +000080
Antoine Pitrouc83ea132010-05-09 14:46:46 +000081 /* SF bug 132008: PyList_Reverse segfaults */
Tim Peters5c4d5bf2001-02-12 22:13:26 +000082#define NLIST 30
Antoine Pitrouc83ea132010-05-09 14:46:46 +000083 list = PyList_New(NLIST);
84 if (list == (PyObject*)NULL)
85 return (PyObject*)NULL;
86 /* list = range(NLIST) */
87 for (i = 0; i < NLIST; ++i) {
88 PyObject* anint = PyInt_FromLong(i);
89 if (anint == (PyObject*)NULL) {
90 Py_DECREF(list);
91 return (PyObject*)NULL;
92 }
93 PyList_SET_ITEM(list, i, anint);
94 }
95 /* list.reverse(), via PyList_Reverse() */
96 i = PyList_Reverse(list); /* should not blow up! */
97 if (i != 0) {
98 Py_DECREF(list);
99 return (PyObject*)NULL;
100 }
101 /* Check that list == range(29, -1, -1) now */
102 for (i = 0; i < NLIST; ++i) {
103 PyObject* anint = PyList_GET_ITEM(list, i);
104 if (PyInt_AS_LONG(anint) != NLIST-1-i) {
105 PyErr_SetString(TestError,
106 "test_list_api: reverse screwed up");
107 Py_DECREF(list);
108 return (PyObject*)NULL;
109 }
110 }
111 Py_DECREF(list);
Tim Peters5c4d5bf2001-02-12 22:13:26 +0000112#undef NLIST
113
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000114 Py_INCREF(Py_None);
115 return Py_None;
Tim Peters5c4d5bf2001-02-12 22:13:26 +0000116}
117
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000118static int
119test_dict_inner(int count)
120{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000121 Py_ssize_t pos = 0, iterations = 0;
122 int i;
123 PyObject *dict = PyDict_New();
124 PyObject *v, *k;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000125
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000126 if (dict == NULL)
127 return -1;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000128
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000129 for (i = 0; i < count; i++) {
130 v = PyInt_FromLong(i);
131 PyDict_SetItem(dict, v, v);
132 Py_DECREF(v);
133 }
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000134
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000135 while (PyDict_Next(dict, &pos, &k, &v)) {
136 PyObject *o;
137 iterations++;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000138
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000139 i = PyInt_AS_LONG(v) + 1;
140 o = PyInt_FromLong(i);
141 if (o == NULL)
142 return -1;
143 if (PyDict_SetItem(dict, k, o) < 0) {
144 Py_DECREF(o);
145 return -1;
146 }
147 Py_DECREF(o);
148 }
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000149
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000150 Py_DECREF(dict);
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000151
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000152 if (iterations != count) {
153 PyErr_SetString(
154 TestError,
155 "test_dict_iteration: dict iteration went wrong ");
156 return -1;
157 } else {
158 return 0;
159 }
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000160}
161
162static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +0000163test_dict_iteration(PyObject* self)
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000164{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000165 int i;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000166
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000167 for (i = 0; i < 200; i++) {
168 if (test_dict_inner(i) < 0) {
169 return NULL;
170 }
171 }
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000172
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000173 Py_INCREF(Py_None);
174 return Py_None;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000175}
176
Tim Peters91621db2001-06-12 20:10:01 +0000177
Nick Coghlan180e4002008-12-30 01:18:48 +0000178/* Issue #4701: Check that PyObject_Hash implicitly calls
179 * PyType_Ready if it hasn't already been called
180 */
181static PyTypeObject _HashInheritanceTester_Type = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000182 PyObject_HEAD_INIT(NULL)
183 0, /* Number of items for varobject */
184 "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 */
191 0, /* tp_compare */
192 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 */
Nick Coghlan180e4002008-12-30 01:18:48 +0000221};
222
223static PyObject*
224test_lazy_hash_inheritance(PyObject* self)
225{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000226 PyTypeObject *type;
227 PyObject *obj;
228 long hash;
Nick Coghlan180e4002008-12-30 01:18:48 +0000229
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000230 type = &_HashInheritanceTester_Type;
Benjamin Petersona98c8e12009-05-05 21:09:21 +0000231
Antoine Pitrouc83ea132010-05-09 14:46:46 +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 Petersona98c8e12009-05-05 21:09:21 +0000236
237
Antoine Pitrouc83ea132010-05-09 14:46:46 +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 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000246
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000247 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 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000254
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000255 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 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000264
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000265 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 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000272
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000273 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 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000280
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000281 Py_DECREF(obj);
Benjamin Petersona98c8e12009-05-05 21:09:21 +0000282
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000283 Py_RETURN_NONE;
Nick Coghlan180e4002008-12-30 01:18:48 +0000284}
285
286
Antoine Pitrou526e4212010-02-02 22:36:17 +0000287/* Issue #7385: Check that memoryview() does not crash
288 * when bf_getbuffer returns an error
289 */
290
291static int
292broken_buffer_getbuffer(PyObject *self, Py_buffer *view, int flags)
293{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000294 PyErr_SetString(
295 TestError,
296 "test_broken_memoryview: expected error in bf_getbuffer");
297 return -1;
Antoine Pitrou526e4212010-02-02 22:36:17 +0000298}
299
300static PyBufferProcs memoryviewtester_as_buffer = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000301 0, /* bf_getreadbuffer */
302 0, /* bf_getwritebuffer */
303 0, /* bf_getsegcount */
304 0, /* bf_getcharbuffer */
305 (getbufferproc)broken_buffer_getbuffer, /* bf_getbuffer */
306 0, /* bf_releasebuffer */
Antoine Pitrou526e4212010-02-02 22:36:17 +0000307};
308
309static PyTypeObject _MemoryViewTester_Type = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000310 PyObject_HEAD_INIT(NULL)
311 0, /* Number of items for varobject */
312 "memoryviewtester", /* Name of this type */
313 sizeof(PyObject), /* Basic object size */
314 0, /* Item size for varobject */
315 (destructor)PyObject_Del, /* tp_dealloc */
316 0, /* tp_print */
317 0, /* tp_getattr */
318 0, /* tp_setattr */
319 0, /* tp_compare */
320 0, /* tp_repr */
321 0, /* tp_as_number */
322 0, /* tp_as_sequence */
323 0, /* tp_as_mapping */
324 0, /* tp_hash */
325 0, /* tp_call */
326 0, /* tp_str */
327 PyObject_GenericGetAttr, /* tp_getattro */
328 0, /* tp_setattro */
329 &memoryviewtester_as_buffer, /* tp_as_buffer */
330 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_NEWBUFFER, /* tp_flags */
331 0, /* tp_doc */
332 0, /* tp_traverse */
333 0, /* tp_clear */
334 0, /* tp_richcompare */
335 0, /* tp_weaklistoffset */
336 0, /* tp_iter */
337 0, /* tp_iternext */
338 0, /* tp_methods */
339 0, /* tp_members */
340 0, /* tp_getset */
341 0, /* tp_base */
342 0, /* tp_dict */
343 0, /* tp_descr_get */
344 0, /* tp_descr_set */
345 0, /* tp_dictoffset */
346 0, /* tp_init */
347 0, /* tp_alloc */
348 PyType_GenericNew, /* tp_new */
Antoine Pitrou526e4212010-02-02 22:36:17 +0000349};
350
351static PyObject*
352test_broken_memoryview(PyObject* self)
353{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000354 PyObject *obj = PyObject_New(PyObject, &_MemoryViewTester_Type);
355 PyObject *res;
Antoine Pitrou526e4212010-02-02 22:36:17 +0000356
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000357 if (obj == NULL) {
358 PyErr_Clear();
359 PyErr_SetString(
360 TestError,
361 "test_broken_memoryview: failed to create object");
362 return NULL;
363 }
Antoine Pitrou526e4212010-02-02 22:36:17 +0000364
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000365 res = PyMemoryView_FromObject(obj);
366 if (res || !PyErr_Occurred()){
367 PyErr_SetString(
368 TestError,
369 "test_broken_memoryview: memoryview() didn't raise an Exception");
370 Py_XDECREF(res);
371 Py_DECREF(obj);
372 return NULL;
373 }
Antoine Pitrou526e4212010-02-02 22:36:17 +0000374
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000375 PyErr_Clear();
376 Py_DECREF(obj);
377 Py_RETURN_NONE;
Antoine Pitrou526e4212010-02-02 22:36:17 +0000378}
379
Stefan Krah5606cd92015-01-30 20:11:10 +0100380static PyObject *
381test_to_contiguous(PyObject* self, PyObject *noargs)
382{
383 int data[9] = {0, -1, 1, -1, 2, -1, 3, -1, 4};
384 int result[5];
385 Py_ssize_t itemsize = sizeof(int);
386 Py_ssize_t shape = 5;
387 Py_ssize_t strides = 2 * itemsize;
388 Py_buffer view = {
389 data,
390 NULL,
391 5 * itemsize,
392 itemsize,
393 1,
394 1,
395 NULL,
396 &shape,
397 &strides,
398 NULL,
399 {0, 0},
400 NULL
401 };
402 int i;
403
404 PyBuffer_ToContiguous(result, &view, view.len, 'C');
405 for (i = 0; i < 5; i++) {
406 if (result[i] != i) {
407 PyErr_SetString(TestError,
408 "test_to_contiguous: incorrect result");
409 return NULL;
410 }
411 }
412
413 view.buf = &data[8];
414 view.strides[0] = -2 * itemsize;
415
416 PyBuffer_ToContiguous(result, &view, view.len, 'C');
417 for (i = 0; i < 5; i++) {
418 if (result[i] != 4-i) {
419 PyErr_SetString(TestError,
420 "test_to_contiguous: incorrect result");
421 return NULL;
422 }
423 }
424
425 Py_RETURN_NONE;
426}
Antoine Pitrou526e4212010-02-02 22:36:17 +0000427
Stefan Krah0a7fc532015-02-01 16:10:35 +0100428static PyObject *
429test_from_contiguous(PyObject* self, PyObject *noargs)
430{
431 int data[9] = {-1,-1,-1,-1,-1,-1,-1,-1,-1};
432 int init[5] = {0, 1, 2, 3, 4};
433 Py_ssize_t itemsize = sizeof(int);
434 Py_ssize_t shape = 5;
435 Py_ssize_t strides = 2 * itemsize;
436 Py_buffer view = {
437 data,
438 NULL,
439 5 * itemsize,
440 itemsize,
441 1,
442 1,
443 NULL,
444 &shape,
445 &strides,
446 NULL,
447 {0, 0},
448 NULL
449 };
450 int *ptr;
451 int i;
452
453 PyBuffer_FromContiguous(&view, init, view.len, 'C');
454 ptr = view.buf;
455 for (i = 0; i < 5; i++) {
456 if (ptr[2*i] != i) {
457 PyErr_SetString(TestError,
458 "test_from_contiguous: incorrect result");
459 return NULL;
460 }
461 }
462
463 view.buf = &data[8];
464 view.strides[0] = -2 * itemsize;
465
466 PyBuffer_FromContiguous(&view, init, view.len, 'C');
467 ptr = view.buf;
468 for (i = 0; i < 5; i++) {
469 if (*(ptr-2*i) != i) {
470 PyErr_SetString(TestError,
471 "test_from_contiguous: incorrect result");
472 return NULL;
473 }
474 }
475
476 Py_RETURN_NONE;
477}
478
479
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000480/* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
Tim Petersff70d3c2001-06-14 01:11:03 +0000481 PyLong_{As, From}{Unsigned,}LongLong().
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000482
483 Note that the meat of the test is contained in testcapi_long.h.
484 This is revolting, but delicate code duplication is worse: "almost
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000485 exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000486 dependence on type names makes it impossible to use a parameterized
487 function. A giant macro would be even worse than this. A C++ template
488 would be perfect.
489
490 The "report an error" functions are deliberately not part of the #include
491 file: if the test fails, you can set a breakpoint in the appropriate
492 error function directly, and crawl back from there in the debugger.
493*/
494
495#define UNBIND(X) Py_DECREF(X); (X) = NULL
496
497static PyObject *
498raise_test_long_error(const char* msg)
499{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000500 return raiseTestError("test_long_api", msg);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000501}
502
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000503#define TESTNAME test_long_api_inner
504#define TYPENAME long
505#define F_S_TO_PY PyLong_FromLong
506#define F_PY_TO_S PyLong_AsLong
507#define F_U_TO_PY PyLong_FromUnsignedLong
508#define F_PY_TO_U PyLong_AsUnsignedLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000509
510#include "testcapi_long.h"
511
512static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000513test_long_api(PyObject* self)
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000514{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000515 return TESTNAME(raise_test_long_error);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000516}
517
518#undef TESTNAME
519#undef TYPENAME
520#undef F_S_TO_PY
521#undef F_PY_TO_S
522#undef F_U_TO_PY
523#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000524
525#ifdef HAVE_LONG_LONG
Tim Peters91621db2001-06-12 20:10:01 +0000526
527static PyObject *
Tim Petersd1a7da62001-06-13 00:35:57 +0000528raise_test_longlong_error(const char* msg)
529{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000530 return raiseTestError("test_longlong_api", msg);
Tim Petersd1a7da62001-06-13 00:35:57 +0000531}
532
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000533#define TESTNAME test_longlong_api_inner
534#define TYPENAME PY_LONG_LONG
535#define F_S_TO_PY PyLong_FromLongLong
536#define F_PY_TO_S PyLong_AsLongLong
537#define F_U_TO_PY PyLong_FromUnsignedLongLong
538#define F_PY_TO_U PyLong_AsUnsignedLongLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000539
540#include "testcapi_long.h"
Tim Petersd1a7da62001-06-13 00:35:57 +0000541
542static PyObject *
Skip Montanaro9582c142006-04-18 01:01:41 +0000543test_longlong_api(PyObject* self, PyObject *args)
Tim Peters91621db2001-06-12 20:10:01 +0000544{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000545 return TESTNAME(raise_test_longlong_error);
Tim Peters91621db2001-06-12 20:10:01 +0000546}
547
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000548#undef TESTNAME
549#undef TYPENAME
550#undef F_S_TO_PY
551#undef F_PY_TO_S
552#undef F_U_TO_PY
553#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000554
Mark Dickinsone31d3002009-12-21 11:21:25 +0000555/* Test the PyLong_AsLongAndOverflow API. General conversion to PY_LONG
556 is tested by test_long_api_inner. This test will concentrate on proper
557 handling of overflow.
558*/
559
560static PyObject *
561test_long_and_overflow(PyObject *self)
562{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000563 PyObject *num, *one, *temp;
564 long value;
565 int overflow;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000566
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000567 /* Test that overflow is set properly for a large value. */
568 /* num is a number larger than LONG_MAX even on 64-bit platforms */
569 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
570 if (num == NULL)
571 return NULL;
572 overflow = 1234;
573 value = PyLong_AsLongAndOverflow(num, &overflow);
574 Py_DECREF(num);
575 if (value == -1 && PyErr_Occurred())
576 return NULL;
577 if (value != -1)
578 return raiseTestError("test_long_and_overflow",
579 "return value was not set to -1");
580 if (overflow != 1)
581 return raiseTestError("test_long_and_overflow",
582 "overflow was not set to 1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000583
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000584 /* Same again, with num = LONG_MAX + 1 */
585 num = PyLong_FromLong(LONG_MAX);
586 if (num == NULL)
587 return NULL;
588 one = PyLong_FromLong(1L);
589 if (one == NULL) {
590 Py_DECREF(num);
591 return NULL;
592 }
593 temp = PyNumber_Add(num, one);
594 Py_DECREF(one);
595 Py_DECREF(num);
596 num = temp;
597 if (num == NULL)
598 return NULL;
599 overflow = 0;
600 value = PyLong_AsLongAndOverflow(num, &overflow);
601 Py_DECREF(num);
602 if (value == -1 && PyErr_Occurred())
603 return NULL;
604 if (value != -1)
605 return raiseTestError("test_long_and_overflow",
606 "return value was not set to -1");
607 if (overflow != 1)
608 return raiseTestError("test_long_and_overflow",
609 "overflow was not set to 1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000610
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000611 /* Test that overflow is set properly for a large negative value. */
612 /* num is a number smaller than LONG_MIN even on 64-bit platforms */
613 num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
614 if (num == NULL)
615 return NULL;
616 overflow = 1234;
617 value = PyLong_AsLongAndOverflow(num, &overflow);
618 Py_DECREF(num);
619 if (value == -1 && PyErr_Occurred())
620 return NULL;
621 if (value != -1)
622 return raiseTestError("test_long_and_overflow",
623 "return value was not set to -1");
624 if (overflow != -1)
625 return raiseTestError("test_long_and_overflow",
626 "overflow was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000627
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000628 /* Same again, with num = LONG_MIN - 1 */
629 num = PyLong_FromLong(LONG_MIN);
630 if (num == NULL)
631 return NULL;
632 one = PyLong_FromLong(1L);
633 if (one == NULL) {
634 Py_DECREF(num);
635 return NULL;
636 }
637 temp = PyNumber_Subtract(num, one);
638 Py_DECREF(one);
639 Py_DECREF(num);
640 num = temp;
641 if (num == NULL)
642 return NULL;
643 overflow = 0;
644 value = PyLong_AsLongAndOverflow(num, &overflow);
645 Py_DECREF(num);
646 if (value == -1 && PyErr_Occurred())
647 return NULL;
648 if (value != -1)
649 return raiseTestError("test_long_and_overflow",
650 "return value was not set to -1");
651 if (overflow != -1)
652 return raiseTestError("test_long_and_overflow",
653 "overflow was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000654
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000655 /* Test that overflow is cleared properly for small values. */
656 num = PyLong_FromString("FF", NULL, 16);
657 if (num == NULL)
658 return NULL;
659 overflow = 1234;
660 value = PyLong_AsLongAndOverflow(num, &overflow);
661 Py_DECREF(num);
662 if (value == -1 && PyErr_Occurred())
663 return NULL;
664 if (value != 0xFF)
665 return raiseTestError("test_long_and_overflow",
666 "expected return value 0xFF");
667 if (overflow != 0)
668 return raiseTestError("test_long_and_overflow",
669 "overflow was not cleared");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000670
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000671 num = PyLong_FromString("-FF", NULL, 16);
672 if (num == NULL)
673 return NULL;
674 overflow = 0;
675 value = PyLong_AsLongAndOverflow(num, &overflow);
676 Py_DECREF(num);
677 if (value == -1 && PyErr_Occurred())
678 return NULL;
679 if (value != -0xFF)
680 return raiseTestError("test_long_and_overflow",
681 "expected return value 0xFF");
682 if (overflow != 0)
683 return raiseTestError("test_long_and_overflow",
684 "overflow was set incorrectly");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000685
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000686 num = PyLong_FromLong(LONG_MAX);
687 if (num == NULL)
688 return NULL;
689 overflow = 1234;
690 value = PyLong_AsLongAndOverflow(num, &overflow);
691 Py_DECREF(num);
692 if (value == -1 && PyErr_Occurred())
693 return NULL;
694 if (value != LONG_MAX)
695 return raiseTestError("test_long_and_overflow",
696 "expected return value LONG_MAX");
697 if (overflow != 0)
698 return raiseTestError("test_long_and_overflow",
699 "overflow was not cleared");
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000700
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000701 num = PyLong_FromLong(LONG_MIN);
702 if (num == NULL)
703 return NULL;
704 overflow = 0;
705 value = PyLong_AsLongAndOverflow(num, &overflow);
706 Py_DECREF(num);
707 if (value == -1 && PyErr_Occurred())
708 return NULL;
709 if (value != LONG_MIN)
710 return raiseTestError("test_long_and_overflow",
711 "expected return value LONG_MIN");
712 if (overflow != 0)
713 return raiseTestError("test_long_and_overflow",
714 "overflow was not cleared");
Mark Dickinsoned02b3f2009-12-21 11:31:54 +0000715
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000716 Py_INCREF(Py_None);
717 return Py_None;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000718}
719
Mark Dickinsona36507c2010-01-30 10:08:33 +0000720/* Test the PyLong_AsLongLongAndOverflow API. General conversion to
721 PY_LONG_LONG is tested by test_long_api_inner. This test will
722 concentrate on proper handling of overflow.
723*/
724
725static PyObject *
726test_long_long_and_overflow(PyObject *self)
727{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000728 PyObject *num, *one, *temp;
729 PY_LONG_LONG value;
730 int overflow;
Mark Dickinsona36507c2010-01-30 10:08:33 +0000731
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000732 /* Test that overflow is set properly for a large value. */
733 /* num is a number larger than PY_LLONG_MAX on a typical machine. */
734 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
735 if (num == NULL)
736 return NULL;
737 overflow = 1234;
738 value = PyLong_AsLongLongAndOverflow(num, &overflow);
739 Py_DECREF(num);
740 if (value == -1 && PyErr_Occurred())
741 return NULL;
742 if (value != -1)
743 return raiseTestError("test_long_long_and_overflow",
744 "return value was not set to -1");
745 if (overflow != 1)
746 return raiseTestError("test_long_long_and_overflow",
747 "overflow was not set to 1");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000748
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000749 /* Same again, with num = PY_LLONG_MAX + 1 */
750 num = PyLong_FromLongLong(PY_LLONG_MAX);
751 if (num == NULL)
752 return NULL;
753 one = PyLong_FromLong(1L);
754 if (one == NULL) {
755 Py_DECREF(num);
756 return NULL;
757 }
758 temp = PyNumber_Add(num, one);
759 Py_DECREF(one);
760 Py_DECREF(num);
761 num = temp;
762 if (num == NULL)
763 return NULL;
764 overflow = 0;
765 value = PyLong_AsLongLongAndOverflow(num, &overflow);
766 Py_DECREF(num);
767 if (value == -1 && PyErr_Occurred())
768 return NULL;
769 if (value != -1)
770 return raiseTestError("test_long_long_and_overflow",
771 "return value was not set to -1");
772 if (overflow != 1)
773 return raiseTestError("test_long_long_and_overflow",
774 "overflow was not set to 1");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000775
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000776 /* Test that overflow is set properly for a large negative value. */
777 /* num is a number smaller than PY_LLONG_MIN on a typical platform */
778 num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
779 if (num == NULL)
780 return NULL;
781 overflow = 1234;
782 value = PyLong_AsLongLongAndOverflow(num, &overflow);
783 Py_DECREF(num);
784 if (value == -1 && PyErr_Occurred())
785 return NULL;
786 if (value != -1)
787 return raiseTestError("test_long_long_and_overflow",
788 "return value was not set to -1");
789 if (overflow != -1)
790 return raiseTestError("test_long_long_and_overflow",
791 "overflow was not set to -1");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000792
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000793 /* Same again, with num = PY_LLONG_MIN - 1 */
794 num = PyLong_FromLongLong(PY_LLONG_MIN);
795 if (num == NULL)
796 return NULL;
797 one = PyLong_FromLong(1L);
798 if (one == NULL) {
799 Py_DECREF(num);
800 return NULL;
801 }
802 temp = PyNumber_Subtract(num, one);
803 Py_DECREF(one);
804 Py_DECREF(num);
805 num = temp;
806 if (num == NULL)
807 return NULL;
808 overflow = 0;
809 value = PyLong_AsLongLongAndOverflow(num, &overflow);
810 Py_DECREF(num);
811 if (value == -1 && PyErr_Occurred())
812 return NULL;
813 if (value != -1)
814 return raiseTestError("test_long_long_and_overflow",
815 "return value was not set to -1");
816 if (overflow != -1)
817 return raiseTestError("test_long_long_and_overflow",
818 "overflow was not set to -1");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000819
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000820 /* Test that overflow is cleared properly for small values. */
821 num = PyLong_FromString("FF", NULL, 16);
822 if (num == NULL)
823 return NULL;
824 overflow = 1234;
825 value = PyLong_AsLongLongAndOverflow(num, &overflow);
826 Py_DECREF(num);
827 if (value == -1 && PyErr_Occurred())
828 return NULL;
829 if (value != 0xFF)
830 return raiseTestError("test_long_long_and_overflow",
831 "expected return value 0xFF");
832 if (overflow != 0)
833 return raiseTestError("test_long_long_and_overflow",
834 "overflow was not cleared");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000835
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000836 num = PyLong_FromString("-FF", NULL, 16);
837 if (num == NULL)
838 return NULL;
839 overflow = 0;
840 value = PyLong_AsLongLongAndOverflow(num, &overflow);
841 Py_DECREF(num);
842 if (value == -1 && PyErr_Occurred())
843 return NULL;
844 if (value != -0xFF)
845 return raiseTestError("test_long_long_and_overflow",
846 "expected return value 0xFF");
847 if (overflow != 0)
848 return raiseTestError("test_long_long_and_overflow",
849 "overflow was set incorrectly");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000850
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000851 num = PyLong_FromLongLong(PY_LLONG_MAX);
852 if (num == NULL)
853 return NULL;
854 overflow = 1234;
855 value = PyLong_AsLongLongAndOverflow(num, &overflow);
856 Py_DECREF(num);
857 if (value == -1 && PyErr_Occurred())
858 return NULL;
859 if (value != PY_LLONG_MAX)
860 return raiseTestError("test_long_long_and_overflow",
861 "expected return value PY_LLONG_MAX");
862 if (overflow != 0)
863 return raiseTestError("test_long_long_and_overflow",
864 "overflow was not cleared");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000865
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000866 num = PyLong_FromLongLong(PY_LLONG_MIN);
867 if (num == NULL)
868 return NULL;
869 overflow = 0;
870 value = PyLong_AsLongLongAndOverflow(num, &overflow);
871 Py_DECREF(num);
872 if (value == -1 && PyErr_Occurred())
873 return NULL;
874 if (value != PY_LLONG_MIN)
875 return raiseTestError("test_long_long_and_overflow",
876 "expected return value PY_LLONG_MIN");
877 if (overflow != 0)
878 return raiseTestError("test_long_long_and_overflow",
879 "overflow was not cleared");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000880
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000881 Py_INCREF(Py_None);
882 return Py_None;
Mark Dickinsona36507c2010-01-30 10:08:33 +0000883}
884
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000885/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
Tim Petersd38b1c72001-09-30 05:09:37 +0000886 for both long and int arguments. The test may leak a little memory if
887 it fails.
888*/
889static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000890test_L_code(PyObject *self)
Tim Petersd38b1c72001-09-30 05:09:37 +0000891{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000892 PyObject *tuple, *num;
893 PY_LONG_LONG value;
Tim Petersd38b1c72001-09-30 05:09:37 +0000894
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000895 tuple = PyTuple_New(1);
896 if (tuple == NULL)
897 return NULL;
Tim Petersd38b1c72001-09-30 05:09:37 +0000898
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000899 num = PyLong_FromLong(42);
900 if (num == NULL)
901 return NULL;
Tim Petersd38b1c72001-09-30 05:09:37 +0000902
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000903 PyTuple_SET_ITEM(tuple, 0, num);
Tim Petersd38b1c72001-09-30 05:09:37 +0000904
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000905 value = -1;
906 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
907 return NULL;
908 if (value != 42)
909 return raiseTestError("test_L_code",
910 "L code returned wrong value for long 42");
Tim Petersd38b1c72001-09-30 05:09:37 +0000911
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000912 Py_DECREF(num);
913 num = PyInt_FromLong(42);
914 if (num == NULL)
915 return NULL;
Tim Petersd38b1c72001-09-30 05:09:37 +0000916
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000917 PyTuple_SET_ITEM(tuple, 0, num);
Tim Petersd38b1c72001-09-30 05:09:37 +0000918
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000919 value = -1;
920 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
921 return NULL;
922 if (value != 42)
923 return raiseTestError("test_L_code",
924 "L code returned wrong value for int 42");
Tim Petersd38b1c72001-09-30 05:09:37 +0000925
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000926 Py_DECREF(tuple);
927 Py_INCREF(Py_None);
928 return Py_None;
Tim Petersd38b1c72001-09-30 05:09:37 +0000929}
930
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000931#endif /* ifdef HAVE_LONG_LONG */
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000932
Serhiy Storchakac6a7c962016-05-16 09:55:32 +0300933static PyObject *
934get_args(PyObject *self, PyObject *args)
935{
936 if (args == NULL) {
937 args = Py_None;
938 }
939 Py_INCREF(args);
940 return args;
941}
942
943static PyObject *
944get_kwargs(PyObject *self, PyObject *args, PyObject *kwargs)
945{
946 if (kwargs == NULL) {
947 kwargs = Py_None;
948 }
949 Py_INCREF(kwargs);
950 return kwargs;
951}
952
Georg Brandl5f135782006-07-26 08:03:10 +0000953/* Test tuple argument processing */
954static PyObject *
955getargs_tuple(PyObject *self, PyObject *args)
956{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000957 int a, b, c;
958 if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
959 return NULL;
960 return Py_BuildValue("iii", a, b, c);
Georg Brandl5f135782006-07-26 08:03:10 +0000961}
962
Christian Heimesea837932008-02-26 17:23:51 +0000963/* test PyArg_ParseTupleAndKeywords */
964static PyObject *getargs_keywords(PyObject *self, PyObject *args, PyObject *kwargs)
965{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000966 static char *keywords[] = {"arg1","arg2","arg3","arg4","arg5", NULL};
967 static char *fmt="(ii)i|(i(ii))(iii)i";
968 int int_args[10]={-1, -1, -1, -1, -1, -1, -1, -1, -1, -1};
Christian Heimesea837932008-02-26 17:23:51 +0000969
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000970 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
971 &int_args[0], &int_args[1], &int_args[2], &int_args[3], &int_args[4],
972 &int_args[5], &int_args[6], &int_args[7], &int_args[8], &int_args[9]))
973 return NULL;
974 return Py_BuildValue("iiiiiiiiii",
975 int_args[0], int_args[1], int_args[2], int_args[3], int_args[4],
976 int_args[5], int_args[6], int_args[7], int_args[8], int_args[9]);
Christian Heimesea837932008-02-26 17:23:51 +0000977}
978
Thomas Heller3457e4b2003-04-24 16:14:27 +0000979/* Functions to call PyArg_ParseTuple with integer format codes,
980 and return the result.
981*/
Thomas Hellera4ea6032003-04-17 18:55:45 +0000982static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000983getargs_b(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000984{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000985 unsigned char value;
986 if (!PyArg_ParseTuple(args, "b", &value))
987 return NULL;
988 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000989}
990
Thomas Heller3457e4b2003-04-24 16:14:27 +0000991static PyObject *
992getargs_B(PyObject *self, PyObject *args)
993{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000994 unsigned char value;
995 if (!PyArg_ParseTuple(args, "B", &value))
996 return NULL;
997 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Heller3457e4b2003-04-24 16:14:27 +0000998}
999
1000static PyObject *
Mark Dickinsonb5e348b2009-12-20 15:57:56 +00001001getargs_h(PyObject *self, PyObject *args)
1002{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001003 short value;
1004 if (!PyArg_ParseTuple(args, "h", &value))
1005 return NULL;
1006 return PyLong_FromLong((long)value);
Mark Dickinsonb5e348b2009-12-20 15:57:56 +00001007}
1008
1009static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +00001010getargs_H(PyObject *self, PyObject *args)
1011{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001012 unsigned short value;
1013 if (!PyArg_ParseTuple(args, "H", &value))
1014 return NULL;
1015 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001016}
1017
1018static PyObject *
1019getargs_I(PyObject *self, PyObject *args)
1020{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001021 unsigned int value;
1022 if (!PyArg_ParseTuple(args, "I", &value))
1023 return NULL;
1024 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001025}
1026
1027static PyObject *
1028getargs_k(PyObject *self, PyObject *args)
1029{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001030 unsigned long value;
1031 if (!PyArg_ParseTuple(args, "k", &value))
1032 return NULL;
1033 return PyLong_FromUnsignedLong(value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001034}
1035
1036static PyObject *
1037getargs_i(PyObject *self, PyObject *args)
1038{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001039 int value;
1040 if (!PyArg_ParseTuple(args, "i", &value))
1041 return NULL;
1042 return PyLong_FromLong((long)value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001043}
1044
Thomas Hellera4ea6032003-04-17 18:55:45 +00001045static PyObject *
1046getargs_l(PyObject *self, PyObject *args)
1047{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001048 long value;
1049 if (!PyArg_ParseTuple(args, "l", &value))
1050 return NULL;
1051 return PyLong_FromLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001052}
1053
Georg Brandl7f573f72006-04-13 07:59:30 +00001054static PyObject *
1055getargs_n(PyObject *self, PyObject *args)
1056{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001057 Py_ssize_t value;
1058 if (!PyArg_ParseTuple(args, "n", &value))
1059 return NULL;
1060 return PyInt_FromSsize_t(value);
Georg Brandl7f573f72006-04-13 07:59:30 +00001061}
1062
Thomas Hellera4ea6032003-04-17 18:55:45 +00001063#ifdef HAVE_LONG_LONG
Thomas Hellera4ea6032003-04-17 18:55:45 +00001064static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +00001065getargs_L(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +00001066{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001067 PY_LONG_LONG value;
1068 if (!PyArg_ParseTuple(args, "L", &value))
1069 return NULL;
1070 return PyLong_FromLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001071}
1072
Thomas Hellera4ea6032003-04-17 18:55:45 +00001073static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +00001074getargs_K(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +00001075{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001076 unsigned PY_LONG_LONG value;
1077 if (!PyArg_ParseTuple(args, "K", &value))
1078 return NULL;
1079 return PyLong_FromUnsignedLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001080}
1081#endif
1082
1083/* This function not only tests the 'k' getargs code, but also the
1084 PyInt_AsUnsignedLongMask() and PyInt_AsUnsignedLongMask() functions. */
1085static PyObject *
1086test_k_code(PyObject *self)
1087{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001088 PyObject *tuple, *num;
1089 unsigned long value;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001090
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001091 tuple = PyTuple_New(1);
1092 if (tuple == NULL)
1093 return NULL;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001094
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001095 /* a number larger than ULONG_MAX even on 64-bit platforms */
1096 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
1097 if (num == NULL)
1098 return NULL;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001099
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001100 value = PyInt_AsUnsignedLongMask(num);
1101 if (value != ULONG_MAX)
1102 return raiseTestError("test_k_code",
1103 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +00001104
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001105 PyTuple_SET_ITEM(tuple, 0, num);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001106
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001107 value = 0;
1108 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
1109 return NULL;
1110 if (value != ULONG_MAX)
1111 return raiseTestError("test_k_code",
1112 "k code returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +00001113
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001114 Py_DECREF(num);
1115 num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
1116 if (num == NULL)
1117 return NULL;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001118
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001119 value = PyInt_AsUnsignedLongMask(num);
1120 if (value != (unsigned long)-0x42)
1121 return raiseTestError("test_k_code",
1122 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +00001123
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001124 PyTuple_SET_ITEM(tuple, 0, num);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001125
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001126 value = 0;
1127 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
1128 return NULL;
1129 if (value != (unsigned long)-0x42)
1130 return raiseTestError("test_k_code",
1131 "k code returned wrong value for long -0xFFF..000042");
Thomas Hellera4ea6032003-04-17 18:55:45 +00001132
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001133 Py_DECREF(tuple);
1134 Py_INCREF(Py_None);
1135 return Py_None;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001136}
1137
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001138static PyObject *
Serhiy Storchakad0dc72b2016-05-16 10:12:02 +03001139getargs_f(PyObject *self, PyObject *args)
1140{
1141 float f;
1142 if (!PyArg_ParseTuple(args, "f", &f))
1143 return NULL;
1144 return PyFloat_FromDouble(f);
1145}
1146
1147static PyObject *
1148getargs_d(PyObject *self, PyObject *args)
1149{
1150 double d;
1151 if (!PyArg_ParseTuple(args, "d", &d))
1152 return NULL;
1153 return PyFloat_FromDouble(d);
1154}
1155
1156static PyObject *
1157getargs_D(PyObject *self, PyObject *args)
1158{
1159 Py_complex cval;
1160 if (!PyArg_ParseTuple(args, "D", &cval))
1161 return NULL;
1162 return PyComplex_FromCComplex(cval);
1163}
1164
1165static PyObject *
1166getargs_S(PyObject *self, PyObject *args)
1167{
1168 PyObject *obj;
1169 if (!PyArg_ParseTuple(args, "S", &obj))
1170 return NULL;
1171 Py_INCREF(obj);
1172 return obj;
1173}
1174
1175static PyObject *
1176getargs_Y(PyObject *self, PyObject *args)
1177{
1178 PyObject *obj;
1179 if (!PyArg_ParseTuple(args, "Y", &obj))
1180 return NULL;
1181 Py_INCREF(obj);
1182 return obj;
1183}
1184
1185#ifdef Py_USING_UNICODE
1186static PyObject *
1187getargs_U(PyObject *self, PyObject *args)
1188{
1189 PyObject *obj;
1190 if (!PyArg_ParseTuple(args, "U", &obj))
1191 return NULL;
1192 Py_INCREF(obj);
1193 return obj;
1194}
1195#endif
1196
1197static PyObject *
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001198getargs_c(PyObject *self, PyObject *args)
1199{
1200 char c;
1201 if (!PyArg_ParseTuple(args, "c", &c))
1202 return NULL;
1203 return PyInt_FromLong((unsigned char)c);
1204}
1205
1206static PyObject *
1207getargs_s(PyObject *self, PyObject *args)
1208{
1209 const char *str;
1210 if (!PyArg_ParseTuple(args, "s", &str))
1211 return NULL;
1212 return PyBytes_FromString(str);
1213}
1214
1215static PyObject *
1216getargs_s_star(PyObject *self, PyObject *args)
1217{
1218 Py_buffer buffer;
1219 PyObject *bytes;
1220 if (!PyArg_ParseTuple(args, "s*", &buffer))
1221 return NULL;
1222 bytes = PyBytes_FromStringAndSize(buffer.buf, buffer.len);
1223 PyBuffer_Release(&buffer);
1224 return bytes;
1225}
1226
1227static PyObject *
1228getargs_s_hash(PyObject *self, PyObject *args)
1229{
1230 const char *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001231 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001232 if (!PyArg_ParseTuple(args, "s#", &str, &size))
1233 return NULL;
1234 return PyBytes_FromStringAndSize(str, size);
1235}
1236
1237static PyObject *
1238getargs_t_hash(PyObject *self, PyObject *args)
1239{
1240 const char *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001241 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001242 if (!PyArg_ParseTuple(args, "t#", &str, &size))
1243 return NULL;
1244 return PyBytes_FromStringAndSize(str, size);
1245}
1246
1247static PyObject *
1248getargs_z(PyObject *self, PyObject *args)
1249{
1250 const char *str;
1251 if (!PyArg_ParseTuple(args, "z", &str))
1252 return NULL;
1253 if (str != NULL)
1254 return PyBytes_FromString(str);
1255 else
1256 Py_RETURN_NONE;
1257}
1258
1259static PyObject *
1260getargs_z_star(PyObject *self, PyObject *args)
1261{
1262 Py_buffer buffer;
1263 PyObject *bytes;
1264 if (!PyArg_ParseTuple(args, "z*", &buffer))
1265 return NULL;
1266 if (buffer.buf != NULL)
1267 bytes = PyBytes_FromStringAndSize(buffer.buf, buffer.len);
1268 else {
1269 Py_INCREF(Py_None);
1270 bytes = Py_None;
1271 }
1272 PyBuffer_Release(&buffer);
1273 return bytes;
1274}
1275
1276static PyObject *
1277getargs_z_hash(PyObject *self, PyObject *args)
1278{
1279 const char *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001280 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001281 if (!PyArg_ParseTuple(args, "z#", &str, &size))
1282 return NULL;
1283 if (str != NULL)
1284 return PyBytes_FromStringAndSize(str, size);
1285 else
1286 Py_RETURN_NONE;
1287}
1288
1289static PyObject *
1290getargs_w(PyObject *self, PyObject *args)
1291{
1292 char *str;
1293 Py_ssize_t size;
1294
1295 if (!PyArg_ParseTuple(args, "wn", &str, &size))
1296 return NULL;
1297
1298 if (2 <= size) {
1299 str[0] = '[';
1300 str[size-1] = ']';
1301 }
1302
1303 return PyBytes_FromStringAndSize(str, size);
1304}
1305
1306static PyObject *
1307getargs_w_hash(PyObject *self, PyObject *args)
1308{
1309 char *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001310 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001311
1312 if (!PyArg_ParseTuple(args, "w#", &str, &size))
1313 return NULL;
1314
1315 if (2 <= size) {
1316 str[0] = '[';
1317 str[size-1] = ']';
1318 }
1319
1320 return PyBytes_FromStringAndSize(str, size);
1321}
1322
1323static PyObject *
1324getargs_w_star(PyObject *self, PyObject *args)
1325{
1326 Py_buffer buffer;
1327 PyObject *result;
1328 char *str;
1329
1330 if (!PyArg_ParseTuple(args, "w*", &buffer))
1331 return NULL;
1332
1333 if (2 <= buffer.len) {
1334 str = buffer.buf;
1335 str[0] = '[';
1336 str[buffer.len-1] = ']';
1337 }
1338
1339 result = PyBytes_FromStringAndSize(buffer.buf, buffer.len);
1340 PyBuffer_Release(&buffer);
1341 return result;
1342}
1343
1344#ifdef Py_USING_UNICODE
1345
1346static int
1347_ustrlen(const Py_UNICODE *u)
1348{
1349 int i = 0;
1350 const Py_UNICODE *v = u;
1351 while (*v != 0) { i++; v++; }
1352 return i;
1353}
1354
1355static PyObject *
1356getargs_u(PyObject *self, PyObject *args)
1357{
1358 const Py_UNICODE *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001359 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001360 if (!PyArg_ParseTuple(args, "u", &str))
1361 return NULL;
1362 size = _ustrlen(str);
1363 return PyUnicode_FromUnicode(str, size);
1364}
1365
1366static PyObject *
1367getargs_u_hash(PyObject *self, PyObject *args)
1368{
1369 const Py_UNICODE *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001370 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001371 if (!PyArg_ParseTuple(args, "u#", &str, &size))
1372 return NULL;
1373 return PyUnicode_FromUnicode(str, size);
1374}
1375
1376#endif
1377
1378static PyObject *
1379getargs_es(PyObject *self, PyObject *args)
1380{
1381 PyObject *arg, *result;
1382 const char *encoding = NULL;
1383 char *str;
1384
1385 if (!PyArg_ParseTuple(args, "O|s", &arg, &encoding))
1386 return NULL;
1387 if (!PyArg_Parse(arg, "es", encoding, &str))
1388 return NULL;
1389 result = PyBytes_FromString(str);
1390 PyMem_Free(str);
1391 return result;
1392}
1393
1394static PyObject *
1395getargs_et(PyObject *self, PyObject *args)
1396{
1397 PyObject *arg, *result;
1398 const char *encoding = NULL;
1399 char *str;
1400
1401 if (!PyArg_ParseTuple(args, "O|s", &arg, &encoding))
1402 return NULL;
1403 if (!PyArg_Parse(arg, "et", encoding, &str))
1404 return NULL;
1405 result = PyBytes_FromString(str);
1406 PyMem_Free(str);
1407 return result;
1408}
1409
1410static PyObject *
1411getargs_es_hash(PyObject *self, PyObject *args)
1412{
1413 PyObject *arg, *result;
1414 const char *encoding = NULL;
1415 PyByteArrayObject *buffer = NULL;
1416 char *str = NULL;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001417 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001418
1419 if (!PyArg_ParseTuple(args, "O|sO!",
1420 &arg, &encoding, &PyByteArray_Type, &buffer))
1421 return NULL;
1422 if (buffer != NULL) {
1423 str = PyByteArray_AS_STRING(buffer);
1424 size = PyByteArray_GET_SIZE(buffer);
1425 }
1426 if (!PyArg_Parse(arg, "es#", encoding, &str, &size))
1427 return NULL;
1428 result = PyBytes_FromStringAndSize(str, size);
1429 if (buffer == NULL)
1430 PyMem_Free(str);
1431 return result;
1432}
1433
1434static PyObject *
1435getargs_et_hash(PyObject *self, PyObject *args)
1436{
1437 PyObject *arg, *result;
1438 const char *encoding = NULL;
1439 PyByteArrayObject *buffer = NULL;
1440 char *str = NULL;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001441 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001442
1443 if (!PyArg_ParseTuple(args, "O|sO!",
1444 &arg, &encoding, &PyByteArray_Type, &buffer))
1445 return NULL;
1446 if (buffer != NULL) {
1447 str = PyByteArray_AS_STRING(buffer);
1448 size = PyByteArray_GET_SIZE(buffer);
1449 }
1450 if (!PyArg_Parse(arg, "et#", encoding, &str, &size))
1451 return NULL;
1452 result = PyBytes_FromStringAndSize(str, size);
1453 if (buffer == NULL)
1454 PyMem_Free(str);
1455 return result;
1456}
1457
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001458#ifdef Py_USING_UNICODE
1459
Benjamin Petersone098c4a2008-12-23 20:12:33 +00001460static volatile int x;
Benjamin Petersonb6a53b52008-12-23 20:09:28 +00001461
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001462/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
1463 of an error.
1464*/
1465static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +00001466test_u_code(PyObject *self)
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001467{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001468 PyObject *tuple, *obj;
1469 Py_UNICODE *value;
1470 int len;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001471
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001472 /* issue4122: Undefined reference to _Py_ascii_whitespace on Windows */
1473 /* Just use the macro and check that it compiles */
1474 x = Py_UNICODE_ISSPACE(25);
Amaury Forgeot d'Arc07d539d2008-10-14 21:47:22 +00001475
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001476 tuple = PyTuple_New(1);
1477 if (tuple == NULL)
1478 return NULL;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001479
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001480 obj = PyUnicode_Decode("test", strlen("test"),
1481 "ascii", NULL);
1482 if (obj == NULL)
1483 return NULL;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001484
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001485 PyTuple_SET_ITEM(tuple, 0, obj);
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001486
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001487 value = 0;
1488 if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
1489 return NULL;
1490 if (value != PyUnicode_AS_UNICODE(obj))
1491 return raiseTestError("test_u_code",
1492 "u code returned wrong value for u'test'");
1493 value = 0;
1494 if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
1495 return NULL;
1496 if (value != PyUnicode_AS_UNICODE(obj) ||
1497 len != PyUnicode_GET_SIZE(obj))
1498 return raiseTestError("test_u_code",
1499 "u# code returned wrong values for u'test'");
Tim Petersbaefd9e2003-01-28 20:37:45 +00001500
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001501 Py_DECREF(tuple);
1502 Py_INCREF(Py_None);
1503 return Py_None;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001504}
1505
Tim Peters8931ff12006-05-13 23:28:20 +00001506static PyObject *
Mark Dickinson6b265f12009-03-18 16:07:26 +00001507test_widechar(PyObject *self)
1508{
1509#if defined(SIZEOF_WCHAR_T) && (SIZEOF_WCHAR_T == 4)
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001510 const wchar_t wtext[2] = {(wchar_t)0x10ABCDu};
1511 size_t wtextlen = 1;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001512#else
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001513 const wchar_t wtext[3] = {(wchar_t)0xDBEAu, (wchar_t)0xDFCDu};
1514 size_t wtextlen = 2;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001515#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001516 PyObject *wide, *utf8;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001517
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001518 wide = PyUnicode_FromWideChar(wtext, wtextlen);
1519 if (wide == NULL)
1520 return NULL;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001521
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001522 utf8 = PyUnicode_FromString("\xf4\x8a\xaf\x8d");
1523 if (utf8 == NULL) {
1524 Py_DECREF(wide);
1525 return NULL;
1526 }
Mark Dickinson6b265f12009-03-18 16:07:26 +00001527
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001528 if (PyUnicode_GET_SIZE(wide) != PyUnicode_GET_SIZE(utf8)) {
1529 Py_DECREF(wide);
1530 Py_DECREF(utf8);
1531 return raiseTestError("test_widechar",
1532 "wide string and utf8 string have different length");
1533 }
1534 if (PyUnicode_Compare(wide, utf8)) {
1535 Py_DECREF(wide);
1536 Py_DECREF(utf8);
1537 if (PyErr_Occurred())
1538 return NULL;
1539 return raiseTestError("test_widechar",
1540 "wide string and utf8 string are differents");
1541 }
Mark Dickinson6b265f12009-03-18 16:07:26 +00001542
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001543 Py_DECREF(wide);
1544 Py_DECREF(utf8);
1545 Py_RETURN_NONE;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001546}
1547
1548static PyObject *
Victor Stinner975134e2011-11-22 01:54:19 +01001549unicode_encodedecimal(PyObject *self, PyObject *args)
1550{
1551 Py_UNICODE *unicode;
Victor Stinner9f915d92011-11-29 00:53:09 +01001552 int length;
Victor Stinner975134e2011-11-22 01:54:19 +01001553 char *errors = NULL;
1554 PyObject *decimal;
1555 Py_ssize_t decimal_length, new_length;
1556 int res;
1557
1558 if (!PyArg_ParseTuple(args, "u#|s", &unicode, &length, &errors))
1559 return NULL;
1560
Serhiy Storchakae822b032013-08-06 16:56:26 +03001561 decimal_length = length * 10; /* len('&#1114111;') */
Victor Stinner975134e2011-11-22 01:54:19 +01001562 decimal = PyBytes_FromStringAndSize(NULL, decimal_length);
1563 if (decimal == NULL)
1564 return NULL;
1565
1566 res = PyUnicode_EncodeDecimal(unicode, length,
1567 PyBytes_AS_STRING(decimal),
1568 errors);
1569 if (res < 0) {
1570 Py_DECREF(decimal);
1571 return NULL;
1572 }
1573
1574 new_length = strlen(PyBytes_AS_STRING(decimal));
1575 assert(new_length <= decimal_length);
1576 res = _PyBytes_Resize(&decimal, new_length);
1577 if (res < 0)
1578 return NULL;
1579
1580 return decimal;
1581}
1582
1583static PyObject *
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001584test_empty_argparse(PyObject *self)
1585{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001586 /* Test that formats can begin with '|'. See issue #4720. */
1587 PyObject *tuple, *dict = NULL;
1588 static char *kwlist[] = {NULL};
1589 int result;
1590 tuple = PyTuple_New(0);
1591 if (!tuple)
1592 return NULL;
1593 if ((result = PyArg_ParseTuple(tuple, "|:test_empty_argparse")) < 0)
1594 goto done;
1595 dict = PyDict_New();
1596 if (!dict)
1597 goto done;
1598 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|:test_empty_argparse", kwlist);
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001599 done:
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001600 Py_DECREF(tuple);
1601 Py_XDECREF(dict);
1602 if (result < 0)
1603 return NULL;
1604 else {
1605 Py_RETURN_NONE;
1606 }
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001607}
1608
1609static PyObject *
Tim Peters8931ff12006-05-13 23:28:20 +00001610codec_incrementalencoder(PyObject *self, PyObject *args)
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001611{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001612 const char *encoding, *errors = NULL;
1613 if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
1614 &encoding, &errors))
1615 return NULL;
1616 return PyCodec_IncrementalEncoder(encoding, errors);
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001617}
1618
Tim Peters8931ff12006-05-13 23:28:20 +00001619static PyObject *
1620codec_incrementaldecoder(PyObject *self, PyObject *args)
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001621{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001622 const char *encoding, *errors = NULL;
1623 if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
1624 &encoding, &errors))
1625 return NULL;
1626 return PyCodec_IncrementalDecoder(encoding, errors);
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001627}
1628
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001629#endif
1630
Tim Peters5b8132f2003-01-31 15:52:05 +00001631/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
Tim Petersbaefd9e2003-01-28 20:37:45 +00001632static PyObject *
1633test_long_numbits(PyObject *self)
1634{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001635 struct triple {
1636 long input;
1637 size_t nbits;
1638 int sign;
1639 } testcases[] = {{0, 0, 0},
1640 {1L, 1, 1},
1641 {-1L, 1, -1},
1642 {2L, 2, 1},
1643 {-2L, 2, -1},
1644 {3L, 2, 1},
1645 {-3L, 2, -1},
1646 {4L, 3, 1},
1647 {-4L, 3, -1},
1648 {0x7fffL, 15, 1}, /* one Python long digit */
1649 {-0x7fffL, 15, -1},
1650 {0xffffL, 16, 1},
1651 {-0xffffL, 16, -1},
1652 {0xfffffffL, 28, 1},
1653 {-0xfffffffL, 28, -1}};
1654 int i;
Tim Petersbaefd9e2003-01-28 20:37:45 +00001655
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001656 for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
1657 PyObject *plong = PyLong_FromLong(testcases[i].input);
1658 size_t nbits = _PyLong_NumBits(plong);
1659 int sign = _PyLong_Sign(plong);
Tim Petersbaefd9e2003-01-28 20:37:45 +00001660
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001661 Py_DECREF(plong);
1662 if (nbits != testcases[i].nbits)
1663 return raiseTestError("test_long_numbits",
1664 "wrong result for _PyLong_NumBits");
1665 if (sign != testcases[i].sign)
1666 return raiseTestError("test_long_numbits",
1667 "wrong result for _PyLong_Sign");
1668 }
1669 Py_INCREF(Py_None);
1670 return Py_None;
Tim Petersbaefd9e2003-01-28 20:37:45 +00001671}
1672
Neal Norwitz4a53dad2006-03-15 05:43:10 +00001673/* Example passing NULLs to PyObject_Str(NULL) and PyObject_Unicode(NULL). */
1674
1675static PyObject *
1676test_null_strings(PyObject *self)
1677{
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001678 PyObject *o1 = PyObject_Str(NULL);
1679#ifdef Py_USING_UNICODE
1680 PyObject *o2 = PyObject_Unicode(NULL);
1681#else
1682 PyObject *o2 = PyObject_Str(NULL);
1683#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001684 PyObject *tuple = PyTuple_Pack(2, o1, o2);
1685 Py_XDECREF(o1);
1686 Py_XDECREF(o2);
1687 return tuple;
Neal Norwitz4a53dad2006-03-15 05:43:10 +00001688}
1689
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001690static PyObject *
1691raise_exception(PyObject *self, PyObject *args)
1692{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001693 PyObject *exc;
1694 PyObject *exc_args, *v;
1695 int num_args, i;
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001696
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001697 if (!PyArg_ParseTuple(args, "Oi:raise_exception",
1698 &exc, &num_args))
1699 return NULL;
1700 if (!PyExceptionClass_Check(exc)) {
1701 PyErr_Format(PyExc_TypeError, "an exception class is required");
1702 return NULL;
1703 }
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001704
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001705 exc_args = PyTuple_New(num_args);
1706 if (exc_args == NULL)
1707 return NULL;
1708 for (i = 0; i < num_args; ++i) {
1709 v = PyInt_FromLong(i);
1710 if (v == NULL) {
1711 Py_DECREF(exc_args);
1712 return NULL;
1713 }
1714 PyTuple_SET_ITEM(exc_args, i, v);
1715 }
1716 PyErr_SetObject(exc, exc_args);
1717 Py_DECREF(exc_args);
1718 return NULL;
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001719}
Tim Peters91621db2001-06-12 20:10:01 +00001720
Antoine Pitrou0d2fac12015-06-23 14:31:11 +02001721static PyObject *
1722set_errno(PyObject *self, PyObject *args)
1723{
1724 int new_errno;
1725
1726 if (!PyArg_ParseTuple(args, "i:set_errno", &new_errno))
1727 return NULL;
1728
1729 errno = new_errno;
1730 Py_RETURN_NONE;
1731}
Benjamin Peterson46bff792010-01-30 23:28:38 +00001732
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001733#ifdef Py_USING_UNICODE
Benjamin Peterson46bff792010-01-30 23:28:38 +00001734static int test_run_counter = 0;
1735
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001736static PyObject *
1737test_datetime_capi(PyObject *self, PyObject *args) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001738 if (PyDateTimeAPI) {
1739 if (test_run_counter) {
1740 /* Probably regrtest.py -R */
1741 Py_RETURN_NONE;
1742 }
1743 else {
1744 PyErr_SetString(PyExc_AssertionError,
1745 "PyDateTime_CAPI somehow initialized");
1746 return NULL;
1747 }
1748 }
1749 test_run_counter++;
1750 PyDateTime_IMPORT;
1751 if (PyDateTimeAPI)
1752 Py_RETURN_NONE;
1753 else
1754 return NULL;
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001755}
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001756#endif
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001757
Benjamin Petersona04ae012010-01-30 23:26:05 +00001758
1759#ifdef WITH_THREAD
1760
Tim Peters59b96c12006-03-21 03:58:41 +00001761/* test_thread_state spawns a thread of its own, and that thread releases
1762 * `thread_done` when it's finished. The driver code has to know when the
1763 * thread finishes, because the thread uses a PyObject (the callable) that
1764 * may go away when the driver finishes. The former lack of this explicit
1765 * synchronization caused rare segfaults, so rare that they were seen only
1766 * on a Mac buildbot (although they were possible on any box).
1767 */
1768static PyThread_type_lock thread_done = NULL;
1769
Benjamin Peterson37346b22008-08-23 20:27:43 +00001770static int
Tim Peters59b96c12006-03-21 03:58:41 +00001771_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001772{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001773 PyObject *rc;
1774 int success;
1775 PyGILState_STATE s = PyGILState_Ensure();
1776 rc = PyObject_CallFunction((PyObject *)callable, "");
1777 success = (rc != NULL);
1778 Py_XDECREF(rc);
1779 PyGILState_Release(s);
1780 return success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001781}
1782
Tim Peters59b96c12006-03-21 03:58:41 +00001783/* Same thing, but releases `thread_done` when it returns. This variant
1784 * should be called only from threads spawned by test_thread_state().
1785 */
1786static void
1787_make_call_from_thread(void *callable)
1788{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001789 _make_call(callable);
1790 PyThread_release_lock(thread_done);
Tim Peters59b96c12006-03-21 03:58:41 +00001791}
1792
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001793static PyObject *
1794test_thread_state(PyObject *self, PyObject *args)
1795{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001796 PyObject *fn;
1797 int success = 1;
Tim Peters59b96c12006-03-21 03:58:41 +00001798
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001799 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
1800 return NULL;
Tim Peters59b96c12006-03-21 03:58:41 +00001801
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001802 if (!PyCallable_Check(fn)) {
1803 PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
1804 fn->ob_type->tp_name);
1805 return NULL;
1806 }
Benjamin Peterson37346b22008-08-23 20:27:43 +00001807
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001808 /* Ensure Python is set up for threading */
1809 PyEval_InitThreads();
1810 thread_done = PyThread_allocate_lock();
1811 if (thread_done == NULL)
1812 return PyErr_NoMemory();
1813 PyThread_acquire_lock(thread_done, 1);
Tim Peters59b96c12006-03-21 03:58:41 +00001814
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001815 /* Start a new thread with our callback. */
1816 PyThread_start_new_thread(_make_call_from_thread, fn);
1817 /* Make the callback with the thread lock held by this thread */
1818 success &= _make_call(fn);
1819 /* Do it all again, but this time with the thread-lock released */
1820 Py_BEGIN_ALLOW_THREADS
1821 success &= _make_call(fn);
1822 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
1823 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001824
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001825 /* And once more with and without a thread
1826 XXX - should use a lock and work out exactly what we are trying
1827 to test <wink>
1828 */
1829 Py_BEGIN_ALLOW_THREADS
1830 PyThread_start_new_thread(_make_call_from_thread, fn);
1831 success &= _make_call(fn);
1832 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
1833 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001834
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001835 /* Release lock we acquired above. This is required on HP-UX. */
1836 PyThread_release_lock(thread_done);
Neal Norwitz97a57222006-10-28 21:17:51 +00001837
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001838 PyThread_free_lock(thread_done);
1839 if (!success)
1840 return NULL;
1841 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001842}
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001843
1844/* test Py_AddPendingCalls using threads */
1845static int _pending_callback(void *arg)
1846{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001847 /* we assume the argument is callable object to which we own a reference */
1848 PyObject *callable = (PyObject *)arg;
1849 PyObject *r = PyObject_CallObject(callable, NULL);
1850 Py_DECREF(callable);
1851 Py_XDECREF(r);
1852 return r != NULL ? 0 : -1;
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001853}
1854
1855/* The following requests n callbacks to _pending_callback. It can be
1856 * run from any python thread.
1857 */
1858PyObject *pending_threadfunc(PyObject *self, PyObject *arg)
1859{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001860 PyObject *callable;
1861 int r;
1862 if (PyArg_ParseTuple(arg, "O", &callable) == 0)
1863 return NULL;
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001864
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001865 /* create the reference for the callbackwhile we hold the lock */
1866 Py_INCREF(callable);
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001867
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001868 Py_BEGIN_ALLOW_THREADS
1869 r = Py_AddPendingCall(&_pending_callback, callable);
1870 Py_END_ALLOW_THREADS
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001871
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001872 if (r<0) {
1873 Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
1874 Py_INCREF(Py_False);
1875 return Py_False;
1876 }
1877 Py_INCREF(Py_True);
1878 return Py_True;
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001879}
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001880#endif
1881
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001882/* Some tests of PyString_FromFormat(). This needs more tests. */
Tim Peters8931ff12006-05-13 23:28:20 +00001883static PyObject *
1884test_string_from_format(PyObject *self, PyObject *args)
1885{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001886 PyObject *result;
1887 char *msg;
Tim Peters8931ff12006-05-13 23:28:20 +00001888
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001889#define CHECK_1_FORMAT(FORMAT, TYPE) \
1890 result = PyString_FromFormat(FORMAT, (TYPE)1); \
1891 if (result == NULL) \
1892 return NULL; \
1893 if (strcmp(PyString_AsString(result), "1")) { \
1894 msg = FORMAT " failed at 1"; \
1895 goto Fail; \
1896 } \
1897 Py_DECREF(result)
Tim Peters8931ff12006-05-13 23:28:20 +00001898
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001899 CHECK_1_FORMAT("%d", int);
1900 CHECK_1_FORMAT("%ld", long);
1901 /* The z width modifier was added in Python 2.5. */
1902 CHECK_1_FORMAT("%zd", Py_ssize_t);
Tim Peters8931ff12006-05-13 23:28:20 +00001903
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001904 /* The u type code was added in Python 2.5. */
1905 CHECK_1_FORMAT("%u", unsigned int);
1906 CHECK_1_FORMAT("%lu", unsigned long);
1907 CHECK_1_FORMAT("%zu", size_t);
Tim Peters8931ff12006-05-13 23:28:20 +00001908
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001909 /* "%lld" and "%llu" support added in Python 2.7. */
Mark Dickinson82864d12009-11-15 16:18:58 +00001910#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001911 CHECK_1_FORMAT("%llu", unsigned PY_LONG_LONG);
1912 CHECK_1_FORMAT("%lld", PY_LONG_LONG);
Mark Dickinson82864d12009-11-15 16:18:58 +00001913#endif
1914
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001915 Py_RETURN_NONE;
Tim Peters8931ff12006-05-13 23:28:20 +00001916
1917 Fail:
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001918 Py_XDECREF(result);
1919 return raiseTestError("test_string_from_format", msg);
Tim Peters8931ff12006-05-13 23:28:20 +00001920
1921#undef CHECK_1_FORMAT
1922}
1923
Larry Hastings402b73f2010-03-25 00:54:54 +00001924/* Coverage testing of capsule objects. */
1925
1926static const char *capsule_name = "capsule name";
1927static char *capsule_pointer = "capsule pointer";
1928static char *capsule_context = "capsule context";
1929static const char *capsule_error = NULL;
1930static int
1931capsule_destructor_call_count = 0;
1932
1933static void
1934capsule_destructor(PyObject *o) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001935 capsule_destructor_call_count++;
1936 if (PyCapsule_GetContext(o) != capsule_context) {
1937 capsule_error = "context did not match in destructor!";
1938 } else if (PyCapsule_GetDestructor(o) != capsule_destructor) {
1939 capsule_error = "destructor did not match in destructor! (woah!)";
1940 } else if (PyCapsule_GetName(o) != capsule_name) {
1941 capsule_error = "name did not match in destructor!";
1942 } else if (PyCapsule_GetPointer(o, capsule_name) != capsule_pointer) {
1943 capsule_error = "pointer did not match in destructor!";
1944 }
Larry Hastings402b73f2010-03-25 00:54:54 +00001945}
1946
1947typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001948 char *name;
1949 char *module;
1950 char *attribute;
Larry Hastings402b73f2010-03-25 00:54:54 +00001951} known_capsule;
1952
1953static PyObject *
1954test_capsule(PyObject *self, PyObject *args)
1955{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001956 PyObject *object;
1957 const char *error = NULL;
1958 void *pointer;
1959 void *pointer2;
1960 known_capsule known_capsules[] = {
1961 #define KNOWN_CAPSULE(module, name) { module "." name, module, name }
1962 KNOWN_CAPSULE("_socket", "CAPI"),
1963 KNOWN_CAPSULE("_curses", "_C_API"),
1964 KNOWN_CAPSULE("datetime", "datetime_CAPI"),
1965 { NULL, NULL },
1966 };
1967 known_capsule *known = &known_capsules[0];
Larry Hastings402b73f2010-03-25 00:54:54 +00001968
1969#define FAIL(x) { error = (x); goto exit; }
1970
1971#define CHECK_DESTRUCTOR \
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001972 if (capsule_error) { \
1973 FAIL(capsule_error); \
1974 } \
1975 else if (!capsule_destructor_call_count) { \
1976 FAIL("destructor not called!"); \
1977 } \
1978 capsule_destructor_call_count = 0; \
Larry Hastings402b73f2010-03-25 00:54:54 +00001979
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001980 object = PyCapsule_New(capsule_pointer, capsule_name, capsule_destructor);
1981 PyCapsule_SetContext(object, capsule_context);
1982 capsule_destructor(object);
1983 CHECK_DESTRUCTOR;
1984 Py_DECREF(object);
1985 CHECK_DESTRUCTOR;
Larry Hastings402b73f2010-03-25 00:54:54 +00001986
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001987 object = PyCapsule_New(known, "ignored", NULL);
1988 PyCapsule_SetPointer(object, capsule_pointer);
1989 PyCapsule_SetName(object, capsule_name);
1990 PyCapsule_SetDestructor(object, capsule_destructor);
1991 PyCapsule_SetContext(object, capsule_context);
1992 capsule_destructor(object);
1993 CHECK_DESTRUCTOR;
1994 /* intentionally access using the wrong name */
1995 pointer2 = PyCapsule_GetPointer(object, "the wrong name");
1996 if (!PyErr_Occurred()) {
1997 FAIL("PyCapsule_GetPointer should have failed but did not!");
1998 }
1999 PyErr_Clear();
2000 if (pointer2) {
2001 if (pointer2 == capsule_pointer) {
2002 FAIL("PyCapsule_GetPointer should not have"
2003 " returned the internal pointer!");
2004 } else {
2005 FAIL("PyCapsule_GetPointer should have "
2006 "returned NULL pointer but did not!");
2007 }
2008 }
2009 PyCapsule_SetDestructor(object, NULL);
2010 Py_DECREF(object);
2011 if (capsule_destructor_call_count) {
2012 FAIL("destructor called when it should not have been!");
2013 }
Larry Hastings402b73f2010-03-25 00:54:54 +00002014
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002015 for (known = &known_capsules[0]; known->module != NULL; known++) {
2016 /* yeah, ordinarily I wouldn't do this either,
2017 but it's fine for this test harness.
2018 */
2019 static char buffer[256];
Larry Hastings402b73f2010-03-25 00:54:54 +00002020#undef FAIL
2021#define FAIL(x) \
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002022 { \
2023 sprintf(buffer, "%s module: \"%s\" attribute: \"%s\"", \
2024 x, known->module, known->attribute); \
2025 error = buffer; \
2026 goto exit; \
2027 } \
Larry Hastings402b73f2010-03-25 00:54:54 +00002028
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002029 PyObject *module = PyImport_ImportModule(known->module);
2030 if (module) {
2031 pointer = PyCapsule_Import(known->name, 0);
2032 if (!pointer) {
2033 Py_DECREF(module);
2034 FAIL("PyCapsule_GetPointer returned NULL unexpectedly!");
2035 }
2036 object = PyObject_GetAttrString(module, known->attribute);
2037 if (!object) {
2038 Py_DECREF(module);
2039 return NULL;
2040 }
2041 pointer2 = PyCapsule_GetPointer(object,
2042 "weebles wobble but they don't fall down");
2043 if (!PyErr_Occurred()) {
2044 Py_DECREF(object);
2045 Py_DECREF(module);
2046 FAIL("PyCapsule_GetPointer should have failed but did not!");
2047 }
2048 PyErr_Clear();
2049 if (pointer2) {
2050 Py_DECREF(module);
2051 Py_DECREF(object);
2052 if (pointer2 == pointer) {
2053 FAIL("PyCapsule_GetPointer should not have"
2054 " returned its internal pointer!");
2055 } else {
2056 FAIL("PyCapsule_GetPointer should have"
2057 " returned NULL pointer but did not!");
2058 }
2059 }
2060 Py_DECREF(object);
2061 Py_DECREF(module);
2062 }
2063 else
2064 PyErr_Clear();
2065 }
Larry Hastings402b73f2010-03-25 00:54:54 +00002066
2067 exit:
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002068 if (error) {
2069 return raiseTestError("test_capsule", error);
2070 }
2071 Py_RETURN_NONE;
Larry Hastings402b73f2010-03-25 00:54:54 +00002072#undef FAIL
2073}
2074
Georg Brandle9462c72006-08-04 18:03:37 +00002075/* This is here to provide a docstring for test_descr. */
2076static PyObject *
2077test_with_docstring(PyObject *self)
2078{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002079 Py_RETURN_NONE;
Georg Brandle9462c72006-08-04 18:03:37 +00002080}
2081
Brett Cannon141534e2008-04-28 03:23:50 +00002082/* To test the format of tracebacks as printed out. */
2083static PyObject *
Brett Cannon8dc43032008-04-28 04:50:06 +00002084traceback_print(PyObject *self, PyObject *args)
Brett Cannon141534e2008-04-28 03:23:50 +00002085{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002086 PyObject *file;
2087 PyObject *traceback;
2088 int result;
2089
2090 if (!PyArg_ParseTuple(args, "OO:traceback_print",
2091 &traceback, &file))
2092 return NULL;
2093
2094 result = PyTraceBack_Print(traceback, file);
2095 if (result < 0)
2096 return NULL;
2097 Py_RETURN_NONE;
Brett Cannon141534e2008-04-28 03:23:50 +00002098}
2099
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00002100/* To test that the result of PyCode_NewEmpty has the right members. */
2101static PyObject *
2102code_newempty(PyObject *self, PyObject *args)
2103{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002104 const char *filename;
2105 const char *funcname;
2106 int firstlineno;
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00002107
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002108 if (!PyArg_ParseTuple(args, "ssi:code_newempty",
2109 &filename, &funcname, &firstlineno))
2110 return NULL;
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00002111
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002112 return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00002113}
2114
Georg Brandl740cdc32009-12-28 08:34:58 +00002115/* Test PyErr_NewExceptionWithDoc (also exercise PyErr_NewException).
2116 Run via Lib/test/test_exceptions.py */
2117static PyObject *
2118make_exception_with_doc(PyObject *self, PyObject *args, PyObject *kwargs)
2119{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002120 char *name;
2121 char *doc = NULL;
2122 PyObject *base = NULL;
2123 PyObject *dict = NULL;
Georg Brandl740cdc32009-12-28 08:34:58 +00002124
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002125 static char *kwlist[] = {"name", "doc", "base", "dict", NULL};
Georg Brandl740cdc32009-12-28 08:34:58 +00002126
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002127 if (!PyArg_ParseTupleAndKeywords(args, kwargs,
2128 "s|sOO:make_exception_with_doc", kwlist,
2129 &name, &doc, &base, &dict))
2130 return NULL;
Georg Brandl740cdc32009-12-28 08:34:58 +00002131
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002132 return PyErr_NewExceptionWithDoc(name, doc, base, dict);
Georg Brandl740cdc32009-12-28 08:34:58 +00002133}
2134
Benjamin Petersona7b09762011-10-15 13:43:21 -04002135static PyObject *
2136sequence_delitem(PyObject *self, PyObject *args)
2137{
2138 PyObject *seq;
2139 Py_ssize_t i;
2140
2141 if (!PyArg_ParseTuple(args, "On", &seq, &i))
2142 return NULL;
2143 if (PySequence_DelItem(seq, i) < 0)
2144 return NULL;
2145 Py_RETURN_NONE;
2146}
2147
Victor Stinner66c6e9d2013-12-13 02:37:09 +01002148#ifdef WITH_THREAD
2149typedef struct {
2150 PyThread_type_lock start_event;
2151 PyThread_type_lock exit_event;
2152 PyObject *callback;
2153} test_c_thread_t;
2154
2155static void
2156temporary_c_thread(void *data)
2157{
2158 test_c_thread_t *test_c_thread = data;
2159 PyGILState_STATE state;
2160 PyObject *res;
2161
2162 PyThread_release_lock(test_c_thread->start_event);
2163
2164 /* Allocate a Python thread state for this thread */
2165 state = PyGILState_Ensure();
2166
2167 res = PyObject_CallFunction(test_c_thread->callback, "", NULL);
2168 Py_CLEAR(test_c_thread->callback);
2169
2170 if (res == NULL) {
2171 PyErr_Print();
2172 }
2173 else {
2174 Py_DECREF(res);
2175 }
2176
2177 /* Destroy the Python thread state for this thread */
2178 PyGILState_Release(state);
2179
2180 PyThread_release_lock(test_c_thread->exit_event);
2181
2182 PyThread_exit_thread();
2183}
2184
2185static PyObject *
2186call_in_temporary_c_thread(PyObject *self, PyObject *callback)
2187{
2188 PyObject *res = NULL;
2189 test_c_thread_t test_c_thread;
2190 long thread;
2191
2192 PyEval_InitThreads();
2193
2194 test_c_thread.start_event = PyThread_allocate_lock();
2195 test_c_thread.exit_event = PyThread_allocate_lock();
2196 test_c_thread.callback = NULL;
2197 if (!test_c_thread.start_event || !test_c_thread.exit_event) {
2198 PyErr_SetString(PyExc_RuntimeError, "could not allocate lock");
2199 goto exit;
2200 }
2201
2202 Py_INCREF(callback);
2203 test_c_thread.callback = callback;
2204
2205 PyThread_acquire_lock(test_c_thread.start_event, 1);
2206 PyThread_acquire_lock(test_c_thread.exit_event, 1);
2207
2208 thread = PyThread_start_new_thread(temporary_c_thread, &test_c_thread);
2209 if (thread == -1) {
2210 PyErr_SetString(PyExc_RuntimeError, "unable to start the thread");
2211 PyThread_release_lock(test_c_thread.start_event);
2212 PyThread_release_lock(test_c_thread.exit_event);
2213 goto exit;
2214 }
2215
2216 PyThread_acquire_lock(test_c_thread.start_event, 1);
2217 PyThread_release_lock(test_c_thread.start_event);
2218
2219 Py_BEGIN_ALLOW_THREADS
2220 PyThread_acquire_lock(test_c_thread.exit_event, 1);
2221 PyThread_release_lock(test_c_thread.exit_event);
2222 Py_END_ALLOW_THREADS
2223
2224 Py_INCREF(Py_None);
2225 res = Py_None;
2226
2227exit:
2228 Py_CLEAR(test_c_thread.callback);
2229 if (test_c_thread.start_event)
2230 PyThread_free_lock(test_c_thread.start_event);
2231 if (test_c_thread.exit_event)
2232 PyThread_free_lock(test_c_thread.exit_event);
2233 return res;
2234}
2235#endif /* WITH_THREAD */
2236
Serhiy Storchaka4068b012015-02-06 08:56:33 +02002237/* marshal */
2238
2239static PyObject*
2240pymarshal_write_long_to_file(PyObject* self, PyObject *args)
2241{
2242 long value;
2243 char *filename;
2244 int version;
2245 FILE *fp;
2246
2247 if (!PyArg_ParseTuple(args, "lsi:pymarshal_write_long_to_file",
2248 &value, &filename, &version))
2249 return NULL;
2250
2251 fp = fopen(filename, "wb");
2252 if (fp == NULL) {
2253 PyErr_SetFromErrno(PyExc_OSError);
2254 return NULL;
2255 }
2256
2257 PyMarshal_WriteLongToFile(value, fp, version);
2258
2259 fclose(fp);
2260 if (PyErr_Occurred())
2261 return NULL;
2262 Py_RETURN_NONE;
2263}
2264
2265static PyObject*
2266pymarshal_write_object_to_file(PyObject* self, PyObject *args)
2267{
2268 PyObject *obj;
2269 char *filename;
2270 int version;
2271 FILE *fp;
2272
2273 if (!PyArg_ParseTuple(args, "Osi:pymarshal_write_object_to_file",
2274 &obj, &filename, &version))
2275 return NULL;
2276
2277 fp = fopen(filename, "wb");
2278 if (fp == NULL) {
2279 PyErr_SetFromErrno(PyExc_OSError);
2280 return NULL;
2281 }
2282
2283 PyMarshal_WriteObjectToFile(obj, fp, version);
2284
2285 fclose(fp);
2286 if (PyErr_Occurred())
2287 return NULL;
2288 Py_RETURN_NONE;
2289}
2290
2291static PyObject*
2292pymarshal_read_short_from_file(PyObject* self, PyObject *args)
2293{
2294 int value;
2295 long pos;
2296 char *filename;
2297 FILE *fp;
2298
2299 if (!PyArg_ParseTuple(args, "s:pymarshal_read_short_from_file", &filename))
2300 return NULL;
2301
2302 fp = fopen(filename, "rb");
2303 if (fp == NULL) {
2304 PyErr_SetFromErrno(PyExc_OSError);
2305 return NULL;
2306 }
2307
2308 value = PyMarshal_ReadShortFromFile(fp);
2309 pos = ftell(fp);
2310
2311 fclose(fp);
2312 if (PyErr_Occurred())
2313 return NULL;
2314 return Py_BuildValue("il", value, pos);
2315}
2316
2317static PyObject*
2318pymarshal_read_long_from_file(PyObject* self, PyObject *args)
2319{
2320 long value, pos;
2321 char *filename;
2322 FILE *fp;
2323
2324 if (!PyArg_ParseTuple(args, "s:pymarshal_read_long_from_file", &filename))
2325 return NULL;
2326
2327 fp = fopen(filename, "rb");
2328 if (fp == NULL) {
2329 PyErr_SetFromErrno(PyExc_OSError);
2330 return NULL;
2331 }
2332
2333 value = PyMarshal_ReadLongFromFile(fp);
2334 pos = ftell(fp);
2335
2336 fclose(fp);
2337 if (PyErr_Occurred())
2338 return NULL;
2339 return Py_BuildValue("ll", value, pos);
2340}
2341
2342static PyObject*
2343pymarshal_read_last_object_from_file(PyObject* self, PyObject *args)
2344{
2345 PyObject *obj;
2346 long pos;
2347 char *filename;
2348 FILE *fp;
2349
2350 if (!PyArg_ParseTuple(args, "s:pymarshal_read_last_object_from_file", &filename))
2351 return NULL;
2352
2353 fp = fopen(filename, "rb");
2354 if (fp == NULL) {
2355 PyErr_SetFromErrno(PyExc_OSError);
2356 return NULL;
2357 }
2358
2359 obj = PyMarshal_ReadLastObjectFromFile(fp);
2360 pos = ftell(fp);
2361
2362 fclose(fp);
2363 return Py_BuildValue("Nl", obj, pos);
2364}
2365
2366static PyObject*
2367pymarshal_read_object_from_file(PyObject* self, PyObject *args)
2368{
2369 PyObject *obj;
2370 long pos;
2371 char *filename;
2372 FILE *fp;
2373
2374 if (!PyArg_ParseTuple(args, "s:pymarshal_read_object_from_file", &filename))
2375 return NULL;
2376
2377 fp = fopen(filename, "rb");
2378 if (fp == NULL) {
2379 PyErr_SetFromErrno(PyExc_OSError);
2380 return NULL;
2381 }
2382
2383 obj = PyMarshal_ReadObjectFromFile(fp);
2384 pos = ftell(fp);
2385
2386 fclose(fp);
2387 return Py_BuildValue("Nl", obj, pos);
2388}
2389
Victor Stinner66c6e9d2013-12-13 02:37:09 +01002390
Tim Peters9ea17ac2001-02-02 05:57:15 +00002391static PyMethodDef TestMethods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002392 {"raise_exception", raise_exception, METH_VARARGS},
Antoine Pitrou0d2fac12015-06-23 14:31:11 +02002393 {"set_errno", set_errno, METH_VARARGS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002394 {"test_config", (PyCFunction)test_config, METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002395#ifdef Py_USING_UNICODE
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002396 {"test_datetime_capi", test_datetime_capi, METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002397#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002398 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
2399 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
2400 {"test_lazy_hash_inheritance", (PyCFunction)test_lazy_hash_inheritance,METH_NOARGS},
2401 {"test_broken_memoryview", (PyCFunction)test_broken_memoryview,METH_NOARGS},
Stefan Krah5606cd92015-01-30 20:11:10 +01002402 {"test_to_contiguous", (PyCFunction)test_to_contiguous, METH_NOARGS},
Stefan Krah0a7fc532015-02-01 16:10:35 +01002403 {"test_from_contiguous", (PyCFunction)test_from_contiguous, METH_NOARGS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002404 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
2405 {"test_long_and_overflow", (PyCFunction)test_long_and_overflow,
2406 METH_NOARGS},
2407 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
2408 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002409#ifdef Py_USING_UNICODE
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002410 {"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002411#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002412 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
2413 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
2414 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
2415 PyDoc_STR("This is a pretty normal docstring.")},
Thomas Heller3457e4b2003-04-24 16:14:27 +00002416
Serhiy Storchakac6a7c962016-05-16 09:55:32 +03002417 {"get_args", get_args, METH_VARARGS},
2418 {"get_kwargs", (PyCFunction)get_kwargs, METH_VARARGS|METH_KEYWORDS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002419 {"getargs_tuple", getargs_tuple, METH_VARARGS},
2420 {"getargs_keywords", (PyCFunction)getargs_keywords,
2421 METH_VARARGS|METH_KEYWORDS},
2422 {"getargs_b", getargs_b, METH_VARARGS},
2423 {"getargs_B", getargs_B, METH_VARARGS},
2424 {"getargs_h", getargs_h, METH_VARARGS},
2425 {"getargs_H", getargs_H, METH_VARARGS},
2426 {"getargs_I", getargs_I, METH_VARARGS},
2427 {"getargs_k", getargs_k, METH_VARARGS},
2428 {"getargs_i", getargs_i, METH_VARARGS},
2429 {"getargs_l", getargs_l, METH_VARARGS},
2430 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00002431#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002432 {"getargs_L", getargs_L, METH_VARARGS},
2433 {"getargs_K", getargs_K, METH_VARARGS},
2434 {"test_longlong_api", test_longlong_api, METH_NOARGS},
2435 {"test_long_long_and_overflow",
2436 (PyCFunction)test_long_long_and_overflow, METH_NOARGS},
2437 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002438#endif
Serhiy Storchakad0dc72b2016-05-16 10:12:02 +03002439 {"getargs_f", getargs_f, METH_VARARGS},
2440 {"getargs_d", getargs_d, METH_VARARGS},
2441 {"getargs_D", getargs_D, METH_VARARGS},
2442 {"getargs_S", getargs_S, METH_VARARGS},
2443 {"getargs_Y", getargs_Y, METH_VARARGS},
2444#ifdef Py_USING_UNICODE
2445 {"getargs_U", getargs_U, METH_VARARGS},
2446#endif
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002447 {"getargs_c", getargs_c, METH_VARARGS},
2448 {"getargs_s", getargs_s, METH_VARARGS},
2449 {"getargs_s_star", getargs_s_star, METH_VARARGS},
2450 {"getargs_s_hash", getargs_s_hash, METH_VARARGS},
2451 {"getargs_t_hash", getargs_t_hash, METH_VARARGS},
2452 {"getargs_z", getargs_z, METH_VARARGS},
2453 {"getargs_z_star", getargs_z_star, METH_VARARGS},
2454 {"getargs_z_hash", getargs_z_hash, METH_VARARGS},
2455 {"getargs_w", getargs_w, METH_VARARGS},
2456 {"getargs_w_star", getargs_w_star, METH_VARARGS},
2457 {"getargs_w_hash", getargs_w_hash, METH_VARARGS},
2458#ifdef Py_USING_UNICODE
2459 {"getargs_u", getargs_u, METH_VARARGS},
2460 {"getargs_u_hash", getargs_u_hash, METH_VARARGS},
2461#endif
2462 {"getargs_es", getargs_es, METH_VARARGS},
2463 {"getargs_et", getargs_et, METH_VARARGS},
2464 {"getargs_es_hash", getargs_es_hash, METH_VARARGS},
2465 {"getargs_et_hash", getargs_et_hash, METH_VARARGS},
2466#ifdef Py_USING_UNICODE
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002467 {"codec_incrementalencoder",
2468 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
2469 {"codec_incrementaldecoder",
2470 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002471 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
2472 {"test_widechar", (PyCFunction)test_widechar, METH_NOARGS},
Victor Stinner975134e2011-11-22 01:54:19 +01002473 {"unicode_encodedecimal", unicode_encodedecimal, METH_VARARGS},
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00002474#endif
Mark Hammond8d98d2c2003-04-19 15:41:53 +00002475#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002476 {"_test_thread_state", test_thread_state, METH_VARARGS},
2477 {"_pending_threadfunc", pending_threadfunc, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00002478#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002479 {"test_capsule", (PyCFunction)test_capsule, METH_NOARGS},
2480 {"traceback_print", traceback_print, METH_VARARGS},
2481 {"code_newempty", code_newempty, METH_VARARGS},
2482 {"make_exception_with_doc", (PyCFunction)make_exception_with_doc,
2483 METH_VARARGS | METH_KEYWORDS},
Benjamin Petersona7b09762011-10-15 13:43:21 -04002484 {"sequence_delitem", (PyCFunction)sequence_delitem, METH_VARARGS},
Victor Stinner66c6e9d2013-12-13 02:37:09 +01002485#ifdef WITH_THREAD
2486 {"call_in_temporary_c_thread", call_in_temporary_c_thread, METH_O,
2487 PyDoc_STR("set_error_class(error_class) -> None")},
2488#endif
Serhiy Storchaka4068b012015-02-06 08:56:33 +02002489 {"pymarshal_write_long_to_file",
2490 pymarshal_write_long_to_file, METH_VARARGS},
2491 {"pymarshal_write_object_to_file",
2492 pymarshal_write_object_to_file, METH_VARARGS},
2493 {"pymarshal_read_short_from_file",
2494 pymarshal_read_short_from_file, METH_VARARGS},
2495 {"pymarshal_read_long_from_file",
2496 pymarshal_read_long_from_file, METH_VARARGS},
2497 {"pymarshal_read_last_object_from_file",
2498 pymarshal_read_last_object_from_file, METH_VARARGS},
2499 {"pymarshal_read_object_from_file",
2500 pymarshal_read_object_from_file, METH_VARARGS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002501 {NULL, NULL} /* sentinel */
Tim Peters9ea17ac2001-02-02 05:57:15 +00002502};
2503
Thomas Hellera4ea6032003-04-17 18:55:45 +00002504#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
2505
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002506typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002507 char bool_member;
2508 char byte_member;
2509 unsigned char ubyte_member;
2510 short short_member;
2511 unsigned short ushort_member;
2512 int int_member;
2513 unsigned int uint_member;
2514 long long_member;
2515 unsigned long ulong_member;
2516 float float_member;
2517 double double_member;
2518 char inplace_member[6];
Martin v. Löwis6371cd82007-06-09 07:42:52 +00002519#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002520 PY_LONG_LONG longlong_member;
2521 unsigned PY_LONG_LONG ulonglong_member;
Martin v. Löwis6371cd82007-06-09 07:42:52 +00002522#endif
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002523} all_structmembers;
2524
2525typedef struct {
2526 PyObject_HEAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002527 all_structmembers structmembers;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002528} test_structmembers;
2529
2530static struct PyMemberDef test_members[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002531 {"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
2532 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
2533 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
2534 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
2535 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
2536 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
2537 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
2538 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
2539 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
2540 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
2541 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
2542 {"T_STRING_INPLACE", T_STRING_INPLACE, offsetof(test_structmembers, structmembers.inplace_member), 0, NULL},
Martin v. Löwis6371cd82007-06-09 07:42:52 +00002543#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002544 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
2545 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
Martin v. Löwis6371cd82007-06-09 07:42:52 +00002546#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002547 {NULL}
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002548};
2549
2550
Georg Brandlc09b94e2008-01-21 21:28:32 +00002551static PyObject *
2552test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
2553{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002554 static char *keywords[] = {
2555 "T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
2556 "T_INT", "T_UINT", "T_LONG", "T_ULONG",
2557 "T_FLOAT", "T_DOUBLE", "T_STRING_INPLACE",
Georg Brandlc09b94e2008-01-21 21:28:32 +00002558#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002559 "T_LONGLONG", "T_ULONGLONG",
Georg Brandlc09b94e2008-01-21 21:28:32 +00002560#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002561 NULL};
2562 static char *fmt = "|bbBhHiIlkfds#"
Georg Brandlc09b94e2008-01-21 21:28:32 +00002563#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002564 "LK"
Georg Brandlc09b94e2008-01-21 21:28:32 +00002565#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002566 ;
2567 test_structmembers *ob;
2568 const char *s = NULL;
Antoine Pitrou1e181022013-05-08 02:07:13 +02002569 int string_len = 0;
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002570 ob = PyObject_New(test_structmembers, type);
2571 if (ob == NULL)
2572 return NULL;
2573 memset(&ob->structmembers, 0, sizeof(all_structmembers));
2574 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
2575 &ob->structmembers.bool_member,
2576 &ob->structmembers.byte_member,
2577 &ob->structmembers.ubyte_member,
2578 &ob->structmembers.short_member,
2579 &ob->structmembers.ushort_member,
2580 &ob->structmembers.int_member,
2581 &ob->structmembers.uint_member,
2582 &ob->structmembers.long_member,
2583 &ob->structmembers.ulong_member,
2584 &ob->structmembers.float_member,
2585 &ob->structmembers.double_member,
2586 &s, &string_len
2587#ifdef HAVE_LONG_LONG
2588 , &ob->structmembers.longlong_member,
2589 &ob->structmembers.ulonglong_member
2590#endif
2591 )) {
2592 Py_DECREF(ob);
2593 return NULL;
2594 }
2595 if (s != NULL) {
2596 if (string_len > 5) {
2597 Py_DECREF(ob);
2598 PyErr_SetString(PyExc_ValueError, "string too long");
2599 return NULL;
2600 }
2601 strcpy(ob->structmembers.inplace_member, s);
2602 }
2603 else {
2604 strcpy(ob->structmembers.inplace_member, "");
2605 }
2606 return (PyObject *)ob;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002607}
2608
Georg Brandlc09b94e2008-01-21 21:28:32 +00002609static void
2610test_structmembers_free(PyObject *ob)
2611{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002612 PyObject_FREE(ob);
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002613}
2614
2615static PyTypeObject test_structmembersType = {
Martin v. Löwis68192102007-07-21 06:55:02 +00002616 PyVarObject_HEAD_INIT(NULL, 0)
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002617 "test_structmembersType",
2618 sizeof(test_structmembers), /* tp_basicsize */
2619 0, /* tp_itemsize */
2620 test_structmembers_free, /* destructor tp_dealloc */
2621 0, /* tp_print */
2622 0, /* tp_getattr */
2623 0, /* tp_setattr */
2624 0, /* tp_compare */
2625 0, /* tp_repr */
2626 0, /* tp_as_number */
2627 0, /* tp_as_sequence */
2628 0, /* tp_as_mapping */
2629 0, /* tp_hash */
2630 0, /* tp_call */
2631 0, /* tp_str */
2632 PyObject_GenericGetAttr, /* tp_getattro */
2633 PyObject_GenericSetAttr, /* tp_setattro */
2634 0, /* tp_as_buffer */
2635 0, /* tp_flags */
2636 "Type containing all structmember types",
2637 0, /* traverseproc tp_traverse */
2638 0, /* tp_clear */
2639 0, /* tp_richcompare */
2640 0, /* tp_weaklistoffset */
2641 0, /* tp_iter */
2642 0, /* tp_iternext */
2643 0, /* tp_methods */
2644 test_members, /* tp_members */
2645 0,
2646 0,
2647 0,
2648 0,
2649 0,
2650 0,
2651 0,
2652 0,
2653 test_structmembers_new, /* tp_new */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002654};
2655
2656
Mark Hammond62b1ab12002-07-23 06:31:15 +00002657PyMODINIT_FUNC
Tim Petersd66595f2001-02-04 03:09:53 +00002658init_testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +00002659{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002660 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00002661
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002662 m = Py_InitModule("_testcapi", TestMethods);
2663 if (m == NULL)
2664 return;
Tim Peters9ea17ac2001-02-02 05:57:15 +00002665
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002666 Py_TYPE(&_HashInheritanceTester_Type)=&PyType_Type;
Hirokazu Yamamoto3cda1db2008-12-31 05:47:19 +00002667
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002668 Py_TYPE(&test_structmembersType)=&PyType_Type;
2669 Py_INCREF(&test_structmembersType);
2670 /* don't use a name starting with "test", since we don't want
2671 test_capi to automatically call this */
2672 PyModule_AddObject(m, "_test_structmembersType", (PyObject *)&test_structmembersType);
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002673
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002674 PyModule_AddObject(m, "CHAR_MAX", PyInt_FromLong(CHAR_MAX));
2675 PyModule_AddObject(m, "CHAR_MIN", PyInt_FromLong(CHAR_MIN));
2676 PyModule_AddObject(m, "UCHAR_MAX", PyInt_FromLong(UCHAR_MAX));
2677 PyModule_AddObject(m, "SHRT_MAX", PyInt_FromLong(SHRT_MAX));
2678 PyModule_AddObject(m, "SHRT_MIN", PyInt_FromLong(SHRT_MIN));
2679 PyModule_AddObject(m, "USHRT_MAX", PyInt_FromLong(USHRT_MAX));
2680 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
2681 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
2682 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
2683 PyModule_AddObject(m, "LONG_MAX", PyInt_FromLong(LONG_MAX));
2684 PyModule_AddObject(m, "LONG_MIN", PyInt_FromLong(LONG_MIN));
2685 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
2686 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
2687 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
2688 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
2689 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
2690 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
2691 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
2692 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
2693 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyInt_FromSsize_t(PY_SSIZE_T_MAX));
2694 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyInt_FromSsize_t(PY_SSIZE_T_MIN));
2695 PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyInt_FromSsize_t(sizeof(PyGC_Head)));
Thomas Hellera4ea6032003-04-17 18:55:45 +00002696
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002697 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
2698 Py_INCREF(TestError);
2699 PyModule_AddObject(m, "error", TestError);
Tim Peters9ea17ac2001-02-02 05:57:15 +00002700}