- Second fixing-things pass: it even links now.
[tinc] / src / tincd.c
1 /*
2     tincd.c -- the main file for tincd
3     Copyright (C) 1998,1999,2000 Ivo Timmermans <itimmermans@bigfoot.com>
4                             2000 Guus Sliepen <guus@sliepen.warande.net>
5
6     This program is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     This program is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with this program; if not, write to the Free Software
18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19
20     $Id: tincd.c,v 1.10.4.11 2000/10/14 17:04:16 guus Exp $
21 */
22
23 #include "config.h"
24
25 #include <errno.h>
26 #include <fcntl.h> 
27 #include <getopt.h>
28 #include <signal.h>
29 #include <stdio.h>
30 #include <sys/types.h>
31 #include <syslog.h>
32 #include <unistd.h>
33 #include <signal.h>
34
35 #ifdef HAVE_SYS_IOCTL_H
36 # include <sys/ioctl.h>
37 #endif
38
39 #include <pidfile.h>
40 #include <utils.h>
41 #include <xalloc.h>
42
43 #include "conf.h"
44 #include "encr.h"
45 #include "net.h"
46 #include "netutl.h"
47 #include "protocol.h"
48
49 #include "system.h"
50
51 /* The name this program was run with. */
52 char *program_name;
53
54 /* If nonzero, display usage information and exit. */
55 static int show_help;
56
57 /* If nonzero, print the version on standard output and exit.  */
58 static int show_version;
59
60 /* If nonzero, it will attempt to kill a running tincd and exit. */
61 static int kill_tincd = 0;
62
63 /* If zero, don't detach from the terminal. */
64 static int do_detach = 1;
65
66 char *identname;                 /* program name for syslog */
67 char *netname = NULL;            /* name of the vpn network */
68 char *pidfilename;               /* pid file location */
69 static pid_t ppid;               /* pid of non-detached part */
70 char **g_argv;                   /* a copy of the cmdline arguments */
71
72 void cleanup_and_exit(int);
73 int detach(void);
74 int kill_other(void);
75 void make_names(void);
76 RETSIGTYPE parent_exit(int a);
77 void setup_signals(void);
78 int write_pidfile(void);
79
80 static struct option const long_options[] =
81 {
82   { "kill", no_argument, NULL, 'k' },
83   { "net", required_argument, NULL, 'n' },
84   { "timeout", required_argument, NULL, 'p' },
85   { "help", no_argument, &show_help, 1 },
86   { "version", no_argument, &show_version, 1 },
87   { "no-detach", no_argument, &do_detach, 0 },
88   { NULL, 0, NULL, 0 }
89 };
90
91 static void
92 usage(int status)
93 {
94   if(status != 0)
95     fprintf(stderr, _("Try `%s --help\' for more information.\n"), program_name);
96   else
97     {
98       printf(_("Usage: %s [option]...\n\n"), program_name);
99       printf(_("  -c, --config=DIR      Read configuration options from DIR.\n"
100                "  -D, --no-detach       Don't fork and detach.\n"
101                "  -d                    Increase debug level.\n"
102                "  -k, --kill            Attempt to kill a running tincd and exit.\n"
103                "  -n, --net=NETNAME     Connect to net NETNAME.\n"
104                "  -t, --timeout=TIMEOUT Seconds to wait before giving a timeout.\n"));
105       printf(_("      --help            Display this help and exit.\n"
106                "      --version         Output version information and exit.\n\n"));
107       printf(_("Report bugs to tinc@nl.linux.org.\n"));
108     }
109   exit(status);
110 }
111
112 void
113 parse_options(int argc, char **argv, char **envp)
114 {
115   int r;
116   int option_index = 0;
117   config_t *p;
118
119   while((r = getopt_long(argc, argv, "c:Ddkn:t:", long_options, &option_index)) != EOF)
120     {
121       switch(r)
122         {
123         case 0: /* long option */
124           break;
125         case 'c': /* config file */
126           confbase = xmalloc(strlen(optarg)+1);
127           strcpy(confbase, optarg);
128           break;
129         case 'D': /* no detach */
130           do_detach = 0;
131           break;
132         case 'd': /* inc debug level */
133           debug_lvl++;
134           break;
135         case 'k': /* kill old tincds */
136           kill_tincd = 1;
137           break;
138         case 'n': /* net name given */
139           netname = xmalloc(strlen(optarg)+1);
140           strcpy(netname, optarg);
141           break;
142         case 't': /* timeout */
143           if(!(p = add_config_val(&config, TYPE_INT, optarg)))
144             {
145               printf(_("Invalid timeout value `%s'.\n"), optarg);
146               usage(1);
147             }
148           break;
149         case '?':
150           usage(1);
151         default:
152           break;
153         }
154     }
155 }
156
157 void memory_full(int size)
158 {
159   syslog(LOG_ERR, _("Memory exhausted (last is %s:%d) (couldn't allocate %d bytes), exiting."), cp_file, cp_line, size);
160   exit(1);
161 }
162
163 /*
164   Detach from current terminal, write pidfile, kill parent
165 */
166 int detach(void)
167 {
168   int fd;
169   pid_t pid;
170
171   if(do_detach)
172     {
173       ppid = getpid();
174
175       if((pid = fork()) < 0)
176         {
177           perror("fork");
178           return -1;
179         }
180       if(pid) /* parent process */
181         {
182           signal(SIGTERM, parent_exit);
183           sleep(600); /* wait 10 minutes */
184           exit(1);
185         }
186     }
187   
188   if(write_pidfile())
189     return -1;
190
191   if(do_detach)
192     {
193       if((fd = open("/dev/tty", O_RDWR)) >= 0)
194         {
195           if(ioctl(fd, TIOCNOTTY, NULL))
196             {
197               perror("ioctl");
198               return -1;
199             }
200           close(fd);
201         }
202
203       if(setsid() < 0)
204         return -1;
205
206       kill(ppid, SIGTERM);
207     }
208   
209   chdir("/"); /* avoid keeping a mointpoint busy */
210
211   openlog(identname, LOG_CONS | LOG_PID, LOG_DAEMON);
212
213   if(debug_lvl > 0)
214     syslog(LOG_NOTICE, _("tincd %s (%s %s) starting, debug level %d"),
215            VERSION, __DATE__, __TIME__, debug_lvl);
216   else
217     syslog(LOG_NOTICE, _("tincd %s starting"), VERSION, debug_lvl);
218
219   xalloc_fail_func = memory_full;
220
221   return 0;
222 }
223
224 /*
225   Close network connections, and terminate neatly
226 */
227 void cleanup_and_exit(int c)
228 {
229   close_network_connections();
230
231   if(debug_lvl > 0)
232     syslog(LOG_INFO, _("Total bytes written: tap %d, socket %d; bytes read: tap %d, socket %d"),
233            total_tap_out, total_socket_out, total_tap_in, total_socket_in);
234
235   closelog();
236   kill(ppid, SIGTERM);
237   exit(c);
238 }
239
240 /*
241   check for an existing tinc for this net, and write pid to pidfile
242 */
243 int write_pidfile(void)
244 {
245   int pid;
246
247   if((pid = check_pid(pidfilename)))
248     {
249       if(netname)
250         fprintf(stderr, _("A tincd is already running for net `%s' with pid %d.\n"),
251                 netname, pid);
252       else
253         fprintf(stderr, _("A tincd is already running with pid %d.\n"), pid);
254       return 1;
255     }
256
257   /* if it's locked, write-protected, or whatever */
258   if(!write_pid(pidfilename))
259     return 1;
260
261   return 0;
262 }
263
264 /*
265   kill older tincd for this net
266 */
267 int kill_other(void)
268 {
269   int pid;
270
271   if(!(pid = read_pid(pidfilename)))
272     {
273       if(netname)
274         fprintf(stderr, _("No other tincd is running for net `%s'.\n"), netname);
275       else
276         fprintf(stderr, _("No other tincd is running.\n"));
277       return 1;
278     }
279
280   errno = 0;    /* No error, sometimes errno is only changed on error */
281   /* ESRCH is returned when no process with that pid is found */
282   if(kill(pid, SIGTERM) && errno == ESRCH)
283     fprintf(stderr, _("Removing stale lock file.\n"));
284   remove_pid(pidfilename);
285
286   return 0;
287 }
288
289 /*
290   Set all files and paths according to netname
291 */
292 void make_names(void)
293 {
294   if(netname)
295     {
296       if(!pidfilename)
297         asprintf(&pidfilename, "/var/run/tinc.%s.pid", netname);
298       if(!confbase)
299         asprintf(&confbase, "%s/tinc/%s", CONFDIR, netname);
300       if(!identname)
301         asprintf(&identname, "tinc.%s", netname);
302     }
303   else
304     {
305       if(!pidfilename)
306         pidfilename = "/var/run/tinc.pid";
307       if(!confbase)
308         asprintf(&confbase, "%s/tinc", CONFDIR);
309       if(!identname)
310         identname = "tinc";
311     }
312 }
313
314 int
315 main(int argc, char **argv, char **envp)
316 {
317   program_name = argv[0];
318
319   setlocale (LC_ALL, "");
320   bindtextdomain (PACKAGE, LOCALEDIR);
321   textdomain (PACKAGE);
322
323   parse_options(argc, argv, envp);
324
325   if(show_version)
326     {
327       printf(_("%s version %s (built %s %s, protocol %d)\n"), PACKAGE, VERSION, __DATE__, __TIME__, PROT_CURRENT);
328       printf(_("Copyright (C) 1998,1999,2000 Ivo Timmermans and others,\n"
329                "see the AUTHORS file for a complete list.\n\n"
330                "tinc comes with ABSOLUTELY NO WARRANTY.  This is free software,\n"
331                "and you are welcome to redistribute it under certain conditions;\n"
332                "see the file COPYING for details.\n\n"));
333       printf(_("This product includes software developed by Eric Young (eay@mincom.oz.au)\n"));
334
335       return 0;
336     }
337
338   if(show_help)
339     usage(0);
340
341   if(geteuid())
342     {
343       fprintf(stderr, _("You must be root to run this program. Sorry.\n"));
344       return 1;
345     }
346
347   g_argv = argv;
348
349   make_names();
350
351   if(kill_tincd)
352     exit(kill_other());
353
354   if(read_server_config())
355     return 1;
356
357   setup_signals();
358
359   if(detach())
360     exit(0);
361
362 /* FIXME: wt* is this suppose to do?
363   if(security_init())
364     return 1;
365 */
366   for(;;)
367     {
368       setup_network_connections();
369
370       main_loop();
371
372       cleanup_and_exit(1);
373
374       syslog(LOG_ERR, _("Unrecoverable error, restarting in %d seconds!"), MAXTIMEOUT);
375       sleep(MAXTIMEOUT);
376     }
377 }
378
379 RETSIGTYPE
380 sigterm_handler(int a)
381 {
382   if(debug_lvl > 0)
383     syslog(LOG_NOTICE, _("Got TERM signal"));
384   cleanup_and_exit(0);
385 }
386
387 RETSIGTYPE
388 sigquit_handler(int a)
389 {
390   if(debug_lvl > 0)
391     syslog(LOG_NOTICE, _("Got QUIT signal"));
392   cleanup_and_exit(0);
393 }
394
395 RETSIGTYPE
396 sigsegv_square(int a)
397 {
398   syslog(LOG_NOTICE, _("Got another SEGV signal: not restarting"));
399   exit(0);
400 }
401
402 RETSIGTYPE
403 sigsegv_handler(int a)
404 {
405   if(cp_file)
406     syslog(LOG_NOTICE, _("Got SEGV signal after %s line %d, trying to re-execute"),
407            cp_file, cp_line);
408   else
409     syslog(LOG_NOTICE, _("Got SEGV signal, trying to re-execute"));
410
411   signal(SIGSEGV, sigsegv_square);
412   close_network_connections();
413   remove_pid(pidfilename);
414   execvp(g_argv[0], g_argv);
415 }
416
417 RETSIGTYPE
418 sighup_handler(int a)
419 {
420   if(debug_lvl > 0)
421     syslog(LOG_NOTICE, _("Got HUP signal, rereading configuration and restarting"));
422   sighup = 1;
423 }
424
425 RETSIGTYPE
426 sigint_handler(int a)
427 {
428   if(debug_lvl > 0)
429     syslog(LOG_NOTICE, _("Got INT signal, exiting"));
430   cleanup_and_exit(0);
431 }
432
433 RETSIGTYPE
434 sigusr1_handler(int a)
435 {
436   dump_conn_list();
437 }
438
439 RETSIGTYPE
440 sigusr2_handler(int a)
441 {
442   if(debug_lvl > 1)
443     syslog(LOG_NOTICE, _("Got USR2 signal, forcing new key generation"));
444 /* FIXME: reprogram this.
445   regenerate_keys();
446 */
447 }
448
449 RETSIGTYPE
450 sighuh(int a)
451 {
452   if(cp_file)
453     syslog(LOG_NOTICE, _("Got unexpected %s after %s line %d"),
454            strsignal(a), cp_file, cp_line);
455   else
456     syslog(LOG_NOTICE, _("Got unexpected %s"), strsignal(a));
457 }
458
459 void
460 setup_signals(void)
461 {
462   int i;
463
464   for(i=0;i<32;i++)
465     signal(i,sighuh);
466
467   if(signal(SIGTERM, SIG_IGN) != SIG_ERR)
468     signal(SIGTERM, sigterm_handler);
469   if(signal(SIGQUIT, SIG_IGN) != SIG_ERR)
470     signal(SIGQUIT, sigquit_handler);
471   if(signal(SIGSEGV, SIG_IGN) != SIG_ERR)
472     signal(SIGSEGV, sigsegv_handler);
473   if(signal(SIGHUP, SIG_IGN) != SIG_ERR)
474     signal(SIGHUP, sighup_handler);
475   signal(SIGPIPE, SIG_IGN);
476   if(signal(SIGINT, SIG_IGN) != SIG_ERR)
477     signal(SIGINT, sigint_handler);
478   signal(SIGUSR1, sigusr1_handler);
479   signal(SIGUSR2, sigusr2_handler);
480 //  signal(SIGCHLD, parent_exit);
481 }
482
483 RETSIGTYPE parent_exit(int a)
484 {
485   exit(0);
486 }