Make it actually compile...

[matches/MCTX3420.git] / server / main.c

diff --git a/server/main.c b/server/main.c
index c032eb1..ecdf068 100644 (file)
--- a/server/main.c
+++ b/server/main.c
@@ -9,6 +9,8 @@
 #include "sensor.h"
 #include "actuator.h"
 #include "control.h"
+#include "pin_test.h"
+#include "bbb_pin_defines.h"
 
 // --- Standard headers --- //
 #include <syslog.h> // for system logging
@@ -26,10 +28,72 @@ Options g_options; // options passed to program through command line arguments
  */
 void ParseArguments(int argc, char ** argv)
 {
+
+
        g_options.program = argv[0]; // program name
        g_options.verbosity = LOGDEBUG; // default log level
        gettimeofday(&(g_options.start_time), NULL); // Start time
-       Log(LOGDEBUG, "Called as %s with %d arguments.", g_options.program, argc);
+
+
+       g_options.auth_method = AUTH_NONE;  // Don't use authentication
+       g_options.auth_uri = ""; // 
+       g_options.ldap_base_dn = "";
+       
+
+
+       for (int i = 1; i < argc; ++i)
+       {
+               if (argv[i][0] != '-')
+                       Fatal("Unexpected argv[%d] - %s", i, argv[i]);
+
+               if (i+1 >= argc || argv[i+1][0] == '-')
+                       Fatal("No argument following switch %s", argv[i]);
+               
+               if (strlen(argv[i]) > 2)
+                       Fatal("Human readable switches are not supported.");
+
+               char * end = NULL;
+               switch (argv[i][1])
+               {
+                       // Set program verbosity
+                       case 'v':
+                               g_options.verbosity = strtol(argv[++i], &end, 10);
+                               break;
+                       // Enable/Disable pin test
+                       case 'p':
+                               g_options.enable_pin = !(strtol(argv[++i], &end, 10));
+                               break;
+                       // LDAP URI
+                       case 'A':
+                               g_options.auth_uri = argv[++i];
+                               break;
+                       // LDAP DN
+                       case 'd':
+                               g_options.ldap_base_dn = argv[++i];
+                               break;
+                       default:
+                               Fatal("Unrecognised switch %s", argv[i]);
+                               break;
+               }
+
+               if (end != NULL && *end != '\0')
+                       Fatal("argv[%d] -%c requires an integer (got \"%s\" instead)", i-1, argv[i-1][0], argv[i]);
+       }       
+
+       Log(LOGDEBUG, "Verbosity: %d", g_options.verbosity);
+       Log(LOGDEBUG, "Pin Module Enabled: %d", g_options.enable_pin);
+       Log(LOGDEBUG, "Auth URI: %s", g_options.auth_uri);
+       Log(LOGDEBUG, "LDAP Base DN: %s", g_options.ldap_base_dn);
+
+       if (g_options.auth_uri[0] != '\0')
+       {
+               //HACK...
+               if (PathExists(g_options.auth_uri))
+                       g_options.auth_method = AUTH_SHADOW;
+               else
+                       g_options.auth_method = AUTH_LDAP;
+       }
+       
 }
 
 /**
@@ -67,11 +131,14 @@ void Cleanup()
  */
 int main(int argc, char ** argv)
 {
+       // Open log before calling ParseArguments (since ParseArguments may call the Log functions)
+       openlog("mctxserv", LOG_PID | LOG_PERROR, LOG_USER);
+       Log(LOGINFO, "Server started");
+
        ParseArguments(argc, argv);
 
        //Open the system log
-       openlog("mctxserv", LOG_PID | LOG_PERROR, LOG_USER);
-       Log(LOGINFO, "Server started");
+
        // signal handler
        //TODO: Make this work
        /*
@@ -83,6 +150,7 @@ int main(int argc, char ** argv)
        */
        Sensor_Init();
        Actuator_Init();
+       Pin_Init();
        //Sensor_StartAll("test");
        //Actuator_StartAll("test");
        const char *ret;
@@ -97,6 +165,8 @@ int main(int argc, char ** argv)
        //Sensor_StopAll();
        //Actuator_StopAll();
 
+       Pin_Close();
+
        Cleanup();
        return 0;
 }