Python调用Rust加密库：EphemeralSecret类型序列化问题求助

我正在开发一个供Python调用的Rust加密库，实现了多种加密协议，但在处理EphemeralSecret类型时遇到报错：

"no method named to_bytes found for struct EphemeralSecret in the current scope
method not found in EphemeralSecret"

我需要在Python端操作生成的密钥，虽清楚序列化私钥存在安全风险，但当前场景必须实现。需要解决以下问题：

1. 如何在Rust与Python间传递EphemeralSecret私钥
2. 如何将Python端传回的字节重新转换为EphemeralSecret

问题集中在以下代码段：

// Elliptic Curve Diffie-Hellman
#[pyfunction]
fn generate_ecdh_key() -> PyResult<(Vec<u8>, Vec<u8>)> {
    let private_key = EphemeralSecret::random_from_rng(OsRng);
    let public_key = PublicKey::from(&private_key);
    
    Ok((private_key.to_bytes().to_vec(), public_key.as_bytes().to_vec()))
}

#[pyfunction]
fn derive_ecdh_shared_key(private_key_bytes: Vec<u8>, server_public_key_bytes: Vec<u8>) -> PyResult<Vec<u8>> {
    let private_key = EphemeralSecret::from(private_key_bytes.as_slice().try_into().unwrap());
    let server_public_key = PublicKey::from(server_public_key_bytes.as_slice().try_into().unwrap());

    let shared_secret = private_key.diffie_hellman(&server_public_key);
    Ok(shared_secret.as_bytes().to_vec())
}

1. 序列化EphemeralSecret到字节数组 #

EphemeralSecret并没有提供to_bytes()方法，但它实现了AsRef<[u8]> trait，可直接通过as_ref()获取字节切片，再转换为Vec<u8>传递给Python。

修改generate_ecdh_key函数：

#[pyfunction]
fn generate_ecdh_key() -> PyResult<(Vec<u8>, Vec<u8>)> {
    let private_key = EphemeralSecret::random_from_rng(OsRng);
    let public_key = PublicKey::from(&private_key);
    
    // 利用AsRef<[u8]>获取私钥字节
    let private_key_bytes = private_key.as_ref().to_vec();
    Ok((private_key_bytes, public_key.as_bytes().to_vec()))
}

2. 从字节数组反序列化为EphemeralSecret #

EphemeralSecret::from()方法要求输入&[u8; 32]（固定32字节的数组切片），Python传递的Vec<u8>需要先验证长度，再安全转换为固定长度数组，同时替换unwrap()为错误处理，返回Python可识别的异常。

修改derive_ecdh_shared_key函数：

#[pyfunction]
fn derive_ecdh_shared_key(private_key_bytes: Vec<u8>, server_public_key_bytes: Vec<u8>) -> PyResult<Vec<u8>> {
    // 验证私钥字节长度是否为32
    let private_key_array = private_key_bytes.try_into()
        .map_err(|_| pyo3::exceptions::PyValueError::new_err("Invalid private key length: must be 32 bytes"))?;
    let private_key = EphemeralSecret::from(&private_key_array);

    // 验证公钥字节长度是否为32
    let server_public_key_array = server_public_key_bytes.try_into()
        .map_err(|_| pyo3::exceptions::PyValueError::new_err("Invalid public key length: must be 32 bytes"))?;
    let server_public_key = PublicKey::from(&server_public_key_array);

    let shared_secret = private_key.diffie_hellman(&server_public_key);
    Ok(shared_secret.as_bytes().to_vec())
}

3. 额外注意事项 #

• 安全风险：序列化并传递EphemeralSecret私钥会暴露密钥材料，务必确保传输过程加密，且避免在Python端长期存储私钥。
• 错误处理：替换原代码中的unwrap()为map_err()转换为Python异常，让Python端能捕获并处理错误，提升库的健壮性。

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