Issue #1621: Fix undefined behaviour in bytes.__hash__, str.__hash__, tuple.__hash__, frozenset.__hash__ and set indexing operations.

This commit is contained in:
Mark Dickinson 2011-09-24 18:18:40 +01:00
parent 0390151100
commit 57e683e53e
5 changed files with 26 additions and 25 deletions

View file

@ -869,16 +869,16 @@ bytes_hash(PyBytesObject *a)
{ {
register Py_ssize_t len; register Py_ssize_t len;
register unsigned char *p; register unsigned char *p;
register Py_hash_t x; register Py_uhash_t x;
if (a->ob_shash != -1) if (a->ob_shash != -1)
return a->ob_shash; return a->ob_shash;
len = Py_SIZE(a); len = Py_SIZE(a);
p = (unsigned char *) a->ob_sval; p = (unsigned char *) a->ob_sval;
x = *p << 7; x = (Py_uhash_t)*p << 7;
while (--len >= 0) while (--len >= 0)
x = (1000003*x) ^ *p++; x = (1000003U*x) ^ (Py_uhash_t)*p++;
x ^= Py_SIZE(a); x ^= (Py_uhash_t)Py_SIZE(a);
if (x == -1) if (x == -1)
x = -2; x = -2;
a->ob_shash = x; a->ob_shash = x;

View file

@ -418,7 +418,7 @@ lookdict_unicode(PyDictObject *mp, PyObject *key, register Py_hash_t hash)
mp->ma_lookup = lookdict; mp->ma_lookup = lookdict;
return lookdict(mp, key, hash); return lookdict(mp, key, hash);
} }
i = hash & mask; i = (size_t)hash & mask;
ep = &ep0[i]; ep = &ep0[i];
if (ep->me_key == NULL || ep->me_key == key) if (ep->me_key == NULL || ep->me_key == key)
return ep; return ep;
@ -572,7 +572,7 @@ insertdict_clean(register PyDictObject *mp, PyObject *key, Py_hash_t hash,
register PyDictEntry *ep; register PyDictEntry *ep;
MAINTAIN_TRACKING(mp, key, value); MAINTAIN_TRACKING(mp, key, value);
i = hash & mask; i = (size_t)hash & mask;
ep = &ep0[i]; ep = &ep0[i];
for (perturb = hash; ep->me_key != NULL; perturb >>= PERTURB_SHIFT) { for (perturb = hash; ep->me_key != NULL; perturb >>= PERTURB_SHIFT) {
i = (i << 2) + i + perturb + 1; i = (i << 2) + i + perturb + 1;

View file

@ -77,7 +77,7 @@ NULL if the rich comparison returns an error.
static setentry * static setentry *
set_lookkey(PySetObject *so, PyObject *key, register Py_hash_t hash) set_lookkey(PySetObject *so, PyObject *key, register Py_hash_t hash)
{ {
register Py_ssize_t i; register size_t i;
register size_t perturb; register size_t perturb;
register setentry *freeslot; register setentry *freeslot;
register size_t mask = so->mask; register size_t mask = so->mask;
@ -86,7 +86,7 @@ set_lookkey(PySetObject *so, PyObject *key, register Py_hash_t hash)
register int cmp; register int cmp;
PyObject *startkey; PyObject *startkey;
i = hash & mask; i = (size_t)hash & mask;
entry = &table[i]; entry = &table[i];
if (entry->key == NULL || entry->key == key) if (entry->key == NULL || entry->key == key)
return entry; return entry;
@ -159,7 +159,7 @@ set_lookkey(PySetObject *so, PyObject *key, register Py_hash_t hash)
static setentry * static setentry *
set_lookkey_unicode(PySetObject *so, PyObject *key, register Py_hash_t hash) set_lookkey_unicode(PySetObject *so, PyObject *key, register Py_hash_t hash)
{ {
register Py_ssize_t i; register size_t i;
register size_t perturb; register size_t perturb;
register setentry *freeslot; register setentry *freeslot;
register size_t mask = so->mask; register size_t mask = so->mask;
@ -174,7 +174,7 @@ set_lookkey_unicode(PySetObject *so, PyObject *key, register Py_hash_t hash)
so->lookup = set_lookkey; so->lookup = set_lookkey;
return set_lookkey(so, key, hash); return set_lookkey(so, key, hash);
} }
i = hash & mask; i = (size_t)hash & mask;
entry = &table[i]; entry = &table[i];
if (entry->key == NULL || entry->key == key) if (entry->key == NULL || entry->key == key)
return entry; return entry;
@ -256,7 +256,7 @@ set_insert_clean(register PySetObject *so, PyObject *key, Py_hash_t hash)
setentry *table = so->table; setentry *table = so->table;
register setentry *entry; register setentry *entry;
i = hash & mask; i = (size_t)hash & mask;
entry = &table[i]; entry = &table[i];
for (perturb = hash; entry->key != NULL; perturb >>= PERTURB_SHIFT) { for (perturb = hash; entry->key != NULL; perturb >>= PERTURB_SHIFT) {
i = (i << 2) + i + perturb + 1; i = (i << 2) + i + perturb + 1;
@ -770,14 +770,14 @@ static Py_hash_t
frozenset_hash(PyObject *self) frozenset_hash(PyObject *self)
{ {
PySetObject *so = (PySetObject *)self; PySetObject *so = (PySetObject *)self;
Py_hash_t h, hash = 1927868237L; Py_uhash_t h, hash = 1927868237U;
setentry *entry; setentry *entry;
Py_ssize_t pos = 0; Py_ssize_t pos = 0;
if (so->hash != -1) if (so->hash != -1)
return so->hash; return so->hash;
hash *= PySet_GET_SIZE(self) + 1; hash *= (Py_uhash_t)PySet_GET_SIZE(self) + 1;
while (set_next(so, &pos, &entry)) { while (set_next(so, &pos, &entry)) {
/* Work to increase the bit dispersion for closely spaced hash /* Work to increase the bit dispersion for closely spaced hash
values. The is important because some use cases have many values. The is important because some use cases have many
@ -785,11 +785,11 @@ frozenset_hash(PyObject *self)
hashes so that many distinct combinations collapse to only hashes so that many distinct combinations collapse to only
a handful of distinct hash values. */ a handful of distinct hash values. */
h = entry->hash; h = entry->hash;
hash ^= (h ^ (h << 16) ^ 89869747L) * 3644798167u; hash ^= (h ^ (h << 16) ^ 89869747U) * 3644798167U;
} }
hash = hash * 69069L + 907133923L; hash = hash * 69069U + 907133923U;
if (hash == -1) if (hash == -1)
hash = 590923713L; hash = 590923713U;
so->hash = hash; so->hash = hash;
return hash; return hash;
} }

View file

@ -315,11 +315,12 @@ Done:
static Py_hash_t static Py_hash_t
tuplehash(PyTupleObject *v) tuplehash(PyTupleObject *v)
{ {
register Py_hash_t x, y; register Py_uhash_t x;
register Py_hash_t y;
register Py_ssize_t len = Py_SIZE(v); register Py_ssize_t len = Py_SIZE(v);
register PyObject **p; register PyObject **p;
Py_hash_t mult = 1000003L; Py_uhash_t mult = 1000003;
x = 0x345678L; x = 0x345678;
p = v->ob_item; p = v->ob_item;
while (--len >= 0) { while (--len >= 0) {
y = PyObject_Hash(*p++); y = PyObject_Hash(*p++);
@ -330,7 +331,7 @@ tuplehash(PyTupleObject *v)
mult += (Py_hash_t)(82520L + len + len); mult += (Py_hash_t)(82520L + len + len);
} }
x += 97531L; x += 97531L;
if (x == -1) if (x == (Py_uhash_t)-1)
x = -2; x = -2;
return x; return x;
} }

View file

@ -7721,22 +7721,22 @@ unicode_getitem(PyUnicodeObject *self, Py_ssize_t index)
} }
/* Believe it or not, this produces the same value for ASCII strings /* Believe it or not, this produces the same value for ASCII strings
as string_hash(). */ as bytes_hash(). */
static Py_hash_t static Py_hash_t
unicode_hash(PyUnicodeObject *self) unicode_hash(PyUnicodeObject *self)
{ {
Py_ssize_t len; Py_ssize_t len;
Py_UNICODE *p; Py_UNICODE *p;
Py_hash_t x; Py_uhash_t x;
if (self->hash != -1) if (self->hash != -1)
return self->hash; return self->hash;
len = Py_SIZE(self); len = Py_SIZE(self);
p = self->str; p = self->str;
x = *p << 7; x = (Py_uhash_t)*p << 7;
while (--len >= 0) while (--len >= 0)
x = (1000003*x) ^ *p++; x = (1000003U*x) ^ (Py_uhash_t)*p++;
x ^= Py_SIZE(self); x ^= (Py_uhash_t)Py_SIZE(self);
if (x == -1) if (x == -1)
x = -2; x = -2;
self->hash = x; self->hash = x;