X-Git-Url: https://git.dlugolecki.net.pl/?a=blobdiff_plain;f=src%2Fprocess1.c;h=2492ebecfe874f5ed4b4bae357b8e05aa56122f0;hb=61f14079d6d16459dfa2b008392fccd78bc423a4;hp=9fdee30b55edb5020a3eb2c05201fa24154f44ca;hpb=717e5b533ec94ef8f9a703bd4f019a2f45e138df;p=wsti_so.git

diff --git a/src/process1.c b/src/process1.c
index 9fdee30..2492ebe 100644
--- a/src/process1.c
+++ b/src/process1.c
@@ -8,6 +8,10 @@
 /* Signals handling.. */
 #include <signal.h>
 
+#include <sys/shm.h>
+#include <sys/msg.h>
+
+
 /** If buffer is too small to hold entire string, it is incremented by this value */
 #define BUFFER_STEP 16
 
@@ -20,15 +24,88 @@ 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 {
+	long mtype;
+	int signo[1];
+};
+
+void notify_other_processes(int signo) {
+	int i = 0;
+	struct queue_message msg;
+ 	msg.signo[0] = signo;
+
+	for (; i < 3; i++) {
+		pid_t pid = processes->pids[i];
+		// Bleh
+		if (i != 0 && pid != 0) {
+			msg.mtype = i+1;
+			fprintf(stderr, "[%s] Sending message of type (%d) with value %d\n", "process1", msg.mtype, msg.signo[0]);
+			msgsnd(pid, &msg, sizeof(int), 0);
+			fprintf(stderr, "[%s] Sending signal %s (%d) to PID: %d\n", "process1", strsignal(SIGUSR1), SIGUSR1, pid);
+			kill(pid, SIGUSR1);
+		}
+	}
+}
+
 /**
  * Handler for signals.
  */
 void sig_handler(int signo)
 {
 	fprintf(stderr, "[%s] Received %s!\n", "process1", strsignal(signo));
+
 	if (signo == SIGUSR1) {
+		fprintf(stderr, "[%s] > Notified!\n", "process1");
+		struct queue_message msg;
+		/* Check queues from both other processes */
+		if (msgrcv(processes->pids[1], &msg, sizeof(int), 1, 0) > 0) {
+			fprintf(stderr, "[%s] > Notified with value: %s!\n", "process1", strsignal(msg.signo[0]));
+			raise(msg.signo[0]);
+			break;
+		}
+		else if (msgrcv(processes->pids[2], &msg, sizeof(int), 1, 0) > 0) {
+			fprintf(stderr, "[%s] > Notified with value: %s!\n", "process1", strsignal(msg.signo[0]));
+			raise(msg.signo[0]);
+			break;
+		}
 	}
 	else if (signo == SIGTERM) {
+		fprintf(stderr, "[%s] > Signalling other processes..\n", "process1");
+		processes->pids[1] = 0;
+		notify_other_processes(signo);
+
 		fprintf(stderr, "[%s] > Releasing resources\n", "process1");
 		close(write_pipe);
 		unlink(write_pipe);
@@ -88,6 +165,26 @@ int main(void) {
 		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());
+
+	/**
+	 * 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);