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 3 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, see <http://www.gnu.org/licenses/>.
  17  *
  18  * Contributor(s):
  19  *   Tom Cort <tom.cort@state.vt.us>
  20  */
  21
  22 #include <errno.h>
  23 #include <idn-int.h>
  24 #include <inotifytools/inotifytools.h>
  25 #include <inotifytools/inotify.h>
  26 #include <libdaemon/dlog.h>
  27 #include <stdlib.h>
  28 #include <string.h>
  29
  30 #include "inotify.h"
  31 #include "sig.h"
  32
  33 int __inotify_init(char *basedir) {
  34         if (!inotifytools_initialize() || !inotifytools_watch_recursively(basedir, IN_ALL_EVENTS)) {
  35                 daemon_log(LOG_ERR, "(%s:%u) Failed init inotify: %s", __FILE__, __LINE__, strerror(inotifytools_error()));
  36                 return 1;
  37         } else {
  38                 return -1;
  39         }
  40 }
  41
  42 int __inotify_exit() {
  43 /*      inotifytools_cleanup();
  44 */      return 1;
  45 }
  46
  47 /**
  48  * Watch the specified directory for changes and call contains_virus()
  49  * @param basedir the directory to watch.
  50  */
  51 void inotify_main(char *basedir) {
  52         struct inotify_event * event;
  53         char *filename;
  54         int length;
  55
  56         if (__inotify_init(basedir) == -1) {
  57                 return;
  58         }
  59
  60         daemon_log(LOG_INFO, "(%s:%u) entering inotify loop", __FILE__, __LINE__);
  61
  62         while (!exit_now && (event = inotifytools_next_event(-1))) {
  63                 if (event && event->name && event->wd) {
  64                         length = strlen(inotifytools_filename_from_wd(event->wd)) + strlen(event->name) + 2;
  65                         filename = (char *) malloc(sizeof(char) * length);
  66
  67                         if (!filename) {
  68                                 daemon_log(LOG_ERR, "(%s:%u) malloc() Failed: %s", __FILE__, __LINE__, strerror(errno));
  69                         }
  70                         memset(filename, '\0', length);
  71
  72                         snprintf(filename, length-1, "%s%s", inotifytools_filename_from_wd( event->wd ), event->name);
  73
  74                         if (((event->mask & IN_CLOSE) || (event->mask & IN_MOVED_TO)) && !(event->mask & IN_ISDIR)) {
  75                                 daemon_log(LOG_ERR, "(%s:%u) SCAN -> %s\n", __FILE__, __LINE__, filename);
  76                         } else if ((event->mask & IN_CREATE) && (event->mask & IN_ISDIR)) {
  77                                 __inotify_exit();
  78                                 __inotify_init(basedir);
  79                         }
  80                 }
  81         }
  82
  83         daemon_log(LOG_INFO, "(%s:%u) leaving inotify loop", __FILE__, __LINE__);
  84         __inotify_exit();
  85 }