]> git.proxmox.com Git - mirror_iproute2.git/blobdiff - lib/ll_map.c
ll_map: Add function to remove link cache entry by index
[mirror_iproute2.git] / lib / ll_map.c
index e5a95e6a4fa09a0289b358d4b118a3213297389c..8e8a0b1e9c9d9880bcbecccffd225fe4ebe9a1c6 100644 (file)
 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
-#include <syslog.h>
 #include <fcntl.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <string.h>
+#include <net/if.h>
 
 #include "libnetlink.h"
 #include "ll_map.h"
+#include "list.h"
 
-struct idxmap
-{
-       struct idxmap * next;
-       int             index;
-       int             type;
-       int             alen;
+struct ll_cache {
+       struct hlist_node idx_hash;
+       struct hlist_node name_hash;
        unsigned        flags;
-       unsigned char   addr[8];
-       char            name[16];
+       unsigned        index;
+       unsigned short  type;
+       char            name[];
 };
 
-static struct idxmap *idxmap[16];
+#define IDXMAP_SIZE    1024
+static struct hlist_head idx_head[IDXMAP_SIZE];
+static struct hlist_head name_head[IDXMAP_SIZE];
+
+static struct ll_cache *ll_get_by_index(unsigned index)
+{
+       struct hlist_node *n;
+       unsigned h = index & (IDXMAP_SIZE - 1);
+
+       hlist_for_each(n, &idx_head[h]) {
+               struct ll_cache *im
+                       = container_of(n, struct ll_cache, idx_hash);
+               if (im->index == index)
+                       return im;
+       }
+
+       return NULL;
+}
+
+unsigned namehash(const char *str)
+{
+       unsigned hash = 5381;
+
+       while (*str)
+               hash = ((hash << 5) + hash) + *str++; /* hash * 33 + c */
+
+       return hash;
+}
+
+static struct ll_cache *ll_get_by_name(const char *name)
+{
+       struct hlist_node *n;
+       unsigned h = namehash(name) & (IDXMAP_SIZE - 1);
+
+       hlist_for_each(n, &name_head[h]) {
+               struct ll_cache *im
+                       = container_of(n, struct ll_cache, name_hash);
+
+               if (strncmp(im->name, name, IFNAMSIZ) == 0)
+                       return im;
+       }
+
+       return NULL;
+}
 
-int ll_remember_index(struct sockaddr_nl *who, struct nlmsghdr *n, void *arg)
+int ll_remember_index(struct nlmsghdr *n, void *arg)
 {
-       int h;
+       unsigned int h;
+       const char *ifname;
        struct ifinfomsg *ifi = NLMSG_DATA(n);
-       struct idxmap *im, **imp;
+       struct ll_cache *im;
        struct rtattr *tb[IFLA_MAX+1];
 
-       if (n->nlmsg_type != RTM_NEWLINK)
+       if (n->nlmsg_type != RTM_NEWLINK && n->nlmsg_type != RTM_DELLINK)
                return 0;
 
-       if (n->nlmsg_len < NLMSG_LENGTH(sizeof(ifi)))
+       if (n->nlmsg_len < NLMSG_LENGTH(sizeof(*ifi)))
                return -1;
 
+       im = ll_get_by_index(ifi->ifi_index);
+       if (n->nlmsg_type == RTM_DELLINK) {
+               if (im) {
+                       hlist_del(&im->name_hash);
+                       hlist_del(&im->idx_hash);
+                       free(im);
+               }
+               return 0;
+       }
 
-       memset(tb, 0, sizeof(tb));
        parse_rtattr(tb, IFLA_MAX, IFLA_RTA(ifi), IFLA_PAYLOAD(n));
-       if (tb[IFLA_IFNAME] == NULL)
+       ifname = rta_getattr_str(tb[IFLA_IFNAME]);
+       if (ifname == NULL)
                return 0;
 
-       h = ifi->ifi_index&0xF;
-
-       for (imp=&idxmap[h]; (im=*imp)!=NULL; imp = &im->next)
-               if (im->index == ifi->ifi_index)
-                       break;
+       if (im) {
+               /* change to existing entry */
+               if (strcmp(im->name, ifname) != 0) {
+                       hlist_del(&im->name_hash);
+                       h = namehash(ifname) & (IDXMAP_SIZE - 1);
+                       hlist_add_head(&im->name_hash, &name_head[h]);
+               }
 
-       if (im == NULL) {
-               im = malloc(sizeof(*im));
-               if (im == NULL)
-                       return 0;
-               im->next = *imp;
-               im->index = ifi->ifi_index;
-               *imp = im;
+               im->flags = ifi->ifi_flags;
+               return 0;
        }
 
+       im = malloc(sizeof(*im) + strlen(ifname) + 1);
+       if (im == NULL)
+               return 0;
+       im->index = ifi->ifi_index;
+       strcpy(im->name, ifname);
        im->type = ifi->ifi_type;
        im->flags = ifi->ifi_flags;
-       if (tb[IFLA_ADDRESS]) {
-               int alen;
-               im->alen = alen = RTA_PAYLOAD(tb[IFLA_ADDRESS]);
-               if (alen > sizeof(im->addr))
-                       alen = sizeof(im->addr);
-               memcpy(im->addr, RTA_DATA(tb[IFLA_ADDRESS]), alen);
-       } else {
-               im->alen = 0;
-               memset(im->addr, 0, sizeof(im->addr));
-       }
-       strcpy(im->name, RTA_DATA(tb[IFLA_IFNAME]));
+
+       h = ifi->ifi_index & (IDXMAP_SIZE - 1);
+       hlist_add_head(&im->idx_hash, &idx_head[h]);
+
+       h = namehash(ifname) & (IDXMAP_SIZE - 1);
+       hlist_add_head(&im->name_hash, &name_head[h]);
+
        return 0;
 }
 
-const char *ll_idx_n2a(int idx, char *buf)
+const char *ll_idx_n2a(unsigned int idx)
 {
-       struct idxmap *im;
+       static char buf[IFNAMSIZ];
 
-       if (idx == 0)
-               return "*";
-       for (im = idxmap[idx&0xF]; im; im = im->next)
-               if (im->index == idx)
-                       return im->name;
-       snprintf(buf, 16, "if%d", idx);
+       snprintf(buf, sizeof(buf), "if%u", idx);
        return buf;
 }
 
+static unsigned int ll_idx_a2n(const char *name)
+{
+       unsigned int idx;
+
+       if (sscanf(name, "if%u", &idx) != 1)
+               return 0;
+       return idx;
+}
 
-const char *ll_index_to_name(int idx)
+const char *ll_index_to_name(unsigned int idx)
 {
-       static char nbuf[16];
+       static char buf[IFNAMSIZ];
+       const struct ll_cache *im;
+
+       if (idx == 0)
+               return "*";
 
-       return ll_idx_n2a(idx, nbuf);
+       im = ll_get_by_index(idx);
+       if (im)
+               return im->name;
+
+       if (if_indextoname(idx, buf) == NULL)
+               snprintf(buf, IFNAMSIZ, "if%u", idx);
+
+       return buf;
 }
 
-int ll_index_to_type(int idx)
+int ll_index_to_type(unsigned idx)
 {
-       struct idxmap *im;
+       const struct ll_cache *im;
 
        if (idx == 0)
                return -1;
-       for (im = idxmap[idx&0xF]; im; im = im->next)
-               if (im->index == idx)
-                       return im->type;
-       return -1;
+
+       im = ll_get_by_index(idx);
+       return im ? im->type : -1;
 }
 
-unsigned ll_index_to_flags(int idx)
+int ll_index_to_flags(unsigned idx)
 {
-       struct idxmap *im;
+       const struct ll_cache *im;
 
        if (idx == 0)
                return 0;
 
-       for (im = idxmap[idx&0xF]; im; im = im->next)
-               if (im->index == idx)
-                       return im->flags;
-       return 0;
+       im = ll_get_by_index(idx);
+       return im ? im->flags : -1;
 }
 
-int ll_name_to_index(char *name)
+unsigned ll_name_to_index(const char *name)
 {
-       static char ncache[16];
-       static int icache;
-       struct idxmap *im;
-       int i;
+       const struct ll_cache *im;
+       unsigned idx;
 
        if (name == NULL)
                return 0;
-       if (icache && strcmp(name, ncache) == 0)
-               return icache;
-       for (i=0; i<16; i++) {
-               for (im = idxmap[i]; im; im = im->next) {
-                       if (strcmp(im->name, name) == 0) {
-                               icache = im->index;
-                               strcpy(ncache, name);
-                               return im->index;
-                       }
-               }
-       }
-       return 0;
+
+       im = ll_get_by_name(name);
+       if (im)
+               return im->index;
+
+       idx = if_nametoindex(name);
+       if (idx == 0)
+               idx = ll_idx_a2n(name);
+       return idx;
+}
+
+void ll_drop_by_index(unsigned index)
+{
+       struct ll_cache *im;
+
+       im = ll_get_by_index(index);
+       if (!im)
+               return;
+
+       hlist_del(&im->idx_hash);
+       hlist_del(&im->name_hash);
+
+       free(im);
 }
 
-int ll_init_map(struct rtnl_handle *rth)
+void ll_init_map(struct rtnl_handle *rth)
 {
-       if (rtnl_wilddump_request(rth, AF_UNSPEC, RTM_GETLINK) < 0) {
+       static int initialized;
+
+       if (initialized)
+               return;
+
+       if (rtnl_linkdump_req(rth, AF_UNSPEC) < 0) {
                perror("Cannot send dump request");
                exit(1);
        }
 
-       if (rtnl_dump_filter(rth, ll_remember_index, &idxmap, NULL, NULL) < 0) {
+       if (rtnl_dump_filter(rth, ll_remember_index, NULL) < 0) {
                fprintf(stderr, "Dump terminated\n");
                exit(1);
        }
-       return 0;
+
+       initialized = 1;
 }