]>
Commit | Line | Data |
---|---|---|
4759758c YH |
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 | ||
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 | ||
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; | |
4759758c YH |
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 | ||
4759758c YH |
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 | ||
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 | ||
cd70f3f5 | 117 | if (rtnl_dump_filter(&rth, print_addrlabel, stdout) < 0) { |
4759758c YH |
118 | fprintf(stderr, "Dump terminated\n"); |
119 | return 1; | |
120 | } | |
121 | ||
122 | return 0; | |
123 | } | |
124 | ||
125 | ||
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; | |
6579feea VC |
136 | char *p = NULL; |
137 | char *l = NULL; | |
4759758c YH |
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(); | |
6579feea | 156 | p = *argv; |
4759758c YH |
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(); | |
6579feea | 164 | l = *argv; |
4759758c YH |
165 | if (get_u32(&label, *argv, 0) || label == 0xffffffffUL) |
166 | invarg("label is invalid\n", *argv); | |
167 | } | |
168 | argc--; | |
169 | argv++; | |
170 | } | |
6579feea VC |
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 | } | |
4759758c YH |
179 | addattr32(&req.n, sizeof(req), IFAL_LABEL, label); |
180 | addattr_l(&req.n, sizeof(req), IFAL_ADDRESS, &prefix.data, prefix.bytelen); | |
3490740b | 181 | req.ifal.ifal_prefixlen = prefix.bitlen; |
4759758c YH |
182 | |
183 | if (req.ifal.ifal_family == AF_UNSPEC) | |
184 | req.ifal.ifal_family = AF_INET6; | |
185 | ||
cd70f3f5 | 186 | if (rtnl_talk(&rth, &req.n, 0, 0, NULL) < 0) |
4759758c YH |
187 | return 2; |
188 | ||
189 | return 0; | |
190 | } | |
191 | ||
192 | ||
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 | ||
cd70f3f5 | 213 | if (rtnl_talk(&rth2, n, 0, 0, NULL) < 0) |
4759758c YH |
214 | return -2; |
215 | ||
216 | rtnl_close(&rth2); | |
217 | } | |
218 | ||
219 | return 0; | |
220 | } | |
221 | ||
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 | ||
cd70f3f5 | 239 | if (rtnl_dump_filter(&rth, flush_addrlabel, NULL) < 0) { |
4759758c YH |
240 | fprintf(stderr, "Flush terminated\n"); |
241 | return 1; | |
242 | } | |
243 | ||
244 | return 0; | |
245 | } | |
246 | ||
247 | int do_ipaddrlabel(int argc, char **argv) | |
248 | { | |
249 | if (argc < 1) { | |
250 | return ipaddrlabel_list(0, NULL); | |
251 | } else if (matches(argv[0], "list") == 0 || | |
252 | matches(argv[0], "show") == 0) { | |
253 | return ipaddrlabel_list(argc-1, argv+1); | |
254 | } else if (matches(argv[0], "add") == 0) { | |
255 | return ipaddrlabel_modify(RTM_NEWADDRLABEL, argc-1, argv+1); | |
256 | } else if (matches(argv[0], "delete") == 0) { | |
257 | return ipaddrlabel_modify(RTM_DELADDRLABEL, argc-1, argv+1); | |
258 | } else if (matches(argv[0], "flush") == 0) { | |
259 | return ipaddrlabel_flush(argc-1, argv+1); | |
260 | } else if (matches(argv[0], "help") == 0) | |
261 | usage(); | |
262 | ||
263 | fprintf(stderr, "Command \"%s\" is unknown, try \"ip addrlabel help\".\n", *argv); | |
264 | exit(-1); | |
265 | } | |
266 |