www.pipapo.org Git - rxpd/blob - src/main.c

   1 /*
   2     main.c - regex policy daemon
   3
   4   Copyright (C)
   5     2007,               Christian Thaeter <ct@pipapo.org>
   6
   7   This program is free software; you can redistribute it and/or
   8   modify it under the terms of the GNU General Public License as
   9   published by the Free Software Foundation; either version 2 of the
  10   License, or (at your option) any later version.
  11
  12   This program is distributed in the hope that it will be useful,
  13   but WITHOUT ANY WARRANTY; without even the implied warranty of
  14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15   GNU General Public License for more details.
  16
  17   You should have received a copy of the GNU General Public License
  18   along with this program; if not, write to the Free Software
  19   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "rxpd.h"
  23
  24 #include <string.h>
  25 #include <sys/time.h>
  26 #include <sys/types.h>
  27 #include <sys/socket.h>
  28 #include <netinet/in.h>
  29
  30 #include <event.h>
  31
  32
  33 void
  34 version (void)
  35 {
  36   printf (
  37           "  Regex Policy Daemon ("PACKAGE_STRING")\n\n"
  38           "  Copyright (C)\n"
  39           "    2007,               Christian Thaeter <ct@pipapo.org>\n\n"
  40           "  This is free software.  You may redistribute copies of it under the terms of\n"
  41           "  the GNU General Public License <http://www.gnu.org/licenses/gpl.html>.\n"
  42           "  There is NO WARRANTY, to the extent permitted by law.\n\n"
  43           "  http://www.pipapo.org/pipawiki/RegexPolicyDaemon\n"
  44           );
  45 }
  46
  47 void
  48 usage (void)
  49 {
  50   /*
  51     rxpd [OPTIONS] RULES..
  52     -v          verbosity
  53     -V          version
  54     -d          daemonize
  55     -D          debug
  56     -b dir      basedir for rules
  57     -q          quiet
  58     -t port     tcp
  59     -u name     unix
  60     -p policy   access policies
  61     -i          case insensitive
  62     -4          ipv4
  63     -6          ipv6
  64     -r          resolve names
  65     -l log      log hits to logfile
  66     -h          usage
  67
  68     -U user     switch to user
  69   */
  70
  71 }
  72
  73
  74 int
  75 main (int argc, char** argv)
  76 {
  77   struct rxpd_base* rxpd;
  78
  79   openlog (PACKAGE_NAME, LOG_PID, LOG_DAEMON);
  80
  81   rxpd = rxpd_init (event_init ());
  82
  83   rxpd_log (rxpd, LOG_NOTICE, PACKAGE_STRING" starting up\n");
  84
  85   // parse commandline args
  86   opterr = 0;
  87
  88   int opt;
  89
  90   while ((opt = getopt (argc, argv, "vVdDb:qt:u:p:i46rl:h")) != -1)
  91     switch (opt)
  92       {
  93       case 'v':
  94         if (rxpd->verbosity < LOG_DEBUG)
  95           ++rxpd->verbosity;
  96         break;
  97       case 'V':
  98         version ();
  99         exit (EXIT_SUCCESS);
 100       case 'd':
 101         rxpd->daemonize = 1;
 102         break;
 103       case 'D':
 104         if (rxpd->verbosity < LOG_INFO)
 105           rxpd->verbosity = LOG_INFO;
 106         rxpd->daemonize = 0;
 107         break;
 108       case 'b':
 109         if (!rxpd->basedir)
 110           rxpd->basedir = rxpd_strdup (optarg);
 111         else
 112           rxpd_die ("basedir already set\n");
 113         break;
 114       case 'q':
 115         rxpd->verbosity = LOG_ALERT;
 116         break;
 117       case 't':
 118         {
 119           int port = atoi (optarg);
 120           if (port > 0 && port < 65536)
 121             {
 122               rxpd_socket_new_tcp4 (rxpd, NULL, port);
 123               // TODO error handling
 124             }
 125           else
 126             {
 127               rxpd_log (rxpd, LOG_ALERT, "Illegal port number\n");
 128               exit (EXIT_FAILURE);
 129             }
 130         }
 131         // TODO rxpd_socket_new_tcp6 (rxpd, NULL, 2374)
 132         break;
 133 #if 0 /*not yet implemented*/
 134       case 'u':
 135         //rxpd_socket_new_unix (rxpd, NULL, 2374);
 136         break;
 137 #endif
 138       case 'p':
 139         if (!rxpd->policy)
 140           rxpd->policy = rxpd_file_new (rxpd, optarg);
 141         else
 142           rxpd_die ("policy already set\n");
 143         break;
 144       case 'i':
 145         rxpd->regflags |= REG_ICASE;
 146         break;
 147 #if 0 /*not yet implemented*/
 148       case '4':
 149         break;
 150       case '6':
 151         break;
 152       case 'r':
 153         break;
 154       case 'l:':
 155         break;
 156 #endif
 157       case 'h':
 158         usage ();
 159         exit (0);
 160         break;
 161       default:
 162         rxpd_log (rxpd, LOG_ALERT, "Unknown option\n");
 163         exit (EXIT_FAILURE);
 164       }
 165
 166   if (!rxpd->basedir)
 167     {
 168       rxpd_log (rxpd, LOG_ALERT, "Basedir not set (use -b BASEDIR)\n");
 169       exit (EXIT_FAILURE);
 170     }
 171
 172   if (llist_is_empty (&rxpd->sockets))
 173     {
 174       rxpd_log (rxpd, LOG_ALERT, "No listening sockets (use -t TCP or -u UNIX)\n");
 175       exit (EXIT_FAILURE);
 176     }
 177
 178   if (rxpd->policy)
 179     {
 180       if (rxpd_file_load (rxpd->policy))
 181         rxpd_log (rxpd, LOG_INFO, "Loaded policy '%s'\n", rxpd->policy->filename);
 182       else
 183         {
 184           rxpd_log (rxpd, LOG_ALERT, "Failed loading policy file\n");
 185           exit (EXIT_FAILURE);
 186         }
 187     }
 188
 189   for (int i = optind; i < argc; ++i)
 190     {
 191       if (!rxpd_file_load (rxpd_file_new (rxpd, argv[i])))
 192         {
 193           //rxpd_log (rxpd, LOG_ALERT, "Failed loading file '%s'\n");
 194           exit (EXIT_FAILURE);
 195         }
 196     }
 197
 198   if (rxpd->daemonize && daemon(1, 0))
 199     rxpd_die ("Couldn't daemonize\n");
 200
 201   LLIST_FOREACH (&rxpd->sockets, n)
 202     {
 203       struct rxpd_socket* socket = (struct rxpd_socket*)n;
 204       rxpd_socket_schedule (socket);
 205     }
 206
 207   // eventloop
 208   event_dispatch ();
 209
 210   rxpd_log (rxpd, LOG_NOTICE, PACKAGE_STRING" exited\n");
 211   rxpd_destroy ();
 212   return EXIT_SUCCESS;
 213 }