1 files changed, 15 insertions, 0 deletions

diff --git a/src/kernel_iface.c b/src/kernel_iface.c
index 33d56df..4ded334 100644
--- a/src/kernel_iface.c
+++ b/src/kernel_iface.c
@@ -75,6 +75,21 @@ void exit_np(void)
         }
 }
 
+int exit_np_trace(void)
+{
+        if (likely(ctrl_page != NULL) &&
+            ctrl_page->sched.np.flag &&
+            !(--ctrl_page->sched.np.flag)) {
+                __sync_synchronize();
+                if (ctrl_page->sched.np.preempt) {
+                        ctrl_page->ts_syscall_start = (uint64_t) get_cycles();
+                        sched_yield();
+                        return 1;
+                }
+        }
+        return 0;
+}
+
 int requested_to_preempt(void)
 {
         return (likely(ctrl_page != NULL) && ctrl_page->sched.np.preempt);