cpython/Modules/_testcapi/hash.c

73 lines
1.6 KiB
C

#include "parts.h"
#include "util.h"
static PyObject *
hash_getfuncdef(PyObject *Py_UNUSED(module), PyObject *Py_UNUSED(args))
{
// bind PyHash_GetFuncDef()
PyHash_FuncDef *def = PyHash_GetFuncDef();
PyObject *types = PyImport_ImportModule("types");
if (types == NULL) {
return NULL;
}
PyObject *result = PyObject_CallMethod(types, "SimpleNamespace", NULL);
Py_DECREF(types);
if (result == NULL) {
return NULL;
}
// ignore PyHash_FuncDef.hash
PyObject *value = PyUnicode_FromString(def->name);
int res = PyObject_SetAttrString(result, "name", value);
Py_DECREF(value);
if (res < 0) {
return NULL;
}
value = PyLong_FromLong(def->hash_bits);
res = PyObject_SetAttrString(result, "hash_bits", value);
Py_DECREF(value);
if (res < 0) {
return NULL;
}
value = PyLong_FromLong(def->seed_bits);
res = PyObject_SetAttrString(result, "seed_bits", value);
Py_DECREF(value);
if (res < 0) {
return NULL;
}
return result;
}
static PyObject *
hash_pointer(PyObject *Py_UNUSED(module), PyObject *arg)
{
void *ptr = PyLong_AsVoidPtr(arg);
if (ptr == NULL && PyErr_Occurred()) {
return NULL;
}
Py_hash_t hash = Py_HashPointer(ptr);
Py_BUILD_ASSERT(sizeof(long long) >= sizeof(hash));
return PyLong_FromLongLong(hash);
}
static PyMethodDef test_methods[] = {
{"hash_getfuncdef", hash_getfuncdef, METH_NOARGS},
{"hash_pointer", hash_pointer, METH_O},
{NULL},
};
int
_PyTestCapi_Init_Hash(PyObject *m)
{
return PyModule_AddFunctions(m, test_methods);
}