562f74b94d3df3ee7194b87aec2bc46bb8f4b1ea
[olsrd.git] / src / gateway_default_handler.c
1 /*
2  * gateway_default_handler.c
3  *
4  *  Created on: Jan 29, 2010
5  *      Author: rogge
6  */
7
8 #include "defs.h"
9 #include "gateway.h"
10 #include "gateway_default_handler.h"
11 #include "scheduler.h"
12 #include "tc_set.h"
13 #include "log.h"
14
15 #include "assert.h"
16
17 #ifndef WIN32
18 static uint32_t gw_def_nodecount, gw_def_stablecount;
19 static bool gw_def_finished_ipv4, gw_def_finished_ipv6;
20
21 static struct timer_entry *gw_def_timer;
22
23 static void gw_default_startup_handler(void);
24 static void gw_default_choosegw_handler(bool ipv4, bool ipv6);
25 static void gw_default_update_handler(struct gateway_entry *);
26 static void gw_default_delete_handler(struct gateway_entry *);
27
28 static struct olsr_gw_handler gw_def_handler = {
29   &gw_default_startup_handler,
30   &gw_default_choosegw_handler,
31   &gw_default_update_handler,
32   &gw_default_delete_handler
33 };
34
35 /**
36  * Look through the gateway list and select the best gateway
37  * depending on the distance to this router
38  */
39 static void gw_default_choose_gateway(void) {
40   struct tc_entry *tc;
41   struct gateway_entry *inet_ipv4, *inet_ipv6;
42   olsr_linkcost cost_ipv4, cost_ipv6;
43   struct gateway_entry *gw;
44   bool dual;
45
46   cost_ipv4 = UINT32_MAX;
47   cost_ipv6 = UINT32_MAX;
48
49   inet_ipv4 = NULL;
50   inet_ipv6 = NULL;
51
52   OLSR_FOR_ALL_GATEWAY_ENTRIES(gw) {
53     /* gateways should not exist without tc entry */
54     if ((tc = olsr_lookup_tc_entry(&gw->originator)) == NULL) {
55       continue;
56     }
57
58     if (!gw_def_finished_ipv4 && gw->ipv4 && gw->ipv4nat == olsr_cnf->smart_gw_allow_nat && tc->path_cost < cost_ipv4) {
59       inet_ipv4 = gw;
60     }
61     if (!gw_def_finished_ipv6 && gw->ipv6 && tc->path_cost < cost_ipv6) {
62       inet_ipv6 = gw;
63     }
64   } OLSR_FOR_ALL_GATEWAY_ENTRIES_END(gw)
65
66   /* found an IPv4 gateway ? */
67   gw_def_finished_ipv4 |= inet_ipv4 != NULL;
68   gw_def_finished_ipv6 |= inet_ipv6 != NULL;
69   dual = inet_ipv4 == inet_ipv6;
70   if (inet_ipv4) {
71     olsr_set_inet_gateway(&inet_ipv4->originator, true, dual, false);
72   }
73   if (inet_ipv6 && !dual) {
74     olsr_set_inet_gateway(&inet_ipv6->originator, false, true, false);
75   }
76
77   /* finished ? */
78   if (gw_def_finished_ipv4 && gw_def_finished_ipv6) {
79     olsr_stop_timer(gw_def_timer);
80     gw_def_timer = NULL;
81   }
82 }
83
84 /* timer for laze gateway selection */
85 static void gw_default_timer(void *unused __attribute__ ((unused))) {
86   /* accept a 10% increase without trigger a stablecount reset */
87   if (tc_tree.count * 10 <= gw_def_nodecount * 11) {
88     gw_def_nodecount = tc_tree.count;
89   }
90   if (tc_tree.count <= gw_def_nodecount) {
91     gw_def_stablecount++;
92   }
93   else {
94     gw_def_nodecount = tc_tree.count;
95     gw_def_stablecount = 0;
96   }
97
98   if (gw_def_stablecount >= GW_DEFAULT_STABLE_COUNT) {
99     gw_default_choose_gateway();
100   }
101 }
102
103 /* gateway handler callbacks */
104 static void gw_default_startup_handler(void) {
105   /* reset node count */
106   gw_def_nodecount = tc_tree.count;
107   gw_def_stablecount = 0;
108
109   /* get new ipv4 GW if we use OLSRv4 or NIIT */
110   gw_def_finished_ipv4 = olsr_cnf->ip_version == AF_INET6 && !olsr_cnf->use_niit;
111
112   /* get new ipv6 GW if we use OLSRv6 */
113   gw_def_finished_ipv6 = olsr_cnf->ip_version == AF_INET;
114
115   olsr_set_timer(&gw_def_timer, GW_DEFAULT_TIMER_INTERVAL, 0, true, &gw_default_timer, NULL, 0);
116 }
117
118 static void gw_default_update_handler(struct gateway_entry *gw) {
119   bool v4changed, v6changed;
120
121   v4changed = (gw == olsr_get_inet_gateway(false))
122       && (!gw->ipv4 || (gw->ipv4nat && !olsr_cnf->smart_gw_allow_nat));
123   v6changed = (gw == olsr_get_inet_gateway(true)) && !gw->ipv6;
124
125   if (v4changed || v6changed) {
126     olsr_gw_default_lookup_gateway(v4changed, v6changed);
127   }
128 }
129
130 static void gw_default_delete_handler(struct gateway_entry *gw) {
131   bool isv4, isv6;
132
133   isv4 = gw == olsr_get_inet_gateway(false);
134   isv6 = gw == olsr_get_inet_gateway(true);
135
136   if (gw != NULL && (isv4 || isv6)) {
137     olsr_gw_default_lookup_gateway(isv4, isv6);
138   }
139 }
140
141 static void gw_default_choosegw_handler(bool ipv4, bool ipv6) {
142   olsr_gw_default_lookup_gateway(ipv4, ipv6);
143
144   if (!(gw_def_finished_ipv4 && gw_def_finished_ipv6)) {
145     gw_default_startup_handler();
146   }
147 }
148
149 /**
150  * initialization of default gateway handler
151  */
152 void olsr_gw_default_init(void) {
153   /* initialize values */
154   gw_def_timer = NULL;
155   gw_def_finished_ipv4 = false;
156   gw_def_finished_ipv6 = false;
157   gw_def_nodecount = 0;
158   gw_def_stablecount = 0;
159   gw_def_timer = NULL;
160
161   /* setup default handler */
162   olsr_set_inetgw_handler(&gw_def_handler);
163 }
164
165 /**
166  * Lookup a new gateway based on distance metric
167  *
168  * @param ipv4 lookup new v4 gateway
169  * @param ipv6 lookup new v6 gateway
170  */
171 void olsr_gw_default_lookup_gateway(bool ipv4, bool ipv6) {
172   if (ipv4) {
173     /* get new ipv4 GW if we use OLSRv4 or NIIT */
174     gw_def_finished_ipv4 = olsr_cnf->ip_version == AF_INET6 && !olsr_cnf->use_niit;
175   }
176   if (ipv6) {
177     /* get new ipv6 GW if we use OLSRv6 */
178     gw_def_finished_ipv6 = olsr_cnf->ip_version == AF_INET;
179   }
180
181   if (!(gw_def_finished_ipv4 && gw_def_finished_ipv6)) {
182     gw_default_choose_gateway();
183   }
184 }
185 #endif