X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=server%2Fmain.c;h=bd649a7fa7680544f22a64180980c39fd6d1bd3e;hb=f153e203bc4215026abd0dcce8427d633b5a6398;hp=7933bab7c3394b60f07746fdec199e143dc2ed57;hpb=f7e1e1e4b7c22ef34702cff9b01025612809aab8;p=matches%2FMCTX3420.git

diff --git a/server/main.c b/server/main.c
index 7933bab..bd649a7 100644
--- a/server/main.c
+++ b/server/main.c
@@ -17,6 +17,13 @@
 #include <signal.h> // for signal handling
 
 
+#ifdef REALTIME_VERSION
+#include <time.h>
+#include <sched.h>
+#include <sys/mman.h>
+#include <sys/utsname.h>
+#endif //REALTIME_VERSION
+
 // --- Variable definitions --- //
 Options g_options; // options passed to program through command line arguments
 
@@ -29,23 +36,20 @@ 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
 	// Set the main directory
-	if (getcwd(g_options.root_dir, sizeof(g_options.root_dir)) == NULL)
-		Fatal("Couldn't get current working directory - %s", strerror(errno));
+	//if (getcwd(g_options.root_dir, sizeof(g_options.root_dir)) == NULL)
+	//	Fatal("Couldn't get current working directory - %s", strerror(errno));
 
-	gettimeofday(&(g_options.start_time), NULL); // Start time
+	clock_gettime(CLOCK_MONOTONIC, &(g_options.start_time)); // Start time
 
 
 	g_options.auth_method = AUTH_NONE;  // Don't use authentication
 	g_options.auth_uri = ""; // 
 	g_options.ldap_base_dn = "";
+	g_options.experiment_dir = ".";
 	
-
-
 	for (int i = 1; i < argc; ++i)
 	{
 		if (argv[i][0] != '-')
@@ -76,6 +80,10 @@ void ParseArguments(int argc, char ** argv)
 			case 'd':
 				g_options.ldap_base_dn = argv[++i];
 				break;
+			case 'e':
+			// Experiments directory
+				g_options.experiment_dir = argv[++i];
+				break;
 			default:
 				Fatal("Unrecognised switch %s", argv[i]);
 				break;
@@ -89,7 +97,13 @@ void ParseArguments(int argc, char ** argv)
 	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);
-	Log(LOGDEBUG, "Root directory: %s", g_options.root_dir);
+	//Log(LOGDEBUG, "Root directory: %s", g_options.root_dir);
+	Log(LOGDEBUG, "Experiment directory: %s", g_options.experiment_dir);
+
+	if (!DirExists(g_options.experiment_dir))
+	{
+		Fatal("Experiment directory '%s' does not exist.", g_options.experiment_dir);
+	}
 
 	if (g_options.auth_uri[0] != '\0')
 	{
@@ -113,6 +127,38 @@ void Cleanup()
 	Log(LOGDEBUG, "Finish cleanup.");
 }
 
+
+#ifdef REALTIME_VERSION
+
+#define MAX_SAFE_STACK (8*1024)
+#define NSEC_PER_SEC (1000000000)
+
+void stack_prefault()
+{
+	unsigned char dummy[MAX_SAFE_STACK];
+	memset(dummy, 0, MAX_SAFE_STACK);
+	return;
+}
+
+bool is_realtime()
+{
+	struct utsname u;
+	bool crit1 = 0;
+	bool crit2 = 0;
+	FILE * f;
+	uname(&u);
+	crit1 = (strcasestr(u.version, "PREEMPT RT") != NULL);
+	if (crit1 && ((f = fopen("/sys/kernel/realtime", "r")) != NULL))
+	{
+		int flag;
+		crit2 = ((fscanf(f, "%d", &flag) == 1) && (flag == 1));
+		fclose(f);
+	}
+	return (crit1 && crit2);
+}
+
+#endif //REALTIME_VERSION
+
 /**
  * Main entry point; start worker threads, setup signal handling, wait for threads to exit, exit
  * @param argc - Num args
@@ -122,30 +168,56 @@ 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); // Setup the g_options structure from program arguments
 
+	Log(LOGINFO, "Server started");
+
+
+	
+	#ifdef REALTIME_VERSION
+	
+	if (is_realtime())
+	{
+		Log(LOGDEBUG, "Running under realtime kernel");
+	}
+	else
+	{
+		Fatal("Not running under realtime kernel");
+	}
+	struct sched_param param;
+	param.sched_priority = 49;
+	if (sched_setscheduler(0, SCHED_FIFO, &param) < 0)
+		Fatal("sched_setscheduler failed - %s", strerror(errno));
+	if (mlockall(MCL_CURRENT | MCL_FUTURE) == -1)
+		Fatal("mlockall failed - %s", strerror(errno));
+	stack_prefault();
+	#endif //REALTIME_VERSION
+
+	
+
 	Sensor_Init();
 	Actuator_Init();
 	Pin_Init();
 	
 	// Try and start things
-	/*
-	const char *ret;
-	if ((ret = Control_SetMode(CONTROL_START, "test")) != NULL)
-		Fatal("Control_SetMode failed with '%s'", ret);
-	*/
+	
+	//const char *ret;
+	//if ((ret = Control_SetMode(CONTROL_START, "test")) != NULL)
+	//	Fatal("Control_SetMode failed with '%s'", ret);
+	
 
 	// run request thread in the main thread
 	FCGI_RequestLoop(NULL);
 
-	/*
-	if ((ret = Control_SetMode(CONTROL_STOP, "test")) != NULL)
-		Fatal("Control_SetMode failed with '%s'", ret);
-	*/
+	
+	Conrol_SetMode(CONTROL_STOP, NULL);
+	//if ((ret = Control_SetMode(CONTROL_STOP, "test")) != NULL)
+	//	Fatal("Control_SetMode failed with '%s'", ret);
+	
 	//Sensor_StopAll();
 	//Actuator_StopAll();