Contents of /trunk/mkinitrd-magellan/busybox/networking/arp.c
Parent Directory | Revision Log
Revision 532 -
(show annotations)
(download)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 11643 byte(s)
Sat Sep 1 22:45:15 2007 UTC (16 years, 8 months ago) by niro
File MIME type: text/plain
File size: 11643 byte(s)
-import if magellan mkinitrd; it is a fork of redhats mkinitrd-5.0.8 with all magellan patches and features; deprecates magellan-src/mkinitrd
1 | /* vi: set sw=4 ts=4: */ |
2 | /* |
3 | * arp.c - Manipulate the system ARP cache |
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 | * Author: Fred N. van Kempen, <waltje at uwalt.nl.mugnet.org> |
11 | * Busybox port: Paul van Gool <pvangool at mimotech.com> |
12 | * |
13 | * modified for getopt32 by Arne Bernin <arne [at] alamut.de> |
14 | */ |
15 | |
16 | #include "busybox.h" |
17 | #include "inet_common.h" |
18 | |
19 | #include <arpa/inet.h> |
20 | #include <net/if.h> |
21 | #include <net/if_arp.h> |
22 | #include <netinet/ether.h> |
23 | #include <netpacket/packet.h> |
24 | |
25 | #define DEBUG 0 |
26 | |
27 | #define DFLT_AF "inet" |
28 | #define DFLT_HW "ether" |
29 | |
30 | #define _PATH_PROCNET_ARP "/proc/net/arp" |
31 | |
32 | #define ARP_OPT_A (0x1) |
33 | #define ARP_OPT_p (0x2) |
34 | #define ARP_OPT_H (0x4) |
35 | #define ARP_OPT_t (0x8) |
36 | #define ARP_OPT_i (0x10) |
37 | #define ARP_OPT_a (0x20) |
38 | #define ARP_OPT_d (0x40) |
39 | #define ARP_OPT_n (0x80) /* do not resolve addresses */ |
40 | #define ARP_OPT_D (0x100) /* HW-address is devicename */ |
41 | #define ARP_OPT_s (0x200) |
42 | #define ARP_OPT_v (0x400 * DEBUG) /* debugging output flag */ |
43 | |
44 | |
45 | static const struct aftype *ap; /* current address family */ |
46 | static const struct hwtype *hw; /* current hardware type */ |
47 | static int sockfd; /* active socket descriptor */ |
48 | static smallint hw_set; /* flag if hw-type was set (-H) */ |
49 | static char *device = ""; /* current device */ |
50 | |
51 | static const char *const options[] = { |
52 | "pub", |
53 | "priv", |
54 | "temp", |
55 | "trail", |
56 | "dontpub", |
57 | "auto", |
58 | "dev", |
59 | "netmask", |
60 | NULL |
61 | }; |
62 | |
63 | /* Delete an entry from the ARP cache. */ |
64 | /* Called only from main, once */ |
65 | static int arp_del(char **args) |
66 | { |
67 | char host[128]; |
68 | struct arpreq req; |
69 | struct sockaddr sa; |
70 | int flags = 0; |
71 | int err; |
72 | |
73 | memset(&req, 0, sizeof(req)); |
74 | |
75 | /* Resolve the host name. */ |
76 | safe_strncpy(host, *args, 128); |
77 | if (ap->input(0, host, &sa) < 0) { |
78 | bb_herror_msg_and_die("%s", host); |
79 | } |
80 | |
81 | /* If a host has more than one address, use the correct one! */ |
82 | memcpy(&req.arp_pa, &sa, sizeof(struct sockaddr)); |
83 | |
84 | if (hw_set) |
85 | req.arp_ha.sa_family = hw->type; |
86 | |
87 | req.arp_flags = ATF_PERM; |
88 | args++; |
89 | while (*args != NULL) { |
90 | switch (index_in_str_array(options, *args)) { |
91 | case 0: /* "pub" */ |
92 | flags |= 1; |
93 | args++; |
94 | break; |
95 | case 1: /* "priv" */ |
96 | flags |= 2; |
97 | args++; |
98 | break; |
99 | case 2: /* "temp" */ |
100 | req.arp_flags &= ~ATF_PERM; |
101 | args++; |
102 | break; |
103 | case 3: /* "trail" */ |
104 | req.arp_flags |= ATF_USETRAILERS; |
105 | args++; |
106 | break; |
107 | case 4: /* "dontpub" */ |
108 | #ifdef HAVE_ATF_DONTPUB |
109 | req.arp_flags |= ATF_DONTPUB; |
110 | #else |
111 | bb_error_msg("feature ATF_DONTPUB is not supported"); |
112 | #endif |
113 | args++; |
114 | break; |
115 | case 5: /* "auto" */ |
116 | #ifdef HAVE_ATF_MAGIC |
117 | req.arp_flags |= ATF_MAGIC; |
118 | #else |
119 | bb_error_msg("feature ATF_MAGIC is not supported"); |
120 | #endif |
121 | args++; |
122 | break; |
123 | case 6: /* "dev" */ |
124 | if (*++args == NULL) |
125 | bb_show_usage(); |
126 | device = *args; |
127 | args++; |
128 | break; |
129 | case 7: /* "netmask" */ |
130 | if (*++args == NULL) |
131 | bb_show_usage(); |
132 | if (strcmp(*args, "255.255.255.255") != 0) { |
133 | safe_strncpy(host, *args, 128); |
134 | if (ap->input(0, host, &sa) < 0) { |
135 | bb_herror_msg_and_die("%s", host); |
136 | } |
137 | memcpy(&req.arp_netmask, &sa, sizeof(struct sockaddr)); |
138 | req.arp_flags |= ATF_NETMASK; |
139 | } |
140 | args++; |
141 | break; |
142 | default: |
143 | bb_show_usage(); |
144 | break; |
145 | } |
146 | } |
147 | if (flags == 0) |
148 | flags = 3; |
149 | |
150 | strncpy(req.arp_dev, device, sizeof(req.arp_dev)); |
151 | |
152 | err = -1; |
153 | |
154 | /* Call the kernel. */ |
155 | if (flags & 2) { |
156 | if (option_mask32 & ARP_OPT_v) |
157 | bb_error_msg("SIOCDARP(nopub)"); |
158 | err = ioctl(sockfd, SIOCDARP, &req); |
159 | if (err < 0) { |
160 | if (errno == ENXIO) { |
161 | if (flags & 1) |
162 | goto nopub; |
163 | printf("No ARP entry for %s\n", host); |
164 | return -1; |
165 | } |
166 | bb_perror_msg_and_die("SIOCDARP(priv)"); |
167 | } |
168 | } |
169 | if ((flags & 1) && err) { |
170 | nopub: |
171 | req.arp_flags |= ATF_PUBL; |
172 | if (option_mask32 & ARP_OPT_v) |
173 | bb_error_msg("SIOCDARP(pub)"); |
174 | if (ioctl(sockfd, SIOCDARP, &req) < 0) { |
175 | if (errno == ENXIO) { |
176 | printf("No ARP entry for %s\n", host); |
177 | return -1; |
178 | } |
179 | bb_perror_msg_and_die("SIOCDARP(pub)"); |
180 | } |
181 | } |
182 | return 0; |
183 | } |
184 | |
185 | /* Get the hardware address to a specified interface name */ |
186 | static void arp_getdevhw(char *ifname, struct sockaddr *sa, |
187 | const struct hwtype *hwt) |
188 | { |
189 | struct ifreq ifr; |
190 | const struct hwtype *xhw; |
191 | |
192 | strcpy(ifr.ifr_name, ifname); |
193 | if (ioctl(sockfd, SIOCGIFHWADDR, &ifr) < 0) { |
194 | bb_perror_msg_and_die("cant get HW-Address for '%s'", ifname); |
195 | } |
196 | if (hwt && (ifr.ifr_hwaddr.sa_family != hw->type)) { |
197 | bb_error_msg_and_die("protocol type mismatch"); |
198 | } |
199 | memcpy(sa, &(ifr.ifr_hwaddr), sizeof(struct sockaddr)); |
200 | |
201 | if (option_mask32 & ARP_OPT_v) { |
202 | xhw = get_hwntype(ifr.ifr_hwaddr.sa_family); |
203 | if (!xhw || !xhw->print) { |
204 | xhw = get_hwntype(-1); |
205 | } |
206 | bb_error_msg("device '%s' has HW address %s '%s'", |
207 | ifname, xhw->name, |
208 | xhw->print((char *) &ifr.ifr_hwaddr.sa_data)); |
209 | } |
210 | } |
211 | |
212 | /* Set an entry in the ARP cache. */ |
213 | /* Called only from main, once */ |
214 | static int arp_set(char **args) |
215 | { |
216 | char host[128]; |
217 | struct arpreq req; |
218 | struct sockaddr sa; |
219 | int flags; |
220 | |
221 | memset(&req, 0, sizeof(req)); |
222 | |
223 | safe_strncpy(host, *args++, 128); |
224 | if (ap->input(0, host, &sa) < 0) { |
225 | bb_herror_msg_and_die("%s", host); |
226 | } |
227 | /* If a host has more than one address, use the correct one! */ |
228 | memcpy(&req.arp_pa, &sa, sizeof(struct sockaddr)); |
229 | |
230 | /* Fetch the hardware address. */ |
231 | if (*args == NULL) { |
232 | bb_error_msg_and_die("need hardware address"); |
233 | } |
234 | if (option_mask32 & ARP_OPT_D) { |
235 | arp_getdevhw(*args++, &req.arp_ha, hw_set ? hw : NULL); |
236 | } else { |
237 | if (hw->input(*args++, &req.arp_ha) < 0) { |
238 | bb_error_msg_and_die("invalid hardware address"); |
239 | } |
240 | } |
241 | |
242 | /* Check out any modifiers. */ |
243 | flags = ATF_PERM | ATF_COM; |
244 | while (*args != NULL) { |
245 | switch (index_in_str_array(options, *args)) { |
246 | case 0: /* "pub" */ |
247 | flags |= ATF_PUBL; |
248 | args++; |
249 | break; |
250 | case 1: /* "priv" */ |
251 | flags &= ~ATF_PUBL; |
252 | args++; |
253 | break; |
254 | case 2: /* "temp" */ |
255 | flags &= ~ATF_PERM; |
256 | args++; |
257 | break; |
258 | case 3: /* "trail" */ |
259 | flags |= ATF_USETRAILERS; |
260 | args++; |
261 | break; |
262 | case 4: /* "dontpub" */ |
263 | #ifdef HAVE_ATF_DONTPUB |
264 | flags |= ATF_DONTPUB; |
265 | #else |
266 | bb_error_msg("feature ATF_DONTPUB is not supported"); |
267 | #endif |
268 | args++; |
269 | break; |
270 | case 5: /* "auto" */ |
271 | #ifdef HAVE_ATF_MAGIC |
272 | flags |= ATF_MAGIC; |
273 | #else |
274 | bb_error_msg("feature ATF_MAGIC is not supported"); |
275 | #endif |
276 | args++; |
277 | break; |
278 | case 6: /* "dev" */ |
279 | if (*++args == NULL) |
280 | bb_show_usage(); |
281 | device = *args; |
282 | args++; |
283 | break; |
284 | case 7: /* "netmask" */ |
285 | if (*++args == NULL) |
286 | bb_show_usage(); |
287 | if (strcmp(*args, "255.255.255.255") != 0) { |
288 | safe_strncpy(host, *args++, 128); |
289 | if (ap->input(0, host, &sa) < 0) { |
290 | bb_herror_msg_and_die("%s", host); |
291 | } |
292 | memcpy(&req.arp_netmask, &sa, sizeof(struct sockaddr)); |
293 | flags |= ATF_NETMASK; |
294 | } |
295 | args++; |
296 | break; |
297 | default: |
298 | bb_show_usage(); |
299 | break; |
300 | } |
301 | } |
302 | |
303 | /* Fill in the remainder of the request. */ |
304 | req.arp_flags = flags; |
305 | |
306 | strncpy(req.arp_dev, device, sizeof(req.arp_dev)); |
307 | |
308 | /* Call the kernel. */ |
309 | if (option_mask32 & ARP_OPT_v) |
310 | bb_error_msg("SIOCSARP()"); |
311 | if (ioctl(sockfd, SIOCSARP, &req) < 0) { |
312 | bb_perror_msg_and_die("SIOCSARP"); |
313 | } |
314 | return 0; |
315 | } |
316 | |
317 | |
318 | /* Print the contents of an ARP request block. */ |
319 | static void |
320 | arp_disp(char *name, char *ip, int type, int arp_flags, |
321 | char *hwa, char *mask, char *dev) |
322 | { |
323 | const struct hwtype *xhw; |
324 | |
325 | xhw = get_hwntype(type); |
326 | if (xhw == NULL) |
327 | xhw = get_hwtype(DFLT_HW); |
328 | |
329 | printf("%s (%s) at ", name, ip); |
330 | |
331 | if (!(arp_flags & ATF_COM)) { |
332 | if (arp_flags & ATF_PUBL) |
333 | printf("* "); |
334 | else |
335 | printf("<incomplete> "); |
336 | } else { |
337 | printf("%s [%s] ", hwa, xhw->name); |
338 | } |
339 | |
340 | if (arp_flags & ATF_NETMASK) |
341 | printf("netmask %s ", mask); |
342 | |
343 | if (arp_flags & ATF_PERM) |
344 | printf("PERM "); |
345 | if (arp_flags & ATF_PUBL) |
346 | printf("PUP "); |
347 | #ifdef HAVE_ATF_MAGIC |
348 | if (arp_flags & ATF_MAGIC) |
349 | printf("AUTO "); |
350 | #endif |
351 | #ifdef HAVE_ATF_DONTPUB |
352 | if (arp_flags & ATF_DONTPUB) |
353 | printf("DONTPUB "); |
354 | #endif |
355 | if (arp_flags & ATF_USETRAILERS) |
356 | printf("TRAIL "); |
357 | |
358 | printf("on %s\n", dev); |
359 | } |
360 | |
361 | /* Display the contents of the ARP cache in the kernel. */ |
362 | /* Called only from main, once */ |
363 | static int arp_show(char *name) |
364 | { |
365 | char host[100]; |
366 | struct sockaddr sa; |
367 | char ip[100]; |
368 | char hwa[100]; |
369 | char mask[100]; |
370 | char line[200]; |
371 | char dev[100]; |
372 | int type, flags; |
373 | FILE *fp; |
374 | char *hostname; |
375 | int num; |
376 | unsigned entries = 0, shown = 0; |
377 | |
378 | host[0] = '\0'; |
379 | |
380 | if (name != NULL) { |
381 | /* Resolve the host name. */ |
382 | safe_strncpy(host, name, (sizeof host)); |
383 | if (ap->input(0, host, &sa) < 0) { |
384 | bb_herror_msg_and_die("%s", host); |
385 | } |
386 | safe_strncpy(host, ap->sprint(&sa, 1), sizeof(host)); |
387 | } |
388 | /* Open the PROCps kernel table. */ |
389 | fp = xfopen(_PATH_PROCNET_ARP, "r"); |
390 | /* Bypass header -- read until newline */ |
391 | if (fgets(line, sizeof(line), fp) != (char *) NULL) { |
392 | mask[0] = '-'; mask[1] = '\0'; |
393 | dev[0] = '-'; dev[1] = '\0'; |
394 | /* Read the ARP cache entries. */ |
395 | for (; fgets(line, sizeof(line), fp);) { |
396 | num = sscanf(line, "%s 0x%x 0x%x %100s %100s %100s\n", |
397 | ip, &type, &flags, hwa, mask, dev); |
398 | if (num < 4) |
399 | break; |
400 | |
401 | entries++; |
402 | /* if the user specified hw-type differs, skip it */ |
403 | if (hw_set && (type != hw->type)) |
404 | continue; |
405 | |
406 | /* if the user specified address differs, skip it */ |
407 | if (host[0] && strcmp(ip, host) != 0) |
408 | continue; |
409 | |
410 | /* if the user specified device differs, skip it */ |
411 | if (device[0] && strcmp(dev, device) != 0) |
412 | continue; |
413 | |
414 | shown++; |
415 | /* This IS ugly but it works -be */ |
416 | if (option_mask32 & ARP_OPT_n) |
417 | hostname = "?"; |
418 | else { |
419 | if (ap->input(0, ip, &sa) < 0) |
420 | hostname = ip; |
421 | else |
422 | hostname = ap->sprint(&sa, (option_mask32 & ARP_OPT_n) | 0x8000); |
423 | if (strcmp(hostname, ip) == 0) |
424 | hostname = "?"; |
425 | } |
426 | |
427 | arp_disp(hostname, ip, type, flags, hwa, mask, dev); |
428 | } |
429 | } |
430 | if (option_mask32 & ARP_OPT_v) |
431 | printf("Entries: %d\tSkipped: %d\tFound: %d\n", |
432 | entries, entries - shown, shown); |
433 | |
434 | if (!shown) { |
435 | if (hw_set || host[0] || device[0]) |
436 | printf("No match found in %d entries\n", entries); |
437 | } |
438 | |
439 | fclose(fp); |
440 | |
441 | return 0; |
442 | } |
443 | |
444 | int arp_main(int argc, char **argv) |
445 | { |
446 | char *hw_type; |
447 | char *protocol; |
448 | |
449 | /* Initialize variables... */ |
450 | ap = get_aftype(DFLT_AF); |
451 | if (!ap) |
452 | bb_error_msg_and_die("%s: %s not supported", DFLT_AF, "address family"); |
453 | |
454 | getopt32(argc, argv, "A:p:H:t:i:adnDsv", &protocol, &protocol, |
455 | &hw_type, &hw_type, &device); |
456 | argv += optind; |
457 | if (option_mask32 & ARP_OPT_A || option_mask32 & ARP_OPT_p) { |
458 | ap = get_aftype(protocol); |
459 | if (ap == NULL) |
460 | bb_error_msg_and_die("%s: unknown %s", protocol, "address family"); |
461 | } |
462 | if (option_mask32 & ARP_OPT_A || option_mask32 & ARP_OPT_p) { |
463 | hw = get_hwtype(hw_type); |
464 | if (hw == NULL) |
465 | bb_error_msg_and_die("%s: unknown %s", hw_type, "hardware type"); |
466 | hw_set = 1; |
467 | } |
468 | //if (option_mask32 & ARP_OPT_i)... -i |
469 | |
470 | if (ap->af != AF_INET) { |
471 | bb_error_msg_and_die("%s: kernel only supports 'inet'", ap->name); |
472 | } |
473 | |
474 | /* If no hw type specified get default */ |
475 | if (!hw) { |
476 | hw = get_hwtype(DFLT_HW); |
477 | if (!hw) |
478 | bb_error_msg_and_die("%s: %s not supported", DFLT_HW, "hardware type"); |
479 | } |
480 | |
481 | if (hw->alen <= 0) { |
482 | bb_error_msg_and_die("%s: %s without ARP support", |
483 | hw->name, "hardware type"); |
484 | } |
485 | sockfd = xsocket(AF_INET, SOCK_DGRAM, 0); |
486 | |
487 | /* Now see what we have to do here... */ |
488 | if (option_mask32 & (ARP_OPT_d|ARP_OPT_s)) { |
489 | if (argv[0] == NULL) |
490 | bb_error_msg_and_die("need host name"); |
491 | if (option_mask32 & ARP_OPT_s) |
492 | return arp_set(argv); |
493 | return arp_del(argv); |
494 | } |
495 | //if (option_mask32 & ARP_OPT_a) - default |
496 | return arp_show(argv[0]); |
497 | } |