Mercurial > cpython-lru_cache-weakref
diff Objects/setobject.c @ 64389:d56b3cafb1e6
Issue 8420: Fix obscure set crashers.
| author | Raymond Hettinger <python@rcn.com> |
|---|---|
| date | Fri, 03 Sep 2010 10:00:50 +0000 |
| parents | 81a9ac73d619 |
| children | b65d0371a731 |
line wrap: on
line diff
--- a/Objects/setobject.c Fri Sep 03 04:22:36 2010 +0000 +++ b/Objects/setobject.c Fri Sep 03 10:00:50 2010 +0000 @@ -364,12 +364,13 @@ set_add_entry(register PySetObject *so, setentry *entry) { register Py_ssize_t n_used; + PyObject *key = entry->key; assert(so->fill <= so->mask); /* at least one empty slot */ n_used = so->used; - Py_INCREF(entry->key); - if (set_insert_key(so, entry->key, (long) entry->hash) == -1) { - Py_DECREF(entry->key); + Py_INCREF(key); + if (set_insert_key(so, key, (long) entry->hash) == -1) { + Py_DECREF(key); return -1; } if (!(so->used > n_used && so->fill*3 >= (so->mask+1)*2)) @@ -637,6 +638,7 @@ set_merge(PySetObject *so, PyObject *otherset) { PySetObject *other; + PyObject *key; register Py_ssize_t i; register setentry *entry; @@ -657,11 +659,12 @@ } for (i = 0; i <= other->mask; i++) { entry = &other->table[i]; - if (entry->key != NULL && - entry->key != dummy) { - Py_INCREF(entry->key); - if (set_insert_key(so, entry->key, (long) entry->hash) == -1) { - Py_DECREF(entry->key); + key = entry->key; + if (key != NULL && + key != dummy) { + Py_INCREF(key); + if (set_insert_key(so, key, (long) entry->hash) == -1) { + Py_DECREF(key); return -1; } } @@ -1642,15 +1645,22 @@ while (_PyDict_Next(other, &pos, &key, &value, &hash)) { setentry an_entry; + Py_INCREF(key); an_entry.hash = hash; an_entry.key = key; + rv = set_discard_entry(so, &an_entry); - if (rv == -1) + if (rv == -1) { + Py_DECREF(key); return NULL; + } if (rv == DISCARD_NOTFOUND) { - if (set_add_entry(so, &an_entry) == -1) + if (set_add_entry(so, &an_entry) == -1) { + Py_DECREF(key); return NULL; + } } + Py_DECREF(key); } Py_RETURN_NONE; }
