From ec1c9f9410330ec7de313ae131a5bfb0d440fc3f Mon Sep 17 00:00:00 2001
From: Noah <33094578+coolreader18@users.noreply.github.com>
Date: Wed, 10 Jun 2020 11:25:04 -0500
Subject: [PATCH] Add os.kill

---
 vm/Cargo.toml       |  2 +-
 vm/src/stdlib/os.rs | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/vm/Cargo.toml b/vm/Cargo.toml
index 7a123c260..d8bc737b4 100644
--- a/vm/Cargo.toml
+++ b/vm/Cargo.toml
@@ -122,7 +122,7 @@ schannel = "0.1"
 
 [target.'cfg(windows)'.dependencies.winapi]
 version = "0.3"
-features = ["winsock2", "handleapi", "ws2def", "std", "winbase", "wincrypt", "fileapi"]
+features = ["winsock2", "handleapi", "ws2def", "std", "winbase", "wincrypt", "fileapi", "wincon"]
 
 [target.'cfg(target_arch = "wasm32")'.dependencies]
 wasm-bindgen = "0.2"
diff --git a/vm/src/stdlib/os.rs b/vm/src/stdlib/os.rs
index a960f980a..59bccb4a3 100644
--- a/vm/src/stdlib/os.rs
+++ b/vm/src/stdlib/os.rs
@@ -1823,6 +1823,43 @@ fn os_wait(vm: &VirtualMachine) -> PyResult<(libc::pid_t, i32)> {
     os_waitpid(-1, 0, vm)
 }
 
+fn os_kill(pid: i32, sig: isize, vm: &VirtualMachine) -> PyResult<()> {
+    #[cfg(unix)]
+    {
+        let ret = unsafe { libc::kill(pid, sig as i32) };
+        if ret == -1 {
+            Err(errno_err(vm))
+        } else {
+            Ok(())
+        }
+    }
+    #[cfg(windows)]
+    {
+        use winapi::um::{handleapi, processthreadsapi, wincon, winnt};
+        let sig = sig as u32;
+        let pid = pid as u32;
+
+        if sig == wincon::CTRL_C_EVENT || sig == wincon::CTRL_BREAK_EVENT {
+            let ret = unsafe { wincon::GenerateConsoleCtrlEvent(sig, pid) };
+            let res = if ret == 0 { Err(errno_err(vm)) } else { Ok(()) };
+            return res;
+        }
+
+        let h = unsafe { processthreadsapi::OpenProcess(winnt::PROCESS_ALL_ACCESS, 0, pid) };
+        if h.is_null() {
+            return Err(errno_err(vm));
+        }
+        let ret = unsafe { processthreadsapi::TerminateProcess(h, sig) };
+        let res = if ret == 0 { Err(errno_err(vm)) } else { Ok(()) };
+        unsafe { handleapi::CloseHandle(h) };
+        res
+    }
+    #[cfg(not(any(unix, windows)))]
+    {
+        unimplemented!()
+    }
+}
+
 pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
     let ctx = &vm.ctx;
 
@@ -1932,6 +1969,7 @@ pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
         "lseek" => ctx.new_function(os_lseek),
         "set_inheritable" => ctx.new_function(os_set_inheritable),
         "link" => ctx.new_function(os_link),
+        "kill" => ctx.new_function(os_kill),
 
         "O_RDONLY" => ctx.new_int(libc::O_RDONLY),
         "O_WRONLY" => ctx.new_int(libc::O_WRONLY),