ndbootd.c revision 1.6 1 1.6 scw /* $NetBSD: ndbootd.c,v 1.6 2002/07/11 19:47:17 scw Exp $ */
2 1.2 fredette
3 1.1 fredette /* ndbootd.c - the Sun Network Disk (nd) daemon: */
4 1.1 fredette
5 1.1 fredette /*
6 1.1 fredette * Copyright (c) 2001 Matthew Fredette. All rights reserved.
7 1.1 fredette *
8 1.1 fredette * Redistribution and use in source and binary forms, with or without
9 1.1 fredette * modification, are permitted provided that the following conditions
10 1.1 fredette * are met:
11 1.1 fredette * 1. Redistributions of source code must retain the above copyright
12 1.1 fredette * notice, this list of conditions and the following disclaimer.
13 1.1 fredette * 2. Redistributions in binary form must reproduce the above copyright
14 1.1 fredette * notice, this list of conditions and the following disclaimer in the
15 1.1 fredette * documentation and/or other materials provided with the distribution.
16 1.1 fredette * 3. All advertising materials mentioning features or use of this software
17 1.1 fredette * must display the following acknowledgement:
18 1.1 fredette * This product includes software developed by Matthew Fredette.
19 1.1 fredette * 4. The name of Matthew Fredette may not be used to endorse or promote
20 1.1 fredette * products derived from this software without specific prior written
21 1.1 fredette * permission.
22 1.1 fredette *
23 1.1 fredette * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
24 1.1 fredette * WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
25 1.1 fredette * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
26 1.1 fredette */
27 1.1 fredette
28 1.4 fredette /* <<Header: /data/home/fredette/project/THE-WEIGHT-CVS/ndbootd/ndbootd.c,v 1.9 2001/06/13 21:19:11 fredette Exp >> */
29 1.1 fredette
30 1.1 fredette /*
31 1.1 fredette * <<Log: ndbootd.c,v >>
32 1.4 fredette * Revision 1.9 2001/06/13 21:19:11 fredette
33 1.4 fredette * (main): Don't assume that a successful, but short, read
34 1.4 fredette * leaves a zero in errno. Instead, just check for the short
35 1.4 fredette * read by looking at the byte count that read returned.
36 1.4 fredette *
37 1.3 fredette * Revision 1.8 2001/05/23 02:35:36 fredette
38 1.3 fredette * Changed many debugging printfs to compile quietly on the
39 1.3 fredette * alpha. Patch from Andrew Brown <atatat (at) atatdot.net>.
40 1.3 fredette *
41 1.1 fredette * Revision 1.7 2001/05/22 13:13:20 fredette
42 1.1 fredette * Ran indent(1) with NetBSD's KNF-approximating profile.
43 1.1 fredette *
44 1.1 fredette * Revision 1.6 2001/05/22 12:53:40 fredette
45 1.1 fredette * [HAVE_STRICT_ALIGNMENT]: Added code to copy packet headers
46 1.1 fredette * between the buffer and local variables, to satisfy
47 1.1 fredette * alignment constraints.
48 1.1 fredette *
49 1.1 fredette * Revision 1.5 2001/05/15 14:43:24 fredette
50 1.1 fredette * Now have prototypes for the allocation functions.
51 1.1 fredette * (main): Now handle boot blocks that aren't an integral
52 1.1 fredette * multiple of the block size.
53 1.1 fredette *
54 1.1 fredette * Revision 1.4 2001/05/09 20:53:38 fredette
55 1.1 fredette * (main): Now insert a small delay before sending each packet.
56 1.1 fredette * Sending packets too quickly apparently overwhelms clients.
57 1.1 fredette * Added new single-letter versions of all options that didn't
58 1.1 fredette * already have them. Expanded some debug messages, and fixed
59 1.1 fredette * others to display Ethernet addresses correctly.
60 1.1 fredette *
61 1.1 fredette * Revision 1.3 2001/01/31 17:35:50 fredette
62 1.1 fredette * (main): Fixed various printf argument lists.
63 1.1 fredette *
64 1.1 fredette * Revision 1.2 2001/01/30 15:35:38 fredette
65 1.1 fredette * Now, ndbootd assembles disk images for clients on-the-fly.
66 1.1 fredette * Defined many new macros related to this.
67 1.1 fredette * (main): Added support for the --boot2 option. Turned the
68 1.1 fredette * original disk-image filename into the filename of the
69 1.1 fredette * first-stage boot program. Now do better multiple-client
70 1.1 fredette * support, especially when it comes to checking if a client
71 1.1 fredette * is really ours. Now assemble client-specific disk images
72 1.1 fredette * on-the-fly, potentially serving each client a different
73 1.1 fredette * second-stage boot.
74 1.1 fredette *
75 1.1 fredette * Revision 1.1 2001/01/29 15:12:13 fredette
76 1.1 fredette * Added.
77 1.1 fredette *
78 1.1 fredette */
79 1.1 fredette
80 1.5 thorpej #include <sys/cdefs.h>
81 1.5 thorpej #if 0
82 1.4 fredette static const char _ndbootd_c_rcsid[] = "<<Id: ndbootd.c,v 1.9 2001/06/13 21:19:11 fredette Exp >>";
83 1.5 thorpej #else
84 1.6 scw __RCSID("$NetBSD: ndbootd.c,v 1.6 2002/07/11 19:47:17 scw Exp $");
85 1.5 thorpej #endif
86 1.1 fredette
87 1.1 fredette /* includes: */
88 1.1 fredette #include "ndbootd.h"
89 1.1 fredette
90 1.1 fredette /* the number of blocks that Sun-2 PROMs load, starting from block
91 1.1 fredette zero: */
92 1.1 fredette #define NDBOOTD_PROM_BLOCK_COUNT (16)
93 1.1 fredette
94 1.1 fredette /* the first block number of the (dummy) Sun disklabel: */
95 1.1 fredette #define NDBOOTD_SUNDK_BLOCK_FIRST (0)
96 1.1 fredette
97 1.1 fredette /* the number of blocks in the (dummy) Sun disklabel: */
98 1.1 fredette #define NDBOOTD_SUNDK_BLOCK_COUNT (1)
99 1.1 fredette
100 1.1 fredette /* the first block number of the first-stage boot program.
101 1.1 fredette the first-stage boot program begins right after the (dummy)
102 1.1 fredette Sun disklabel: */
103 1.1 fredette #define NDBOOTD_BOOT1_BLOCK_FIRST (NDBOOTD_SUNDK_BLOCK_FIRST + NDBOOTD_SUNDK_BLOCK_COUNT)
104 1.1 fredette
105 1.1 fredette /* the number of blocks in the first-stage boot program: */
106 1.1 fredette #define NDBOOTD_BOOT1_BLOCK_COUNT (NDBOOTD_PROM_BLOCK_COUNT - NDBOOTD_BOOT1_BLOCK_FIRST)
107 1.1 fredette
108 1.1 fredette /* the first block number of any second-stage boot program.
109 1.1 fredette any second-stage boot program begins right after the first-stage boot program: */
110 1.1 fredette #define NDBOOTD_BOOT2_BLOCK_FIRST (NDBOOTD_BOOT1_BLOCK_FIRST + NDBOOTD_BOOT1_BLOCK_COUNT)
111 1.1 fredette
112 1.1 fredette /* this macro returns the number of bytes available in an object starting at a given offset: */
113 1.1 fredette #define NDBOOTD_BYTES_AVAIL(block_number, byte_offset, obj_block_first, obj_block_count) \
114 1.1 fredette ((((ssize_t) (obj_block_count) - (ssize_t) ((block_number) - (obj_block_first))) * NDBOOT_BSIZE) - (ssize_t) (byte_offset))
115 1.1 fredette
116 1.1 fredette /* this determines how long we can cache file descriptors and RARP
117 1.1 fredette information: */
118 1.1 fredette #define NDBOOTD_CLIENT_TTL_SECONDS (10)
119 1.1 fredette
120 1.1 fredette /* this determines how long we wait before sending a packet: */
121 1.1 fredette #define NDBOOTD_SEND_DELAY_USECONDS (10000)
122 1.1 fredette
123 1.1 fredette /* this macro helps us size a struct ifreq: */
124 1.1 fredette #ifdef HAVE_SOCKADDR_SA_LEN
125 1.1 fredette #define SIZEOF_IFREQ(ifr) (sizeof(ifr->ifr_name) + ifr->ifr_addr.sa_len)
126 1.1 fredette #else /* !HAVE_SOCKADDR_SA_LEN */
127 1.1 fredette #define SIZEOF_IFREQ(ifr) (sizeof(ifr->ifr_name) + sizeof(struct sockaddr))
128 1.1 fredette #endif /* !HAVE_SOCKADDR_SA_LEN */
129 1.1 fredette
130 1.1 fredette /* prototypes: */
131 1.1 fredette void *ndbootd_malloc _NDBOOTD_P((size_t));
132 1.1 fredette void *ndbootd_malloc0 _NDBOOTD_P((size_t));
133 1.1 fredette void *ndbootd_memdup _NDBOOTD_P((void *, size_t));
134 1.1 fredette
135 1.1 fredette /* globals: */
136 1.1 fredette const char *_ndbootd_argv0;
137 1.1 fredette #ifdef _NDBOOTD_DO_DEBUG
138 1.1 fredette int _ndbootd_debug;
139 1.1 fredette #endif /* _NDBOOTD_DO_DEBUG */
140 1.1 fredette
141 1.1 fredette /* allocators: */
142 1.1 fredette void *
143 1.1 fredette ndbootd_malloc(size_t size)
144 1.1 fredette {
145 1.1 fredette void *buffer;
146 1.1 fredette if ((buffer = malloc(size)) == NULL) {
147 1.1 fredette abort();
148 1.1 fredette }
149 1.1 fredette return (buffer);
150 1.1 fredette }
151 1.1 fredette void *
152 1.1 fredette ndbootd_malloc0(size_t size)
153 1.1 fredette {
154 1.1 fredette void *buffer;
155 1.1 fredette buffer = ndbootd_malloc(size);
156 1.1 fredette memset(buffer, 0, size);
157 1.1 fredette return (buffer);
158 1.1 fredette }
159 1.1 fredette void *
160 1.1 fredette ndbootd_memdup(void *buffer0, size_t size)
161 1.1 fredette {
162 1.1 fredette void *buffer1;
163 1.1 fredette buffer1 = ndbootd_malloc(size);
164 1.1 fredette memcpy(buffer1, buffer0, size);
165 1.1 fredette return (buffer1);
166 1.1 fredette }
167 1.1 fredette #define ndbootd_free free
168 1.1 fredette #define ndbootd_new(t, c) ((t *) ndbootd_malloc(sizeof(t) * (c)))
169 1.1 fredette #define ndbootd_new0(t, c) ((t *) ndbootd_malloc0(sizeof(t) * (c)))
170 1.1 fredette #define ndbootd_dup(t, b, c) ((t *) ndbootd_memdup(b, c))
171 1.1 fredette
172 1.1 fredette /* this calculates an IP packet header checksum: */
173 1.1 fredette static void
174 1.1 fredette _ndbootd_ip_cksum(struct ip * ip_packet)
175 1.1 fredette {
176 1.1 fredette u_int16_t *_word, word;
177 1.1 fredette u_int32_t checksum;
178 1.1 fredette unsigned int byte_count, bytes_left;
179 1.1 fredette
180 1.1 fredette /* we assume that the IP packet header is 16-bit aligned: */
181 1.1 fredette assert((((unsigned long) ip_packet) % sizeof(word)) == 0);
182 1.1 fredette
183 1.1 fredette /* initialize for the checksum: */
184 1.1 fredette checksum = 0;
185 1.1 fredette
186 1.1 fredette /* sum up the packet contents: */
187 1.1 fredette _word = (u_int16_t *) ip_packet;
188 1.1 fredette byte_count = ip_packet->ip_hl << 2;
189 1.1 fredette for (bytes_left = byte_count; bytes_left >= sizeof(*_word);) {
190 1.1 fredette checksum += *(_word++);
191 1.1 fredette bytes_left -= sizeof(*_word);
192 1.1 fredette }
193 1.1 fredette word = 0;
194 1.1 fredette memcpy(&word, _word, bytes_left);
195 1.1 fredette checksum += word;
196 1.1 fredette
197 1.1 fredette /* finish the checksum: */
198 1.1 fredette checksum = (checksum >> 16) + (checksum & 0xffff);
199 1.1 fredette checksum += (checksum >> 16);
200 1.6 scw checksum = ~checksum;
201 1.6 scw ip_packet->ip_sum = checksum;
202 1.1 fredette }
203 1.1 fredette /* this finds a network interface: */
204 1.1 fredette static struct ndbootd_interface *
205 1.1 fredette _ndbootd_find_interface(const char *ifr_name_user)
206 1.1 fredette {
207 1.1 fredette int saved_errno;
208 1.1 fredette int dummy_fd;
209 1.1 fredette char ifreq_buffer[16384]; /* FIXME - magic constant. */
210 1.1 fredette struct ifconf ifc;
211 1.1 fredette struct ifreq *ifr;
212 1.1 fredette struct ifreq *ifr_user;
213 1.1 fredette size_t ifr_offset;
214 1.1 fredette struct sockaddr_in saved_ip_address;
215 1.1 fredette short saved_flags;
216 1.1 fredette #ifdef HAVE_AF_LINK
217 1.1 fredette struct ifreq *link_ifreqs[20]; /* FIXME - magic constant. */
218 1.1 fredette size_t link_ifreqs_count;
219 1.1 fredette size_t link_ifreqs_i;
220 1.1 fredette struct sockaddr_dl *sadl;
221 1.1 fredette #endif /* HAVE_AF_LINK */
222 1.1 fredette struct ndbootd_interface *interface;
223 1.1 fredette
224 1.1 fredette /* make a dummy socket so we can read the interface list: */
225 1.1 fredette if ((dummy_fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
226 1.1 fredette return (NULL);
227 1.1 fredette }
228 1.1 fredette /* read the interface list: */
229 1.1 fredette ifc.ifc_len = sizeof(ifreq_buffer);
230 1.1 fredette ifc.ifc_buf = ifreq_buffer;
231 1.1 fredette if (ioctl(dummy_fd, SIOCGIFCONF, &ifc) < 0) {
232 1.1 fredette saved_errno = errno;
233 1.1 fredette close(dummy_fd);
234 1.1 fredette errno = saved_errno;
235 1.1 fredette return (NULL);
236 1.1 fredette }
237 1.1 fredette #ifdef HAVE_AF_LINK
238 1.1 fredette /* start our list of link address ifreqs: */
239 1.1 fredette link_ifreqs_count = 0;
240 1.1 fredette #endif /* HAVE_AF_LINK */
241 1.1 fredette
242 1.1 fredette /* walk the interface list: */
243 1.1 fredette ifr_user = NULL;
244 1.1 fredette for (ifr_offset = 0;; ifr_offset += SIZEOF_IFREQ(ifr)) {
245 1.1 fredette
246 1.1 fredette /* stop walking if we have run out of space in the buffer.
247 1.1 fredette * note that before we can use SIZEOF_IFREQ, we have to make
248 1.1 fredette * sure that there is a minimum number of bytes in the buffer
249 1.1 fredette * to use it (namely, that there's a whole struct sockaddr
250 1.1 fredette * available): */
251 1.1 fredette ifr = (struct ifreq *) (ifreq_buffer + ifr_offset);
252 1.1 fredette if ((ifr_offset + sizeof(ifr->ifr_name) + sizeof(struct sockaddr)) > ifc.ifc_len
253 1.1 fredette || (ifr_offset + SIZEOF_IFREQ(ifr)) > ifc.ifc_len) {
254 1.1 fredette errno = ENOENT;
255 1.1 fredette break;
256 1.1 fredette }
257 1.1 fredette #ifdef HAVE_AF_LINK
258 1.1 fredette /* if this is a hardware address, save it: */
259 1.1 fredette if (ifr->ifr_addr.sa_family == AF_LINK) {
260 1.1 fredette if (link_ifreqs_count < (sizeof(link_ifreqs) / sizeof(link_ifreqs[0]))) {
261 1.1 fredette link_ifreqs[link_ifreqs_count++] = ifr;
262 1.1 fredette }
263 1.1 fredette continue;
264 1.1 fredette }
265 1.1 fredette #endif /* HAVE_AF_LINK */
266 1.1 fredette
267 1.1 fredette /* ignore this interface if it doesn't do IP: */
268 1.1 fredette if (ifr->ifr_addr.sa_family != AF_INET) {
269 1.1 fredette continue;
270 1.1 fredette }
271 1.1 fredette /* get the interface flags, preserving the IP address in the
272 1.1 fredette * struct ifreq across the call: */
273 1.1 fredette saved_ip_address = *((struct sockaddr_in *) & ifr->ifr_addr);
274 1.1 fredette if (ioctl(dummy_fd, SIOCGIFFLAGS, ifr) < 0) {
275 1.1 fredette ifr = NULL;
276 1.1 fredette break;
277 1.1 fredette }
278 1.1 fredette saved_flags = ifr->ifr_flags;
279 1.1 fredette *((struct sockaddr_in *) & ifr->ifr_addr) = saved_ip_address;
280 1.1 fredette
281 1.1 fredette /* ignore this interface if it isn't up and running: */
282 1.1 fredette if ((saved_flags & (IFF_UP | IFF_RUNNING)) != (IFF_UP | IFF_RUNNING)) {
283 1.1 fredette continue;
284 1.1 fredette }
285 1.1 fredette /* if we don't have an interface yet, take this one depending
286 1.1 fredette * on whether the user asked for an interface by name or not.
287 1.1 fredette * if he did, and this is it, take this one. if he didn't,
288 1.1 fredette * and this isn't a loopback interface, take this one: */
289 1.1 fredette if (ifr_user == NULL
290 1.1 fredette && (ifr_name_user != NULL
291 1.1 fredette ? !strncmp(ifr->ifr_name, ifr_name_user, sizeof(ifr->ifr_name))
292 1.1 fredette : !(ifr->ifr_flags & IFF_LOOPBACK))) {
293 1.1 fredette ifr_user = ifr;
294 1.1 fredette }
295 1.1 fredette }
296 1.1 fredette
297 1.1 fredette /* close the dummy socket: */
298 1.1 fredette saved_errno = errno;
299 1.1 fredette close(dummy_fd);
300 1.1 fredette errno = saved_errno;
301 1.1 fredette
302 1.1 fredette /* if we don't have an interface to return: */
303 1.1 fredette if (ifr_user == NULL) {
304 1.1 fredette return (NULL);
305 1.1 fredette }
306 1.1 fredette /* start the interface description: */
307 1.1 fredette interface = ndbootd_new0(struct ndbootd_interface, 1);
308 1.1 fredette
309 1.1 fredette #ifdef HAVE_AF_LINK
310 1.1 fredette
311 1.1 fredette /* we must be able to find an AF_LINK ifreq that gives us the
312 1.1 fredette * interface's Ethernet address. */
313 1.1 fredette ifr = NULL;
314 1.1 fredette for (link_ifreqs_i = 0; link_ifreqs_i < link_ifreqs_count; link_ifreqs_i++) {
315 1.1 fredette if (!strncmp(link_ifreqs[link_ifreqs_i]->ifr_name,
316 1.1 fredette ifr_user->ifr_name,
317 1.1 fredette sizeof(ifr_user->ifr_name))) {
318 1.1 fredette ifr = link_ifreqs[link_ifreqs_i];
319 1.1 fredette break;
320 1.1 fredette }
321 1.1 fredette }
322 1.1 fredette if (ifr == NULL) {
323 1.1 fredette free(interface);
324 1.1 fredette return (NULL);
325 1.1 fredette }
326 1.1 fredette /* copy out the Ethernet address: */
327 1.1 fredette sadl = (struct sockaddr_dl *) & ifr->ifr_addr;
328 1.1 fredette memcpy(interface->ndbootd_interface_ether, LLADDR(sadl), sadl->sdl_alen);
329 1.1 fredette
330 1.1 fredette #else /* !HAVE_AF_LINK */
331 1.1 fredette #error "must have AF_LINK for now"
332 1.1 fredette #endif /* !HAVE_AF_LINK */
333 1.1 fredette
334 1.1 fredette /* finish this interface and return it: */
335 1.1 fredette interface->ndbootd_interface_ifreq = (struct ifreq *) ndbootd_memdup(ifr_user, SIZEOF_IFREQ(ifr_user));
336 1.1 fredette interface->ndbootd_interface_fd = -1;
337 1.1 fredette return (interface);
338 1.1 fredette }
339 1.1 fredette
340 1.1 fredette int
341 1.1 fredette main(int argc, char *argv[])
342 1.1 fredette {
343 1.1 fredette int argv_i;
344 1.1 fredette int show_usage;
345 1.1 fredette const char *interface_name;
346 1.1 fredette const char *boot1_file_name;
347 1.1 fredette const char *boot2_x_name;
348 1.1 fredette char *boot2_file_name;
349 1.1 fredette int boot2_x_name_is_dir;
350 1.1 fredette time_t last_open_time;
351 1.1 fredette int boot1_fd;
352 1.1 fredette int boot2_fd;
353 1.1 fredette time_t last_rarp_time;
354 1.1 fredette char last_client_ether[ETHER_ADDR_LEN];
355 1.1 fredette struct in_addr last_client_ip;
356 1.1 fredette struct stat stat_buffer;
357 1.1 fredette int32_t boot1_block_count;
358 1.1 fredette int32_t boot2_block_count;
359 1.1 fredette size_t boot1_byte_count;
360 1.1 fredette size_t boot2_byte_count;
361 1.1 fredette ssize_t byte_count_read;
362 1.1 fredette struct ndbootd_interface *interface;
363 1.1 fredette char pid_buffer[(sizeof(pid_t) * 3) + 2];
364 1.1 fredette unsigned char packet_buffer[sizeof(struct ether_header) + IP_MAXPACKET];
365 1.1 fredette unsigned char disk_buffer[NDBOOT_MAX_BYTE_COUNT];
366 1.1 fredette char hostname_buffer[MAXHOSTNAMELEN + 1];
367 1.1 fredette struct hostent *the_hostent;
368 1.1 fredette ssize_t packet_length;
369 1.1 fredette time_t now;
370 1.1 fredette struct ether_header *ether_packet;
371 1.1 fredette struct ip *ip_packet;
372 1.1 fredette struct ndboot_packet *nd_packet;
373 1.1 fredette #ifdef HAVE_STRICT_ALIGNMENT
374 1.1 fredette struct ether_header ether_packet_buffer;
375 1.1 fredette unsigned char ip_packet_buffer[IP_MAXPACKET];
376 1.1 fredette struct ndboot_packet nd_packet_buffer;
377 1.1 fredette #endif /* HAVE_STRICT_ALIGNMENT */
378 1.1 fredette int nd_window_size;
379 1.1 fredette int nd_window_filled;
380 1.1 fredette off_t file_offset;
381 1.1 fredette size_t disk_buffer_offset;
382 1.1 fredette size_t block_number;
383 1.1 fredette size_t byte_offset;
384 1.1 fredette ssize_t byte_count;
385 1.1 fredette ssize_t byte_count_wanted;
386 1.1 fredette struct timeval send_delay;
387 1.1 fredette int fd;
388 1.1 fredette
389 1.1 fredette /* check our command line: */
390 1.1 fredette if ((_ndbootd_argv0 = strrchr(argv[0], '/')) == NULL)
391 1.1 fredette _ndbootd_argv0 = argv[0];
392 1.1 fredette else
393 1.1 fredette _ndbootd_argv0++;
394 1.1 fredette show_usage = FALSE;
395 1.1 fredette #ifdef _NDBOOTD_DO_DEBUG
396 1.1 fredette _ndbootd_debug = FALSE;
397 1.1 fredette #endif /* _NDBOOTD_DO_DEBUG */
398 1.1 fredette boot1_file_name = NULL;
399 1.1 fredette boot2_x_name = NULL;
400 1.1 fredette interface_name = NULL;
401 1.1 fredette nd_window_size = NDBOOT_WINDOW_SIZE_DEFAULT;
402 1.1 fredette for (argv_i = 1; argv_i < argc; argv_i++) {
403 1.1 fredette if (argv[argv_i][0] != '-'
404 1.1 fredette || argv[argv_i][1] == '\0') {
405 1.1 fredette break;
406 1.1 fredette } else if (!strcmp(argv[argv_i], "-s")
407 1.1 fredette || !strcmp(argv[argv_i], "--boot2")) {
408 1.1 fredette if (++argv_i < argc) {
409 1.1 fredette boot2_x_name = argv[argv_i];
410 1.1 fredette } else {
411 1.1 fredette show_usage = TRUE;
412 1.1 fredette break;
413 1.1 fredette }
414 1.1 fredette } else if (!strcmp(argv[argv_i], "-i")
415 1.1 fredette || !strcmp(argv[argv_i], "--interface")) {
416 1.1 fredette if (++argv_i < argc) {
417 1.1 fredette interface_name = argv[argv_i];
418 1.1 fredette } else {
419 1.1 fredette show_usage = TRUE;
420 1.1 fredette break;
421 1.1 fredette }
422 1.1 fredette } else if (!strcmp(argv[argv_i], "-w")
423 1.1 fredette || !strcmp(argv[argv_i], "--window-size")) {
424 1.1 fredette if (++argv_i == argc || (nd_window_size = atoi(argv[argv_i])) <= 0) {
425 1.1 fredette show_usage = TRUE;
426 1.1 fredette break;
427 1.1 fredette }
428 1.1 fredette }
429 1.1 fredette #ifdef _NDBOOTD_DO_DEBUG
430 1.1 fredette else if (!strcmp(argv[argv_i], "-d")
431 1.1 fredette || !strcmp(argv[argv_i], "--debug")) {
432 1.1 fredette _ndbootd_debug = TRUE;
433 1.1 fredette }
434 1.1 fredette #endif /* _NDBOOTD_DO_DEBUG */
435 1.1 fredette else {
436 1.1 fredette if (strcmp(argv[argv_i], "-h")
437 1.1 fredette && strcmp(argv[argv_i], "--help")) {
438 1.1 fredette fprintf(stderr, "%s error: unknown switch '%s'\n",
439 1.1 fredette _ndbootd_argv0, argv[argv_i]);
440 1.1 fredette }
441 1.1 fredette show_usage = TRUE;
442 1.1 fredette break;
443 1.1 fredette }
444 1.1 fredette }
445 1.1 fredette if (argv_i + 1 == argc) {
446 1.1 fredette boot1_file_name = argv[argv_i];
447 1.1 fredette } else {
448 1.1 fredette show_usage = TRUE;
449 1.1 fredette }
450 1.1 fredette
451 1.1 fredette if (show_usage) {
452 1.1 fredette fprintf(stderr, "\
453 1.1 fredette usage: %s [OPTIONS] BOOT1-BIN\n\
454 1.1 fredette where OPTIONS are:\n\
455 1.1 fredette -s, --boot2 { BOOT2-BIN | DIR }\n\
456 1.1 fredette find a second-stage boot program in the file\n\
457 1.1 fredette BOOT2-BIN or in the directory DIR\n\
458 1.1 fredette -i, --interface NAME use interface NAME\n\
459 1.1 fredette -w, --window-size COUNT \n\
460 1.1 fredette send at most COUNT unacknowledged packets [default=%d]\n",
461 1.1 fredette _ndbootd_argv0,
462 1.1 fredette NDBOOT_WINDOW_SIZE_DEFAULT);
463 1.1 fredette #ifdef _NDBOOTD_DO_DEBUG
464 1.1 fredette fprintf(stderr, "\
465 1.1 fredette -d, --debug set debug mode\n");
466 1.1 fredette #endif /* _NDBOOTD_DO_DEBUG */
467 1.1 fredette exit(1);
468 1.1 fredette }
469 1.1 fredette /* if we have been given a name for the second-stage boot, see if it's
470 1.1 fredette * a filename or a directory: */
471 1.1 fredette boot2_x_name_is_dir = FALSE;
472 1.1 fredette if (boot2_x_name != NULL) {
473 1.1 fredette if (stat(boot2_x_name, &stat_buffer) < 0) {
474 1.1 fredette fprintf(stderr, "%s error: could not stat %s: %s\n",
475 1.1 fredette _ndbootd_argv0, boot2_x_name, strerror(errno));
476 1.1 fredette exit(1);
477 1.1 fredette }
478 1.1 fredette if (S_ISDIR(stat_buffer.st_mode)) {
479 1.1 fredette boot2_x_name_is_dir = TRUE;
480 1.1 fredette } else if (!S_ISREG(stat_buffer.st_mode)) {
481 1.1 fredette fprintf(stderr, "%s error: %s is neither a regular file nor a directory\n",
482 1.1 fredette _ndbootd_argv0, boot2_x_name);
483 1.1 fredette exit(1);
484 1.1 fredette }
485 1.1 fredette }
486 1.1 fredette /* find the interface we will use: */
487 1.1 fredette if ((interface = _ndbootd_find_interface(interface_name)) == NULL) {
488 1.1 fredette fprintf(stderr, "%s error: could not find the interface to use: %s\n",
489 1.1 fredette _ndbootd_argv0, strerror(errno));
490 1.1 fredette exit(1);
491 1.1 fredette }
492 1.1 fredette _NDBOOTD_DEBUG((fp, "opening interface %s", interface->ndbootd_interface_ifreq->ifr_name));
493 1.1 fredette
494 1.1 fredette /* open the network interface: */
495 1.1 fredette if (ndbootd_raw_open(interface)) {
496 1.1 fredette fprintf(stderr, "%s error: could not open the %s interface: %s\n",
497 1.1 fredette _ndbootd_argv0, interface->ndbootd_interface_ifreq->ifr_name, strerror(errno));
498 1.1 fredette exit(1);
499 1.1 fredette }
500 1.1 fredette _NDBOOTD_DEBUG((fp, "opened interface %s (ip %s ether %02x:%02x:%02x:%02x:%02x:%02x)",
501 1.1 fredette interface->ndbootd_interface_ifreq->ifr_name,
502 1.1 fredette inet_ntoa(((struct sockaddr_in *) & interface->ndbootd_interface_ifreq->ifr_addr)->sin_addr),
503 1.1 fredette ((unsigned char *) interface->ndbootd_interface_ether)[0],
504 1.1 fredette ((unsigned char *) interface->ndbootd_interface_ether)[1],
505 1.1 fredette ((unsigned char *) interface->ndbootd_interface_ether)[2],
506 1.1 fredette ((unsigned char *) interface->ndbootd_interface_ether)[3],
507 1.1 fredette ((unsigned char *) interface->ndbootd_interface_ether)[4],
508 1.1 fredette ((unsigned char *) interface->ndbootd_interface_ether)[5]));
509 1.1 fredette
510 1.1 fredette /* become a daemon: */
511 1.1 fredette #ifdef _NDBOOTD_DO_DEBUG
512 1.1 fredette if (!_ndbootd_debug)
513 1.1 fredette #endif /* _NDBOOTD_DO_DEBUG */
514 1.1 fredette {
515 1.1 fredette
516 1.1 fredette /* fork and exit: */
517 1.1 fredette switch (fork()) {
518 1.1 fredette case 0:
519 1.1 fredette break;
520 1.1 fredette case -1:
521 1.1 fredette fprintf(stderr, "%s error: could not fork: %s\n",
522 1.1 fredette _ndbootd_argv0, strerror(errno));
523 1.1 fredette exit(1);
524 1.1 fredette default:
525 1.1 fredette exit(0);
526 1.1 fredette }
527 1.1 fredette
528 1.1 fredette /* close all file descriptors: */
529 1.1 fredette #ifdef HAVE_GETDTABLESIZE
530 1.1 fredette fd = getdtablesize();
531 1.1 fredette #else /* !HAVE_GETDTABLESIZE */
532 1.1 fredette fd = -1;
533 1.1 fredette #endif /* !HAVE_GETDTABLESIZE */
534 1.1 fredette for (; fd >= 0; fd--) {
535 1.1 fredette if (fd != interface->ndbootd_interface_fd) {
536 1.1 fredette close(fd);
537 1.1 fredette }
538 1.1 fredette }
539 1.1 fredette
540 1.1 fredette #ifdef HAVE_SETSID
541 1.1 fredette /* become our own session: */
542 1.1 fredette setsid();
543 1.1 fredette #endif /* HAVE_SETSID */
544 1.1 fredette }
545 1.1 fredette /* write the pid file: */
546 1.1 fredette if ((fd = open(NDBOOTD_PID_FILE, O_WRONLY | O_CREAT | O_TRUNC, 0644)) >= 0) {
547 1.1 fredette sprintf(pid_buffer, "%u\n", getpid());
548 1.1 fredette write(fd, pid_buffer, strlen(pid_buffer));
549 1.1 fredette close(fd);
550 1.1 fredette }
551 1.1 fredette #ifdef HAVE_STRICT_ALIGNMENT
552 1.1 fredette /* we will be dealing with all packet headers in separate buffers, to
553 1.1 fredette * make sure everything is correctly aligned: */
554 1.1 fredette ether_packet = ðer_packet_buffer;
555 1.1 fredette ip_packet = (struct ip *) & ip_packet_buffer[0];
556 1.1 fredette nd_packet = &nd_packet_buffer;
557 1.1 fredette #else /* !HAVE_STRICT_ALIGNMENT */
558 1.1 fredette /* we will always find the Ethernet header and the IP packet at the
559 1.1 fredette * front of the buffer: */
560 1.1 fredette ether_packet = (struct ether_header *) packet_buffer;
561 1.1 fredette ip_packet = (struct ip *) (ether_packet + 1);
562 1.1 fredette #endif /* !HAVE_STRICT_ALIGNMENT */
563 1.1 fredette
564 1.1 fredette /* initialize our state: */
565 1.1 fredette last_rarp_time = 0;
566 1.1 fredette last_open_time = 0;
567 1.1 fredette boot1_fd = -1;
568 1.1 fredette boot2_file_name = NULL;
569 1.1 fredette boot2_fd = -1;
570 1.1 fredette
571 1.1 fredette /* loop processing packets: */
572 1.1 fredette for (;;) {
573 1.1 fredette
574 1.1 fredette /* receive another packet: */
575 1.1 fredette packet_length = ndbootd_raw_read(interface, packet_buffer, sizeof(packet_buffer));
576 1.1 fredette if (packet_length < 0) {
577 1.1 fredette _NDBOOTD_DEBUG((fp, "failed to receive packet: %s", strerror(errno)));
578 1.1 fredette exit(1);
579 1.1 fredette continue;
580 1.1 fredette }
581 1.1 fredette now = time(NULL);
582 1.1 fredette
583 1.1 fredette /* check the Ethernet and IP parts of the packet: */
584 1.1 fredette if (packet_length
585 1.1 fredette < (sizeof(struct ether_header)
586 1.1 fredette + sizeof(struct ip)
587 1.1 fredette + sizeof(struct ndboot_packet))) {
588 1.3 fredette _NDBOOTD_DEBUG((fp, "ignoring a too-short packet of length %ld", (long) packet_length));
589 1.1 fredette continue;
590 1.1 fredette }
591 1.1 fredette #ifdef HAVE_STRICT_ALIGNMENT
592 1.1 fredette memcpy(ether_packet, packet_buffer, sizeof(struct ether_header));
593 1.1 fredette memcpy(ip_packet, packet_buffer + sizeof(struct ether_header),
594 1.1 fredette (((struct ip *) (packet_buffer + sizeof(struct ether_header)))->ip_hl << 2));
595 1.1 fredette #endif /* !HAVE_STRICT_ALIGNMENT */
596 1.1 fredette if (ether_packet->ether_type != htons(ETHERTYPE_IP)
597 1.1 fredette || ip_packet->ip_p != IPPROTO_ND) {
598 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with the wrong Ethernet or IP protocol"));
599 1.1 fredette continue;
600 1.1 fredette }
601 1.1 fredette _ndbootd_ip_cksum(ip_packet);
602 1.1 fredette if (ip_packet->ip_sum != 0) {
603 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with a bad IP checksum"));
604 1.1 fredette continue;
605 1.1 fredette }
606 1.1 fredette if (packet_length
607 1.1 fredette != (sizeof(struct ether_header)
608 1.1 fredette + (ip_packet->ip_hl << 2)
609 1.1 fredette + sizeof(struct ndboot_packet))) {
610 1.3 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with bad total length %ld", (long) packet_length));
611 1.1 fredette continue;
612 1.1 fredette }
613 1.1 fredette /* if we need to, refresh our RARP cache: */
614 1.1 fredette if ((last_rarp_time + NDBOOTD_CLIENT_TTL_SECONDS) < now
615 1.1 fredette || memcmp(last_client_ether, ether_packet->ether_shost, ETHER_ADDR_LEN)) {
616 1.1 fredette
617 1.1 fredette /* turn the Ethernet address into a hostname: */
618 1.1 fredette if (ether_ntohost(hostname_buffer, (struct ether_addr *) ether_packet->ether_shost)) {
619 1.1 fredette _NDBOOTD_DEBUG((fp, "could not resolve %02x:%02x:%02x:%02x:%02x:%02x into a hostname: %s",
620 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[0],
621 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[1],
622 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[2],
623 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[3],
624 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[4],
625 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[5],
626 1.1 fredette strerror(errno)));
627 1.1 fredette continue;
628 1.1 fredette }
629 1.1 fredette /* turn the hostname into an IP address: */
630 1.1 fredette hostname_buffer[sizeof(hostname_buffer) - 1] = '\0';
631 1.1 fredette if ((the_hostent = gethostbyname(hostname_buffer)) == NULL
632 1.1 fredette || the_hostent->h_addrtype != AF_INET) {
633 1.1 fredette _NDBOOTD_DEBUG((fp, "could not resolve %s into an IP address: %s",
634 1.1 fredette hostname_buffer,
635 1.1 fredette strerror(errno)));
636 1.1 fredette continue;
637 1.1 fredette }
638 1.1 fredette /* save these new results in our RARP cache: */
639 1.1 fredette last_rarp_time = now;
640 1.1 fredette memcpy(last_client_ether, ether_packet->ether_shost, ETHER_ADDR_LEN);
641 1.1 fredette memcpy(&last_client_ip, the_hostent->h_addr, sizeof(last_client_ip));
642 1.1 fredette _NDBOOTD_DEBUG((fp, "IP address for %02x:%02x:%02x:%02x:%02x:%02x is %s",
643 1.1 fredette ((unsigned char *) last_client_ether)[0],
644 1.1 fredette ((unsigned char *) last_client_ether)[1],
645 1.1 fredette ((unsigned char *) last_client_ether)[2],
646 1.1 fredette ((unsigned char *) last_client_ether)[3],
647 1.1 fredette ((unsigned char *) last_client_ether)[4],
648 1.1 fredette ((unsigned char *) last_client_ether)[5],
649 1.1 fredette inet_ntoa(last_client_ip)));
650 1.1 fredette
651 1.1 fredette /* this will cause the file descriptor cache to be
652 1.1 fredette * reloaded, the next time we make it that far: */
653 1.1 fredette last_open_time = 0;
654 1.1 fredette }
655 1.1 fredette /* if this IP packet was broadcast, rewrite the source IP
656 1.1 fredette * address to be the client, else, check that the client is
657 1.1 fredette * using the correct IP addresses: */
658 1.1 fredette if (ip_packet->ip_dst.s_addr == htonl(0)) {
659 1.1 fredette ip_packet->ip_src = last_client_ip;
660 1.1 fredette } else {
661 1.1 fredette if (ip_packet->ip_src.s_addr !=
662 1.1 fredette last_client_ip.s_addr) {
663 1.1 fredette _NDBOOTD_DEBUG((fp, "machine %02x:%02x:%02x:%02x:%02x:%02x is using the wrong IP address\n",
664 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[0],
665 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[1],
666 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[2],
667 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[3],
668 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[4],
669 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[5]));
670 1.1 fredette continue;
671 1.1 fredette }
672 1.1 fredette if (ip_packet->ip_dst.s_addr
673 1.1 fredette != ((struct sockaddr_in *) & interface->ndbootd_interface_ifreq->ifr_addr)->sin_addr.s_addr) {
674 1.1 fredette _NDBOOTD_DEBUG((fp, "machine %02x:%02x:%02x:%02x:%02x:%02x is sending to the wrong IP address\n",
675 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[0],
676 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[1],
677 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[2],
678 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[3],
679 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[4],
680 1.1 fredette ((unsigned char *) ether_packet->ether_shost)[5]));
681 1.1 fredette continue;
682 1.1 fredette }
683 1.1 fredette }
684 1.1 fredette
685 1.1 fredette /* if we need to, refresh our "cache" of file descriptors for
686 1.1 fredette * the boot programs: */
687 1.1 fredette if ((last_open_time + NDBOOTD_CLIENT_TTL_SECONDS) < now) {
688 1.1 fredette
689 1.1 fredette /* close any previously opened programs: */
690 1.1 fredette if (boot1_fd >= 0) {
691 1.1 fredette close(boot1_fd);
692 1.1 fredette }
693 1.1 fredette if (boot2_file_name != NULL) {
694 1.1 fredette free(boot2_file_name);
695 1.1 fredette }
696 1.1 fredette if (boot2_fd >= 0) {
697 1.1 fredette close(boot2_fd);
698 1.1 fredette }
699 1.1 fredette /* open the first-stage boot program: */
700 1.1 fredette if ((boot1_fd = open(boot1_file_name, O_RDONLY)) < 0) {
701 1.1 fredette _NDBOOTD_DEBUG((fp, "could not open %s: %s",
702 1.1 fredette boot1_file_name, strerror(errno)));
703 1.1 fredette continue;
704 1.1 fredette }
705 1.1 fredette if (fstat(boot1_fd, &stat_buffer) < 0) {
706 1.1 fredette _NDBOOTD_DEBUG((fp, "could not stat %s: %s",
707 1.1 fredette boot1_file_name, strerror(errno)));
708 1.1 fredette continue;
709 1.1 fredette }
710 1.1 fredette boot1_byte_count = stat_buffer.st_size;
711 1.1 fredette boot1_block_count = (boot1_byte_count + (NDBOOT_BSIZE - 1)) / NDBOOT_BSIZE;
712 1.1 fredette if (boot1_block_count > NDBOOTD_BOOT1_BLOCK_COUNT) {
713 1.1 fredette _NDBOOTD_DEBUG((fp, "first-stage boot program %s has too many blocks (%d, max is %d)",
714 1.1 fredette boot1_file_name, boot1_block_count, NDBOOTD_BOOT1_BLOCK_COUNT));
715 1.1 fredette }
716 1.1 fredette _NDBOOTD_DEBUG((fp, "first-stage boot program %s has %d blocks",
717 1.1 fredette boot1_file_name, boot1_block_count));
718 1.1 fredette
719 1.1 fredette /* open any second-stage boot program: */
720 1.1 fredette if (boot2_x_name != NULL) {
721 1.1 fredette
722 1.1 fredette /* determine what the name of the second-stage
723 1.1 fredette * boot program will be: */
724 1.1 fredette if (boot2_x_name_is_dir) {
725 1.1 fredette if ((boot2_file_name = malloc(strlen(boot2_x_name) + strlen("/00000000.SUN2") + 1)) != NULL) {
726 1.1 fredette sprintf(boot2_file_name, "%s/%02X%02X%02X%02X.SUN2",
727 1.1 fredette boot2_x_name,
728 1.1 fredette ((unsigned char *) &last_client_ip)[0],
729 1.1 fredette ((unsigned char *) &last_client_ip)[1],
730 1.1 fredette ((unsigned char *) &last_client_ip)[2],
731 1.1 fredette ((unsigned char *) &last_client_ip)[3]);
732 1.1 fredette }
733 1.1 fredette } else {
734 1.1 fredette boot2_file_name = strdup(boot2_x_name);
735 1.1 fredette }
736 1.1 fredette if (boot2_file_name == NULL) {
737 1.1 fredette abort();
738 1.1 fredette }
739 1.1 fredette /* open the second-stage boot program: */
740 1.1 fredette if ((boot2_fd = open(boot2_file_name, O_RDONLY)) < 0) {
741 1.1 fredette _NDBOOTD_DEBUG((fp, "could not open %s: %s",
742 1.1 fredette boot2_file_name, strerror(errno)));
743 1.1 fredette continue;
744 1.1 fredette }
745 1.1 fredette if (fstat(boot2_fd, &stat_buffer) < 0) {
746 1.1 fredette _NDBOOTD_DEBUG((fp, "could not stat %s: %s",
747 1.1 fredette boot2_file_name, strerror(errno)));
748 1.1 fredette continue;
749 1.1 fredette }
750 1.1 fredette boot2_byte_count = stat_buffer.st_size;
751 1.1 fredette boot2_block_count = (boot2_byte_count + (NDBOOT_BSIZE - 1)) / NDBOOT_BSIZE;
752 1.1 fredette _NDBOOTD_DEBUG((fp, "second-stage boot program %s has %d blocks",
753 1.1 fredette boot2_file_name, boot2_block_count));
754 1.1 fredette }
755 1.1 fredette /* success: */
756 1.1 fredette last_open_time = now;
757 1.1 fredette }
758 1.1 fredette /* check the nd packet: */
759 1.1 fredette #ifdef HAVE_STRICT_ALIGNMENT
760 1.1 fredette memcpy(nd_packet, packet_buffer + sizeof(struct ether_header) + (ip_packet->ip_hl << 2), sizeof(struct ndboot_packet));
761 1.1 fredette #else /* !HAVE_STRICT_ALIGNMENT */
762 1.1 fredette nd_packet = (struct ndboot_packet *) (((char *) ip_packet) + (ip_packet->ip_hl << 2));
763 1.1 fredette #endif /* !HAVE_STRICT_ALIGNMENT */
764 1.1 fredette
765 1.1 fredette /* dump a bunch of debug information: */
766 1.1 fredette _NDBOOTD_DEBUG((fp, "recv: op 0x%02x minor 0x%02x error %d vers %d seq %d blk %d bcount %d off %d count %d",
767 1.1 fredette nd_packet->ndboot_packet_op,
768 1.1 fredette nd_packet->ndboot_packet_minor,
769 1.1 fredette nd_packet->ndboot_packet_error,
770 1.1 fredette nd_packet->ndboot_packet_disk_version,
771 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_sequence),
772 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_block_number),
773 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_byte_count),
774 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_current_byte_offset),
775 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_current_byte_count)));
776 1.1 fredette
777 1.1 fredette /* ignore this packet if it has a bad opcode, a bad minor
778 1.1 fredette * number, a bad disk version, a bad block number, a bad byte
779 1.1 fredette * count, a bad current byte offset, or a bad current byte
780 1.1 fredette * count: */
781 1.1 fredette /* FIXME - for some of these conditions, we probably should
782 1.1 fredette * return an NDBOOT_OP_ERROR packet: */
783 1.1 fredette if ((nd_packet->ndboot_packet_op & NDBOOT_OP_MASK) != NDBOOT_OP_READ) {
784 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with bad op %d",
785 1.1 fredette nd_packet->ndboot_packet_op & NDBOOT_OP_MASK));
786 1.1 fredette continue;
787 1.1 fredette }
788 1.1 fredette if (nd_packet->ndboot_packet_minor != NDBOOT_MINOR_NDP0) {
789 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with device minor %d",
790 1.1 fredette nd_packet->ndboot_packet_minor));
791 1.1 fredette continue;
792 1.1 fredette }
793 1.1 fredette if (nd_packet->ndboot_packet_disk_version != 0) {
794 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with disk version %d",
795 1.1 fredette nd_packet->ndboot_packet_disk_version));
796 1.1 fredette continue;
797 1.1 fredette }
798 1.1 fredette if (ntohl(nd_packet->ndboot_packet_block_number) < 0) {
799 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with bad block number %d",
800 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_block_number)));
801 1.1 fredette continue;
802 1.1 fredette }
803 1.1 fredette if (ntohl(nd_packet->ndboot_packet_byte_count) <= 0 ||
804 1.1 fredette ntohl(nd_packet->ndboot_packet_byte_count) > NDBOOT_MAX_BYTE_COUNT) {
805 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with bad byte count %d",
806 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_byte_count)));
807 1.1 fredette continue;
808 1.1 fredette }
809 1.1 fredette if (ntohl(nd_packet->ndboot_packet_current_byte_offset) < 0 ||
810 1.1 fredette ntohl(nd_packet->ndboot_packet_current_byte_offset)
811 1.1 fredette >= ntohl(nd_packet->ndboot_packet_byte_count)) {
812 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with bad current offset %d",
813 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_current_byte_offset)));
814 1.1 fredette continue;
815 1.1 fredette }
816 1.1 fredette if (ntohl(nd_packet->ndboot_packet_current_byte_count) < 0 ||
817 1.1 fredette ntohl(nd_packet->ndboot_packet_current_byte_count)
818 1.1 fredette > (ntohl(nd_packet->ndboot_packet_byte_count)
819 1.1 fredette - ntohl(nd_packet->ndboot_packet_current_byte_offset))) {
820 1.1 fredette _NDBOOTD_DEBUG((fp, "ignoring a packet with bad current count %d",
821 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_current_byte_count)));
822 1.1 fredette continue;
823 1.1 fredette }
824 1.1 fredette /* if we were given a current byte count of zero, rewrite it
825 1.1 fredette * to be the maximum: */
826 1.1 fredette if (ntohl(nd_packet->ndboot_packet_current_byte_count) == 0) {
827 1.1 fredette nd_packet->ndboot_packet_current_byte_count =
828 1.1 fredette htonl(ntohl(nd_packet->ndboot_packet_byte_count)
829 1.1 fredette - ntohl(nd_packet->ndboot_packet_current_byte_offset));
830 1.1 fredette }
831 1.1 fredette /* read the data: */
832 1.1 fredette disk_buffer_offset = 0;
833 1.1 fredette block_number = ntohl(nd_packet->ndboot_packet_block_number);
834 1.1 fredette byte_offset = ntohl(nd_packet->ndboot_packet_current_byte_offset);
835 1.1 fredette byte_count = ntohl(nd_packet->ndboot_packet_current_byte_count);
836 1.1 fredette for (; byte_count > 0;) {
837 1.1 fredette
838 1.1 fredette /* adjust the current block number and byte offset
839 1.1 fredette * such that the byte offset is always < NDBOOT_BSIZE: */
840 1.1 fredette block_number += (byte_offset / NDBOOT_BSIZE);
841 1.1 fredette byte_offset = byte_offset % NDBOOT_BSIZE;
842 1.1 fredette
843 1.1 fredette /* dispatch on the beginning block number: */
844 1.1 fredette byte_count_read = 0;
845 1.1 fredette
846 1.1 fredette /* the (dummy) Sun disk label: */
847 1.1 fredette if (block_number >= NDBOOTD_SUNDK_BLOCK_FIRST
848 1.1 fredette && block_number < (NDBOOTD_SUNDK_BLOCK_FIRST + NDBOOTD_SUNDK_BLOCK_COUNT)) {
849 1.1 fredette byte_count_read = MIN(NDBOOTD_BYTES_AVAIL(block_number, byte_offset,
850 1.1 fredette NDBOOTD_SUNDK_BLOCK_FIRST, NDBOOTD_SUNDK_BLOCK_COUNT),
851 1.1 fredette byte_count);
852 1.1 fredette }
853 1.1 fredette /* the first-stage boot program: */
854 1.1 fredette else if (block_number >= NDBOOTD_BOOT1_BLOCK_FIRST
855 1.1 fredette && block_number < (NDBOOTD_BOOT1_BLOCK_FIRST + NDBOOTD_BOOT1_BLOCK_COUNT)) {
856 1.1 fredette
857 1.1 fredette /* if any real part of the first-stage boot
858 1.1 fredette * program is needed to satisfy the request,
859 1.1 fredette * read it (otherwise we return garbage as
860 1.1 fredette * padding): */
861 1.1 fredette byte_count_wanted = MIN(NDBOOTD_BYTES_AVAIL(block_number, byte_offset,
862 1.1 fredette NDBOOTD_BOOT1_BLOCK_FIRST, boot1_block_count),
863 1.1 fredette byte_count);
864 1.1 fredette if (byte_count_wanted > 0) {
865 1.1 fredette
866 1.1 fredette file_offset = ((block_number - NDBOOTD_BOOT1_BLOCK_FIRST) * NDBOOT_BSIZE) + byte_offset;
867 1.1 fredette if (lseek(boot1_fd, file_offset, SEEK_SET) < 0) {
868 1.3 fredette _NDBOOTD_DEBUG((fp, "could not seek %s to block %ld offset %ld: %s",
869 1.1 fredette boot1_file_name,
870 1.3 fredette (long) (block_number - NDBOOTD_BOOT1_BLOCK_FIRST),
871 1.3 fredette (long) byte_offset,
872 1.1 fredette strerror(errno)));
873 1.1 fredette break;
874 1.1 fredette }
875 1.1 fredette byte_count_read = read(boot1_fd, disk_buffer + disk_buffer_offset, byte_count_wanted);
876 1.1 fredette /* pretend that the size of the
877 1.1 fredette * first-stage boot program is a
878 1.1 fredette * multiple of NDBOOT_BSIZE: */
879 1.1 fredette if (byte_count_read != byte_count_wanted
880 1.4 fredette && byte_count_read > 0
881 1.1 fredette && file_offset + byte_count_read == boot1_byte_count) {
882 1.1 fredette byte_count_read = byte_count_wanted;
883 1.1 fredette }
884 1.1 fredette if (byte_count_read != byte_count_wanted) {
885 1.3 fredette _NDBOOTD_DEBUG((fp, "could not read %ld bytes at block %ld offset %ld from %s: %s (read %ld bytes)",
886 1.3 fredette (long) byte_count_wanted,
887 1.3 fredette (long) (block_number - NDBOOTD_BOOT1_BLOCK_FIRST),
888 1.3 fredette (long) byte_offset,
889 1.1 fredette boot1_file_name,
890 1.1 fredette strerror(errno),
891 1.3 fredette (long) byte_count_read));
892 1.1 fredette break;
893 1.1 fredette }
894 1.1 fredette }
895 1.1 fredette /* the number of bytes we read, including any
896 1.1 fredette * padding garbage: */
897 1.1 fredette byte_count_read = MIN(NDBOOTD_BYTES_AVAIL(block_number, byte_offset,
898 1.1 fredette NDBOOTD_BOOT1_BLOCK_FIRST, NDBOOTD_BOOT1_BLOCK_COUNT),
899 1.1 fredette byte_count);
900 1.1 fredette }
901 1.1 fredette /* any second-stage boot program: */
902 1.1 fredette else if (block_number >= NDBOOTD_BOOT2_BLOCK_FIRST) {
903 1.1 fredette
904 1.1 fredette /* if any real part of any first-stage boot
905 1.1 fredette * program is needed to satisfy the request,
906 1.1 fredette * read it (otherwise we return garbage as
907 1.1 fredette * padding): */
908 1.1 fredette byte_count_wanted = MIN(NDBOOTD_BYTES_AVAIL(block_number, byte_offset,
909 1.1 fredette NDBOOTD_BOOT2_BLOCK_FIRST, boot2_block_count),
910 1.1 fredette byte_count);
911 1.1 fredette if (boot2_fd >= 0
912 1.1 fredette && byte_count_wanted > 0) {
913 1.1 fredette
914 1.1 fredette file_offset = ((block_number - NDBOOTD_BOOT2_BLOCK_FIRST) * NDBOOT_BSIZE) + byte_offset;
915 1.1 fredette if (lseek(boot2_fd, file_offset, SEEK_SET) < 0) {
916 1.3 fredette _NDBOOTD_DEBUG((fp, "could not seek %s to block %ld offset %ld: %s",
917 1.1 fredette boot2_file_name,
918 1.3 fredette (long) (block_number - NDBOOTD_BOOT2_BLOCK_FIRST),
919 1.3 fredette (long) byte_offset,
920 1.1 fredette strerror(errno)));
921 1.1 fredette break;
922 1.1 fredette }
923 1.1 fredette byte_count_read = read(boot2_fd, disk_buffer + disk_buffer_offset, byte_count_wanted);
924 1.1 fredette /* pretend that the size of the
925 1.1 fredette * second-stage boot program is a
926 1.1 fredette * multiple of NDBOOT_BSIZE: */
927 1.1 fredette if (byte_count_read != byte_count_wanted
928 1.4 fredette && byte_count_read > 0
929 1.1 fredette && file_offset + byte_count_read == boot2_byte_count) {
930 1.1 fredette byte_count_read = byte_count_wanted;
931 1.1 fredette }
932 1.1 fredette if (byte_count_read != byte_count_wanted) {
933 1.3 fredette _NDBOOTD_DEBUG((fp, "could not read %ld bytes at block %ld offset %ld from %s: %s (read %ld bytes)",
934 1.3 fredette (long) byte_count_wanted,
935 1.3 fredette (long) (block_number - NDBOOTD_BOOT2_BLOCK_FIRST),
936 1.3 fredette (long) byte_offset,
937 1.1 fredette boot2_file_name,
938 1.1 fredette strerror(errno),
939 1.3 fredette (long) byte_count_read));
940 1.1 fredette break;
941 1.1 fredette }
942 1.1 fredette }
943 1.1 fredette /* the number of bytes we read, including any
944 1.1 fredette * padding garbage: */
945 1.1 fredette byte_count_read = byte_count;
946 1.1 fredette }
947 1.1 fredette /* update for the amount that we read: */
948 1.1 fredette assert(byte_count_read > 0);
949 1.1 fredette disk_buffer_offset += byte_count_read;
950 1.1 fredette byte_offset += byte_count_read;
951 1.1 fredette byte_count -= byte_count_read;
952 1.1 fredette }
953 1.1 fredette if (byte_count > 0) {
954 1.1 fredette /* an error occurred: */
955 1.1 fredette continue;
956 1.1 fredette }
957 1.1 fredette /* set the Ethernet and IP destination and source addresses,
958 1.1 fredette * and the IP TTL: */
959 1.1 fredette memcpy(ether_packet->ether_dhost, ether_packet->ether_shost, ETHER_ADDR_LEN);
960 1.1 fredette memcpy(ether_packet->ether_shost, interface->ndbootd_interface_ether, ETHER_ADDR_LEN);
961 1.1 fredette #ifdef HAVE_STRICT_ALIGNMENT
962 1.1 fredette memcpy(packet_buffer, ether_packet, sizeof(struct ether_header));
963 1.1 fredette #endif /* !HAVE_STRICT_ALIGNMENT */
964 1.1 fredette ip_packet->ip_dst = ip_packet->ip_src;
965 1.1 fredette ip_packet->ip_src = ((struct sockaddr_in *) & interface->ndbootd_interface_ifreq->ifr_addr)->sin_addr;
966 1.1 fredette ip_packet->ip_ttl = 4;
967 1.1 fredette
968 1.1 fredette /* return the data: */
969 1.1 fredette nd_window_filled = 0;
970 1.1 fredette disk_buffer_offset = 0;
971 1.1 fredette byte_count = ntohl(nd_packet->ndboot_packet_current_byte_count);
972 1.1 fredette for (;;) {
973 1.1 fredette
974 1.1 fredette /* set the byte count on this packet: */
975 1.1 fredette nd_packet->ndboot_packet_current_byte_count = htonl(MIN(byte_count, NDBOOT_MAX_PACKET_DATA));
976 1.1 fredette
977 1.1 fredette /* set our opcode. the opcode is always
978 1.1 fredette * NDBOOT_OP_READ, ORed with NDBOOT_OP_FLAG_DONE |
979 1.1 fredette * NDBOOT_OP_FLAG_WAIT if this packet finishes the
980 1.1 fredette * request, or ORed with NDBOOT_OP_FLAG_WAIT if this
981 1.1 fredette * packet fills the window: */
982 1.1 fredette nd_window_filled++;
983 1.1 fredette nd_packet->ndboot_packet_op =
984 1.1 fredette (NDBOOT_OP_READ
985 1.1 fredette | ((ntohl(nd_packet->ndboot_packet_current_byte_offset)
986 1.1 fredette + ntohl(nd_packet->ndboot_packet_current_byte_count))
987 1.1 fredette == ntohl(nd_packet->ndboot_packet_byte_count)
988 1.1 fredette ? (NDBOOT_OP_FLAG_DONE
989 1.1 fredette | NDBOOT_OP_FLAG_WAIT)
990 1.1 fredette : (nd_window_filled == nd_window_size
991 1.1 fredette ? NDBOOT_OP_FLAG_WAIT
992 1.1 fredette : 0)));
993 1.1 fredette
994 1.1 fredette /* copy the data into the packet: */
995 1.1 fredette memcpy(packet_buffer +
996 1.1 fredette sizeof(struct ether_header) + (ip_packet->ip_hl << 2) + sizeof(struct ndboot_packet),
997 1.1 fredette disk_buffer + disk_buffer_offset,
998 1.1 fredette ntohl(nd_packet->ndboot_packet_current_byte_count));
999 1.1 fredette
1000 1.1 fredette /* finish the IP packet and calculate the checksum: */
1001 1.1 fredette ip_packet->ip_len = htons((ip_packet->ip_hl << 2)
1002 1.1 fredette + sizeof(struct ndboot_packet)
1003 1.1 fredette + ntohl(nd_packet->ndboot_packet_current_byte_count));
1004 1.1 fredette ip_packet->ip_sum = 0;
1005 1.1 fredette _ndbootd_ip_cksum(ip_packet);
1006 1.1 fredette
1007 1.1 fredette #ifdef HAVE_STRICT_ALIGNMENT
1008 1.1 fredette memcpy(packet_buffer + sizeof(struct ether_header), ip_packet, ip_packet->ip_hl << 2);
1009 1.1 fredette memcpy(packet_buffer + sizeof(struct ether_header) + (ip_packet->ip_hl << 2), nd_packet, sizeof(struct ndboot_packet));
1010 1.1 fredette #endif /* !HAVE_STRICT_ALIGNMENT */
1011 1.1 fredette
1012 1.1 fredette /* dump a bunch of debug information: */
1013 1.1 fredette _NDBOOTD_DEBUG((fp, "send: op 0x%02x minor 0x%02x error %d vers %d seq %d blk %d bcount %d off %d count %d (win %d)",
1014 1.1 fredette nd_packet->ndboot_packet_op,
1015 1.1 fredette nd_packet->ndboot_packet_minor,
1016 1.1 fredette nd_packet->ndboot_packet_error,
1017 1.1 fredette nd_packet->ndboot_packet_disk_version,
1018 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_sequence),
1019 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_block_number),
1020 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_byte_count),
1021 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_current_byte_offset),
1022 1.1 fredette (int) ntohl(nd_packet->ndboot_packet_current_byte_count),
1023 1.1 fredette nd_window_filled - 1));
1024 1.1 fredette
1025 1.1 fredette /* delay before sending the packet: */
1026 1.1 fredette send_delay.tv_sec = 0;
1027 1.1 fredette send_delay.tv_usec = NDBOOTD_SEND_DELAY_USECONDS;
1028 1.1 fredette select(0, NULL, NULL, NULL, &send_delay);
1029 1.1 fredette
1030 1.1 fredette /* transmit the packet: */
1031 1.1 fredette if (ndbootd_raw_write(interface, packet_buffer,
1032 1.1 fredette sizeof(struct ether_header) + (ip_packet->ip_hl << 2) + sizeof(struct ndboot_packet) + ntohl(nd_packet->ndboot_packet_current_byte_count)) < 0) {
1033 1.1 fredette _NDBOOTD_DEBUG((fp, "could not write a packet: %s",
1034 1.1 fredette strerror(errno)));
1035 1.1 fredette }
1036 1.1 fredette /* if we set NDBOOT_OP_FLAG_DONE or
1037 1.1 fredette * NDBOOT_OP_FLAG_WAIT in the packet we just sent,
1038 1.1 fredette * we're done sending: */
1039 1.1 fredette if (nd_packet->ndboot_packet_op != NDBOOT_OP_READ) {
1040 1.1 fredette break;
1041 1.1 fredette }
1042 1.1 fredette /* advance to the next packet: */
1043 1.1 fredette byte_count -= ntohl(nd_packet->ndboot_packet_current_byte_count);
1044 1.1 fredette disk_buffer_offset += ntohl(nd_packet->ndboot_packet_current_byte_count);
1045 1.1 fredette nd_packet->ndboot_packet_current_byte_offset =
1046 1.1 fredette htonl(ntohl(nd_packet->ndboot_packet_current_byte_offset)
1047 1.1 fredette + ntohl(nd_packet->ndboot_packet_current_byte_count));
1048 1.1 fredette }
1049 1.1 fredette }
1050 1.1 fredette /* NOTREACHED */
1051 1.1 fredette }
1052 1.1 fredette /* the raw Ethernet access code: */
1053 1.2 fredette #include "config/ndbootd-bpf.c"
1054