2 files changed, 24 insertions, 0 deletions

diff --git a/cgit.h b/cgit.h
index a38981d..f402466 100644
--- a/cgit.h
+++ b/cgit.h
@@ -15,12 +15,23 @@
 #include <revision.h>
 #include <log-tree.h>
 #include <archive.h>
 #include <xdiff/xdiff.h>
 
 
+/*
+ * The valid cgit repo-commands
+ */
+#define CMD_LOG      = 1;
+#define CMD_COMMIT   = 1;
+#define CMD_DIFF     = 1;
+#define CMD_TREE     = 1;
+#define CMD_VIEW     = 1;
+#define CMD_BLOB     = 1;
+#define CMD_SNAPSHOT = 1;
+
 typedef void (*configfn)(const char *name, const char *value);
 typedef void (*filepair_fn)(struct diff_filepair *pair);
 typedef void (*linediff_fn)(char *line, int len);
 
 struct cacheitem {
         char *name;
@@ -109,12 +120,13 @@ extern char *cgit_query_sha2;
 extern char *cgit_query_path;
 extern char *cgit_query_name;
 extern int   cgit_query_ofs;
 
 extern int htmlfd;
 
+extern int cgit_get_cmd_index(const char *cmd);
 extern struct repoinfo *cgit_get_repoinfo(const char *url);
 extern void cgit_global_config_cb(const char *name, const char *value);
 extern void cgit_repo_config_cb(const char *name, const char *value);
 extern void cgit_querystring_cb(const char *name, const char *value);
 
 extern int chk_zero(int result, char *msg);
diff --git a/shared.c b/shared.c
index b164d81..fb05427 100644
--- a/shared.c
+++ b/shared.c
@@ -48,12 +48,24 @@ char *cgit_query_sha2   = NULL;
 char *cgit_query_path   = NULL;
 char *cgit_query_name   = NULL;
 int   cgit_query_ofs    = 0;
 
 int htmlfd = 0;
 
+
+int cgit_get_cmd_index(const char *cmd)
+{
+        static char *cmds[] = {"log", "commit", "diff", "tree", "view", "blob", "snapshot", NULL};
+        int i;
+
+        for(i = 0; cmds[i]; i++)
+                if (!strcmp(cmd, cmds[i]))
+                        return i + 1;
+        return 0;
+}
+
 int chk_zero(int result, char *msg)
 {
         if (result != 0)
                 die("%s: %s", msg, strerror(errno));
         return result;
 }