netdev-dpdk: fix mbuf leaks
[cascardo/ovs.git] / lib / ovs-router.c
1 /*
2  * Copyright (c) 2014, 2015 Nicira, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at:
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include <config.h>
18
19 #include "ovs-router.h"
20
21 #include <arpa/inet.h>
22 #include <errno.h>
23 #include <inttypes.h>
24 #include <sys/socket.h>
25 #include <net/if.h>
26 #include <netinet/in.h>
27 #include <stdarg.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <unistd.h>
31
32 #include "classifier.h"
33 #include "command-line.h"
34 #include "compiler.h"
35 #include "dpif.h"
36 #include "dynamic-string.h"
37 #include "netdev.h"
38 #include "packets.h"
39 #include "seq.h"
40 #include "ovs-thread.h"
41 #include "route-table.h"
42 #include "tnl-ports.h"
43 #include "unixctl.h"
44 #include "util.h"
45
46 static struct ovs_mutex mutex = OVS_MUTEX_INITIALIZER;
47 static struct classifier cls;
48
49 struct ovs_router_entry {
50     struct cls_rule cr;
51     char output_bridge[IFNAMSIZ];
52     struct in6_addr gw;
53     struct in6_addr nw_addr;
54     uint8_t plen;
55     uint8_t priority;
56 };
57
58 static struct ovs_router_entry *
59 ovs_router_entry_cast(const struct cls_rule *cr)
60 {
61     if (offsetof(struct ovs_router_entry, cr) == 0) {
62         return CONTAINER_OF(cr, struct ovs_router_entry, cr);
63     } else {
64         return cr ? CONTAINER_OF(cr, struct ovs_router_entry, cr) : NULL;
65     }
66 }
67
68 bool
69 ovs_router_lookup(const struct in6_addr *ip6_dst, char output_bridge[],
70                   struct in6_addr *gw)
71 {
72     const struct cls_rule *cr;
73     struct flow flow = {.ipv6_dst = *ip6_dst};
74
75     cr = classifier_lookup(&cls, CLS_MAX_VERSION, &flow, NULL);
76     if (cr) {
77         struct ovs_router_entry *p = ovs_router_entry_cast(cr);
78
79         ovs_strlcpy(output_bridge, p->output_bridge, IFNAMSIZ);
80         *gw = p->gw;
81         return true;
82     }
83     return false;
84 }
85
86 bool
87 ovs_router_lookup4(ovs_be32 ip_dst, char output_bridge[], ovs_be32 *gw)
88 {
89     struct in6_addr ip6_dst = in6_addr_mapped_ipv4(ip_dst);
90     struct in6_addr gw6;
91
92     if (ovs_router_lookup(&ip6_dst, output_bridge, &gw6)) {
93         *gw = in6_addr_get_mapped_ipv4(&gw6);
94         return true;
95     }
96     return route_table_fallback_lookup(ip_dst, output_bridge, gw);
97 }
98
99 static void
100 rt_entry_free(struct ovs_router_entry *p)
101 {
102     cls_rule_destroy(&p->cr);
103     free(p);
104 }
105
106 static void rt_init_match(struct match *match, const struct in6_addr *ip6_dst,
107                           uint8_t plen)
108 {
109     struct in6_addr dst;
110     struct in6_addr mask;
111
112     mask = ipv6_create_mask(plen);
113
114     dst = ipv6_addr_bitand(ip6_dst, &mask);
115     memset(match, 0, sizeof *match);
116     match->flow.ipv6_dst = dst;
117     match->wc.masks.ipv6_dst = mask;
118 }
119
120 static void
121 ovs_router_insert__(uint8_t priority, const struct in6_addr *ip6_dst,
122                     uint8_t plen, const char output_bridge[],
123                     const struct in6_addr *gw)
124 {
125     const struct cls_rule *cr;
126     struct ovs_router_entry *p;
127     struct match match;
128
129     rt_init_match(&match, ip6_dst, plen);
130
131     p = xzalloc(sizeof *p);
132     ovs_strlcpy(p->output_bridge, output_bridge, sizeof p->output_bridge);
133     if (ipv6_addr_is_set(gw)) {
134         p->gw = *gw;
135     }
136     p->nw_addr = match.flow.ipv6_dst;
137     p->plen = plen;
138     p->priority = priority;
139     /* Longest prefix matches first. */
140     cls_rule_init(&p->cr, &match, priority);
141
142     ovs_mutex_lock(&mutex);
143     cr = classifier_replace(&cls, &p->cr, CLS_MIN_VERSION, NULL, 0);
144     ovs_mutex_unlock(&mutex);
145
146     if (cr) {
147         /* An old rule with the same match was displaced. */
148         ovsrcu_postpone(rt_entry_free, ovs_router_entry_cast(cr));
149     }
150     tnl_port_map_insert_ipdev(output_bridge);
151     seq_change(tnl_conf_seq);
152 }
153
154 void
155 ovs_router_insert(const struct in6_addr *ip_dst, uint8_t plen,
156                   const char output_bridge[], const struct in6_addr *gw)
157 {
158     ovs_router_insert__(plen, ip_dst, plen, output_bridge, gw);
159 }
160
161
162 static bool
163 __rt_entry_delete(const struct cls_rule *cr)
164 {
165     struct ovs_router_entry *p = ovs_router_entry_cast(cr);
166
167     tnl_port_map_delete_ipdev(p->output_bridge);
168     /* Remove it. */
169     cr = classifier_remove(&cls, cr);
170     if (cr) {
171         ovsrcu_postpone(rt_entry_free, ovs_router_entry_cast(cr));
172         return true;
173     }
174     return false;
175 }
176
177 static bool
178 rt_entry_delete(uint8_t priority, const struct in6_addr *ip6_dst, uint8_t plen)
179 {
180     const struct cls_rule *cr;
181     struct cls_rule rule;
182     struct match match;
183     bool res = false;
184
185     rt_init_match(&match, ip6_dst, plen);
186
187     cls_rule_init(&rule, &match, priority);
188
189     /* Find the exact rule. */
190     cr = classifier_find_rule_exactly(&cls, &rule, CLS_MAX_VERSION);
191     if (cr) {
192         ovs_mutex_lock(&mutex);
193         res = __rt_entry_delete(cr);
194         ovs_mutex_unlock(&mutex);
195     }
196     return res;
197 }
198
199 static bool
200 scan_ipv6_route(const char *s, struct in6_addr *addr, unsigned int *plen)
201 {
202     char *error = ipv6_parse_cidr(s, addr, plen);
203     if (error) {
204         free(error);
205         return false;
206     }
207     return true;
208 }
209
210 static bool
211 scan_ipv4_route(const char *s, ovs_be32 *addr, unsigned int *plen)
212 {
213     char *error = ip_parse_cidr(s, addr, plen);
214     if (error) {
215         free(error);
216         return false;
217     }
218     return true;
219 }
220
221 static void
222 ovs_router_add(struct unixctl_conn *conn, int argc,
223               const char *argv[], void *aux OVS_UNUSED)
224 {
225     ovs_be32 ip;
226     unsigned int plen;
227     struct in6_addr ip6;
228     struct in6_addr gw6;
229
230     if (scan_ipv4_route(argv[1], &ip, &plen)) {
231         ovs_be32 gw = 0;
232         if (argc > 3 && !ip_parse(argv[3], &gw)) {
233             unixctl_command_reply_error(conn, "Invalid gateway");
234             return;
235         }
236         in6_addr_set_mapped_ipv4(&ip6, ip);
237         in6_addr_set_mapped_ipv4(&gw6, gw);
238         plen += 96;
239     } else if (scan_ipv6_route(argv[1], &ip6, &plen)) {
240         gw6 = in6addr_any;
241         if (argc > 3 && !ipv6_parse(argv[3], &gw6)) {
242             unixctl_command_reply_error(conn, "Invalid IPv6 gateway");
243             return;
244         }
245     } else {
246         unixctl_command_reply_error(conn, "Invalid parameters");
247         return;
248     }
249     ovs_router_insert__(plen + 32, &ip6, plen, argv[2], &gw6);
250     unixctl_command_reply(conn, "OK");
251 }
252
253 static void
254 ovs_router_del(struct unixctl_conn *conn, int argc OVS_UNUSED,
255               const char *argv[], void *aux OVS_UNUSED)
256 {
257     ovs_be32 ip;
258     unsigned int plen;
259     struct in6_addr ip6;
260
261     if (scan_ipv4_route(argv[1], &ip, &plen)) {
262         in6_addr_set_mapped_ipv4(&ip6, ip);
263         plen += 96;
264     } else if (!scan_ipv6_route(argv[1], &ip6, &plen)) {
265         unixctl_command_reply_error(conn, "Invalid parameters");
266         return;
267     }
268     if (rt_entry_delete(plen + 32, &ip6, plen)) {
269         unixctl_command_reply(conn, "OK");
270         seq_change(tnl_conf_seq);
271     } else {
272         unixctl_command_reply_error(conn, "Not found");
273     }
274 }
275
276 static void
277 ovs_router_show(struct unixctl_conn *conn, int argc OVS_UNUSED,
278                const char *argv[] OVS_UNUSED, void *aux OVS_UNUSED)
279 {
280     struct ovs_router_entry *rt;
281     struct ds ds = DS_EMPTY_INITIALIZER;
282
283     ds_put_format(&ds, "Route Table:\n");
284     CLS_FOR_EACH(rt, cr, &cls) {
285         uint8_t plen;
286         if (rt->priority == rt->plen) {
287             ds_put_format(&ds, "Cached: ");
288         } else {
289             ds_put_format(&ds, "User: ");
290         }
291         ipv6_format_mapped(&rt->nw_addr, &ds);
292         plen = rt->plen;
293         if (IN6_IS_ADDR_V4MAPPED(&rt->nw_addr)) {
294             plen -= 96;
295         }
296         ds_put_format(&ds, "/%"PRIu16" dev %s", plen, rt->output_bridge);
297         if (ipv6_addr_is_set(&rt->gw)) {
298             ds_put_format(&ds, " GW ");
299             ipv6_format_mapped(&rt->gw, &ds);
300         }
301         ds_put_format(&ds, "\n");
302     }
303     unixctl_command_reply(conn, ds_cstr(&ds));
304     ds_destroy(&ds);
305 }
306
307 static void
308 ovs_router_lookup_cmd(struct unixctl_conn *conn, int argc OVS_UNUSED,
309                       const char *argv[], void *aux OVS_UNUSED)
310 {
311     ovs_be32 ip;
312     struct in6_addr ip6;
313     unsigned int plen;
314     char iface[IFNAMSIZ];
315     struct in6_addr gw;
316
317     if (scan_ipv4_route(argv[1], &ip, &plen) && plen == 32) {
318         in6_addr_set_mapped_ipv4(&ip6, ip);
319     } else if (!(scan_ipv6_route(argv[1], &ip6, &plen) && plen == 128)) {
320         unixctl_command_reply_error(conn, "Invalid parameters");
321         return;
322     }
323
324     if (ovs_router_lookup(&ip6, iface, &gw)) {
325         struct ds ds = DS_EMPTY_INITIALIZER;
326         ds_put_format(&ds, "gateway ");
327         ipv6_format_mapped(&ip6, &ds);
328         ds_put_format(&ds, "\ndev %s\n", iface);
329         unixctl_command_reply(conn, ds_cstr(&ds));
330         ds_destroy(&ds);
331     } else {
332         unixctl_command_reply_error(conn, "Not found");
333     }
334 }
335
336 void
337 ovs_router_flush(void)
338 {
339     struct ovs_router_entry *rt;
340
341     ovs_mutex_lock(&mutex);
342     classifier_defer(&cls);
343     CLS_FOR_EACH(rt, cr, &cls) {
344         if (rt->priority == rt->plen) {
345             __rt_entry_delete(&rt->cr);
346         }
347     }
348     classifier_publish(&cls);
349     ovs_mutex_unlock(&mutex);
350     seq_change(tnl_conf_seq);
351 }
352
353 /* May not be called more than once. */
354 void
355 ovs_router_init(void)
356 {
357     classifier_init(&cls, NULL);
358     unixctl_command_register("ovs/route/add", "ip_addr/prefix_len out_br_name gw", 2, 3,
359                              ovs_router_add, NULL);
360     unixctl_command_register("ovs/route/show", "", 0, 0, ovs_router_show, NULL);
361     unixctl_command_register("ovs/route/del", "ip_addr/prefix_len", 1, 1, ovs_router_del,
362                              NULL);
363     unixctl_command_register("ovs/route/lookup", "ip_addr", 1, 1,
364                              ovs_router_lookup_cmd, NULL);
365 }