More cleanup of OS-specific interface, this times with win32
[olsrd.git] / src / main.c
1
2 /*
3  * The olsr.org Optimized Link-State Routing daemon(olsrd)
4  * Copyright (c) 2004-2009, the olsr.org team - see HISTORY file
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  * * Redistributions of source code must retain the above copyright
12  *   notice, this list of conditions and the following disclaimer.
13  * * Redistributions in binary form must reproduce the above copyright
14  *   notice, this list of conditions and the following disclaimer in
15  *   the documentation and/or other materials provided with the
16  *   distribution.
17  * * Neither the name of olsr.org, olsrd nor the names of its
18  *   contributors may be used to endorse or promote products derived
19  *   from this software without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
25  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
29  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
31  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32  * POSSIBILITY OF SUCH DAMAGE.
33  *
34  * Visit http://www.olsr.org for more information.
35  *
36  * If you find this software useful feel free to make a donation
37  * to the project. For more information see the website or contact
38  * the copyright holders.
39  *
40  */
41
42 #include <unistd.h>
43 #include <signal.h>
44 #include <sys/stat.h>
45 #include <assert.h>
46 #include <errno.h>
47
48 #include "defs.h"
49 #include "common/avl.h"
50 #include "common/avl_olsr_comp.h"
51 #include "olsr.h"
52 #include "ipcalc.h"
53 #include "scheduler.h"
54 #include "parser.h"
55 #include "plugin_loader.h"
56 #include "apm.h"
57 #include "net_olsr.h"
58 #include "olsr_cfg_gen.h"
59 #include "common/string.h"
60 #include "mid_set.h"
61 #include "duplicate_set.h"
62 #include "olsr_comport.h"
63 #include "neighbor_table.h"
64 #include "olsr_logging.h"
65 #include "os_net.h"
66 #include "os_kernel_routes.h"
67
68 #if defined linux
69 #include <linux/types.h>
70 #include <linux/rtnetlink.h>
71 #include <fcntl.h>
72 #endif
73
74 #define STDOUT_PULSE_INT 600    /* msec */
75
76 #ifdef WIN32
77 int __stdcall SignalHandler(unsigned long signo);
78 #else
79 static void signal_shutdown(int);
80 #endif
81 static void olsr_shutdown(void);
82
83 /*
84  * Local function prototypes
85  */
86 #ifndef WIN32
87 static void signal_reconfigure(int);
88 #endif
89
90 /* Global stuff externed in olsr_cfg.h */
91 struct olsr_config *olsr_cnf;          /* The global configuration */
92
93 volatile enum app_state app_state = STATE_INIT;
94
95 static char copyright_string[] __attribute__ ((unused)) =
96   "The olsr.org Optimized Link-State Routing daemon(olsrd) Copyright (c) 2004, Andreas Tonnesen(andreto@olsr.org) All rights reserved.";
97
98 static char pulsedata[] = "\\|/-";
99 static uint8_t pulse_state = 0;
100
101 static struct timer_entry *hna_gen_timer;
102 static struct timer_entry *mid_gen_timer;
103 static struct timer_entry *tc_gen_timer;
104
105 static void
106 generate_stdout_pulse(void *foo __attribute__ ((unused)))
107 {
108   if (pulsedata[++pulse_state] == '\0') {
109     pulse_state = 0;
110   }
111   fprintf(stderr, "%c\r", pulsedata[pulse_state]);
112 }
113
114 /**
115  * Main entrypoint
116  */
117
118 int
119 main(int argc, char *argv[])
120 {
121   /* Some cookies for stats keeping */
122   static struct olsr_timer_info *pulse_timer_info = NULL;
123   static struct olsr_timer_info *tc_gen_timer_info = NULL;
124   static struct olsr_timer_info *mid_gen_timer_info = NULL;
125   static struct olsr_timer_info *hna_gen_timer_info = NULL;
126
127   char conf_file_name[FILENAME_MAX];
128   char parse_msg[FILENAME_MAX + 256];
129   int exitcode = 0;
130 #if !defined(REMOVE_LOG_INFO) || !defined(REMOVE_LOG_ERROR)
131   struct ipaddr_str buf;
132 #endif
133 #ifdef WIN32
134   WSADATA WsaData;
135   size_t len;
136 #endif
137
138   /* paranoia checks */
139   assert(sizeof(uint8_t) == 1);
140   assert(sizeof(uint16_t) == 2);
141   assert(sizeof(uint32_t) == 4);
142   assert(sizeof(int8_t) == 1);
143   assert(sizeof(int16_t) == 2);
144   assert(sizeof(int32_t) == 4);
145
146   setbuf(stdout, NULL);
147   setbuf(stderr, NULL);
148
149   /* Using PID as random seed */
150   srandom(getpid());
151
152   /*
153    * Set default configfile name
154    */
155 #ifdef WIN32
156 #ifndef WINCE
157   GetWindowsDirectory(conf_file_name, sizeof(conf_file_name) - 1 - strlen(OLSRD_CONF_FILE_NAME));
158 #else
159   conf_file_name[0] = 0;
160 #endif
161
162   len = strlen(conf_file_name);
163
164   if (len == 0 || conf_file_name[len - 1] != '\\') {
165     conf_file_name[len++] = '\\';
166   }
167
168   strscpy(conf_file_name + len, OLSRD_CONF_FILE_NAME, sizeof(conf_file_name) - len);
169 #else
170   strscpy(conf_file_name, OLSRD_GLOBAL_CONF_FILE, sizeof(conf_file_name));
171 #endif
172
173   /*
174    * set up configuration prior to processing commandline options
175    */
176   switch (olsr_parse_cfg(argc, argv, conf_file_name, parse_msg, &olsr_cnf)) {
177   case CFG_ERROR:
178     if (parse_msg[0])
179       fprintf(stderr, "Error: %s\n", parse_msg);
180     exit(EXIT_FAILURE);
181     break;
182   case CFG_WARN:
183     if (parse_msg[0])
184       fprintf(stderr, "Warning: %s\n", parse_msg);
185     /* No exit */
186     break;
187   case CFG_EXIT:
188     exit(EXIT_SUCCESS);
189     break;
190   case CFG_OK:
191     /* Continue */
192     break;
193   }                             /* switch */
194
195   OLSR_INFO(LOG_MAIN, "\n *** %s ***\n Build date: %s on %s\n http://www.olsr.org\n\n", olsrd_version, build_date, build_host);
196
197   /* Sanity check configuration */
198   if (olsr_sanity_check_cfg(olsr_cnf) < 0) {
199     olsr_exit(EXIT_FAILURE);
200   }
201
202   /* set global interface options */
203   os_init_global_ifoptions();
204
205 #ifndef WIN32
206   /* Check if user is root */
207   if (geteuid()) {
208     fprintf(stderr, "You must be root(uid = 0) to run olsrd!\nExiting\n\n");
209     exit(EXIT_FAILURE);
210   }
211 #else
212   if (WSAStartup(0x0202, &WsaData)) {
213     fprintf(stderr, "Could not initialize WinSock.\n");
214     olsr_exit(EXIT_FAILURE);
215   }
216 #endif
217
218   /* Set avl tree comparator */
219   if (olsr_cnf->ipsize == 4) {
220     avl_comp_default = avl_comp_ipv4;
221     avl_comp_addr_origin_default = avl_comp_ipv4_addr_origin;
222     avl_comp_prefix_default = avl_comp_ipv4_prefix;
223     avl_comp_prefix_origin_default = avl_comp_ipv4_prefix_origin;
224   } else {
225     avl_comp_default = avl_comp_ipv6;
226     avl_comp_addr_origin_default = avl_comp_ipv6_addr_origin;
227     avl_comp_prefix_default = avl_comp_ipv6_prefix;
228     avl_comp_prefix_origin_default = avl_comp_ipv6_prefix_origin;
229   }
230
231   /* initialize logging */
232   olsr_log_init();
233
234   /* initialize cookie system */
235   olsr_cookie_init();
236
237   /* Initialize timers and scheduler part */
238   olsr_init_timers();
239
240   pulse_timer_info = olsr_alloc_timerinfo("Stdout pulse", &generate_stdout_pulse, true);
241   tc_gen_timer_info = olsr_alloc_timerinfo("TC generation", &olsr_output_lq_tc, true);
242   mid_gen_timer_info = olsr_alloc_timerinfo("MID generation", &generate_mid, true);
243   hna_gen_timer_info = olsr_alloc_timerinfo("HNA generation", &generate_hna, true);
244
245   /* initialize plugin system */
246   olsr_init_pluginsystem();
247   olsr_plugins_init(true);
248
249   /* initialisation of different tables to be used. */
250   olsr_init_tables();
251
252   /* enable lq-plugins */
253   olsr_plugins_enable(PLUGIN_TYPE_LQ, true);
254
255   /* initialize built in server services */
256   olsr_com_init(true);
257
258   /* Initialize net */
259   init_net();
260
261   /* Initialize SPF */
262   olsr_init_spf();
263
264   /*
265    * socket for ioctl calls
266    */
267   olsr_cnf->ioctl_s = socket(olsr_cnf->ip_version, SOCK_DGRAM, 0);
268   if (olsr_cnf->ioctl_s < 0) {
269     OLSR_ERROR(LOG_MAIN, "ioctl socket: %s\n", strerror(errno));
270     olsr_exit(EXIT_FAILURE);
271   }
272 #if defined linux
273   olsr_cnf->rtnl_s = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE);
274   if (olsr_cnf->rtnl_s < 0) {
275     OLSR_ERROR(LOG_MAIN, "rtnetlink socket: %s\n", strerror(errno));
276     olsr_exit(EXIT_FAILURE);
277   }
278   os_socket_set_nonblocking(olsr_cnf->rtnl_s);
279
280   /* Create rule for RtTable to resolve route insertion problems*/
281   if ( ( olsr_cnf->rt_table < 253) & ( olsr_cnf->rt_table > 0 ) ) {
282     OLSR_WARN(LOG_NETWORKING,"make sure to have correct policy routing rules (destination based rules are required, or a dummy rule with prio like 65535)");
283     /*olsr_netlink_rule(olsr_cnf->ip_version, olsr_cnf->rt_table, RTM_NEWRULE);*/
284   }
285 #endif
286
287 /*
288  * create routing socket
289  */
290 #if defined __FreeBSD__ || defined __MacOSX__ || defined __NetBSD__ || defined __OpenBSD__
291   olsr_cnf->rts_bsd = socket(PF_ROUTE, SOCK_RAW, 0);
292   if (olsr_cnf->rts_bsd < 0) {
293     OLSR_ERROR(LOG_MAIN, "routing socket: %s\n", strerror(errno));
294     olsr_exit(EXIT_FAILURE);
295   }
296 #endif
297
298   /* Initialize parser */
299   olsr_init_parser();
300
301   /* Initialize route-exporter */
302   olsr_init_export_route();
303
304   /* Initialize message sequencnumber */
305   init_msg_seqno();
306
307   /* Initialize dynamic willingness calculation */
308   olsr_init_willingness();
309
310   /*
311    *Set up willingness/APM
312    */
313   if (olsr_cnf->willingness_auto) {
314     if (apm_init() < 0) {
315       OLSR_INFO(LOG_MAIN, "Could not read APM info - setting default willingness(%d)\n", WILL_DEFAULT);
316
317       olsr_cnf->willingness_auto = 0;
318       olsr_cnf->willingness = WILL_DEFAULT;
319     } else {
320 #if REMOVE_LOG_INFO != 1
321       struct millitxt_buf tbuf;
322 #endif
323       olsr_cnf->willingness = olsr_calculate_willingness();
324
325       OLSR_INFO(LOG_MAIN, "Willingness set to %d - next update in %s secs\n",
326           olsr_cnf->willingness, olsr_milli_to_txt(&tbuf, olsr_cnf->will_int));
327     }
328   }
329
330 #if defined linux
331   /* Initializing lo:olsr if necessary */
332   if (olsr_cnf->source_ip_mode) {
333     OLSR_INFO(LOG_NETWORKING, "Initializing lo:olsr interface for source ip mode...\n");
334     if (olsr_os_localhost_if(&olsr_cnf->router_id, true) <= 0) {
335       OLSR_ERROR(LOG_NETWORKING, "Cannot create lo:olsr interface for ip '%s'\n", olsr_ip_to_string(&buf, &olsr_cnf->router_id));
336       olsr_exit(EXIT_FAILURE);
337     }
338   }
339 #endif
340
341   /* Initializing networkinterfaces */
342   if (!init_interfaces()) {
343     if (olsr_cnf->allow_no_interfaces) {
344       OLSR_INFO(LOG_MAIN,
345                 "No interfaces detected! This might be intentional, but it also might mean that your configuration is fubar.\nI will continue after 5 seconds...\n");
346       sleep(5);
347     } else {
348       OLSR_ERROR(LOG_MAIN, "No interfaces detected!\nBailing out!\n");
349       olsr_exit(EXIT_FAILURE);
350     }
351   }
352
353   /* Print heartbeat to stdout */
354
355 #if !defined WINCE
356   if (olsr_cnf->log_target_stderr > 0 && isatty(STDOUT_FILENO)) {
357     olsr_start_timer(STDOUT_PULSE_INT, 0, NULL, pulse_timer_info);
358   }
359 #endif
360
361   /* daemon mode */
362 #ifndef WIN32
363   if (!olsr_cnf->no_fork) {
364     OLSR_INFO(LOG_MAIN, "%s detaching from the current process...\n", olsrd_version);
365     if (daemon(0, 0) < 0) {
366       OLSR_ERROR(LOG_MAIN, "daemon(3) failed: %s\n", strerror(errno));
367       exit(EXIT_FAILURE);
368     }
369   }
370 #endif
371
372   /* activate LQ algorithm */
373   init_lq_handler();
374
375   OLSR_INFO(LOG_MAIN, "Main address: %s\n\n", olsr_ip_to_string(&buf, &olsr_cnf->router_id));
376
377   /* Start syslog entry */
378   OLSR_INFO(LOG_MAIN, "%s successfully started", olsrd_version);
379
380   /*
381    *signal-handlers
382    */
383
384   /* ctrl-C and friends */
385 #ifdef WIN32
386 #ifndef WINCE
387   SetConsoleCtrlHandler(SignalHandler, true);
388 #endif
389 #else
390   {
391     struct sigaction act;
392     sigemptyset(&act.sa_mask);
393     act.sa_flags = 0;
394     act.sa_handler = signal_reconfigure;
395     sigaction(SIGHUP, &act, NULL);
396     act.sa_handler = signal_shutdown;
397     sigaction(SIGINT, &act, NULL);
398     sigaction(SIGQUIT, &act, NULL);
399     sigaction(SIGILL, &act, NULL);
400     sigaction(SIGABRT, &act, NULL);
401 //  sigaction(SIGSEGV, &act, NULL);
402     sigaction(SIGTERM, &act, NULL);
403     act.sa_handler = SIG_IGN;
404     sigaction(SIGPIPE, &act, NULL);
405     // Ignoring SIGUSR1 and SIGUSR1 by default to be able to use them in plugins
406     sigaction(SIGUSR1, &act, NULL);
407     sigaction(SIGUSR2, &act, NULL);
408   }
409 #endif
410
411   link_changes = false;
412
413   tc_gen_timer =
414     olsr_start_timer(olsr_cnf->tc_params.emission_interval, TC_JITTER, NULL, tc_gen_timer_info);
415   mid_gen_timer =
416     olsr_start_timer(olsr_cnf->mid_params.emission_interval, MID_JITTER, NULL, mid_gen_timer_info);
417   hna_gen_timer =
418     olsr_start_timer(olsr_cnf->hna_params.emission_interval, HNA_JITTER, NULL, hna_gen_timer_info);
419
420   /* enable default plugins */
421   olsr_plugins_enable(PLUGIN_TYPE_DEFAULT, true);
422
423   /* Starting scheduler */
424   app_state = STATE_RUNNING;
425   olsr_scheduler();
426
427   olsr_stop_timer(tc_gen_timer);
428   tc_gen_timer = NULL;
429
430   olsr_stop_timer(mid_gen_timer);
431   mid_gen_timer = NULL;
432
433   olsr_stop_timer(hna_gen_timer);
434   hna_gen_timer = NULL;
435
436   exitcode = olsr_cnf->exit_value;
437   switch (app_state) {
438   case STATE_INIT:
439     OLSR_ERROR(LOG_MAIN, "terminating and got \"init\"?");
440     exitcode = EXIT_FAILURE;
441     break;
442   case STATE_RUNNING:
443     OLSR_ERROR(LOG_MAIN, "terminating and got \"running\"?");
444     exitcode = EXIT_FAILURE;
445     break;
446 #ifndef WIN32
447   case STATE_RECONFIGURE:
448     /* if we are started with -nofork, we do not weant to go into the
449      * background here. So we can simply be the child process.
450      */
451     switch (olsr_cnf->no_fork ? 0 : fork()) {
452       int i;
453     case 0:
454       /* child process */
455       for (i = sysconf(_SC_OPEN_MAX); --i > STDERR_FILENO;) {
456         close(i);
457       }
458       sleep(1);
459       OLSR_INFO(LOG_MAIN, "Restarting %s\n", argv[0]);
460       execv(argv[0], argv);
461       /* if we reach this, the exev() failed */
462       OLSR_ERROR(LOG_MAIN, "execv() failed: %s", strerror(errno));
463       /* and we simply shutdown */
464       exitcode = EXIT_FAILURE;
465       break;
466     case -1:
467       /* fork() failes */
468       OLSR_ERROR(LOG_MAIN, "fork() failed: %s", strerror(errno));
469       /* and we simply shutdown */
470       exitcode = EXIT_FAILURE;
471       break;
472     default:
473       /* parent process */
474       OLSR_INFO(LOG_MAIN, "Reconfiguring OLSR\n");
475       break;
476     }
477     /* fall through */
478 #endif
479   case STATE_SHUTDOWN:
480     olsr_shutdown();
481     break;
482   };
483
484   return exitcode;
485 }                               /* main */
486
487 #ifndef WIN32
488
489 /**
490  * Request reconfiguration of olsrd.
491  *
492  *@param signal the signal that triggered this callback
493  */
494 static void
495 signal_reconfigure(int signo __attribute__ ((unused)))
496 {
497   const int save_errno = errno;
498   OLSR_INFO(LOG_MAIN, "Received signal %d - requesting reconfiguration", signo);
499   app_state = STATE_RECONFIGURE;
500   errno = save_errno;
501 }
502
503 #endif
504
505 /**
506  *Function called at shutdown. Signal handler
507  *
508  * @param signal the signal that triggered this call
509  */
510 #ifdef WIN32
511 int __stdcall
512 SignalHandler(unsigned long signo)
513 #else
514 static void
515 signal_shutdown(int signo __attribute__ ((unused)))
516 #endif
517 {
518   const int save_errno = errno;
519   OLSR_INFO(LOG_MAIN, "Received signal %d - requesting shutdown", (int)signo);
520   app_state = STATE_SHUTDOWN;
521   errno = save_errno;
522 #ifdef WIN32
523   return 0;
524 #endif
525 }
526
527 static void
528 olsr_shutdown(void)
529 {
530   struct mid_entry *mid;
531   struct list_iterator iterator;
532
533   olsr_delete_all_kernel_routes();
534
535   /* Flush MID database */
536   OLSR_FOR_ALL_MID_ENTRIES(mid, iterator) {
537     olsr_delete_mid_entry(mid);
538   }
539
540   /* Flush TC database */
541   olsr_delete_all_tc_entries();
542
543   OLSR_INFO(LOG_MAIN, "Closing sockets...\n");
544
545   /* kill http/telnet server */
546   olsr_com_destroy();
547
548   /* Flush duplicate set */
549   olsr_flush_duplicate_entries();
550
551   /* Shut down LQ plugin */
552   deinit_lq_handler();
553
554   /* Closing plug-ins */
555   olsr_destroy_pluginsystem();
556
557   /* Remove active interfaces */
558   destroy_interfaces();
559
560 #if defined linux
561   /* delete lo:olsr if neccesarry */
562   if (olsr_cnf->source_ip_mode) {
563     olsr_os_localhost_if(&olsr_cnf->router_id, false);
564   }
565 #endif
566
567   /* Reset network settings */
568   os_cleanup_global_ifoptions();
569
570   /* ioctl socket */
571   CLOSESOCKET(olsr_cnf->ioctl_s);
572
573 #if defined linux
574   /*if ((olsr_cnf->rttable < 253) & (olsr_cnf->rttable > 0)) {
575     olsr_netlink_rule(olsr_cnf->ip_version, olsr_cnf->rttable, RTM_DELRULE);
576   }*/
577
578   CLOSESOCKET(olsr_cnf->rtnl_s);
579 #endif
580
581 #if defined __FreeBSD__ || defined __MacOSX__ || defined __NetBSD__ || defined __OpenBSD__
582   /* routing socket */
583   CLOSESOCKET(olsr_cnf->rts_bsd);
584 #endif
585
586   /* Close and delete all sockets */
587   olsr_flush_sockets();
588
589   /* Stop and delete all timers. */
590   olsr_flush_timers();
591
592   /* Remove parser hooks */
593   olsr_deinit_parser();
594
595   /* Remove IP filters */
596   deinit_netfilters();
597
598   OLSR_INFO(LOG_MAIN, "\n <<<< %s - terminating >>>>\n           http://www.olsr.org\n", olsrd_version);
599
600   olsr_log_cleanup();
601
602   /* Free cookies and memory pools attached. */
603   olsr_cookie_cleanup();
604
605   /* Flush config */
606   olsr_free_cfg(olsr_cnf);
607 }
608
609 /*
610  * Local Variables:
611  * c-basic-offset: 2
612  * indent-tabs-mode: nil
613  * End:
614  */