Add enter_np/exit_np callbacks for non-preemptable sections support

1 files changed, 10 insertions, 0 deletions

diff --git a/include/linux/sched_plugin.h b/include/linux/sched_plugin.h
index 5c223695ab..3926388a0f 100644
--- a/include/linux/sched_plugin.h
+++ b/include/linux/sched_plugin.h
@@ -56,6 +56,12 @@ typedef void (*task_blocks_t)  (struct task_struct *task);
 /* called when a real-time task exits. Free any allocated resources */
 typedef long (*tear_down_t)    (struct task_struct *);
 
+/* called when a real-time task wants to enter a non-preemptable section */
+typedef long (*enter_np_t) (struct task_struct *);
+/* called when a real-time task wants to leave a non-preemptable section */
+typedef long (*exit_np_t) (struct task_struct *);
+
+
 
 /********************* sys call backends  ********************/
 /* This function causes the caller to sleep until the next release */
@@ -89,6 +95,10 @@ struct sched_plugin {
         wake_up_task_t          wake_up_task;
         task_blocks_t           task_blocks;
         tear_down_t             tear_down;
+
+        /*   non-preemptable sections   */
+        enter_np_t              enter_np;
+        exit_np_t               exit_np;
 } __attribute__ ((__aligned__(SMP_CACHE_BYTES)));
 
 typedef struct sched_plugin sched_plugin_t;