1 files changed, 15 insertions, 0 deletions

diff --git a/include/litmus/rt_param.h b/include/litmus/rt_param.h
index a3dde39753bd..fa0f5f4647bd 100644
--- a/include/litmus/rt_param.h
+++ b/include/litmus/rt_param.h
@@ -51,6 +51,14 @@ typedef enum {
         TASK_EARLY
 } release_policy_t;
 
+typedef enum {
+        PGM_NOT_A_NODE,
+        PGM_SRC,
+        PGM_SINK,
+        PGM_SRC_SINK,
+        PGM_INTERNAL
+} pgm_node_type_t;
+
 /* We use the common priority interpretation "lower index == higher priority",
  * which is commonly used in fixed-priority schedulability analysis papers.
  * So, a numerically lower priority value implies higher scheduling priority,
@@ -122,6 +130,8 @@ struct rt_task {
         task_class_t    cls;
         budget_policy_t  budget_policy;  /* ignored by pfair */
         release_policy_t release_policy;
+        pgm_node_type_t  pgm_type;
+        lt_t             pgm_expected_etoe;
 };
 
 union np_flag {
@@ -162,6 +172,11 @@ struct control_page {
         uint64_t irq_syscall_start; /* Snapshot of irq_count when the syscall
                                      * started. */
 
+        /* Flags from userspace signifying PGM wait states. */
+        volatile uint32_t       pgm_waiting;    /* waiting for tokens */
+        volatile uint32_t       pgm_sending;    /* sending tokens */
+        volatile uint32_t       pgm_satisfied;  /* done waiting/sending */
+
         /* to be extended */
 };