bootptest.c revision 1.6 1 1.6 mrg /* $NetBSD: bootptest.c,v 1.6 1998/07/06 07:02:17 mrg Exp $ */
2 1.4 perry
3 1.1 gwr /*
4 1.1 gwr * bootptest.c - Test out a bootp server.
5 1.1 gwr *
6 1.1 gwr * This simple program was put together from pieces taken from
7 1.1 gwr * various places, including the CMU BOOTP client and server.
8 1.1 gwr * The packet printing routine is from the Berkeley "tcpdump"
9 1.1 gwr * program with some enhancements I added. The print-bootp.c
10 1.2 gwr * file was shared with my copy of "tcpdump" and therefore uses
11 1.1 gwr * some unusual utility routines that would normally be provided
12 1.2 gwr * by various parts of the tcpdump program. Gordon W. Ross
13 1.1 gwr *
14 1.1 gwr * Boilerplate:
15 1.1 gwr *
16 1.1 gwr * This program includes software developed by the University of
17 1.1 gwr * California, Lawrence Berkeley Laboratory and its contributors.
18 1.1 gwr * (See the copyright notice in print-bootp.c)
19 1.1 gwr *
20 1.1 gwr * The remainder of this program is public domain. You may do
21 1.1 gwr * whatever you like with it except claim that you wrote it.
22 1.1 gwr *
23 1.1 gwr * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
24 1.1 gwr * WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
25 1.1 gwr * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
26 1.1 gwr *
27 1.1 gwr * HISTORY:
28 1.1 gwr *
29 1.1 gwr * 12/02/93 Released version 1.4 (with bootp-2.3.2)
30 1.1 gwr * 11/05/93 Released version 1.3
31 1.1 gwr * 10/14/93 Released version 1.2
32 1.1 gwr * 10/11/93 Released version 1.1
33 1.1 gwr * 09/28/93 Released version 1.0
34 1.1 gwr * 09/93 Original developed by Gordon W. Ross <gwr (at) mc.com>
35 1.1 gwr */
36 1.1 gwr
37 1.5 lukem #include <sys/cdefs.h>
38 1.5 lukem #ifndef lint
39 1.6 mrg __RCSID("$NetBSD: bootptest.c,v 1.6 1998/07/06 07:02:17 mrg Exp $");
40 1.5 lukem #endif
41 1.5 lukem
42 1.1 gwr char *usage = "bootptest [-h] server-name [vendor-data-template-file]";
43 1.1 gwr
44 1.1 gwr #include <sys/types.h>
45 1.1 gwr #include <sys/socket.h>
46 1.1 gwr #include <sys/ioctl.h>
47 1.1 gwr #include <sys/file.h>
48 1.1 gwr #include <sys/time.h>
49 1.1 gwr #include <sys/stat.h>
50 1.1 gwr
51 1.1 gwr #include <net/if.h>
52 1.1 gwr #include <netinet/in.h>
53 1.1 gwr #include <arpa/inet.h> /* inet_ntoa */
54 1.1 gwr
55 1.1 gwr #include <stdlib.h>
56 1.1 gwr #include <signal.h>
57 1.1 gwr #include <stdio.h>
58 1.1 gwr #include <string.h>
59 1.1 gwr #include <errno.h>
60 1.1 gwr #include <ctype.h>
61 1.1 gwr #include <netdb.h>
62 1.1 gwr #include <assert.h>
63 1.5 lukem #include <unistd.h>
64 1.1 gwr
65 1.1 gwr #include "bootp.h"
66 1.1 gwr #include "bootptest.h"
67 1.1 gwr #include "getif.h"
68 1.5 lukem #include "report.h"
69 1.1 gwr #include "patchlevel.h"
70 1.1 gwr
71 1.1 gwr #define LOG_ERR 1
72 1.1 gwr #define BUFLEN 1024
73 1.1 gwr #define WAITSECS 1
74 1.1 gwr #define MAXWAIT 10
75 1.1 gwr
76 1.1 gwr int vflag = 1;
77 1.1 gwr int tflag = 0;
78 1.1 gwr int thiszone;
79 1.1 gwr char *progname;
80 1.1 gwr unsigned char *packetp;
81 1.1 gwr unsigned char *snapend;
82 1.1 gwr int snaplen;
83 1.1 gwr
84 1.1 gwr
85 1.1 gwr /*
86 1.1 gwr * IP port numbers for client and server obtained from /etc/services
87 1.1 gwr */
88 1.1 gwr
89 1.1 gwr u_short bootps_port, bootpc_port;
90 1.1 gwr
91 1.1 gwr
92 1.1 gwr /*
93 1.1 gwr * Internet socket and interface config structures
94 1.1 gwr */
95 1.1 gwr
96 1.1 gwr struct sockaddr_in sin_server; /* where to send requests */
97 1.1 gwr struct sockaddr_in sin_client; /* for bind and listen */
98 1.1 gwr struct sockaddr_in sin_from; /* Packet source */
99 1.1 gwr u_char eaddr[16]; /* Ethernet address */
100 1.1 gwr
101 1.1 gwr /*
102 1.1 gwr * General
103 1.1 gwr */
104 1.1 gwr
105 1.1 gwr int debug = 1; /* Debugging flag (level) */
106 1.6 mrg char hostname[MAXHOSTNAMELEN + 1];
107 1.1 gwr char *sndbuf; /* Send packet buffer */
108 1.1 gwr char *rcvbuf; /* Receive packet buffer */
109 1.1 gwr
110 1.1 gwr /*
111 1.1 gwr * Vendor magic cookies for CMU and RFC1048
112 1.1 gwr */
113 1.1 gwr
114 1.1 gwr unsigned char vm_cmu[4] = VM_CMU;
115 1.1 gwr unsigned char vm_rfc1048[4] = VM_RFC1048;
116 1.1 gwr short secs; /* How long client has waited */
117 1.1 gwr
118 1.5 lukem
119 1.5 lukem #ifdef __STDC__
120 1.5 lukem #define P(args) args
121 1.5 lukem #else
122 1.5 lukem #define P(args) ()
123 1.5 lukem #endif
124 1.5 lukem
125 1.5 lukem extern int getether P((char *, char *));
126 1.5 lukem int main P((int, char **));
127 1.5 lukem void send_request P((int));
128 1.5 lukem
129 1.5 lukem #undef P
130 1.1 gwr
131 1.1 gwr /*
132 1.1 gwr * Initialization such as command-line processing is done, then
133 1.1 gwr * the receiver loop is started. Die when interrupted.
134 1.1 gwr */
135 1.1 gwr
136 1.5 lukem int
137 1.1 gwr main(argc, argv)
138 1.1 gwr int argc;
139 1.1 gwr char **argv;
140 1.1 gwr {
141 1.1 gwr struct bootp *bp;
142 1.1 gwr struct servent *sep;
143 1.1 gwr struct hostent *hep;
144 1.1 gwr
145 1.1 gwr char *servername = NULL;
146 1.1 gwr char *vendor_file = NULL;
147 1.1 gwr char *bp_file = NULL;
148 1.3 mycroft int s; /* Socket file descriptor */
149 1.5 lukem int n, fromlen, recvcnt;
150 1.1 gwr int use_hwa = 0;
151 1.2 gwr int32 vend_magic;
152 1.1 gwr int32 xid;
153 1.1 gwr
154 1.1 gwr progname = strrchr(argv[0], '/');
155 1.1 gwr if (progname)
156 1.1 gwr progname++;
157 1.1 gwr else
158 1.1 gwr progname = argv[0];
159 1.1 gwr argc--;
160 1.1 gwr argv++;
161 1.1 gwr
162 1.1 gwr if (debug)
163 1.1 gwr printf("%s: version %s.%d\n", progname, VERSION, PATCHLEVEL);
164 1.1 gwr
165 1.2 gwr /*
166 1.2 gwr * Verify that "struct bootp" has the correct official size.
167 1.2 gwr * (Catch evil compilers that do struct padding.)
168 1.2 gwr */
169 1.1 gwr assert(sizeof(struct bootp) == BP_MINPKTSZ);
170 1.1 gwr
171 1.1 gwr sndbuf = malloc(BUFLEN);
172 1.1 gwr rcvbuf = malloc(BUFLEN);
173 1.1 gwr if (!sndbuf || !rcvbuf) {
174 1.1 gwr printf("malloc failed\n");
175 1.1 gwr exit(1);
176 1.1 gwr }
177 1.2 gwr
178 1.2 gwr /* default magic number */
179 1.2 gwr bcopy(vm_rfc1048, (char*)&vend_magic, 4);
180 1.2 gwr
181 1.1 gwr /* Handle option switches. */
182 1.1 gwr while (argc > 0) {
183 1.1 gwr if (argv[0][0] != '-')
184 1.1 gwr break;
185 1.1 gwr switch (argv[0][1]) {
186 1.1 gwr
187 1.1 gwr case 'f': /* File name to reqest. */
188 1.1 gwr if (argc < 2)
189 1.1 gwr goto error;
190 1.1 gwr argc--; argv++;
191 1.1 gwr bp_file = *argv;
192 1.1 gwr break;
193 1.1 gwr
194 1.1 gwr case 'h': /* Use hardware address. */
195 1.1 gwr use_hwa = 1;
196 1.1 gwr break;
197 1.1 gwr
198 1.2 gwr case 'm': /* Magic number value. */
199 1.2 gwr if (argc < 2)
200 1.2 gwr goto error;
201 1.2 gwr argc--; argv++;
202 1.2 gwr vend_magic = inet_addr(*argv);
203 1.2 gwr break;
204 1.2 gwr
205 1.1 gwr error:
206 1.1 gwr default:
207 1.1 gwr puts(usage);
208 1.1 gwr exit(1);
209 1.1 gwr
210 1.1 gwr }
211 1.1 gwr argc--;
212 1.1 gwr argv++;
213 1.1 gwr }
214 1.1 gwr
215 1.1 gwr /* Get server name (or address) for query. */
216 1.1 gwr if (argc > 0) {
217 1.1 gwr servername = *argv;
218 1.1 gwr argc--;
219 1.1 gwr argv++;
220 1.1 gwr }
221 1.1 gwr /* Get optional vendor-data-template-file. */
222 1.1 gwr if (argc > 0) {
223 1.1 gwr vendor_file = *argv;
224 1.1 gwr argc--;
225 1.1 gwr argv++;
226 1.1 gwr }
227 1.1 gwr if (!servername) {
228 1.1 gwr printf("missing server name.\n");
229 1.1 gwr puts(usage);
230 1.1 gwr exit(1);
231 1.1 gwr }
232 1.1 gwr /*
233 1.1 gwr * Create a socket.
234 1.1 gwr */
235 1.1 gwr if ((s = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
236 1.1 gwr perror("socket");
237 1.1 gwr exit(1);
238 1.1 gwr }
239 1.1 gwr /*
240 1.1 gwr * Get server's listening port number
241 1.1 gwr */
242 1.1 gwr sep = getservbyname("bootps", "udp");
243 1.1 gwr if (sep) {
244 1.1 gwr bootps_port = ntohs((u_short) sep->s_port);
245 1.1 gwr } else {
246 1.1 gwr fprintf(stderr, "udp/bootps: unknown service -- using port %d\n",
247 1.1 gwr IPPORT_BOOTPS);
248 1.1 gwr bootps_port = (u_short) IPPORT_BOOTPS;
249 1.1 gwr }
250 1.1 gwr
251 1.1 gwr /*
252 1.1 gwr * Set up server socket address (for send)
253 1.1 gwr */
254 1.1 gwr if (servername) {
255 1.3 mycroft if (inet_aton(servername, &sin_server.sin_addr) == 0) {
256 1.1 gwr hep = gethostbyname(servername);
257 1.1 gwr if (!hep) {
258 1.1 gwr fprintf(stderr, "%s: unknown host\n", servername);
259 1.1 gwr exit(1);
260 1.1 gwr }
261 1.3 mycroft memcpy(&sin_server.sin_addr, hep->h_addr,
262 1.3 mycroft sizeof(sin_server.sin_addr));
263 1.1 gwr }
264 1.1 gwr } else {
265 1.1 gwr /* Get broadcast address */
266 1.1 gwr /* XXX - not yet */
267 1.3 mycroft sin_server.sin_addr.s_addr = INADDR_ANY;
268 1.1 gwr }
269 1.1 gwr sin_server.sin_family = AF_INET;
270 1.1 gwr sin_server.sin_port = htons(bootps_port);
271 1.1 gwr
272 1.1 gwr /*
273 1.1 gwr * Get client's listening port number
274 1.1 gwr */
275 1.1 gwr sep = getservbyname("bootpc", "udp");
276 1.1 gwr if (sep) {
277 1.1 gwr bootpc_port = ntohs(sep->s_port);
278 1.1 gwr } else {
279 1.1 gwr fprintf(stderr, "udp/bootpc: unknown service -- using port %d\n",
280 1.1 gwr IPPORT_BOOTPC);
281 1.1 gwr bootpc_port = (u_short) IPPORT_BOOTPC;
282 1.1 gwr }
283 1.1 gwr
284 1.1 gwr /*
285 1.1 gwr * Set up client socket address (for listen)
286 1.1 gwr */
287 1.1 gwr sin_client.sin_family = AF_INET;
288 1.1 gwr sin_client.sin_port = htons(bootpc_port);
289 1.1 gwr sin_client.sin_addr.s_addr = INADDR_ANY;
290 1.1 gwr
291 1.1 gwr /*
292 1.1 gwr * Bind client socket to BOOTPC port.
293 1.1 gwr */
294 1.1 gwr if (bind(s, (struct sockaddr *) &sin_client, sizeof(sin_client)) < 0) {
295 1.1 gwr perror("bind BOOTPC port");
296 1.1 gwr if (errno == EACCES)
297 1.1 gwr fprintf(stderr, "You need to run this as root\n");
298 1.1 gwr exit(1);
299 1.1 gwr }
300 1.1 gwr /*
301 1.1 gwr * Build a request.
302 1.1 gwr */
303 1.1 gwr bp = (struct bootp *) sndbuf;
304 1.1 gwr bzero(bp, sizeof(*bp));
305 1.1 gwr bp->bp_op = BOOTREQUEST;
306 1.1 gwr xid = (int32) getpid();
307 1.1 gwr bp->bp_xid = (u_int32) htonl(xid);
308 1.1 gwr if (bp_file)
309 1.1 gwr strncpy(bp->bp_file, bp_file, BP_FILE_LEN);
310 1.1 gwr
311 1.1 gwr /*
312 1.1 gwr * Fill in the hardware address (or client IP address)
313 1.1 gwr */
314 1.1 gwr if (use_hwa) {
315 1.1 gwr struct ifreq *ifr;
316 1.1 gwr
317 1.1 gwr ifr = getif(s, &sin_server.sin_addr);
318 1.1 gwr if (!ifr) {
319 1.1 gwr printf("No interface for %s\n", servername);
320 1.1 gwr exit(1);
321 1.1 gwr }
322 1.1 gwr if (getether(ifr->ifr_name, eaddr)) {
323 1.1 gwr printf("Can not get ether addr for %s\n", ifr->ifr_name);
324 1.1 gwr exit(1);
325 1.1 gwr }
326 1.1 gwr /* Copy Ethernet address into request packet. */
327 1.1 gwr bp->bp_htype = 1;
328 1.1 gwr bp->bp_hlen = 6;
329 1.1 gwr bcopy(eaddr, bp->bp_chaddr, bp->bp_hlen);
330 1.1 gwr } else {
331 1.1 gwr /* Fill in the client IP address. */
332 1.1 gwr gethostname(hostname, sizeof(hostname));
333 1.6 mrg hostname[sizeof(hostname) - 1] = '\0';
334 1.1 gwr hep = gethostbyname(hostname);
335 1.1 gwr if (!hep) {
336 1.1 gwr printf("Can not get my IP address\n");
337 1.1 gwr exit(1);
338 1.1 gwr }
339 1.1 gwr bcopy(hep->h_addr, &bp->bp_ciaddr, hep->h_length);
340 1.1 gwr }
341 1.1 gwr
342 1.1 gwr /*
343 1.1 gwr * Copy in the default vendor data.
344 1.1 gwr */
345 1.2 gwr bcopy((char*)&vend_magic, bp->bp_vend, 4);
346 1.2 gwr if (vend_magic)
347 1.2 gwr bp->bp_vend[4] = TAG_END;
348 1.1 gwr
349 1.1 gwr /*
350 1.1 gwr * Read in the "options" part of the request.
351 1.1 gwr * This also determines the size of the packet.
352 1.1 gwr */
353 1.1 gwr snaplen = sizeof(*bp);
354 1.1 gwr if (vendor_file) {
355 1.1 gwr int fd = open(vendor_file, 0);
356 1.1 gwr if (fd < 0) {
357 1.1 gwr perror(vendor_file);
358 1.1 gwr exit(1);
359 1.1 gwr }
360 1.1 gwr /* Compute actual space for options. */
361 1.1 gwr n = BUFLEN - sizeof(*bp) + BP_VEND_LEN;
362 1.1 gwr n = read(fd, bp->bp_vend, n);
363 1.1 gwr close(fd);
364 1.1 gwr if (n < 0) {
365 1.1 gwr perror(vendor_file);
366 1.1 gwr exit(1);
367 1.1 gwr }
368 1.1 gwr printf("read %d bytes of vendor template\n", n);
369 1.1 gwr if (n > BP_VEND_LEN) {
370 1.1 gwr printf("warning: extended options in use (len > %d)\n",
371 1.1 gwr BP_VEND_LEN);
372 1.1 gwr snaplen += (n - BP_VEND_LEN);
373 1.1 gwr }
374 1.1 gwr }
375 1.1 gwr /*
376 1.1 gwr * Set globals needed by print_bootp
377 1.1 gwr * (called by send_request)
378 1.1 gwr */
379 1.1 gwr packetp = (unsigned char *) eaddr;
380 1.1 gwr snapend = (unsigned char *) sndbuf + snaplen;
381 1.1 gwr
382 1.1 gwr /* Send a request once per second while waiting for replies. */
383 1.1 gwr recvcnt = 0;
384 1.1 gwr bp->bp_secs = secs = 0;
385 1.1 gwr send_request(s);
386 1.1 gwr while (1) {
387 1.1 gwr struct timeval tv;
388 1.1 gwr int readfds;
389 1.1 gwr
390 1.1 gwr tv.tv_sec = WAITSECS;
391 1.1 gwr tv.tv_usec = 0L;
392 1.1 gwr readfds = (1 << s);
393 1.1 gwr n = select(s + 1, (fd_set *) & readfds, NULL, NULL, &tv);
394 1.1 gwr if (n < 0) {
395 1.1 gwr perror("select");
396 1.1 gwr break;
397 1.1 gwr }
398 1.1 gwr if (n == 0) {
399 1.1 gwr /*
400 1.2 gwr * We have not received a response in the last second.
401 1.1 gwr * If we have ever received any responses, exit now.
402 1.1 gwr * Otherwise, bump the "wait time" field and re-send.
403 1.1 gwr */
404 1.1 gwr if (recvcnt > 0)
405 1.1 gwr exit(0);
406 1.1 gwr secs += WAITSECS;
407 1.1 gwr if (secs > MAXWAIT)
408 1.1 gwr break;
409 1.1 gwr bp->bp_secs = htons(secs);
410 1.1 gwr send_request(s);
411 1.1 gwr continue;
412 1.1 gwr }
413 1.1 gwr fromlen = sizeof(sin_from);
414 1.1 gwr n = recvfrom(s, rcvbuf, BUFLEN, 0,
415 1.1 gwr (struct sockaddr *) &sin_from, &fromlen);
416 1.1 gwr if (n <= 0) {
417 1.1 gwr continue;
418 1.1 gwr }
419 1.1 gwr if (n < sizeof(struct bootp)) {
420 1.1 gwr printf("received short packet\n");
421 1.1 gwr continue;
422 1.1 gwr }
423 1.1 gwr recvcnt++;
424 1.1 gwr
425 1.1 gwr /* Print the received packet. */
426 1.1 gwr printf("Recvd from %s", inet_ntoa(sin_from.sin_addr));
427 1.1 gwr /* set globals needed by bootp_print() */
428 1.1 gwr snaplen = n;
429 1.1 gwr snapend = (unsigned char *) rcvbuf + snaplen;
430 1.5 lukem bootp_print((struct bootp *)rcvbuf, n, sin_from.sin_port, 0);
431 1.1 gwr putchar('\n');
432 1.1 gwr /*
433 1.1 gwr * This no longer exits immediately after receiving
434 1.1 gwr * one response because it is useful to know if the
435 1.1 gwr * client might get multiple responses. This code
436 1.1 gwr * will now listen for one second after a response.
437 1.1 gwr */
438 1.1 gwr }
439 1.1 gwr fprintf(stderr, "no response from %s\n", servername);
440 1.1 gwr exit(1);
441 1.1 gwr }
442 1.1 gwr
443 1.5 lukem void
444 1.1 gwr send_request(s)
445 1.1 gwr int s;
446 1.1 gwr {
447 1.1 gwr /* Print the request packet. */
448 1.1 gwr printf("Sending to %s", inet_ntoa(sin_server.sin_addr));
449 1.5 lukem bootp_print((struct bootp *)sndbuf, snaplen, sin_from.sin_port, 0);
450 1.1 gwr putchar('\n');
451 1.1 gwr
452 1.1 gwr /* Send the request packet. */
453 1.1 gwr if (sendto(s, sndbuf, snaplen, 0,
454 1.1 gwr (struct sockaddr *) &sin_server,
455 1.1 gwr sizeof(sin_server)) < 0)
456 1.1 gwr {
457 1.1 gwr perror("sendto server");
458 1.1 gwr exit(1);
459 1.1 gwr }
460 1.1 gwr }
461 1.1 gwr
462 1.1 gwr /*
463 1.1 gwr * Print out a filename (or other ascii string).
464 1.1 gwr * Return true if truncated.
465 1.1 gwr */
466 1.1 gwr int
467 1.1 gwr printfn(s, ep)
468 1.1 gwr register u_char *s, *ep;
469 1.1 gwr {
470 1.1 gwr register u_char c;
471 1.1 gwr
472 1.1 gwr putchar('"');
473 1.5 lukem while ((c = *s++) != 0) {
474 1.1 gwr if (s > ep) {
475 1.1 gwr putchar('"');
476 1.1 gwr return (1);
477 1.1 gwr }
478 1.1 gwr if (!isascii(c)) {
479 1.1 gwr c = toascii(c);
480 1.1 gwr putchar('M');
481 1.1 gwr putchar('-');
482 1.1 gwr }
483 1.1 gwr if (!isprint(c)) {
484 1.1 gwr c ^= 0x40; /* DEL to ?, others to alpha */
485 1.1 gwr putchar('^');
486 1.1 gwr }
487 1.1 gwr putchar(c);
488 1.1 gwr }
489 1.1 gwr putchar('"');
490 1.1 gwr return (0);
491 1.1 gwr }
492 1.1 gwr
493 1.1 gwr /*
494 1.1 gwr * Convert an IP addr to a string.
495 1.1 gwr * (like inet_ntoa, but ina is a pointer)
496 1.1 gwr */
497 1.1 gwr char *
498 1.1 gwr ipaddr_string(ina)
499 1.1 gwr struct in_addr *ina;
500 1.1 gwr {
501 1.1 gwr static char b[24];
502 1.1 gwr u_char *p;
503 1.1 gwr
504 1.1 gwr p = (u_char *) ina;
505 1.1 gwr sprintf(b, "%d.%d.%d.%d", p[0], p[1], p[2], p[3]);
506 1.1 gwr return (b);
507 1.1 gwr }
508 1.1 gwr
509 1.1 gwr /*
510 1.1 gwr * Local Variables:
511 1.1 gwr * tab-width: 4
512 1.1 gwr * c-indent-level: 4
513 1.1 gwr * c-argdecl-indent: 4
514 1.1 gwr * c-continued-statement-offset: 4
515 1.1 gwr * c-continued-brace-offset: -4
516 1.1 gwr * c-label-offset: -4
517 1.1 gwr * c-brace-offset: 0
518 1.1 gwr * End:
519 1.1 gwr */
520