Add combined PGM_SRC_SINK node type.

2 files changed, 7 insertions, 0 deletions

diff --git a/include/litmus/rt_param.h b/include/litmus/rt_param.h
index 0ddc1973b0aa..7b4045069b27 100644
--- a/include/litmus/rt_param.h
+++ b/include/litmus/rt_param.h
@@ -55,6 +55,7 @@ typedef enum {
         PGM_NOT_A_NODE,
         PGM_SRC,
         PGM_SINK,
+        PGM_SRC_SINK,
         PGM_INTERNAL
 } pgm_node_type_t;
 
diff --git a/litmus/litmus.c b/litmus/litmus.c
index 6f6479b10db0..058cb956fb44 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -144,6 +144,12 @@ asmlinkage long sys_set_rt_task_param(pid_t pid, struct rt_task __user * param)
                        pid, tp.budget_policy);
                 goto out_unlock;
         }
+        if (tp.pgm_type < PGM_NOT_A_NODE || tp.pgm_type > PGM_INTERNAL) {
+                printk(KERN_INFO "litmus: real-time task %d rejected "
+                           "because of unknown PGM node type specified (%d)\n",
+                           pid, tp.pgm_type);
+                goto out_unlock;
+        }
 
         target->rt_param.task_params = tp;