Skip to content

Commit

Permalink
pythongh-111545: Add PyHash_Pointer() function
Browse files Browse the repository at this point in the history
* Keep _Py_HashPointer() function as an alias to PyHash_Pointer().
* Add PyHash_Pointer() tests to test_capi.test_hash.
* Remove _Py_HashPointerRaw() function: inline code in
  _Py_hashtable_hash_ptr().
  • Loading branch information
vstinner committed Nov 15, 2023
1 parent d4f83e1 commit ee4eed7
Show file tree
Hide file tree
Showing 10 changed files with 110 additions and 23 deletions.
9 changes: 9 additions & 0 deletions Doc/c-api/hash.rst
Original file line number Diff line number Diff line change
Expand Up @@ -46,3 +46,12 @@ See also the :c:member:`PyTypeObject.tp_hash` member.
Get the hash function definition.
.. versionadded:: 3.4
.. c:function:: Py_hash_t PyHash_Pointer(const void *ptr)
Hash a pointer.
The function cannot fail (cannot return ``-1``).
.. versionadded:: 3.13
3 changes: 3 additions & 0 deletions Doc/whatsnew/3.13.rst
Original file line number Diff line number Diff line change
Expand Up @@ -1181,6 +1181,9 @@ New Features
:exc:`KeyError` if the key missing.
(Contributed by Stefan Behnel and Victor Stinner in :gh:`111262`.)

* Add :c:func:`PyHash_Pointer` function to hash a pointer.
(Contributed by Victor Stinner in :gh:`111545`.)


Porting to Python 3.13
----------------------
Expand Down
2 changes: 2 additions & 0 deletions Include/cpython/pyhash.h
Original file line number Diff line number Diff line change
Expand Up @@ -11,3 +11,5 @@ typedef struct {
} PyHash_FuncDef;

PyAPI_FUNC(PyHash_FuncDef*) PyHash_GetFuncDef(void);

PyAPI_FUNC(Py_hash_t) PyHash_Pointer(const void *ptr);
7 changes: 2 additions & 5 deletions Include/internal/pycore_pyhash.h
Original file line number Diff line number Diff line change
Expand Up @@ -8,11 +8,8 @@
/* Helpers for hash functions */
extern Py_hash_t _Py_HashDouble(PyObject *, double);

// Export for '_decimal' shared extension
PyAPI_FUNC(Py_hash_t) _Py_HashPointer(const void*);

// Similar to _Py_HashPointer(), but don't replace -1 with -2
extern Py_hash_t _Py_HashPointerRaw(const void*);
// Kept for backward compatibility
#define _Py_HashPointer PyHash_Pointer

// Export for '_datetime' shared extension
PyAPI_FUNC(Py_hash_t) _Py_HashBytes(const void*, Py_ssize_t);
Expand Down
49 changes: 48 additions & 1 deletion Lib/test/test_capi/test_hash.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,8 @@
_testcapi = import_helper.import_module('_testcapi')


SIZEOF_PY_HASH_T = _testcapi.SIZEOF_VOID_P
SIZEOF_VOID_P = _testcapi.SIZEOF_VOID_P
SIZEOF_PY_HASH_T = SIZEOF_VOID_P


class CAPITest(unittest.TestCase):
Expand All @@ -31,3 +32,49 @@ def test_hash_getfuncdef(self):
self.assertEqual(func_def.name, hash_info.algorithm)
self.assertEqual(func_def.hash_bits, hash_info.hash_bits)
self.assertEqual(func_def.seed_bits, hash_info.seed_bits)

def test_hash_pointer(self):
# Test PyHash_Pointer()
hash_pointer = _testcapi.hash_pointer

UHASH_T_MASK = ((2 ** (8 * SIZEOF_PY_HASH_T)) - 1)
HASH_T_MAX = (2 ** (8 * SIZEOF_PY_HASH_T - 1) - 1)

# PyHash_Pointer() rotates the pointer bits by 4 bits to the right
if SIZEOF_VOID_P == 8:
values = [
0xABCDEF1234567890,
0x1234567890ABCDEF,
0xFEE4ABEDD1CECA5E,
]
else:
values = [
0x12345678,
0x1234ABCD,
0xDEADCAFE,
]

def python_hash_pointer(x):
# rotate x by 4 bits to the right and cast to signed Py_hash_t
x = (x >> 4) | (x << (8 * SIZEOF_VOID_P - 4))
x &= UHASH_T_MASK
# Convert unsigned Py_uhash_t to signed Py_hash_t
if HASH_T_MAX < x:
x = (~x) + 1
x &= UHASH_T_MASK
x = (~x) + 1
return x

for value in values:
expected = python_hash_pointer(value)
with self.subTest(value=value):
self.assertEqual(hash_pointer(value), expected,
f"hash_pointer({value:x}) = "
f"{hash_pointer(value):x} != {expected:x}")

# PyHash_Pointer(NULL) returns 0
self.assertEqual(hash_pointer(0), 0)

# PyHash_Pointer((void*)(uintptr_t)-1) doesn't return -1 but -2
VOID_P_MAX = -1 & (2 ** (8 * SIZEOF_VOID_P) - 1)
self.assertEqual(hash_pointer(VOID_P_MAX), -2)
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
Add :c:func:`PyHash_Pointer` function to hash a pointer. Patch by Victor
Stinner.
16 changes: 16 additions & 0 deletions Modules/_testcapi/hash.c
Original file line number Diff line number Diff line change
Expand Up @@ -44,8 +44,24 @@ hash_getfuncdef(PyObject *Py_UNUSED(module), PyObject *Py_UNUSED(args))
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 = PyHash_Pointer(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},
};

Expand Down
2 changes: 2 additions & 0 deletions PC/pyconfig.h
Original file line number Diff line number Diff line change
Expand Up @@ -355,6 +355,8 @@ Py_NO_ENABLE_SHARED to find out. Also support MS_NO_COREDLL for b/w compat */
# define ALIGNOF_MAX_ALIGN_T 8
#endif

#define SIZEOF_UINTPTR_T SIZEOF_VOID_P

#ifdef _DEBUG
# define Py_DEBUG
#endif
Expand Down
16 changes: 13 additions & 3 deletions Python/hashtable.c
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,7 @@
*/

#include "Python.h"
#include "pycore_hashtable.h"
#include "pycore_pyhash.h" // _Py_HashPointerRaw()
#include "pycore_hashtable.h" // export _Py_hashtable_new()

#define HASHTABLE_MIN_SIZE 16
#define HASHTABLE_HIGH 0.50
Expand Down Expand Up @@ -89,10 +88,21 @@ _Py_slist_remove(_Py_slist_t *list, _Py_slist_item_t *previous,
}


// Similar to PyHash_Pointer() but avoid "if (x == -1) x = -2;" for best
// performance. The value (Py_uhash_t)-1 is not special for
// _Py_hashtable_t.hash_func function, there is no need to replace it with -2.
Py_uhash_t
_Py_hashtable_hash_ptr(const void *key)
{
return (Py_uhash_t)_Py_HashPointerRaw(key);
uintptr_t x = (uintptr_t)key;
Py_BUILD_ASSERT(sizeof(x) == sizeof(key));

// Bottom 3 or 4 bits are likely to be 0; rotate x by 4 to the right
// to avoid excessive hash collisions.
x = (x >> 4) | (x << (8 * SIZEOF_UINTPTR_T - 4));

Py_BUILD_ASSERT(sizeof(x) == sizeof(Py_hash_t));
return (Py_uhash_t)x;
}


Expand Down
27 changes: 13 additions & 14 deletions Python/pyhash.c
Original file line number Diff line number Diff line change
Expand Up @@ -131,24 +131,23 @@ _Py_HashDouble(PyObject *inst, double v)
return (Py_hash_t)x;
}

// See also _Py_hashtable_hash_ptr() which is similar but can return -1.
Py_hash_t
_Py_HashPointerRaw(const void *p)
PyHash_Pointer(const void *ptr)
{
size_t y = (size_t)p;
/* bottom 3 or 4 bits are likely to be 0; rotate y by 4 to avoid
excessive hash collisions for dicts and sets */
y = (y >> 4) | (y << (8 * SIZEOF_VOID_P - 4));
return (Py_hash_t)y;
}
uintptr_t x = (uintptr_t)ptr;
Py_BUILD_ASSERT(sizeof(x) == sizeof(ptr));

Py_hash_t
_Py_HashPointer(const void *p)
{
Py_hash_t x = _Py_HashPointerRaw(p);
if (x == -1) {
x = -2;
// Bottom 3 or 4 bits are likely to be 0; rotate x by 4 to the right
// to avoid excessive hash collisions for dicts and sets.
x = (x >> 4) | (x << (8 * SIZEOF_UINTPTR_T - 4));

Py_BUILD_ASSERT(sizeof(x) == sizeof(Py_hash_t));
Py_hash_t result = (Py_hash_t)x;
if (result == -1) {
result = -2;
}
return x;
return result;
}

Py_hash_t
Expand Down

0 comments on commit ee4eed7

Please sign in to comment.