/* * swlib.c: Switch configuration API (user space part) * * Copyright (C) 2008 Felix Fietkau * * 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 #include #include #include #include #include #include #include #include #include #include "swlib.h" #include #include #include //#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 struct nla_policy link_policy[SWITCH_LINK_ATTR_MAX] = { [SWITCH_LINK_FLAG_LINK] = { .type = NLA_FLAG }, [SWITCH_LINK_FLAG_DUPLEX] = { .type = NLA_FLAG }, [SWITCH_LINK_FLAG_ANEG] = { .type = NLA_FLAG }, [SWITCH_LINK_SPEED] = { .type = NLA_U32 }, [SWITCH_LINK_FLAG_EEE_100BASET] = { .type = NLA_FLAG }, [SWITCH_LINK_FLAG_EEE_1000BASET] = { .type = NLA_FLAG }, }; 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_link_val(struct nl_msg *msg, struct nlattr *nla, struct switch_val *val) { struct nlattr *tb[SWITCH_LINK_ATTR_MAX + 1]; struct switch_port_link *link; int err = 0; if (!val->value.link) val->value.link = malloc(sizeof(struct switch_port_link)); err = nla_parse_nested(tb, SWITCH_LINK_ATTR_MAX, nla, link_policy); if (err < 0) goto out; link = val->value.link; link->link = !!tb[SWITCH_LINK_FLAG_LINK]; link->duplex = !!tb[SWITCH_LINK_FLAG_DUPLEX]; link->aneg = !!tb[SWITCH_LINK_FLAG_ANEG]; link->tx_flow = !!tb[SWITCH_LINK_FLAG_TX_FLOW]; link->rx_flow = !!tb[SWITCH_LINK_FLAG_RX_FLOW]; link->speed = nla_get_u32(tb[SWITCH_LINK_SPEED]); link->eee = 0; if (tb[SWITCH_LINK_FLAG_EEE_100BASET]) link->eee |= SWLIB_LINK_FLAG_EEE_100BASET; if (tb[SWITCH_LINK_FLAG_EEE_1000BASET]) link->eee |= SWLIB_LINK_FLAG_EEE_1000BASET; 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); else if (tb[SWITCH_ATTR_OP_VALUE_LINK]) val->err = store_link_val(msg, tb[SWITCH_ATTR_OP_VALUE_LINK], 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 = (char *)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 (family) nl_object_put((struct nl_object*)family); if (cache) nl_cache_free(cache); if (handle) nl_socket_free(handle); family = NULL; 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->name); free(a->description); free(a); a = next; } *head = NULL; } static void swlib_free_port_map(struct switch_dev *dev) { int i; if (!dev || !dev->maps) return; for (i = 0; i < dev->ports; i++) free(dev->maps[i].segment); free(dev->maps); } void swlib_free(struct switch_dev *dev) { swlib_free_attributes(&dev->ops); swlib_free_attributes(&dev->port_ops); swlib_free_attributes(&dev->vlan_ops); swlib_free_port_map(dev); free(dev->name); free(dev->alias); 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; } }