* @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;
+ 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);
* @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 (!strcmp(action, "query")) {
+
+ //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_JSONPair("state", Control_GetModeName(Control_GetMode()));
+ 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_RejectJSON(context, "Unknown action specified.");
return;
- } 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.");
+ }
+
+ 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;
}
- } else {
- FCGI_RejectJSONEx(context, STATUS_UNAUTHORIZED,
- "Invalid control key specified.");
- return;
}
+
+
+ //HACK
+ chdir(context->user_name);
+
void *arg = NULL;
if (desired_mode == CONTROL_START) {
- int len = strlen(name);
- if (len <= 0) {
- FCGI_RejectJSON(context, "An experiment name must be specified.");
- return;
- } else if (PathExists(name) && !force) {
+ 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);
}
/**
const char *ret = NULL;
pthread_mutex_lock(&(g_controls.mutex));
- if (g_controls.current_mode == CONTROL_EMERGENCY && desired_mode != CONTROL_STOP) {
- ret = "In emergency mode. Stop before doing anything else.";
- } else if (g_controls.current_mode == desired_mode) {
- ret = "Already in desired mode.";
- } else if (desired_mode == CONTROL_START) {
- if (g_controls.current_mode == CONTROL_STOP) {
- //TODO Sanitise name (ensure it contains no special chars eg \ / .. .
- FILE *fp = fopen((const char*) arg, "a");
- if (fp) {
- fclose(fp);
- gettimeofday(&(g_controls.start_time), NULL);
- } else {
- ret = "Cannot open experiment name marker";
- }
- } else {
- ret = "Cannot start when not in a stopped state.";
- }
- } else if (desired_mode == CONTROL_RESUME) {
- if (g_controls.current_mode != CONTROL_PAUSE)
- ret = "Cannot resume when not in a paused state.";
+ if (g_controls.current_mode == desired_mode)
+ ret = "Already in the desired mode.";
+ else if (g_controls.current_mode == CONTROL_EMERGENCY && desired_mode != CONTROL_STOP)
+ ret = "In emergency mode. You must stop before continuing.";
+ else switch (desired_mode) {
+ case CONTROL_START:
+ if (g_controls.current_mode == CONTROL_STOP) {
+ const char * name = arg;
+ if (!*name)
+ ret = "An experiment name must be specified";
+ else if (strpbrk(name, INVALID_CHARACTERS))
+ ret = "The experiment name must not contain: " INVALID_CHARACTERS_JSON;
+ else {
+ FILE *fp = fopen((const char*) arg, "a");
+ if (fp) {
+ fclose(fp);
+ gettimeofday(&(g_controls.start_time), NULL);
+ } else
+ ret = "Cannot open experiment name marker";
+ }
+ } else
+ ret = "Cannot start when not in a stopped state.";
+ break;
+ case CONTROL_PAUSE:
+ if (g_controls.current_mode != CONTROL_START)
+ ret = "Cannot pause when not in a running state.";
+ break;
+ case CONTROL_RESUME:
+ if (g_controls.current_mode != CONTROL_PAUSE)
+ ret = "Cannot resume when not in a paused state.";
+ break;
+ case CONTROL_EMERGENCY:
+ if (g_controls.current_mode != CONTROL_START) //pfft
+ ret = "Not running so how can there be an emergency?";
+ break;
+ default:
+ break;
}
if (ret == NULL) {
}
/**
- * Gets the current mode.
- * @return The current mode
- */
-ControlModes Control_GetMode() {
- return g_controls.current_mode;
-}
-
-/**
- * Gets a string representation of a mode
+ * Gets a string representation of the current mode
* @param mode The mode to get a string representation of
* @return The string representation of the mode
*/
-const char * Control_GetModeName(ControlModes mode) {
+const char * Control_GetModeName() {
const char * ret = "Unknown";
- switch (mode) {
+ switch (g_controls.current_mode) {
case CONTROL_START: ret = "Running"; break;
case CONTROL_PAUSE: ret = "Paused"; break;
case CONTROL_RESUME: ret = "Resumed"; break;
return ret;
}
-/*
-bool Control_Lock() {
- pthread_mutex_lock(&(g_controls.mutex));
- if (g_controls.state == STATE_RUNNING || g_controls.state == STATE_PAUSED)
- return true;
- pthread_mutex_unlock(&(g_controls.mutex));
- return false;
-}
-
-void Control_Unlock() {
- pthread_mutex_unlock(&(g_controls.mutex));
-}*/
-
/**
* Gets the start time for the current experiment
* @return the start time
*/
const struct timeval* Control_GetStartTime() {
return &g_controls.start_time;
-}
\ No newline at end of file
+}