Add locking protocol name <--> id translation

2 files changed, 42 insertions, 0 deletions

diff --git a/include/litmus.h b/include/litmus.h
index ec548b1..2b6b374 100644
--- a/include/litmus.h
+++ b/include/litmus.h
@@ -58,6 +58,9 @@ typedef enum  {
         OMLP_SEM        = 5,
 } obj_type_t;
 
+int lock_protocol_for_name(const char* name);
+const char* name_for_lock_protocol(int id);
+
 int od_openx(int fd, obj_type_t type, int obj_id, void* config);
 int od_close(int od);
 
diff --git a/src/litmus.c b/src/litmus.c
index d478ffc..a1ce873 100644
--- a/src/litmus.c
+++ b/src/litmus.c
@@ -10,6 +10,45 @@
 #include "litmus.h"
 #include "internal.h"
 
+#define LP(name) {name ## _SEM, #name}
+
+static struct {
+        int id;
+        const char* name;
+} protocol[] = {
+        LP(FMLP),
+        LP(SRP),
+        LP(MPCP),
+        LP(MPCP_VS),
+        LP(DPCP),
+        LP(OMLP),
+};
+
+#define NUM_PROTOS (sizeof(protocol)/sizeof(protocol[0]))
+
+int lock_protocol_for_name(const char* name)
+{
+        int i;
+
+        for (i = 0; i < NUM_PROTOS; i++)
+                if (strcmp(name, protocol[i].name) == 0)
+                        return protocol[i].id;
+
+        return -1;
+}
+
+const char* name_for_lock_protocol(int id)
+{
+        int i;
+
+        for (i = 0; i < NUM_PROTOS; i++)
+                if (protocol[i].id == id)
+                        return protocol[i].name;
+
+        return "<UNKNOWN>";
+}
+
+
 void show_rt_param(struct rt_task* tp)
 {
         printf("rt params:\n\t"