123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780 |
- /*
- * swlib.c: Switch configuration API (user space part)
- *
- * Copyright (C) 2008 Felix Fietkau <nbd@openwrt.org>
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public License
- * version 2.1 as published by the Free Software Foundation.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- */
- #include <stdio.h>
- #include <string.h>
- #include <stdlib.h>
- #include <inttypes.h>
- #include <errno.h>
- #include <stdint.h>
- #include <getopt.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <linux/switch.h>
- #include "swlib.h"
- #include <netlink/netlink.h>
- #include <netlink/genl/genl.h>
- #include <netlink/genl/family.h>
- //#define DEBUG 1
- #ifdef DEBUG
- #define DPRINTF(fmt, ...) fprintf(stderr, "%s(%d): " fmt, __func__, __LINE__, ##__VA_ARGS__)
- #else
- #define DPRINTF(fmt, ...) do {} while (0)
- #endif
- static struct nl_sock *handle;
- static struct nl_cache *cache;
- static struct genl_family *family;
- static struct nlattr *tb[SWITCH_ATTR_MAX + 1];
- static int refcount = 0;
- static struct nla_policy port_policy[SWITCH_ATTR_MAX] = {
- [SWITCH_PORT_ID] = { .type = NLA_U32 },
- [SWITCH_PORT_FLAG_TAGGED] = { .type = NLA_FLAG },
- };
- static struct nla_policy portmap_policy[SWITCH_PORTMAP_MAX] = {
- [SWITCH_PORTMAP_SEGMENT] = { .type = NLA_STRING },
- [SWITCH_PORTMAP_VIRT] = { .type = NLA_U32 },
- };
- static inline void *
- swlib_alloc(size_t size)
- {
- void *ptr;
- ptr = malloc(size);
- if (!ptr)
- goto done;
- memset(ptr, 0, size);
- done:
- return ptr;
- }
- static int
- wait_handler(struct nl_msg *msg, void *arg)
- {
- int *finished = arg;
- *finished = 1;
- return NL_STOP;
- }
- /* helper function for performing netlink requests */
- static int
- swlib_call(int cmd, int (*call)(struct nl_msg *, void *),
- int (*data)(struct nl_msg *, void *), void *arg)
- {
- struct nl_msg *msg;
- struct nl_cb *cb = NULL;
- int finished;
- int flags = 0;
- int err;
- msg = nlmsg_alloc();
- if (!msg) {
- fprintf(stderr, "Out of memory!\n");
- exit(1);
- }
- if (!data)
- flags |= NLM_F_DUMP;
- genlmsg_put(msg, NL_AUTO_PID, NL_AUTO_SEQ, genl_family_get_id(family), 0, flags, cmd, 0);
- if (data) {
- if (data(msg, arg) < 0)
- goto nla_put_failure;
- }
- cb = nl_cb_alloc(NL_CB_CUSTOM);
- if (!cb) {
- fprintf(stderr, "nl_cb_alloc failed.\n");
- exit(1);
- }
- err = nl_send_auto_complete(handle, msg);
- if (err < 0) {
- fprintf(stderr, "nl_send_auto_complete failed: %d\n", err);
- goto out;
- }
- finished = 0;
- if (call)
- nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, call, arg);
- if (data)
- nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, wait_handler, &finished);
- else
- nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, wait_handler, &finished);
- err = nl_recvmsgs(handle, cb);
- if (err < 0) {
- goto out;
- }
- if (!finished)
- err = nl_wait_for_ack(handle);
- out:
- if (cb)
- nl_cb_put(cb);
- nla_put_failure:
- nlmsg_free(msg);
- return err;
- }
- static int
- send_attr(struct nl_msg *msg, void *arg)
- {
- struct switch_val *val = arg;
- struct switch_attr *attr = val->attr;
- NLA_PUT_U32(msg, SWITCH_ATTR_ID, attr->dev->id);
- NLA_PUT_U32(msg, SWITCH_ATTR_OP_ID, attr->id);
- switch(attr->atype) {
- case SWLIB_ATTR_GROUP_PORT:
- NLA_PUT_U32(msg, SWITCH_ATTR_OP_PORT, val->port_vlan);
- break;
- case SWLIB_ATTR_GROUP_VLAN:
- NLA_PUT_U32(msg, SWITCH_ATTR_OP_VLAN, val->port_vlan);
- break;
- default:
- break;
- }
- return 0;
- nla_put_failure:
- return -1;
- }
- static int
- store_port_val(struct nl_msg *msg, struct nlattr *nla, struct switch_val *val)
- {
- struct nlattr *p;
- int ports = val->attr->dev->ports;
- int err = 0;
- int remaining;
- if (!val->value.ports)
- val->value.ports = malloc(sizeof(struct switch_port) * ports);
- nla_for_each_nested(p, nla, remaining) {
- struct nlattr *tb[SWITCH_PORT_ATTR_MAX+1];
- struct switch_port *port;
- if (val->len >= ports)
- break;
- err = nla_parse_nested(tb, SWITCH_PORT_ATTR_MAX, p, port_policy);
- if (err < 0)
- goto out;
- if (!tb[SWITCH_PORT_ID])
- continue;
- port = &val->value.ports[val->len];
- port->id = nla_get_u32(tb[SWITCH_PORT_ID]);
- port->flags = 0;
- if (tb[SWITCH_PORT_FLAG_TAGGED])
- port->flags |= SWLIB_PORT_FLAG_TAGGED;
- val->len++;
- }
- out:
- return err;
- }
- static int
- store_val(struct nl_msg *msg, void *arg)
- {
- struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
- struct switch_val *val = arg;
- if (!val)
- goto error;
- if (nla_parse(tb, SWITCH_ATTR_MAX - 1, genlmsg_attrdata(gnlh, 0),
- genlmsg_attrlen(gnlh, 0), NULL) < 0) {
- goto error;
- }
- if (tb[SWITCH_ATTR_OP_VALUE_INT])
- val->value.i = nla_get_u32(tb[SWITCH_ATTR_OP_VALUE_INT]);
- else if (tb[SWITCH_ATTR_OP_VALUE_STR])
- val->value.s = strdup(nla_get_string(tb[SWITCH_ATTR_OP_VALUE_STR]));
- else if (tb[SWITCH_ATTR_OP_VALUE_PORTS])
- val->err = store_port_val(msg, tb[SWITCH_ATTR_OP_VALUE_PORTS], val);
- val->err = 0;
- return 0;
- error:
- return NL_SKIP;
- }
- int
- swlib_get_attr(struct switch_dev *dev, struct switch_attr *attr, struct switch_val *val)
- {
- int cmd;
- int err;
- switch(attr->atype) {
- case SWLIB_ATTR_GROUP_GLOBAL:
- cmd = SWITCH_CMD_GET_GLOBAL;
- break;
- case SWLIB_ATTR_GROUP_PORT:
- cmd = SWITCH_CMD_GET_PORT;
- break;
- case SWLIB_ATTR_GROUP_VLAN:
- cmd = SWITCH_CMD_GET_VLAN;
- break;
- default:
- return -EINVAL;
- }
- memset(&val->value, 0, sizeof(val->value));
- val->len = 0;
- val->attr = attr;
- val->err = -EINVAL;
- err = swlib_call(cmd, store_val, send_attr, val);
- if (!err)
- err = val->err;
- return err;
- }
- static int
- send_attr_ports(struct nl_msg *msg, struct switch_val *val)
- {
- struct nlattr *n;
- int i;
- /* TODO implement multipart? */
- if (val->len == 0)
- goto done;
- n = nla_nest_start(msg, SWITCH_ATTR_OP_VALUE_PORTS);
- if (!n)
- goto nla_put_failure;
- for (i = 0; i < val->len; i++) {
- struct switch_port *port = &val->value.ports[i];
- struct nlattr *np;
- np = nla_nest_start(msg, SWITCH_ATTR_PORT);
- if (!np)
- goto nla_put_failure;
- NLA_PUT_U32(msg, SWITCH_PORT_ID, port->id);
- if (port->flags & SWLIB_PORT_FLAG_TAGGED)
- NLA_PUT_FLAG(msg, SWITCH_PORT_FLAG_TAGGED);
- nla_nest_end(msg, np);
- }
- nla_nest_end(msg, n);
- done:
- return 0;
- nla_put_failure:
- return -1;
- }
- static int
- send_attr_val(struct nl_msg *msg, void *arg)
- {
- struct switch_val *val = arg;
- struct switch_attr *attr = val->attr;
- if (send_attr(msg, arg))
- goto nla_put_failure;
- switch(attr->type) {
- case SWITCH_TYPE_NOVAL:
- break;
- case SWITCH_TYPE_INT:
- NLA_PUT_U32(msg, SWITCH_ATTR_OP_VALUE_INT, val->value.i);
- break;
- case SWITCH_TYPE_STRING:
- if (!val->value.s)
- goto nla_put_failure;
- NLA_PUT_STRING(msg, SWITCH_ATTR_OP_VALUE_STR, val->value.s);
- break;
- case SWITCH_TYPE_PORTS:
- if (send_attr_ports(msg, val) < 0)
- goto nla_put_failure;
- break;
- default:
- goto nla_put_failure;
- }
- return 0;
- nla_put_failure:
- return -1;
- }
- int
- swlib_set_attr(struct switch_dev *dev, struct switch_attr *attr, struct switch_val *val)
- {
- int cmd;
- switch(attr->atype) {
- case SWLIB_ATTR_GROUP_GLOBAL:
- cmd = SWITCH_CMD_SET_GLOBAL;
- break;
- case SWLIB_ATTR_GROUP_PORT:
- cmd = SWITCH_CMD_SET_PORT;
- break;
- case SWLIB_ATTR_GROUP_VLAN:
- cmd = SWITCH_CMD_SET_VLAN;
- break;
- default:
- return -EINVAL;
- }
- val->attr = attr;
- return swlib_call(cmd, NULL, send_attr_val, val);
- }
- int swlib_set_attr_string(struct switch_dev *dev, struct switch_attr *a, int port_vlan, const char *str)
- {
- struct switch_port *ports;
- struct switch_val val;
- char *ptr;
- memset(&val, 0, sizeof(val));
- val.port_vlan = port_vlan;
- switch(a->type) {
- case SWITCH_TYPE_INT:
- val.value.i = atoi(str);
- break;
- case SWITCH_TYPE_STRING:
- val.value.s = str;
- break;
- case SWITCH_TYPE_PORTS:
- ports = alloca(sizeof(struct switch_port) * dev->ports);
- memset(ports, 0, sizeof(struct switch_port) * dev->ports);
- val.len = 0;
- ptr = (char *)str;
- while(ptr && *ptr)
- {
- while(*ptr && isspace(*ptr))
- ptr++;
- if (!*ptr)
- break;
- if (!isdigit(*ptr))
- return -1;
- if (val.len >= dev->ports)
- return -1;
- ports[val.len].flags = 0;
- ports[val.len].id = strtoul(ptr, &ptr, 10);
- while(*ptr && !isspace(*ptr)) {
- if (*ptr == 't')
- ports[val.len].flags |= SWLIB_PORT_FLAG_TAGGED;
- else
- return -1;
- ptr++;
- }
- if (*ptr)
- ptr++;
- val.len++;
- }
- val.value.ports = ports;
- break;
- case SWITCH_TYPE_NOVAL:
- if (str && !strcmp(str, "0"))
- return 0;
- break;
- default:
- return -1;
- }
- return swlib_set_attr(dev, a, &val);
- }
- struct attrlist_arg {
- int id;
- int atype;
- struct switch_dev *dev;
- struct switch_attr *prev;
- struct switch_attr **head;
- };
- static int
- add_id(struct nl_msg *msg, void *arg)
- {
- struct attrlist_arg *l = arg;
- NLA_PUT_U32(msg, SWITCH_ATTR_ID, l->id);
- return 0;
- nla_put_failure:
- return -1;
- }
- static int
- add_attr(struct nl_msg *msg, void *ptr)
- {
- struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
- struct attrlist_arg *arg = ptr;
- struct switch_attr *new;
- if (nla_parse(tb, SWITCH_ATTR_MAX - 1, genlmsg_attrdata(gnlh, 0),
- genlmsg_attrlen(gnlh, 0), NULL) < 0)
- goto done;
- new = swlib_alloc(sizeof(struct switch_attr));
- if (!new)
- goto done;
- new->dev = arg->dev;
- new->atype = arg->atype;
- if (arg->prev) {
- arg->prev->next = new;
- } else {
- arg->prev = *arg->head;
- }
- *arg->head = new;
- arg->head = &new->next;
- if (tb[SWITCH_ATTR_OP_ID])
- new->id = nla_get_u32(tb[SWITCH_ATTR_OP_ID]);
- if (tb[SWITCH_ATTR_OP_TYPE])
- new->type = nla_get_u32(tb[SWITCH_ATTR_OP_TYPE]);
- if (tb[SWITCH_ATTR_OP_NAME])
- new->name = strdup(nla_get_string(tb[SWITCH_ATTR_OP_NAME]));
- if (tb[SWITCH_ATTR_OP_DESCRIPTION])
- new->description = strdup(nla_get_string(tb[SWITCH_ATTR_OP_DESCRIPTION]));
- done:
- return NL_SKIP;
- }
- int
- swlib_scan(struct switch_dev *dev)
- {
- struct attrlist_arg arg;
- if (dev->ops || dev->port_ops || dev->vlan_ops)
- return 0;
- arg.atype = SWLIB_ATTR_GROUP_GLOBAL;
- arg.dev = dev;
- arg.id = dev->id;
- arg.prev = NULL;
- arg.head = &dev->ops;
- swlib_call(SWITCH_CMD_LIST_GLOBAL, add_attr, add_id, &arg);
- arg.atype = SWLIB_ATTR_GROUP_PORT;
- arg.prev = NULL;
- arg.head = &dev->port_ops;
- swlib_call(SWITCH_CMD_LIST_PORT, add_attr, add_id, &arg);
- arg.atype = SWLIB_ATTR_GROUP_VLAN;
- arg.prev = NULL;
- arg.head = &dev->vlan_ops;
- swlib_call(SWITCH_CMD_LIST_VLAN, add_attr, add_id, &arg);
- return 0;
- }
- struct switch_attr *swlib_lookup_attr(struct switch_dev *dev,
- enum swlib_attr_group atype, const char *name)
- {
- struct switch_attr *head;
- if (!name || !dev)
- return NULL;
- switch(atype) {
- case SWLIB_ATTR_GROUP_GLOBAL:
- head = dev->ops;
- break;
- case SWLIB_ATTR_GROUP_PORT:
- head = dev->port_ops;
- break;
- case SWLIB_ATTR_GROUP_VLAN:
- head = dev->vlan_ops;
- break;
- }
- while(head) {
- if (!strcmp(name, head->name))
- return head;
- head = head->next;
- }
- return NULL;
- }
- static void
- swlib_priv_free(void)
- {
- if (cache)
- nl_cache_free(cache);
- if (handle)
- nl_socket_free(handle);
- handle = NULL;
- cache = NULL;
- }
- static int
- swlib_priv_init(void)
- {
- int ret;
- handle = nl_socket_alloc();
- if (!handle) {
- DPRINTF("Failed to create handle\n");
- goto err;
- }
- if (genl_connect(handle)) {
- DPRINTF("Failed to connect to generic netlink\n");
- goto err;
- }
- ret = genl_ctrl_alloc_cache(handle, &cache);
- if (ret < 0) {
- DPRINTF("Failed to allocate netlink cache\n");
- goto err;
- }
- family = genl_ctrl_search_by_name(cache, "switch");
- if (!family) {
- DPRINTF("Switch API not present\n");
- goto err;
- }
- return 0;
- err:
- swlib_priv_free();
- return -EINVAL;
- }
- struct swlib_scan_arg {
- const char *name;
- struct switch_dev *head;
- struct switch_dev *ptr;
- };
- static int
- add_port_map(struct switch_dev *dev, struct nlattr *nla)
- {
- struct nlattr *p;
- int err = 0, idx = 0;
- int remaining;
- dev->maps = malloc(sizeof(struct switch_portmap) * dev->ports);
- if (!dev->maps)
- return -1;
- memset(dev->maps, 0, sizeof(struct switch_portmap) * dev->ports);
- nla_for_each_nested(p, nla, remaining) {
- struct nlattr *tb[SWITCH_PORTMAP_MAX+1];
- if (idx >= dev->ports)
- continue;
- err = nla_parse_nested(tb, SWITCH_PORTMAP_MAX, p, portmap_policy);
- if (err < 0)
- continue;
- if (tb[SWITCH_PORTMAP_SEGMENT] && tb[SWITCH_PORTMAP_VIRT]) {
- dev->maps[idx].segment = strdup(nla_get_string(tb[SWITCH_PORTMAP_SEGMENT]));
- dev->maps[idx].virt = nla_get_u32(tb[SWITCH_PORTMAP_VIRT]);
- }
- idx++;
- }
- out:
- return err;
- }
- static int
- add_switch(struct nl_msg *msg, void *arg)
- {
- struct swlib_scan_arg *sa = arg;
- struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
- struct switch_dev *dev;
- const char *name;
- const char *alias;
- if (nla_parse(tb, SWITCH_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NULL) < 0)
- goto done;
- if (!tb[SWITCH_ATTR_DEV_NAME])
- goto done;
- name = nla_get_string(tb[SWITCH_ATTR_DEV_NAME]);
- alias = nla_get_string(tb[SWITCH_ATTR_ALIAS]);
- if (sa->name && (strcmp(name, sa->name) != 0) && (strcmp(alias, sa->name) != 0))
- goto done;
- dev = swlib_alloc(sizeof(struct switch_dev));
- if (!dev)
- goto done;
- strncpy(dev->dev_name, name, IFNAMSIZ - 1);
- dev->alias = strdup(alias);
- if (tb[SWITCH_ATTR_ID])
- dev->id = nla_get_u32(tb[SWITCH_ATTR_ID]);
- if (tb[SWITCH_ATTR_NAME])
- dev->name = strdup(nla_get_string(tb[SWITCH_ATTR_NAME]));
- if (tb[SWITCH_ATTR_PORTS])
- dev->ports = nla_get_u32(tb[SWITCH_ATTR_PORTS]);
- if (tb[SWITCH_ATTR_VLANS])
- dev->vlans = nla_get_u32(tb[SWITCH_ATTR_VLANS]);
- if (tb[SWITCH_ATTR_CPU_PORT])
- dev->cpu_port = nla_get_u32(tb[SWITCH_ATTR_CPU_PORT]);
- if (tb[SWITCH_ATTR_PORTMAP])
- add_port_map(dev, tb[SWITCH_ATTR_PORTMAP]);
- if (!sa->head) {
- sa->head = dev;
- sa->ptr = dev;
- } else {
- sa->ptr->next = dev;
- sa->ptr = dev;
- }
- refcount++;
- done:
- return NL_SKIP;
- }
- static int
- list_switch(struct nl_msg *msg, void *arg)
- {
- struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
- if (nla_parse(tb, SWITCH_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NULL) < 0)
- goto done;
- if (!tb[SWITCH_ATTR_DEV_NAME] || !tb[SWITCH_ATTR_NAME])
- goto done;
- printf("Found: %s - %s\n", nla_get_string(tb[SWITCH_ATTR_DEV_NAME]),
- nla_get_string(tb[SWITCH_ATTR_ALIAS]));
- done:
- return NL_SKIP;
- }
- void
- swlib_list(void)
- {
- if (swlib_priv_init() < 0)
- return;
- swlib_call(SWITCH_CMD_GET_SWITCH, list_switch, NULL, NULL);
- swlib_priv_free();
- }
- void
- swlib_print_portmap(struct switch_dev *dev, char *segment)
- {
- int i;
- if (segment) {
- if (!strcmp(segment, "cpu")) {
- printf("%d ", dev->cpu_port);
- } else if (!strcmp(segment, "disabled")) {
- for (i = 0; i < dev->ports; i++)
- if (!dev->maps[i].segment)
- printf("%d ", i);
- } else for (i = 0; i < dev->ports; i++) {
- if (dev->maps[i].segment && !strcmp(dev->maps[i].segment, segment))
- printf("%d ", i);
- }
- } else {
- printf("%s - %s\n", dev->dev_name, dev->name);
- for (i = 0; i < dev->ports; i++)
- if (i == dev->cpu_port)
- printf("port%d:\tcpu\n", i);
- else if (dev->maps[i].segment)
- printf("port%d:\t%s.%d\n", i, dev->maps[i].segment, dev->maps[i].virt);
- else
- printf("port%d:\tdisabled\n", i);
- }
- }
- struct switch_dev *
- swlib_connect(const char *name)
- {
- struct swlib_scan_arg arg;
- if (!refcount) {
- if (swlib_priv_init() < 0)
- return NULL;
- };
- arg.head = NULL;
- arg.ptr = NULL;
- arg.name = name;
- swlib_call(SWITCH_CMD_GET_SWITCH, add_switch, NULL, &arg);
- if (!refcount)
- swlib_priv_free();
- return arg.head;
- }
- static void
- swlib_free_attributes(struct switch_attr **head)
- {
- struct switch_attr *a = *head;
- struct switch_attr *next;
- while (a) {
- next = a->next;
- free(a);
- a = next;
- }
- *head = NULL;
- }
- void
- swlib_free(struct switch_dev *dev)
- {
- swlib_free_attributes(&dev->ops);
- swlib_free_attributes(&dev->port_ops);
- swlib_free_attributes(&dev->vlan_ops);
- free(dev);
- if (--refcount == 0)
- swlib_priv_free();
- }
- void
- swlib_free_all(struct switch_dev *dev)
- {
- struct switch_dev *p;
- while (dev) {
- p = dev->next;
- swlib_free(dev);
- dev = p;
- }
- }
|