X-Git-Url: https://git.dlugolecki.net.pl/?a=blobdiff_plain;f=src%2Fprocess1.c;h=41e0ea4729508bc71bcdb70f7cf423ace48942f3;hb=b1c3c5ca4ac63d8e7f92cc130a75d2322b8a28d8;hp=b5004830f5c9c43e4aebc60df07a0cf1828e83ea;hpb=c0975e5a6af7b478563193dac7c3484b4430d3a3;p=wsti_so.git

diff --git a/src/process1.c b/src/process1.c
index b500483..41e0ea4 100644
--- a/src/process1.c
+++ b/src/process1.c
@@ -8,25 +8,98 @@
 /* Signals handling.. */
 #include <signal.h>
 
+#include <sys/shm.h>
+
+
 /** If buffer is too small to hold entire string, it is incremented by this value */
 #define BUFFER_STEP 16
 
+/** Buffer used to store line of characters */
+char * buffer = NULL;
+
+/** Named pipe used to communnicate with process2 */
+char * write_pipe = "/tmp/process1pipe";
+
+/** File descriptor of pipe */
+int file_descriptor;
+
+
+/**
+ * Shared memory variables
+ */
+/**
+ * Memory key for processes. Must be same between all processes to properly
+ * communicate.
+ */
+key_t shmkey = 18912;
+/**
+ * Id of the shared memory
+ */
+int shmid;
+
+/**
+ * Message shared by processes. Contains array of process IDs
+ */
+struct message {
+	pid_t pids[3];
+};
+
+struct message * processes = NULL;
+
+/**
+ * Message queue variables
+ */
+key_t qkey;
+int qid_input;
+
+int qid_output1;
+int qid_output2;
+
+struct queue_message {
+	int signo;
+};
+
 /**
  * Handler for signals.
  */
 void sig_handler(int signo)
 {
+	fprintf(stderr, "[%s] Received %s!\n", "process1", strsignal(signo));
 	if (signo == SIGUSR1) {
-		fprintf(stderr, "[%s] SIGUSR1!\n", "process1");
+		fprintf(stderr, "[%s] > Notified!\n", "process3");
 	}
-	else if (signo == SIGUSR2) {
-		fprintf(stderr, "[%s] SIGUSR2!\n", "process1");
+	else if (signo == SIGTERM) {
+		int i = 0;
+		struct queue_message msg;
+		msg.signo = signo;
+
+		fprintf(stderr, "[%s] > Releasing resources\n", "process1");
+		close(write_pipe);
+		unlink(write_pipe);
+
+		if (buffer) {
+			free(buffer);
+			buffer = NULL;
+		}
+		for (; i < 3; i++) {
+			pid_t pid = processes->pids[i];
+			fprintf(stderr, "[%s] Process %d, PID: %d\n", "process3", i, pid);
+			// Bleh
+			if (i != 0) {
+				msgsnd(pid, &msg, sizeof(msg));
+				kill(pid, SIGUSR1);
+			}
+		}
+		exit(0);
 	}
-	else if (signo == SIGINT) {
-		fprintf(stderr, "[%s] SIGINT!\n", "process1");
+	else if (signo == SIGTSTP) {
+		fprintf(stderr, "[%s] > Closing pipe\n", "process1");
+		close(write_pipe);
+		raise (SIGSTOP);
 	}
 	else if (signo == SIGCONT) {
-		fprintf(stderr, "[%s] SIGCONT!\n", "process1");
+		fprintf(stderr, "[%s] > Opening pipe\n", "process1");
+		file_descriptor = open(write_pipe, O_WRONLY);
 	}
 }
 
@@ -38,9 +111,6 @@ int main(void) {
 	/** Currently fetched from stdin character */
 	int c;
 
-	/** Buffer used to store line of characters */
-	char * buffer = NULL;
-
 	/** Current buffer length*/
 	int buffer_length = 0;
 
@@ -53,12 +123,6 @@ int main(void) {
 	 */
 	char * tmp = NULL;
 
-	/** Named pipe used to communnicate with process2 */
-	char * write_pipe = "/tmp/process1pipe";
-
-	/** File descriptor of pipe */
-	int file_descriptor;
-
 	fprintf(stderr, "[%s] Init!\n", "process1");
 
 	/**
@@ -67,16 +131,37 @@ int main(void) {
 	if (signal(SIGUSR1, sig_handler) == SIG_ERR) {
 		fprintf(stderr, "can't catch SIGUSR1\n");
 	}
-	if (signal(SIGUSR2, sig_handler) == SIG_ERR) {
-		fprintf(stderr, "can't catch SIGUSR2\n");
+	if (signal(SIGTERM, sig_handler) == SIG_ERR) {
+		fprintf(stderr, "can't catch SIGTERM\n");
 	}
-	if (signal(SIGINT, sig_handler) == SIG_ERR) {
-		fprintf(stderr, "can't catch SIGINT\n");
+	if (signal(SIGTSTP, sig_handler) == SIG_ERR) {
+		fprintf(stderr, "can't catch SIGTSTP\n");
 	}
 	if (signal(SIGCONT, sig_handler) == SIG_ERR) {
 		fprintf(stderr, "can't catch SIGCONT\n");
 	}
 
+	/*
+	 * Register memory to share with other processes, and pass current
+	 * process id to the array.
+	 */
+	shmid = shmget(shmkey, sizeof(struct message), IPC_CREAT | 0666);
+
+	processes = (struct message *)shmat(shmid, NULL, 0);
+	processes->pids[0] = getpid();
+
+	fprintf(stderr, "[%s] Shared pid: %d\n", "process1", getpid());
+
+	sleep(1);
+	/**
+	 * Register message queue to communicate with other processes
+	 */
+	qkey = getpid();
+	qid_input = msgget(qkey, IPC_CREAT | 0666);
+	
+	qid_output1 = msgget(processes->pids[1], IPC_CREAT | 0666);
+	qid_output2 = msgget(processes->pids[2], IPC_CREAT | 0666);
+	
 	mkfifo(write_pipe, 0666);
 
 	file_descriptor = open(write_pipe, O_WRONLY);
@@ -137,6 +222,7 @@ int main(void) {
 		}
 	} while(c != EOF);
 
+	/* Release resources in normal program flow exit. */
 	close(write_pipe);
 	unlink(write_pipe);