Merge branch 'master' of https://github.com/szmoore/MCTX3420.git
[matches/MCTX3420.git] / server / fastcgi.c
index 49bf197..e41e375 100644 (file)
@@ -1,6 +1,6 @@
 /**
  * @file fastcgi.c
- * @purpose Runs the FCGI request loop to handle web interface requests.
+ * @brief Runs the FCGI request loop to handle web interface requests.
  *
  * fcgi_stdio.h must be included before all else so the stdio function
  * redirection works ok.
@@ -8,20 +8,21 @@
 
 #include <fcgi_stdio.h>
 #include <openssl/sha.h>
-#include <time.h>
 
 #include "common.h"
 #include "sensor.h"
-#include "log.h"
+#include "control.h"
 #include "options.h"
 
-#define LOGIN_TIMEOUT 180
+/**The time period (in seconds) before the control key expires @ */
+#define CONTROL_TIMEOUT 180
 
+/**Contextual information related to FCGI requests*/
 struct FCGIContext {
-       /**The time of last valid logged-in user access*/
-       time_t login_timestamp;
-       char login_key[41];
-       char login_ip[16];
+       /**The time of last valid user access possessing the control key*/
+       time_t control_timestamp;
+       char control_key[41];
+       char control_ip[16];
        /**The name of the current module**/
        const char *current_module;
        /**For debugging purposes?**/
@@ -29,32 +30,68 @@ struct FCGIContext {
 };
 
 /**
- * Handles user logins.
+ * Identifies build information and the current API version to the user.
+ * Also useful for testing that the API is running and identifying the 
+ * sensors and actuators present.
  * @param context The context to work in
- * @param params User specified parameters
- */
-static void LoginHandler(FCGIContext *context, char *params) {
+ * @param params User specified paramters: [actuators, sensors]
+ */ 
+static void IdentifyHandler(FCGIContext *context, char *params) {
+       bool identSensors = false, identActuators = false;
        const char *key, *value;
-       bool force = 0, end = 0;
+       int i;
 
        while ((params = FCGI_KeyPair(params, &key, &value))) {
-               if (!strcmp(key, "force"))
-                       force = !force;
-               else if (!strcmp(key, "end"))
-                       end = !end;
+               if (!strcmp(key, "sensors")) {
+                       identSensors = !identSensors;
+               } else if (!strcmp(key, "actuators")) {
+                       identActuators = !identActuators;
+               }
        }
 
-       if (end) {
-               *(context->login_key) = 0;
-               FCGI_BeginJSON(context, STATUS_OK);
-               FCGI_EndJSON();
-               return;
+       FCGI_BeginJSON(context, STATUS_OK);
+       FCGI_JSONPair("description", "MCTX3420 Server API (2013)");
+       FCGI_JSONPair("build_date", __DATE__ " " __TIME__);
+       FCGI_JSONLong("api_version", API_VERSION);
+       if (identSensors) {
+               FCGI_JSONKey("sensors");
+               FCGI_JSONValue("{\n\t\t");
+               for (i = 0; i < NUMSENSORS; i++) {
+                       if (i > 0) {
+                               FCGI_JSONValue(",\n\t\t");
+                       }
+                       FCGI_JSONValue("\"%d\" : \"%s\"", i, g_sensor_names[i]); 
+               }
+               FCGI_JSONValue("\n\t}");
        }
+       if (identActuators) {
+               FCGI_JSONKey("actuators");
+               FCGI_JSONValue("{\n\t\t");
+               for (i = 0; i < NUMACTUATORS; i++) {
+                       if (i > 0) {
+                               FCGI_JSONValue(",\n\t\t");
+                       }
+                       FCGI_JSONValue("\"%d\" : \"%s\"", i, g_actuator_names[i]); 
+               }
+               FCGI_JSONValue("\n\t}");
+       }
+       FCGI_EndJSON();
+}
 
+/**
+ * Gives the user a key that determines who has control over
+ * the system at any one time. The key can be forcibly generated, revoking
+ * any previous control keys. To be used in conjunction with HTTP 
+ * basic authentication.
+ * This function will generate a JSON response that indicates success/failure.
+ * @param context The context to work in
+ * @param force Whether to force key generation or not.
+ */ 
+void FCGI_BeginControl(FCGIContext *context, bool force) {
        time_t now = time(NULL);
-       if (force || !*(context->login_key) || 
-          (now - context->login_timestamp > LOGIN_TIMEOUT)) 
-       {
+       bool expired = now - context->control_timestamp > CONTROL_TIMEOUT;
+       
+       if (force || !*(context->control_key) || expired) {
                SHA_CTX sha1ctx;
                unsigned char sha1[20];
                int i = rand();
@@ -64,62 +101,55 @@ static void LoginHandler(FCGIContext *context, char *params) {
                SHA1_Update(&sha1ctx, &i, sizeof(i));
                SHA1_Final(sha1, &sha1ctx);
 
-               context->login_timestamp = now;
+               context->control_timestamp = now;
                for (i = 0; i < 20; i++)
-                       sprintf(context->login_key + i * 2, "%02x", sha1[i]);
-               snprintf(context->login_ip, 16, "%s", getenv("REMOTE_ADDR"));
+                       sprintf(context->control_key + i * 2, "%02x", sha1[i]);
+               snprintf(context->control_ip, 16, "%s", getenv("REMOTE_ADDR"));
                FCGI_BeginJSON(context, STATUS_OK);
-               FCGI_JSONPair("key", context->login_key);
-               FCGI_EndJSON();
+               FCGI_JSONPair("key", context->control_key);
+               FCGI_EndJSON();         
        } else {
                char buf[128];
                strftime(buf, 128, "%H:%M:%S %d-%m-%Y",
-                       localtime(&(context->login_timestamp))); 
-               FCGI_BeginJSON(context, STATUS_UNAUTHORIZED);
-               FCGI_JSONPair("description", "Already logged in");
-               FCGI_JSONPair("user", context->login_ip); 
-               FCGI_JSONPair("time", buf);
-               FCGI_EndJSON();
-       }
-}
-
-/*TODO: Remove and replace with the actual actuator code*/
-static void ActuatorHandler(FCGIContext *context, char *params) {
-       const char *key, *value, *loginkey = NULL;
-       while ((params = FCGI_KeyPair(params, &key, &value))) {
-               if (!strcmp(key, "key")) {
-                       loginkey = value;
-               }
-       }
-       if (!loginkey || !FCGI_Authorized(context, loginkey)) {
+                       localtime(&(context->control_timestamp))); 
                FCGI_BeginJSON(context, STATUS_UNAUTHORIZED);
-               FCGI_JSONPair("description", "Invalid key specified.");
-               FCGI_EndJSON();
-       } else {
-               FCGI_BeginJSON(context, STATUS_OK);
-               FCGI_JSONPair("description", "Logged in!");
+               FCGI_JSONPair("description", "Another user already has control");
+               FCGI_JSONPair("current_user", context->control_ip); 
+               FCGI_JSONPair("when", buf);
                FCGI_EndJSON();
        }
 }
 
 /**
  * Given an FCGIContext, determines if the current user (as specified by
- * the key) is authorized or not. If validated, the context login_timestamp is
+ * the key) has control or not. If validated, the context control_timestamp is
  * updated.
  * @param context The context to work in
- * @param key The login key to be validated.
+ * @param key The control key to be validated.
  * @return TRUE if authorized, FALSE if not.
  */
-bool FCGI_Authorized(FCGIContext *context, const char *key) {
+bool FCGI_HasControl(FCGIContext *context, const char *key) {
        time_t now = time(NULL);
-       int result = (now - context->login_timestamp) <= LOGIN_TIMEOUT &&
-                                !strcmp(context->login_key, key);
+       int result = (now - context->control_timestamp) <= CONTROL_TIMEOUT &&
+                                key != NULL && !strcmp(context->control_key, key);
        if (result) {
-               context->login_timestamp = now; //Update the login_timestamp
+               context->control_timestamp = now; //Update the control_timestamp
        }
        return result;
 }
 
+
+/**
+ * Revokes the current control key, if present.
+ * @param context The context to work in
+ */
+void FCGI_EndControl(FCGIContext *context) {
+       *(context->control_key) = 0;
+       FCGI_BeginJSON(context, STATUS_OK);
+       FCGI_EndJSON();
+       return;
+}
+
 /**
  * Extracts a key/value pair from a request string.
  * Note that the input is modified by this function.
@@ -179,14 +209,13 @@ void FCGI_BeginJSON(FCGIContext *context, StatusCodes status_code)
        FCGI_JSONDouble("start_time", start_time);
        FCGI_JSONDouble("current_time", current_time);
        FCGI_JSONDouble("running_time", current_time - start_time);
-       
 }
 
 /**
  * Adds a key/value pair to a JSON response. The response must have already
  * been initiated by FCGI_BeginJSON. Note that characters are not escaped.
  * @param key The key of the JSON entry
- * &param value The value associated with the key.
+ * @param value The value associated with the key.
  */
 void FCGI_JSONPair(const char *key, const char *value)
 {
@@ -233,19 +262,6 @@ void FCGI_JSONKey(const char *key)
        printf(",\r\n\t\"%s\" : ", key);
 }
 
-/**
- * Should be used to write out the value of a JSON key. This has
- * the same format as the printf functions. Care should be taken to format
- * the output in valid JSON. 
- */
-void FCGI_JSONValue(const char *format, ...)
-{
-       va_list list;
-       va_start(list, format);
-       vprintf(format, list);
-       va_end(list);
-}
-
 /**
  * Ends a JSON response that was initiated by FCGI_BeginJSON.
  */
@@ -255,18 +271,20 @@ void FCGI_EndJSON()
 }
 
 /**
- * To be used when the input parameters are invalid.
- * Sends a response with HTTP status 400 Bad request, along with
- * JSON data for debugging.
+ * To be used when the input parameters are rejected. The return data
+ * will also have debugging information provided.
  * @param context The context to work in
- * @param params The parameters that the module handler received.
+ * @param status The status the return data should have.
+ * @param description A short description of why the input was rejected.
  */
-void FCGI_RejectJSON(FCGIContext *context)
+void FCGI_RejectJSONEx(FCGIContext *context, StatusCodes status, const char *description)
 {
-       printf("Status: 400 Bad Request\r\n");
+       if (description == NULL)
+               description = "Unknown";
        
-       FCGI_BeginJSON(context, STATUS_ERROR);
-       FCGI_JSONPair("description", "Invalid request");
+       Log(LOGINFO, "%s: Rejected query with: %d: %s", context->current_module, status, description);
+       FCGI_BeginJSON(context, status);
+       FCGI_JSONPair("description", description);
        FCGI_JSONLong("responsenumber", context->response_number);
        FCGI_JSONPair("params", getenv("QUERY_STRING"));
        FCGI_JSONPair("host", getenv("SERVER_HOSTNAME"));
@@ -275,6 +293,22 @@ void FCGI_RejectJSON(FCGIContext *context)
        FCGI_EndJSON();
 }
 
+/**
+ * Generates a response to the client as described by the format parameter and
+ * extra arguments (exactly like printf). To be used when none of the other
+ * predefined functions will work exactly as needed. Extra care should be taken
+ * to ensure the correctness of the output.
+ * @param format The format string
+ * @param ... Any extra arguments as required by the format string.
+ */
+void FCGI_PrintRaw(const char *format, ...)
+{
+       va_list list;
+       va_start(list, format);
+       vprintf(format, list);
+       va_end(list);
+}
+
 /**
  * Main FCGI request loop that receives/responds to client requests.
  * @param data Reserved.
@@ -311,22 +345,24 @@ void * FCGI_RequestLoop (void *data)
                size_t lastchar = strlen(module) - 1;
                if (lastchar > 0 && module[lastchar] == '/')
                        module[lastchar] = 0;
-               
 
-               if (!strcmp("login", module)) {
-                       module_handler = LoginHandler;
+               //Default to the 'identify' module if none specified
+               if (!*module) 
+                       strcpy(module, "identify");
+               
+               if (!strcmp("identify", module)) {
+                       module_handler = IdentifyHandler;
+               } else if (!strcmp("control", module)) {
+                       module_handler = Control_Handler;
                } else if (!strcmp("sensors", module)) {
-                       module_handler = Sensor_Handler;
-               } else if (!strcmp("actuators", module)) {
-                       module_handler = ActuatorHandler;
+                       module_handler = Sensor_Handler2;
                }
 
                context.current_module = module;
                if (module_handler) {
                        module_handler(&context, params);
                } else {
-                       strncat(module, " [unknown]", BUFSIZ);
-                       FCGI_RejectJSON(&context);
+                       FCGI_RejectJSON(&context, "Unhandled module");
                }
                context.response_number++;
 
@@ -337,6 +373,4 @@ void * FCGI_RequestLoop (void *data)
        Thread_QuitProgram(false);
        // NOTE: Don't call pthread_exit, because this runs in the main thread. Just return.
        return NULL;
-
-       
 }

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