将Windows HICON转换为HTML前端显示的原始数据

Question

我正在使用Rust和SolidJS制作一个Tauri窗口切换器，并利用winapi与Windows系统交互。目前，我已经实现了以下功能：

创建了一个HashMap，键为窗口标题（字符串），值为HWND（窗口句柄）。
将窗口标题发送至前端展示。
当前端点击某个标题时，能够成功切换到相应的窗口。

现在，我希望在每个窗口标题旁显示应用程序图标，以增强可识别性。为此，我编写了如下方法：

unsafe extern "system" fn enum_windows_proc(hwnd: HWND, lparam: LPARAM) -> BOOL {
    let title_length = GetWindowTextLengthW(hwnd);
    if title_length > 0 {
        let mut buffer: Vec = vec![0; (title_length + 1) as usize]; 
        GetWindowTextW(hwnd, buffer.as_mut_ptr(), buffer.len() as i32);

        if IsWindowVisible(hwnd) != 0 {
            let window_title = OsString::from_wide(&buffer).to_string_lossy().to_string();
            let title_map = &mut *(lparam as *mut HashMap);
            let app_icon = get_app_icon(hwnd);

            if let Some(icon) = app_icon {
                title_map.insert(window_title, (hwnd, icon));
            }
        }
    }

    1
}

fn get_app_icon(hwnd: HWND) -> Option {
    let icon_handle: usize = unsafe { GetClassLongPtrW(hwnd, GCLP_HICON as i32) };
    if icon_handle == 0 {
        None
    } else {
        Some(icon_handle as HICON)
    }
}

我已经修改了代码结构，以便在存储窗口标题的同时也能获取并存储其图标句柄（HICON）。我想确认我的HICON获取方法是否正确，以及如何正确地打印icon_handle的值进行调试。当前尝试打印icon_handle时没有输出，但如果替换为任何字符串则能正常打印。

Matt Sheppard · Answer

以下是将 HICON 转换为 PNG 图标 base64 编码表示的方法：

use base64::{engine::general_purpose, Engine as _};
use image::RgbaImage;
use std::{mem::{self, MaybeUninit}, ptr::addr_of_mut};
use windows::Win32::{Foundation::HWND, Graphics::Gdi::{DeleteObject, GetDC, GetDIBits, GetObjectW, ReleaseDC, BITMAP, BITMAPINFOHEADER, BI_RGB, DIB_RGB_COLORS, HDC}, UI::WindowsAndMessaging::{GetIconInfo, HICON}};

unsafe fn icon_to_rgba_image(icon: HICON) -> String {
    let bitmap_size = i32::try_from(mem::size_of::()).unwrap();
    let biheader_size = u32::try_from(mem::size_of::()).unwrap();

    let mut info = MaybeUninit::uninit();
    GetIconInfo(icon, info.as_mut_ptr()).unwrap();
    let info = info.assume_init_ref();
    DeleteObject(info.hbmMask).unwrap();

    let mut bitmap: MaybeUninit = MaybeUninit::uninit();
    let result = GetObjectW(
        info.hbmColor,
        bitmap_size,
        Some(bitmap.as_mut_ptr().cast()),
    );
    assert!(result == bitmap_size);
    let bitmap = bitmap.assume_init_ref();

    let width = u32::try_from(bitmap.bmWidth).unwrap();
    let height = u32::try_from(bitmap.bmHeight).unwrap();
    let w = usize::try_from(bitmap.bmWidth).unwrap();
    let h = usize::try_from(bitmap.bmHeight).unwrap();

    let buf_size = w
        .checked_mul(h)
        .and_then(|size| size.checked_mul(4))
        .unwrap();
    let mut buf: Vec = Vec::with_capacity(buf_size);

    let dc = GetDC(HWND(0));
    assert!(dc != HDC(0));

    let mut bitmap_info = BITMAPINFOHEADER {
        biSize: biheader_size,
        biWidth: bitmap.bmWidth,
        biHeight: -bitmap.bmHeight,
        biPlanes: 1,
        biBitCount: 32,
        biCompression: BI_RGB,
        biSizeImage: 0,
        biXPelsPerMeter: 0,
        biYPelsPerMeter: 0,
        biClrUsed: 0,
        biClrImportant: 0,
    };
    let result = GetDIBits(
        dc,
        info.hbmColor,
        0,
        height,
        Some(buf.as_mut_ptr().cast()),
        addr_of_mut!(bitmap_info).cast(),
        DIB_RGB_COLORS,
    );
    buf.set_len(buf.capacity());

    let result = ReleaseDC(HWND(0), dc);
    assert!(result == 1);
    DeleteObject(info.hbmColor).unwrap();

    for chunk in buf.chunks_exact_mut(4) {
        let [b, _, r, _] = chunk else { unreachable!() };
        mem::swap(b, r);
    }

    let img = RgbaImage::from_vec(width, height, buf).unwrap();

    let mut data: Vec = Vec::new();
    img.write_to(&mut Cursor::new(&mut data), image::ImageOutputFormat::Png).unwrap();

    general_purpose::STANDARD.encode(&data);
}

Tauri 是构建在 TAO 之上的，而 TAO 又是 winit 的一个分支，专门实现了跨平台的窗口管理。

具体来说，在 windows/window.rs 中处理了与窗口相关的实现。因此，你可能需要克隆 TAO 的仓库，并在那里添加必要的修改，然后使用 Cargo 的依赖覆盖功能，使其指向你的 TAO 版本。

TAO 实现了与 hicon 相关的类，如下所示：

let class = WNDCLASSEXW {
    cbSize: mem::size_of::() as u32,
    style: CS_HREDRAW | CS_VREDRAW | CS_OWNDC,
    lpfnWndProc: Some(window_proc),
    cbClsExtra: 0,
    cbWndExtra: 0,
    hInstance: HINSTANCE(GetModuleHandleW(PCWSTR::null()).unwrap_or_default().0),
    hIcon: HICON::default(),
    hCursor: HCURSOR::default(), // 鼠标光标状态正常工作必须设为 null
    hbrBackground: HBRUSH::default(),
    lpszMenuName: PCWSTR::null(),
    lpszClassName: PCWSTR::from_raw(class_name.as_ptr()),
    hIconSm: HICON::default(),
};

值得一提的是，Tauri 使用了 windows 这个库来替代 winapi，尽管下载量较少，但微软提供了更好的支持，且与 apple-sys 对 Mac 的支持方式类似。

如 Jonathan Potter 所指出的，要获取 hicon 的信息，你需要使用 GetIconInfo 函数，它会提供一个包含掩码和彩色位图句柄的 ICONINFO 结构体。

直接打印 hicon 是不可能的，因为它只是一个句柄，用来引用图标资源：

pub type HICON = isize;

附注：这是 TAO 检查 Windows 上窗口是否可见的方式。同时，在同一段代码中，有大约七种方法可以获取窗口的硬件ID，包括以 HWND 及其他格式。

bbum · Answer

感谢@DeveloperMindset.com提供的关键帮助。

我对其做了一些调整，以便使用 winapi 进行工作，因为我未能成功地在 window.rs 中应用它，并且修改了方法，以便从exe路径中提取图标。

unsafe fn icon_to_rgba_image(icon: HICON, hwnd: HWND) -> String {
    let bitmap_size = i32::try_from(mem::size_of::()).unwrap();
    let biheader_size = u32::try_from(mem::size_of::()).unwrap();

    let mut info = MaybeUninit::uninit();
    GetIconInfo(icon, info.as_mut_ptr());
    let info = info.assume_init_ref();
    DeleteObject(info.hbmMask as *mut c_void);

    let mut bitmap: MaybeUninit = MaybeUninit::uninit();
    let result = GetObjectW(
        info.hbmColor as *mut c_void,
        bitmap_size,
        bitmap.as_mut_ptr().cast(),
    );

    if result != bitmap_size {
        return "".to_string();
    }

    let bitmap = bitmap.assume_init_ref();

    let width = u32::try_from(bitmap.bmWidth).unwrap();
    let height = u32::try_from(bitmap.bmHeight).unwrap();
    let w = usize::try_from(bitmap.bmWidth).unwrap();
    let h = usize::try_from(bitmap.bmHeight).unwrap();

    let buf_size = w
        .checked_mul(h)
        .and_then(|size| size.checked_mul(4))
        .unwrap();
    let mut buf: Vec = Vec::with_capacity(buf_size);

    let dc = GetDC(hwnd);

    let mut bitmap_info = BITMAPINFOHEADER {
        biSize: biheader_size,
        biWidth: bitmap.bmWidth,
        biHeight: -bitmap.bmHeight,
        biPlanes: 1,
        biBitCount: 32,
        biCompression: 0, // 修改了biCompression为0，对应BI_RGB常量
        biSizeImage: 0,
        biXPelsPerMeter: 0,
        biYPelsPerMeter: 0,
        biClrUsed: 0,
        biClrImportant: 0,
    };
    GetDIBits(
        dc,
        info.hbmColor,
        0,
        height,
        buf.as_mut_ptr().cast(),
        addr_of_mut!(bitmap_info).cast(),
        DIB_RGB_COLORS,
    );

    buf.set_len(buf.capacity());

    let result = ReleaseDC(hwnd, dc);
    assert!(result == 1);
    DeleteObject(info.hbmColor as *mut c_void);

    for chunk in buf.chunks_exact_mut(4) {
        let [b, _, r, _] = chunk else { unreachable!() };
        mem::swap(b, r);
    }

    let img = RgbaImage::from_vec(width, height, buf).unwrap();

    let mut data: Vec = Vec::new();
    img.write_to(&mut Cursor::new(&mut data), image::ImageOutputFormat::Png)
        .unwrap();

    general_purpose::STANDARD.encode(&data)
}