1 /* $NetBSD: installboot.c,v 1.9 2024/09/15 03:56:57 tsutsui Exp $ */
2
3 /*-
4 * Copyright (c) 2005 Izumi Tsutsui. All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 */
26
27 #if HAVE_NBTOOL_CONFIG_H
28 #include "nbtool_config.h"
29 #endif
30
31 #include <err.h>
32 #include <fcntl.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <unistd.h>
37 #include <sys/stat.h>
38 #include <sys/types.h>
39
40 #include "installboot.h"
41
42 #define BSIZE 512
43 #define MAX_SB_SIZE (64 * 1024) /* XXX */
44 #define roundup(x, y) ((((x)+((y)-1))/(y))*(y))
45
46 static void usage(void);
47
48 static ib_params installboot_params;
49
50 static struct ib_fs cd9660_fstype = {
51 .name = "cd9660",
52 .match = cd9660_match,
53 .findstage2 = cd9660_findstage2
54 };
55
56 int
main(int argc,char ** argv)57 main(int argc, char **argv)
58 {
59 ib_params *params;
60 uint8_t *bb;
61 struct apple_part_map_entry pme;
62 size_t bbi;
63 struct shared_bbinfo *bbinfop;
64 off_t partoff;
65 uint32_t nblk, maxblk, blk_i;
66 int rv;
67 ib_block *blocks;
68 uint64_t block;
69
70 setprogname(argv[0]);
71 params = &installboot_params;
72 memset(params, 0, sizeof(*params));
73 params->fsfd = -1;
74 params->s1fd = -1;
75
76 if (argc != 4)
77 usage();
78
79 params->filesystem = argv[1];
80 params->fstype = &cd9660_fstype;
81
82 if ((params->fsfd = open(params->filesystem, O_RDWR, 0600)) == -1)
83 err(1, "Opening file system `%s' read", params->filesystem);
84 if (fstat(params->fsfd, ¶ms->fsstat) == -1)
85 err(1, "Examining file system `%s'", params->filesystem);
86 if (!params->fstype->match(params))
87 errx(1, "File system `%s' is not of type %s",
88 params->filesystem, params->fstype->name);
89
90 #ifdef DEBUG
91 printf("file system: %s, %ld bytes\n",
92 params->filesystem, (long)params->fsstat.st_size);
93 #endif
94
95 /*
96 * Find space for primary boot from the second (NetBSD_BootBlock)
97 * partition.
98 */
99 if (pread(params->fsfd, &pme, sizeof pme, BSIZE * 2) != sizeof(pme))
100 err(1, "read pme from file system `%s'", params->filesystem);
101
102 if (strcmp((char *)pme.pmPartName, "NetBSD_BootBlock"))
103 err(1, "invalid partition map in file system `%s'",
104 params->filesystem);
105
106 /* pmPyPartStart is written by mkisofs */
107 partoff = BSIZE * be32toh(pme.pmPyPartStart);
108
109 #ifdef DEBUG
110 printf("NetBSD partition offset = %ld\n", (long)partoff);
111 #endif
112
113 params->stage1 = argv[2];
114
115 if ((params->s1fd = open(params->stage1, O_RDONLY, 0600)) == -1)
116 err(1, "Opening primary bootstrap `%s'", params->stage1);
117 if (fstat(params->s1fd, ¶ms->s1stat) == -1)
118 err(1, "Examining primary bootstrap `%s'", params->stage1);
119 if (!S_ISREG(params->s1stat.st_mode))
120 err(1, "`%s' must be a regular file", params->stage1);
121
122 if (params->s1stat.st_size > MACPPC_BOOT_BLOCK_MAX_SIZE)
123 err(1, "primary bootrap `%s' too large (%ld bytes)",
124 params->stage1, (long)params->s1stat.st_size);
125
126 #ifdef DEBUG
127 printf("primary boot: %s, %ld bytes\n",
128 params->stage1, (long)params->s1stat.st_size);
129 #endif
130
131 params->stage2 = argv[3];
132
133 bb = calloc(1, MACPPC_BOOT_BLOCK_MAX_SIZE);
134 if (bb == NULL)
135 err(1, "Allocating %ul bytes for bbinfo",
136 MACPPC_BOOT_BLOCK_MAX_SIZE);
137
138 rv = read(params->s1fd, bb, params->s1stat.st_size);
139
140 if (rv == -1)
141 err(1, "Reading `%s'", params->stage1);
142
143 if (memcmp(bb + 1, "ELF", strlen("ELF")) == 0) {
144 warnx("`%s' is an ELF executable; need raw binary",
145 params->stage1);
146 }
147
148 /* look for the bbinfo structure */
149 for (bbi = 0; bbi < MACPPC_BOOT_BLOCK_MAX_SIZE;
150 bbi += sizeof(uint32_t)) {
151 bbinfop = (void *)(bb + bbi);
152 if (memcmp(bbinfop->bbi_magic, MACPPC_BBINFO_MAGIC,
153 sizeof(bbinfop->bbi_magic)) == 0) {
154 #ifdef DEBUG
155 printf("magic found: %s\n", bbinfop->bbi_magic);
156 #endif
157 break;
158 }
159 }
160 if (bbi >= MACPPC_BOOT_BLOCK_MAX_SIZE)
161 err(1, "bbinfo structure not found in `%s'", params->stage1);
162
163 maxblk = be32toh(bbinfop->bbi_block_count);
164 if (maxblk == 0 ||
165 maxblk > (MACPPC_BOOT_BLOCK_MAX_SIZE / sizeof(uint32_t)))
166 err(1, "bbinfo structure in `%s' has preposterous size `%u'",
167 params->stage1, maxblk);
168
169 blocks = malloc(sizeof(*blocks) * maxblk);
170 if (blocks == NULL) {
171 err(1, "Allocating %lu bytes for blocks",
172 (unsigned long)sizeof(*blocks) * maxblk);
173 }
174
175 if (S_ISREG(params->fsstat.st_mode)) {
176 if (fsync(params->fsfd) == -1)
177 err(1, "Synchronising file system `%s'",
178 params->filesystem);
179 }
180
181 nblk = maxblk;
182 if (!params->fstype->findstage2(params, &nblk, blocks)) {
183 exit(1);
184 }
185
186 bbinfop->bbi_block_count = htobe32(nblk);
187 bbinfop->bbi_block_size = htobe32(blocks[0].blocksize);
188 for (blk_i = 0; blk_i < nblk; blk_i++) {
189 /* XXX bootxx assumes blocksize is 512 */
190 block = blocks[blk_i].block * (params->fstype->blocksize / 512);
191 bbinfop->bbi_block_table[blk_i] = htobe32(block);
192 if (blocks[blk_i].blocksize < blocks[0].blocksize &&
193 blk_i + 1 != nblk) {
194 warnx("Secondary bootstrap `%s' blocks do not have "
195 "a uniform size", params->stage2);
196 exit(1);
197 }
198 }
199
200 /* XXX no write option */
201
202 if (pwrite(params->fsfd, bb, MACPPC_BOOT_BLOCK_MAX_SIZE, partoff) !=
203 MACPPC_BOOT_BLOCK_MAX_SIZE)
204 err(1, "write bootblock");
205
206 if (S_ISREG(params->fsstat.st_mode)) {
207 if (fsync(params->fsfd) == -1)
208 err(1, "Synchronising file system `%s'",
209 params->filesystem);
210 }
211
212 free(bb);
213
214 if (close(params->fsfd) == -1)
215 err(1, "Closing file system `%s'", params->filesystem);
216 if (close(params->s1fd) == -1)
217 err(1, "Closing primary bootstrap `%s'", params->stage1);
218
219 return 0;
220 }
221
222 static void
usage(void)223 usage(void)
224 {
225 const char *prog;
226
227 prog = getprogname();
228 fprintf(stderr, "usage: %s hybrid-cd-image primary secondary\n", prog);
229 exit(1);
230 }
231