feat: Init _thread.interrupt_main

Karatuss · Karatuss · commit 57714d0345a6 · 2022-08-16T20:43:06.000+09:00
diff --git a/vm/src/stdlib/signal.rs b/vm/src/stdlib/signal.rs
@@ -279,4 +279,20 @@ pub(crate) mod _signal {
             Err(vm.new_value_error("signal number out of range".to_owned()))
         }
     }
+
+    /// Similar to `PyErr_SetInterruptEx` in CPython
+    ///
+    /// Missing signal handler for the given signal number is silently ignored.
+    pub fn set_interrupt_ex(signum: i32, vm: &VirtualMachine) -> PyResult<()> {
+        assert_in_range(signum, vm)?;
+
+        match signum as usize {
+            SIG_DFL | SIG_IGN => Ok(()),
+            _ => {
+                // interrupt the main thread with given signal number
+                run_signal(signum);
+                Ok(())
+            }
+        }
+    }
 }
diff --git a/vm/src/stdlib/thread.rs b/vm/src/stdlib/thread.rs
@@ -304,6 +304,11 @@ pub(crate) mod _thread {
         vm.state.thread_count.fetch_sub(1);
     }
 
+    #[pyfunction]
+    fn interrupt_main(signum: OptionalArg<i32>, vm: &VirtualMachine) -> PyResult<()> {
+        crate::stdlib::signal::_signal::set_interrupt_ex(signum.unwrap_or(libc::SIGINT), vm)
+    }
+
     #[pyfunction]
     fn exit(vm: &VirtualMachine) -> PyResult {
         Err(vm.new_exception_empty(vm.ctx.exceptions.system_exit.to_owned()))
