python - PyDict_SetItem 是否会增加键的引用计数？如果是，它发生在代码中的哪个位置？

Question

TLDR:PyDict_SetItem 递增键和值，但是这在代码中的什么位置发生？

PyDict_SetItem调用 insertdict。

insertdict立即对键和值执行Py_INCREF。然而，在 end of the success path ，然后它对键(但不是值)执行 Py_DECREF 。我一定缺少这段代码的某些部分，在执行此 PY_DECREF 之前，它在键上执行了额外的 PY_INCREF 操作。 我的问题是这个额外的 PY_INCREF 在哪里以及为什么发生？为什么 insertdict 开头的初始 Py_INCREF 不足？

由此看来，乍一看PyDict_SetItem只是增加了value的引用计数，而没有增加key的引用计数。当然，这不是真的。例如，在PyDict_SetItemString中，它接受一个 char *，通过 PyUnicode_FromString 将其转换为 PyObject(返回一个新值)，对该新值执行 Py_DECREF调用 PyDict_SetItem 后。如果 PyDict_SetItem 不递增该键，并且 PyDict_SetItemString 递减刚刚创建的键，则程序最终可能会出现段错误。鉴于这种情况没有发生，看来我在这里遗漏了一些东西。

<小时/>

最后，此代码应证明 PyDict_SetItem 会递增键和值，并且调用者应同时递减键和值，除非它们是借用引用/或将给出键和值给别人。

#include <Python.h>
#include <stdio.h>

int main(void)
{
    Py_Initialize();
    PyObject *dict = NULL, *key = NULL, *value = NULL;
    int i = 5000;
    char *o = "foo";

    if (!(dict = PyDict_New())) {
        goto error;
    }
    if (!(key = Py_BuildValue("i", i))) {
        goto error;
    }
    if (!(value = Py_BuildValue("s", o))) {
        goto error;
    }
    printf("Before PyDict_SetItem\n");
    printf("key is %i\n", key->ob_refcnt);  /* Prints 1 */
    printf("value is %i\n", value->ob_refcnt);  /* Prints 1 */

    printf("Calling PyDict_SetItem\n");
    if (PyDict_SetItem(dict, key, value) < 0) {
        goto error;
    }
    printf("key is %i\n", key->ob_refcnt);  /* Prints 2 */
    printf("value is %i\n", value->ob_refcnt);  /* Prints 2 */

    printf("Decrefing key and value\n");
    Py_DECREF(key);
    Py_DECREF(value);
    printf("key is %i\n", key->ob_refcnt);   /* Prints 1 */
    printf("value is %i\n", value->ob_refcnt);   /* Prints 1 */

    Py_Finalize();
    return 0; // would return the dict in normal code
error:
    printf("error");
    Py_XDECREF(dict);
    Py_XDECREF(key);
    Py_XDECREF(value);
    Py_Finalize();
    return 1;
}

你可以这样编译:

gcc -c -I/path/to/python/include/python3.7m dict.c
gcc dict.o -L/path/to/python/lib/python3.7/config-3.7m-i386-linux-gnu -L/path/to/python/lib -Wl,-rpath=/path/to/python/lib -lpython3.7m -lpthread -lm -ldl -lutil -lrt -Xlinker -export-dynamic -m32

Answer 1

insertdict 中的 Py_DECREF(key); 并非所有成功都会发生。当相等的键已经存在时，就会发生这种情况，要么是因为存在现有条目，要么是因为该字典是一个与具有该键的其他字典共享键的拆分表字典。在该路径上，未插入提供的 key ，因此需要取消原始 Py_INCREF(key);。

在键不存在的路径上，insertdict 命中 different return statement并且不减少 key :

if (ix == DKIX_EMPTY) {
    /* Insert into new slot. */
    assert(old_value == NULL);
    if (mp->ma_keys->dk_usable <= 0) {
        /* Need to resize. */
        if (insertion_resize(mp) < 0)
            goto Fail;
    }
    Py_ssize_t hashpos = find_empty_slot(mp->ma_keys, hash);
    ep = &DK_ENTRIES(mp->ma_keys)[mp->ma_keys->dk_nentries];
    dictkeys_set_index(mp->ma_keys, hashpos, mp->ma_keys->dk_nentries);
    ep->me_key = key;
    ep->me_hash = hash;
    if (mp->ma_values) {
        assert (mp->ma_values[mp->ma_keys->dk_nentries] == NULL);
        mp->ma_values[mp->ma_keys->dk_nentries] = value;
    }
    else {
        ep->me_value = value;
    }
    mp->ma_used++;
    mp->ma_version_tag = DICT_NEXT_VERSION();
    mp->ma_keys->dk_usable--;
    mp->ma_keys->dk_nentries++;
    assert(mp->ma_keys->dk_usable >= 0);
    ASSERT_CONSISTENT(mp);
    return 0;
}