for users ensured proper destructors, and ctrl c should kill it

users/mainReceiver.cpp

@@ -12,6 +12,9 @@
 #include "utilities.h"
 #include "logger.h"
 
+#include <sys/types.h>	//wait
+#include <sys/wait.h>	//wait
+
 #include <string>
 using namespace std;
 
@@ -19,35 +22,35 @@ using namespace std;
 #define START_TCP_PORT	1954
 #define PRINT_IN_COLOR(c,f, ...) 	printf ("\033[%dm" f RESET, 30 + c+1, ##__VA_ARGS__)
 
-slsReceiverUsers *receivers[NUM_RECEIVERS];
 
-void deleteReceiver(slsReceiverUsers* r[]){
-	for (int i = 0; i < NUM_RECEIVERS; ++i)
-	if (r[i]) {
-		delete r[i];
-		r[i] = 0;
-	}
-}
+pid_t childPid[NUM_RECEIVERS];
+bool keeprunning;
+int numrunning;
 
-void closeFile(int p){
-	deleteReceiver(receivers);
+
+
+void sigChildExitedHandler(int sig) {
+  pid_t pid = wait(NULL);
+  bprintf(GRAY, "\nChild Process Pid %d exited.\n", pid);
+  numrunning--;
 }
 
 
-
+void sigInterruptHandler(int p){
+	keeprunning = false;
+}
 
 int StartAcq(char* filepath, char* filename, uint64_t fileindex, uint32_t datasize, void*p){
-  printf("--StartAcq:  filepath:%s  filename:%s fileindex:%llu  datasize:%u\n",
+  bprintf(BLUE, "#### StartAcq:  filepath:%s  filename:%s fileindex:%llu  datasize:%u ####\n",
 	 filepath, filename, fileindex, datasize);
 
-  printf("--StartAcq: returning 0\n");
+  bprintf(BLUE, "--StartAcq: returning 0\n");
   return 0;
 }
 
 
 void AcquisitionFinished(uint64_t frames, void*p){
-  printf("AcquisitionFinished: frames:%llu \n",frames);
-
+  bprintf(BLUE, "#### AcquisitionFinished: frames:%llu ####\n",frames);
 }
 
 
@@ -56,7 +59,7 @@ void GetData(uint64_t frameNumber, uint32_t expLength, uint32_t packetNumber, ui
 		char* datapointer, uint32_t datasize, void* p){
 
 	PRINT_IN_COLOR (xCoord,
-			"%d GetData: \n"
+			"#### %d GetData: ####\n"
 			"frameNumber: %llu\t\texpLength: %u\t\tpacketNumber: %u\t\tbunchId: %llu\t\ttimestamp: %llu\t\tmodId: %u\t\t"
 			"xCoord: %u\t\tyCoord: %u\t\tzCoord: %u\t\tdebug: %u\t\troundRNumber: %u\t\tdetType: %u\t\t"
 			"version: %u\t\tfirstbytedata: 0x%x\t\tdatsize: %u\n\n",
@@ -70,23 +73,56 @@ void GetData(uint64_t frameNumber, uint32_t expLength, uint32_t packetNumber, ui
 
 int main(int argc, char *argv[]) {
 
-	//Catch signal SIGINT to close files properly
-	signal(SIGINT,closeFile);
+	// set default child process pid values
+	for (int i = 0; i < NUM_RECEIVERS; ++i)
+		childPid[i] = -1;
+
+	keeprunning = true;
+	numrunning = 0;
+
+	// Catch signal SIGINT to close files and call destructors properly
+	struct sigaction sa;
+	sa.sa_flags=0;							// no flags
+	sa.sa_handler=sigInterruptHandler;		// handler function
+	sigemptyset(&sa.sa_mask);				// dont block additional signals during invocation of handler
+	if (sigaction(SIGINT, &sa, NULL) == -1) {
+		bprintf(RED, "Could not set handler function for SIGINT\n");
+	}
+
+	// wait for all the SIGCHILD signals
+	struct sigaction asa;
+	asa.sa_flags=0;							// no flags
+	asa.sa_handler=sigChildExitedHandler;	// handler function
+	sigemptyset(&asa.sa_mask);				// dont block additional signals during invocation of handler
+	if (sigaction(SIGCHLD, &asa, NULL) == -1) {
+		bprintf(RED, "Could not set handler function for SICHILD\n");
+	}
+
 
-	int ret = slsReceiverDefs::OK;
 	int narg= 3;
 
 	for (int i = 0; i < NUM_RECEIVERS; ++i) {
 
+		childPid[i] = fork();
+
+		// fork failed
+		if (childPid[i] < 0) {
+			bprintf(RED,"fork() failed. Killing all the receiver objects\n");
+			raise(SIGINT);
+		}
+
+		// child process
+		else if (childPid[i] == 0) {
+			bprintf(BLUE,"Starting Receiver %d with pid %ld\n", i, (long)getpid());
+
 			char temp[10];
 			sprintf(temp,"%d",START_TCP_PORT + i);
 			char* args[] = {(char*)"ignored", (char*)"--rx_tcpport", temp};
-
-		cprintf(BLUE,"Starting Receiver %d\n", i);
-		receivers[i] = new slsReceiverUsers(narg, args, ret);
+			int ret = slsReceiverDefs::OK;
+			slsReceiverUsers *receiver = new slsReceiverUsers(narg, args, ret);
 			if(ret==slsReceiverDefs::FAIL){
-			deleteReceiver(receivers);
-			return -1;
+				delete receiver;
+				exit(EXIT_FAILURE);
 			}
 
 			//register callbacks
@@ -105,16 +141,16 @@ int main(int argc, char *argv[]) {
 			 * we write depending on file write enable
 			 * users get data to write depending on call backs registered
 			 */
-		printf("Registering 	StartAcq()\n");
-		receivers[i]->registerCallBackStartAcquisition(StartAcq, NULL);
+			bprintf(BLUE, "Registering 	StartAcq()\n");
+			receiver->registerCallBackStartAcquisition(StartAcq, NULL);
 
 			/**
 			 * Call back for acquisition finished
 			 * callback argument is
 			 * total frames caught
 			 */
-		printf("Registering 	AcquisitionFinished()\n");
-		receivers[i]->registerCallBackAcquisitionFinished(AcquisitionFinished, NULL);
+			bprintf(BLUE, "Registering 	AcquisitionFinished()\n");
+			receiver->registerCallBackAcquisitionFinished(AcquisitionFinished, NULL);
 
 			/**
 			 * Call back for raw data
@@ -135,39 +171,33 @@ int main(int argc, char *argv[]) {
 			 * dataPointer is the pointer to the data
 			 * dataSize in bytes is the size of the data in bytes
 			 */
-		printf("Registering     GetData() \n");
-		receivers[i]->registerCallBackRawDataReady(GetData,NULL);
+			bprintf(BLUE, "Registering     GetData() \n");
+			receiver->registerCallBackRawDataReady(GetData,NULL);
 
 
 			//start tcp server thread
-		ret = receivers[i]->start();
-		if(ret == slsReceiverDefs::FAIL){
-			for (int i = 0; i < i; ++i)
-				receivers[i]->stop();
-			deleteReceiver(receivers);
-			return -1;
-		}
+			if (receiver->start() == slsReceiverDefs::FAIL){
+				delete receiver;
+				exit(EXIT_FAILURE);
 			}
 
-
-	FILE_LOG(logDEBUG1) << "DONE!" << endl;
-	cprintf( BLUE, "Type \'q\' to exit\n");
-	string str;
-	cin>>str;
-	//wait and look for an exit keyword
-	while(str.find("q") == string::npos)
-		cin>>str;
-	//stop tcp server thread, stop udp socket
-	for (int i = 0; i < NUM_RECEIVERS; ++i) {
-		cprintf(BLUE,"Stopping Receiver %d\n",i);
-		receivers[i]->stop();
+			while(keeprunning);
+			delete receiver;
+			exit(EXIT_SUCCESS);
 		}
 
+		// parent process
+		else
+			numrunning++;
 
-	deleteReceiver(receivers);
-
+	}
 
+	FILE_LOG(logINFO) << "Ready ... ";
+	bprintf(GRAY, "\n[ Press \'Ctrl+c\' to exit ]\n");
 
+	// wait for all child processes to exit
+	while(numrunning);
+	FILE_LOG(logINFO) << "Goodbye!";
 	return 0;
 }