Allow interrupting memtrace report with Ctrl+C

Performance impact: Before: 948.1 ms ± 9.8 ms After: 960.0 ms ± 41.7 ms Closes #103
mephi42 · Aug 17, 2024 · cd2eede · cd2eede
1 parent 7f2b613
commit cd2eede
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/memtrace_ext/memtrace_ext.cc b/memtrace_ext/memtrace_ext.cc
@@ -757,6 +757,9 @@ class Trace : public TraceBase {
 
   template <typename V, typename F>
   int VisitOne(V* visitor, const F& filter) {
+    if (PyErr_CheckSignals()) {
+      boost::python::throw_error_already_set();
+    }
     if (!Have(Tlv<E, W>::kFixedLength)) return -EINVAL;
     Tlv<E, W> tlv(cur_);
     if (!Have(tlv.GetAlignedLength())) return -EINVAL;