9d439225678b12d1afdfcd66774c8ef2d346be73
[tinc] / src / protocol_edge.c
1 /*
2     protocol_edge.c -- handle the meta-protocol, edges
3     Copyright (C) 1999-2005 Ivo Timmermans,
4                   2000-2009 Guus Sliepen <guus@tinc-vpn.org>
5                   2009      Michael Tokarev <mjt@corpit.ru>
6
7     This program is free software; you can redistribute it and/or modify
8     it under the terms of the GNU General Public License as published by
9     the Free Software Foundation; either version 2 of the License, or
10     (at your option) any later version.
11
12     This program is distributed in the hope that it will be useful,
13     but WITHOUT ANY WARRANTY; without even the implied warranty of
14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15     GNU General Public License for more details.
16
17     You should have received a copy of the GNU General Public License along
18     with this program; if not, write to the Free Software Foundation, Inc.,
19     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20 */
21
22 #include "system.h"
23
24 #include "avl_tree.h"
25 #include "conf.h"
26 #include "connection.h"
27 #include "edge.h"
28 #include "graph.h"
29 #include "logger.h"
30 #include "meta.h"
31 #include "net.h"
32 #include "netutl.h"
33 #include "node.h"
34 #include "protocol.h"
35 #include "utils.h"
36 #include "xalloc.h"
37
38 bool send_add_edge(connection_t *c, const edge_t *e) {
39         bool x;
40         char *address, *port;
41
42         sockaddr2str(&e->address, &address, &port);
43
44         x = send_request(c, "%d %x %s %s %s %s %lx %d", ADD_EDGE, rand(),
45                                          e->from->name, e->to->name, address, port,
46                                          e->options, e->weight);
47         free(address);
48         free(port);
49
50         return x;
51 }
52
53 bool add_edge_h(connection_t *c) {
54         edge_t *e;
55         node_t *from, *to;
56         char from_name[MAX_STRING_SIZE];
57         char to_name[MAX_STRING_SIZE];
58         char to_address[MAX_STRING_SIZE];
59         char to_port[MAX_STRING_SIZE];
60         sockaddr_t address;
61         long int options;
62         int weight;
63
64         if(sscanf(c->buffer, "%*d %*x "MAX_STRING" "MAX_STRING" "MAX_STRING" "MAX_STRING" %lx %d",
65                           from_name, to_name, to_address, to_port, &options, &weight) != 6) {
66                 logger(LOG_ERR, "Got bad %s from %s (%s)", "ADD_EDGE", c->name,
67                            c->hostname);
68                 return false;
69         }
70
71         /* Check if names are valid */
72
73         if(!check_id(from_name) || !check_id(to_name)) {
74                 logger(LOG_ERR, "Got bad %s from %s (%s): %s", "ADD_EDGE", c->name,
75                            c->hostname, "invalid name");
76                 return false;
77         }
78
79         if(seen_request(c->buffer))
80                 return true;
81
82         /* Lookup nodes */
83
84         from = lookup_node(from_name);
85         to = lookup_node(to_name);
86
87         if(tunnelserver &&
88            from != myself && from != c->node &&
89            to != myself && to != c->node) {
90                 /* ignore indirect edge registrations for tunnelserver */
91                 ifdebug(PROTOCOL) logger(LOG_WARNING,
92                    "Ignoring indirect %s from %s (%s)",
93                    "ADD_EDGE", c->name, c->hostname);
94                 return true;
95         }
96
97         if(!from) {
98                 from = new_node();
99                 from->name = xstrdup(from_name);
100                 node_add(from);
101         }
102
103         if(!to) {
104                 to = new_node();
105                 to->name = xstrdup(to_name);
106                 node_add(to);
107         }
108
109
110         /* Convert addresses */
111
112         address = str2sockaddr(to_address, to_port);
113
114         /* Check if edge already exists */
115
116         e = lookup_edge(from, to);
117
118         if(e) {
119                 if(e->weight != weight || e->options != options || sockaddrcmp(&e->address, &address)) {
120                         if(from == myself) {
121                                 ifdebug(PROTOCOL) logger(LOG_WARNING, "Got %s from %s (%s) for ourself which does not match existing entry",
122                                                    "ADD_EDGE", c->name, c->hostname);
123                                 send_add_edge(c, e);
124                                 return true;
125                         } else {
126                                 ifdebug(PROTOCOL) logger(LOG_WARNING, "Got %s from %s (%s) which does not match existing entry",
127                                                    "ADD_EDGE", c->name, c->hostname);
128                                 edge_del(e);
129                                 graph();
130                         }
131                 } else
132                         return true;
133         } else if(from == myself) {
134                 ifdebug(PROTOCOL) logger(LOG_WARNING, "Got %s from %s (%s) for ourself which does not exist",
135                                    "ADD_EDGE", c->name, c->hostname);
136                 e = new_edge();
137                 e->from = from;
138                 e->to = to;
139                 send_del_edge(c, e);
140                 free_edge(e);
141                 return true;
142         }
143
144         e = new_edge();
145         e->from = from;
146         e->to = to;
147         e->address = address;
148         e->options = options;
149         e->weight = weight;
150         edge_add(e);
151
152         /* Tell the rest about the new edge */
153
154         if(!tunnelserver)
155                 forward_request(c);
156
157         /* Run MST before or after we tell the rest? */
158
159         graph();
160
161         return true;
162 }
163
164 bool send_del_edge(connection_t *c, const edge_t *e) {
165         return send_request(c, "%d %x %s %s", DEL_EDGE, rand(),
166                                                 e->from->name, e->to->name);
167 }
168
169 bool del_edge_h(connection_t *c) {
170         edge_t *e;
171         char from_name[MAX_STRING_SIZE];
172         char to_name[MAX_STRING_SIZE];
173         node_t *from, *to;
174
175         if(sscanf(c->buffer, "%*d %*x "MAX_STRING" "MAX_STRING, from_name, to_name) != 2) {
176                 logger(LOG_ERR, "Got bad %s from %s (%s)", "DEL_EDGE", c->name,
177                            c->hostname);
178                 return false;
179         }
180
181         /* Check if names are valid */
182
183         if(!check_id(from_name) || !check_id(to_name)) {
184                 logger(LOG_ERR, "Got bad %s from %s (%s): %s", "DEL_EDGE", c->name,
185                            c->hostname, "invalid name");
186                 return false;
187         }
188
189         if(seen_request(c->buffer))
190                 return true;
191
192         /* Lookup nodes */
193
194         from = lookup_node(from_name);
195         to = lookup_node(to_name);
196
197         if(tunnelserver &&
198            from != myself && from != c->node &&
199            to != myself && to != c->node) {
200                 /* ignore indirect edge registrations for tunnelserver */
201                 ifdebug(PROTOCOL) logger(LOG_WARNING,
202                    "Ignoring indirect %s from %s (%s)",
203                    "DEL_EDGE", c->name, c->hostname);
204                 return true;
205         }
206
207         if(!from) {
208                 ifdebug(PROTOCOL) logger(LOG_ERR, "Got %s from %s (%s) which does not appear in the edge tree",
209                                    "DEL_EDGE", c->name, c->hostname);
210                 return true;
211         }
212
213         if(!to) {
214                 ifdebug(PROTOCOL) logger(LOG_ERR, "Got %s from %s (%s) which does not appear in the edge tree",
215                                    "DEL_EDGE", c->name, c->hostname);
216                 return true;
217         }
218
219         /* Check if edge exists */
220
221         e = lookup_edge(from, to);
222
223         if(!e) {
224                 ifdebug(PROTOCOL) logger(LOG_WARNING, "Got %s from %s (%s) which does not appear in the edge tree",
225                                    "DEL_EDGE", c->name, c->hostname);
226                 return true;
227         }
228
229         if(e->from == myself) {
230                 ifdebug(PROTOCOL) logger(LOG_WARNING, "Got %s from %s (%s) for ourself",
231                                    "DEL_EDGE", c->name, c->hostname);
232                 send_add_edge(c, e);    /* Send back a correction */
233                 return true;
234         }
235
236         /* Tell the rest about the deleted edge */
237
238         if(!tunnelserver)
239                 forward_request(c);
240
241         /* Delete the edge */
242
243         edge_del(e);
244
245         /* Run MST before or after we tell the rest? */
246
247         graph();
248
249         /* If the node is not reachable anymore but we remember it had an edge to us, clean it up */
250
251         if(!to->status.reachable) {
252                 e = lookup_edge(to, myself);
253                 if(e) {
254                         if(!tunnelserver)
255                                 send_del_edge(broadcast, e);
256                         edge_del(e);
257                 }
258         }
259
260         return true;
261 }