Add SIGSTOP in SIGTSTP to properly stop the process.

[wsti_so.git] / src / process1.c

diff --git a/src/process1.c b/src/process1.c
index f4a1392d84398e72d440e0d63cfb6b64690b46a7..9fdee30b55edb5020a3eb2c05201fa24154f44ca 100644 (file)
--- a/src/process1.c
+++ b/src/process1.c
@@ -5,9 +5,51 @@
 /* open/read/write/close */
 #include <fcntl.h>
 
+/* Signals handling.. */
+#include <signal.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;
+
+/**
+ * Handler for signals.
+ */
+void sig_handler(int signo)
+{
+       fprintf(stderr, "[%s] Received %s!\n", "process1", strsignal(signo));
+       if (signo == SIGUSR1) {
+       }
+       else if (signo == SIGTERM) {
+               fprintf(stderr, "[%s] > Releasing resources\n", "process1");
+               close(write_pipe);
+               unlink(write_pipe);
+
+               if (buffer) {
+                       free(buffer);
+                       buffer = NULL;
+               }
+               exit(0);
+       }
+       else if (signo == SIGTSTP) {
+               fprintf(stderr, "[%s] > Closing pipe\n", "process1");
+               close(write_pipe);
+               raise (SIGSTOP);
+       }
+       else if (signo == SIGCONT) {
+               fprintf(stderr, "[%s] > Opening pipe\n", "process1");
+               file_descriptor = open(write_pipe, O_WRONLY);
+       }
+}
+
 /**
  * Program reads entire lines of text from the standard input and pass them
  * to the process2 using created pipe.
@@ -16,9 +58,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;
 
@@ -31,16 +70,28 @@ int main(void) {
         */
        char * tmp = NULL;
 
-       /** Named pipe used to communnicate with process2 */
-       char * write_pipe = "/tmp/process1pipe";
+       fprintf(stderr, "[%s] Init!\n", "process1");
 
-       /** File descriptor of pipe */
-       int file_descriptor;
+       /**
+        * Register signals handled by process
+        */
+       if (signal(SIGUSR1, sig_handler) == SIG_ERR) {
+               fprintf(stderr, "can't catch SIGUSR1\n");
+       }
+       if (signal(SIGTERM, sig_handler) == SIG_ERR) {
+               fprintf(stderr, "can't catch SIGTERM\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");
+       }
 
        mkfifo(write_pipe, 0666);
 
        file_descriptor = open(write_pipe, O_WRONLY);
-       
+
        do {
                c = fgetc(stdin);
 
@@ -88,7 +139,7 @@ int main(void) {
                                buffer = tmp;
                                buffer[i] = c;
 
-                               /* Used for debug..*/
+                               /* Used only for debugging..*/
                                /*
                                printf("c: %c/%d, i: %d, bl: %d\n", c, c, i, buffer_length);
                                */
@@ -97,6 +148,7 @@ int main(void) {
                }
        } while(c != EOF);
 
+       /* Release resources in normal program flow exit. */
        close(write_pipe);
        unlink(write_pipe);