olsr: add olsr_realloc function
[olsrd.git] / src / olsr.c
1 /*
2  * The olsr.org Optimized Link-State Routing daemon (olsrd)
3  *
4  * (c) by the OLSR project
5  *
6  * See our Git repository to find out who worked on this file
7  * and thus is a copyright holder on it.
8  *
9  * All rights reserved.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  *
15  * * Redistributions of source code must retain the above copyright
16  *   notice, this list of conditions and the following disclaimer.
17  * * Redistributions in binary form must reproduce the above copyright
18  *   notice, this list of conditions and the following disclaimer in
19  *   the documentation and/or other materials provided with the
20  *   distribution.
21  * * Neither the name of olsr.org, olsrd nor the names of its
22  *   contributors may be used to endorse or promote products derived
23  *   from this software without specific prior written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
26  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
27  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
28  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
29  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
30  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
31  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
32  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
33  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
35  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36  * POSSIBILITY OF SUCH DAMAGE.
37  *
38  * Visit http://www.olsr.org for more information.
39  *
40  * If you find this software useful feel free to make a donation
41  * to the project. For more information see the website or contact
42  * the copyright holders.
43  *
44  */
45
46 /**
47  * All these functions are global
48  */
49
50 #include "defs.h"
51 #include "builddata.h"
52 #include "olsr.h"
53 #include "link_set.h"
54 #include "two_hop_neighbor_table.h"
55 #include "tc_set.h"
56 #include "duplicate_set.h"
57 #include "mpr_selector_set.h"
58 #include "mid_set.h"
59 #include "mpr.h"
60 #include "lq_mpr.h"
61 #include "olsr_spf.h"
62 #include "scheduler.h"
63 #include "apm.h"
64 #include "misc.h"
65 #include "neighbor_table.h"
66 #include "log.h"
67 #include "lq_packet.h"
68 #include "common/avl.h"
69 #include "net_olsr.h"
70 #include "lq_plugin.h"
71 #include "gateway.h"
72 #include "duplicate_handler.h"
73 #include "olsr_random.h"
74
75 #include <stdarg.h>
76 #include <signal.h>
77 #include <unistd.h>
78
79 bool changes_topology;
80 bool changes_neighborhood;
81 bool changes_hna;
82 bool changes_force;
83
84 /*COLLECT startup sleeps caused by warnings*/
85
86 #ifdef OLSR_COLLECT_STARTUP_SLEEP
87 static int max_startup_sleep = 0;
88 #endif /* OLSR_COLLECT_STARTUP_SLEEP */
89 static int sum_startup_sleep = 0;
90
91 void olsr_startup_sleep(int s)
92 {
93   sum_startup_sleep += s;
94 #ifdef OLSR_COLLECT_STARTUP_SLEEP
95   if (s > max_startup_sleep) max_startup_sleep=s;
96 #else /* OLSR_COLLECT_STARTUP_SLEEP */
97   sleep(s);
98 #endif /* OLSR_COLLECT_STARTUP_SLEEP */
99 }
100
101 void olsr_do_startup_sleep(void)
102 {
103 #ifdef OLSR_COLLECT_STARTUP_SLEEP
104   if (sum_startup_sleep > max_startup_sleep)
105     printf("OLSR encountered multiple problems on startup, which should delay startup by %i seconds.\nAs this is quite much time, OLSR will sleep only %i seconds.\nBUT YOU SHOULD FIX ABOVE PROBLEMS!\n",
106            sum_startup_sleep,max_startup_sleep);
107   sleep(max_startup_sleep);
108 #else /* OLSR_COLLECT_STARTUP_SLEEP */
109   if (sum_startup_sleep > 0) 
110     printf("olsrd startup was delayed %i seconds due to various nasty error messages.\nYOU SHOULD REALLY FIX ABOVE PROBLEMS!\n",
111            sum_startup_sleep);
112 #endif /* OLSR_COLLECT_STARTUP_SLEEP */
113 }
114
115 /**
116  * Process changes functions
117  */
118
119 struct pcf {
120   int (*function) (int, int, int);
121   struct pcf *next;
122 };
123
124 static struct pcf *pcf_list;
125
126 static uint16_t message_seqno;
127 union olsr_ip_addr all_zero;
128
129 /**
130  *Initialize the message sequence number as a random value
131  */
132 void
133 init_msg_seqno(void)
134 {
135   message_seqno = olsr_random() & 0xFFFF;
136 }
137
138 /**
139  * Get and increment the message sequence number
140  *
141  *@return the seqno
142  */
143 uint16_t
144 get_msg_seqno(void)
145 {
146   return message_seqno++;
147 }
148
149 bool
150 olsr_is_bad_duplicate_msg_seqno(uint16_t seqno) {
151   int32_t diff = (int32_t) seqno - (int32_t) message_seqno;
152
153   if (diff < -32768) {
154     diff += 65536;
155   }
156   else if (diff > 32767) {
157     diff -= 65536;
158   }
159   return diff > 0;
160 }
161
162 void
163 register_pcf(int (*f) (int, int, int))
164 {
165   struct pcf *new_pcf;
166
167   OLSR_PRINTF(1, "Registering pcf function\n");
168
169   new_pcf = olsr_malloc(sizeof(struct pcf), "New PCF");
170
171   new_pcf->function = f;
172   new_pcf->next = pcf_list;
173   pcf_list = new_pcf;
174
175 }
176
177 /**
178  *Process changes in neighborhood or/and topology.
179  *Re-calculates the neighborhood/topology if there
180  *are any updates - then calls the right functions to
181  *update the routing table.
182  *@return 0
183  */
184 void
185 olsr_process_changes(void)
186 {
187   struct pcf *tmp_pc_list;
188
189 #ifdef DEBUG
190   if (changes_neighborhood)
191     OLSR_PRINTF(3, "CHANGES IN NEIGHBORHOOD\n");
192   if (changes_topology)
193     OLSR_PRINTF(3, "CHANGES IN TOPOLOGY\n");
194   if (changes_hna)
195     OLSR_PRINTF(3, "CHANGES IN HNA\n");
196 #endif /* DEBUG */
197
198   if (!changes_neighborhood && !changes_topology && !changes_hna)
199     return;
200
201   if (olsr_cnf->debug_level > 0 && olsr_cnf->clear_screen && isatty(1)) {
202     clear_console();
203     printf("       *** %s (%s on %s) ***\n", olsrd_version, build_date, build_host);
204   }
205
206   if (changes_neighborhood) {
207     if (olsr_cnf->lq_level < 1) {
208       olsr_calculate_mpr();
209     } else {
210       olsr_calculate_lq_mpr();
211     }
212   }
213
214   /* calculate the routing table */
215   if (changes_neighborhood || changes_topology || changes_hna) {
216     olsr_calculate_routing_table(false);
217   }
218
219   if (olsr_cnf->debug_level > 0) {
220     if (olsr_cnf->debug_level > 2) {
221       olsr_print_mid_set();
222 #ifdef __linux__
223     olsr_print_gateway_entries();
224 #endif /* __linux__ */
225
226       if (olsr_cnf->debug_level > 3) {
227         if (olsr_cnf->debug_level > 8) {
228           olsr_print_duplicate_table();
229         }
230         olsr_print_hna_set();
231       }
232     }
233     olsr_print_link_set();
234     olsr_print_neighbor_table();
235     olsr_print_two_hop_neighbor_table();
236     if (olsr_cnf->debug_level > 3) {
237       olsr_print_tc_table();
238     }
239   }
240
241   for (tmp_pc_list = pcf_list; tmp_pc_list != NULL; tmp_pc_list = tmp_pc_list->next) {
242     tmp_pc_list->function(changes_neighborhood, changes_topology, changes_hna);
243   }
244
245   changes_neighborhood = false;
246   changes_topology = false;
247   changes_hna = false;
248   changes_force = false;
249 }
250
251 /**
252  *Initialize all the tables used(neighbor,
253  *topology, MID,  HNA, MPR, dup).
254  *Also initalizes other variables
255  */
256 void
257 olsr_init_tables(void)
258 {
259   changes_topology = false;
260   changes_neighborhood = false;
261   changes_hna = false;
262
263   /* Set avl tree comparator */
264   if (olsr_cnf->ipsize == 4) {
265     avl_comp_default = avl_comp_ipv4;
266     avl_comp_prefix_default = avl_comp_ipv4_prefix;
267   } else {
268     avl_comp_default = avl_comp_ipv6;
269     avl_comp_prefix_default = avl_comp_ipv6_prefix;
270   }
271
272   /* Initialize lq plugin set */
273   init_lq_handler_tree();
274
275   /* Initialize link set */
276   olsr_init_link_set();
277
278   /* Initialize duplicate table */
279   olsr_init_duplicate_set();
280
281   /* Initialize neighbor table */
282   olsr_init_neighbor_table();
283
284   /* Initialize routing table */
285   olsr_init_routing_table();
286
287   /* Initialize two hop table */
288   olsr_init_two_hop_table();
289
290   /* Initialize topology */
291   olsr_init_tc();
292
293   /* Initialize mpr selector table */
294   olsr_init_mprs_set();
295
296   /* Initialize MID set */
297   olsr_init_mid_set();
298
299   /* Initialize HNA set */
300   olsr_init_hna_set();
301
302   /* Initialize duplicate handler */
303 #ifndef NO_DUPLICATE_DETECTION_HANDLER
304   olsr_duplicate_handler_init();
305 #endif /* NO_DUPLICATE_DETECTION_HANDLER */
306 }
307
308 /**
309  *Check if a message is to be forwarded and forward
310  *it if necessary.
311  *
312  *@param m the OLSR message to be forwarded
313  *@param in_if the incoming interface
314  *@param from_addr neighbour we received message from
315  *
316  *@returns positive if forwarded
317  */
318 int
319 olsr_forward_message(union olsr_message *m, struct interface_olsr *in_if, union olsr_ip_addr *from_addr)
320 {
321   union olsr_ip_addr *src;
322   struct neighbor_entry *neighbor;
323   int msgsize;
324   struct interface_olsr *ifn;
325   bool is_ttl_1 = false;
326
327   /*
328    * Sven-Ola: We should not flood the mesh with overdue messages. Because
329    * of a bug in parser.c:parse_packet, we have a lot of messages because
330    * all older olsrd's have lq_fish enabled.
331    */
332   if (AF_INET == olsr_cnf->ip_version) {
333     if (m->v4.ttl < 2 || 255 < (int)m->v4.hopcnt + (int)m->v4.ttl)
334       is_ttl_1 = true;
335   } else {
336     if (m->v6.ttl < 2 || 255 < (int)m->v6.hopcnt + (int)m->v6.ttl)
337       is_ttl_1 = true;
338   }
339
340   /* Lookup sender address */
341   src = mid_lookup_main_addr(from_addr);
342   if (!src)
343     src = from_addr;
344
345   neighbor = olsr_lookup_neighbor_table(src);
346   if (!neighbor)
347     return 0;
348
349   if (neighbor->status != SYM)
350     return 0;
351
352   /* Check MPR */
353   if (olsr_lookup_mprs_set(src) == NULL) {
354 #ifdef DEBUG
355     struct ipaddr_str buf;
356     OLSR_PRINTF(5, "Forward - sender %s not MPR selector\n", olsr_ip_to_string(&buf, src));
357 #endif /* DEBUG */
358     return 0;
359   }
360
361   if (olsr_message_is_duplicate(m)) {
362     return 0;
363   }
364
365   /* Treat TTL hopcnt except for ethernet link */
366   if (!is_ttl_1) {
367     if (olsr_cnf->ip_version == AF_INET) {
368       /* IPv4 */
369       m->v4.hopcnt++;
370       m->v4.ttl--;
371     } else {
372       /* IPv6 */
373       m->v6.hopcnt++;
374       m->v6.ttl--;
375     }
376   }
377
378   /* Update packet data */
379   msgsize = ntohs(m->v4.olsr_msgsize);
380
381   /* looping trough interfaces */
382   for (ifn = ifnet; ifn; ifn = ifn->int_next) {
383     /* do not retransmit out through the same interface if it has mode == ether */
384     if (ifn == in_if && ifn->mode == IF_MODE_ETHER) continue;
385
386     /* do not forward TTL 1 messages to non-ether interfaces */
387     if (is_ttl_1 && ifn->mode != IF_MODE_ETHER) continue;
388
389     if (net_output_pending(ifn)) {
390       /*
391        * Check if message is to big to be piggybacked
392        */
393       if (net_outbuffer_push(ifn, m, msgsize) != msgsize) {
394         /* Send */
395         net_output(ifn);
396         /* Buffer message */
397         set_buffer_timer(ifn);
398
399         if (net_outbuffer_push(ifn, m, msgsize) != msgsize) {
400           OLSR_PRINTF(1, "Received message to big to be forwarded in %s(%d bytes)!", ifn->int_name, msgsize);
401           olsr_syslog(OLSR_LOG_ERR, "Received message to big to be forwarded on %s(%d bytes)!", ifn->int_name, msgsize);
402         }
403       }
404     } else {
405       /* No forwarding pending */
406       set_buffer_timer(ifn);
407
408       if (net_outbuffer_push(ifn, m, msgsize) != msgsize) {
409         OLSR_PRINTF(1, "Received message to big to be forwarded in %s(%d bytes)!", ifn->int_name, msgsize);
410         olsr_syslog(OLSR_LOG_ERR, "Received message to big to be forwarded on %s(%d bytes)!", ifn->int_name, msgsize);
411       }
412     }
413   }
414   return 1;
415 }
416
417 void
418 set_buffer_timer(struct interface_olsr *ifn)
419 {
420   /* Set timer */
421   ifn->fwdtimer = GET_TIMESTAMP(olsr_random() * olsr_cnf->max_jitter * MSEC_PER_SEC / OLSR_RANDOM_MAX);
422 }
423
424 void
425 olsr_init_willingness(void)
426 {
427   if (olsr_cnf->willingness_auto) {
428
429     /* Run it first and then periodic. */
430     olsr_update_willingness(NULL);
431
432     olsr_start_timer((unsigned int)olsr_cnf->will_int * MSEC_PER_SEC, 5, OLSR_TIMER_PERIODIC, &olsr_update_willingness, NULL, 0);
433   }
434 }
435
436 void
437 olsr_update_willingness(void *foo __attribute__ ((unused)))
438 {
439   int tmp_will = olsr_cnf->willingness;
440
441   /* Re-calculate willingness */
442   olsr_cnf->willingness = olsr_calculate_willingness();
443
444   if (tmp_will != olsr_cnf->willingness) {
445     OLSR_PRINTF(1, "Local willingness updated: old %d new %d\n", tmp_will, olsr_cnf->willingness);
446   }
447 }
448
449 /**
450  *Calculate this nodes willingness to act as a MPR
451  *based on either a fixed value or the power status
452  *of the node using APM
453  *
454  *@return a 8bit value from 0-7 representing the willingness
455  */
456
457 uint8_t
458 olsr_calculate_willingness(void)
459 {
460   struct olsr_apm_info ainfo;
461
462   /* If fixed willingness */
463   if (!olsr_cnf->willingness_auto)
464     return olsr_cnf->willingness;
465
466   if (apm_read(&ainfo) < 1)
467     return WILL_DEFAULT;
468
469   apm_printinfo(&ainfo);
470
471   /* If AC powered */
472   if (ainfo.ac_line_status == OLSR_AC_POWERED)
473     return 6;
474
475   /* If battery powered
476    *
477    * juice > 78% will: 3
478    * 78% > juice > 26% will: 2
479    * 26% > juice will: 1
480    */
481   return (ainfo.battery_percentage / 26);
482 }
483
484 const char *
485 olsr_msgtype_to_string(uint8_t msgtype)
486 {
487   static char type[20];
488
489   switch (msgtype) {
490   case (HELLO_MESSAGE):
491     return "HELLO";
492   case (TC_MESSAGE):
493     return "TC";
494   case (MID_MESSAGE):
495     return "MID";
496   case (HNA_MESSAGE):
497     return "HNA";
498   case (LQ_HELLO_MESSAGE):
499     return ("LQ-HELLO");
500   case (LQ_TC_MESSAGE):
501     return ("LQ-TC");
502   default:
503     break;
504   }
505
506   snprintf(type, sizeof(type), "UNKNOWN(%d)", msgtype);
507   return type;
508 }
509
510 const char *
511 olsr_link_to_string(uint8_t linktype)
512 {
513   static char type[20];
514
515   switch (linktype) {
516   case (UNSPEC_LINK):
517     return "UNSPEC";
518   case (ASYM_LINK):
519     return "ASYM";
520   case (SYM_LINK):
521     return "SYM";
522   case (LOST_LINK):
523     return "LOST";
524   case (HIDE_LINK):
525     return "HIDE";
526   default:
527     break;
528   }
529
530   snprintf(type, sizeof(type), "UNKNOWN(%d)", linktype);
531   return type;
532 }
533
534 const char *
535 olsr_status_to_string(uint8_t status)
536 {
537   static char type[20];
538
539   switch (status) {
540   case (NOT_NEIGH):
541     return "NOT NEIGH";
542   case (SYM_NEIGH):
543     return "NEIGHBOR";
544   case (MPR_NEIGH):
545     return "MPR";
546   default:
547     break;
548   }
549
550   snprintf(type, sizeof(type), "UNKNOWN(%d)", status);
551   return type;
552 }
553
554 /**
555  *Termination function to be called whenever a error occures
556  *that requires the daemon to terminate
557  *
558  *@param msg the message to write to the syslog and possibly stdout
559  *@param val the exit code
560  */
561
562 void
563 olsr_exit(const char *msg, int val)
564 {
565   if (msg) {
566     OLSR_PRINTF(1, "OLSR EXIT: %s\n", msg);
567     olsr_syslog(OLSR_LOG_ERR, "olsrd exit: %s\n", msg);
568   }
569   fflush(stdout);
570   fflush(stderr);
571
572   if (olsr_cnf) {
573     olsr_cnf->exit_value = val;
574   }
575
576   raise(SIGTERM);
577
578   /* in case the signal handler was not setup yet */
579   exit(val);
580 }
581
582 /**
583  * Wrapper for malloc(3) that does error-checking
584  *
585  * @param size the number of bytes to allocate
586  * @param id a string identifying the caller for
587  * use in error messaging
588  *
589  * @return a void pointer to the memory allocated
590  */
591 void *
592 olsr_malloc(size_t size, const char *id)
593 {
594   void *ptr;
595
596   /*
597    * Not all the callers do a proper cleaning of memory.
598    * Clean it on behalf of those.
599    */
600   ptr = calloc(1, size);
601
602   if (!ptr) {
603     char buf[1024];
604     snprintf(buf, sizeof(buf), "%s: out of memory!: %s\n", id, strerror(errno));
605     olsr_exit(buf, EXIT_FAILURE);
606   }
607
608   return ptr;
609 }
610
611 /**
612  * Wrapper for realloc(3) that does error-checking
613  *
614  * @param ptr pointer to the buffer
615  * @param size the number of bytes to (re)allocate
616  * @param id a string identifying the caller for
617  * use in error messaging
618  *
619  * @return a void pointer to the memory allocated
620  */
621 void *
622 olsr_realloc(void * ptr, size_t size, const char *id)
623 {
624   ptr = realloc(ptr, size);
625   if (!ptr) {
626     char buf[1024];
627     snprintf(buf, sizeof(buf), "%s: out of memory!: %s\n", id, strerror(errno));
628     olsr_exit(buf, EXIT_FAILURE);
629   }
630
631   return ptr;
632 }
633
634 /**
635  *Wrapper for printf that prints to a specific
636  *debuglevel upper limit
637  *
638  */
639
640 int
641 olsr_printf(int loglevel, const char *format, ...)
642 {
643   if ((loglevel <= olsr_cnf->debug_level) && debug_handle) {
644     va_list arglist;
645     va_start(arglist, format);
646     vfprintf(debug_handle, format, arglist);
647     va_end(arglist);
648   }
649   return 0;
650 }
651
652 /*
653  * Local Variables:
654  * c-basic-offset: 2
655  * indent-tabs-mode: nil
656  * End:
657  */