hostinfod/monitor_sockets.c

   1 /* virt-hostinfo
   2  * Copyright (C) 2009 Red Hat Inc.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 #ifdef HAVE_CONFIG_H
  20 #include <config.h>
  21 #endif
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <stdarg.h>
  26 #include <unistd.h>
  27 #include <fcntl.h>
  28 #include <sys/inotify.h>
  29 #include <sys/stat.h>
  30
  31 #include <apr_general.h>
  32
  33 #include "hostinfod.h"
  34
  35 int sockets_inotify_fd = -1;
  36
  37 void
  38 monitor_socket_dir (void)
  39 {
  40   struct stat statbuf;
  41
  42   if (!socket_dir) {
  43     error ("no socket directory was configured - see hostinfo(8)");
  44     exit (1);
  45   }
  46
  47   if (stat (socket_dir, &statbuf) == -1) {
  48     perrorf ("%s (socket directory): does not exist or is not accessible",
  49              socket_dir);
  50     exit (1);
  51   }
  52   if (!S_ISDIR (statbuf.st_mode)) {
  53     error ("%s (socket directory): not a directory", socket_dir);
  54     exit (1);
  55   }
  56
  57   debug ("creating inotify watch on '%s'", socket_dir);
  58
  59 #ifdef HAVE_INOTIFY_INIT1
  60   sockets_inotify_fd = inotify_init1 (IN_NONBLOCK | IN_CLOEXEC);
  61   if (sockets_inotify_fd == -1) {
  62     perrorf ("inotify_init1");
  63     exit (1);
  64   }
  65 #else
  66   sockets_inotify_fd = inotify_init ();
  67   if (sockets_inotify_fd == -1) {
  68     perrorf ("inotify_init");
  69     exit (1);
  70   }
  71   if (fcntl (sockets_inotify_fd, F_SETFL, O_NONBLOCK) == -1) {
  72     perrorf ("fcntl: O_NONBLOCK");
  73     exit (1);
  74   }
  75   if (fcntl (sockets_inotify_fd, F_SETFD, FD_CLOEXEC) == -1) {
  76     perrorf ("fcntl: FD_CLOEXEC");
  77     exit (1);
  78   }
  79 #endif
  80
  81   if (inotify_add_watch (sockets_inotify_fd, socket_dir,
  82                          IN_CREATE|IN_DELETE|IN_MOVE) == -1) {
  83     perrorf ("inotify_add_watch: %s", socket_dir);
  84     exit (1);
  85   }
  86
  87   debug ("inotify of %s successful, fd = %d",
  88          socket_dir, sockets_inotify_fd);
  89 }