Question

我正在寻找更方便的方法在Rust中的winapi调用中使用std::String。

使用rust v 0.12.0-nigtly与winapi 0.1.22和user32-sys 0.1.1

现在我使用这样的东西：

use winapi;
use user32;

pub fn get_window_title(handle: i32) -> String {
    let mut v: Vec<u16> = Vec::new();
    v.reserve(255);
    let mut p = v.as_mut_ptr();
    let len = v.len();
    let cap = v.capacity();
    let mut read_len = 0;
    unsafe {
        mem::forget(v);
        read_len = unsafe { user32::GetWindowTextW(handle as winapi::HWND, p, 255) };
        if read_len > 0 {
            return String::from_utf16_lossy(Vec::from_raw_parts(p, read_len as usize, cap).as_slice());
        } else {
            return "".to_string();
        }
    }
}

我认为，这种基于矢量的内存分配相当奇怪。因此，我正在寻找更简单的方式将LPCWSTR投射到std::String

Answer 1

在您的情况下，您总是希望最多255个字节，因此您可以使用数组而不是向量。这会将整个样板减少为mem::uninitialized()调用，as_mut_ptr()调用和切片操作。

unsafe {
    let mut v: [u16; 255] = mem::uninitialized();
    let read_len = user32::GetWindowTextW(
        handle as winapi::HWND,
        v.as_mut_ptr(),
        255,
    );
    String::from_utf16_lossy(&v[0..read_len])
}

如果你想使用Vec，有一种比销毁vec并重新创建它更容易的方法。您可以直接写入Vec的内容，让Rust处理其他所有内容。

let mut v: Vec<u16> = Vec::with_capacity(255);
unsafe {
    let read_len = user32::GetWindowTextW(
        handle as winapi::HWND,
        v.as_mut_ptr(),
        v.capacity(),
    );
    v.set_len(read_len); // this is undefined behavior if read_len > v.capacity()
    String::from_utf16_lossy(&v)
}

作为旁注，Rust中的惯用语不是在函数的最后一个语句中使用return，而是简单地让表达式在没有分号的情况下站在那里。在原始代码中，最终的if-expression可以写成

if read_len > 0 {
    String::from_utf16_lossy(Vec::from_raw_parts(p, read_len as usize, cap).as_slice())
} else {
    "".to_string()
}

但我从我的示例中删除了整个条件，因为没有必要以0字符的方式处理n个字符。

在winapi调用中使用字符串的更方便的方法

1 个答案: