|
1
|
1 /***************************************************************************
|
|
|
2 * Copyright (C) 2005 by 510 Software Group *
|
|
|
3 * *
|
|
|
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 2 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, write to the *
|
|
|
17 * Free Software Foundation, Inc., *
|
|
|
18 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
|
|
|
19 ***************************************************************************/
|
|
|
20
|
|
|
21
|
|
|
22 #ifdef HAVE_CONFIG_H
|
|
|
23 #include <config.h>
|
|
|
24 #endif
|
|
|
25
|
|
|
26 #include <iostream>
|
|
|
27 #include <cstdlib>
|
|
|
28 #include <errno.h>
|
|
|
29 #include <sysexits.h>
|
|
|
30 #include <unistd.h>
|
|
|
31 #include <pthread.h>
|
|
|
32 #include <syslog.h>
|
|
|
33 #include <sys/types.h>
|
|
|
34 #include <sys/stat.h>
|
|
|
35 #include <sys/wait.h> /* header for waitpid() and various macros */
|
|
|
36 #include <signal.h> /* header for signal functions */
|
|
|
37 #include "includes.h"
|
|
|
38
|
|
2
|
39 extern "C" {
|
|
|
40 void sig_chld(int signo);
|
|
|
41 }
|
|
1
|
42 int debug_syslog = 0;
|
|
|
43 bool syslog_opened = false;
|
|
|
44 bool use_syslog = true; // false to printf
|
|
|
45 bool loader_run = true; // used to stop the config loader thread
|
|
|
46 CONFIG *config = NULL; // protected by the config_mutex
|
|
|
47 int generation = 0; // protected by the config_mutex
|
|
|
48 const int maxlen = 1000; // used for snprintf buffers
|
|
|
49
|
|
|
50 pthread_mutex_t config_mutex;
|
|
|
51 pthread_mutex_t syslog_mutex;
|
|
|
52
|
|
|
53
|
|
|
54 ////////////////////////////////////////////////
|
|
|
55 // syslog a message
|
|
|
56 //
|
|
|
57 void my_syslog(char *text) {
|
|
|
58 if (use_syslog) {
|
|
|
59 pthread_mutex_lock(&syslog_mutex);
|
|
|
60 if (!syslog_opened) {
|
|
|
61 openlog("syslog2iptables", LOG_PID, LOG_MAIL);
|
|
|
62 syslog_opened = true;
|
|
|
63 }
|
|
|
64 syslog(LOG_NOTICE, "%s", text);
|
|
|
65 pthread_mutex_unlock(&syslog_mutex);
|
|
|
66 }
|
|
|
67 else {
|
|
3
|
68 printf("%s\n", text);
|
|
1
|
69 }
|
|
|
70 }
|
|
|
71
|
|
|
72 ////////////////////////////////////////////////
|
|
|
73 // reload the config
|
|
|
74 //
|
|
|
75 CONFIG* new_conf();
|
|
|
76 CONFIG* new_conf() {
|
|
|
77 CONFIG *newc = new CONFIG;
|
|
|
78 pthread_mutex_lock(&config_mutex);
|
|
|
79 newc->generation = generation++;
|
|
|
80 pthread_mutex_unlock(&config_mutex);
|
|
|
81 if (debug_syslog) {
|
|
|
82 char buf[maxlen];
|
|
|
83 snprintf(buf, sizeof(buf), "loading configuration generation %d", newc->generation);
|
|
|
84 my_syslog(buf);
|
|
|
85 }
|
|
|
86 if (load_conf(*newc, "syslog2iptables.conf")) {
|
|
|
87 newc->load_time = time(NULL);
|
|
|
88 return newc;
|
|
|
89 }
|
|
|
90 delete newc;
|
|
|
91 return NULL;
|
|
|
92 }
|
|
|
93
|
|
|
94
|
|
|
95 ////////////////////////////////////////////////
|
|
|
96 // thread to watch the old config files for changes
|
|
|
97 // and reload when needed. we also cleanup old
|
|
|
98 // configs whose reference count has gone to zero.
|
|
|
99 //
|
|
|
100 void* config_loader(void *arg);
|
|
|
101 void* config_loader(void *arg) {
|
|
|
102 typedef set<CONFIG *> configp_set;
|
|
|
103 configp_set old_configs;
|
|
|
104 while (loader_run) {
|
|
|
105 sleep(180); // look for modifications every 3 minutes
|
|
|
106 if (!loader_run) break;
|
|
|
107 CONFIG &dc = *config;
|
|
|
108 time_t then = dc.load_time;
|
|
|
109 struct stat st;
|
|
|
110 bool reload = false;
|
|
|
111 for (string_set::iterator i=dc.config_files.begin(); i!=dc.config_files.end(); i++) {
|
|
|
112 char *fn = *i;
|
|
|
113 if (stat(fn, &st)) reload = true; // file disappeared
|
|
|
114 else if (st.st_mtime > then) reload = true; // file modified
|
|
|
115 if (reload) break;
|
|
|
116 }
|
|
|
117 if (reload) {
|
|
|
118 CONFIG *newc = new_conf();
|
|
|
119 if (newc) {
|
|
|
120 // replace the global config pointer
|
|
|
121 pthread_mutex_lock(&config_mutex);
|
|
|
122 CONFIG *old = config;
|
|
|
123 config = newc;
|
|
|
124 pthread_mutex_unlock(&config_mutex);
|
|
|
125 if (old) old_configs.insert(old);
|
|
|
126 }
|
|
|
127 else {
|
|
|
128 // failed to load new config
|
|
|
129 my_syslog("failed to load new configuration");
|
|
|
130 system("echo 'failed to load new /etc/syslog2iptables.conf' | mail -s 'error in /etc/syslog2iptables.conf' root");
|
|
|
131 // update the load time on the current config to prevent complaining every 3 minutes
|
|
|
132 dc.load_time = time(NULL);
|
|
|
133 }
|
|
|
134 }
|
|
|
135 // now look for old configs with zero ref counts
|
|
|
136 for (configp_set::iterator i=old_configs.begin(); i!=old_configs.end(); ) {
|
|
|
137 CONFIG *old = *i;
|
|
|
138 if (!old->reference_count) {
|
|
|
139 if (debug_syslog) {
|
|
|
140 char buf[maxlen];
|
|
|
141 snprintf(buf, sizeof(buf), "freeing memory for old configuration generation %d", old->generation);
|
|
|
142 my_syslog(buf);
|
|
|
143 }
|
|
|
144 delete old; // destructor does all the work
|
|
|
145 old_configs.erase(i++);
|
|
|
146 }
|
|
|
147 else i++;
|
|
|
148 }
|
|
|
149 }
|
|
|
150 return NULL;
|
|
|
151 }
|
|
|
152
|
|
|
153
|
|
|
154 ////////////////////////////////////////////////
|
|
|
155 // The signal handler function -- only gets called when a SIGCHLD
|
|
|
156 // is received, ie when a child terminates
|
|
|
157 //
|
|
|
158 void sig_chld(int signo)
|
|
|
159 {
|
|
|
160 int status;
|
|
|
161 /* Wait for any child without blocking */
|
|
|
162 while (waitpid(-1, &status, WNOHANG) > 0) {
|
|
|
163 // ignore child exit status, we only do this to cleanup zombies
|
|
|
164 }
|
|
|
165 }
|
|
|
166
|
|
|
167
|
|
|
168 void usage(char *prog);
|
|
|
169 void usage(char *prog)
|
|
|
170 {
|
|
|
171 }
|
|
|
172
|
|
|
173
|
|
|
174 int main(int argc, char *argv[])
|
|
|
175 {
|
|
|
176 token_init();
|
|
|
177 bool check = true;
|
|
|
178 int c;
|
|
|
179 const char *args = "ch";
|
|
|
180 extern char *optarg;
|
|
|
181
|
|
|
182 // Process command line options
|
|
|
183 while ((c = getopt(argc, argv, args)) != -1) {
|
|
|
184 switch (c) {
|
|
|
185 case 'c':
|
|
|
186 check = true;
|
|
|
187 break;
|
|
|
188
|
|
|
189 case 'h':
|
|
|
190 default:
|
|
|
191 usage(argv[0]);
|
|
|
192 exit(EX_USAGE);
|
|
|
193 }
|
|
|
194 }
|
|
|
195
|
|
|
196 if (check) {
|
|
|
197 use_syslog = false;
|
|
|
198 debug_syslog = 10;
|
|
|
199 CONFIG *conf = new_conf();
|
|
|
200 if (conf) {
|
|
|
201 conf->dump();
|
|
3
|
202 for (int i=0; i<2; i++) {
|
|
2
|
203 conf->read();
|
|
3
|
204 conf->sleep(5);
|
|
2
|
205 }
|
|
1
|
206 delete conf;
|
|
|
207 return 0;
|
|
|
208 }
|
|
|
209 else {
|
|
|
210 return 1; // config failed to load
|
|
|
211 }
|
|
|
212 }
|
|
|
213
|
|
|
214 // switch to background mode
|
|
|
215 if (daemon(1,0) < 0) {
|
|
|
216 fprintf(stderr, "daemon() call failed\n");
|
|
|
217 exit(EX_UNAVAILABLE);
|
|
|
218 }
|
|
|
219
|
|
|
220 // write the pid
|
|
|
221 const char *pidpath = "/var/run/syslog2iptables.pid";
|
|
|
222 unlink(pidpath);
|
|
|
223 FILE *f = fopen(pidpath, "w");
|
|
|
224 if (f) {
|
|
|
225 #ifdef linux
|
|
|
226 // from a comment in the DCC source code:
|
|
|
227 // Linux threads are broken. Signals given the
|
|
|
228 // original process are delivered to only the
|
|
|
229 // thread that happens to have that PID. The
|
|
|
230 // sendmail libmilter thread that needs to hear
|
|
|
231 // SIGINT and other signals does not, and that breaks
|
|
|
232 // scripts that need to stop milters.
|
|
|
233 // However, signaling the process group works.
|
|
|
234 fprintf(f, "-%d\n", (u_int)getpgrp());
|
|
|
235 #else
|
|
|
236 fprintf(f, "%d\n", (u_int)getpid());
|
|
|
237 #endif
|
|
|
238 fclose(f);
|
|
|
239 }
|
|
|
240
|
|
|
241 // initialize the thread sync objects
|
|
|
242 pthread_mutex_init(&config_mutex, 0);
|
|
|
243 pthread_mutex_init(&syslog_mutex, 0);
|
|
|
244
|
|
|
245 // load the initial config
|
|
|
246 config = new_conf();
|
|
|
247 if (!config) {
|
|
|
248 my_syslog("failed to load initial configuration, quitting");
|
|
|
249 exit(1);
|
|
|
250 }
|
|
|
251
|
|
2
|
252 // setup sigchld handler to prevent zombies
|
|
|
253 struct sigaction act;
|
|
|
254 act.sa_handler = sig_chld; // Assign sig_chld as our SIGCHLD handler
|
|
|
255 sigemptyset(&act.sa_mask); // We don't want to block any other signals in this example
|
|
|
256 act.sa_flags = SA_NOCLDSTOP; // only want children that have terminated
|
|
|
257 if (sigaction(SIGCHLD, &act, NULL) < 0) {
|
|
|
258 my_syslog("failed to setup SIGCHLD handler");
|
|
|
259 exit(1);
|
|
|
260 }
|
|
|
261
|
|
1
|
262 // only create threads after the fork() in daemon
|
|
|
263 pthread_t tid;
|
|
|
264 if (pthread_create(&tid, 0, config_loader, 0))
|
|
|
265 my_syslog("failed to create config loader thread");
|
|
|
266 if (pthread_detach(tid))
|
|
|
267 my_syslog("failed to detach config loader thread");
|
|
|
268
|
|
|
269 loader_run = false; // eventually the config loader thread will terminate
|
|
|
270 return EXIT_SUCCESS;
|
|
|
271 }
|
