inotify.c

   1 /*
   2  * inoclam - Inotify+ClamAV virus scanner
   3  * Copyright (C) 2007 Vermont Department of Taxes
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  *
  19  * Contributor(s):
  20  *   Tom Cort <tom.cort@state.vt.us>
  21  */
  22
  23 #include <errno.h>
  24 #include <idn-int.h>
  25 #include <inotifytools/inotifytools.h>
  26 #include <inotifytools/inotify.h>
  27 #include <libdaemon/dlog.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30
  31 #include "inotify.h"
  32 #include "sig.h"
  33
  34 int __inotify_init(char *basedir) {
  35         if (!inotifytools_initialize() || !inotifytools_watch_recursively(basedir, IN_ALL_EVENTS)) {
  36                 daemon_log(LOG_ERR, "(%s:%u) Failed init inotify: %s", __FILE__, __LINE__, strerror(inotifytools_error()));
  37                 return -1;
  38         } else {
  39                 daemon_log(LOG_DEBUG, "(%s:%u) init inotify", __FILE__, __LINE__);
  40                 return 1;
  41         }
  42 }
  43
  44 int __inotify_exit() {
  45         inotifytools_cleanup();
  46         return 1;
  47 }
  48
  49 /**
  50  * Watch the specified directory for changes and call contains_virus()
  51  * @param basedir the directory to watch.
  52  */
  53 void inotify_main(char *basedir) {
  54         struct inotify_event * event;
  55         char *filename;
  56         int length;
  57
  58         event = NULL;
  59         filename = NULL;
  60         length = 0;
  61
  62         if (basedir == NULL || __inotify_init(basedir) == -1) {
  63                 exit_now = 1;
  64                 return;
  65         }
  66
  67         daemon_log(LOG_INFO, "(%s:%u) entering inotify loop", __FILE__, __LINE__);
  68
  69         while (!exit_now && (event = inotifytools_next_event(-1))) {
  70                 if (event && event->name && event->wd) {
  71                         length = strlen(inotifytools_filename_from_wd(event->wd)) + strlen(event->name) + 2;
  72                         filename = (char *) malloc(sizeof(char) * length);
  73                         if (!filename) {
  74                                 daemon_log(LOG_ERR, "(%s:%u) malloc() Failed: %s", __FILE__, __LINE__, strerror(errno));
  75                                 exit_now = 1;
  76                                 return;
  77                         }
  78                         memset(filename, '\0', length);
  79
  80                         snprintf(filename, length-1, "%s%s", inotifytools_filename_from_wd( event->wd ), event->name);
  81
  82                         if ((event->mask & (IN_CLOSE|IN_MOVED_TO)) && !(event->mask & IN_ISDIR)) {
  83                                 /*
  84                                  * Scan files that are being closed.
  85                                  * Scan files that have been moved to a watched directory.
  86                                  */
  87                                 daemon_log(LOG_ERR, "(%s:%u) SCAN -> %s", __FILE__, __LINE__, filename);
  88                         } else if ((event->mask & (IN_CREATE|IN_MOVED_TO|IN_DELETE)) && (event->mask & IN_ISDIR)) {
  89                                 /*
  90                                  * If the directory structure has changed, reload the watch list.
  91                                  * Benchmark this to determine if this is too much of a performance hit.
  92                                  */
  93                                 __inotify_exit();
  94                                 __inotify_init(basedir);
  95                         }
  96
  97                         if (filename) {
  98                                 free(filename);
  99                                 filename = NULL;
 100                         }
 101                 }
 102         }
 103
 104         daemon_log(LOG_INFO, "(%s:%u) leaving inotify loop", __FILE__, __LINE__);
 105         __inotify_exit();
 106 }