使用Python.h高效传递C uint32_t数组拼接大整数至Python

你当前的方案存在两个核心问题：

• 类型不匹配：s#格式返回的是Python bytes 类型，不是你预期的整数类型，Python侧需要额外转换步骤。
• 字节序问题：直接memcpy uint32数组的原始内存会受宿主平台字节序影响：x86等小端平台上，uint32 0x00000300 的低地址到高地址字节为 0x00 0x03 0x00 0x00，和你期望的按十六进制顺序拼接的大端布局（0x00 0x00 0x03 0x00）完全不符，最终拼接的整数自然出错。

根据你最终需要的返回类型，可选择以下两种方案，优先推荐方案1：

方案1：C侧直接构建Python整数（最高效） #

如果最终需求就是返回整数，直接在C侧按位拼接数值后构建Python长整数对象，完全避免中间bytes的生成开销，性能最优。
你的需求本质是计算：整数值 = (uint64_t)state[0] << 64 | (uint64_t)state[1] << 32 | state[2]
实现代码：

#include <Python.h>
#include <stdint.h>

static PyObject* get_combined_state(PyObject* self, PyObject* args) {
    uint32_t state[3] = {0x00000300, 0x00001100, 0x00022200};
    uint64_t high = ((uint64_t)state[0] << 32) | state[1];
    
    PyObject *py_high = NULL, *py_low = NULL, *py_shift = NULL, *py_shifted = NULL, *result = NULL;
    
    // 构建各部分的Python长整数对象
    py_high = PyLong_FromUnsignedLongLong(high);
    if (!py_high) goto cleanup;
    py_low = PyLong_FromUnsignedLong(state[2]);
    if (!py_low) goto cleanup;
    py_shift = PyLong_FromLong(32);
    if (!py_shift) goto cleanup;

    // 计算 high << 32 后拼接低32位
    py_shifted = PyNumber_Lshift(py_high, py_shift);
    if (!py_shifted) goto cleanup;
    result = PyNumber_Add(py_shifted, py_low);

cleanup:
    // 释放临时对象，避免内存泄漏
    Py_XDECREF(py_high);
    Py_XDECREF(py_low);
    Py_XDECREF(py_shift);
    Py_XDECREF(py_shifted);
    return result;
}

C侧返回后，Python侧直接拿到的就是目标整数 0x000003000000110000022200，无需额外转换。

方案2：修正Bytes返回方案（需序列化场景使用） #

如果后续有其他序列化需求需要用到bytes对象，需要手动将每个uint32转为大端字节后再拼接，不能直接memcpy原始内存：

// 不受平台字节序影响的uint32转大端字节工具函数
static void uint32_to_be_bytes(uint32_t val, char out[4]) {
    out[0] = (val >> 24) & 0xFF;
    out[1] = (val >> 16) & 0xFF;
    out[2] = (val >> 8) & 0xFF;
    out[3] = val & 0xFF;
}

static PyObject* get_combined_state_bytes(PyObject* self, PyObject* args) {
    uint32_t state[3] = {0x00000300, 0x00001100, 0x00022200};
    char charstate[12];
    // 按数组顺序拼接每个uint32的大端字节
    uint32_to_be_bytes(state[0], charstate);
    uint32_to_be_bytes(state[1], charstate + 4);
    uint32_to_be_bytes(state[2], charstate + 8);
    // 返回bytes对象，明确使用PyBytes接口避免兼容问题
    return PyBytes_FromStringAndSize(charstate, 12);
}

Python侧仅需执行 int.from_bytes(返回的bytes, byteorder='big') 即可得到目标整数。

内容的提问来源于stack exchange，提问作者Juan