blob: ecc473cf4c1521e322dbc1b4a0eedd02b2c1104a [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"
Victor Stinnerd2aff602017-05-09 13:57:20 +020013#include <signal.h>
Victor Stinner95a91df2017-06-27 14:31:40 +020014#ifdef MS_WINDOWS
15# include <crtdbg.h>
16#endif
Tim Peters9ea17ac2001-02-02 05:57:15 +000017
Victor Stinner4dea0652017-08-11 14:38:37 +020018#ifdef HAVE_SYS_WAIT_H
19#include <sys/wait.h> /* For W_STOPCODE */
20#endif
21
Mark Hammond8d98d2c2003-04-19 15:41:53 +000022#ifdef WITH_THREAD
23#include "pythread.h"
24#endif /* WITH_THREAD */
Antoine Pitrouc83ea132010-05-09 14:46:46 +000025static PyObject *TestError; /* set to exception object in init */
Tim Peters9ea17ac2001-02-02 05:57:15 +000026
Tim Peters91621db2001-06-12 20:10:01 +000027/* Raise TestError with test_name + ": " + msg, and return NULL. */
28
29static PyObject *
30raiseTestError(const char* test_name, const char* msg)
31{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000032 char buf[2048];
Tim Peters91621db2001-06-12 20:10:01 +000033
Antoine Pitrouc83ea132010-05-09 14:46:46 +000034 if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
35 PyErr_SetString(TestError, "internal error msg too large");
36 else {
37 PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
38 PyErr_SetString(TestError, buf);
39 }
40 return NULL;
Tim Peters91621db2001-06-12 20:10:01 +000041}
42
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +000043/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
Tim Peters9ea17ac2001-02-02 05:57:15 +000044
45 The ones derived from autoconf on the UNIX-like OSes can be relied
46 upon (in the absence of sloppy cross-compiling), but the Windows
47 platforms have these hardcoded. Better safe than sorry.
48*/
49static PyObject*
Martin v. Löwisb5bc5372006-10-27 06:16:31 +000050sizeof_error(const char* fatname, const char* typname,
Antoine Pitrouc83ea132010-05-09 14:46:46 +000051 int expected, int got)
Tim Peters9ea17ac2001-02-02 05:57:15 +000052{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000053 char buf[1024];
54 PyOS_snprintf(buf, sizeof(buf),
55 "%.200s #define == %d but sizeof(%.200s) == %d",
56 fatname, expected, typname, got);
57 PyErr_SetString(TestError, buf);
58 return (PyObject*)NULL;
Tim Peters9ea17ac2001-02-02 05:57:15 +000059}
60
61static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000062test_config(PyObject *self)
Tim Peters9ea17ac2001-02-02 05:57:15 +000063{
Tim Peters9ea17ac2001-02-02 05:57:15 +000064#define CHECK_SIZEOF(FATNAME, TYPE) \
Antoine Pitrouc83ea132010-05-09 14:46:46 +000065 if (FATNAME != sizeof(TYPE)) \
66 return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
Tim Peters9ea17ac2001-02-02 05:57:15 +000067
Antoine Pitrouc83ea132010-05-09 14:46:46 +000068 CHECK_SIZEOF(SIZEOF_SHORT, short);
69 CHECK_SIZEOF(SIZEOF_INT, int);
70 CHECK_SIZEOF(SIZEOF_LONG, long);
71 CHECK_SIZEOF(SIZEOF_VOID_P, void*);
72 CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
Tim Peters9ea17ac2001-02-02 05:57:15 +000073#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +000074 CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
Tim Peters9ea17ac2001-02-02 05:57:15 +000075#endif
76
77#undef CHECK_SIZEOF
78
Antoine Pitrouc83ea132010-05-09 14:46:46 +000079 Py_INCREF(Py_None);
80 return Py_None;
Tim Peters9ea17ac2001-02-02 05:57:15 +000081}
82
Tim Peters5c4d5bf2001-02-12 22:13:26 +000083static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000084test_list_api(PyObject *self)
Tim Peters5c4d5bf2001-02-12 22:13:26 +000085{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000086 PyObject* list;
87 int i;
Tim Peters5c4d5bf2001-02-12 22:13:26 +000088
Antoine Pitrouc83ea132010-05-09 14:46:46 +000089 /* SF bug 132008: PyList_Reverse segfaults */
Tim Peters5c4d5bf2001-02-12 22:13:26 +000090#define NLIST 30
Antoine Pitrouc83ea132010-05-09 14:46:46 +000091 list = PyList_New(NLIST);
92 if (list == (PyObject*)NULL)
93 return (PyObject*)NULL;
94 /* list = range(NLIST) */
95 for (i = 0; i < NLIST; ++i) {
96 PyObject* anint = PyInt_FromLong(i);
97 if (anint == (PyObject*)NULL) {
98 Py_DECREF(list);
99 return (PyObject*)NULL;
100 }
101 PyList_SET_ITEM(list, i, anint);
102 }
103 /* list.reverse(), via PyList_Reverse() */
104 i = PyList_Reverse(list); /* should not blow up! */
105 if (i != 0) {
106 Py_DECREF(list);
107 return (PyObject*)NULL;
108 }
109 /* Check that list == range(29, -1, -1) now */
110 for (i = 0; i < NLIST; ++i) {
111 PyObject* anint = PyList_GET_ITEM(list, i);
112 if (PyInt_AS_LONG(anint) != NLIST-1-i) {
113 PyErr_SetString(TestError,
114 "test_list_api: reverse screwed up");
115 Py_DECREF(list);
116 return (PyObject*)NULL;
117 }
118 }
119 Py_DECREF(list);
Tim Peters5c4d5bf2001-02-12 22:13:26 +0000120#undef NLIST
121
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000122 Py_INCREF(Py_None);
123 return Py_None;
Tim Peters5c4d5bf2001-02-12 22:13:26 +0000124}
125
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000126static int
127test_dict_inner(int count)
128{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000129 Py_ssize_t pos = 0, iterations = 0;
130 int i;
131 PyObject *dict = PyDict_New();
132 PyObject *v, *k;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000133
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000134 if (dict == NULL)
135 return -1;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000136
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000137 for (i = 0; i < count; i++) {
138 v = PyInt_FromLong(i);
139 PyDict_SetItem(dict, v, v);
140 Py_DECREF(v);
141 }
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000142
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000143 while (PyDict_Next(dict, &pos, &k, &v)) {
144 PyObject *o;
145 iterations++;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000146
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000147 i = PyInt_AS_LONG(v) + 1;
148 o = PyInt_FromLong(i);
149 if (o == NULL)
150 return -1;
151 if (PyDict_SetItem(dict, k, o) < 0) {
152 Py_DECREF(o);
153 return -1;
154 }
155 Py_DECREF(o);
156 }
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000157
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000158 Py_DECREF(dict);
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000159
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000160 if (iterations != count) {
161 PyErr_SetString(
162 TestError,
163 "test_dict_iteration: dict iteration went wrong ");
164 return -1;
165 } else {
166 return 0;
167 }
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000168}
169
170static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +0000171test_dict_iteration(PyObject* self)
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000172{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000173 int i;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000174
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000175 for (i = 0; i < 200; i++) {
176 if (test_dict_inner(i) < 0) {
177 return NULL;
178 }
179 }
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000180
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000181 Py_INCREF(Py_None);
182 return Py_None;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000183}
184
Tim Peters91621db2001-06-12 20:10:01 +0000185
Nick Coghlan180e4002008-12-30 01:18:48 +0000186/* Issue #4701: Check that PyObject_Hash implicitly calls
187 * PyType_Ready if it hasn't already been called
188 */
189static PyTypeObject _HashInheritanceTester_Type = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000190 PyObject_HEAD_INIT(NULL)
191 0, /* Number of items for varobject */
192 "hashinheritancetester", /* Name of this type */
193 sizeof(PyObject), /* Basic object size */
194 0, /* Item size for varobject */
195 (destructor)PyObject_Del, /* tp_dealloc */
196 0, /* tp_print */
197 0, /* tp_getattr */
198 0, /* tp_setattr */
199 0, /* tp_compare */
200 0, /* tp_repr */
201 0, /* tp_as_number */
202 0, /* tp_as_sequence */
203 0, /* tp_as_mapping */
204 0, /* tp_hash */
205 0, /* tp_call */
206 0, /* tp_str */
207 PyObject_GenericGetAttr, /* tp_getattro */
208 0, /* tp_setattro */
209 0, /* tp_as_buffer */
210 Py_TPFLAGS_DEFAULT, /* tp_flags */
211 0, /* tp_doc */
212 0, /* tp_traverse */
213 0, /* tp_clear */
214 0, /* tp_richcompare */
215 0, /* tp_weaklistoffset */
216 0, /* tp_iter */
217 0, /* tp_iternext */
218 0, /* tp_methods */
219 0, /* tp_members */
220 0, /* tp_getset */
221 0, /* tp_base */
222 0, /* tp_dict */
223 0, /* tp_descr_get */
224 0, /* tp_descr_set */
225 0, /* tp_dictoffset */
226 0, /* tp_init */
227 0, /* tp_alloc */
228 PyType_GenericNew, /* tp_new */
Nick Coghlan180e4002008-12-30 01:18:48 +0000229};
230
231static PyObject*
232test_lazy_hash_inheritance(PyObject* self)
233{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000234 PyTypeObject *type;
235 PyObject *obj;
236 long hash;
Nick Coghlan180e4002008-12-30 01:18:48 +0000237
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000238 type = &_HashInheritanceTester_Type;
Benjamin Petersona98c8e12009-05-05 21:09:21 +0000239
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000240 if (type->tp_dict != NULL)
241 /* The type has already been initialized. This probably means
242 -R is being used. */
243 Py_RETURN_NONE;
Benjamin Petersona98c8e12009-05-05 21:09:21 +0000244
245
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000246 obj = PyObject_New(PyObject, type);
247 if (obj == NULL) {
248 PyErr_Clear();
249 PyErr_SetString(
250 TestError,
251 "test_lazy_hash_inheritance: failed to create object");
252 return NULL;
253 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000254
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000255 if (type->tp_dict != NULL) {
256 PyErr_SetString(
257 TestError,
258 "test_lazy_hash_inheritance: type initialised too soon");
259 Py_DECREF(obj);
260 return NULL;
261 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000262
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000263 hash = PyObject_Hash(obj);
264 if ((hash == -1) && PyErr_Occurred()) {
265 PyErr_Clear();
266 PyErr_SetString(
267 TestError,
268 "test_lazy_hash_inheritance: could not hash object");
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_dict == NULL) {
274 PyErr_SetString(
275 TestError,
276 "test_lazy_hash_inheritance: type not initialised by hash()");
277 Py_DECREF(obj);
278 return NULL;
279 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000280
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000281 if (type->tp_hash != PyType_Type.tp_hash) {
282 PyErr_SetString(
283 TestError,
284 "test_lazy_hash_inheritance: unexpected hash function");
285 Py_DECREF(obj);
286 return NULL;
287 }
Nick Coghlan180e4002008-12-30 01:18:48 +0000288
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000289 Py_DECREF(obj);
Benjamin Petersona98c8e12009-05-05 21:09:21 +0000290
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000291 Py_RETURN_NONE;
Nick Coghlan180e4002008-12-30 01:18:48 +0000292}
293
294
Antoine Pitrou526e4212010-02-02 22:36:17 +0000295/* Issue #7385: Check that memoryview() does not crash
296 * when bf_getbuffer returns an error
297 */
298
299static int
300broken_buffer_getbuffer(PyObject *self, Py_buffer *view, int flags)
301{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000302 PyErr_SetString(
303 TestError,
304 "test_broken_memoryview: expected error in bf_getbuffer");
305 return -1;
Antoine Pitrou526e4212010-02-02 22:36:17 +0000306}
307
308static PyBufferProcs memoryviewtester_as_buffer = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000309 0, /* bf_getreadbuffer */
310 0, /* bf_getwritebuffer */
311 0, /* bf_getsegcount */
312 0, /* bf_getcharbuffer */
313 (getbufferproc)broken_buffer_getbuffer, /* bf_getbuffer */
314 0, /* bf_releasebuffer */
Antoine Pitrou526e4212010-02-02 22:36:17 +0000315};
316
317static PyTypeObject _MemoryViewTester_Type = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000318 PyObject_HEAD_INIT(NULL)
319 0, /* Number of items for varobject */
320 "memoryviewtester", /* Name of this type */
321 sizeof(PyObject), /* Basic object size */
322 0, /* Item size for varobject */
323 (destructor)PyObject_Del, /* tp_dealloc */
324 0, /* tp_print */
325 0, /* tp_getattr */
326 0, /* tp_setattr */
327 0, /* tp_compare */
328 0, /* tp_repr */
329 0, /* tp_as_number */
330 0, /* tp_as_sequence */
331 0, /* tp_as_mapping */
332 0, /* tp_hash */
333 0, /* tp_call */
334 0, /* tp_str */
335 PyObject_GenericGetAttr, /* tp_getattro */
336 0, /* tp_setattro */
337 &memoryviewtester_as_buffer, /* tp_as_buffer */
338 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_NEWBUFFER, /* tp_flags */
339 0, /* tp_doc */
340 0, /* tp_traverse */
341 0, /* tp_clear */
342 0, /* tp_richcompare */
343 0, /* tp_weaklistoffset */
344 0, /* tp_iter */
345 0, /* tp_iternext */
346 0, /* tp_methods */
347 0, /* tp_members */
348 0, /* tp_getset */
349 0, /* tp_base */
350 0, /* tp_dict */
351 0, /* tp_descr_get */
352 0, /* tp_descr_set */
353 0, /* tp_dictoffset */
354 0, /* tp_init */
355 0, /* tp_alloc */
356 PyType_GenericNew, /* tp_new */
Antoine Pitrou526e4212010-02-02 22:36:17 +0000357};
358
359static PyObject*
360test_broken_memoryview(PyObject* self)
361{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000362 PyObject *obj = PyObject_New(PyObject, &_MemoryViewTester_Type);
363 PyObject *res;
Antoine Pitrou526e4212010-02-02 22:36:17 +0000364
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000365 if (obj == NULL) {
366 PyErr_Clear();
367 PyErr_SetString(
368 TestError,
369 "test_broken_memoryview: failed to create object");
370 return NULL;
371 }
Antoine Pitrou526e4212010-02-02 22:36:17 +0000372
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000373 res = PyMemoryView_FromObject(obj);
374 if (res || !PyErr_Occurred()){
375 PyErr_SetString(
376 TestError,
377 "test_broken_memoryview: memoryview() didn't raise an Exception");
378 Py_XDECREF(res);
379 Py_DECREF(obj);
380 return NULL;
381 }
Antoine Pitrou526e4212010-02-02 22:36:17 +0000382
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000383 PyErr_Clear();
384 Py_DECREF(obj);
385 Py_RETURN_NONE;
Antoine Pitrou526e4212010-02-02 22:36:17 +0000386}
387
Stefan Krah5606cd92015-01-30 20:11:10 +0100388static PyObject *
389test_to_contiguous(PyObject* self, PyObject *noargs)
390{
391 int data[9] = {0, -1, 1, -1, 2, -1, 3, -1, 4};
392 int result[5];
393 Py_ssize_t itemsize = sizeof(int);
394 Py_ssize_t shape = 5;
395 Py_ssize_t strides = 2 * itemsize;
396 Py_buffer view = {
397 data,
398 NULL,
399 5 * itemsize,
400 itemsize,
401 1,
402 1,
403 NULL,
404 &shape,
405 &strides,
406 NULL,
407 {0, 0},
408 NULL
409 };
410 int i;
411
412 PyBuffer_ToContiguous(result, &view, view.len, 'C');
413 for (i = 0; i < 5; i++) {
414 if (result[i] != i) {
415 PyErr_SetString(TestError,
416 "test_to_contiguous: incorrect result");
417 return NULL;
418 }
419 }
420
421 view.buf = &data[8];
422 view.strides[0] = -2 * itemsize;
423
424 PyBuffer_ToContiguous(result, &view, view.len, 'C');
425 for (i = 0; i < 5; i++) {
426 if (result[i] != 4-i) {
427 PyErr_SetString(TestError,
428 "test_to_contiguous: incorrect result");
429 return NULL;
430 }
431 }
432
433 Py_RETURN_NONE;
434}
Antoine Pitrou526e4212010-02-02 22:36:17 +0000435
Stefan Krah0a7fc532015-02-01 16:10:35 +0100436static PyObject *
437test_from_contiguous(PyObject* self, PyObject *noargs)
438{
439 int data[9] = {-1,-1,-1,-1,-1,-1,-1,-1,-1};
440 int init[5] = {0, 1, 2, 3, 4};
441 Py_ssize_t itemsize = sizeof(int);
442 Py_ssize_t shape = 5;
443 Py_ssize_t strides = 2 * itemsize;
444 Py_buffer view = {
445 data,
446 NULL,
447 5 * itemsize,
448 itemsize,
449 1,
450 1,
451 NULL,
452 &shape,
453 &strides,
454 NULL,
455 {0, 0},
456 NULL
457 };
458 int *ptr;
459 int i;
460
461 PyBuffer_FromContiguous(&view, init, view.len, 'C');
462 ptr = view.buf;
463 for (i = 0; i < 5; i++) {
464 if (ptr[2*i] != i) {
465 PyErr_SetString(TestError,
466 "test_from_contiguous: incorrect result");
467 return NULL;
468 }
469 }
470
471 view.buf = &data[8];
472 view.strides[0] = -2 * itemsize;
473
474 PyBuffer_FromContiguous(&view, init, view.len, 'C');
475 ptr = view.buf;
476 for (i = 0; i < 5; i++) {
477 if (*(ptr-2*i) != i) {
478 PyErr_SetString(TestError,
479 "test_from_contiguous: incorrect result");
480 return NULL;
481 }
482 }
483
484 Py_RETURN_NONE;
485}
486
487
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000488/* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
Tim Petersff70d3c2001-06-14 01:11:03 +0000489 PyLong_{As, From}{Unsigned,}LongLong().
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000490
491 Note that the meat of the test is contained in testcapi_long.h.
492 This is revolting, but delicate code duplication is worse: "almost
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000493 exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000494 dependence on type names makes it impossible to use a parameterized
495 function. A giant macro would be even worse than this. A C++ template
496 would be perfect.
497
498 The "report an error" functions are deliberately not part of the #include
499 file: if the test fails, you can set a breakpoint in the appropriate
500 error function directly, and crawl back from there in the debugger.
501*/
502
503#define UNBIND(X) Py_DECREF(X); (X) = NULL
504
505static PyObject *
506raise_test_long_error(const char* msg)
507{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000508 return raiseTestError("test_long_api", msg);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000509}
510
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000511#define TESTNAME test_long_api_inner
512#define TYPENAME long
513#define F_S_TO_PY PyLong_FromLong
514#define F_PY_TO_S PyLong_AsLong
515#define F_U_TO_PY PyLong_FromUnsignedLong
516#define F_PY_TO_U PyLong_AsUnsignedLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000517
518#include "testcapi_long.h"
519
520static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000521test_long_api(PyObject* self)
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000522{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000523 return TESTNAME(raise_test_long_error);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000524}
525
526#undef TESTNAME
527#undef TYPENAME
528#undef F_S_TO_PY
529#undef F_PY_TO_S
530#undef F_U_TO_PY
531#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000532
533#ifdef HAVE_LONG_LONG
Tim Peters91621db2001-06-12 20:10:01 +0000534
535static PyObject *
Tim Petersd1a7da62001-06-13 00:35:57 +0000536raise_test_longlong_error(const char* msg)
537{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000538 return raiseTestError("test_longlong_api", msg);
Tim Petersd1a7da62001-06-13 00:35:57 +0000539}
540
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000541#define TESTNAME test_longlong_api_inner
542#define TYPENAME PY_LONG_LONG
543#define F_S_TO_PY PyLong_FromLongLong
544#define F_PY_TO_S PyLong_AsLongLong
545#define F_U_TO_PY PyLong_FromUnsignedLongLong
546#define F_PY_TO_U PyLong_AsUnsignedLongLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000547
548#include "testcapi_long.h"
Tim Petersd1a7da62001-06-13 00:35:57 +0000549
550static PyObject *
Skip Montanaro9582c142006-04-18 01:01:41 +0000551test_longlong_api(PyObject* self, PyObject *args)
Tim Peters91621db2001-06-12 20:10:01 +0000552{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000553 return TESTNAME(raise_test_longlong_error);
Tim Peters91621db2001-06-12 20:10:01 +0000554}
555
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000556#undef TESTNAME
557#undef TYPENAME
558#undef F_S_TO_PY
559#undef F_PY_TO_S
560#undef F_U_TO_PY
561#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000562
Mark Dickinsone31d3002009-12-21 11:21:25 +0000563/* Test the PyLong_AsLongAndOverflow API. General conversion to PY_LONG
564 is tested by test_long_api_inner. This test will concentrate on proper
565 handling of overflow.
566*/
567
568static PyObject *
569test_long_and_overflow(PyObject *self)
570{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000571 PyObject *num, *one, *temp;
572 long value;
573 int overflow;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000574
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000575 /* Test that overflow is set properly for a large value. */
576 /* num is a number larger than LONG_MAX even on 64-bit platforms */
577 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
578 if (num == NULL)
579 return NULL;
580 overflow = 1234;
581 value = PyLong_AsLongAndOverflow(num, &overflow);
582 Py_DECREF(num);
583 if (value == -1 && PyErr_Occurred())
584 return NULL;
585 if (value != -1)
586 return raiseTestError("test_long_and_overflow",
587 "return value was not set to -1");
588 if (overflow != 1)
589 return raiseTestError("test_long_and_overflow",
590 "overflow was not set to 1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000591
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000592 /* Same again, with num = LONG_MAX + 1 */
593 num = PyLong_FromLong(LONG_MAX);
594 if (num == NULL)
595 return NULL;
596 one = PyLong_FromLong(1L);
597 if (one == NULL) {
598 Py_DECREF(num);
599 return NULL;
600 }
601 temp = PyNumber_Add(num, one);
602 Py_DECREF(one);
603 Py_DECREF(num);
604 num = temp;
605 if (num == NULL)
606 return NULL;
607 overflow = 0;
608 value = PyLong_AsLongAndOverflow(num, &overflow);
609 Py_DECREF(num);
610 if (value == -1 && PyErr_Occurred())
611 return NULL;
612 if (value != -1)
613 return raiseTestError("test_long_and_overflow",
614 "return value was not set to -1");
615 if (overflow != 1)
616 return raiseTestError("test_long_and_overflow",
617 "overflow was not set to 1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000618
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000619 /* Test that overflow is set properly for a large negative value. */
620 /* num is a number smaller than LONG_MIN even on 64-bit platforms */
621 num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
622 if (num == NULL)
623 return NULL;
624 overflow = 1234;
625 value = PyLong_AsLongAndOverflow(num, &overflow);
626 Py_DECREF(num);
627 if (value == -1 && PyErr_Occurred())
628 return NULL;
629 if (value != -1)
630 return raiseTestError("test_long_and_overflow",
631 "return value was not set to -1");
632 if (overflow != -1)
633 return raiseTestError("test_long_and_overflow",
634 "overflow was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000635
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000636 /* Same again, with num = LONG_MIN - 1 */
637 num = PyLong_FromLong(LONG_MIN);
638 if (num == NULL)
639 return NULL;
640 one = PyLong_FromLong(1L);
641 if (one == NULL) {
642 Py_DECREF(num);
643 return NULL;
644 }
645 temp = PyNumber_Subtract(num, one);
646 Py_DECREF(one);
647 Py_DECREF(num);
648 num = temp;
649 if (num == NULL)
650 return NULL;
651 overflow = 0;
652 value = PyLong_AsLongAndOverflow(num, &overflow);
653 Py_DECREF(num);
654 if (value == -1 && PyErr_Occurred())
655 return NULL;
656 if (value != -1)
657 return raiseTestError("test_long_and_overflow",
658 "return value was not set to -1");
659 if (overflow != -1)
660 return raiseTestError("test_long_and_overflow",
661 "overflow was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000662
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000663 /* Test that overflow is cleared properly for small values. */
664 num = PyLong_FromString("FF", NULL, 16);
665 if (num == NULL)
666 return NULL;
667 overflow = 1234;
668 value = PyLong_AsLongAndOverflow(num, &overflow);
669 Py_DECREF(num);
670 if (value == -1 && PyErr_Occurred())
671 return NULL;
672 if (value != 0xFF)
673 return raiseTestError("test_long_and_overflow",
674 "expected return value 0xFF");
675 if (overflow != 0)
676 return raiseTestError("test_long_and_overflow",
677 "overflow was not cleared");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000678
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000679 num = PyLong_FromString("-FF", NULL, 16);
680 if (num == NULL)
681 return NULL;
682 overflow = 0;
683 value = PyLong_AsLongAndOverflow(num, &overflow);
684 Py_DECREF(num);
685 if (value == -1 && PyErr_Occurred())
686 return NULL;
687 if (value != -0xFF)
688 return raiseTestError("test_long_and_overflow",
689 "expected return value 0xFF");
690 if (overflow != 0)
691 return raiseTestError("test_long_and_overflow",
692 "overflow was set incorrectly");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000693
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000694 num = PyLong_FromLong(LONG_MAX);
695 if (num == NULL)
696 return NULL;
697 overflow = 1234;
698 value = PyLong_AsLongAndOverflow(num, &overflow);
699 Py_DECREF(num);
700 if (value == -1 && PyErr_Occurred())
701 return NULL;
702 if (value != LONG_MAX)
703 return raiseTestError("test_long_and_overflow",
704 "expected return value LONG_MAX");
705 if (overflow != 0)
706 return raiseTestError("test_long_and_overflow",
707 "overflow was not cleared");
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000708
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000709 num = PyLong_FromLong(LONG_MIN);
710 if (num == NULL)
711 return NULL;
712 overflow = 0;
713 value = PyLong_AsLongAndOverflow(num, &overflow);
714 Py_DECREF(num);
715 if (value == -1 && PyErr_Occurred())
716 return NULL;
717 if (value != LONG_MIN)
718 return raiseTestError("test_long_and_overflow",
719 "expected return value LONG_MIN");
720 if (overflow != 0)
721 return raiseTestError("test_long_and_overflow",
722 "overflow was not cleared");
Mark Dickinsoned02b3f2009-12-21 11:31:54 +0000723
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000724 Py_INCREF(Py_None);
725 return Py_None;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000726}
727
Mark Dickinsona36507c2010-01-30 10:08:33 +0000728/* Test the PyLong_AsLongLongAndOverflow API. General conversion to
729 PY_LONG_LONG is tested by test_long_api_inner. This test will
730 concentrate on proper handling of overflow.
731*/
732
733static PyObject *
734test_long_long_and_overflow(PyObject *self)
735{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000736 PyObject *num, *one, *temp;
737 PY_LONG_LONG value;
738 int overflow;
Mark Dickinsona36507c2010-01-30 10:08:33 +0000739
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000740 /* Test that overflow is set properly for a large value. */
741 /* num is a number larger than PY_LLONG_MAX on a typical machine. */
742 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
743 if (num == NULL)
744 return NULL;
745 overflow = 1234;
746 value = PyLong_AsLongLongAndOverflow(num, &overflow);
747 Py_DECREF(num);
748 if (value == -1 && PyErr_Occurred())
749 return NULL;
750 if (value != -1)
751 return raiseTestError("test_long_long_and_overflow",
752 "return value was not set to -1");
753 if (overflow != 1)
754 return raiseTestError("test_long_long_and_overflow",
755 "overflow was not set to 1");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000756
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000757 /* Same again, with num = PY_LLONG_MAX + 1 */
758 num = PyLong_FromLongLong(PY_LLONG_MAX);
759 if (num == NULL)
760 return NULL;
761 one = PyLong_FromLong(1L);
762 if (one == NULL) {
763 Py_DECREF(num);
764 return NULL;
765 }
766 temp = PyNumber_Add(num, one);
767 Py_DECREF(one);
768 Py_DECREF(num);
769 num = temp;
770 if (num == NULL)
771 return NULL;
772 overflow = 0;
773 value = PyLong_AsLongLongAndOverflow(num, &overflow);
774 Py_DECREF(num);
775 if (value == -1 && PyErr_Occurred())
776 return NULL;
777 if (value != -1)
778 return raiseTestError("test_long_long_and_overflow",
779 "return value was not set to -1");
780 if (overflow != 1)
781 return raiseTestError("test_long_long_and_overflow",
782 "overflow was not set to 1");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000783
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000784 /* Test that overflow is set properly for a large negative value. */
785 /* num is a number smaller than PY_LLONG_MIN on a typical platform */
786 num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
787 if (num == NULL)
788 return NULL;
789 overflow = 1234;
790 value = PyLong_AsLongLongAndOverflow(num, &overflow);
791 Py_DECREF(num);
792 if (value == -1 && PyErr_Occurred())
793 return NULL;
794 if (value != -1)
795 return raiseTestError("test_long_long_and_overflow",
796 "return value was not set to -1");
797 if (overflow != -1)
798 return raiseTestError("test_long_long_and_overflow",
799 "overflow was not set to -1");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000800
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000801 /* Same again, with num = PY_LLONG_MIN - 1 */
802 num = PyLong_FromLongLong(PY_LLONG_MIN);
803 if (num == NULL)
804 return NULL;
805 one = PyLong_FromLong(1L);
806 if (one == NULL) {
807 Py_DECREF(num);
808 return NULL;
809 }
810 temp = PyNumber_Subtract(num, one);
811 Py_DECREF(one);
812 Py_DECREF(num);
813 num = temp;
814 if (num == NULL)
815 return NULL;
816 overflow = 0;
817 value = PyLong_AsLongLongAndOverflow(num, &overflow);
818 Py_DECREF(num);
819 if (value == -1 && PyErr_Occurred())
820 return NULL;
821 if (value != -1)
822 return raiseTestError("test_long_long_and_overflow",
823 "return value was not set to -1");
824 if (overflow != -1)
825 return raiseTestError("test_long_long_and_overflow",
826 "overflow was not set to -1");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000827
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000828 /* Test that overflow is cleared properly for small values. */
829 num = PyLong_FromString("FF", NULL, 16);
830 if (num == NULL)
831 return NULL;
832 overflow = 1234;
833 value = PyLong_AsLongLongAndOverflow(num, &overflow);
834 Py_DECREF(num);
835 if (value == -1 && PyErr_Occurred())
836 return NULL;
837 if (value != 0xFF)
838 return raiseTestError("test_long_long_and_overflow",
839 "expected return value 0xFF");
840 if (overflow != 0)
841 return raiseTestError("test_long_long_and_overflow",
842 "overflow was not cleared");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000843
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000844 num = PyLong_FromString("-FF", NULL, 16);
845 if (num == NULL)
846 return NULL;
847 overflow = 0;
848 value = PyLong_AsLongLongAndOverflow(num, &overflow);
849 Py_DECREF(num);
850 if (value == -1 && PyErr_Occurred())
851 return NULL;
852 if (value != -0xFF)
853 return raiseTestError("test_long_long_and_overflow",
854 "expected return value 0xFF");
855 if (overflow != 0)
856 return raiseTestError("test_long_long_and_overflow",
857 "overflow was set incorrectly");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000858
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000859 num = PyLong_FromLongLong(PY_LLONG_MAX);
860 if (num == NULL)
861 return NULL;
862 overflow = 1234;
863 value = PyLong_AsLongLongAndOverflow(num, &overflow);
864 Py_DECREF(num);
865 if (value == -1 && PyErr_Occurred())
866 return NULL;
867 if (value != PY_LLONG_MAX)
868 return raiseTestError("test_long_long_and_overflow",
869 "expected return value PY_LLONG_MAX");
870 if (overflow != 0)
871 return raiseTestError("test_long_long_and_overflow",
872 "overflow was not cleared");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000873
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000874 num = PyLong_FromLongLong(PY_LLONG_MIN);
875 if (num == NULL)
876 return NULL;
877 overflow = 0;
878 value = PyLong_AsLongLongAndOverflow(num, &overflow);
879 Py_DECREF(num);
880 if (value == -1 && PyErr_Occurred())
881 return NULL;
882 if (value != PY_LLONG_MIN)
883 return raiseTestError("test_long_long_and_overflow",
884 "expected return value PY_LLONG_MIN");
885 if (overflow != 0)
886 return raiseTestError("test_long_long_and_overflow",
887 "overflow was not cleared");
Mark Dickinsona36507c2010-01-30 10:08:33 +0000888
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000889 Py_INCREF(Py_None);
890 return Py_None;
Mark Dickinsona36507c2010-01-30 10:08:33 +0000891}
892
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000893/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
Tim Petersd38b1c72001-09-30 05:09:37 +0000894 for both long and int arguments. The test may leak a little memory if
895 it fails.
896*/
897static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000898test_L_code(PyObject *self)
Tim Petersd38b1c72001-09-30 05:09:37 +0000899{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000900 PyObject *tuple, *num;
901 PY_LONG_LONG value;
Tim Petersd38b1c72001-09-30 05:09:37 +0000902
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000903 tuple = PyTuple_New(1);
904 if (tuple == NULL)
905 return NULL;
Tim Petersd38b1c72001-09-30 05:09:37 +0000906
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000907 num = PyLong_FromLong(42);
908 if (num == NULL)
909 return NULL;
Tim Petersd38b1c72001-09-30 05:09:37 +0000910
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000911 PyTuple_SET_ITEM(tuple, 0, num);
Tim Petersd38b1c72001-09-30 05:09:37 +0000912
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000913 value = -1;
Oren Milman20958e62017-08-29 19:16:12 +0300914 if (!PyArg_ParseTuple(tuple, "L:test_L_code", &value)) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000915 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +0300916 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000917 if (value != 42)
918 return raiseTestError("test_L_code",
919 "L code returned wrong value for long 42");
Tim Petersd38b1c72001-09-30 05:09:37 +0000920
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000921 Py_DECREF(num);
922 num = PyInt_FromLong(42);
923 if (num == NULL)
924 return NULL;
Tim Petersd38b1c72001-09-30 05:09:37 +0000925
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000926 PyTuple_SET_ITEM(tuple, 0, num);
Tim Petersd38b1c72001-09-30 05:09:37 +0000927
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000928 value = -1;
Oren Milman20958e62017-08-29 19:16:12 +0300929 if (!PyArg_ParseTuple(tuple, "L:test_L_code", &value)) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000930 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +0300931 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000932 if (value != 42)
933 return raiseTestError("test_L_code",
934 "L code returned wrong value for int 42");
Tim Petersd38b1c72001-09-30 05:09:37 +0000935
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000936 Py_DECREF(tuple);
937 Py_INCREF(Py_None);
938 return Py_None;
Tim Petersd38b1c72001-09-30 05:09:37 +0000939}
940
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000941#endif /* ifdef HAVE_LONG_LONG */
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000942
Serhiy Storchakac6a7c962016-05-16 09:55:32 +0300943static PyObject *
Serhiy Storchaka12cf60c2016-05-20 22:31:24 +0300944return_none(void *unused)
945{
946 Py_RETURN_NONE;
947}
948
949static PyObject *
950raise_error(void *unused)
951{
952 PyErr_SetNone(PyExc_ValueError);
953 return NULL;
954}
955
956static int
957test_buildvalue_N_error(const char *fmt)
958{
959 PyObject *arg, *res;
960
961 arg = PyList_New(0);
962 if (arg == NULL) {
963 return -1;
964 }
965
966 Py_INCREF(arg);
967 res = Py_BuildValue(fmt, return_none, NULL, arg);
968 if (res == NULL) {
969 return -1;
970 }
971 Py_DECREF(res);
972 if (Py_REFCNT(arg) != 1) {
973 PyErr_Format(TestError, "test_buildvalue_N: "
974 "arg was not decrefed in successful "
975 "Py_BuildValue(\"%s\")", fmt);
976 return -1;
977 }
978
979 Py_INCREF(arg);
980 res = Py_BuildValue(fmt, raise_error, NULL, arg);
981 if (res != NULL || !PyErr_Occurred()) {
982 PyErr_Format(TestError, "test_buildvalue_N: "
983 "Py_BuildValue(\"%s\") didn't complain", fmt);
984 return -1;
985 }
986 PyErr_Clear();
987 if (Py_REFCNT(arg) != 1) {
988 PyErr_Format(TestError, "test_buildvalue_N: "
989 "arg was not decrefed in failed "
990 "Py_BuildValue(\"%s\")", fmt);
991 return -1;
992 }
993 Py_DECREF(arg);
994 return 0;
995}
996
997static PyObject *
998test_buildvalue_N(PyObject *self, PyObject *noargs)
999{
1000 PyObject *arg, *res;
1001
1002 arg = PyList_New(0);
1003 if (arg == NULL) {
1004 return NULL;
1005 }
1006 Py_INCREF(arg);
1007 res = Py_BuildValue("N", arg);
1008 if (res == NULL) {
1009 return NULL;
1010 }
1011 if (res != arg) {
1012 return raiseTestError("test_buildvalue_N",
1013 "Py_BuildValue(\"N\") returned wrong result");
1014 }
1015 if (Py_REFCNT(arg) != 2) {
1016 return raiseTestError("test_buildvalue_N",
1017 "arg was not decrefed in Py_BuildValue(\"N\")");
1018 }
1019 Py_DECREF(res);
1020 Py_DECREF(arg);
1021
1022 if (test_buildvalue_N_error("O&N") < 0)
1023 return NULL;
1024 if (test_buildvalue_N_error("(O&N)") < 0)
1025 return NULL;
1026 if (test_buildvalue_N_error("[O&N]") < 0)
1027 return NULL;
1028 if (test_buildvalue_N_error("{O&N}") < 0)
1029 return NULL;
1030 if (test_buildvalue_N_error("{()O&(())N}") < 0)
1031 return NULL;
1032
1033 Py_RETURN_NONE;
1034}
1035
1036
1037static PyObject *
Serhiy Storchakac6a7c962016-05-16 09:55:32 +03001038get_args(PyObject *self, PyObject *args)
1039{
1040 if (args == NULL) {
1041 args = Py_None;
1042 }
1043 Py_INCREF(args);
1044 return args;
1045}
1046
1047static PyObject *
1048get_kwargs(PyObject *self, PyObject *args, PyObject *kwargs)
1049{
1050 if (kwargs == NULL) {
1051 kwargs = Py_None;
1052 }
1053 Py_INCREF(kwargs);
1054 return kwargs;
1055}
1056
Georg Brandl5f135782006-07-26 08:03:10 +00001057/* Test tuple argument processing */
1058static PyObject *
1059getargs_tuple(PyObject *self, PyObject *args)
1060{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001061 int a, b, c;
1062 if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
1063 return NULL;
1064 return Py_BuildValue("iii", a, b, c);
Georg Brandl5f135782006-07-26 08:03:10 +00001065}
1066
Christian Heimesea837932008-02-26 17:23:51 +00001067/* test PyArg_ParseTupleAndKeywords */
1068static PyObject *getargs_keywords(PyObject *self, PyObject *args, PyObject *kwargs)
1069{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001070 static char *keywords[] = {"arg1","arg2","arg3","arg4","arg5", NULL};
1071 static char *fmt="(ii)i|(i(ii))(iii)i";
1072 int int_args[10]={-1, -1, -1, -1, -1, -1, -1, -1, -1, -1};
Christian Heimesea837932008-02-26 17:23:51 +00001073
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001074 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
1075 &int_args[0], &int_args[1], &int_args[2], &int_args[3], &int_args[4],
1076 &int_args[5], &int_args[6], &int_args[7], &int_args[8], &int_args[9]))
1077 return NULL;
1078 return Py_BuildValue("iiiiiiiiii",
1079 int_args[0], int_args[1], int_args[2], int_args[3], int_args[4],
1080 int_args[5], int_args[6], int_args[7], int_args[8], int_args[9]);
Christian Heimesea837932008-02-26 17:23:51 +00001081}
1082
Thomas Heller3457e4b2003-04-24 16:14:27 +00001083/* Functions to call PyArg_ParseTuple with integer format codes,
1084 and return the result.
1085*/
Thomas Hellera4ea6032003-04-17 18:55:45 +00001086static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +00001087getargs_b(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +00001088{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001089 unsigned char value;
1090 if (!PyArg_ParseTuple(args, "b", &value))
1091 return NULL;
1092 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001093}
1094
Thomas Heller3457e4b2003-04-24 16:14:27 +00001095static PyObject *
1096getargs_B(PyObject *self, PyObject *args)
1097{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001098 unsigned char value;
1099 if (!PyArg_ParseTuple(args, "B", &value))
1100 return NULL;
1101 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001102}
1103
1104static PyObject *
Mark Dickinsonb5e348b2009-12-20 15:57:56 +00001105getargs_h(PyObject *self, PyObject *args)
1106{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001107 short value;
1108 if (!PyArg_ParseTuple(args, "h", &value))
1109 return NULL;
1110 return PyLong_FromLong((long)value);
Mark Dickinsonb5e348b2009-12-20 15:57:56 +00001111}
1112
1113static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +00001114getargs_H(PyObject *self, PyObject *args)
1115{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001116 unsigned short value;
1117 if (!PyArg_ParseTuple(args, "H", &value))
1118 return NULL;
1119 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001120}
1121
1122static PyObject *
1123getargs_I(PyObject *self, PyObject *args)
1124{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001125 unsigned int value;
1126 if (!PyArg_ParseTuple(args, "I", &value))
1127 return NULL;
1128 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001129}
1130
1131static PyObject *
1132getargs_k(PyObject *self, PyObject *args)
1133{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001134 unsigned long value;
1135 if (!PyArg_ParseTuple(args, "k", &value))
1136 return NULL;
1137 return PyLong_FromUnsignedLong(value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001138}
1139
1140static PyObject *
1141getargs_i(PyObject *self, PyObject *args)
1142{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001143 int value;
1144 if (!PyArg_ParseTuple(args, "i", &value))
1145 return NULL;
1146 return PyLong_FromLong((long)value);
Thomas Heller3457e4b2003-04-24 16:14:27 +00001147}
1148
Thomas Hellera4ea6032003-04-17 18:55:45 +00001149static PyObject *
1150getargs_l(PyObject *self, PyObject *args)
1151{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001152 long value;
1153 if (!PyArg_ParseTuple(args, "l", &value))
1154 return NULL;
1155 return PyLong_FromLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001156}
1157
Georg Brandl7f573f72006-04-13 07:59:30 +00001158static PyObject *
1159getargs_n(PyObject *self, PyObject *args)
1160{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001161 Py_ssize_t value;
1162 if (!PyArg_ParseTuple(args, "n", &value))
1163 return NULL;
1164 return PyInt_FromSsize_t(value);
Georg Brandl7f573f72006-04-13 07:59:30 +00001165}
1166
Thomas Hellera4ea6032003-04-17 18:55:45 +00001167#ifdef HAVE_LONG_LONG
Thomas Hellera4ea6032003-04-17 18:55:45 +00001168static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +00001169getargs_L(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +00001170{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001171 PY_LONG_LONG value;
1172 if (!PyArg_ParseTuple(args, "L", &value))
1173 return NULL;
1174 return PyLong_FromLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001175}
1176
Thomas Hellera4ea6032003-04-17 18:55:45 +00001177static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +00001178getargs_K(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +00001179{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001180 unsigned PY_LONG_LONG value;
1181 if (!PyArg_ParseTuple(args, "K", &value))
1182 return NULL;
1183 return PyLong_FromUnsignedLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001184}
1185#endif
1186
1187/* This function not only tests the 'k' getargs code, but also the
1188 PyInt_AsUnsignedLongMask() and PyInt_AsUnsignedLongMask() functions. */
1189static PyObject *
1190test_k_code(PyObject *self)
1191{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001192 PyObject *tuple, *num;
1193 unsigned long value;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001194
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001195 tuple = PyTuple_New(1);
1196 if (tuple == NULL)
1197 return NULL;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001198
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001199 /* a number larger than ULONG_MAX even on 64-bit platforms */
1200 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
1201 if (num == NULL)
1202 return NULL;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001203
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001204 value = PyInt_AsUnsignedLongMask(num);
1205 if (value != ULONG_MAX)
1206 return raiseTestError("test_k_code",
1207 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +00001208
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001209 PyTuple_SET_ITEM(tuple, 0, num);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001210
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001211 value = 0;
Oren Milman20958e62017-08-29 19:16:12 +03001212 if (!PyArg_ParseTuple(tuple, "k:test_k_code", &value)) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001213 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +03001214 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001215 if (value != ULONG_MAX)
1216 return raiseTestError("test_k_code",
1217 "k code returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +00001218
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001219 Py_DECREF(num);
1220 num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
1221 if (num == NULL)
1222 return NULL;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001223
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001224 value = PyInt_AsUnsignedLongMask(num);
1225 if (value != (unsigned long)-0x42)
1226 return raiseTestError("test_k_code",
1227 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
Thomas Hellera4ea6032003-04-17 18:55:45 +00001228
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001229 PyTuple_SET_ITEM(tuple, 0, num);
Thomas Hellera4ea6032003-04-17 18:55:45 +00001230
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001231 value = 0;
Oren Milman20958e62017-08-29 19:16:12 +03001232 if (!PyArg_ParseTuple(tuple, "k:test_k_code", &value)) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001233 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +03001234 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001235 if (value != (unsigned long)-0x42)
1236 return raiseTestError("test_k_code",
1237 "k code returned wrong value for long -0xFFF..000042");
Thomas Hellera4ea6032003-04-17 18:55:45 +00001238
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001239 Py_DECREF(tuple);
1240 Py_INCREF(Py_None);
1241 return Py_None;
Thomas Hellera4ea6032003-04-17 18:55:45 +00001242}
1243
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001244static PyObject *
Serhiy Storchakad0dc72b2016-05-16 10:12:02 +03001245getargs_f(PyObject *self, PyObject *args)
1246{
1247 float f;
1248 if (!PyArg_ParseTuple(args, "f", &f))
1249 return NULL;
1250 return PyFloat_FromDouble(f);
1251}
1252
1253static PyObject *
1254getargs_d(PyObject *self, PyObject *args)
1255{
1256 double d;
1257 if (!PyArg_ParseTuple(args, "d", &d))
1258 return NULL;
1259 return PyFloat_FromDouble(d);
1260}
1261
1262static PyObject *
1263getargs_D(PyObject *self, PyObject *args)
1264{
1265 Py_complex cval;
1266 if (!PyArg_ParseTuple(args, "D", &cval))
1267 return NULL;
1268 return PyComplex_FromCComplex(cval);
1269}
1270
1271static PyObject *
1272getargs_S(PyObject *self, PyObject *args)
1273{
1274 PyObject *obj;
1275 if (!PyArg_ParseTuple(args, "S", &obj))
1276 return NULL;
1277 Py_INCREF(obj);
1278 return obj;
1279}
1280
1281static PyObject *
1282getargs_Y(PyObject *self, PyObject *args)
1283{
1284 PyObject *obj;
1285 if (!PyArg_ParseTuple(args, "Y", &obj))
1286 return NULL;
1287 Py_INCREF(obj);
1288 return obj;
1289}
1290
1291#ifdef Py_USING_UNICODE
1292static PyObject *
1293getargs_U(PyObject *self, PyObject *args)
1294{
1295 PyObject *obj;
1296 if (!PyArg_ParseTuple(args, "U", &obj))
1297 return NULL;
1298 Py_INCREF(obj);
1299 return obj;
1300}
1301#endif
1302
1303static PyObject *
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001304getargs_c(PyObject *self, PyObject *args)
1305{
1306 char c;
1307 if (!PyArg_ParseTuple(args, "c", &c))
1308 return NULL;
1309 return PyInt_FromLong((unsigned char)c);
1310}
1311
1312static PyObject *
1313getargs_s(PyObject *self, PyObject *args)
1314{
1315 const char *str;
1316 if (!PyArg_ParseTuple(args, "s", &str))
1317 return NULL;
1318 return PyBytes_FromString(str);
1319}
1320
1321static PyObject *
1322getargs_s_star(PyObject *self, PyObject *args)
1323{
1324 Py_buffer buffer;
1325 PyObject *bytes;
1326 if (!PyArg_ParseTuple(args, "s*", &buffer))
1327 return NULL;
1328 bytes = PyBytes_FromStringAndSize(buffer.buf, buffer.len);
1329 PyBuffer_Release(&buffer);
1330 return bytes;
1331}
1332
1333static PyObject *
1334getargs_s_hash(PyObject *self, PyObject *args)
1335{
1336 const char *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001337 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001338 if (!PyArg_ParseTuple(args, "s#", &str, &size))
1339 return NULL;
1340 return PyBytes_FromStringAndSize(str, size);
1341}
1342
1343static PyObject *
1344getargs_t_hash(PyObject *self, PyObject *args)
1345{
1346 const char *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001347 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001348 if (!PyArg_ParseTuple(args, "t#", &str, &size))
1349 return NULL;
1350 return PyBytes_FromStringAndSize(str, size);
1351}
1352
1353static PyObject *
1354getargs_z(PyObject *self, PyObject *args)
1355{
1356 const char *str;
1357 if (!PyArg_ParseTuple(args, "z", &str))
1358 return NULL;
1359 if (str != NULL)
1360 return PyBytes_FromString(str);
1361 else
1362 Py_RETURN_NONE;
1363}
1364
1365static PyObject *
1366getargs_z_star(PyObject *self, PyObject *args)
1367{
1368 Py_buffer buffer;
1369 PyObject *bytes;
1370 if (!PyArg_ParseTuple(args, "z*", &buffer))
1371 return NULL;
1372 if (buffer.buf != NULL)
1373 bytes = PyBytes_FromStringAndSize(buffer.buf, buffer.len);
1374 else {
1375 Py_INCREF(Py_None);
1376 bytes = Py_None;
1377 }
1378 PyBuffer_Release(&buffer);
1379 return bytes;
1380}
1381
1382static PyObject *
1383getargs_z_hash(PyObject *self, PyObject *args)
1384{
1385 const char *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001386 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001387 if (!PyArg_ParseTuple(args, "z#", &str, &size))
1388 return NULL;
1389 if (str != NULL)
1390 return PyBytes_FromStringAndSize(str, size);
1391 else
1392 Py_RETURN_NONE;
1393}
1394
1395static PyObject *
1396getargs_w(PyObject *self, PyObject *args)
1397{
1398 char *str;
1399 Py_ssize_t size;
1400
1401 if (!PyArg_ParseTuple(args, "wn", &str, &size))
1402 return NULL;
1403
1404 if (2 <= size) {
1405 str[0] = '[';
1406 str[size-1] = ']';
1407 }
1408
1409 return PyBytes_FromStringAndSize(str, size);
1410}
1411
1412static PyObject *
1413getargs_w_hash(PyObject *self, PyObject *args)
1414{
1415 char *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001416 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001417
1418 if (!PyArg_ParseTuple(args, "w#", &str, &size))
1419 return NULL;
1420
1421 if (2 <= size) {
1422 str[0] = '[';
1423 str[size-1] = ']';
1424 }
1425
1426 return PyBytes_FromStringAndSize(str, size);
1427}
1428
1429static PyObject *
1430getargs_w_star(PyObject *self, PyObject *args)
1431{
1432 Py_buffer buffer;
1433 PyObject *result;
1434 char *str;
1435
1436 if (!PyArg_ParseTuple(args, "w*", &buffer))
1437 return NULL;
1438
1439 if (2 <= buffer.len) {
1440 str = buffer.buf;
1441 str[0] = '[';
1442 str[buffer.len-1] = ']';
1443 }
1444
1445 result = PyBytes_FromStringAndSize(buffer.buf, buffer.len);
1446 PyBuffer_Release(&buffer);
1447 return result;
1448}
1449
1450#ifdef Py_USING_UNICODE
1451
1452static int
1453_ustrlen(const Py_UNICODE *u)
1454{
1455 int i = 0;
1456 const Py_UNICODE *v = u;
1457 while (*v != 0) { i++; v++; }
1458 return i;
1459}
1460
1461static PyObject *
1462getargs_u(PyObject *self, PyObject *args)
1463{
1464 const Py_UNICODE *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001465 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001466 if (!PyArg_ParseTuple(args, "u", &str))
1467 return NULL;
1468 size = _ustrlen(str);
1469 return PyUnicode_FromUnicode(str, size);
1470}
1471
1472static PyObject *
1473getargs_u_hash(PyObject *self, PyObject *args)
1474{
1475 const Py_UNICODE *str;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001476 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001477 if (!PyArg_ParseTuple(args, "u#", &str, &size))
1478 return NULL;
1479 return PyUnicode_FromUnicode(str, size);
1480}
1481
1482#endif
1483
1484static PyObject *
1485getargs_es(PyObject *self, PyObject *args)
1486{
1487 PyObject *arg, *result;
1488 const char *encoding = NULL;
1489 char *str;
1490
1491 if (!PyArg_ParseTuple(args, "O|s", &arg, &encoding))
1492 return NULL;
1493 if (!PyArg_Parse(arg, "es", encoding, &str))
1494 return NULL;
1495 result = PyBytes_FromString(str);
1496 PyMem_Free(str);
1497 return result;
1498}
1499
1500static PyObject *
1501getargs_et(PyObject *self, PyObject *args)
1502{
1503 PyObject *arg, *result;
1504 const char *encoding = NULL;
1505 char *str;
1506
1507 if (!PyArg_ParseTuple(args, "O|s", &arg, &encoding))
1508 return NULL;
1509 if (!PyArg_Parse(arg, "et", encoding, &str))
1510 return NULL;
1511 result = PyBytes_FromString(str);
1512 PyMem_Free(str);
1513 return result;
1514}
1515
1516static PyObject *
1517getargs_es_hash(PyObject *self, PyObject *args)
1518{
1519 PyObject *arg, *result;
1520 const char *encoding = NULL;
1521 PyByteArrayObject *buffer = NULL;
1522 char *str = NULL;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001523 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001524
1525 if (!PyArg_ParseTuple(args, "O|sO!",
1526 &arg, &encoding, &PyByteArray_Type, &buffer))
1527 return NULL;
1528 if (buffer != NULL) {
1529 str = PyByteArray_AS_STRING(buffer);
1530 size = PyByteArray_GET_SIZE(buffer);
1531 }
1532 if (!PyArg_Parse(arg, "es#", encoding, &str, &size))
1533 return NULL;
1534 result = PyBytes_FromStringAndSize(str, size);
1535 if (buffer == NULL)
1536 PyMem_Free(str);
1537 return result;
1538}
1539
1540static PyObject *
1541getargs_et_hash(PyObject *self, PyObject *args)
1542{
1543 PyObject *arg, *result;
1544 const char *encoding = NULL;
1545 PyByteArrayObject *buffer = NULL;
1546 char *str = NULL;
Serhiy Storchaka4b40c3e2016-01-29 00:55:37 +02001547 int size;
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001548
1549 if (!PyArg_ParseTuple(args, "O|sO!",
1550 &arg, &encoding, &PyByteArray_Type, &buffer))
1551 return NULL;
1552 if (buffer != NULL) {
1553 str = PyByteArray_AS_STRING(buffer);
1554 size = PyByteArray_GET_SIZE(buffer);
1555 }
1556 if (!PyArg_Parse(arg, "et#", encoding, &str, &size))
1557 return NULL;
1558 result = PyBytes_FromStringAndSize(str, size);
1559 if (buffer == NULL)
1560 PyMem_Free(str);
1561 return result;
1562}
1563
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001564#ifdef Py_USING_UNICODE
1565
Benjamin Petersone098c4a2008-12-23 20:12:33 +00001566static volatile int x;
Benjamin Petersonb6a53b52008-12-23 20:09:28 +00001567
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001568/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
1569 of an error.
1570*/
1571static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +00001572test_u_code(PyObject *self)
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001573{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001574 PyObject *tuple, *obj;
1575 Py_UNICODE *value;
1576 int len;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001577
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001578 /* issue4122: Undefined reference to _Py_ascii_whitespace on Windows */
1579 /* Just use the macro and check that it compiles */
1580 x = Py_UNICODE_ISSPACE(25);
Amaury Forgeot d'Arc07d539d2008-10-14 21:47:22 +00001581
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001582 tuple = PyTuple_New(1);
1583 if (tuple == NULL)
1584 return NULL;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001585
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001586 obj = PyUnicode_Decode("test", strlen("test"),
1587 "ascii", NULL);
1588 if (obj == NULL)
1589 return NULL;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001590
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001591 PyTuple_SET_ITEM(tuple, 0, obj);
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001592
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001593 value = 0;
Oren Milman20958e62017-08-29 19:16:12 +03001594 if (!PyArg_ParseTuple(tuple, "u:test_u_code", &value)) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001595 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +03001596 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001597 if (value != PyUnicode_AS_UNICODE(obj))
1598 return raiseTestError("test_u_code",
1599 "u code returned wrong value for u'test'");
1600 value = 0;
Oren Milman20958e62017-08-29 19:16:12 +03001601 if (!PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len)) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001602 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +03001603 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001604 if (value != PyUnicode_AS_UNICODE(obj) ||
1605 len != PyUnicode_GET_SIZE(obj))
1606 return raiseTestError("test_u_code",
1607 "u# code returned wrong values for u'test'");
Tim Petersbaefd9e2003-01-28 20:37:45 +00001608
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001609 Py_DECREF(tuple);
1610 Py_INCREF(Py_None);
1611 return Py_None;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001612}
1613
Tim Peters8931ff12006-05-13 23:28:20 +00001614static PyObject *
Mark Dickinson6b265f12009-03-18 16:07:26 +00001615test_widechar(PyObject *self)
1616{
1617#if defined(SIZEOF_WCHAR_T) && (SIZEOF_WCHAR_T == 4)
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001618 const wchar_t wtext[2] = {(wchar_t)0x10ABCDu};
1619 size_t wtextlen = 1;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001620#else
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001621 const wchar_t wtext[3] = {(wchar_t)0xDBEAu, (wchar_t)0xDFCDu};
1622 size_t wtextlen = 2;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001623#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001624 PyObject *wide, *utf8;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001625
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001626 wide = PyUnicode_FromWideChar(wtext, wtextlen);
1627 if (wide == NULL)
1628 return NULL;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001629
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001630 utf8 = PyUnicode_FromString("\xf4\x8a\xaf\x8d");
1631 if (utf8 == NULL) {
1632 Py_DECREF(wide);
1633 return NULL;
1634 }
Mark Dickinson6b265f12009-03-18 16:07:26 +00001635
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001636 if (PyUnicode_GET_SIZE(wide) != PyUnicode_GET_SIZE(utf8)) {
1637 Py_DECREF(wide);
1638 Py_DECREF(utf8);
1639 return raiseTestError("test_widechar",
1640 "wide string and utf8 string have different length");
1641 }
1642 if (PyUnicode_Compare(wide, utf8)) {
1643 Py_DECREF(wide);
1644 Py_DECREF(utf8);
1645 if (PyErr_Occurred())
1646 return NULL;
1647 return raiseTestError("test_widechar",
1648 "wide string and utf8 string are differents");
1649 }
Mark Dickinson6b265f12009-03-18 16:07:26 +00001650
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001651 Py_DECREF(wide);
1652 Py_DECREF(utf8);
1653 Py_RETURN_NONE;
Mark Dickinson6b265f12009-03-18 16:07:26 +00001654}
1655
1656static PyObject *
Victor Stinner975134e2011-11-22 01:54:19 +01001657unicode_encodedecimal(PyObject *self, PyObject *args)
1658{
1659 Py_UNICODE *unicode;
Victor Stinner9f915d92011-11-29 00:53:09 +01001660 int length;
Victor Stinner975134e2011-11-22 01:54:19 +01001661 char *errors = NULL;
1662 PyObject *decimal;
1663 Py_ssize_t decimal_length, new_length;
1664 int res;
1665
1666 if (!PyArg_ParseTuple(args, "u#|s", &unicode, &length, &errors))
1667 return NULL;
1668
Serhiy Storchakae822b032013-08-06 16:56:26 +03001669 decimal_length = length * 10; /* len('&#1114111;') */
Victor Stinner975134e2011-11-22 01:54:19 +01001670 decimal = PyBytes_FromStringAndSize(NULL, decimal_length);
1671 if (decimal == NULL)
1672 return NULL;
1673
1674 res = PyUnicode_EncodeDecimal(unicode, length,
1675 PyBytes_AS_STRING(decimal),
1676 errors);
1677 if (res < 0) {
1678 Py_DECREF(decimal);
1679 return NULL;
1680 }
1681
1682 new_length = strlen(PyBytes_AS_STRING(decimal));
1683 assert(new_length <= decimal_length);
1684 res = _PyBytes_Resize(&decimal, new_length);
1685 if (res < 0)
1686 return NULL;
1687
1688 return decimal;
1689}
1690
1691static PyObject *
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001692test_empty_argparse(PyObject *self)
1693{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001694 /* Test that formats can begin with '|'. See issue #4720. */
1695 PyObject *tuple, *dict = NULL;
1696 static char *kwlist[] = {NULL};
1697 int result;
1698 tuple = PyTuple_New(0);
1699 if (!tuple)
1700 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +03001701 if (!(result = PyArg_ParseTuple(tuple, "|:test_empty_argparse"))) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001702 goto done;
Oren Milman20958e62017-08-29 19:16:12 +03001703 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001704 dict = PyDict_New();
1705 if (!dict)
1706 goto done;
1707 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|:test_empty_argparse", kwlist);
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001708 done:
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001709 Py_DECREF(tuple);
1710 Py_XDECREF(dict);
Oren Milman20958e62017-08-29 19:16:12 +03001711 if (!result) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001712 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +03001713 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001714 else {
1715 Py_RETURN_NONE;
1716 }
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001717}
1718
1719static PyObject *
Tim Peters8931ff12006-05-13 23:28:20 +00001720codec_incrementalencoder(PyObject *self, PyObject *args)
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001721{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001722 const char *encoding, *errors = NULL;
1723 if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
1724 &encoding, &errors))
1725 return NULL;
1726 return PyCodec_IncrementalEncoder(encoding, errors);
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001727}
1728
Tim Peters8931ff12006-05-13 23:28:20 +00001729static PyObject *
1730codec_incrementaldecoder(PyObject *self, PyObject *args)
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001731{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001732 const char *encoding, *errors = NULL;
1733 if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
1734 &encoding, &errors))
1735 return NULL;
1736 return PyCodec_IncrementalDecoder(encoding, errors);
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001737}
1738
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001739#endif
1740
Tim Peters5b8132f2003-01-31 15:52:05 +00001741/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
Tim Petersbaefd9e2003-01-28 20:37:45 +00001742static PyObject *
1743test_long_numbits(PyObject *self)
1744{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001745 struct triple {
1746 long input;
1747 size_t nbits;
1748 int sign;
1749 } testcases[] = {{0, 0, 0},
1750 {1L, 1, 1},
1751 {-1L, 1, -1},
1752 {2L, 2, 1},
1753 {-2L, 2, -1},
1754 {3L, 2, 1},
1755 {-3L, 2, -1},
1756 {4L, 3, 1},
1757 {-4L, 3, -1},
1758 {0x7fffL, 15, 1}, /* one Python long digit */
1759 {-0x7fffL, 15, -1},
1760 {0xffffL, 16, 1},
1761 {-0xffffL, 16, -1},
1762 {0xfffffffL, 28, 1},
1763 {-0xfffffffL, 28, -1}};
1764 int i;
Tim Petersbaefd9e2003-01-28 20:37:45 +00001765
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001766 for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
1767 PyObject *plong = PyLong_FromLong(testcases[i].input);
1768 size_t nbits = _PyLong_NumBits(plong);
1769 int sign = _PyLong_Sign(plong);
Tim Petersbaefd9e2003-01-28 20:37:45 +00001770
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001771 Py_DECREF(plong);
1772 if (nbits != testcases[i].nbits)
1773 return raiseTestError("test_long_numbits",
1774 "wrong result for _PyLong_NumBits");
1775 if (sign != testcases[i].sign)
1776 return raiseTestError("test_long_numbits",
1777 "wrong result for _PyLong_Sign");
1778 }
1779 Py_INCREF(Py_None);
1780 return Py_None;
Tim Petersbaefd9e2003-01-28 20:37:45 +00001781}
1782
Neal Norwitz4a53dad2006-03-15 05:43:10 +00001783/* Example passing NULLs to PyObject_Str(NULL) and PyObject_Unicode(NULL). */
1784
1785static PyObject *
1786test_null_strings(PyObject *self)
1787{
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001788 PyObject *o1 = PyObject_Str(NULL);
1789#ifdef Py_USING_UNICODE
1790 PyObject *o2 = PyObject_Unicode(NULL);
1791#else
1792 PyObject *o2 = PyObject_Str(NULL);
1793#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001794 PyObject *tuple = PyTuple_Pack(2, o1, o2);
1795 Py_XDECREF(o1);
1796 Py_XDECREF(o2);
1797 return tuple;
Neal Norwitz4a53dad2006-03-15 05:43:10 +00001798}
1799
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001800static PyObject *
1801raise_exception(PyObject *self, PyObject *args)
1802{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001803 PyObject *exc;
1804 PyObject *exc_args, *v;
1805 int num_args, i;
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001806
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001807 if (!PyArg_ParseTuple(args, "Oi:raise_exception",
1808 &exc, &num_args))
1809 return NULL;
1810 if (!PyExceptionClass_Check(exc)) {
1811 PyErr_Format(PyExc_TypeError, "an exception class is required");
1812 return NULL;
1813 }
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001814
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001815 exc_args = PyTuple_New(num_args);
1816 if (exc_args == NULL)
1817 return NULL;
1818 for (i = 0; i < num_args; ++i) {
1819 v = PyInt_FromLong(i);
1820 if (v == NULL) {
1821 Py_DECREF(exc_args);
1822 return NULL;
1823 }
1824 PyTuple_SET_ITEM(exc_args, i, v);
1825 }
1826 PyErr_SetObject(exc, exc_args);
1827 Py_DECREF(exc_args);
1828 return NULL;
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001829}
Tim Peters91621db2001-06-12 20:10:01 +00001830
Antoine Pitrou0d2fac12015-06-23 14:31:11 +02001831static PyObject *
1832set_errno(PyObject *self, PyObject *args)
1833{
1834 int new_errno;
1835
1836 if (!PyArg_ParseTuple(args, "i:set_errno", &new_errno))
1837 return NULL;
1838
1839 errno = new_errno;
1840 Py_RETURN_NONE;
1841}
Benjamin Peterson46bff792010-01-30 23:28:38 +00001842
Benjamin Peterson888a29f2017-01-15 23:57:56 -08001843#if defined(Py_USING_UNICODE) && !defined(Py_BUILD_CORE)
Benjamin Peterson46bff792010-01-30 23:28:38 +00001844static int test_run_counter = 0;
1845
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001846static PyObject *
1847test_datetime_capi(PyObject *self, PyObject *args) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001848 if (PyDateTimeAPI) {
1849 if (test_run_counter) {
1850 /* Probably regrtest.py -R */
1851 Py_RETURN_NONE;
1852 }
1853 else {
1854 PyErr_SetString(PyExc_AssertionError,
1855 "PyDateTime_CAPI somehow initialized");
1856 return NULL;
1857 }
1858 }
1859 test_run_counter++;
1860 PyDateTime_IMPORT;
1861 if (PyDateTimeAPI)
1862 Py_RETURN_NONE;
1863 else
1864 return NULL;
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001865}
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02001866#endif
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001867
Benjamin Petersona04ae012010-01-30 23:26:05 +00001868
1869#ifdef WITH_THREAD
1870
Tim Peters59b96c12006-03-21 03:58:41 +00001871/* test_thread_state spawns a thread of its own, and that thread releases
1872 * `thread_done` when it's finished. The driver code has to know when the
1873 * thread finishes, because the thread uses a PyObject (the callable) that
1874 * may go away when the driver finishes. The former lack of this explicit
1875 * synchronization caused rare segfaults, so rare that they were seen only
1876 * on a Mac buildbot (although they were possible on any box).
1877 */
1878static PyThread_type_lock thread_done = NULL;
1879
Benjamin Peterson37346b22008-08-23 20:27:43 +00001880static int
Tim Peters59b96c12006-03-21 03:58:41 +00001881_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001882{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001883 PyObject *rc;
1884 int success;
1885 PyGILState_STATE s = PyGILState_Ensure();
1886 rc = PyObject_CallFunction((PyObject *)callable, "");
1887 success = (rc != NULL);
1888 Py_XDECREF(rc);
1889 PyGILState_Release(s);
1890 return success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001891}
1892
Tim Peters59b96c12006-03-21 03:58:41 +00001893/* Same thing, but releases `thread_done` when it returns. This variant
1894 * should be called only from threads spawned by test_thread_state().
1895 */
1896static void
1897_make_call_from_thread(void *callable)
1898{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001899 _make_call(callable);
1900 PyThread_release_lock(thread_done);
Tim Peters59b96c12006-03-21 03:58:41 +00001901}
1902
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001903static PyObject *
1904test_thread_state(PyObject *self, PyObject *args)
1905{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001906 PyObject *fn;
1907 int success = 1;
Tim Peters59b96c12006-03-21 03:58:41 +00001908
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001909 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
1910 return NULL;
Tim Peters59b96c12006-03-21 03:58:41 +00001911
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001912 if (!PyCallable_Check(fn)) {
1913 PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
1914 fn->ob_type->tp_name);
1915 return NULL;
1916 }
Benjamin Peterson37346b22008-08-23 20:27:43 +00001917
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001918 /* Ensure Python is set up for threading */
1919 PyEval_InitThreads();
1920 thread_done = PyThread_allocate_lock();
1921 if (thread_done == NULL)
1922 return PyErr_NoMemory();
1923 PyThread_acquire_lock(thread_done, 1);
Tim Peters59b96c12006-03-21 03:58:41 +00001924
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001925 /* Start a new thread with our callback. */
1926 PyThread_start_new_thread(_make_call_from_thread, fn);
1927 /* Make the callback with the thread lock held by this thread */
1928 success &= _make_call(fn);
1929 /* Do it all again, but this time with the thread-lock released */
1930 Py_BEGIN_ALLOW_THREADS
1931 success &= _make_call(fn);
1932 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
1933 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001934
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001935 /* And once more with and without a thread
1936 XXX - should use a lock and work out exactly what we are trying
1937 to test <wink>
1938 */
1939 Py_BEGIN_ALLOW_THREADS
1940 PyThread_start_new_thread(_make_call_from_thread, fn);
1941 success &= _make_call(fn);
1942 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
1943 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001944
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001945 /* Release lock we acquired above. This is required on HP-UX. */
1946 PyThread_release_lock(thread_done);
Neal Norwitz97a57222006-10-28 21:17:51 +00001947
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001948 PyThread_free_lock(thread_done);
1949 if (!success)
1950 return NULL;
1951 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001952}
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001953
1954/* test Py_AddPendingCalls using threads */
1955static int _pending_callback(void *arg)
1956{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001957 /* we assume the argument is callable object to which we own a reference */
1958 PyObject *callable = (PyObject *)arg;
1959 PyObject *r = PyObject_CallObject(callable, NULL);
1960 Py_DECREF(callable);
1961 Py_XDECREF(r);
1962 return r != NULL ? 0 : -1;
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001963}
1964
1965/* The following requests n callbacks to _pending_callback. It can be
1966 * run from any python thread.
1967 */
1968PyObject *pending_threadfunc(PyObject *self, PyObject *arg)
1969{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001970 PyObject *callable;
1971 int r;
1972 if (PyArg_ParseTuple(arg, "O", &callable) == 0)
1973 return NULL;
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001974
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001975 /* create the reference for the callbackwhile we hold the lock */
1976 Py_INCREF(callable);
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001977
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001978 Py_BEGIN_ALLOW_THREADS
1979 r = Py_AddPendingCall(&_pending_callback, callable);
1980 Py_END_ALLOW_THREADS
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001981
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001982 if (r<0) {
1983 Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
1984 Py_INCREF(Py_False);
1985 return Py_False;
1986 }
1987 Py_INCREF(Py_True);
1988 return Py_True;
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001989}
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001990#endif
1991
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001992/* Some tests of PyString_FromFormat(). This needs more tests. */
Tim Peters8931ff12006-05-13 23:28:20 +00001993static PyObject *
1994test_string_from_format(PyObject *self, PyObject *args)
1995{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001996 PyObject *result;
1997 char *msg;
Tim Peters8931ff12006-05-13 23:28:20 +00001998
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001999#define CHECK_1_FORMAT(FORMAT, TYPE) \
2000 result = PyString_FromFormat(FORMAT, (TYPE)1); \
2001 if (result == NULL) \
2002 return NULL; \
2003 if (strcmp(PyString_AsString(result), "1")) { \
2004 msg = FORMAT " failed at 1"; \
2005 goto Fail; \
2006 } \
2007 Py_DECREF(result)
Tim Peters8931ff12006-05-13 23:28:20 +00002008
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002009 CHECK_1_FORMAT("%d", int);
2010 CHECK_1_FORMAT("%ld", long);
2011 /* The z width modifier was added in Python 2.5. */
2012 CHECK_1_FORMAT("%zd", Py_ssize_t);
Tim Peters8931ff12006-05-13 23:28:20 +00002013
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002014 /* The u type code was added in Python 2.5. */
2015 CHECK_1_FORMAT("%u", unsigned int);
2016 CHECK_1_FORMAT("%lu", unsigned long);
2017 CHECK_1_FORMAT("%zu", size_t);
Tim Peters8931ff12006-05-13 23:28:20 +00002018
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002019 /* "%lld" and "%llu" support added in Python 2.7. */
Mark Dickinson82864d12009-11-15 16:18:58 +00002020#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002021 CHECK_1_FORMAT("%llu", unsigned PY_LONG_LONG);
2022 CHECK_1_FORMAT("%lld", PY_LONG_LONG);
Mark Dickinson82864d12009-11-15 16:18:58 +00002023#endif
2024
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002025 Py_RETURN_NONE;
Tim Peters8931ff12006-05-13 23:28:20 +00002026
2027 Fail:
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002028 Py_XDECREF(result);
2029 return raiseTestError("test_string_from_format", msg);
Tim Peters8931ff12006-05-13 23:28:20 +00002030
2031#undef CHECK_1_FORMAT
2032}
2033
Larry Hastings402b73f2010-03-25 00:54:54 +00002034/* Coverage testing of capsule objects. */
2035
2036static const char *capsule_name = "capsule name";
2037static char *capsule_pointer = "capsule pointer";
2038static char *capsule_context = "capsule context";
2039static const char *capsule_error = NULL;
2040static int
2041capsule_destructor_call_count = 0;
2042
2043static void
2044capsule_destructor(PyObject *o) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002045 capsule_destructor_call_count++;
2046 if (PyCapsule_GetContext(o) != capsule_context) {
2047 capsule_error = "context did not match in destructor!";
2048 } else if (PyCapsule_GetDestructor(o) != capsule_destructor) {
2049 capsule_error = "destructor did not match in destructor! (woah!)";
2050 } else if (PyCapsule_GetName(o) != capsule_name) {
2051 capsule_error = "name did not match in destructor!";
2052 } else if (PyCapsule_GetPointer(o, capsule_name) != capsule_pointer) {
2053 capsule_error = "pointer did not match in destructor!";
2054 }
Larry Hastings402b73f2010-03-25 00:54:54 +00002055}
2056
2057typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002058 char *name;
2059 char *module;
2060 char *attribute;
Larry Hastings402b73f2010-03-25 00:54:54 +00002061} known_capsule;
2062
2063static PyObject *
2064test_capsule(PyObject *self, PyObject *args)
2065{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002066 PyObject *object;
2067 const char *error = NULL;
2068 void *pointer;
2069 void *pointer2;
2070 known_capsule known_capsules[] = {
2071 #define KNOWN_CAPSULE(module, name) { module "." name, module, name }
2072 KNOWN_CAPSULE("_socket", "CAPI"),
2073 KNOWN_CAPSULE("_curses", "_C_API"),
2074 KNOWN_CAPSULE("datetime", "datetime_CAPI"),
2075 { NULL, NULL },
2076 };
2077 known_capsule *known = &known_capsules[0];
Larry Hastings402b73f2010-03-25 00:54:54 +00002078
2079#define FAIL(x) { error = (x); goto exit; }
2080
2081#define CHECK_DESTRUCTOR \
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002082 if (capsule_error) { \
2083 FAIL(capsule_error); \
2084 } \
2085 else if (!capsule_destructor_call_count) { \
2086 FAIL("destructor not called!"); \
2087 } \
2088 capsule_destructor_call_count = 0; \
Larry Hastings402b73f2010-03-25 00:54:54 +00002089
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002090 object = PyCapsule_New(capsule_pointer, capsule_name, capsule_destructor);
2091 PyCapsule_SetContext(object, capsule_context);
2092 capsule_destructor(object);
2093 CHECK_DESTRUCTOR;
2094 Py_DECREF(object);
2095 CHECK_DESTRUCTOR;
Larry Hastings402b73f2010-03-25 00:54:54 +00002096
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002097 object = PyCapsule_New(known, "ignored", NULL);
2098 PyCapsule_SetPointer(object, capsule_pointer);
2099 PyCapsule_SetName(object, capsule_name);
2100 PyCapsule_SetDestructor(object, capsule_destructor);
2101 PyCapsule_SetContext(object, capsule_context);
2102 capsule_destructor(object);
2103 CHECK_DESTRUCTOR;
2104 /* intentionally access using the wrong name */
2105 pointer2 = PyCapsule_GetPointer(object, "the wrong name");
2106 if (!PyErr_Occurred()) {
2107 FAIL("PyCapsule_GetPointer should have failed but did not!");
2108 }
2109 PyErr_Clear();
2110 if (pointer2) {
2111 if (pointer2 == capsule_pointer) {
2112 FAIL("PyCapsule_GetPointer should not have"
2113 " returned the internal pointer!");
2114 } else {
2115 FAIL("PyCapsule_GetPointer should have "
2116 "returned NULL pointer but did not!");
2117 }
2118 }
2119 PyCapsule_SetDestructor(object, NULL);
2120 Py_DECREF(object);
2121 if (capsule_destructor_call_count) {
2122 FAIL("destructor called when it should not have been!");
2123 }
Larry Hastings402b73f2010-03-25 00:54:54 +00002124
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002125 for (known = &known_capsules[0]; known->module != NULL; known++) {
2126 /* yeah, ordinarily I wouldn't do this either,
2127 but it's fine for this test harness.
2128 */
2129 static char buffer[256];
Larry Hastings402b73f2010-03-25 00:54:54 +00002130#undef FAIL
2131#define FAIL(x) \
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002132 { \
2133 sprintf(buffer, "%s module: \"%s\" attribute: \"%s\"", \
2134 x, known->module, known->attribute); \
2135 error = buffer; \
2136 goto exit; \
2137 } \
Larry Hastings402b73f2010-03-25 00:54:54 +00002138
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002139 PyObject *module = PyImport_ImportModule(known->module);
2140 if (module) {
2141 pointer = PyCapsule_Import(known->name, 0);
2142 if (!pointer) {
2143 Py_DECREF(module);
2144 FAIL("PyCapsule_GetPointer returned NULL unexpectedly!");
2145 }
2146 object = PyObject_GetAttrString(module, known->attribute);
2147 if (!object) {
2148 Py_DECREF(module);
2149 return NULL;
2150 }
2151 pointer2 = PyCapsule_GetPointer(object,
2152 "weebles wobble but they don't fall down");
2153 if (!PyErr_Occurred()) {
2154 Py_DECREF(object);
2155 Py_DECREF(module);
2156 FAIL("PyCapsule_GetPointer should have failed but did not!");
2157 }
2158 PyErr_Clear();
2159 if (pointer2) {
2160 Py_DECREF(module);
2161 Py_DECREF(object);
2162 if (pointer2 == pointer) {
2163 FAIL("PyCapsule_GetPointer should not have"
2164 " returned its internal pointer!");
2165 } else {
2166 FAIL("PyCapsule_GetPointer should have"
2167 " returned NULL pointer but did not!");
2168 }
2169 }
2170 Py_DECREF(object);
2171 Py_DECREF(module);
2172 }
2173 else
2174 PyErr_Clear();
2175 }
Larry Hastings402b73f2010-03-25 00:54:54 +00002176
2177 exit:
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002178 if (error) {
2179 return raiseTestError("test_capsule", error);
2180 }
2181 Py_RETURN_NONE;
Larry Hastings402b73f2010-03-25 00:54:54 +00002182#undef FAIL
2183}
2184
Georg Brandle9462c72006-08-04 18:03:37 +00002185/* This is here to provide a docstring for test_descr. */
2186static PyObject *
2187test_with_docstring(PyObject *self)
2188{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002189 Py_RETURN_NONE;
Georg Brandle9462c72006-08-04 18:03:37 +00002190}
2191
Brett Cannon141534e2008-04-28 03:23:50 +00002192/* To test the format of tracebacks as printed out. */
2193static PyObject *
Brett Cannon8dc43032008-04-28 04:50:06 +00002194traceback_print(PyObject *self, PyObject *args)
Brett Cannon141534e2008-04-28 03:23:50 +00002195{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002196 PyObject *file;
2197 PyObject *traceback;
2198 int result;
2199
2200 if (!PyArg_ParseTuple(args, "OO:traceback_print",
2201 &traceback, &file))
2202 return NULL;
2203
2204 result = PyTraceBack_Print(traceback, file);
2205 if (result < 0)
2206 return NULL;
2207 Py_RETURN_NONE;
Brett Cannon141534e2008-04-28 03:23:50 +00002208}
2209
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00002210/* To test that the result of PyCode_NewEmpty has the right members. */
2211static PyObject *
2212code_newempty(PyObject *self, PyObject *args)
2213{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002214 const char *filename;
2215 const char *funcname;
2216 int firstlineno;
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00002217
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002218 if (!PyArg_ParseTuple(args, "ssi:code_newempty",
2219 &filename, &funcname, &firstlineno))
2220 return NULL;
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00002221
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002222 return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00002223}
2224
Georg Brandl740cdc32009-12-28 08:34:58 +00002225/* Test PyErr_NewExceptionWithDoc (also exercise PyErr_NewException).
2226 Run via Lib/test/test_exceptions.py */
2227static PyObject *
2228make_exception_with_doc(PyObject *self, PyObject *args, PyObject *kwargs)
2229{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002230 char *name;
2231 char *doc = NULL;
2232 PyObject *base = NULL;
2233 PyObject *dict = NULL;
Georg Brandl740cdc32009-12-28 08:34:58 +00002234
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002235 static char *kwlist[] = {"name", "doc", "base", "dict", NULL};
Georg Brandl740cdc32009-12-28 08:34:58 +00002236
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002237 if (!PyArg_ParseTupleAndKeywords(args, kwargs,
2238 "s|sOO:make_exception_with_doc", kwlist,
2239 &name, &doc, &base, &dict))
2240 return NULL;
Georg Brandl740cdc32009-12-28 08:34:58 +00002241
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002242 return PyErr_NewExceptionWithDoc(name, doc, base, dict);
Georg Brandl740cdc32009-12-28 08:34:58 +00002243}
2244
Benjamin Petersona7b09762011-10-15 13:43:21 -04002245static PyObject *
2246sequence_delitem(PyObject *self, PyObject *args)
2247{
2248 PyObject *seq;
2249 Py_ssize_t i;
2250
2251 if (!PyArg_ParseTuple(args, "On", &seq, &i))
2252 return NULL;
2253 if (PySequence_DelItem(seq, i) < 0)
2254 return NULL;
2255 Py_RETURN_NONE;
2256}
2257
Victor Stinner66c6e9d2013-12-13 02:37:09 +01002258#ifdef WITH_THREAD
2259typedef struct {
2260 PyThread_type_lock start_event;
2261 PyThread_type_lock exit_event;
2262 PyObject *callback;
2263} test_c_thread_t;
2264
2265static void
2266temporary_c_thread(void *data)
2267{
2268 test_c_thread_t *test_c_thread = data;
2269 PyGILState_STATE state;
2270 PyObject *res;
2271
2272 PyThread_release_lock(test_c_thread->start_event);
2273
2274 /* Allocate a Python thread state for this thread */
2275 state = PyGILState_Ensure();
2276
2277 res = PyObject_CallFunction(test_c_thread->callback, "", NULL);
2278 Py_CLEAR(test_c_thread->callback);
2279
2280 if (res == NULL) {
2281 PyErr_Print();
2282 }
2283 else {
2284 Py_DECREF(res);
2285 }
2286
2287 /* Destroy the Python thread state for this thread */
2288 PyGILState_Release(state);
2289
2290 PyThread_release_lock(test_c_thread->exit_event);
2291
2292 PyThread_exit_thread();
2293}
2294
2295static PyObject *
2296call_in_temporary_c_thread(PyObject *self, PyObject *callback)
2297{
2298 PyObject *res = NULL;
2299 test_c_thread_t test_c_thread;
2300 long thread;
2301
2302 PyEval_InitThreads();
2303
2304 test_c_thread.start_event = PyThread_allocate_lock();
2305 test_c_thread.exit_event = PyThread_allocate_lock();
2306 test_c_thread.callback = NULL;
2307 if (!test_c_thread.start_event || !test_c_thread.exit_event) {
2308 PyErr_SetString(PyExc_RuntimeError, "could not allocate lock");
2309 goto exit;
2310 }
2311
2312 Py_INCREF(callback);
2313 test_c_thread.callback = callback;
2314
2315 PyThread_acquire_lock(test_c_thread.start_event, 1);
2316 PyThread_acquire_lock(test_c_thread.exit_event, 1);
2317
2318 thread = PyThread_start_new_thread(temporary_c_thread, &test_c_thread);
2319 if (thread == -1) {
2320 PyErr_SetString(PyExc_RuntimeError, "unable to start the thread");
2321 PyThread_release_lock(test_c_thread.start_event);
2322 PyThread_release_lock(test_c_thread.exit_event);
2323 goto exit;
2324 }
2325
2326 PyThread_acquire_lock(test_c_thread.start_event, 1);
2327 PyThread_release_lock(test_c_thread.start_event);
2328
2329 Py_BEGIN_ALLOW_THREADS
2330 PyThread_acquire_lock(test_c_thread.exit_event, 1);
2331 PyThread_release_lock(test_c_thread.exit_event);
2332 Py_END_ALLOW_THREADS
2333
2334 Py_INCREF(Py_None);
2335 res = Py_None;
2336
2337exit:
2338 Py_CLEAR(test_c_thread.callback);
2339 if (test_c_thread.start_event)
2340 PyThread_free_lock(test_c_thread.start_event);
2341 if (test_c_thread.exit_event)
2342 PyThread_free_lock(test_c_thread.exit_event);
2343 return res;
2344}
2345#endif /* WITH_THREAD */
2346
Serhiy Storchaka4068b012015-02-06 08:56:33 +02002347/* marshal */
2348
2349static PyObject*
2350pymarshal_write_long_to_file(PyObject* self, PyObject *args)
2351{
2352 long value;
2353 char *filename;
2354 int version;
2355 FILE *fp;
2356
2357 if (!PyArg_ParseTuple(args, "lsi:pymarshal_write_long_to_file",
2358 &value, &filename, &version))
2359 return NULL;
2360
2361 fp = fopen(filename, "wb");
2362 if (fp == NULL) {
2363 PyErr_SetFromErrno(PyExc_OSError);
2364 return NULL;
2365 }
2366
2367 PyMarshal_WriteLongToFile(value, fp, version);
2368
2369 fclose(fp);
2370 if (PyErr_Occurred())
2371 return NULL;
2372 Py_RETURN_NONE;
2373}
2374
2375static PyObject*
2376pymarshal_write_object_to_file(PyObject* self, PyObject *args)
2377{
2378 PyObject *obj;
2379 char *filename;
2380 int version;
2381 FILE *fp;
2382
2383 if (!PyArg_ParseTuple(args, "Osi:pymarshal_write_object_to_file",
2384 &obj, &filename, &version))
2385 return NULL;
2386
2387 fp = fopen(filename, "wb");
2388 if (fp == NULL) {
2389 PyErr_SetFromErrno(PyExc_OSError);
2390 return NULL;
2391 }
2392
2393 PyMarshal_WriteObjectToFile(obj, fp, version);
2394
2395 fclose(fp);
2396 if (PyErr_Occurred())
2397 return NULL;
2398 Py_RETURN_NONE;
2399}
2400
2401static PyObject*
2402pymarshal_read_short_from_file(PyObject* self, PyObject *args)
2403{
2404 int value;
2405 long pos;
2406 char *filename;
2407 FILE *fp;
2408
2409 if (!PyArg_ParseTuple(args, "s:pymarshal_read_short_from_file", &filename))
2410 return NULL;
2411
2412 fp = fopen(filename, "rb");
2413 if (fp == NULL) {
2414 PyErr_SetFromErrno(PyExc_OSError);
2415 return NULL;
2416 }
2417
2418 value = PyMarshal_ReadShortFromFile(fp);
2419 pos = ftell(fp);
2420
2421 fclose(fp);
2422 if (PyErr_Occurred())
2423 return NULL;
2424 return Py_BuildValue("il", value, pos);
2425}
2426
2427static PyObject*
2428pymarshal_read_long_from_file(PyObject* self, PyObject *args)
2429{
2430 long value, pos;
2431 char *filename;
2432 FILE *fp;
2433
2434 if (!PyArg_ParseTuple(args, "s:pymarshal_read_long_from_file", &filename))
2435 return NULL;
2436
2437 fp = fopen(filename, "rb");
2438 if (fp == NULL) {
2439 PyErr_SetFromErrno(PyExc_OSError);
2440 return NULL;
2441 }
2442
2443 value = PyMarshal_ReadLongFromFile(fp);
2444 pos = ftell(fp);
2445
2446 fclose(fp);
2447 if (PyErr_Occurred())
2448 return NULL;
2449 return Py_BuildValue("ll", value, pos);
2450}
2451
2452static PyObject*
2453pymarshal_read_last_object_from_file(PyObject* self, PyObject *args)
2454{
2455 PyObject *obj;
2456 long pos;
2457 char *filename;
2458 FILE *fp;
2459
2460 if (!PyArg_ParseTuple(args, "s:pymarshal_read_last_object_from_file", &filename))
2461 return NULL;
2462
2463 fp = fopen(filename, "rb");
2464 if (fp == NULL) {
2465 PyErr_SetFromErrno(PyExc_OSError);
2466 return NULL;
2467 }
2468
2469 obj = PyMarshal_ReadLastObjectFromFile(fp);
2470 pos = ftell(fp);
2471
2472 fclose(fp);
2473 return Py_BuildValue("Nl", obj, pos);
2474}
2475
2476static PyObject*
2477pymarshal_read_object_from_file(PyObject* self, PyObject *args)
2478{
2479 PyObject *obj;
2480 long pos;
2481 char *filename;
2482 FILE *fp;
2483
2484 if (!PyArg_ParseTuple(args, "s:pymarshal_read_object_from_file", &filename))
2485 return NULL;
2486
2487 fp = fopen(filename, "rb");
2488 if (fp == NULL) {
2489 PyErr_SetFromErrno(PyExc_OSError);
2490 return NULL;
2491 }
2492
2493 obj = PyMarshal_ReadObjectFromFile(fp);
2494 pos = ftell(fp);
2495
2496 fclose(fp);
2497 return Py_BuildValue("Nl", obj, pos);
2498}
2499
Victor Stinnerd2aff602017-05-09 13:57:20 +02002500static PyObject*
2501test_raise_signal(PyObject* self, PyObject *args)
2502{
2503 int signum, err;
2504
Oren Milman20958e62017-08-29 19:16:12 +03002505 if (!PyArg_ParseTuple(args, "i:raise_signal", &signum)) {
Victor Stinnerd2aff602017-05-09 13:57:20 +02002506 return NULL;
Oren Milman20958e62017-08-29 19:16:12 +03002507 }
Victor Stinnerd2aff602017-05-09 13:57:20 +02002508
2509 err = raise(signum);
2510 if (err)
2511 return PyErr_SetFromErrno(PyExc_OSError);
2512
2513 if (PyErr_CheckSignals() < 0)
2514 return NULL;
2515
2516 Py_RETURN_NONE;
2517}
2518
Victor Stinner66c6e9d2013-12-13 02:37:09 +01002519
Victor Stinner95a91df2017-06-27 14:31:40 +02002520#ifdef MS_WINDOWS
2521static PyObject*
2522msvcrt_CrtSetReportMode(PyObject* self, PyObject *args)
2523{
2524 int type, mode;
2525 int res;
2526
2527 if (!PyArg_ParseTuple(args, "ii:CrtSetReportMode", &type, &mode)) {
2528 return NULL;
2529 }
2530
2531 res = _CrtSetReportMode(type, mode);
2532 if (res == -1) {
2533 PyErr_SetFromErrno(PyExc_OSError);
2534 return NULL;
2535 }
2536 return PyInt_FromLong(res);
2537}
2538
Victor Stinner4dea0652017-08-11 14:38:37 +02002539
Victor Stinner95a91df2017-06-27 14:31:40 +02002540static PyObject*
2541msvcrt_CrtSetReportFile(PyObject* self, PyObject *args)
2542{
2543 int type, file;
2544 long res;
2545
2546 if (!PyArg_ParseTuple(args, "ii:CrtSetReportFile", &type, &file)) {
2547 return NULL;
2548 }
2549
2550 res = (long)_CrtSetReportFile(type, (_HFILE)file);
2551
2552 return PyInt_FromLong(res);
2553}
2554#endif
2555
2556
Victor Stinner4dea0652017-08-11 14:38:37 +02002557#ifdef W_STOPCODE
2558static PyObject*
2559py_w_stopcode(PyObject *self, PyObject *args)
2560{
2561 int sig, status;
2562 if (!PyArg_ParseTuple(args, "i", &sig)) {
2563 return NULL;
2564 }
2565 status = W_STOPCODE(sig);
2566 return PyLong_FromLong(status);
2567}
2568#endif
2569
2570
Tim Peters9ea17ac2001-02-02 05:57:15 +00002571static PyMethodDef TestMethods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002572 {"raise_exception", raise_exception, METH_VARARGS},
Antoine Pitrou0d2fac12015-06-23 14:31:11 +02002573 {"set_errno", set_errno, METH_VARARGS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002574 {"test_config", (PyCFunction)test_config, METH_NOARGS},
Benjamin Peterson888a29f2017-01-15 23:57:56 -08002575#if defined(Py_USING_UNICODE) && !defined(Py_BUILD_CORE)
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002576 {"test_datetime_capi", test_datetime_capi, METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002577#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002578 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
2579 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
2580 {"test_lazy_hash_inheritance", (PyCFunction)test_lazy_hash_inheritance,METH_NOARGS},
2581 {"test_broken_memoryview", (PyCFunction)test_broken_memoryview,METH_NOARGS},
Stefan Krah5606cd92015-01-30 20:11:10 +01002582 {"test_to_contiguous", (PyCFunction)test_to_contiguous, METH_NOARGS},
Stefan Krah0a7fc532015-02-01 16:10:35 +01002583 {"test_from_contiguous", (PyCFunction)test_from_contiguous, METH_NOARGS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002584 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
2585 {"test_long_and_overflow", (PyCFunction)test_long_and_overflow,
2586 METH_NOARGS},
2587 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
2588 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002589#ifdef Py_USING_UNICODE
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002590 {"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002591#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002592 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
2593 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
2594 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
2595 PyDoc_STR("This is a pretty normal docstring.")},
Thomas Heller3457e4b2003-04-24 16:14:27 +00002596
Serhiy Storchaka12cf60c2016-05-20 22:31:24 +03002597 {"test_buildvalue_N", test_buildvalue_N, METH_NOARGS},
Serhiy Storchakac6a7c962016-05-16 09:55:32 +03002598 {"get_args", get_args, METH_VARARGS},
2599 {"get_kwargs", (PyCFunction)get_kwargs, METH_VARARGS|METH_KEYWORDS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002600 {"getargs_tuple", getargs_tuple, METH_VARARGS},
2601 {"getargs_keywords", (PyCFunction)getargs_keywords,
2602 METH_VARARGS|METH_KEYWORDS},
2603 {"getargs_b", getargs_b, METH_VARARGS},
2604 {"getargs_B", getargs_B, METH_VARARGS},
2605 {"getargs_h", getargs_h, METH_VARARGS},
2606 {"getargs_H", getargs_H, METH_VARARGS},
2607 {"getargs_I", getargs_I, METH_VARARGS},
2608 {"getargs_k", getargs_k, METH_VARARGS},
2609 {"getargs_i", getargs_i, METH_VARARGS},
2610 {"getargs_l", getargs_l, METH_VARARGS},
2611 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00002612#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002613 {"getargs_L", getargs_L, METH_VARARGS},
2614 {"getargs_K", getargs_K, METH_VARARGS},
2615 {"test_longlong_api", test_longlong_api, METH_NOARGS},
2616 {"test_long_long_and_overflow",
2617 (PyCFunction)test_long_long_and_overflow, METH_NOARGS},
2618 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002619#endif
Serhiy Storchakad0dc72b2016-05-16 10:12:02 +03002620 {"getargs_f", getargs_f, METH_VARARGS},
2621 {"getargs_d", getargs_d, METH_VARARGS},
2622 {"getargs_D", getargs_D, METH_VARARGS},
2623 {"getargs_S", getargs_S, METH_VARARGS},
2624 {"getargs_Y", getargs_Y, METH_VARARGS},
2625#ifdef Py_USING_UNICODE
2626 {"getargs_U", getargs_U, METH_VARARGS},
2627#endif
Serhiy Storchaka3ecb6ab2016-01-28 19:56:04 +02002628 {"getargs_c", getargs_c, METH_VARARGS},
2629 {"getargs_s", getargs_s, METH_VARARGS},
2630 {"getargs_s_star", getargs_s_star, METH_VARARGS},
2631 {"getargs_s_hash", getargs_s_hash, METH_VARARGS},
2632 {"getargs_t_hash", getargs_t_hash, METH_VARARGS},
2633 {"getargs_z", getargs_z, METH_VARARGS},
2634 {"getargs_z_star", getargs_z_star, METH_VARARGS},
2635 {"getargs_z_hash", getargs_z_hash, METH_VARARGS},
2636 {"getargs_w", getargs_w, METH_VARARGS},
2637 {"getargs_w_star", getargs_w_star, METH_VARARGS},
2638 {"getargs_w_hash", getargs_w_hash, METH_VARARGS},
2639#ifdef Py_USING_UNICODE
2640 {"getargs_u", getargs_u, METH_VARARGS},
2641 {"getargs_u_hash", getargs_u_hash, METH_VARARGS},
2642#endif
2643 {"getargs_es", getargs_es, METH_VARARGS},
2644 {"getargs_et", getargs_et, METH_VARARGS},
2645 {"getargs_es_hash", getargs_es_hash, METH_VARARGS},
2646 {"getargs_et_hash", getargs_et_hash, METH_VARARGS},
2647#ifdef Py_USING_UNICODE
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002648 {"codec_incrementalencoder",
2649 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
2650 {"codec_incrementaldecoder",
2651 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002652 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
2653 {"test_widechar", (PyCFunction)test_widechar, METH_NOARGS},
Victor Stinner975134e2011-11-22 01:54:19 +01002654 {"unicode_encodedecimal", unicode_encodedecimal, METH_VARARGS},
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00002655#endif
Mark Hammond8d98d2c2003-04-19 15:41:53 +00002656#ifdef WITH_THREAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002657 {"_test_thread_state", test_thread_state, METH_VARARGS},
2658 {"_pending_threadfunc", pending_threadfunc, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00002659#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002660 {"test_capsule", (PyCFunction)test_capsule, METH_NOARGS},
2661 {"traceback_print", traceback_print, METH_VARARGS},
2662 {"code_newempty", code_newempty, METH_VARARGS},
2663 {"make_exception_with_doc", (PyCFunction)make_exception_with_doc,
2664 METH_VARARGS | METH_KEYWORDS},
Benjamin Petersona7b09762011-10-15 13:43:21 -04002665 {"sequence_delitem", (PyCFunction)sequence_delitem, METH_VARARGS},
Victor Stinner66c6e9d2013-12-13 02:37:09 +01002666#ifdef WITH_THREAD
2667 {"call_in_temporary_c_thread", call_in_temporary_c_thread, METH_O,
2668 PyDoc_STR("set_error_class(error_class) -> None")},
2669#endif
Serhiy Storchaka4068b012015-02-06 08:56:33 +02002670 {"pymarshal_write_long_to_file",
2671 pymarshal_write_long_to_file, METH_VARARGS},
2672 {"pymarshal_write_object_to_file",
2673 pymarshal_write_object_to_file, METH_VARARGS},
2674 {"pymarshal_read_short_from_file",
2675 pymarshal_read_short_from_file, METH_VARARGS},
2676 {"pymarshal_read_long_from_file",
2677 pymarshal_read_long_from_file, METH_VARARGS},
2678 {"pymarshal_read_last_object_from_file",
2679 pymarshal_read_last_object_from_file, METH_VARARGS},
2680 {"pymarshal_read_object_from_file",
2681 pymarshal_read_object_from_file, METH_VARARGS},
Victor Stinnerd2aff602017-05-09 13:57:20 +02002682 {"raise_signal", (PyCFunction)test_raise_signal, METH_VARARGS},
Victor Stinner95a91df2017-06-27 14:31:40 +02002683#ifdef MS_WINDOWS
2684 {"CrtSetReportMode", (PyCFunction)msvcrt_CrtSetReportMode, METH_VARARGS},
2685 {"CrtSetReportFile", (PyCFunction)msvcrt_CrtSetReportFile, METH_VARARGS},
2686#endif
Victor Stinner4dea0652017-08-11 14:38:37 +02002687#ifdef W_STOPCODE
2688 {"W_STOPCODE", py_w_stopcode, METH_VARARGS},
2689#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002690 {NULL, NULL} /* sentinel */
Tim Peters9ea17ac2001-02-02 05:57:15 +00002691};
2692
Thomas Hellera4ea6032003-04-17 18:55:45 +00002693#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
2694
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002695typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002696 char bool_member;
2697 char byte_member;
2698 unsigned char ubyte_member;
2699 short short_member;
2700 unsigned short ushort_member;
2701 int int_member;
2702 unsigned int uint_member;
2703 long long_member;
2704 unsigned long ulong_member;
2705 float float_member;
2706 double double_member;
2707 char inplace_member[6];
Martin v. Löwis6371cd82007-06-09 07:42:52 +00002708#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002709 PY_LONG_LONG longlong_member;
2710 unsigned PY_LONG_LONG ulonglong_member;
Martin v. Löwis6371cd82007-06-09 07:42:52 +00002711#endif
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002712} all_structmembers;
2713
2714typedef struct {
2715 PyObject_HEAD
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002716 all_structmembers structmembers;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002717} test_structmembers;
2718
2719static struct PyMemberDef test_members[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002720 {"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
2721 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
2722 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
2723 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
2724 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
2725 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
2726 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
2727 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
2728 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
2729 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
2730 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
2731 {"T_STRING_INPLACE", T_STRING_INPLACE, offsetof(test_structmembers, structmembers.inplace_member), 0, NULL},
Martin v. Löwis6371cd82007-06-09 07:42:52 +00002732#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002733 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
2734 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
Martin v. Löwis6371cd82007-06-09 07:42:52 +00002735#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002736 {NULL}
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002737};
2738
2739
Georg Brandlc09b94e2008-01-21 21:28:32 +00002740static PyObject *
2741test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
2742{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002743 static char *keywords[] = {
2744 "T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
2745 "T_INT", "T_UINT", "T_LONG", "T_ULONG",
2746 "T_FLOAT", "T_DOUBLE", "T_STRING_INPLACE",
Georg Brandlc09b94e2008-01-21 21:28:32 +00002747#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002748 "T_LONGLONG", "T_ULONGLONG",
Georg Brandlc09b94e2008-01-21 21:28:32 +00002749#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002750 NULL};
2751 static char *fmt = "|bbBhHiIlkfds#"
Georg Brandlc09b94e2008-01-21 21:28:32 +00002752#ifdef HAVE_LONG_LONG
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002753 "LK"
Georg Brandlc09b94e2008-01-21 21:28:32 +00002754#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002755 ;
2756 test_structmembers *ob;
2757 const char *s = NULL;
Antoine Pitrou1e181022013-05-08 02:07:13 +02002758 int string_len = 0;
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002759 ob = PyObject_New(test_structmembers, type);
2760 if (ob == NULL)
2761 return NULL;
2762 memset(&ob->structmembers, 0, sizeof(all_structmembers));
2763 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
2764 &ob->structmembers.bool_member,
2765 &ob->structmembers.byte_member,
2766 &ob->structmembers.ubyte_member,
2767 &ob->structmembers.short_member,
2768 &ob->structmembers.ushort_member,
2769 &ob->structmembers.int_member,
2770 &ob->structmembers.uint_member,
2771 &ob->structmembers.long_member,
2772 &ob->structmembers.ulong_member,
2773 &ob->structmembers.float_member,
2774 &ob->structmembers.double_member,
2775 &s, &string_len
2776#ifdef HAVE_LONG_LONG
2777 , &ob->structmembers.longlong_member,
2778 &ob->structmembers.ulonglong_member
2779#endif
2780 )) {
2781 Py_DECREF(ob);
2782 return NULL;
2783 }
2784 if (s != NULL) {
2785 if (string_len > 5) {
2786 Py_DECREF(ob);
2787 PyErr_SetString(PyExc_ValueError, "string too long");
2788 return NULL;
2789 }
2790 strcpy(ob->structmembers.inplace_member, s);
2791 }
2792 else {
2793 strcpy(ob->structmembers.inplace_member, "");
2794 }
2795 return (PyObject *)ob;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002796}
2797
Georg Brandlc09b94e2008-01-21 21:28:32 +00002798static void
2799test_structmembers_free(PyObject *ob)
2800{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002801 PyObject_FREE(ob);
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002802}
2803
2804static PyTypeObject test_structmembersType = {
Martin v. Löwis68192102007-07-21 06:55:02 +00002805 PyVarObject_HEAD_INIT(NULL, 0)
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002806 "test_structmembersType",
2807 sizeof(test_structmembers), /* tp_basicsize */
2808 0, /* tp_itemsize */
2809 test_structmembers_free, /* destructor tp_dealloc */
2810 0, /* tp_print */
2811 0, /* tp_getattr */
2812 0, /* tp_setattr */
2813 0, /* tp_compare */
2814 0, /* tp_repr */
2815 0, /* tp_as_number */
2816 0, /* tp_as_sequence */
2817 0, /* tp_as_mapping */
2818 0, /* tp_hash */
2819 0, /* tp_call */
2820 0, /* tp_str */
2821 PyObject_GenericGetAttr, /* tp_getattro */
2822 PyObject_GenericSetAttr, /* tp_setattro */
2823 0, /* tp_as_buffer */
2824 0, /* tp_flags */
2825 "Type containing all structmember types",
2826 0, /* traverseproc tp_traverse */
2827 0, /* tp_clear */
2828 0, /* tp_richcompare */
2829 0, /* tp_weaklistoffset */
2830 0, /* tp_iter */
2831 0, /* tp_iternext */
2832 0, /* tp_methods */
2833 test_members, /* tp_members */
2834 0,
2835 0,
2836 0,
2837 0,
2838 0,
2839 0,
2840 0,
2841 0,
2842 test_structmembers_new, /* tp_new */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002843};
2844
2845
Mark Hammond62b1ab12002-07-23 06:31:15 +00002846PyMODINIT_FUNC
Tim Petersd66595f2001-02-04 03:09:53 +00002847init_testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +00002848{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002849 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00002850
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002851 m = Py_InitModule("_testcapi", TestMethods);
2852 if (m == NULL)
2853 return;
Tim Peters9ea17ac2001-02-02 05:57:15 +00002854
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002855 Py_TYPE(&_HashInheritanceTester_Type)=&PyType_Type;
Hirokazu Yamamoto3cda1db2008-12-31 05:47:19 +00002856
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002857 Py_TYPE(&test_structmembersType)=&PyType_Type;
2858 Py_INCREF(&test_structmembersType);
2859 /* don't use a name starting with "test", since we don't want
2860 test_capi to automatically call this */
2861 PyModule_AddObject(m, "_test_structmembersType", (PyObject *)&test_structmembersType);
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00002862
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002863 PyModule_AddObject(m, "CHAR_MAX", PyInt_FromLong(CHAR_MAX));
2864 PyModule_AddObject(m, "CHAR_MIN", PyInt_FromLong(CHAR_MIN));
2865 PyModule_AddObject(m, "UCHAR_MAX", PyInt_FromLong(UCHAR_MAX));
2866 PyModule_AddObject(m, "SHRT_MAX", PyInt_FromLong(SHRT_MAX));
2867 PyModule_AddObject(m, "SHRT_MIN", PyInt_FromLong(SHRT_MIN));
2868 PyModule_AddObject(m, "USHRT_MAX", PyInt_FromLong(USHRT_MAX));
2869 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
2870 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
2871 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
2872 PyModule_AddObject(m, "LONG_MAX", PyInt_FromLong(LONG_MAX));
2873 PyModule_AddObject(m, "LONG_MIN", PyInt_FromLong(LONG_MIN));
2874 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
2875 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
2876 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
2877 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
2878 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
2879 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
2880 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
2881 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
2882 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyInt_FromSsize_t(PY_SSIZE_T_MAX));
2883 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyInt_FromSsize_t(PY_SSIZE_T_MIN));
2884 PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyInt_FromSsize_t(sizeof(PyGC_Head)));
Thomas Hellera4ea6032003-04-17 18:55:45 +00002885
Victor Stinner95a91df2017-06-27 14:31:40 +02002886#ifdef MS_WINDOWS
2887 PyModule_AddIntConstant(m, "CRT_WARN", _CRT_WARN);
2888 PyModule_AddIntConstant(m, "CRT_ERROR", _CRT_ERROR);
2889 PyModule_AddIntConstant(m, "CRT_ASSERT", _CRT_ASSERT);
2890 PyModule_AddIntConstant(m, "CRTDBG_MODE_FILE", _CRTDBG_MODE_FILE);
2891 PyModule_AddIntConstant(m, "CRTDBG_FILE_STDERR", (int)_CRTDBG_FILE_STDERR);
2892#endif
2893
Antoine Pitrouc83ea132010-05-09 14:46:46 +00002894 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
2895 Py_INCREF(TestError);
2896 PyModule_AddObject(m, "error", TestError);
Tim Peters9ea17ac2001-02-02 05:57:15 +00002897}