blob: f1c05fa37905b9482c6dfb58e0331571452e14f8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
//@only-target: windows # this directly tests windows-only functions
use core::ffi::c_void;
use std::ffi::OsStr;
use std::os::windows::ffi::OsStrExt;
type HANDLE = *mut c_void;
type PWSTR = *mut u16;
type PCWSTR = *const u16;
type HRESULT = i32;
type HLOCAL = *mut ::core::ffi::c_void;
extern "system" {
fn GetCurrentThread() -> HANDLE;
fn GetThreadDescription(hthread: HANDLE, lpthreaddescription: *mut PWSTR) -> HRESULT;
fn SetThreadDescription(hthread: HANDLE, lpthreaddescription: PCWSTR) -> HRESULT;
fn LocalFree(hmem: HLOCAL) -> HLOCAL;
}
fn to_u16s<S: AsRef<OsStr>>(s: S) -> Vec<u16> {
let mut result: Vec<_> = s.as_ref().encode_wide().collect();
result.push(0);
result
}
fn main() {
unsafe {
let name = c"mythreadname";
let utf16 = to_u16s(name.to_str().unwrap());
SetThreadDescription(GetCurrentThread(), utf16.as_ptr());
let mut ptr = core::ptr::null_mut::<u16>();
let result = GetThreadDescription(GetCurrentThread(), &mut ptr);
assert!(result >= 0);
let name_gotten = String::from_utf16_lossy({
let mut len = 0;
while *ptr.add(len) != 0 {
len += 1;
}
core::slice::from_raw_parts(ptr, len)
});
assert_eq!(name_gotten, name.to_str().unwrap());
let r = LocalFree(ptr.cast());
assert!(r.is_null());
}
}
|
