loading files from optargs
[rxpd] / 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"
43           );
44 }
45
46 void
47 usage (void)
48 {
49   /*
50     rxpd [OPTIONS] RULES..
51     -v          verbosity
52     -V          version
53     -d          daemonize
54     -D          debug
55     -b dir      basedir for rules
56     -q          quiet
57     -t port     tcp
58     -u name     unix
59     -p policy   access policies
60     -i          case insensitive
61     -4          ipv4
62     -6          ipv6
63     -r          resolve names
64     -l log      log hits to logfile
65     -h          usage
66
67     -U user     switch to user
68   */
69
70 }
71
72
73 int
74 main (int argc, char** argv)
75 {
76   struct rxpd_base* rxpd;
77   rxpd = rxpd_init (event_init ());
78
79   // parse commandline args
80   opterr = 0;
81
82   int opt;
83
84   while ((opt = getopt (argc, argv, "vVdDb:qt:u:p:i46rl:h")) != -1)
85     switch (opt)
86       {
87       case 'v':
88         if (rxpd->verbosity < LOG_DEBUG)
89           ++rxpd->verbosity;
90         break;
91       case 'V':
92         version ();
93         exit (EXIT_SUCCESS);
94       case 'd':
95         rxpd->daemonize = 1;
96         break;
97       case 'D':
98         if (rxpd->verbosity < LOG_NOTICE)
99           rxpd->verbosity = LOG_NOTICE;
100         rxpd->daemonize = 0;
101         break;
102       case 'b':
103         if (!rxpd->basedir)
104           rxpd->basedir = rxpd_strdup (optarg);
105         else
106           rxpd_die ("basedir already set\n");
107         break;
108       case 'q':
109         rxpd->verbosity = LOG_ALERT;
110         break;
111       case 't':
112         {
113           int port = atoi (optarg);
114           if (port > 0 && port < 65536)
115             {
116               rxpd_socket_new_tcp4 (rxpd, NULL, port);
117               // TODO error handling
118             }
119           else
120             {
121               rxpd_log (rxpd, LOG_ALERT, "Illegal port number\n");
122               exit (EXIT_FAILURE);
123             }
124         }
125         // TODO rxpd_socket_new_tcp6 (rxpd, NULL, 2374)
126         break;
127 #if 0 /*not yet implemented*/
128       case 'u':
129         //rxpd_socket_new_unix (rxpd, NULL, 2374);
130         break;
131 #endif
132       case 'p':
133         if (!rxpd->policy)
134           rxpd->policy = rxpd_file_new (rxpd, optarg);
135         else
136           rxpd_die ("policy already set\n");
137         break;
138       case 'i':
139         rxpd->regflags |= REG_ICASE; 
140         break;
141 #if 0 /*not yet implemented*/
142       case '4':
143         break;
144       case '6':
145         break;
146       case 'r':
147         break;
148       case 'l:':
149         break;
150 #endif
151       case 'h':
152         usage ();
153         exit (0);
154         break;
155       default:
156         rxpd_log (rxpd, LOG_ALERT, "Unknown option\n");
157         exit (EXIT_FAILURE);
158       }
159
160   for (int i = optind; i < argc; ++i)
161     {
162       if (!rxpd_file_load (rxpd_file_new (rxpd, argv[i])))
163         {
164           //rxpd_log (rxpd, LOG_ALERT, "Failed loading file '%s'\n");
165           exit (EXIT_FAILURE);
166         }
167     }
168
169   if (rxpd->daemonize && daemon(0, 0))
170     rxpd_die ("Couldn't daemonize\n");
171
172   if (!rxpd->basedir)
173     {
174       rxpd_log (rxpd, LOG_ALERT, "Basedir not set (use -b BASEDIR)\n");
175       exit (EXIT_FAILURE);
176     }
177
178   if (llist_is_empty (&rxpd->sockets))
179     {
180       rxpd_log (rxpd, LOG_ALERT, "No listening sockets (use -t TCP or -u UNIX)\n");
181       exit (EXIT_FAILURE);
182     }
183
184   if (rxpd->policy && !rxpd_file_load (rxpd->policy))
185     {
186       rxpd_log (rxpd, LOG_ALERT, "Failed loading policy file\n");
187       exit (EXIT_FAILURE);
188     }
189
190   LLIST_FOREACH (&rxpd->sockets, n)
191     {
192       struct rxpd_socket* socket = (struct rxpd_socket*)n;
193       rxpd_socket_schedule (socket);
194     }
195
196
197   // eventloop
198   event_dispatch ();
199
200   //
201   rxpd_destroy ();
202   printf ("OK\n");
203 }