1 /* $OpenBSD: setnetbootinfo.c,v 1.4 2014/07/12 19:01:49 tedu 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 49 int verbose, force, unset; 50 char *netboot, *outfile, *addr, *host; 51 52 char *outfilename; 53 54 struct ether_addr *ether_addr, _ether_addr; 55 56 static void 57 usage() 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 67 int 68 main(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, 0)) == -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