Trtyr's Blog

1
use std::{mem::size_of, ptr};
2
use windows::core::{PCWSTR, PWSTR};
3
use windows::Win32::Foundation::{HANDLE, HWND};
4
use windows::Win32::Security::WinTrust::{
5
    WinVerifyTrustEx, WINTRUST_ACTION_GENERIC_VERIFY_V2, WINTRUST_DATA, WINTRUST_DATA_0,
6
    WINTRUST_DATA_UICONTEXT, WINTRUST_FILE_INFO, WTD_CHOICE_FILE, WTD_REVOKE_NONE,
7
    WTD_STATEACTION_CLOSE, WTD_STATEACTION_VERIFY, WTD_UI_NONE, WTD_USE_DEFAULT_OSVER_CHECK,
8
};
9

10
fn signature(path: &str) -> bool {
11
    // 6. 把 path 改成 PCWSTR
12
    let mut path_str = path.encode_utf16().collect::<Vec<u16>>();
13
    path_str.push(b'\0' as u16);
14
    let path_ptr = PCWSTR(path_str.as_mut_ptr());
15

16
    // 5. 创建 WINTRUST_FILE_INFO 结构体
17
    let mut file_info = WINTRUST_FILE_INFO {
18
        cbStruct: size_of::<WINTRUST_FILE_INFO>() as u32, // 结构体大小
19
        pcwszFilePath: path_ptr,                          // 一个 PCWSTR 类型，代表文件路径
20
        hFile: HANDLE::default(),                         // 默认
21
        pgKnownSubject: ptr::null_mut(),                  // 默认
22
    };
23

24
    // 4. 定义 WINTRUST_DATA_0，由于我们设置的是`dwUnionChoice: WTD_CHOICE_FILE`，所以我们只需要`pFile`
25
    let wintrust_data_0 = WINTRUST_DATA_0 {
26
        pFile: &mut file_info as *mut _,
27
    };
28

29
    // 3. 准备 WINTRUST_DATA 结构体
30
    let mut wintrust_data = WINTRUST_DATA {
31
        cbStruct: size_of::<WINTRUST_DATA>() as u32, // 结构体大小，为 u32 类型
32
        pPolicyCallbackData: ptr::null_mut(), // 设置为 null_mut() 表示我们不使用任何自定义的回调函数来在验证过程中途干预。
33
        pSIPClientData: ptr::null_mut(), // 设置为 null_mut() 表示不向 SIP（主题接口包）提供任何额外的数据。
34
        dwUIChoice: WTD_UI_NONE, // 不显示 UI，即使签名无效或不受信任，也不要弹出任何警告窗口。
35
        fdwRevocationChecks: WTD_REVOKE_NONE, // 不检查吊销，跳过联网检查证书吊销列表（CRL）
36
        dwUnionChoice: WTD_CHOICE_FILE, // 验证文件
37
        Anonymous: wintrust_data_0, // 准备验证文件的 WINTRUST_DATA_0 联合体
38
        dwStateAction: WTD_STATEACTION_VERIFY, // 启用验证，这会初始化并保留验证状态的句柄（hWVTStateData），这会在内存里占用资源，记得释放资源
39
        hWVTStateData: HANDLE::default(),      // 直接默认
40
        pwszURLReference: PWSTR::null(),       // 我们不需要它，设为null
41
        dwProvFlags: WTD_USE_DEFAULT_OSVER_CHECK, // 告诉系统使用它自己的默认版本检查逻辑
42
        dwUIContext: WINTRUST_DATA_UICONTEXT(0), // 仅在 dwUIChoice 不是 WTD_UI_NONE 时才有意义，我们禁用了UI，这里设为 0
43
        pSignatureSettings: ptr::null_mut(),     // 使用默认行为（验证第一个签名）
44
    };
45

46
    // 2. 定义我们要做的操作的 GUID，需要的类型为: `*mut windows_core::GUID`
47
    let mut action_id = WINTRUST_ACTION_GENERIC_VERIFY_V2;
48

49
    // 1. 调用 WinVerifyTrustEx
50
    let result = unsafe {
51
        WinVerifyTrustEx(
52
            HWND::default(),    // 窗口句柄，默认 0
53
            &mut action_id,     // GUID
54
            &mut wintrust_data, // WINTRUST_DATA 结构体
55
        )
56
    };
57

58
    // 7. WTD_STATEACTION_VERIFY 会验证并返回一个状态句柄到 hWVTStateData，释放这个句柄需要再次调用 WinVerifyTrustEx，并且把 dwStateAction 设置为 WTD_STATEACTION_CLOSE。
59
    wintrust_data.dwStateAction = WTD_STATEACTION_CLOSE;
60
    unsafe {
61
        WinVerifyTrustEx(HWND::default(), &mut action_id, &mut wintrust_data);
62
    }
63

64
    match result {
65
        0 => true,
66
        _ => false,
67
    }
68
}
69

70
fn main() {
71
    let path = "D:/SoftWare/Weixin/Weixin.exe";
72
    let code = signature(path);
73
    println!("Code: {}", code);
74
}