desktoporz 0.2.0

Open source DesktopOK clone: save and restore Windows desktop icon layouts
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

// DesktopORZ — Save and restore Windows desktop icon layouts
// Copyright (C) 2026 ThainanViniciusKatchan
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

use std::ffi::c_void;

use windows::core::Result;
use windows::Win32::Foundation::{CloseHandle, HANDLE};
use windows::Win32::System::Diagnostics::Debug::{ReadProcessMemory, WriteProcessMemory};
use windows::Win32::System::Memory::{
    VirtualAllocEx, VirtualFreeEx, MEM_COMMIT, MEM_RELEASE, MEM_RESERVE, PAGE_READWRITE,
};

pub struct ProcessHandle(HANDLE);

impl ProcessHandle {
    pub fn new(handle: HANDLE) -> Self {
        Self(handle)
    }

    pub fn raw(&self) -> HANDLE {
        self.0
    }
}

impl Drop for ProcessHandle {
    fn drop(&mut self) {
        unsafe {
            let _ = CloseHandle(self.0);
        }
    }
}

pub struct RemoteBuffer<'a> {
    process: &'a ProcessHandle,
    address: *mut c_void,
    size: usize,
}

impl<'a> RemoteBuffer<'a> {
    pub fn new(process: &'a ProcessHandle, size: usize) -> Result<Self> {
        let address = unsafe {
            VirtualAllocEx(
                process.raw(),
                None,
                size,
                MEM_COMMIT | MEM_RESERVE,
                PAGE_READWRITE,
            )
        };
        if address.is_null() {
            return Err(windows::core::Error::from_win32());
        }
        Ok(Self {
            process,
            address,
            size,
        })
    }

    pub fn as_ptr(&self) -> *mut c_void {
        self.address
    }

    pub fn write<T: Copy>(&self, value: &T) -> Result<()> {
        unsafe {
            WriteProcessMemory(
                self.process.raw(),
                self.address,
                value as *const T as *const c_void,
                std::mem::size_of::<T>(),
                None,
            )
        }
    }

    pub fn read<T: Copy + Default>(&self) -> Result<T> {
        let mut value = T::default();
        unsafe {
            ReadProcessMemory(
                self.process.raw(),
                self.address,
                &mut value as *mut T as *mut c_void,
                std::mem::size_of::<T>(),
                None,
            )?;
        }
        Ok(value)
    }

    pub fn read_wide_string(&self) -> Result<String> {
        let mut buf = vec![0u16; self.size / 2];
        unsafe {
            ReadProcessMemory(
                self.process.raw(),
                self.address,
                buf.as_mut_ptr() as *mut c_void,
                buf.len() * 2,
                None,
            )?;
        }
        let end = buf.iter().position(|&c| c == 0).unwrap_or(buf.len());
        Ok(String::from_utf16_lossy(&buf[..end]))
    }
}

impl<'a> Drop for RemoteBuffer<'a> {
    fn drop(&mut self) {
        unsafe {
            let _ = VirtualFreeEx(self.process.raw(), self.address, 0, MEM_RELEASE);
        }
    }
}