4 files changed, 51 insertions, 0 deletions

diff --git a/include/color.h b/include/color.h
new file mode 100644
index 0000000..dc1b1cc
--- /dev/null
+++ b/include/color.h
@@ -0,0 +1,9 @@
+#ifndef COLOR_H
+#define COLOR_H
+
+#include <stddef.h> /* for size_t */
+
+int map_color_ctrl(void **);
+void* color_malloc(size_t);
+
+#endif
diff --git a/include/internal.h b/include/internal.h
index 07253b7..f30ce61 100644
--- a/include/internal.h
+++ b/include/internal.h
@@ -18,8 +18,13 @@ int __launch_rt_task(rt_fn_t rt_prog, void *rt_arg,
         }
 
 
+/* taken from the kernel */
+
 #define likely(x)   __builtin_expect((x), 1)
 #define unlikely(x) __builtin_expect((x), 0)
 
+#define BUILD_BUG_ON_ZERO(e) (sizeof(struct { int:-!!(e); }))
+#define BUILD_BUG_ON(condition) ((void)BUILD_BUG_ON_ZERO(condition))
+
 #endif
 
diff --git a/include/litmus.h b/include/litmus.h
index 045579e..82e2ba2 100644
--- a/include/litmus.h
+++ b/include/litmus.h
@@ -99,6 +99,7 @@ task_class_t str2class(const char* str);
 /* non-preemptive section support */
 void enter_np(void);
 void exit_np(void);
+int  requested_to_preempt(void);
 
 /* task system support */
 int wait_for_ts_release(void);
diff --git a/include/perfcounters.h b/include/perfcounters.h
new file mode 100644
index 0000000..03f94fb
--- /dev/null
+++ b/include/perfcounters.h
@@ -0,0 +1,36 @@
+#ifndef PERFCOUNTERS_H
+#define PERFCOUNTERS_H
+
+#include <stdint.h>
+
+#include "../../litmus-rt/include/linux/perf_event.h"
+
+#define NR_PERF_COUNTERS        4
+
+/*
+ * Retain this information with a performance counter file descriptor.
+ */
+struct perf_counter {
+        int                     fd;
+        enum perf_type_id       type;
+        uint64_t                config;
+};
+
+
+/*
+ * Initialize a set of counters for a CPU.
+ *
+ * This is NOT thread safe!
+ *
+ * @cpu                 CPU
+ * @group_leader        group leader PID or -1 make a new group
+ * @perf_counter        array
+ * @return              0 or error
+ */
+int setup_cpu_perf(const int, const int, struct perf_counter*);
+
+const char* get_perf_name(const struct perf_counter*);
+
+int read_perf_counter(const struct perf_counter*, uint64_t*);
+
+#endif