dont change the current dir when daemonize
[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\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   for (int i = optind; i < argc; ++i)
167     {
168       if (!rxpd_file_load (rxpd_file_new (rxpd, argv[i])))
169         {
170           //rxpd_log (rxpd, LOG_ALERT, "Failed loading file '%s'\n");
171           exit (EXIT_FAILURE);
172         }
173     }
174
175   if (rxpd->daemonize && daemon(1, 0))
176     rxpd_die ("Couldn't daemonize\n");
177
178   if (!rxpd->basedir)
179     {
180       rxpd_log (rxpd, LOG_ALERT, "Basedir not set (use -b BASEDIR)\n");
181       exit (EXIT_FAILURE);
182     }
183
184   if (llist_is_empty (&rxpd->sockets))
185     {
186       rxpd_log (rxpd, LOG_ALERT, "No listening sockets (use -t TCP or -u UNIX)\n");
187       exit (EXIT_FAILURE);
188     }
189
190   if (rxpd->policy)
191     {
192       if (rxpd_file_load (rxpd->policy))
193         rxpd_log (rxpd, LOG_INFO, "Loaded policy '%s'\n", rxpd->policy->filename);
194       else
195         {
196           rxpd_log (rxpd, LOG_ALERT, "Failed loading policy file\n");
197           exit (EXIT_FAILURE);
198         }
199     }
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
208   // eventloop
209   event_dispatch ();
210
211   rxpd_log (rxpd, LOG_NOTICE, PACKAGE_STRING" exited\n");
212   rxpd_destroy ();
213   return EXIT_SUCCESS;
214 }