handle winapis

sha0coder · sha0coder · commit 93982f41ed1a · 2024-10-23T08:11:19.000+02:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -10,7 +10,7 @@ crate-type = ["cdylib"]
 
 [dependencies]
 pyo3 = "0.18.1"
-libscemu = "0.18.2"
+libscemu = "0.18.3"
 #libscemu = { path = "/home/sha0/src/libscemu/" }
 #libscemu = { path = "/Users/sha0coder/src/libscemu/" }
 #libscemu = { path = "/Users/jesus/src/libscemu/" }
diff --git a/DOCUMENTATION.md b/DOCUMENTATION.md
@@ -445,6 +445,24 @@ libscemu provide multipe hooks, but not pyscemu.
 one opion is synchronous way:
 
 ```python
+def GetUserNameA():
+    retaddr = emu.stack_pop64()
+    print('GetUserNameA')
+    emu.write_string(emu.get_reg('rcx'), 'baremetal\x00')
+    emu.write_qword(emu.get_reg('rdx'), 9)
+    emu.set_reg('rax', emu.get_reg('rcx'))
+
+def recv():
+    retaddr = emu.stack_pop64()
+    rip = emu.get_reg('rip')
+    rcx = emu.get_reg('rcx')
+    rdx = emu.get_reg('rdx')
+    r8 = emu.get_reg('r8')
+    print(f'{rip:x}: recv({rcx}, {rdx:x}, {r8})')
+    emu.write_dword(rdx, 3)
+    emu.set_reg('rax', 4)
+
+
 emu.set_reg('rip', comm_protocol)
 while True:
     addr, name = emu.run_until_apicall()
@@ -453,6 +471,6 @@ while True:
     elif name =='recv':
         recv()
     else:
-        print(f'unhandled call {name}')
+        emu.handle_winapi(addr)
 ```
 
diff --git a/src/lib.rs b/src/lib.rs
@@ -800,6 +800,11 @@ impl Emu {
         return Ok(self.emu.bp.get_mem_write());
     }
 
+    /// handle winapi address
+    pub fn handle_winapi(&mut self, addr: u64) {
+        self.emu.handle_winapi(addr);
+    }
+
     /// emulate until next winapi call
     pub fn run_until_apicall(&mut self) -> PyResult<(u64,String)> {
         self.emu.skip_apicall = true;
