X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=server%2Fcontrol.c;h=c6b582384f2f2fe120b7625ddf38cef6fd4d1240;hb=f486831f971d51f983a65b734ebd5af2b598d136;hp=5c8a4103feceb3672c79aa0cde7b1298973854f4;hpb=b7150f722995ea6807182da24653a15693a82698;p=matches%2FMCTX3420.git diff --git a/server/control.c b/server/control.c index 5c8a410..c6b5823 100644 --- a/server/control.c +++ b/server/control.c @@ -3,19 +3,23 @@ * @brief Handles all client control requests (admin related) */ #include "common.h" +#include "options.h" #include "control.h" #include "sensor.h" #include "actuator.h" +#include typedef struct ControlData { ControlModes current_mode; pthread_mutex_t mutex; struct timeval start_time; + char user_name[31]; // The user who owns the currently running experiment } ControlData; ControlData g_controls = {CONTROL_STOP, PTHREAD_MUTEX_INITIALIZER, {0}}; -static bool PathExists(const char *path) { +bool PathExists(const char *path) +{ FILE *fp = fopen(path, "r"); if (fp) { fclose(fp); @@ -31,64 +35,143 @@ static bool PathExists(const char *path) { * @param params The input parameters */ void Control_Handler(FCGIContext *context, char *params) { - const char *action, *key = "", *name = ""; + const char *action = ""; + const char *name = ""; bool force = false; ControlModes desired_mode; + + + + - FCGIValue values[4] = { + // 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[3] = { {"action", &action, FCGI_REQUIRED(FCGI_STRING_T)}, - {"key", &key, FCGI_STRING_T}, {"force", &force, FCGI_BOOL_T}, {"name", &name, FCGI_STRING_T} }; - if (!FCGI_ParseRequest(context, params, values, 4)) - return; - - if (!strcmp(action, "lock")) { - FCGI_LockControl(context, force); + + if (!FCGI_ParseRequest(context, params, values, 3)) return; - } else if (!strcmp(action, "emergency")) { - desired_mode = CONTROL_EMERGENCY; - } else if (FCGI_HasControl(context, key)) { - if (!strcmp(action, "release")) { - FCGI_ReleaseControl(context); - } else if (!strcmp(action, "start")) { - desired_mode = CONTROL_START; - } else if (!strcmp(action, "pause")) { - desired_mode = CONTROL_PAUSE; - } else if (!strcmp(action, "resume")) { - desired_mode = CONTROL_RESUME; - } else if (!strcmp(action, "stop")) { - desired_mode = CONTROL_STOP; - } else { - FCGI_RejectJSON(context, "Unknown action specified."); + + //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, "emergency")) { + desired_mode = CONTROL_EMERGENCY; + } else if (!strcmp(action, "start")) { + desired_mode = CONTROL_START; + } else if (!strcmp(action, "pause")) { + desired_mode = CONTROL_PAUSE; + } else if (!strcmp(action, "resume")) { + desired_mode = CONTROL_RESUME; + } else if (!strcmp(action, "stop")) { + desired_mode = CONTROL_STOP; } else { - FCGI_RejectJSONEx(context, STATUS_UNAUTHORIZED, - "Invalid control key specified."); + 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); } /** @@ -135,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,4 +260,4 @@ const char * Control_GetModeName() { */ const struct timeval* Control_GetStartTime() { return &g_controls.start_time; -} \ No newline at end of file +}