如何正确地填充C字符串指针？

Question

我有一个外国金融机构签名，我需要执行：

pub unsafe extern fn f(header_size: u32, header_ptr: *mut u8) -> i32;

FFI调用者需要提供一个缓冲区header\u ptr和该缓冲区的大小header\u size。Rust将向该缓冲区中填充一个字符串，最大值为标题大小，如果成功，则返回0。FFI调用方应将字符串解释为ASCII。

鉴于我有一个头，我如何以最惯用的方式填充缓冲区：

现在我有：

let header_bytes = slice::from_raw_parts_mut(header_ptr, header_size as usize);

if header_bytes.len() < headers.len() { return Errors::IndexOutOfBounds as i32; }

for (i, byte) in headers.as_bytes().iter().enumerate() {
    header_bytes[i] = *byte;
}

但这感觉不对。

编辑，我认为这不是一个完全重复的问题，因为我的问题与字符串有关，而IIRC在转换时有特殊的考虑




Answer 1

由于C字符串不超过0终止字节数组，从Rust字符串转换是非常直接的。几乎每个有效的Rust字符串也是一个有效的C字符串，但是您必须确保C字符串以0字符结束，并且字符串中的其他任何地方都没有0字符。

Rust提供了一个负责转换的类型：CString。

如果您的输入字符串成功地转换为CString，您可以简单地复制字节，而无需担心细节。

use std::slice;
use std::ffi::CString;

pub unsafe extern fn f(header_size: u32, header_ptr: *mut u8) -> i32 {
    let headers = "abc";
    let c_headers = match CString::new(headers) {
        Ok(cs) => cs,
        Err(_) => return -1,  // failed to convert to C string
    };
    let bytes = c_headers.as_bytes_with_nul();

    let header_bytes = slice::from_raw_parts_mut(header_ptr, header_size as usize);
    header_bytes[..bytes.len()].copy_from_slice(bytes);

    0  // success
}

fn main() {
    let mut h = [1u8; 8];

    unsafe {
        f(h.len() as u32, h.as_mut_ptr());
    }

    println!("{:?}", h);  // [97, 98, 99, 0, 1, 1, 1, 1]
}

请注意，为了简洁起见，我省略了长度检查。header_bytes[...bytes.len（）]将在缓冲区太短时感到恐慌。如果从C调用f，则需要避免这种情况。