1/*	$OpenBSD: setnetbootinfo.c,v 1.5 2021/10/24 21:24:22 deraadt Exp $	*/
2/*	$NetBSD: setnetbootinfo.c,v 1.5 1997/04/06 08:41:37 cgd Exp $	*/
3
4/*
5 * Copyright (c) 1997 Christopher G. Demetriou
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 * 3. All advertising materials mentioning features or use of this software
17 *    must display the following acknowledgement:
18 *      This product includes software developed by Christopher G. Demetriou
19 *	for the NetBSD Project.
20 * 4. The name of the author may not be used to endorse or promote products
21 *    derived from this software without specific prior written permission
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
24 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
25 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
26 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
27 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
28 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
29 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
30 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
32 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
33 */
34
35#include <err.h>
36#include <stdio.h>
37#include <stdlib.h>
38#include <string.h>
39#include <unistd.h>
40#include <sys/fcntl.h>
41#include <sys/stat.h>
42#include <sys/socket.h>						/* XXX */
43#include <net/if.h>						/* XXX */
44#include <netinet/in.h>
45#include <netinet/if_ether.h>
46
47#include "bbinfo.h"
48
49int	verbose, force, unset;
50char	*netboot, *outfile, *addr, *host;
51
52char	*outfilename;
53
54struct ether_addr *ether_addr, _ether_addr;
55
56static void
57usage()
58{
59
60	(void)fprintf(stderr, "usage:\n");
61	(void)fprintf(stderr, "\tsetnetboot [-v] [-f] [-o outfile] \\\n");
62	(void)fprintf(stderr, "\t    [-a ether-address | -h ether-host] infile\n");
63	(void)fprintf(stderr, "\tsetnetboot [-v] -u -o outfile infile\n");
64	exit(1);
65}
66
67int
68main(argc, argv)
69	int argc;
70	char *argv[];
71{
72	struct netbbinfo *netbbinfop;
73	struct stat sb;
74	u_int64_t *qp, csum;
75	char *netbb;
76	int c, fd, i;
77
78	while ((c = getopt(argc, argv, "a:fh:o:uv")) != -1) {
79		switch (c) {
80		case 'a':
81			/* use the argument as an ethernet address */
82			addr = optarg;
83			break;
84		case 'f':
85			/* set force flag in network boot block */
86			force = 1;
87			break;
88		case 'h':
89			/* use the argument as a host to find in /etc/ethers */
90			host = optarg;
91			break;
92		case 'o':
93			/* use the argument as the output file name */
94			outfile = optarg;
95			break;
96		case 'u':
97			/* remove configuration information */
98			unset = 1;
99			break;
100		case 'v':
101			/* Chat */
102			verbose = 1;
103			break;
104		default:
105			usage();
106		}
107	}
108
109	if ((argc - optind) != 1)
110		usage();
111	netboot = argv[optind];
112
113	if (unset && (force || host != NULL || addr != NULL))
114		errx(1, "-u can't be used with -f, -h, or -a");
115
116	if (unset) {
117		if (force || host != NULL || addr != NULL)
118			errx(1, "-u can't be used with -f, -h, or -a");
119		if (outfile == NULL)
120			errx(1, "-u cannot be used without -o");
121	} else {
122		if ((host == NULL && addr == NULL) ||
123		    (host != NULL && addr != NULL))
124			usage();
125
126		if (host != NULL) {
127			if (ether_hostton(host, &_ether_addr) == -1)
128				errx(1, "ethernet address couldn't be found for \"%s\"",
129				    host);
130			ether_addr = &_ether_addr;
131		} else { /* addr != NULL */
132			ether_addr = ether_aton(addr);
133			if (ether_addr == NULL)
134				errx(1, "ethernet address \"%s\" is invalid",
135				    addr);
136		}
137	}
138
139	if (outfile != NULL)
140		outfilename = outfile;
141	else {
142		/* name + 12 for enet addr + '.' before enet addr + NUL */
143		size_t len = strlen(netboot) + 14;
144
145		outfilename = malloc(len);
146		if (outfilename == NULL)
147			err(1, "malloc of output file name failed");
148		snprintf(outfilename, len,
149		    "%s.%02x%02x%02x%02x%02x%02x", netboot,
150		    ether_addr->ether_addr_octet[0],
151		    ether_addr->ether_addr_octet[1],
152		    ether_addr->ether_addr_octet[2],
153		    ether_addr->ether_addr_octet[3],
154		    ether_addr->ether_addr_octet[4],
155		    ether_addr->ether_addr_octet[5]);
156	}
157
158	if (verbose) {
159		printf("netboot: %s\n", netboot);
160		if (unset)
161			printf("unsetting configuration\n");
162		else
163			printf("ethernet address: %s (%s), force = %d\n",
164			    ether_ntoa(ether_addr), host ? host : addr, force);
165		printf("output netboot: %s\n", outfilename);
166	}
167
168
169	if (verbose)
170		printf("opening %s...\n", netboot);
171	if ((fd = open(netboot, O_RDONLY)) == -1)
172		err(1, "open: %s", netboot);
173	if (fstat(fd, &sb) == -1)
174		err(1, "fstat: %s", netboot);
175	if (!S_ISREG(sb.st_mode))
176		errx(1, "%s must be a regular file", netboot);
177
178	if (verbose)
179		printf("reading %s...\n", netboot);
180	netbb = malloc(sb.st_size);
181	if (netbb == NULL)
182		err(1, "malloc of %lu for %s failed",
183		    (unsigned long)sb.st_size, netboot);
184	if (read(fd, netbb, sb.st_size) != sb.st_size)
185		err(1, "read of %lu from %s failed",
186		    (unsigned long)sb.st_size, netboot);
187
188	if (verbose)
189		printf("closing %s...\n", netboot);
190	close(fd);
191
192	if (verbose)
193		printf("looking for netbbinfo...\n");
194	netbbinfop = NULL;
195	for (qp = (u_int64_t *)netbb; qp < (u_int64_t *)(netbb + sb.st_size);
196	    qp++) {
197		if (((struct netbbinfo *)qp)->magic1 == 0xfeedbabedeadbeef &&
198		    ((struct netbbinfo *)qp)->magic2 == 0xfeedbeefdeadbabe) {
199			netbbinfop = (struct netbbinfo *)qp;
200			break;
201		}
202	}
203	if (netbbinfop == NULL)
204		errx(1, "netboot information structure not found in %s",
205		    netboot);
206	if (verbose)
207		printf("found netbbinfo structure at offset 0x%lx.\n",
208		    (unsigned long)((char *)netbbinfop - netbb));
209
210	if (verbose)
211		printf("setting netbbinfo structure...\n");
212	bzero(netbbinfop, sizeof *netbbinfop);
213	netbbinfop->magic1 = 0xfeedbabedeadbeef;
214	netbbinfop->magic2 = 0xfeedbeefdeadbabe;
215	netbbinfop->set = unset ? 0 : 1;
216	if (netbbinfop->set) {
217		for (i = 0; i < 6; i++)
218			netbbinfop->ether_addr[i] =
219			    ether_addr->ether_addr_octet[i];
220		netbbinfop->force = force;
221	}
222	netbbinfop->cksum = 0;
223
224	if (verbose)
225		printf("setting netbbinfo checksum...\n");
226	csum = 0;
227	for (i = 0, qp = (u_int64_t *)netbbinfop;
228	    i < (sizeof *netbbinfop / sizeof (u_int64_t)); i++, qp++)
229		csum += *qp;
230	netbbinfop->cksum = -csum;
231
232	if (verbose)
233		printf("opening %s...\n", outfilename);
234	if ((fd = open(outfilename, O_WRONLY | O_CREAT, 0666)) == -1)
235		err(1, "open: %s", outfilename);
236
237	if (verbose)
238		printf("writing %s...\n", outfilename);
239	if (write(fd, netbb, sb.st_size) != sb.st_size)
240		err(1, "write of %lu to %s failed",
241		    (unsigned long)sb.st_size, outfilename);
242
243	if (verbose)
244		printf("closing %s...\n", outfilename);
245	close(fd);
246
247	free(netbb);
248	if (outfile == NULL)
249		free(outfilename);
250
251	exit (0);
252}
253