Init commit

src/notify_touchy.c

@@ -0,0 +1,178 @@
+/*
+FILENAME :     notify_touchy.c
+DESCRIPTION :  Service which touches a file to vrffile if the return code of program is status_ok.
+AUTHOR :       Bandie
+DATE :         2018-03-17T20:31:48+01:00
+LICENSE :      GNU-GPLv3
+*/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <ctype.h>
+#include <unistd.h>
+#include <signal.h>
+#include <sys/wait.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <string.h>
+#include <libnotify/notify.h>
+
+#define PROGARGVSIZE 256
+
+char *prog = NULL;
+
+void help(){
+  printf("notify_touchy ( arguments )\n\n"
+         "It checks if a program has a specific return code within some time. If it has a file will be touched. "
+         "If not it will be erased.\n"
+         "The purpose is to integrate it with i3status.\n\n"
+         "Arguments:\n\n"
+         "-h, --help                         Shows this here.\n"
+         "-p, --program [program or script]  Executes and checks this program for the return code. (Required)\n"
+         "-r, --return [return code]         Listen to this return code. (Default: 0)\n"
+         "-s, --sleep [seconds]              Interval of when the next check should start. (Default: 20)\n"
+         "-v, --verbose                      Show some unnecessary stuff.\n\n");
+}
+
+void exec_failed(){
+  fprintf(stderr, "ERROR: It is not possible to execute %s: ", prog);
+  if(access(prog, F_OK)) fprintf(stderr, "It does not exist (in your PATH variable).\n");
+  else fprintf(stderr, "Access denied. (Are you root?)\n");
+  exit(EXIT_FAILURE);
+}
+
+void sig_handler(int signo){
+  if(signo == SIGTERM || signo == SIGINT){
+    exit(0);
+  }
+}
+
+int main(int argc, char *argv[]){
+  int returncode = 0;
+  int sleeptm = 20;
+  int verbose = 0;
+  int statval;
+  int progargc = 1;
+  char *progargv[PROGARGVSIZE];
+ 
+
+
+  // Argument handling
+  for(int i=0; i<argc; i++){
+    if(!strcmp(argv[i], "--help") || !strcmp(argv[i], "-h")){
+      help();
+      return 0;
+    }
+    if(!strcmp(argv[i], "--verbose") || !strcmp(argv[i], "-v"))
+      verbose = 1;
+    if((!strcmp(argv[i], "--program") || !strcmp(argv[i], "-p")) && (i+1) < argc){
+      int k = 0;
+      while(argv[i+1][k]){
+        if(argv[i+1][k] == ' ')
+          progargc++;
+        k++;
+      }
+      
+      char *prgargptr;
+      prgargptr = strtok(argv[i+1], " ");
+      prog = prgargptr;
+      progargv[0] = prgargptr;
+      int l=1;
+      while(prgargptr != NULL){
+        prgargptr = strtok(NULL, " ");
+        progargv[l] = prgargptr;
+        l++;
+      }
+      
+    }
+    if((!strcmp(argv[i], "--return") || !strcmp(argv[i], "-r")) && (i+1) < argc){
+      int j = 0;
+      while(argv[i+1][j]){
+        if(j==0 && argv[i+1][j] == '-') j++;
+        if(!isdigit(argv[i+1][j])){
+          fprintf(stderr, "ERROR: Invalid argument. The return code must be a digit.\n");
+          return 1;
+        }
+        j++;
+      }
+      sscanf(argv[i+1], "%d", &returncode);
+    }
+    if((!strcmp(argv[i], "--sleep") || !strcmp(argv[i], "-s")) && (i+1) < argc){
+      int j=0;
+      while(argv[i+1][j]){
+        if(!isdigit(argv[i+1][j]) || argv[i+1][0] == '0'){
+          fprintf(stderr, "ERROR: Invalid argument. The sleep value must be a positive digit.\n");
+          if(argv[i+1][0] == '0')
+            fprintf(stderr, "No, you REALLY don't want to put a 0 in here.\n");
+          return 1;
+        }
+        j++;
+      }
+      sscanf(argv[i+1], "%d", &sleeptm);
+    }
+
+  }
+
+  if(!prog){
+    fprintf(stderr, "ERROR: Missing argument.\nExcuse me but where did you specifiy which program you use? (-p)\n");
+    exit(1);
+  }
+
+  // Signal handling
+  if(signal(SIGTERM, sig_handler) == SIG_ERR || signal(SIGINT, sig_handler) == SIG_ERR)
+    fprintf(stderr, "ERROR: Can't catch signal!");
+  
+  while(1){  
+
+    if(fork() == 0){
+
+      // Exec program silently by default
+      if(!verbose){
+        int fd = open("/dev/null", O_WRONLY | O_CREAT, 0666);    
+
+        dup2(fd, 1);
+        dup2(fd, 2);
+        if(execvp(prog, progargv) < 0)
+          exec_failed();
+        
+        close(fd);
+      }
+      else{
+        if(execvp(prog, progargv) < 0)
+          exec_failed();
+      }
+
+
+    } 
+
+    else {
+
+      if(verbose) 
+        printf("Waiting for %d...\n", getpid());
+
+      wait(&statval);
+      if(WIFEXITED(statval)){
+        if(verbose) 
+          printf("Exit: %d\n", statval);
+
+        if(statval != returncode){
+          char s[100];
+          sprintf(s, "Warning: %s returned something bad.", prog);
+          printf("%s\n", s);
+          notify_init(prog);
+          NotifyNotification *Not = notify_notification_new("Notify_touchy", s, "dialog-warning");
+          notify_notification_show(Not, NULL);
+          g_object_unref(G_OBJECT(Not));
+          notify_uninit();
+        }
+
+      } 
+    }
+
+    sleep(sleeptm);
+  }
+
+
+  sig_handler(SIGTERM);
+  return 0;
+}