X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=server%2Fcontrol.c;h=29d9ef6ced8e1dede815729b936ae84740204c7d;hb=9e7cfbbc6137056bba8ed8644fc6ecfe398553fe;hp=4da1f7507718de6c90e7517efa1b9542a06a6f10;hpb=18027c718bc45eaa191068da1e739eebeb6111e4;p=matches%2FMCTX3420.git diff --git a/server/control.c b/server/control.c index 4da1f75..29d9ef6 100644 --- a/server/control.c +++ b/server/control.c @@ -1,107 +1,199 @@ /** * @file control.c - * @brief Handles all client control requests (admin/actuator related) + * @brief Handles all client control requests (admin related) */ #include "common.h" #include "control.h" +#include "sensor.h" +#include "actuator.h" -const char * g_actuator_names[NUMACTUATORS] = { - "Pressure regulator", "Solenoid 1" -}; +typedef enum ControlState { + STATE_STOPPED, + STATE_PAUSED, + STATE_RUNNING +} ControlState; + +typedef enum Mode { + START, + PAUSE, + RESUME, + STOP +} Mode; + +typedef struct ControlData { + ControlState state; + pthread_mutex_t mutex; + struct timeval start_time; +} ControlData; + +ControlData g_controls = {STATE_STOPPED, PTHREAD_MUTEX_INITIALIZER, {0}}; + +static bool ExperimentExists(const char *experiment_name) { + FILE *fp = fopen(experiment_name, "r"); + if (!fp) + return false; + fclose(fp); + return true; +} /** - * Handles control of the actuators. + * System control handler. This covers high-level control, including + * admin related functions and starting/stopping experiments.. + * @param context The context to work in + * @param params The input parameters */ -void ActuatorHandler(FCGIContext *context, ActuatorId id, const char *set_value) { - char *ptr; +void Control_Handler(FCGIContext *context, char *params) { + const char *action, *key = "", *name = ""; + bool force = false; + Mode mode; + + FCGIValue values[4] = { + {"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; - switch(id) { //Add new actuators here - case ACT_PRESSURE: //Suppose is pressure regulator. 0-700 input (kPa) - { - int value = strtol(set_value, &ptr, 10); - if (*ptr == '\0' && value >= 0 && value <= 700) { + if (!strcmp(action, "lock")) { + FCGI_LockControl(context, force); + return; + } else if (FCGI_HasControl(context, key)) { + if (!strcmp(action, "release")) { + FCGI_ReleaseControl(context); + } else if (!strcmp(action, "start")) { + mode = START; + } else if (!strcmp(action, "pause")) { + mode = PAUSE; + } else if (!strcmp(action, "resume")) { + mode = RESUME; + } else if (!strcmp(action, "stop")) { + mode = STOP; + } else { + FCGI_RejectJSON(context, "Unknown action specified."); + return; + } + } else { + FCGI_RejectJSONEx(context, STATUS_UNAUTHORIZED, + "Invalid control key specified."); + return; + } + + switch(mode) { + case START: + if (!*name) { + FCGI_RejectJSON(context, "An experiment name must be provided"); + } else if (ExperimentExists(name) && !force) { + FCGI_RejectJSONEx(context, STATUS_ALREADYEXISTS, + "An experiment with the specified name already exists."); + } else if (!Control_Start(name)) { + FCGI_RejectJSON(context, "An experiment is already running."); + } else { FCGI_BeginJSON(context, STATUS_OK); - FCGI_JSONKey("description"); - FCGI_JSONValue("\"Set pressure to %d kPa!\"", value); FCGI_EndJSON(); - } else { - FCGI_RejectJSONEx(context, - STATUS_ERROR, "Invalid pressure specified."); } - } break; - case ACT_SOLENOID1: - { - int value = strtol(set_value, &ptr, 10); - if (*ptr == '\0') { - const char *state = "off"; - if (value) - state = "on"; + break; + case PAUSE: + if (!Control_Pause()) { + FCGI_RejectJSON(context, "No experiment to pause."); + } else { FCGI_BeginJSON(context, STATUS_OK); - FCGI_JSONKey("description"); - FCGI_JSONValue("\"Solenoid 1 turned %s!\"", state); FCGI_EndJSON(); + } + break; + case RESUME: + if (!Control_Resume()) { + FCGI_RejectJSON(context, "No experiment to resume."); } else { - FCGI_RejectJSON(context, "Invalid actuator value specified"); + FCGI_BeginJSON(context, STATUS_OK); + FCGI_EndJSON(); } - } break; - default: - FCGI_RejectJSONEx(context, - STATUS_ERROR, "Invalid actuator id specified."); + break; + case STOP: + if (!Control_Stop()) { + FCGI_RejectJSON(context, "No experiment to stop."); + } else { + FCGI_BeginJSON(context, STATUS_OK); + FCGI_EndJSON(); + } + break; } } -/** - * System control handler. This covers control over all aspects of the system. - * E.g: Actuators, system commands (start/stop experiment/recording) etc - * @param context The context to work in - * @param params The input parameters - */ -void Control_Handler(FCGIContext *context, char *params) { - const char *key, *value, *control_key = NULL; - const char *action = NULL, *set_value = NULL; - bool force = false; - char *ptr; - int id = -1; - - while ((params = FCGI_KeyPair(params, &key, &value))) { - if (!strcmp(key, "action")) - action = value; - else if (!strcmp(key, "key")) - control_key = value; - else if (!strcmp(key, "force")) - force = !force; - else if (!strcmp(key, "id") && *value) { //Ensure non-empty value - int parsed = strtol(value, &ptr, 10); - if (*ptr == '\0') { - id = parsed; - } - } else if (!strcmp(key, "value")) { - set_value = value; +bool Control_Start(const char *experiment_name) { + bool ret = false; + + pthread_mutex_lock(&(g_controls.mutex)); + if (g_controls.state == STATE_STOPPED) { + FILE *fp = fopen(experiment_name, "a"); + if (fp) { + fclose(fp); + gettimeofday(&(g_controls.start_time), NULL); + Sensor_StartAll(experiment_name); + Actuator_StartAll(experiment_name); + g_controls.state = STATE_RUNNING; + ret = true; } } - - if (action == NULL) { //Must have an action - FCGI_RejectJSON(context, "No action specified"); - } else if (!strcmp(action, "start")) { - FCGI_BeginControl(context, force); - } else if (!strcmp(action, "stop")) { //Don't require control key to stop... - //EMERGENCY STOP!! TODO - replace! - FCGI_BeginJSON(context, STATUS_OK); - FCGI_JSONPair("description", "stopped! (not)"); - FCGI_EndJSON(); - } else { //Under this section, the user must have the current control key. - if (!FCGI_HasControl(context, control_key)) { - FCGI_RejectJSONEx(context, - STATUS_UNAUTHORIZED, "Invalid control key specified."); - } else if (!strcmp(action, "end")) { - FCGI_EndControl(context); - } else if (!strcmp(action, "set")) { - if (set_value == NULL || *set_value == '\0') { - FCGI_RejectJSONEx(context, - STATUS_ERROR, "Set called but no value specified."); - } else { - ActuatorHandler(context, id, set_value); - } - } + pthread_mutex_unlock(&(g_controls.mutex)); + return ret; +} + + +bool Control_Pause() { + bool ret = false; + pthread_mutex_lock(&(g_controls.mutex)); + if (g_controls.state == STATE_RUNNING) { + Actuator_PauseAll(); + Sensor_PauseAll(); + g_controls.state = STATE_PAUSED; + ret = true; } + pthread_mutex_unlock(&(g_controls.mutex)); + return ret; } + +bool Control_Resume() { + bool ret = false; + pthread_mutex_lock(&(g_controls.mutex)); + if (g_controls.state == STATE_PAUSED) { + Actuator_ResumeAll(); + Sensor_ResumeAll(); + g_controls.state = STATE_RUNNING; + ret = true; + } + pthread_mutex_unlock(&(g_controls.mutex)); + return ret; +} + +bool Control_Stop() { + bool ret = false; + + pthread_mutex_lock(&(g_controls.mutex)); + if (g_controls.state != STATE_STOPPED) { + Actuator_StopAll(); + Sensor_StopAll(); + g_controls.state = STATE_STOPPED; + ret = true; + } + pthread_mutex_unlock(&(g_controls.mutex)); + 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)); +} + +const struct timeval* Control_GetStartTime() { + return &g_controls.start_time; +} \ No newline at end of file