Modules/_testcapimodule.c

   1 /*
   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"
   9 #include <float.h>
  10 #include "structmember.h"
  11
  12 #ifdef WITH_THREAD
  13 #include "pythread.h"
  14 #endif /* WITH_THREAD */
  15 static PyObject *TestError;     /* set to exception object in init */
  16
  17 /* Raise TestError with test_name + ": " + msg, and return NULL. */
  18
  19 static PyObject *
  20 raiseTestError(const char* test_name, const char* msg)
  21 {
  22         char buf[2048];
  23
  24         if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
  25                 PyErr_SetString(TestError, "internal error msg too large");
  26         else {
  27                 PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
  28                 PyErr_SetString(TestError, buf);
  29         }
  30         return NULL;
  31 }
  32
  33 /* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
  34
  35    The ones derived from autoconf on the UNIX-like OSes can be relied
  36    upon (in the absence of sloppy cross-compiling), but the Windows
  37    platforms have these hardcoded.  Better safe than sorry.
  38 */
  39 static PyObject*
  40 sizeof_error(const char* fatname, const char* typname,
  41         int expected, int got)
  42 {
  43         char buf[1024];
  44         PyOS_snprintf(buf, sizeof(buf),
  45                 "%.200s #define == %d but sizeof(%.200s) == %d",
  46                 fatname, expected, typname, got);
  47         PyErr_SetString(TestError, buf);
  48         return (PyObject*)NULL;
  49 }
  50
  51 static PyObject*
  52 test_config(PyObject *self)
  53 {
  54 #define CHECK_SIZEOF(FATNAME, TYPE) \
  55             if (FATNAME != sizeof(TYPE)) \
  56                 return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
  57
  58         CHECK_SIZEOF(SIZEOF_SHORT, short);
  59         CHECK_SIZEOF(SIZEOF_INT, int);
  60         CHECK_SIZEOF(SIZEOF_LONG, long);
  61         CHECK_SIZEOF(SIZEOF_VOID_P, void*);
  62         CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
  63 #ifdef HAVE_LONG_LONG
  64         CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
  65 #endif
  66
  67 #undef CHECK_SIZEOF
  68
  69         Py_INCREF(Py_None);
  70         return Py_None;
  71 }
  72
  73 static PyObject*
  74 test_list_api(PyObject *self)
  75 {
  76         PyObject* list;
  77         int i;
  78
  79         /* SF bug 132008:  PyList_Reverse segfaults */
  80 #define NLIST 30
  81         list = PyList_New(NLIST);
  82         if (list == (PyObject*)NULL)
  83                 return (PyObject*)NULL;
  84         /* list = range(NLIST) */
  85         for (i = 0; i < NLIST; ++i) {
  86                 PyObject* anint = PyInt_FromLong(i);
  87                 if (anint == (PyObject*)NULL) {
  88                         Py_DECREF(list);
  89                         return (PyObject*)NULL;
  90                 }
  91                 PyList_SET_ITEM(list, i, anint);
  92         }
  93         /* list.reverse(), via PyList_Reverse() */
  94         i = PyList_Reverse(list);   /* should not blow up! */
  95         if (i != 0) {
  96                 Py_DECREF(list);
  97                 return (PyObject*)NULL;
  98         }
  99         /* Check that list == range(29, -1, -1) now */
 100         for (i = 0; i < NLIST; ++i) {
 101                 PyObject* anint = PyList_GET_ITEM(list, i);
 102                 if (PyInt_AS_LONG(anint) != NLIST-1-i) {
 103                         PyErr_SetString(TestError,
 104                                         "test_list_api: reverse screwed up");
 105                         Py_DECREF(list);
 106                         return (PyObject*)NULL;
 107                 }
 108         }
 109         Py_DECREF(list);
 110 #undef NLIST
 111
 112         Py_INCREF(Py_None);
 113         return Py_None;
 114 }
 115
 116 static int
 117 test_dict_inner(int count)
 118 {
 119         Py_ssize_t pos = 0, iterations = 0;
 120         int i;
 121         PyObject *dict = PyDict_New();
 122         PyObject *v, *k;
 123
 124         if (dict == NULL)
 125                 return -1;
 126
 127         for (i = 0; i < count; i++) {
 128                 v = PyInt_FromLong(i);
 129                 PyDict_SetItem(dict, v, v);
 130                 Py_DECREF(v);
 131         }
 132
 133         while (PyDict_Next(dict, &pos, &k, &v)) {
 134                 PyObject *o;
 135                 iterations++;
 136
 137                 i = PyInt_AS_LONG(v) + 1;
 138                 o = PyInt_FromLong(i);
 139                 if (o == NULL)
 140                         return -1;
 141                 if (PyDict_SetItem(dict, k, o) < 0) {
 142                         Py_DECREF(o);
 143                         return -1;
 144                 }
 145                 Py_DECREF(o);
 146         }
 147
 148         Py_DECREF(dict);
 149
 150         if (iterations != count) {
 151                 PyErr_SetString(
 152                         TestError,
 153                         "test_dict_iteration: dict iteration went wrong ");
 154                 return -1;
 155         } else {
 156                 return 0;
 157         }
 158 }
 159
 160 static PyObject*
 161 test_dict_iteration(PyObject* self)
 162 {
 163         int i;
 164
 165         for (i = 0; i < 200; i++) {
 166                 if (test_dict_inner(i) < 0) {
 167                         return NULL;
 168                 }
 169         }
 170
 171         Py_INCREF(Py_None);
 172         return Py_None;
 173 }
 174
 175
 176 /* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
 177    PyLong_{As, From}{Unsigned,}LongLong().
 178
 179    Note that the meat of the test is contained in testcapi_long.h.
 180    This is revolting, but delicate code duplication is worse:  "almost
 181    exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
 182    dependence on type names makes it impossible to use a parameterized
 183    function.  A giant macro would be even worse than this.  A C++ template
 184    would be perfect.
 185
 186    The "report an error" functions are deliberately not part of the #include
 187    file:  if the test fails, you can set a breakpoint in the appropriate
 188    error function directly, and crawl back from there in the debugger.
 189 */
 190
 191 #define UNBIND(X)  Py_DECREF(X); (X) = NULL
 192
 193 static PyObject *
 194 raise_test_long_error(const char* msg)
 195 {
 196         return raiseTestError("test_long_api", msg);
 197 }
 198
 199 #define TESTNAME        test_long_api_inner
 200 #define TYPENAME        long
 201 #define F_S_TO_PY       PyLong_FromLong
 202 #define F_PY_TO_S       PyLong_AsLong
 203 #define F_U_TO_PY       PyLong_FromUnsignedLong
 204 #define F_PY_TO_U       PyLong_AsUnsignedLong
 205
 206 #include "testcapi_long.h"
 207
 208 static PyObject *
 209 test_long_api(PyObject* self)
 210 {
 211         return TESTNAME(raise_test_long_error);
 212 }
 213
 214 #undef TESTNAME
 215 #undef TYPENAME
 216 #undef F_S_TO_PY
 217 #undef F_PY_TO_S
 218 #undef F_U_TO_PY
 219 #undef F_PY_TO_U
 220
 221 #ifdef HAVE_LONG_LONG
 222
 223 static PyObject *
 224 raise_test_longlong_error(const char* msg)
 225 {
 226         return raiseTestError("test_longlong_api", msg);
 227 }
 228
 229 #define TESTNAME        test_longlong_api_inner
 230 #define TYPENAME        PY_LONG_LONG
 231 #define F_S_TO_PY       PyLong_FromLongLong
 232 #define F_PY_TO_S       PyLong_AsLongLong
 233 #define F_U_TO_PY       PyLong_FromUnsignedLongLong
 234 #define F_PY_TO_U       PyLong_AsUnsignedLongLong
 235
 236 #include "testcapi_long.h"
 237
 238 static PyObject *
 239 test_longlong_api(PyObject* self, PyObject *args)
 240 {
 241         return TESTNAME(raise_test_longlong_error);
 242 }
 243
 244 #undef TESTNAME
 245 #undef TYPENAME
 246 #undef F_S_TO_PY
 247 #undef F_PY_TO_S
 248 #undef F_U_TO_PY
 249 #undef F_PY_TO_U
 250
 251 /* Test the L code for PyArg_ParseTuple.  This should deliver a PY_LONG_LONG
 252    for both long and int arguments.  The test may leak a little memory if
 253    it fails.
 254 */
 255 static PyObject *
 256 test_L_code(PyObject *self)
 257 {
 258         PyObject *tuple, *num;
 259         PY_LONG_LONG value;
 260
 261         tuple = PyTuple_New(1);
 262         if (tuple == NULL)
 263                 return NULL;
 264
 265         num = PyLong_FromLong(42);
 266         if (num == NULL)
 267                 return NULL;
 268
 269         PyTuple_SET_ITEM(tuple, 0, num);
 270
 271         value = -1;
 272         if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
 273                 return NULL;
 274         if (value != 42)
 275                 return raiseTestError("test_L_code",
 276                         "L code returned wrong value for long 42");
 277
 278         Py_DECREF(num);
 279         num = PyInt_FromLong(42);
 280         if (num == NULL)
 281                 return NULL;
 282
 283         PyTuple_SET_ITEM(tuple, 0, num);
 284
 285         value = -1;
 286         if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
 287                 return NULL;
 288         if (value != 42)
 289                 return raiseTestError("test_L_code",
 290                         "L code returned wrong value for int 42");
 291
 292         Py_DECREF(tuple);
 293         Py_INCREF(Py_None);
 294         return Py_None;
 295 }
 296
 297 #endif  /* ifdef HAVE_LONG_LONG */
 298
 299 /* Test tuple argument processing */
 300 static PyObject *
 301 getargs_tuple(PyObject *self, PyObject *args)
 302 {
 303         int a, b, c;
 304         if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
 305                 return NULL;
 306         return Py_BuildValue("iii", a, b, c);
 307 }
 308
 309 /* Functions to call PyArg_ParseTuple with integer format codes,
 310    and return the result.
 311 */
 312 static PyObject *
 313 getargs_b(PyObject *self, PyObject *args)
 314 {
 315         unsigned char value;
 316         if (!PyArg_ParseTuple(args, "b", &value))
 317                 return NULL;
 318         return PyLong_FromUnsignedLong((unsigned long)value);
 319 }
 320
 321 static PyObject *
 322 getargs_B(PyObject *self, PyObject *args)
 323 {
 324         unsigned char value;
 325         if (!PyArg_ParseTuple(args, "B", &value))
 326                 return NULL;
 327         return PyLong_FromUnsignedLong((unsigned long)value);
 328 }
 329
 330 static PyObject *
 331 getargs_H(PyObject *self, PyObject *args)
 332 {
 333         unsigned short value;
 334         if (!PyArg_ParseTuple(args, "H", &value))
 335                 return NULL;
 336         return PyLong_FromUnsignedLong((unsigned long)value);
 337 }
 338
 339 static PyObject *
 340 getargs_I(PyObject *self, PyObject *args)
 341 {
 342         unsigned int value;
 343         if (!PyArg_ParseTuple(args, "I", &value))
 344                 return NULL;
 345         return PyLong_FromUnsignedLong((unsigned long)value);
 346 }
 347
 348 static PyObject *
 349 getargs_k(PyObject *self, PyObject *args)
 350 {
 351         unsigned long value;
 352         if (!PyArg_ParseTuple(args, "k", &value))
 353                 return NULL;
 354         return PyLong_FromUnsignedLong(value);
 355 }
 356
 357 static PyObject *
 358 getargs_i(PyObject *self, PyObject *args)
 359 {
 360         int value;
 361         if (!PyArg_ParseTuple(args, "i", &value))
 362                 return NULL;
 363         return PyLong_FromLong((long)value);
 364 }
 365
 366 static PyObject *
 367 getargs_l(PyObject *self, PyObject *args)
 368 {
 369         long value;
 370         if (!PyArg_ParseTuple(args, "l", &value))
 371                 return NULL;
 372         return PyLong_FromLong(value);
 373 }
 374
 375 static PyObject *
 376 getargs_n(PyObject *self, PyObject *args)
 377 {
 378         Py_ssize_t value;
 379         if (!PyArg_ParseTuple(args, "n", &value))
 380         return NULL;
 381         return PyInt_FromSsize_t(value);
 382 }
 383
 384 #ifdef HAVE_LONG_LONG
 385 static PyObject *
 386 getargs_L(PyObject *self, PyObject *args)
 387 {
 388         PY_LONG_LONG value;
 389         if (!PyArg_ParseTuple(args, "L", &value))
 390                 return NULL;
 391         return PyLong_FromLongLong(value);
 392 }
 393
 394 static PyObject *
 395 getargs_K(PyObject *self, PyObject *args)
 396 {
 397         unsigned PY_LONG_LONG value;
 398         if (!PyArg_ParseTuple(args, "K", &value))
 399                 return NULL;
 400         return PyLong_FromUnsignedLongLong(value);
 401 }
 402 #endif
 403
 404 /* This function not only tests the 'k' getargs code, but also the
 405    PyInt_AsUnsignedLongMask() and PyInt_AsUnsignedLongMask() functions. */
 406 static PyObject *
 407 test_k_code(PyObject *self)
 408 {
 409         PyObject *tuple, *num;
 410         unsigned long value;
 411
 412         tuple = PyTuple_New(1);
 413         if (tuple == NULL)
 414                 return NULL;
 415
 416         /* a number larger than ULONG_MAX even on 64-bit platforms */
 417         num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
 418         if (num == NULL)
 419                 return NULL;
 420
 421         value = PyInt_AsUnsignedLongMask(num);
 422         if (value != ULONG_MAX)
 423                 return raiseTestError("test_k_code",
 424             "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
 425
 426         PyTuple_SET_ITEM(tuple, 0, num);
 427
 428         value = 0;
 429         if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
 430                 return NULL;
 431         if (value != ULONG_MAX)
 432                 return raiseTestError("test_k_code",
 433                         "k code returned wrong value for long 0xFFF...FFF");
 434
 435         Py_DECREF(num);
 436         num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
 437         if (num == NULL)
 438                 return NULL;
 439
 440         value = PyInt_AsUnsignedLongMask(num);
 441         if (value != (unsigned long)-0x42)
 442                 return raiseTestError("test_k_code",
 443             "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
 444
 445         PyTuple_SET_ITEM(tuple, 0, num);
 446
 447         value = 0;
 448         if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
 449                 return NULL;
 450         if (value != (unsigned long)-0x42)
 451                 return raiseTestError("test_k_code",
 452                         "k code returned wrong value for long -0xFFF..000042");
 453
 454         Py_DECREF(tuple);
 455         Py_INCREF(Py_None);
 456         return Py_None;
 457 }
 458
 459 #ifdef Py_USING_UNICODE
 460
 461 /* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
 462    of an error.
 463 */
 464 static PyObject *
 465 test_u_code(PyObject *self)
 466 {
 467         PyObject *tuple, *obj;
 468         Py_UNICODE *value;
 469         int len;
 470
 471         tuple = PyTuple_New(1);
 472         if (tuple == NULL)
 473                 return NULL;
 474
 475         obj = PyUnicode_Decode("test", strlen("test"),
 476                                "ascii", NULL);
 477         if (obj == NULL)
 478                 return NULL;
 479
 480         PyTuple_SET_ITEM(tuple, 0, obj);
 481
 482         value = 0;
 483         if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
 484                 return NULL;
 485         if (value != PyUnicode_AS_UNICODE(obj))
 486                 return raiseTestError("test_u_code",
 487                         "u code returned wrong value for u'test'");
 488         value = 0;
 489         if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
 490                 return NULL;
 491         if (value != PyUnicode_AS_UNICODE(obj) ||
 492             len != PyUnicode_GET_SIZE(obj))
 493                 return raiseTestError("test_u_code",
 494                         "u# code returned wrong values for u'test'");
 495
 496         Py_DECREF(tuple);
 497         Py_INCREF(Py_None);
 498         return Py_None;
 499 }
 500
 501 static PyObject *
 502 codec_incrementalencoder(PyObject *self, PyObject *args)
 503 {
 504         const char *encoding, *errors = NULL;
 505         if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
 506                               &encoding, &errors))
 507                 return NULL;
 508         return PyCodec_IncrementalEncoder(encoding, errors);
 509 }
 510
 511 static PyObject *
 512 codec_incrementaldecoder(PyObject *self, PyObject *args)
 513 {
 514         const char *encoding, *errors = NULL;
 515         if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
 516                               &encoding, &errors))
 517                 return NULL;
 518         return PyCodec_IncrementalDecoder(encoding, errors);
 519 }
 520
 521 #endif
 522
 523 /* Simple test of _PyLong_NumBits and _PyLong_Sign. */
 524 static PyObject *
 525 test_long_numbits(PyObject *self)
 526 {
 527         struct triple {
 528                 long input;
 529                 size_t nbits;
 530                 int sign;
 531         } testcases[] = {{0, 0, 0},
 532                          {1L, 1, 1},
 533                          {-1L, 1, -1},
 534                          {2L, 2, 1},
 535                          {-2L, 2, -1},
 536                          {3L, 2, 1},
 537                          {-3L, 2, -1},
 538                          {4L, 3, 1},
 539                          {-4L, 3, -1},
 540                          {0x7fffL, 15, 1},      /* one Python long digit */
 541                          {-0x7fffL, 15, -1},
 542                          {0xffffL, 16, 1},
 543                          {-0xffffL, 16, -1},
 544                          {0xfffffffL, 28, 1},
 545                          {-0xfffffffL, 28, -1}};
 546         int i;
 547
 548         for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
 549                 PyObject *plong = PyLong_FromLong(testcases[i].input);
 550                 size_t nbits = _PyLong_NumBits(plong);
 551                 int sign = _PyLong_Sign(plong);
 552
 553                 Py_DECREF(plong);
 554                 if (nbits != testcases[i].nbits)
 555                         return raiseTestError("test_long_numbits",
 556                                         "wrong result for _PyLong_NumBits");
 557                 if (sign != testcases[i].sign)
 558                         return raiseTestError("test_long_numbits",
 559                                         "wrong result for _PyLong_Sign");
 560         }
 561         Py_INCREF(Py_None);
 562         return Py_None;
 563 }
 564
 565 /* Example passing NULLs to PyObject_Str(NULL) and PyObject_Unicode(NULL). */
 566
 567 static PyObject *
 568 test_null_strings(PyObject *self)
 569 {
 570         PyObject *o1 = PyObject_Str(NULL), *o2 = PyObject_Unicode(NULL);
 571         PyObject *tuple = PyTuple_Pack(2, o1, o2);
 572         Py_XDECREF(o1);
 573         Py_XDECREF(o2);
 574         return tuple;
 575 }
 576
 577 static PyObject *
 578 raise_exception(PyObject *self, PyObject *args)
 579 {
 580         PyObject *exc;
 581         PyObject *exc_args, *v;
 582         int num_args, i;
 583
 584         if (!PyArg_ParseTuple(args, "Oi:raise_exception",
 585                               &exc, &num_args))
 586                 return NULL;
 587
 588         exc_args = PyTuple_New(num_args);
 589         if (exc_args == NULL)
 590                 return NULL;
 591         for (i = 0; i < num_args; ++i) {
 592                 v = PyInt_FromLong(i);
 593                 if (v == NULL) {
 594                         Py_DECREF(exc_args);
 595                         return NULL;
 596                 }
 597                 PyTuple_SET_ITEM(exc_args, i, v);
 598         }
 599         PyErr_SetObject(exc, exc_args);
 600         Py_DECREF(exc_args);
 601         return NULL;
 602 }
 603
 604 #ifdef WITH_THREAD
 605
 606 /* test_thread_state spawns a thread of its own, and that thread releases
 607  * `thread_done` when it's finished.  The driver code has to know when the
 608  * thread finishes, because the thread uses a PyObject (the callable) that
 609  * may go away when the driver finishes.  The former lack of this explicit
 610  * synchronization caused rare segfaults, so rare that they were seen only
 611  * on a Mac buildbot (although they were possible on any box).
 612  */
 613 static PyThread_type_lock thread_done = NULL;
 614
 615 static void
 616 _make_call(void *callable)
 617 {
 618         PyObject *rc;
 619         PyGILState_STATE s = PyGILState_Ensure();
 620         rc = PyObject_CallFunction((PyObject *)callable, "");
 621         Py_XDECREF(rc);
 622         PyGILState_Release(s);
 623 }
 624
 625 /* Same thing, but releases `thread_done` when it returns.  This variant
 626  * should be called only from threads spawned by test_thread_state().
 627  */
 628 static void
 629 _make_call_from_thread(void *callable)
 630 {
 631         _make_call(callable);
 632         PyThread_release_lock(thread_done);
 633 }
 634
 635 static PyObject *
 636 test_thread_state(PyObject *self, PyObject *args)
 637 {
 638         PyObject *fn;
 639
 640         if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
 641                 return NULL;
 642
 643         /* Ensure Python is set up for threading */
 644         PyEval_InitThreads();
 645         thread_done = PyThread_allocate_lock();
 646         if (thread_done == NULL)
 647                 return PyErr_NoMemory();
 648         PyThread_acquire_lock(thread_done, 1);
 649
 650         /* Start a new thread with our callback. */
 651         PyThread_start_new_thread(_make_call_from_thread, fn);
 652         /* Make the callback with the thread lock held by this thread */
 653         _make_call(fn);
 654         /* Do it all again, but this time with the thread-lock released */
 655         Py_BEGIN_ALLOW_THREADS
 656         _make_call(fn);
 657         PyThread_acquire_lock(thread_done, 1);  /* wait for thread to finish */
 658         Py_END_ALLOW_THREADS
 659
 660         /* And once more with and without a thread
 661            XXX - should use a lock and work out exactly what we are trying
 662            to test <wink>
 663         */
 664         Py_BEGIN_ALLOW_THREADS
 665         PyThread_start_new_thread(_make_call_from_thread, fn);
 666         _make_call(fn);
 667         PyThread_acquire_lock(thread_done, 1);  /* wait for thread to finish */
 668         Py_END_ALLOW_THREADS
 669
 670         /* Release lock we acquired above.  This is required on HP-UX. */
 671         PyThread_release_lock(thread_done);
 672
 673         PyThread_free_lock(thread_done);
 674         Py_RETURN_NONE;
 675 }
 676 #endif
 677
 678 /* Some tests of PyString_FromFormat().  This needs more tests. */
 679 static PyObject *
 680 test_string_from_format(PyObject *self, PyObject *args)
 681 {
 682         PyObject *result;
 683         char *msg;
 684
 685 #define CHECK_1_FORMAT(FORMAT, TYPE)                    \
 686         result = PyString_FromFormat(FORMAT, (TYPE)1);  \
 687         if (result == NULL)                             \
 688                 return NULL;                            \
 689         if (strcmp(PyString_AsString(result), "1")) {   \
 690                 msg = FORMAT " failed at 1";            \
 691                 goto Fail;                              \
 692         }                                               \
 693         Py_DECREF(result)
 694
 695         CHECK_1_FORMAT("%d", int);
 696         CHECK_1_FORMAT("%ld", long);
 697         /* The z width modifier was added in Python 2.5. */
 698         CHECK_1_FORMAT("%zd", Py_ssize_t);
 699
 700         /* The u type code was added in Python 2.5. */
 701         CHECK_1_FORMAT("%u", unsigned int);
 702         CHECK_1_FORMAT("%lu", unsigned long);
 703         CHECK_1_FORMAT("%zu", size_t);
 704
 705         Py_RETURN_NONE;
 706
 707  Fail:
 708         Py_XDECREF(result);
 709         return raiseTestError("test_string_from_format", msg);
 710
 711 #undef CHECK_1_FORMAT
 712 }
 713
 714 /* This is here to provide a docstring for test_descr. */
 715 static PyObject *
 716 test_with_docstring(PyObject *self)
 717 {
 718         Py_RETURN_NONE;
 719 }
 720
 721 static PyMethodDef TestMethods[] = {
 722         {"raise_exception",     raise_exception,                 METH_VARARGS},
 723         {"test_config",         (PyCFunction)test_config,        METH_NOARGS},
 724         {"test_list_api",       (PyCFunction)test_list_api,      METH_NOARGS},
 725         {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
 726         {"test_long_api",       (PyCFunction)test_long_api,      METH_NOARGS},
 727         {"test_long_numbits",   (PyCFunction)test_long_numbits,  METH_NOARGS},
 728         {"test_k_code",         (PyCFunction)test_k_code,        METH_NOARGS},
 729         {"test_null_strings",   (PyCFunction)test_null_strings,  METH_NOARGS},
 730         {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
 731         {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
 732          PyDoc_STR("This is a pretty normal docstring.")},
 733
 734         {"getargs_tuple",       getargs_tuple,                   METH_VARARGS},
 735         {"getargs_b",           getargs_b,                       METH_VARARGS},
 736         {"getargs_B",           getargs_B,                       METH_VARARGS},
 737         {"getargs_H",           getargs_H,                       METH_VARARGS},
 738         {"getargs_I",           getargs_I,                       METH_VARARGS},
 739         {"getargs_k",           getargs_k,                       METH_VARARGS},
 740         {"getargs_i",           getargs_i,                       METH_VARARGS},
 741         {"getargs_l",           getargs_l,                       METH_VARARGS},
 742         {"getargs_n",           getargs_n,                       METH_VARARGS},
 743 #ifdef HAVE_LONG_LONG
 744         {"getargs_L",           getargs_L,                       METH_VARARGS},
 745         {"getargs_K",           getargs_K,                       METH_VARARGS},
 746         {"test_longlong_api",   test_longlong_api,               METH_NOARGS},
 747         {"test_L_code",         (PyCFunction)test_L_code,        METH_NOARGS},
 748         {"codec_incrementalencoder",
 749          (PyCFunction)codec_incrementalencoder,  METH_VARARGS},
 750         {"codec_incrementaldecoder",
 751          (PyCFunction)codec_incrementaldecoder,  METH_VARARGS},
 752 #endif
 753 #ifdef Py_USING_UNICODE
 754         {"test_u_code",         (PyCFunction)test_u_code,        METH_NOARGS},
 755 #endif
 756 #ifdef WITH_THREAD
 757         {"_test_thread_state",  test_thread_state,               METH_VARARGS},
 758 #endif
 759         {NULL, NULL} /* sentinel */
 760 };
 761
 762 #define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
 763
 764 typedef struct {
 765         char byte_member;
 766         unsigned char ubyte_member;
 767         short short_member;
 768         unsigned short ushort_member;
 769         int int_member;
 770         unsigned int uint_member;
 771         long long_member;
 772         unsigned long ulong_member;
 773         float float_member;
 774         double double_member;
 775 #ifdef HAVE_LONG_LONG
 776         PY_LONG_LONG longlong_member;
 777         unsigned PY_LONG_LONG ulonglong_member;
 778 #endif
 779 } all_structmembers;
 780
 781 typedef struct {
 782     PyObject_HEAD
 783         all_structmembers structmembers;
 784 } test_structmembers;
 785
 786 static struct PyMemberDef test_members[] = {
 787         {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
 788         {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
 789         {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
 790         {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
 791         {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
 792         {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
 793         {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
 794         {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
 795         {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
 796         {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
 797 #ifdef HAVE_LONG_LONG
 798         {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
 799         {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
 800 #endif
 801         {NULL}
 802 };
 803
 804
 805 static PyObject *test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs){
 806         static char *keywords[]={"T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT", "T_INT", "T_UINT",
 807                 "T_LONG", "T_ULONG", "T_FLOAT", "T_DOUBLE",
 808                 #ifdef HAVE_LONG_LONG
 809                 "T_LONGLONG", "T_ULONGLONG",
 810                 #endif
 811                 NULL};
 812         static char *fmt="|bBhHiIlkfd"
 813                 #ifdef HAVE_LONG_LONG
 814                 "LK"
 815                 #endif
 816                 ;
 817         test_structmembers *ob=PyObject_New(test_structmembers, type);
 818         if (ob==NULL)
 819                 return NULL;
 820         memset(&ob->structmembers, 0, sizeof(all_structmembers));
 821         if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
 822                 &ob->structmembers.byte_member, &ob->structmembers.ubyte_member,
 823                 &ob->structmembers.short_member, &ob->structmembers.ushort_member,
 824                 &ob->structmembers.int_member, &ob->structmembers.uint_member,
 825                 &ob->structmembers.long_member, &ob->structmembers.ulong_member,
 826                 &ob->structmembers.float_member, &ob->structmembers.double_member
 827                 #ifdef HAVE_LONG_LONG
 828                 ,&ob->structmembers.longlong_member, &ob->structmembers.ulonglong_member
 829                 #endif
 830                 )){
 831                 Py_DECREF(ob);
 832                 return NULL;
 833                 }
 834         return (PyObject *)ob;
 835 }
 836
 837 static void test_structmembers_free(PyObject *ob){
 838         PyObject_FREE(ob);
 839 }
 840
 841 static PyTypeObject test_structmembersType = {
 842     PyObject_HEAD_INIT(NULL)
 843     0,
 844         "test_structmembersType",
 845         sizeof(test_structmembers),     /* tp_basicsize */
 846         0,                              /* tp_itemsize */
 847         test_structmembers_free,        /* destructor tp_dealloc */
 848         0,                              /* tp_print */
 849         0,                              /* tp_getattr */
 850         0,                              /* tp_setattr */
 851         0,                              /* tp_compare */
 852         0,                              /* tp_repr */
 853         0,                              /* tp_as_number */
 854         0,                              /* tp_as_sequence */
 855         0,                              /* tp_as_mapping */
 856         0,                              /* tp_hash */
 857         0,                              /* tp_call */
 858         0,                              /* tp_str */
 859         PyObject_GenericGetAttr,
 860         PyObject_GenericSetAttr,
 861         0,                              /* tp_as_buffer */
 862         0,                              /* tp_flags */
 863         "Type containing all structmember types",
 864         0,                              /* traverseproc tp_traverse */
 865         0,                              /* tp_clear */
 866         0,                              /* tp_richcompare */
 867         0,                              /* tp_weaklistoffset */
 868         0,                              /* tp_iter */
 869         0,                              /* tp_iternext */
 870         0,                              /* tp_methods */
 871         test_members,   /* tp_members */
 872         0,
 873         0,
 874         0,
 875         0,
 876         0,
 877         0,
 878         0,
 879         0,
 880         test_structmembers_new,                 /* tp_new */
 881 };
 882
 883
 884 PyMODINIT_FUNC
 885 init_testcapi(void)
 886 {
 887         PyObject *m;
 888
 889         m = Py_InitModule("_testcapi", TestMethods);
 890         if (m == NULL)
 891                 return;
 892
 893         test_structmembersType.ob_type=&PyType_Type;
 894         Py_INCREF(&test_structmembersType);
 895         PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
 896
 897         PyModule_AddObject(m, "CHAR_MAX", PyInt_FromLong(CHAR_MAX));
 898         PyModule_AddObject(m, "CHAR_MIN", PyInt_FromLong(CHAR_MIN));
 899         PyModule_AddObject(m, "UCHAR_MAX", PyInt_FromLong(UCHAR_MAX));
 900         PyModule_AddObject(m, "SHRT_MAX", PyInt_FromLong(SHRT_MAX));
 901         PyModule_AddObject(m, "SHRT_MIN", PyInt_FromLong(SHRT_MIN));
 902         PyModule_AddObject(m, "USHRT_MAX", PyInt_FromLong(USHRT_MAX));
 903         PyModule_AddObject(m, "INT_MAX",  PyLong_FromLong(INT_MAX));
 904         PyModule_AddObject(m, "INT_MIN",  PyLong_FromLong(INT_MIN));
 905         PyModule_AddObject(m, "UINT_MAX",  PyLong_FromUnsignedLong(UINT_MAX));
 906         PyModule_AddObject(m, "LONG_MAX", PyInt_FromLong(LONG_MAX));
 907         PyModule_AddObject(m, "LONG_MIN", PyInt_FromLong(LONG_MIN));
 908         PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
 909         PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
 910         PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
 911         PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
 912         PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
 913         PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
 914         PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
 915         PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
 916         PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyInt_FromSsize_t(PY_SSIZE_T_MAX));
 917         PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyInt_FromSsize_t(PY_SSIZE_T_MIN));
 918
 919         TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
 920         Py_INCREF(TestError);
 921         PyModule_AddObject(m, "error", TestError);
 922 }