Initial message queue communication.

[wsti_so.git] / src / process3.c

diff --git a/src/process3.c b/src/process3.c
index e75a4fb610d66bae33e895d8271178a8280fe1e7..54bb6587370a53648f0aecbc63a6e2b299ec5176 100644 (file)
--- a/src/process3.c
+++ b/src/process3.c
@@ -9,34 +9,84 @@
 /* Signals handling.. */
 #include <signal.h>
 
+#include <sys/shm.h>
+
 /** Named pipe used to communicate with process2 */
 char * read_pipe = "/tmp/process2pipe";
 
 /** Descriptor of input pipe */
 int read_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", "process3", strsignal(signo));
        if (signo == SIGUSR1) {
-               fprintf(stderr, "[%s] Received SIGUSR1!\n", "process3");
+               fprintf(stderr, "[%s] > Notified!\n", "process3");
        }
-       else if (signo == SIGQUIT) {
-               fprintf(stderr, "[%s] Received SIGQUIT!\n", "process3");
+       else if (signo == SIGTERM) {
+               int i = 0;
+               struct queue_message msg;
+               msg.signo = signo;
+
                fprintf(stderr, "[%s] > Releasing resources\n", "process3");
                close(read_descriptor);
+               for (; i < 3; i++) {
+                       pid_t pid = processes->pids[i];
+                       fprintf(stderr, "[%s] Process %d, PID: %d\n", "process3", i, pid);
+                       // Bleh
+                       if (i != 2) {
+                               msgsnd(pid, &msg, sizeof(msg));
+                               kill(pid, SIGUSR1);
+                       }
+               }
                exit(0);
        }
-       else if (signo == SIGINT) {
-               fprintf(stderr, "[%s] Received SIGINT!\n", "process3");
-               fprintf(stderr, "[%s] > Closing pipe\n", "process2");
+       else if (signo == SIGTSTP) {
+               fprintf(stderr, "[%s] > Closing pipe\n", "process3");
                close(read_descriptor);
+               raise (SIGSTOP);
        }
        else if (signo == SIGCONT) {
-               fprintf(stderr, "[%s] Received SIGCONT!\n", "process3");
-               fprintf(stderr, "[%s] > Opening pipe\n", "process2");
+               fprintf(stderr, "[%s] > Opening pipe\n", "process3");
                read_descriptor = open(read_pipe, O_RDONLY);
        }
 }
@@ -60,16 +110,37 @@ int main(void) {
        if (signal(SIGUSR1, sig_handler) == SIG_ERR) {
                fprintf(stderr, "can't catch SIGUSR1\n");
        }
-       if (signal(SIGQUIT, sig_handler) == SIG_ERR) {
-               fprintf(stderr, "can't catch SIGQUIT\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[2] = getpid();
+
+       fprintf(stderr, "[%s] Shared pid: %d\n", "process3", 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);
+
        /* Reading from process2 */
        read_descriptor = open(read_pipe, O_RDONLY);