Add define for compiling under RTLinux
[matches/MCTX3420.git] / server / control.c
index 820c37c..776cdd1 100644 (file)
@@ -3,14 +3,17 @@
  * @brief Handles all client control requests (admin related)
  */
 #include "common.h"
+#include "options.h"
 #include "control.h"
 #include "sensor.h"
 #include "actuator.h"
+#include <dirent.h>
 
 typedef struct ControlData {
        ControlModes current_mode;
        pthread_mutex_t mutex;
-       struct timeval start_time;
+       struct timespec start_time;
+       char user_name[31]; // The user who owns the currently running experiment
 } ControlData;
 
 ControlData g_controls = {CONTROL_STOP, PTHREAD_MUTEX_INITIALIZER, {0}};
@@ -36,30 +39,65 @@ void Control_Handler(FCGIContext *context, char *params) {
        const char *name = "";
        bool force = false;
        ControlModes desired_mode;
-
+       
+       
+       
 
 
        // Login/auth now handled entirely in fastcgi.c and login.c
        //TODO: Need to not have the ability for any user to stop someone else' experiment...
        // (achieve by storing the username of the person running the current experiment, even when they log out?)
        // (Our program should only realisitically support a single experiment at a time, so that should be sufficient)
-       FCGIValue values[4] = {
+       FCGIValue values[3] = {
                {"action", &action, FCGI_REQUIRED(FCGI_STRING_T)},
                {"force", &force, FCGI_BOOL_T},
                {"name", &name, FCGI_STRING_T}
        };
 
-       if (!FCGI_ParseRequest(context, params, values, 4))
+
+       if (!FCGI_ParseRequest(context, params, values, 3))
                return;
+
+       //HACKETY HACK HACK (should really be a seperate function)
+       if (strcmp(action, "list") == 0)
+       {
+               DIR * dir = opendir(context->user_name);
+               if (dir == NULL)
+               {
+                       FCGI_RejectJSON(context, "Failed to open user directory");
+                       return;
+               }
+               struct dirent * ent;
+               FCGI_BeginJSON(context, STATUS_OK);
+               FCGI_JSONKey("experiments");
+               FCGI_PrintRaw("[");
+
+               bool first = true;
+               while ((ent = readdir(dir)) != NULL)
+               {
+                       char * c;
+                       for (c = ent->d_name; *c != '\0' && *c != '.'; ++c);
+
+                       if (*c != '\0' && strcmp(c, ".exp") == 0)
+                       {
+                               if (!first)
+                                       FCGI_PrintRaw(",");
+                               *c = '\0'; // Ummm... probably not a great idea
+                               FCGI_PrintRaw(ent->d_name);
+                               first = false;
+                       }
+               }
+               FCGI_PrintRaw("]");
+               FCGI_EndJSON();
+               
+               return; // Dear god this is terrible
+       }
+       //TODO: Need a "load" action to set data files (but not run) from a past experiment
+
+       //TODO: Need a "delete" action so that people can overwrite experiments (without all this "force" shenanigans)
        
-       if (!strcmp(action, "lock")) {
-               FCGI_LockControl(context, force);
-               return;
-       } else if (!strcmp(action, "emergency")) {
+       if (!strcmp(action, "emergency")) {
                desired_mode = CONTROL_EMERGENCY;
-       }
-       else if (!strcmp(action, "release")) {
-               FCGI_ReleaseControl(context);
        } else if (!strcmp(action, "start")) {
                desired_mode = CONTROL_START;
        } else if (!strcmp(action, "pause")) {
@@ -72,25 +110,68 @@ void Control_Handler(FCGIContext *context, char *params) {
                FCGI_RejectJSON(context, "Unknown action specified.");
                return;
        }
+
+       if (*g_controls.user_name != '\0' && strcmp(g_controls.user_name,context->user_name) != 0)
+       {
+               if (context->user_type != USER_ADMIN)
+               {
+                       FCGI_RejectJSON(context, "Another user has an experiment in progress.");
+                       return;
+               }
+               
+               if (!force)
+               {
+                       Log(LOGERR, "User %s is currently running an experiment!", g_controls.user_name);
+                       FCGI_RejectJSON(context, "Pass \"force\" to take control over another user's experiment");
+                       return;
+               }
+       }
+
+       
        
+       //HACK
+       chdir(context->user_name);
+
        void *arg = NULL;
        if (desired_mode == CONTROL_START) {
                if (PathExists(name) && !force) {
                        FCGI_RejectJSON(context, "An experiment with that name already exists.");
+                       chdir(g_options.root_dir); // REVERSE HACK
+                       return;
+               }
+               char * c = (char*)name;
+               for (c = (char*)name; *c != '\0' && *c != '.'; ++c);
+               if (*c == '.')
+               {
+                       FCGI_RejectJSON(context, "Can't include '.' characters in experiment names (at this point we don't care anymore, go find someone who does).");
+                       chdir(g_options.root_dir); // REVERSE HACK
                        return;
                }
-
                arg = (void*)name;
        }
 
+       
+
        const char *ret;
-       if ((ret = Control_SetMode(desired_mode, arg)) != NULL) {
+       if ((ret = Control_SetMode(desired_mode, arg)) != NULL) 
+       {
                FCGI_RejectJSON(context, ret);
-       } else {
+       } 
+       else 
+       {
+               if (desired_mode == CONTROL_STOP)
+                       g_controls.user_name[0] = '\0';
+               else
+               {
+                       snprintf(g_controls.user_name, sizeof(g_controls.user_name), context->user_name);
+               }
                FCGI_BeginJSON(context, STATUS_OK);
                FCGI_JSONPair("description", "ok");
                FCGI_EndJSON();
        }
+
+       //REVERSE HACK
+       chdir(g_options.root_dir);
 }
 
 /**
@@ -120,7 +201,7 @@ const char* Control_SetMode(ControlModes desired_mode, void * arg)
                                        FILE *fp = fopen((const char*) arg, "a");
                                        if (fp) {
                                                fclose(fp);
-                                               gettimeofday(&(g_controls.start_time), NULL);
+                                               clock_gettime(CLOCK_MONOTONIC, &(g_controls.start_time));
                                        } else
                                                ret = "Cannot open experiment name marker";
                                }
@@ -137,7 +218,7 @@ const char* Control_SetMode(ControlModes desired_mode, void * arg)
                break;
                case CONTROL_EMERGENCY:
                        if (g_controls.current_mode != CONTROL_START) //pfft
-                               ret = "Not running so how can there be an emergency.";
+                               ret = "Not running so how can there be an emergency?";
                break;
                default:
                break;
@@ -177,6 +258,6 @@ const char * Control_GetModeName() {
  * Gets the start time for the current experiment
  * @return the start time
  */
-const struct timeval* Control_GetStartTime() {
+const struct timespec * Control_GetStartTime() {
        return &g_controls.start_time;
 }

UCC git Repository :: git.ucc.asn.au