#include "_testcapi/parts.h"

#include "frameobject.h" // PyFrame_New()

#include "marshal.h" // PyMarshal_WriteLongToFile()

#ifdef bool

# error "The public headers should not include <stdbool.h>, see gh-90904"

#include <float.h> // FLT_MAX

#include <signal.h>

#include <stddef.h> // offsetof()

#ifdef HAVE_SYS_WAIT_H

# include <sys/wait.h> // W_STOPCODE

#ifdef HAVE_SYS_SYSCTL_H

# include <sys/sysctl.h> // sysctlbyname()

#include "_testcapi/util.h"

static struct PyModuleDef _testcapimodule;

typedef struct {

PyObject *error; // _testcapi.error object

} testcapistate_t;

static testcapistate_t*

get_testcapi_state(PyObject *module)

{

void *state = PyModule_GetState(module);

assert(state != NULL);

return (testcapistate_t *)state;

}

static PyObject *

get_testerror(PyObject *self) {

testcapistate_t *state = get_testcapi_state(self);

return state->error;

}

static void

simple_object_dealloc(PyObject *self)

{

PyObject_Free(self);

}

static PyObject *

raiseTestError(PyObject *self, const char* test_name, const char* msg)

{

PyErr_Format(get_testerror(self), "%s: %s", test_name, msg);

return NULL;

}

static PyObject*

sizeof_error(PyObject *self, const char* fatname, const char* typname,

int expected, int got)

{

PyErr_Format(get_testerror(self),

"%s #define == %d but sizeof(%s) == %d",

fatname, expected, typname, got);

return (PyObject*)NULL;

}

static PyObject*

test_config(PyObject *self, PyObject *Py_UNUSED(ignored))

{

#define CHECK_SIZEOF(FATNAME, TYPE) \

do { \

if (FATNAME != sizeof(TYPE)) { \

return sizeof_error(self, #FATNAME, #TYPE, FATNAME, sizeof(TYPE)); \

} \

} while (0)

CHECK_SIZEOF(SIZEOF_SHORT, short);

CHECK_SIZEOF(SIZEOF_INT, int);

CHECK_SIZEOF(SIZEOF_LONG, long);

CHECK_SIZEOF(SIZEOF_VOID_P, void*);

CHECK_SIZEOF(SIZEOF_TIME_T, time_t);

CHECK_SIZEOF(SIZEOF_LONG_LONG, long long);

#undef CHECK_SIZEOF

Py_RETURN_NONE;

}

static PyObject*

test_sizeof_c_types(PyObject *self, PyObject *Py_UNUSED(ignored))

{

#if defined(__GNUC__) && ((__GNUC__ > 4) || ((__GNUC__ == 4) && (__GNUC_MINOR__ > 5)))

#pragma GCC diagnostic push

#pragma GCC diagnostic ignored "-Wtype-limits"

#define CHECK_SIZEOF(TYPE, EXPECTED) \

do { \

if (EXPECTED != sizeof(TYPE)) { \

PyErr_Format(get_testerror(self), \

"sizeof(%s) = %zu instead of %u", \

#TYPE, sizeof(TYPE), (unsigned)(EXPECTED)); \

return (PyObject*)NULL; \

} \

} while (0)

#define IS_SIGNED(TYPE) (((TYPE)-1) < (TYPE)0)

#define CHECK_SIGNNESS(TYPE, SIGNED) \

do { \

if (IS_SIGNED(TYPE) != SIGNED) { \

PyErr_Format(get_testerror(self), \

"%s signness is %i, instead of %i", \

#TYPE, IS_SIGNED(TYPE), SIGNED); \

return (PyObject*)NULL; \

} \

} while (0)

/* integer types */

CHECK_SIZEOF(Py_UCS1, 1);

CHECK_SIZEOF(Py_UCS2, 2);

CHECK_SIZEOF(Py_UCS4, 4);

CHECK_SIGNNESS(Py_UCS1, 0);

CHECK_SIGNNESS(Py_UCS2, 0);

CHECK_SIGNNESS(Py_UCS4, 0);

CHECK_SIZEOF(int32_t, 4);

CHECK_SIGNNESS(int32_t, 1);

CHECK_SIZEOF(uint32_t, 4);

CHECK_SIGNNESS(uint32_t, 0);

CHECK_SIZEOF(int64_t, 8);

CHECK_SIGNNESS(int64_t, 1);

CHECK_SIZEOF(uint64_t, 8);

CHECK_SIGNNESS(uint64_t, 0);

/* pointer/size types */

CHECK_SIZEOF(size_t, sizeof(void *));

CHECK_SIGNNESS(size_t, 0);

CHECK_SIZEOF(Py_ssize_t, sizeof(void *));

CHECK_SIGNNESS(Py_ssize_t, 1);

CHECK_SIZEOF(uintptr_t, sizeof(void *));

CHECK_SIGNNESS(uintptr_t, 0);

CHECK_SIZEOF(intptr_t, sizeof(void *));

CHECK_SIGNNESS(intptr_t, 1);

Py_RETURN_NONE;

#undef IS_SIGNED

#undef CHECK_SIGNESS

#undef CHECK_SIZEOF

#if defined(__GNUC__) && ((__GNUC__ > 4) || ((__GNUC__ == 4) && (__GNUC_MINOR__ > 5)))

#pragma GCC diagnostic pop

}

static PyTypeObject _HashInheritanceTester_Type = {

PyVarObject_HEAD_INIT(NULL, 0)

"hashinheritancetester", /* Name of this type */

sizeof(PyObject), /* Basic object size */

0, /* Item size for varobject */

simple_object_dealloc, /* tp_dealloc */

0, /* tp_vectorcall_offset */

0, /* tp_getattr */

0, /* tp_setattr */

0, /* tp_as_async */

0, /* tp_repr */

0, /* tp_as_number */

0, /* tp_as_sequence */

0, /* tp_as_mapping */

0, /* tp_hash */

0, /* tp_call */

0, /* tp_str */

PyObject_GenericGetAttr, /* tp_getattro */

0, /* tp_setattro */

0, /* tp_as_buffer */

Py_TPFLAGS_DEFAULT, /* tp_flags */

0, /* tp_doc */

0, /* tp_traverse */

0, /* tp_clear */

0, /* tp_richcompare */

0, /* tp_weaklistoffset */

0, /* tp_iter */

0, /* tp_iternext */

0, /* tp_methods */

0, /* tp_members */

0, /* tp_getset */

0, /* tp_base */

0, /* tp_dict */

0, /* tp_descr_get */

0, /* tp_descr_set */

0, /* tp_dictoffset */

0, /* tp_init */

0, /* tp_alloc */

PyType_GenericNew, /* tp_new */

};

static PyObject*

pycompilestring(PyObject* self, PyObject *obj) {

if (PyBytes_CheckExact(obj) == 0) {

PyErr_SetString(PyExc_ValueError, "Argument must be a bytes object");

return NULL;

}

const char *the_string = PyBytes_AsString(obj);

if (the_string == NULL) {

return NULL;

}

return Py_CompileString(the_string, "<string>", Py_file_input);

}

static PyObject*

pycompilestringexflags(PyObject *self, PyObject *args) {

const char *the_string, *filename;

int start, flags;

if (!PyArg_ParseTuple(args, "ysii", &the_string, &filename, &start, &flags)) {

return NULL;

}

PyCompilerFlags cf = _PyCompilerFlags_INIT;

cf.cf_flags = flags;

return Py_CompileStringExFlags(the_string, filename, start, &cf, -1);

}

static PyObject*

test_lazy_hash_inheritance(PyObject* self, PyObject *Py_UNUSED(ignored))

{

PyTypeObject *type;

PyObject *obj;

Py_hash_t hash;

type = &_HashInheritanceTester_Type;

if (type->tp_dict != NULL)

/* The type has already been initialized. This probably means

Py_RETURN_NONE;

obj = PyObject_New(PyObject, type);

if (obj == NULL) {

PyErr_Clear();

PyErr_SetString(

get_testerror(self),

"test_lazy_hash_inheritance: failed to create object");

return NULL;

}

if (type->tp_dict != NULL) {

PyErr_SetString(

get_testerror(self),

"test_lazy_hash_inheritance: type initialised too soon");

Py_DECREF(obj);

return NULL;

}

hash = PyObject_Hash(obj);

if ((hash == -1) && PyErr_Occurred()) {

PyErr_Clear();

PyErr_SetString(

get_testerror(self),

"test_lazy_hash_inheritance: could not hash object");

Py_DECREF(obj);

return NULL;

}

if (type->tp_dict == NULL) {

PyErr_SetString(

get_testerror(self),

"test_lazy_hash_inheritance: type not initialised by hash()");

Py_DECREF(obj);

return NULL;

}

if (type->tp_hash != PyType_Type.tp_hash) {

PyErr_SetString(

get_testerror(self),

"test_lazy_hash_inheritance: unexpected hash function");

Py_DECREF(obj);

return NULL;

}

Py_DECREF(obj);

Py_RETURN_NONE;

}

static PyObject *

return_none(void *unused)

{

Py_RETURN_NONE;

}

static PyObject *

raise_error(void *unused)

{

PyErr_SetNone(PyExc_ValueError);

return NULL;

}

static PyObject *

py_buildvalue(PyObject *self, PyObject *args)

{

const char *fmt;

PyObject *objs[10] = {NULL};

if (!PyArg_ParseTuple(args, "s|OOOOOOOOOO", &fmt,

&objs[0], &objs[1], &objs[2], &objs[3], &objs[4],

&objs[5], &objs[6], &objs[7], &objs[8], &objs[9]))

{

return NULL;

}

for(int i = 0; i < 10; i++) {

NULLABLE(objs[i]);

}

return Py_BuildValue(fmt,

objs[0], objs[1], objs[2], objs[3], objs[4],

objs[5], objs[6], objs[7], objs[8], objs[9]);

}

static PyObject *

py_buildvalue_ints(PyObject *self, PyObject *args)

{

const char *fmt;

unsigned int values[10] = {0};

if (!PyArg_ParseTuple(args, "s|IIIIIIIIII", &fmt,

&values[0], &values[1], &values[2], &values[3], &values[4],

&values[5], &values[6], &values[7], &values[8], &values[9]))

{

return NULL;

}

return Py_BuildValue(fmt,

values[0], values[1], values[2], values[3], values[4],

values[5], values[6], values[7], values[8], values[9]);

}

static int

test_buildvalue_N_error(PyObject *self, const char *fmt)

{

PyObject *arg, *res;

arg = PyList_New(0);

if (arg == NULL) {

return -1;

}

Py_INCREF(arg);

res = Py_BuildValue(fmt, return_none, NULL, arg);

if (res == NULL) {

return -1;

}

Py_DECREF(res);

if (Py_REFCNT(arg) != 1) {

PyErr_Format(get_testerror(self), "test_buildvalue_N: "

"arg was not decrefed in successful "

"Py_BuildValue(\"%s\")", fmt);

return -1;

}

Py_INCREF(arg);

res = Py_BuildValue(fmt, raise_error, NULL, arg);

if (res != NULL || !PyErr_Occurred()) {

PyErr_Format(get_testerror(self), "test_buildvalue_N: "

"Py_BuildValue(\"%s\") didn't complain", fmt);

return -1;

}

PyErr_Clear();

if (Py_REFCNT(arg) != 1) {

PyErr_Format(get_testerror(self), "test_buildvalue_N: "

"arg was not decrefed in failed "

"Py_BuildValue(\"%s\")", fmt);

return -1;

}

Py_DECREF(arg);

return 0;

}

static PyObject *

test_buildvalue_N(PyObject *self, PyObject *Py_UNUSED(ignored))

{

PyObject *arg, *res;

arg = PyList_New(0);

if (arg == NULL) {

return NULL;

}

Py_INCREF(arg);

res = Py_BuildValue("N", arg);

if (res == NULL) {

return NULL;

}

if (res != arg) {

return raiseTestError(self, "test_buildvalue_N",

"Py_BuildValue(\"N\") returned wrong result");

}

if (Py_REFCNT(arg) != 2) {

return raiseTestError(self, "test_buildvalue_N",

"arg was not decrefed in Py_BuildValue(\"N\")");

}

Py_DECREF(res);

Py_DECREF(arg);

if (test_buildvalue_N_error(self, "O&N") < 0)

return NULL;

if (test_buildvalue_N_error(self, "(O&N)") < 0)

return NULL;

if (test_buildvalue_N_error(self, "[O&N]") < 0)

return NULL;

if (test_buildvalue_N_error(self, "{O&N}") < 0)

return NULL;

if (test_buildvalue_N_error(self, "{()O&(())N}") < 0)

return NULL;

Py_RETURN_NONE;

}

static PyObject *

test_buildvalue_p(PyObject *self, PyObject *Py_UNUSED(ignored))

{

PyObject *res = Py_BuildValue("p", 3);

if (res == NULL) {

return NULL;

}

if (!Py_IsTrue(res)) {

Py_DECREF(res);

return raiseTestError(self, "test_buildvalue_p", "Py_BuildValue(\"p\", 3) returned wrong result");

}

Py_DECREF(res);

res = Py_BuildValue("p", 0);

if (res == NULL) {

return NULL;

}

if (!Py_IsFalse(res)) {

Py_DECREF(res);

return raiseTestError(self, "test_buildvalue_p", "Py_BuildValue(\"p\", 0) returned wrong result");

}

Py_DECREF(res);

Py_RETURN_NONE;

}

static PyObject *

pyobject_repr_from_null(PyObject *self, PyObject *Py_UNUSED(ignored))

{

return PyObject_Repr(NULL);

}

static PyObject *

pyobject_str_from_null(PyObject *self, PyObject *Py_UNUSED(ignored))

{

return PyObject_Str(NULL);

}

static PyObject *

pyobject_bytes_from_null(PyObject *self, PyObject *Py_UNUSED(ignored))

{

return PyObject_Bytes(NULL);

}

static PyObject *

set_errno(PyObject *self, PyObject *args)

{

int new_errno;

if (!PyArg_ParseTuple(args, "i:set_errno", &new_errno))

return NULL;

errno = new_errno;

Py_RETURN_NONE;

}

static PyThread_type_lock thread_done = NULL;

static int

_make_call(void *callable)

{

PyObject *rc;

int success;

PyGILState_STATE s = PyGILState_Ensure();

rc = PyObject_CallNoArgs((PyObject *)callable);

success = (rc != NULL);

Py_XDECREF(rc);

PyGILState_Release(s);

return success;

}

static void

_make_call_from_thread(void *callable)

{

_make_call(callable);

PyThread_release_lock(thread_done);

}

static PyObject *

test_thread_state(PyObject *self, PyObject *args)

{

PyObject *fn;

int success = 1;

if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))

return NULL;

if (!PyCallable_Check(fn)) {

PyErr_Format(PyExc_TypeError, "'%T' object is not callable", fn);

return NULL;

}

thread_done = PyThread_allocate_lock();

if (thread_done == NULL)

return PyErr_NoMemory();

PyThread_acquire_lock(thread_done, 1);

/* Start a new thread with our callback. */

PyThread_start_new_thread(_make_call_from_thread, fn);

/* Make the callback with the thread lock held by this thread */

success &= _make_call(fn);

/* Do it all again, but this time with the thread-lock released */

Py_BEGIN_ALLOW_THREADS

success &= _make_call(fn);

PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */

Py_END_ALLOW_THREADS

/* And once more with and without a thread

Py_BEGIN_ALLOW_THREADS

PyThread_start_new_thread(_make_call_from_thread, fn);

success &= _make_call(fn);

PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */

Py_END_ALLOW_THREADS

/* Release lock we acquired above. This is required on HP-UX. */

PyThread_release_lock(thread_done);

PyThread_free_lock(thread_done);

if (!success)

return NULL;

Py_RETURN_NONE;

}

static PyObject *

gilstate_ensure_release(PyObject *module, PyObject *Py_UNUSED(ignored))

{

PyGILState_STATE state = PyGILState_Ensure();

PyGILState_Release(state);

Py_RETURN_NONE;

}

#ifndef MS_WINDOWS

static PyThread_type_lock wait_done = NULL;

static void wait_for_lock(void *unused) {

PyThread_acquire_lock(wait_done, 1);

PyThread_release_lock(wait_done);

PyThread_free_lock(wait_done);

wait_done = NULL;

}

static PyObject *

spawn_pthread_waiter(PyObject *self, PyObject *Py_UNUSED(ignored))

{

if (wait_done) {

PyErr_SetString(PyExc_RuntimeError, "thread already running");

return NULL;

}

wait_done = PyThread_allocate_lock();

if (wait_done == NULL)

return PyErr_NoMemory();

PyThread_acquire_lock(wait_done, 1);

PyThread_start_new_thread(wait_for_lock, NULL);

Py_RETURN_NONE;

}

static PyObject *

end_spawned_pthread(PyObject *self, PyObject *Py_UNUSED(ignored))

{

if (!wait_done) {

PyErr_SetString(PyExc_RuntimeError, "call _spawn_pthread_waiter 1st");

return NULL;

}

PyThread_release_lock(wait_done);

Py_RETURN_NONE;

}

#endif // not MS_WINDOWS

static int _pending_callback(void *arg)

{

/* we assume the argument is callable object to which we own a reference */

PyObject *callable = (PyObject *)arg;

PyObject *r = PyObject_CallNoArgs(callable);

Py_DECREF(callable);

Py_XDECREF(r);

return r != NULL ? 0 : -1;

}

static PyObject *

pending_threadfunc(PyObject *self, PyObject *arg, PyObject *kwargs)

{

static char *kwlist[] = {"callback", "num",

"blocking", "ensure_added", NULL};

PyObject *callable;

unsigned int num = 1;

int blocking = 0;

int ensure_added = 0;

if (!PyArg_ParseTupleAndKeywords(arg, kwargs,

"O|I$pp:_pending_threadfunc", kwlist,

&callable, &num, &blocking, &ensure_added))

{

return NULL;

}

/* create the reference for the callbackwhile we hold the lock */

for (unsigned int i = 0; i < num; i++) {

Py_INCREF(callable);

}

PyThreadState *save_tstate = NULL;

if (!blocking) {

save_tstate = PyEval_SaveThread();

}

unsigned int num_added = 0;

for (; num_added < num; num_added++) {

if (ensure_added) {

int r;

do {

r = Py_AddPendingCall(&_pending_callback, callable);

} while (r < 0);

}

else {

if (Py_AddPendingCall(&_pending_callback, callable) < 0) {

break;

}

}

}

if (!blocking) {

PyEval_RestoreThread(save_tstate);

}

for (unsigned int i = num_added; i < num; i++) {

Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */

}

/* The callable is decref'ed above in each added _pending_callback(). */

return PyLong_FromUnsignedLong((unsigned long)num_added);

}

static const char *capsule_name = "capsule name";

static char *capsule_pointer = "capsule pointer";

static char *capsule_context = "capsule context";

static const char *capsule_error = NULL;

static int

capsule_destructor_call_count = 0;

static void

capsule_destructor(PyObject *o) {

capsule_destructor_call_count++;

if (PyCapsule_GetContext(o) != capsule_context) {

capsule_error = "context did not match in destructor!";

} else if (PyCapsule_GetDestructor(o) != capsule_destructor) {

capsule_error = "destructor did not match in destructor! (woah!)";

} else if (PyCapsule_GetName(o) != capsule_name) {

capsule_error = "name did not match in destructor!";

} else if (PyCapsule_GetPointer(o, capsule_name) != capsule_pointer) {

capsule_error = "pointer did not match in destructor!";

}

}

typedef struct {

char *name;

char *module;

char *attribute;

} known_capsule;

static PyObject *

test_capsule(PyObject *self, PyObject *Py_UNUSED(ignored))

{

PyObject *object;

const char *error = NULL;

void *pointer;

void *pointer2;

known_capsule known_capsules[] = {

#define KNOWN_CAPSULE(module, name) { module "." name, module, name }

KNOWN_CAPSULE("_socket", "CAPI"),

KNOWN_CAPSULE("_curses", "_C_API"),

KNOWN_CAPSULE("datetime", "datetime_CAPI"),

{ NULL, NULL },

};

known_capsule *known = &known_capsules[0];

#define FAIL(x) \

do { \

error = (x); \

goto exit; \

} while (0)

#define CHECK_DESTRUCTOR \

object = PyCapsule_New(capsule_pointer, capsule_name, capsule_destructor);

PyCapsule_SetContext(object, capsule_context);

capsule_destructor(object);

CHECK_DESTRUCTOR;

Py_DECREF(object);

CHECK_DESTRUCTOR;

object = PyCapsule_New(known, "ignored", NULL);

PyCapsule_SetPointer(object, capsule_pointer);

PyCapsule_SetName(object, capsule_name);

PyCapsule_SetDestructor(object, capsule_destructor);

PyCapsule_SetContext(object, capsule_context);

capsule_destructor(object);

CHECK_DESTRUCTOR;

/* intentionally access using the wrong name */

pointer2 = PyCapsule_GetPointer(object, "the wrong name");

if (!PyErr_Occurred()) {

FAIL("PyCapsule_GetPointer should have failed but did not!");

}

PyErr_Clear();

if (pointer2) {

if (pointer2 == capsule_pointer) {

FAIL("PyCapsule_GetPointer should not have"

" returned the internal pointer!");

} else {

FAIL("PyCapsule_GetPointer should have "

"returned NULL pointer but did not!");

}

}

PyCapsule_SetDestructor(object, NULL);

Py_DECREF(object);

if (capsule_destructor_call_count) {

FAIL("destructor called when it should not have been!");

}

for (known = &known_capsules[0]; known->module != NULL; known++) {

/* yeah, ordinarily I wouldn't do this either,

static char buffer[256];

#undef FAIL

#define FAIL(x) \

do { \

sprintf(buffer, "%s module: \"%s\" attribute: \"%s\"", \

x, known->module, known->attribute); \

error = buffer; \

goto exit; \

} while (0)

PyObject *module = PyImport_ImportModule(known->module);

if (module) {

pointer = PyCapsule_Import(known->name, 0);

if (!pointer) {

Py_DECREF(module);

FAIL("PyCapsule_GetPointer returned NULL unexpectedly!");

}

object = PyObject_GetAttrString(module, known->attribute);

if (!object) {

Py_DECREF(module);

return NULL;

}

pointer2 = PyCapsule_GetPointer(object,

"weebles wobble but they don't fall down");

if (!PyErr_Occurred()) {

Py_DECREF(object);

Py_DECREF(module);

FAIL("PyCapsule_GetPointer should have failed but did not!");

}

PyErr_Clear();

if (pointer2) {

Py_DECREF(module);

Py_DECREF(object);

if (pointer2 == pointer) {

FAIL("PyCapsule_GetPointer should not have"

" returned its internal pointer!");

} else {

FAIL("PyCapsule_GetPointer should have"

" returned NULL pointer but did not!");

}

}

Py_DECREF(object);

Py_DECREF(module);

}

else

PyErr_Clear();

}

exit:

if (error) {

return raiseTestError(self, "test_capsule", error);

}

Py_RETURN_NONE;

#undef FAIL

}

#ifdef HAVE_GETTIMEOFDAY

static void print_delta(int test, struct timeval *s, struct timeval *e)

{

e->tv_sec -= s->tv_sec;

e->tv_usec -= s->tv_usec;

if (e->tv_usec < 0) {

e->tv_sec -=1;

e->tv_usec += 1000000;

}

printf("Test %d: %d.%06ds\n", test, (int)e->tv_sec, (int)e->tv_usec);

}

static PyObject *

profile_int(PyObject *self, PyObject* args)

{

int i, k;

struct timeval start, stop;

PyObject *single, **multiple, *op1, *result;

/* Test 1: Allocate and immediately deallocate

gettimeofday(&start, NULL);

for(k=0; k < 20000; k++)

for(i=0; i < 1000; i++) {

single = PyLong_FromLong(i);

Py_DECREF(single);

}

gettimeofday(&stop, NULL);

print_delta(1, &start, &stop);

/* Test 2: Allocate and immediately deallocate

gettimeofday(&start, NULL);

for(k=0; k < 20000; k++)

for(i=0; i < 1000; i++) {

single = PyLong_FromLong(i+1000000);

Py_DECREF(single);

}

gettimeofday(&stop, NULL);

print_delta(2, &start, &stop);

/* Test 3: Allocate a few integers, then release

multiple = malloc(sizeof(PyObject*) * 1000);

if (multiple == NULL)

return PyErr_NoMemory();

gettimeofday(&start, NULL);

for(k=0; k < 20000; k++) {

for(i=0; i < 1000; i++) {

multiple[i] = PyLong_FromLong(i+1000000);

}

for(i=0; i < 1000; i++) {

Py_DECREF(multiple[i]);

}

}

gettimeofday(&stop, NULL);

print_delta(3, &start, &stop);

free(multiple);

/* Test 4: Allocate many integers, then release

multiple = malloc(sizeof(PyObject*) * 1000000);

if (multiple == NULL)

return PyErr_NoMemory();

gettimeofday(&start, NULL);

for(k=0; k < 20; k++) {

for(i=0; i < 1000000; i++) {

multiple[i] = PyLong_FromLong(i+1000000);

}

for(i=0; i < 1000000; i++) {

Py_DECREF(multiple[i]);

}

}

gettimeofday(&stop, NULL);

print_delta(4, &start, &stop);

free(multiple);

/* Test 5: Allocate many integers < 32000 */

multiple = malloc(sizeof(PyObject*) * 1000000);

if (multiple == NULL)

return PyErr_NoMemory();

gettimeofday(&start, NULL);

for(k=0; k < 10; k++) {

for(i=0; i < 1000000; i++) {

multiple[i] = PyLong_FromLong(i+1000);

}

for(i=0; i < 1000000; i++) {

Py_DECREF(multiple[i]);

}

}

gettimeofday(&stop, NULL);

print_delta(5, &start, &stop);

free(multiple);

/* Test 6: Perform small int addition */

op1 = PyLong_FromLong(1);

gettimeofday(&start, NULL);

for(i=0; i < 10000000; i++) {

result = PyNumber_Add(op1, op1);

Py_DECREF(result);

}

gettimeofday(&stop, NULL);

Py_DECREF(op1);

print_delta(6, &start, &stop);

/* Test 7: Perform medium int addition */

op1 = PyLong_FromLong(1000);

if (op1 == NULL)

return NULL;

gettimeofday(&start, NULL);

for(i=0; i < 10000000; i++) {

result = PyNumber_Add(op1, op1);

Py_XDECREF(result);

}

gettimeofday(&stop, NULL);

Py_DECREF(op1);

print_delta(7, &start, &stop);

Py_RETURN_NONE;

}

static PyObject *str1, *str2;

static int

failing_converter(PyObject *obj, void *arg)

{

/* Clone str1, then let the conversion fail. */

assert(str1);

str2 = Py_NewRef(str1);

return 0;

}

static PyObject*

argparsing(PyObject *o, PyObject *args)

{

PyObject *res;

str1 = str2 = NULL;

if (!PyArg_ParseTuple(args, "O&O&",

PyUnicode_FSConverter, &str1,

failing_converter, &str2)) {

if (!str2)

/* argument converter not called? */

return NULL;

/* Should be 1 */

res = PyLong_FromSsize_t(Py_REFCNT(str2));

Py_DECREF(str2);

PyErr_Clear();

return res;

}

Py_RETURN_NONE;

}

static PyObject *

code_newempty(PyObject *self, PyObject *args)

{

const char *filename;

const char *funcname;