libiproute: audit callgraph, shortcut error paths into die() functions.
[oweals/busybox.git] / networking / libiproute / ll_map.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * ll_map.c
4  *
5  *              This program is free software; you can redistribute it and/or
6  *              modify it under the terms of the GNU General Public License
7  *              as published by the Free Software Foundation; either version
8  *              2 of the License, or (at your option) any later version.
9  *
10  * Authors:     Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
11  *
12  */
13
14 #include "libbb.h"
15 #include <string.h>
16
17 #include "libnetlink.h"
18 #include "ll_map.h"
19
20 #include <sys/socket.h> /* socket() */
21 #include <net/if.h>     /* struct ifreq and co. */
22 #include <sys/ioctl.h>  /* ioctl() & SIOCGIFINDEX */
23
24 struct idxmap {
25         struct idxmap * next;
26         int             index;
27         int             type;
28         int             alen;
29         unsigned        flags;
30         unsigned char   addr[8];
31         char            name[16];
32 };
33
34 static struct idxmap *idxmap[16];
35
36 int ll_remember_index(struct sockaddr_nl *who, struct nlmsghdr *n, void *arg)
37 {
38         int h;
39         struct ifinfomsg *ifi = NLMSG_DATA(n);
40         struct idxmap *im, **imp;
41         struct rtattr *tb[IFLA_MAX+1];
42
43         if (n->nlmsg_type != RTM_NEWLINK)
44                 return 0;
45
46         if (n->nlmsg_len < NLMSG_LENGTH(sizeof(ifi)))
47                 return -1;
48
49
50         memset(tb, 0, sizeof(tb));
51         parse_rtattr(tb, IFLA_MAX, IFLA_RTA(ifi), IFLA_PAYLOAD(n));
52         if (tb[IFLA_IFNAME] == NULL)
53                 return 0;
54
55         h = ifi->ifi_index&0xF;
56
57         for (imp = &idxmap[h]; (im = *imp) != NULL; imp = &im->next)
58                 if (im->index == ifi->ifi_index)
59                         break;
60
61         if (im == NULL) {
62                 im = xmalloc(sizeof(*im));
63                 im->next = *imp;
64                 im->index = ifi->ifi_index;
65                 *imp = im;
66         }
67
68         im->type = ifi->ifi_type;
69         im->flags = ifi->ifi_flags;
70         if (tb[IFLA_ADDRESS]) {
71                 int alen;
72                 im->alen = alen = RTA_PAYLOAD(tb[IFLA_ADDRESS]);
73                 if (alen > sizeof(im->addr))
74                         alen = sizeof(im->addr);
75                 memcpy(im->addr, RTA_DATA(tb[IFLA_ADDRESS]), alen);
76         } else {
77                 im->alen = 0;
78                 memset(im->addr, 0, sizeof(im->addr));
79         }
80         strcpy(im->name, RTA_DATA(tb[IFLA_IFNAME]));
81         return 0;
82 }
83
84 const char *ll_idx_n2a(int idx, char *buf)
85 {
86         struct idxmap *im;
87
88         if (idx == 0)
89                 return "*";
90         for (im = idxmap[idx & 0xF]; im; im = im->next)
91                 if (im->index == idx)
92                         return im->name;
93         snprintf(buf, 16, "if%d", idx);
94         return buf;
95 }
96
97
98 const char *ll_index_to_name(int idx)
99 {
100         static char nbuf[16];
101
102         return ll_idx_n2a(idx, nbuf);
103 }
104
105 int ll_index_to_type(int idx)
106 {
107         struct idxmap *im;
108
109         if (idx == 0)
110                 return -1;
111         for (im = idxmap[idx & 0xF]; im; im = im->next)
112                 if (im->index == idx)
113                         return im->type;
114         return -1;
115 }
116
117 unsigned ll_index_to_flags(int idx)
118 {
119         struct idxmap *im;
120
121         if (idx == 0)
122                 return 0;
123
124         for (im = idxmap[idx & 0xF]; im; im = im->next)
125                 if (im->index == idx)
126                         return im->flags;
127         return 0;
128 }
129
130 // TODO: caching is not warranted - no users which repeatedly call it
131 int ll_name_to_index(char *name)
132 {
133         static char ncache[16];
134         static int icache;
135
136         struct idxmap *im;
137         int sock_fd;
138         int i;
139
140         if (name == NULL)
141                 return 0;
142         if (icache && strcmp(name, ncache) == 0)
143                 return icache;
144         for (i = 0; i < 16; i++) {
145                 for (im = idxmap[i]; im; im = im->next) {
146                         if (strcmp(im->name, name) == 0) {
147                                 icache = im->index;
148                                 strcpy(ncache, name);
149                                 return im->index;
150                         }
151                 }
152         }
153         /* We have not found the interface in our cache, but the kernel
154          * may still know about it. One reason is that we may be using
155          * module on-demand loading, which means that the kernel will
156          * load the module and make the interface exist only when
157          * we explicitely request it (check for dev_load() in net/core/dev.c).
158          * I can think of other similar scenario, but they are less common...
159          * Jean II */
160         sock_fd = socket(AF_INET, SOCK_DGRAM, 0);
161         if (sock_fd) {
162                 struct ifreq ifr;
163                 int ret;
164                 strncpy(ifr.ifr_name, name, IFNAMSIZ);
165                 ifr.ifr_ifindex = -1;
166                 ret = ioctl(sock_fd, SIOCGIFINDEX, &ifr);
167                 close(sock_fd);
168                 if (ret >= 0)
169                         /* In theory, we should redump the interface list
170                          * to update our cache, this is left as an exercise
171                          * to the reader... Jean II */
172                         return ifr.ifr_ifindex;
173         }
174
175         return 0;
176 }
177
178 int ll_init_map(struct rtnl_handle *rth)
179 {
180         if (rtnl_wilddump_request(rth, AF_UNSPEC, RTM_GETLINK) < 0) {
181                 bb_perror_msg_and_die("cannot send dump request");
182         }
183
184         if (rtnl_dump_filter(rth, ll_remember_index, &idxmap, NULL, NULL) < 0) {
185                 bb_error_msg_and_die("dump terminated");
186         }
187         return 0;
188 }