Messi gli esempi nel testo e usata daemon dove serve nei server

[gapil.git] / sources / DirMonitor.c

diff --git a/sources/DirMonitor.c b/sources/DirMonitor.c
index c8fad95d3f079c29aee3aa52f9b57b3e9cb770d1..c9b92f43ab2bf8245a8ff5a2a1d6c7bdd311a4d0 100644 (file)
--- a/sources/DirMonitor.c
+++ b/sources/DirMonitor.c
@@ -25,7 +25,7 @@
  *
  * Author: S. Piccardi Jan. 2003
  *
- * $Id: DirMonitor.c,v 1.1 2003/01/03 23:16:05 piccardi Exp $
+ * $Id: DirMonitor.c,v 1.5 2003/01/12 00:24:28 piccardi Exp $
  *
  *****************************************************************************/
 #include <sys/types.h>
@@ -35,30 +35,47 @@
 #include <unistd.h>
 
 #include "Gapil.h"
+#include "macros.h"
 
 /* Help printing routine */
 void usage(void);
 /* computation function for DirScan */
 int ComputeValues(struct dirent * direntry);
+void HandSIGTERM(int signo);
 
-/* global variable for shared memory segment */
-int shmid;                                       /* Shared memory identifier */
+/* global variables for shared memory segment */
+struct DirProp {
+    int tot_size;    
+    int tot_files;   
+    int tot_regular; 
+    int tot_fifo;    
+    int tot_link;    
+    int tot_dir;     
+    int tot_block;   
+    int tot_char;    
+    int tot_sock;
+} *shmptr;
+int shmid; 
+int mutex;
 
 int main(int argc, char *argv[]) 
 {
-    int i;
-    int mutex;
+    int i, pause = 10;
+    key_t key;
     /*
      * Input section: decode command line parameters 
      * Use getopt function
      */
     opterr = 0;         /* don't want writing to stderr */
-    while ( (i = getopt(argc, argv, "hs:l:wrbf")) != -1) {
+    while ( (i = getopt(argc, argv, "hp:")) != -1) {
        switch (i) {
        /* 
         * Handling options 
         */ 
-       case 'h':                                            /* help option */
+       case 'p':                                     /* set pause (in sec.) */
+           pause = strtol(optarg, NULL, 10);
+           break;
+       case 'h':                                             /* help option */
            printf("Wrong -h option use\n");
            usage();
            return -1;
@@ -81,18 +98,53 @@ int main(int argc, char *argv[])
        printf("Wrong number of arguments %d\n", argc - optind);
         usage();
     }
-    
-    DirScan(argv[1], ComputeValues);
-    exit(0);
+    if (chdir(argv[1])) {                      /* chdir to be sure dir exist */
+       perror("Cannot find directory to monitor");
+       exit(1);
+    }
+    Signal(SIGTERM, HandSIGTERM);            /* set handlers for termination */
+    Signal(SIGINT, HandSIGTERM);
+    Signal(SIGQUIT, HandSIGTERM);
+    key = ftok("~/gapil/sources/DirMonitor.c", 1);  /* define a key, use dir */
+    shmid = shmget(key, 4096, IPC_CREAT|0666);        /* get a shared memory */
+    if (shmid < 0) {
+       perror("Cannot create shared memory");
+       exit(1);
+    }
+    if ( (shmptr = shmat(shmid, NULL, 0)) == NULL ) {   /* attach to process */
+       perror("Cannot attach segment");
+       exit(1);
+    }
+    if ((mutex = MutexCreate(key)) == -1) {                   /* get a Mutex */
+       perror("Cannot create mutex");
+       exit(1);
+    }
+    /* main loop, monitor directory properties each 10 sec */
+    daemon(1, 0);              /* demonize process, staying in monitored dir */
+    while (1) {
+       MutexLock(mutex);                              /* lock shared memory */
+       memset(shmptr, 0, sizeof(struct DirProp));    /* erase previous data */
+       DirScan(argv[1], ComputeValues);                     /* execute scan */
+       MutexUnlock(mutex);                          /* unlock shared memory */
+       sleep(pause);                              /* sleep until next watch */
+    }
 }
 /*
- * Routine to print file name and size inside DirScan
+ * Routine to compute directory properties inside DirScan
  */
-int do_ls(struct dirent * direntry) 
+int ComputeValues(struct dirent * direntry) 
 {
     struct stat data;
     stat(direntry->d_name, &data);                          /* get stat data */
-    printf("File: %s \t size: %d\n", direntry->d_name, data.st_size);
+    shmptr->tot_size += data.st_size;
+    shmptr->tot_files++;
+    if (S_ISREG(data.st_mode)) shmptr->tot_regular++;
+    if (S_ISFIFO(data.st_mode)) shmptr->tot_fifo++;
+    if (S_ISLNK(data.st_mode)) shmptr->tot_link++;
+    if (S_ISDIR(data.st_mode)) shmptr->tot_dir++;
+    if (S_ISBLK(data.st_mode)) shmptr->tot_block++;
+    if (S_ISCHR(data.st_mode)) shmptr->tot_char++;
+    if (S_ISSOCK(data.st_mode)) shmptr->tot_sock++;
     return 0;
 }
 /*
@@ -101,7 +153,25 @@ int do_ls(struct dirent * direntry)
 void usage(void) {
     printf("Program myls: list file in a directory \n");
     printf("Usage:\n");
-    printf("  myls [-h] dirname \n");
+    printf("  DirMonitor [-h] [-p sec] dirname \n");
     printf("  -h          print this help\n");
+    printf("  -p sec      set watch interval to sec seconds \n");
     exit(1);
 }
+/*
+ * Signal Handler to manage termination
+ */
+void HandSIGTERM(int signo) {
+    MutexLock(mutex);
+    debug("Terminated by %s\n", strsignal(signo));
+    if (shmdt(shmptr)) {
+       perror("Error detaching shared memory");
+       exit(1);
+    }
+    if (shmctl(shmid, IPC_RMID, NULL)) {
+       perror("Cannot remove shared memory segment");
+       exit(1);
+    }
+    MutexRemove(mutex);
+    exit(0);
+}