1 /*
2 * ipaddrlabel.c "ip addrlabel"
3 *
4 * Copyright (C)2007 USAGI/WIDE Project
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 *
20 *
21 * Based on iprule.c.
22 *
23 * Authors: YOSHIFUJI Hideaki <yoshfuji@linux-ipv6.org>
24 *
25 */
26
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <unistd.h>
30 #include <syslog.h>
31 #include <fcntl.h>
32 #include <sys/socket.h>
33 #include <netinet/in.h>
34 #include <netinet/ip.h>
35 #include <arpa/inet.h>
36 #include <string.h>
37 #include <linux/types.h>
38 #include <linux/if_addrlabel.h>
39
40 #include "rt_names.h"
41 #include "utils.h"
42 #include "ip_common.h"
43
44 #define IFAL_RTA(r) ((struct rtattr*)(((char*)(r)) + NLMSG_ALIGN(sizeof(struct ifaddrlblmsg))))
45 #define IFAL_PAYLOAD(n) NLMSG_PAYLOAD(n,sizeof(struct ifaddrlblmsg))
46
47 extern struct rtnl_handle rth;
48
49 static void usage(void) __attribute__((noreturn));
50
usage(void)51 static void usage(void)
52 {
53 fprintf(stderr, "Usage: ip addrlabel [ list | add | del | flush ] prefix PREFIX [ dev DEV ] [ label LABEL ]\n");
54 exit(-1);
55 }
56
print_addrlabel(const struct sockaddr_nl * who,struct nlmsghdr * n,void * arg)57 int print_addrlabel(const struct sockaddr_nl *who, struct nlmsghdr *n, void *arg)
58 {
59 FILE *fp = (FILE*)arg;
60 struct ifaddrlblmsg *ifal = NLMSG_DATA(n);
61 int len = n->nlmsg_len;
62 struct rtattr *tb[IFAL_MAX+1];
63 char abuf[256];
64
65 if (n->nlmsg_type != RTM_NEWADDRLABEL && n->nlmsg_type != RTM_DELADDRLABEL)
66 return 0;
67
68 len -= NLMSG_LENGTH(sizeof(*ifal));
69 if (len < 0)
70 return -1;
71
72 parse_rtattr(tb, IFAL_MAX, IFAL_RTA(ifal), len);
73
74 if (n->nlmsg_type == RTM_DELADDRLABEL)
75 fprintf(fp, "Deleted ");
76
77 if (tb[IFAL_ADDRESS]) {
78 fprintf(fp, "prefix %s/%u ",
79 format_host(ifal->ifal_family,
80 RTA_PAYLOAD(tb[IFAL_ADDRESS]),
81 RTA_DATA(tb[IFAL_ADDRESS]),
82 abuf, sizeof(abuf)),
83 ifal->ifal_prefixlen);
84 }
85
86 if (ifal->ifal_index)
87 fprintf(fp, "dev %s ", ll_index_to_name(ifal->ifal_index));
88
89 if (tb[IFAL_LABEL] && RTA_PAYLOAD(tb[IFAL_LABEL]) == sizeof(int32_t)) {
90 int32_t label;
91 memcpy(&label, RTA_DATA(tb[IFAL_LABEL]), sizeof(label));
92 fprintf(fp, "label %d ", label);
93 }
94
95 fprintf(fp, "\n");
96 fflush(fp);
97 return 0;
98 }
99
ipaddrlabel_list(int argc,char ** argv)100 static int ipaddrlabel_list(int argc, char **argv)
101 {
102 int af = preferred_family;
103
104 if (af == AF_UNSPEC)
105 af = AF_INET6;
106
107 if (argc > 0) {
108 fprintf(stderr, "\"ip addrlabel show\" does not take any arguments.\n");
109 return -1;
110 }
111
112 if (rtnl_wilddump_request(&rth, af, RTM_GETADDRLABEL) < 0) {
113 perror("Cannot send dump request");
114 return 1;
115 }
116
117 if (rtnl_dump_filter(&rth, print_addrlabel, stdout) < 0) {
118 fprintf(stderr, "Dump terminated\n");
119 return 1;
120 }
121
122 return 0;
123 }
124
125
ipaddrlabel_modify(int cmd,int argc,char ** argv)126 static int ipaddrlabel_modify(int cmd, int argc, char **argv)
127 {
128 struct {
129 struct nlmsghdr n;
130 struct ifaddrlblmsg ifal;
131 char buf[1024];
132 } req;
133
134 inet_prefix prefix;
135 uint32_t label = 0xffffffffUL;
136 char *p = NULL;
137 char *l = NULL;
138
139 memset(&req, 0, sizeof(req));
140 memset(&prefix, 0, sizeof(prefix));
141
142 req.n.nlmsg_type = cmd;
143 req.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifaddrlblmsg));
144 req.n.nlmsg_flags = NLM_F_REQUEST;
145 req.ifal.ifal_family = preferred_family;
146 req.ifal.ifal_prefixlen = 0;
147 req.ifal.ifal_index = 0;
148
149 if (cmd == RTM_NEWADDRLABEL) {
150 req.n.nlmsg_flags |= NLM_F_CREATE|NLM_F_EXCL;
151 }
152
153 while (argc > 0) {
154 if (strcmp(*argv, "prefix") == 0) {
155 NEXT_ARG();
156 p = *argv;
157 get_prefix(&prefix, *argv, preferred_family);
158 } else if (strcmp(*argv, "dev") == 0) {
159 NEXT_ARG();
160 if ((req.ifal.ifal_index = ll_name_to_index(*argv)) == 0)
161 invarg("dev is invalid\n", *argv);
162 } else if (strcmp(*argv, "label") == 0) {
163 NEXT_ARG();
164 l = *argv;
165 if (get_u32(&label, *argv, 0) || label == 0xffffffffUL)
166 invarg("label is invalid\n", *argv);
167 }
168 argc--;
169 argv++;
170 }
171 if (p == NULL) {
172 fprintf(stderr, "Not enough information: \"prefix\" argument is required.\n");
173 return -1;
174 }
175 if (l == NULL) {
176 fprintf(stderr, "Not enough information: \"label\" argument is required.\n");
177 return -1;
178 }
179 addattr32(&req.n, sizeof(req), IFAL_LABEL, label);
180 addattr_l(&req.n, sizeof(req), IFAL_ADDRESS, &prefix.data, prefix.bytelen);
181 req.ifal.ifal_prefixlen = prefix.bitlen;
182
183 if (req.ifal.ifal_family == AF_UNSPEC)
184 req.ifal.ifal_family = AF_INET6;
185
186 if (rtnl_talk(&rth, &req.n, 0, 0, NULL) < 0)
187 return 2;
188
189 return 0;
190 }
191
192
flush_addrlabel(const struct sockaddr_nl * who,struct nlmsghdr * n,void * arg)193 static int flush_addrlabel(const struct sockaddr_nl *who, struct nlmsghdr *n, void *arg)
194 {
195 struct rtnl_handle rth2;
196 struct rtmsg *r = NLMSG_DATA(n);
197 int len = n->nlmsg_len;
198 struct rtattr * tb[IFAL_MAX+1];
199
200 len -= NLMSG_LENGTH(sizeof(*r));
201 if (len < 0)
202 return -1;
203
204 parse_rtattr(tb, IFAL_MAX, RTM_RTA(r), len);
205
206 if (tb[IFAL_ADDRESS]) {
207 n->nlmsg_type = RTM_DELADDRLABEL;
208 n->nlmsg_flags = NLM_F_REQUEST;
209
210 if (rtnl_open(&rth2, 0) < 0)
211 return -1;
212
213 if (rtnl_talk(&rth2, n, 0, 0, NULL) < 0)
214 return -2;
215
216 rtnl_close(&rth2);
217 }
218
219 return 0;
220 }
221
ipaddrlabel_flush(int argc,char ** argv)222 static int ipaddrlabel_flush(int argc, char **argv)
223 {
224 int af = preferred_family;
225
226 if (af == AF_UNSPEC)
227 af = AF_INET6;
228
229 if (argc > 0) {
230 fprintf(stderr, "\"ip addrlabel flush\" does not allow extra arguments\n");
231 return -1;
232 }
233
234 if (rtnl_wilddump_request(&rth, af, RTM_GETADDRLABEL) < 0) {
235 perror("Cannot send dump request");
236 return 1;
237 }
238
239 if (rtnl_dump_filter(&rth, flush_addrlabel, NULL) < 0) {
240 fprintf(stderr, "Flush terminated\n");
241 return 1;
242 }
243
244 return 0;
245 }
246
do_ipaddrlabel(int argc,char ** argv)247 int do_ipaddrlabel(int argc, char **argv)
248 {
249 ll_init_map(&rth);
250
251 if (argc < 1) {
252 return ipaddrlabel_list(0, NULL);
253 } else if (matches(argv[0], "list") == 0 ||
254 matches(argv[0], "show") == 0) {
255 return ipaddrlabel_list(argc-1, argv+1);
256 } else if (matches(argv[0], "add") == 0) {
257 return ipaddrlabel_modify(RTM_NEWADDRLABEL, argc-1, argv+1);
258 } else if (matches(argv[0], "delete") == 0) {
259 return ipaddrlabel_modify(RTM_DELADDRLABEL, argc-1, argv+1);
260 } else if (matches(argv[0], "flush") == 0) {
261 return ipaddrlabel_flush(argc-1, argv+1);
262 } else if (matches(argv[0], "help") == 0)
263 usage();
264
265 fprintf(stderr, "Command \"%s\" is unknown, try \"ip addrlabel help\".\n", *argv);
266 exit(-1);
267 }
268
269