- added number of working threads

Source/charon/daemon.c

@@ -21,28 +21,321 @@
  */
  
 #include <stdio.h>
- 
+#include <signal.h>
+#include <pthread.h>
+
+#include "daemon.h" 
+
 
 #include "types.h"
-#include "utils/tester.h"
+#include "socket.h"
+#include "ike_sa_manager.h"
+#include "sender.h"
+#include "receiver.h"
+#include "scheduler.h"
+#include "thread_pool.h"
+#include "utils/allocator.h"
+#include "utils/logger_manager.h"
+#include "queues/event_queue.h"
 #include "queues/job_queue.h"
+#include "queues/send_queue.h"
 
 
-job_queue_t *job_queue;
+/* function declaration (defined and described after main function) */
+
+static status_t initialize_globals();
+static void destroy_globals();
+static status_t start_threads();
+static void end_threads();
+static void main_loop();
+static void register_signals();
+static void destroy_and_exit(int);
+
+/** Global job-queue */
+job_queue_t *global_job_queue = NULL;
+/** Global event-queue */
+event_queue_t *global_event_queue = NULL;
+ /** Global send-queue */
+send_queue_t *global_send_queue = NULL;
+ /** Global socket */
+socket_t *global_socket = NULL;
+/** Global logger manager */
+logger_manager_t *global_logger_manager = NULL;
+/** Global ike_sa-manager */
+ike_sa_manager_t *global_ike_sa_manager = NULL;
+
+/**
+ * logger_t object assigned for daemon things
+ */
+static logger_t *logger = NULL;
+
+/**
+ * Sender-Thread
+ */
+static sender_t *sender_thread = NULL;
+/**
+ * Receiver-Thread
+ */
+static receiver_t *receiver_thread = NULL;
+/**
+ * Scheduler-Thread
+ */
+static scheduler_t *scheduler_thread = NULL;
+/**
+ * Thread pool holding the worker threads
+ */
+static thread_pool_t *thread_pool = NULL;
+
+/**
+ * Signal set used for signal handling
+ */
+sigset_t signal_set;
+
 
- 
 int main()
 {
- 	
- 	job_queue = job_queue_create();
- 	
-	job_queue->destroy(job_queue);
+	/* set signal handler */
+	register_signals();
 	
+	/* initialize all global objects */
+ 	if (initialize_globals() != SUCCESS)
+ 	{
+ 		destroy_globals();
+ 		return -1;
+ 	}
+ 	/* a own logger for the daemon is created */
+ 	logger = global_logger_manager->create_logger(global_logger_manager,DAEMON,NULL);
+ 	if (logger == NULL)
+ 	{
+		printf("could not create logger object");
+ 		destroy_globals();
+ 		return -1;
+ 	}
+ 	
+ 	logger->log(logger,CONTROL,"start daemon %s", DAEMON_NAME); 	
+ 	/* now  its time to create all the different threads :-) */ 
+	if (start_threads() != SUCCESS)
+	{
+		/* ugh, not good */
+	 	logger->log(logger,CONTROL,"Fatal error: Needed Threads could not be started");		
+	 	destroy_and_exit(-1);
+	}
+ 	
+ 	logger->log(logger,CONTROL_MORE,"going to wait for exit signal");
+ 	/* go and handle signals*/
+ 	main_loop();
+ 	
+ 	destroy_and_exit(0);
+ 	
+ 	/* never reached */
+ 	return -1;
+}
+
+/**
+ * Main Loop.
+ * Waits for registered signals and acts dependently
+ */
+static void main_loop()
+{
+	while(1)
+	{
+	   int signal_number;
+       int error;
+
+       error = sigwait(&signal_set, &signal_number);
+
+       if(error)
+       {
+              /* do error code */
+			  logger->log(logger,CONTROL,"Error %d when waiting for signal",error);
+              return;
+       }
+       switch (signal_number)
+       {
+			case SIGHUP:
+			{
+				logger->log(logger,CONTROL,"Signal of type SIGHUP received. Do nothing");
+				break;
+			}
+			case SIGINT:
+			{
+				logger->log(logger,CONTROL,"Signal of type SIGINT received. Exit main loop.");
+				return;
+			}
+			case SIGTERM:
+			{
+				logger->log(logger,CONTROL,"Signal of type SIGTERM received. Exit main loop.");			
+				return;
+			}
+			default:
+			{
+				logger->log(logger,CONTROL,"Unknown signal %d received. Do nothing",signal_number);
+				break;
+			}
+		}
+	}
+}
+
+/**
+ * Registers signals SIGINT, SIGHUP and SIGTERM.
+ * Signals are handled in main_loop()
+ */
+static void register_signals()
+{
+    sigemptyset(&signal_set);
+    sigaddset(&signal_set, SIGINT); 
+    sigaddset(&signal_set, SIGHUP); 
+    sigaddset(&signal_set, SIGTERM); 
+    pthread_sigmask(SIG_BLOCK, &signal_set, 0);
+
+}
+
+/**
+ * Initializes global objects
+ * 
+ * @return
+ * 			- SUCCESS
+ * 			- FAILED
+ */
+static status_t initialize_globals()
+{
+ 	/* initialize global object */
+ 	global_logger_manager = logger_manager_create(ALL);
+	global_socket = socket_create(4600);
+ 	global_job_queue = job_queue_create();
+ 	global_event_queue = event_queue_create();
+ 	global_send_queue = send_queue_create();
+ 	global_ike_sa_manager = ike_sa_manager_create();
+ 	
+ 	if ((global_logger_manager == NULL) ||
+		(global_socket == NULL) ||
+		(global_job_queue == NULL) ||
+		(global_event_queue == NULL) ||
+		(global_send_queue == NULL) ||
+		(global_ike_sa_manager == NULL))
+ 	{
+	 	return FAILED;
+ 	}
+ 	
+ 	return SUCCESS;
+}
+
+/**
+ * Destroy global objects
+ */
+static void destroy_globals()
+{
+	if (global_ike_sa_manager != NULL)
+	{
+		global_job_queue->destroy(global_job_queue);
+	}
+	if (global_event_queue != NULL)
+	{
+		global_event_queue->destroy(global_event_queue);	
+	}
+	if (global_send_queue != NULL)
+	{
+		global_send_queue->destroy(global_send_queue);	
+	}
+	if (global_socket != NULL)
+	{
+		global_socket->destroy(global_socket);
+	}
+	if (global_logger_manager != NULL)
+	{
+		global_logger_manager->destroy(global_logger_manager);
+	}
+	if (global_ike_sa_manager != NULL)
+	{
+		global_ike_sa_manager->destroy(global_ike_sa_manager);
+	}
+}
+
+/**
+ * Creates all needed Threads
+ * 
+ * @return
+ * 			- SUCCESS
+ * 			- FAILED
+ */
+static status_t start_threads()
+{
+	sender_thread = sender_create();
+	if (sender_thread == NULL)
+	{
+		return FAILED;
+	}
+	scheduler_thread = scheduler_create();
+	if (scheduler_thread == NULL)
+	{
+		return FAILED;
+	}
+	thread_pool = thread_pool_create(NUMBER_OF_WORKING_THREADS);	
+	if (thread_pool == NULL)
+	{
+		return FAILED;
+	}
+	receiver_thread = receiver_create();
+	if (receiver_thread == NULL)
+	{
+		return FAILED;
+	}	
+
+	return SUCCESS;
+}
+
+
+/**
+ * Ends all Threads
+ * 
+ */
+static void end_threads()
+{
+ 	logger->log(logger,CONTROL_MORE,"going to stop receiver thread");
+	if (receiver_thread != NULL)
+	{
+		receiver_thread->destroy(receiver_thread);
+	}
+ 	logger->log(logger,CONTROL_MORE,"going to stop scheduler thread");
+	if (scheduler_thread != NULL)
+	{
+		scheduler_thread->destroy(scheduler_thread);	
+	}
+ 	logger->log(logger,CONTROL_MORE,"going to stop sender thread");
+	if (sender_thread != NULL)
+	{
+		sender_thread->destroy(sender_thread);
+	}
+ 	logger->log(logger,CONTROL_MORE,"going to stop all worker threads");
+	if (thread_pool != NULL)
+	{
+		thread_pool->destroy(thread_pool);	
+	}
+
+}
+
+/**
+ * Destroys initialized objects, kills all threads and exits
+ * 
+ * @param exit_code Code to exit with
+ */
+static void destroy_and_exit(int exit_code)
+{
+ 	logger->log(logger,CONTROL,"going to exit daemon"); 	
+
+	end_threads();
+
+	/* logger is destroyed */
+ 	logger->log(logger,CONTROL_MORE,"destroy logger and afterwards all global objects -> No more log from here");
+	global_logger_manager->destroy_logger(global_logger_manager,logger);
+	
+	/* all globals can be destroyed now */
+	destroy_globals();
+
 #ifdef LEAK_DETECTIVE
 	/* Leaks are reported in log file */
-	report_leaks();
+	report_memory_leaks(void);
 #endif
-	
-	return 0;
+
+	exit(exit_code);	
 }
- 

Source/charon/daemon.h

@@ -23,6 +23,11 @@
 #ifndef DAEMON_H_
 #define DAEMON_H_
 
-#define DEAMON_NAME "charon"
+#include "globals.h"
+
+#define DAEMON_NAME "charon"
+
+#define NUMBER_OF_WORKING_THREADS 4
+
 
 #endif /*DAEMON_H_*/

Source/charon/globals.h

@@ -28,12 +28,14 @@
 #include "queues/job_queue.h"
 #include "queues/event_queue.h"
 #include "utils/logger_manager.h"
+#include "ike_sa_manager.h"
 
 
+extern socket_t *global_socket;
 extern send_queue_t *global_send_queue;
 extern job_queue_t *global_job_queue;
 extern event_queue_t *global_event_queue;
-extern socket_t *global_socket;
 extern logger_manager_t *global_logger_manager;
+extern ike_sa_manager_t *global_ike_sa_manager;
 
 #endif /*GLOBALS_H_*/

Source/charon/ike_sa_manager.c

@@ -25,14 +25,13 @@
 
 #include "ike_sa_manager.h"
 
+#include "globals.h"
 #include "ike_sa_id.h"
 #include "utils/allocator.h"
 #include "utils/logger.h"
 #include "utils/logger_manager.h"
 #include "utils/linked_list.h"
 
-extern logger_manager_t *global_logger_manager;
-
 /**
  * @brief An entry in the linked list, contains IKE_SA, locking and lookup data.
  */

Source/charon/receiver.c

@@ -25,11 +25,11 @@
 
 #include "receiver.h"
 
+#include "job.h"
 #include "socket.h"
 #include "packet.h"
-#include "job.h"
-#include "queues/job_queue.h"
 #include "globals.h"
+#include "queues/job_queue.h"
 #include "utils/allocator.h"
 
 /**

Source/charon/utils/logger.c

@@ -277,7 +277,7 @@ logger_t *logger_create(char *logger_name, logger_level_t log_level,FILE * outpu
 	
 	if (output == NULL)
 	{
-		openlog(DEAMON_NAME, 0, LOG_DAEMON);
+		openlog(DAEMON_NAME, 0, LOG_DAEMON);
 	}
 	
 	return (logger_t*)this;