--- /dev/null
+/*
+ * Copyright (c) 2014 Nicira, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <config.h>
+#include <arpa/inet.h>
+#include <errno.h>
+#include <inttypes.h>
+#include <sys/socket.h>
+#include <net/if.h>
+#include <netinet/in.h>
+#include <stdarg.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "classifier.h"
+#include "command-line.h"
+#include "compiler.h"
+#include "dpif.h"
+#include "dynamic-string.h"
+#include "netdev.h"
+#include "packets.h"
+#include "ovs-router.h"
+#include "unixctl.h"
+#include "util.h"
+
+static struct classifier cls;
+
+struct ovs_router_entry {
+ struct cls_rule cr;
+ char output_bridge[IFNAMSIZ];
+ ovs_be32 gw;
+ ovs_be32 nw_addr;
+ uint8_t plen;
+ uint8_t priority;
+};
+
+static struct ovs_router_entry *
+ovs_router_entry_cast(const struct cls_rule *cr)
+{
+ if (offsetof(struct ovs_router_entry, cr) == 0) {
+ return CONTAINER_OF(cr, struct ovs_router_entry, cr);
+ } else {
+ return cr ? CONTAINER_OF(cr, struct ovs_router_entry, cr) : NULL;
+ }
+}
+
+bool
+ovs_router_lookup(ovs_be32 ip_dst, char output_bridge[], ovs_be32 *gw)
+{
+ const struct cls_rule *cr;
+ struct flow flow = {.nw_dst = ip_dst};
+
+ cr = classifier_lookup(&cls, &flow, NULL);
+ if (cr) {
+ struct ovs_router_entry *p = ovs_router_entry_cast(cr);
+
+ strncpy(output_bridge, p->output_bridge, IFNAMSIZ);
+ *gw = p->gw;
+ return true;
+ }
+ return false;
+}
+
+static void
+rt_entry_free(struct ovs_router_entry *p)
+{
+ cls_rule_destroy(&p->cr);
+ free(p);
+}
+
+static void rt_init_match(struct match *match, ovs_be32 ip_dst, uint8_t plen)
+{
+ ovs_be32 mask;
+
+ mask = htonl(UINT32_MAX << (32 - plen));
+
+ ip_dst &= mask; /* Clear out insignificant bits. */
+ memset(match, 0, sizeof *match);
+ match->flow.nw_dst = ip_dst;
+ match->wc.masks.nw_dst = mask;
+}
+
+static void
+ovs_router_insert__(uint8_t priority, ovs_be32 ip_dst, uint8_t plen,
+ const char output_bridge[],
+ ovs_be32 gw)
+{
+ const struct cls_rule *cr;
+ struct ovs_router_entry *p;
+ struct match match;
+
+ rt_init_match(&match, ip_dst, plen);
+
+ p = xzalloc(sizeof *p);
+ strncpy(p->output_bridge, output_bridge, IFNAMSIZ);
+ p->gw = gw;
+ p->nw_addr = match.flow.nw_dst;
+ p->plen = plen;
+ p->priority = priority;
+ cls_rule_init(&p->cr, &match, priority); /* Longest prefix matches first. */
+
+ cr = classifier_replace(&cls, &p->cr);
+ if (cr) {
+ /* An old rule with the same match was displaced. */
+ ovsrcu_postpone(rt_entry_free, ovs_router_entry_cast(cr));
+ }
+}
+
+void
+ovs_router_insert(ovs_be32 ip_dst, uint8_t plen, const char output_bridge[],
+ ovs_be32 gw)
+{
+ ovs_router_insert__(plen, ip_dst, plen, output_bridge, gw);
+}
+
+static bool
+rt_entry_delete(uint8_t priority, ovs_be32 ip_dst, uint8_t plen)
+{
+ struct cls_rule *cr;
+ struct cls_rule rule;
+ struct match match;
+
+ rt_init_match(&match, ip_dst, plen);
+
+ cls_rule_init(&rule, &match, priority);
+
+ /* Find the exact rule. */
+ cr = classifier_find_rule_exactly(&cls, &rule);
+ if (cr) {
+ /* Remove it. */
+ cr = classifier_remove(&cls, cr);
+ if (cr) {
+
+ ovsrcu_postpone(rt_entry_free, ovs_router_entry_cast(cr));
+ return true;
+ }
+ }
+ return false;
+}
+
+static bool
+scan_ipv4_route(const char *s, ovs_be32 *addr, unsigned int *plen)
+{
+ int len, max_plen, n;
+ int slen = strlen(s);
+ uint8_t *ip = (uint8_t *)addr;
+
+ *addr = htons(0);
+ if (!ovs_scan(s, "%"SCNu8"%n", &ip[0], &n)) {
+ return false;
+ }
+ len = n;
+ max_plen = 8;
+ for (int i = 1; i < 4; i++) {
+ if (ovs_scan(s + len, ".%"SCNu8"%n", &ip[i], &n)) {
+ len += n;
+ max_plen += 8;
+ } else {
+ break;
+ }
+ }
+ if (len == slen && max_plen == 32) {
+ *plen = 32;
+ return true;
+ }
+ if (ovs_scan(s + len, "/%u%n", plen, &n)
+ && len + n == slen && *plen <= max_plen) {
+ return true;
+ }
+ return false;
+}
+
+static void
+ovs_router_add(struct unixctl_conn *conn, int argc,
+ const char *argv[], void *aux OVS_UNUSED)
+{
+ ovs_be32 ip, gw;
+ unsigned int plen;
+
+ if (scan_ipv4_route(argv[1], &ip, &plen)) {
+ if (argc > 3) {
+ inet_pton(AF_INET, argv[3], (struct in_addr *)&gw);
+ } else {
+ gw = 0;
+ }
+ ovs_router_insert__(plen + 32, ip, plen, argv[2], gw);
+ unixctl_command_reply(conn, "OK");
+ } else {
+ unixctl_command_reply(conn, "Invalid parameters");
+ }
+}
+
+static void
+ovs_router_del(struct unixctl_conn *conn, int argc OVS_UNUSED,
+ const char *argv[], void *aux OVS_UNUSED)
+{
+ ovs_be32 ip;
+ unsigned int plen;
+
+ if (scan_ipv4_route(argv[1], &ip, &plen)) {
+
+ if (rt_entry_delete(plen + 32, ip, plen)) {
+ unixctl_command_reply(conn, "OK");
+ } else {
+ unixctl_command_reply(conn, "Not found");
+ }
+ } else {
+ unixctl_command_reply(conn, "Invalid parameters");
+ }
+}
+
+static void
+ovs_router_show(struct unixctl_conn *conn, int argc OVS_UNUSED,
+ const char *argv[] OVS_UNUSED, void *aux OVS_UNUSED)
+{
+ struct ovs_router_entry *rt;
+ struct ds ds = DS_EMPTY_INITIALIZER;
+
+ ds_put_format(&ds, "Route Table:\n");
+ CLS_FOR_EACH(rt, cr, &cls) {
+ if (rt->priority == rt->plen) {
+ ds_put_format(&ds, "Cached: ");
+ } else {
+ ds_put_format(&ds, "User: ");
+ }
+ ds_put_format(&ds, IP_FMT"/%"PRIu16" dev %s",
+ IP_ARGS(rt->nw_addr), rt->plen,
+ rt->output_bridge);
+ if (rt->gw) {
+ ds_put_format(&ds, " GW "IP_FMT, IP_ARGS(rt->gw));
+ }
+ ds_put_format(&ds, "\n");
+ }
+ unixctl_command_reply(conn, ds_cstr(&ds));
+ ds_destroy(&ds);
+}
+
+void
+ovs_router_flush(void)
+{
+ struct ovs_router_entry *rt;
+
+ CLS_FOR_EACH_SAFE(rt, cr, &cls) {
+ if (rt->priority == rt->plen) {
+ classifier_remove(&cls, &rt->cr);
+ }
+ }
+}
+
+/* May not be called more than once. */
+void
+ovs_router_unixctl_register(void)
+{
+ classifier_init(&cls, NULL);
+ /* XXX: Add documentation for these commands. */
+ unixctl_command_register("ovs/route/add", "ip mask dev gw", 2, 3,
+ ovs_router_add, NULL);
+ unixctl_command_register("ovs/route/show", "", 0, 0, ovs_router_show, NULL);
+ unixctl_command_register("ovs/route/del", "ip mask", 1, 1, ovs_router_del,
+ NULL);
+}
#include <net/if.h>
#include "hash.h"
-#include "hmap.h"
#include "netlink.h"
#include "netlink-notifier.h"
#include "netlink-socket.h"
#include "ofpbuf.h"
+#include "ovs-router.h"
#include "rtnetlink-link.h"
#include "vlog.h"
unsigned char rtm_dst_len;
/* Extracted from Netlink attributes. */
- uint32_t rta_dst; /* Destination in host byte order. 0 if missing. */
+ ovs_be32 rta_dst; /* 0 if missing. */
char ifname[IFNAMSIZ]; /* Interface name. */
};
struct route_data rd; /* Data parsed from this message. */
};
-struct route_node {
- struct hmap_node node; /* Node in route_map. */
- struct route_data rd; /* Data associated with this node. */
-};
-
static struct ovs_mutex route_table_mutex = OVS_MUTEX_INITIALIZER;
static struct vlog_rate_limit rl = VLOG_RATE_LIMIT_INIT(5, 20);
static struct nln_notifier *name_notifier = NULL;
static bool route_table_valid = false;
-static struct hmap route_map;
static int route_table_reset(void);
static void route_table_handle_msg(const struct route_table_msg *);
static bool route_table_parse(struct ofpbuf *, struct route_table_msg *);
static void route_table_change(const struct route_table_msg *, void *);
-static struct route_node *route_node_lookup(const struct route_data *);
-static struct route_node *route_node_lookup_by_ip(uint32_t ip);
static void route_map_clear(void);
-static uint32_t hash_route_data(const struct route_data *);
static void name_table_init(void);
static void name_table_uninit(void);
static void name_table_change(const struct rtnetlink_link_change *, void *);
-/* Populates 'name' with the name of the interface traffic destined for 'ip'
- * is likely to egress out of.
- *
- * Returns true if successful, otherwise false. */
-bool
-route_table_get_name(ovs_be32 ip_, char name[IFNAMSIZ])
- OVS_REQUIRES(route_table_mutex)
-{
- struct route_node *rn;
- uint32_t ip = ntohl(ip_);
- bool res = false;
-
- ovs_mutex_lock(&route_table_mutex);
- if (!route_table_valid) {
- route_table_reset();
- }
-
- rn = route_node_lookup_by_ip(ip);
-
- if (rn) {
- ovs_strlcpy(name, rn->rd.ifname, IFNAMSIZ);
- res = true;
- goto out;
- }
-
- /* Choose a default route. */
- HMAP_FOR_EACH(rn, node, &route_map) {
- if (rn->rd.rta_dst == 0 && rn->rd.rtm_dst_len == 0) {
- ovs_strlcpy(name, rn->rd.ifname, IFNAMSIZ);
- res = true;
- break;
- }
- }
-
-out:
- ovs_mutex_unlock(&route_table_mutex);
- return res;
-}
-
uint64_t
route_table_get_change_seq(void)
{
nln_notifier_create(nln, (nln_notify_func *) route_table_change,
NULL);
- hmap_init(&route_map);
route_table_reset();
name_table_init();
}
nln = NULL;
route_map_clear();
- hmap_destroy(&route_map);
name_table_uninit();
}
ovs_mutex_unlock(&route_table_mutex);
}
if (attrs[RTA_DST]) {
- change->rd.rta_dst = ntohl(nl_attr_get_be32(attrs[RTA_DST]));
+ change->rd.rta_dst = nl_attr_get_be32(attrs[RTA_DST]);
}
} else {
static void
route_table_handle_msg(const struct route_table_msg *change)
{
- if (change->relevant && change->nlmsg_type == RTM_NEWROUTE &&
- !route_node_lookup(&change->rd)) {
- struct route_node *rn;
-
- rn = xzalloc(sizeof *rn);
- memcpy(&rn->rd, &change->rd, sizeof change->rd);
+ if (change->relevant && change->nlmsg_type == RTM_NEWROUTE) {
+ const struct route_data *rd = &change->rd;
- hmap_insert(&route_map, &rn->node, hash_route_data(&rn->rd));
+ ovs_router_insert(rd->rta_dst, rd->rtm_dst_len, rd->ifname, 0);
}
}
-static struct route_node *
-route_node_lookup(const struct route_data *rd)
-{
- struct route_node *rn;
-
- HMAP_FOR_EACH_WITH_HASH(rn, node, hash_route_data(rd), &route_map) {
- if (!memcmp(&rn->rd, rd, sizeof *rd)) {
- return rn;
- }
- }
-
- return NULL;
-}
-
-static struct route_node *
-route_node_lookup_by_ip(uint32_t ip)
-{
- int dst_len;
- struct route_node *rn, *rn_ret;
-
- dst_len = -1;
- rn_ret = NULL;
-
- HMAP_FOR_EACH(rn, node, &route_map) {
- uint32_t mask = 0xffffffff << (32 - rn->rd.rtm_dst_len);
-
- if (rn->rd.rta_dst == 0 && rn->rd.rtm_dst_len == 0) {
- /* Default route. */
- continue;
- }
-
- if (rn->rd.rtm_dst_len > dst_len &&
- (ip & mask) == (rn->rd.rta_dst & mask)) {
- rn_ret = rn;
- dst_len = rn->rd.rtm_dst_len;
- }
- }
-
- return rn_ret;
-}
-
static void
route_map_clear(void)
{
- struct route_node *rn, *rn_next;
-
- HMAP_FOR_EACH_SAFE(rn, rn_next, node, &route_map) {
- hmap_remove(&route_map, &rn->node);
- free(rn);
- }
+ ovs_router_flush();
}
-static uint32_t
-hash_route_data(const struct route_data *rd)
-{
- return hash_bytes(rd, sizeof *rd, 0);
-}
\f
/* name_table . */
projects / cascardo / ovs.git / commitdiff