mirror of https://github.com/python/cpython
849 lines
25 KiB
C
849 lines
25 KiB
C
#include "pyconfig.h" // Py_GIL_DISABLED
|
|
#ifndef Py_GIL_DISABLED
|
|
// Need limited C API 3.14 to test PyLong_AsInt64()
|
|
# define Py_LIMITED_API 0x030e0000
|
|
#endif
|
|
|
|
#include "parts.h"
|
|
#include "util.h"
|
|
#include "clinic/long.c.h"
|
|
|
|
/*[clinic input]
|
|
module _testlimitedcapi
|
|
[clinic start generated code]*/
|
|
/*[clinic end generated code: output=da39a3ee5e6b4b0d input=2700057f9c1135ba]*/
|
|
|
|
|
|
static PyObject *
|
|
raiseTestError(const char* test_name, const char* msg)
|
|
{
|
|
PyErr_Format(PyExc_AssertionError, "%s: %s", test_name, msg);
|
|
return NULL;
|
|
}
|
|
|
|
/* Tests of PyLong_{As, From}{Unsigned,}Long(), and
|
|
PyLong_{As, From}{Unsigned,}LongLong().
|
|
|
|
Note that the meat of the test is contained in testcapi_long.h.
|
|
This is revolting, but delicate code duplication is worse: "almost
|
|
exactly the same" code is needed to test long long, but the ubiquitous
|
|
dependence on type names makes it impossible to use a parameterized
|
|
function. A giant macro would be even worse than this. A C++ template
|
|
would be perfect.
|
|
|
|
The "report an error" functions are deliberately not part of the #include
|
|
file: if the test fails, you can set a breakpoint in the appropriate
|
|
error function directly, and crawl back from there in the debugger.
|
|
*/
|
|
|
|
#define UNBIND(X) Py_DECREF(X); (X) = NULL
|
|
|
|
static PyObject *
|
|
raise_test_long_error(const char* msg)
|
|
{
|
|
return raiseTestError("test_long_api", msg);
|
|
}
|
|
|
|
// Test PyLong_FromLong()/PyLong_AsLong()
|
|
// and PyLong_FromUnsignedLong()/PyLong_AsUnsignedLong().
|
|
|
|
#define TESTNAME test_long_api_inner
|
|
#define TYPENAME long
|
|
#define F_S_TO_PY PyLong_FromLong
|
|
#define F_PY_TO_S PyLong_AsLong
|
|
#define F_U_TO_PY PyLong_FromUnsignedLong
|
|
#define F_PY_TO_U PyLong_AsUnsignedLong
|
|
|
|
#include "testcapi_long.h"
|
|
|
|
/*[clinic input]
|
|
_testlimitedcapi.test_long_api
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
_testlimitedcapi_test_long_api_impl(PyObject *module)
|
|
/*[clinic end generated code: output=06a2c02366d1853a input=9012b3d6a483df63]*/
|
|
{
|
|
return TESTNAME(raise_test_long_error);
|
|
}
|
|
|
|
#undef TESTNAME
|
|
#undef TYPENAME
|
|
#undef F_S_TO_PY
|
|
#undef F_PY_TO_S
|
|
#undef F_U_TO_PY
|
|
#undef F_PY_TO_U
|
|
|
|
// Test PyLong_FromLongLong()/PyLong_AsLongLong()
|
|
// and PyLong_FromUnsignedLongLong()/PyLong_AsUnsignedLongLong().
|
|
|
|
static PyObject *
|
|
raise_test_longlong_error(const char* msg)
|
|
{
|
|
return raiseTestError("test_longlong_api", msg);
|
|
}
|
|
|
|
#define TESTNAME test_longlong_api_inner
|
|
#define TYPENAME long long
|
|
#define F_S_TO_PY PyLong_FromLongLong
|
|
#define F_PY_TO_S PyLong_AsLongLong
|
|
#define F_U_TO_PY PyLong_FromUnsignedLongLong
|
|
#define F_PY_TO_U PyLong_AsUnsignedLongLong
|
|
|
|
#include "testcapi_long.h"
|
|
|
|
/*[clinic input]
|
|
_testlimitedcapi.test_longlong_api
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
_testlimitedcapi_test_longlong_api_impl(PyObject *module)
|
|
/*[clinic end generated code: output=8faa10e1c35214bf input=2b582a9d25bd68e7]*/
|
|
{
|
|
return TESTNAME(raise_test_longlong_error);
|
|
}
|
|
|
|
#undef TESTNAME
|
|
#undef TYPENAME
|
|
#undef F_S_TO_PY
|
|
#undef F_PY_TO_S
|
|
#undef F_U_TO_PY
|
|
#undef F_PY_TO_U
|
|
|
|
|
|
/*[clinic input]
|
|
_testlimitedcapi.test_long_and_overflow
|
|
|
|
Test the PyLong_AsLongAndOverflow API.
|
|
|
|
General conversion to PY_LONG is tested by test_long_api_inner.
|
|
This test will concentrate on proper handling of overflow.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
_testlimitedcapi_test_long_and_overflow_impl(PyObject *module)
|
|
/*[clinic end generated code: output=fdfd3c1eeabb6d14 input=e3a18791de6519fe]*/
|
|
{
|
|
PyObject *num, *one, *temp;
|
|
long value;
|
|
int overflow;
|
|
|
|
/* Test that overflow is set properly for a large value. */
|
|
/* num is a number larger than LONG_MAX even on 64-bit platforms */
|
|
num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 1234;
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -1)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"return value was not set to -1");
|
|
if (overflow != 1)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"overflow was not set to 1");
|
|
|
|
/* Same again, with num = LONG_MAX + 1 */
|
|
num = PyLong_FromLong(LONG_MAX);
|
|
if (num == NULL)
|
|
return NULL;
|
|
one = PyLong_FromLong(1L);
|
|
if (one == NULL) {
|
|
Py_DECREF(num);
|
|
return NULL;
|
|
}
|
|
temp = PyNumber_Add(num, one);
|
|
Py_DECREF(one);
|
|
Py_DECREF(num);
|
|
num = temp;
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 0;
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -1)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"return value was not set to -1");
|
|
if (overflow != 1)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"overflow was not set to 1");
|
|
|
|
/* Test that overflow is set properly for a large negative value. */
|
|
/* num is a number smaller than LONG_MIN even on 64-bit platforms */
|
|
num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 1234;
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -1)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"return value was not set to -1");
|
|
if (overflow != -1)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"overflow was not set to -1");
|
|
|
|
/* Same again, with num = LONG_MIN - 1 */
|
|
num = PyLong_FromLong(LONG_MIN);
|
|
if (num == NULL)
|
|
return NULL;
|
|
one = PyLong_FromLong(1L);
|
|
if (one == NULL) {
|
|
Py_DECREF(num);
|
|
return NULL;
|
|
}
|
|
temp = PyNumber_Subtract(num, one);
|
|
Py_DECREF(one);
|
|
Py_DECREF(num); num = temp;
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 0;
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -1)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"return value was not set to -1");
|
|
if (overflow != -1)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"overflow was not set to -1");
|
|
|
|
/* Test that overflow is cleared properly for small values. */
|
|
num = PyLong_FromString("FF", NULL, 16);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 1234;
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != 0xFF)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"expected return value 0xFF");
|
|
if (overflow != 0)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"overflow was not cleared");
|
|
|
|
num = PyLong_FromString("-FF", NULL, 16);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 0;
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -0xFF)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"expected return value 0xFF");
|
|
if (overflow != 0)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"overflow was set incorrectly");
|
|
|
|
num = PyLong_FromLong(LONG_MAX);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 1234;
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != LONG_MAX)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"expected return value LONG_MAX");
|
|
if (overflow != 0)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"overflow was not cleared");
|
|
|
|
num = PyLong_FromLong(LONG_MIN);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 0;
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != LONG_MIN)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"expected return value LONG_MIN");
|
|
if (overflow != 0)
|
|
return raiseTestError("test_long_and_overflow",
|
|
"overflow was not cleared");
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/*[clinic input]
|
|
_testlimitedcapi.test_long_long_and_overflow
|
|
|
|
Test the PyLong_AsLongLongAndOverflow API.
|
|
|
|
General conversion to long long is tested by test_long_api_inner.
|
|
This test will concentrate on proper handling of overflow.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
_testlimitedcapi_test_long_long_and_overflow_impl(PyObject *module)
|
|
/*[clinic end generated code: output=3d2721a49c09a307 input=741c593b606cc6b3]*/
|
|
{
|
|
PyObject *num, *one, *temp;
|
|
long long value;
|
|
int overflow;
|
|
|
|
/* Test that overflow is set properly for a large value. */
|
|
/* num is a number larger than LLONG_MAX on a typical machine. */
|
|
num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 1234;
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -1)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"return value was not set to -1");
|
|
if (overflow != 1)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"overflow was not set to 1");
|
|
|
|
/* Same again, with num = LLONG_MAX + 1 */
|
|
num = PyLong_FromLongLong(LLONG_MAX);
|
|
if (num == NULL)
|
|
return NULL;
|
|
one = PyLong_FromLong(1L);
|
|
if (one == NULL) {
|
|
Py_DECREF(num);
|
|
return NULL;
|
|
}
|
|
temp = PyNumber_Add(num, one);
|
|
Py_DECREF(one);
|
|
Py_DECREF(num); num = temp;
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 0;
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -1)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"return value was not set to -1");
|
|
if (overflow != 1)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"overflow was not set to 1");
|
|
|
|
/* Test that overflow is set properly for a large negative value. */
|
|
/* num is a number smaller than LLONG_MIN on a typical platform */
|
|
num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 1234;
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -1)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"return value was not set to -1");
|
|
if (overflow != -1)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"overflow was not set to -1");
|
|
|
|
/* Same again, with num = LLONG_MIN - 1 */
|
|
num = PyLong_FromLongLong(LLONG_MIN);
|
|
if (num == NULL)
|
|
return NULL;
|
|
one = PyLong_FromLong(1L);
|
|
if (one == NULL) {
|
|
Py_DECREF(num);
|
|
return NULL;
|
|
}
|
|
temp = PyNumber_Subtract(num, one);
|
|
Py_DECREF(one);
|
|
Py_DECREF(num); num = temp;
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 0;
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -1)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"return value was not set to -1");
|
|
if (overflow != -1)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"overflow was not set to -1");
|
|
|
|
/* Test that overflow is cleared properly for small values. */
|
|
num = PyLong_FromString("FF", NULL, 16);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 1234;
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != 0xFF)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"expected return value 0xFF");
|
|
if (overflow != 0)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"overflow was not cleared");
|
|
|
|
num = PyLong_FromString("-FF", NULL, 16);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 0;
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != -0xFF)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"expected return value 0xFF");
|
|
if (overflow != 0)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"overflow was set incorrectly");
|
|
|
|
num = PyLong_FromLongLong(LLONG_MAX);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 1234;
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != LLONG_MAX)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"expected return value LLONG_MAX");
|
|
if (overflow != 0)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"overflow was not cleared");
|
|
|
|
num = PyLong_FromLongLong(LLONG_MIN);
|
|
if (num == NULL)
|
|
return NULL;
|
|
overflow = 0;
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
Py_DECREF(num);
|
|
if (value == -1 && PyErr_Occurred())
|
|
return NULL;
|
|
if (value != LLONG_MIN)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"expected return value LLONG_MIN");
|
|
if (overflow != 0)
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
"overflow was not cleared");
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/*[clinic input]
|
|
_testlimitedcapi.test_long_as_size_t
|
|
|
|
Test the PyLong_As{Size,Ssize}_t API.
|
|
|
|
At present this just tests that non-integer arguments are handled correctly.
|
|
It should be extended to test overflow handling.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
_testlimitedcapi_test_long_as_size_t_impl(PyObject *module)
|
|
/*[clinic end generated code: output=297a9f14a42f55af input=8923d8f2038c46f4]*/
|
|
{
|
|
size_t out_u;
|
|
Py_ssize_t out_s;
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
out_u = PyLong_AsSize_t(Py_None);
|
|
if (out_u != (size_t)-1 || !PyErr_Occurred())
|
|
return raiseTestError("test_long_as_size_t",
|
|
"PyLong_AsSize_t(None) didn't complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_TypeError))
|
|
return raiseTestError("test_long_as_size_t",
|
|
"PyLong_AsSize_t(None) raised "
|
|
"something other than TypeError");
|
|
PyErr_Clear();
|
|
|
|
out_s = PyLong_AsSsize_t(Py_None);
|
|
if (out_s != (Py_ssize_t)-1 || !PyErr_Occurred())
|
|
return raiseTestError("test_long_as_size_t",
|
|
"PyLong_AsSsize_t(None) didn't complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_TypeError))
|
|
return raiseTestError("test_long_as_size_t",
|
|
"PyLong_AsSsize_t(None) raised "
|
|
"something other than TypeError");
|
|
PyErr_Clear();
|
|
|
|
/* Py_INCREF(Py_None) omitted - we already have a reference to it. */
|
|
return Py_None;
|
|
}
|
|
|
|
/*[clinic input]
|
|
_testlimitedcapi.test_long_as_unsigned_long_long_mask
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
_testlimitedcapi_test_long_as_unsigned_long_long_mask_impl(PyObject *module)
|
|
/*[clinic end generated code: output=90be09ffeec8ecab input=17c660bd58becad5]*/
|
|
{
|
|
unsigned long long res = PyLong_AsUnsignedLongLongMask(NULL);
|
|
|
|
if (res != (unsigned long long)-1 || !PyErr_Occurred()) {
|
|
return raiseTestError("test_long_as_unsigned_long_long_mask",
|
|
"PyLong_AsUnsignedLongLongMask(NULL) didn't "
|
|
"complain");
|
|
}
|
|
if (!PyErr_ExceptionMatches(PyExc_SystemError)) {
|
|
return raiseTestError("test_long_as_unsigned_long_long_mask",
|
|
"PyLong_AsUnsignedLongLongMask(NULL) raised "
|
|
"something other than SystemError");
|
|
}
|
|
PyErr_Clear();
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/*[clinic input]
|
|
_testlimitedcapi.test_long_as_double
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
_testlimitedcapi_test_long_as_double_impl(PyObject *module)
|
|
/*[clinic end generated code: output=0e688c2acf224f88 input=e7b5712385064a48]*/
|
|
{
|
|
double out;
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
out = PyLong_AsDouble(Py_None);
|
|
if (out != -1.0 || !PyErr_Occurred())
|
|
return raiseTestError("test_long_as_double",
|
|
"PyLong_AsDouble(None) didn't complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_TypeError))
|
|
return raiseTestError("test_long_as_double",
|
|
"PyLong_AsDouble(None) raised "
|
|
"something other than TypeError");
|
|
PyErr_Clear();
|
|
|
|
/* Py_INCREF(Py_None) omitted - we already have a reference to it. */
|
|
return Py_None;
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_check(PyObject *module, PyObject *obj)
|
|
{
|
|
NULLABLE(obj);
|
|
return PyLong_FromLong(PyLong_Check(obj));
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_checkexact(PyObject *module, PyObject *obj)
|
|
{
|
|
NULLABLE(obj);
|
|
return PyLong_FromLong(PyLong_CheckExact(obj));
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_fromdouble(PyObject *module, PyObject *arg)
|
|
{
|
|
double value;
|
|
if (!PyArg_Parse(arg, "d", &value)) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromDouble(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_fromstring(PyObject *module, PyObject *args)
|
|
{
|
|
const char *str;
|
|
Py_ssize_t len;
|
|
int base;
|
|
char *end = UNINITIALIZED_PTR;
|
|
if (!PyArg_ParseTuple(args, "z#i", &str, &len, &base)) {
|
|
return NULL;
|
|
}
|
|
|
|
PyObject *result = PyLong_FromString(str, &end, base);
|
|
if (result == NULL) {
|
|
// XXX 'end' is not always set.
|
|
return NULL;
|
|
}
|
|
return Py_BuildValue("Nn", result, (Py_ssize_t)(end - str));
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_fromvoidptr(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
return PyLong_FromVoidPtr((void *)arg);
|
|
}
|
|
|
|
/*[clinic input]
|
|
_testlimitedcapi.PyLong_AsInt
|
|
arg: object
|
|
/
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
_testlimitedcapi_PyLong_AsInt(PyObject *module, PyObject *arg)
|
|
/*[clinic end generated code: output=d91db4c1287f85fa input=32c66be86f3265a1]*/
|
|
{
|
|
NULLABLE(arg);
|
|
assert(!PyErr_Occurred());
|
|
int value = PyLong_AsInt(arg);
|
|
if (value == -1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromLong(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_aslong(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
long value = PyLong_AsLong(arg);
|
|
if (value == -1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromLong(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_aslongandoverflow(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
int overflow = UNINITIALIZED_INT;
|
|
long value = PyLong_AsLongAndOverflow(arg, &overflow);
|
|
if (value == -1 && PyErr_Occurred()) {
|
|
assert(overflow == -1);
|
|
return NULL;
|
|
}
|
|
return Py_BuildValue("li", value, overflow);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_asunsignedlong(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
unsigned long value = PyLong_AsUnsignedLong(arg);
|
|
if (value == (unsigned long)-1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromUnsignedLong(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_asunsignedlongmask(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
unsigned long value = PyLong_AsUnsignedLongMask(arg);
|
|
if (value == (unsigned long)-1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromUnsignedLong(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_aslonglong(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
long long value = PyLong_AsLongLong(arg);
|
|
if (value == -1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromLongLong(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_aslonglongandoverflow(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
int overflow = UNINITIALIZED_INT;
|
|
long long value = PyLong_AsLongLongAndOverflow(arg, &overflow);
|
|
if (value == -1 && PyErr_Occurred()) {
|
|
assert(overflow == -1);
|
|
return NULL;
|
|
}
|
|
return Py_BuildValue("Li", value, overflow);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_asunsignedlonglong(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
unsigned long long value = PyLong_AsUnsignedLongLong(arg);
|
|
if (value == (unsigned long long)-1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromUnsignedLongLong(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_asunsignedlonglongmask(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
unsigned long long value = PyLong_AsUnsignedLongLongMask(arg);
|
|
if (value == (unsigned long long)-1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromUnsignedLongLong(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_as_ssize_t(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
Py_ssize_t value = PyLong_AsSsize_t(arg);
|
|
if (value == -1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromSsize_t(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_as_size_t(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
size_t value = PyLong_AsSize_t(arg);
|
|
if (value == (size_t)-1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromSize_t(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_asdouble(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
double value = PyLong_AsDouble(arg);
|
|
if (value == -1.0 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyFloat_FromDouble(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_asvoidptr(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
void *value = PyLong_AsVoidPtr(arg);
|
|
if (value == NULL) {
|
|
if (PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
return Py_NewRef((PyObject *)value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_aspid(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
pid_t value = PyLong_AsPid(arg);
|
|
if (value == -1 && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromPid(value);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
pylong_asint32(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
int32_t value;
|
|
if (PyLong_AsInt32(arg, &value) < 0) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromInt32(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_asuint32(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
uint32_t value;
|
|
if (PyLong_AsUInt32(arg, &value) < 0) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromUInt32(value);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
pylong_asint64(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
int64_t value;
|
|
if (PyLong_AsInt64(arg, &value) < 0) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromInt64(value);
|
|
}
|
|
|
|
static PyObject *
|
|
pylong_asuint64(PyObject *module, PyObject *arg)
|
|
{
|
|
NULLABLE(arg);
|
|
uint64_t value;
|
|
if (PyLong_AsUInt64(arg, &value) < 0) {
|
|
return NULL;
|
|
}
|
|
return PyLong_FromUInt64(value);
|
|
}
|
|
|
|
|
|
static PyMethodDef test_methods[] = {
|
|
_TESTLIMITEDCAPI_TEST_LONG_AND_OVERFLOW_METHODDEF
|
|
_TESTLIMITEDCAPI_TEST_LONG_API_METHODDEF
|
|
_TESTLIMITEDCAPI_TEST_LONG_AS_DOUBLE_METHODDEF
|
|
_TESTLIMITEDCAPI_TEST_LONG_AS_SIZE_T_METHODDEF
|
|
_TESTLIMITEDCAPI_TEST_LONG_AS_UNSIGNED_LONG_LONG_MASK_METHODDEF
|
|
_TESTLIMITEDCAPI_TEST_LONG_LONG_AND_OVERFLOW_METHODDEF
|
|
_TESTLIMITEDCAPI_TEST_LONGLONG_API_METHODDEF
|
|
{"pylong_check", pylong_check, METH_O},
|
|
{"pylong_checkexact", pylong_checkexact, METH_O},
|
|
{"pylong_fromdouble", pylong_fromdouble, METH_O},
|
|
{"pylong_fromstring", pylong_fromstring, METH_VARARGS},
|
|
{"pylong_fromvoidptr", pylong_fromvoidptr, METH_O},
|
|
_TESTLIMITEDCAPI_PYLONG_ASINT_METHODDEF
|
|
{"pylong_aslong", pylong_aslong, METH_O},
|
|
{"pylong_aslongandoverflow", pylong_aslongandoverflow, METH_O},
|
|
{"pylong_asunsignedlong", pylong_asunsignedlong, METH_O},
|
|
{"pylong_asunsignedlongmask", pylong_asunsignedlongmask, METH_O},
|
|
{"pylong_aslonglong", pylong_aslonglong, METH_O},
|
|
{"pylong_aslonglongandoverflow", pylong_aslonglongandoverflow, METH_O},
|
|
{"pylong_asunsignedlonglong", pylong_asunsignedlonglong, METH_O},
|
|
{"pylong_asunsignedlonglongmask", pylong_asunsignedlonglongmask, METH_O},
|
|
{"pylong_as_ssize_t", pylong_as_ssize_t, METH_O},
|
|
{"pylong_as_size_t", pylong_as_size_t, METH_O},
|
|
{"pylong_asdouble", pylong_asdouble, METH_O},
|
|
{"pylong_asvoidptr", pylong_asvoidptr, METH_O},
|
|
{"pylong_aspid", pylong_aspid, METH_O},
|
|
{"pylong_asint32", pylong_asint32, METH_O},
|
|
{"pylong_asuint32", pylong_asuint32, METH_O},
|
|
{"pylong_asint64", pylong_asint64, METH_O},
|
|
{"pylong_asuint64", pylong_asuint64, METH_O},
|
|
{NULL},
|
|
};
|
|
|
|
int
|
|
_PyTestLimitedCAPI_Init_Long(PyObject *mod)
|
|
{
|
|
if (PyModule_AddFunctions(mod, test_methods) < 0) {
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|