1 /* $NetBSD: hp300.c,v 1.15 2013/06/14 03:54:43 msaitoh Exp $ */
2
3 /*-
4 * Copyright (c) 2003 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by David Laight.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 #if HAVE_NBTOOL_CONFIG_H
33 #include "nbtool_config.h"
34 #endif
35
36 #include <sys/cdefs.h>
37 #if !defined(__lint)
38 __RCSID("$NetBSD: hp300.c,v 1.15 2013/06/14 03:54:43 msaitoh Exp $");
39 #endif /* !__lint */
40
41 /* We need the target disklabel.h, not the hosts one..... */
42 #ifdef HAVE_NBTOOL_CONFIG_H
43 #include "nbtool_config.h"
44 #include <nbinclude/hp300/disklabel.h>
45 #include <nbinclude/sys/disklabel.h>
46 #else
47 #include <sys/disklabel.h>
48 #endif
49 #include <sys/fcntl.h>
50 #include <sys/ioctl.h>
51 #include <sys/mman.h>
52 #include <sys/param.h>
53
54 #include <assert.h>
55 #include <err.h>
56 #include <md5.h>
57 #include <stddef.h>
58 #include <stdio.h>
59 #include <stdlib.h>
60 #include <string.h>
61 #include <unistd.h>
62
63 #include "installboot.h"
64
65 static int hp300_setboot(ib_params *);
66
67 struct ib_mach ib_mach_hp300 =
68 { "hp300", hp300_setboot, no_clearboot, no_editboot, IB_APPEND };
69
70 static int
hp300_setboot(ib_params * params)71 hp300_setboot(ib_params *params)
72 {
73 int retval;
74 uint8_t *bootstrap;
75 ssize_t rv;
76 struct partition *boot;
77 struct hp300_lifdir *lifdir;
78 int offset;
79 int i;
80 unsigned int secsize = HP300_SECTSIZE;
81 uint64_t boot_size, boot_offset;
82 struct disklabel *label;
83
84 assert(params != NULL);
85 assert(params->fsfd != -1);
86 assert(params->filesystem != NULL);
87 assert(params->s1fd != -1);
88 assert(params->stage1 != NULL);
89
90 retval = 0;
91 bootstrap = MAP_FAILED;
92
93 label = malloc(params->sectorsize);
94 if (label == NULL) {
95 warn("Failed to allocate memory for disklabel");
96 goto done;
97 }
98
99 if (params->flags & IB_APPEND) {
100 if (!S_ISREG(params->fsstat.st_mode)) {
101 warnx(
102 "`%s' must be a regular file to append a bootstrap",
103 params->filesystem);
104 goto done;
105 }
106 boot_offset = roundup(params->fsstat.st_size, HP300_SECTSIZE);
107 } else {
108 /*
109 * The bootstrap can be well over 8k, and must go into a BOOT
110 * partition. Read NetBSD label to locate BOOT partition.
111 */
112 if (pread(params->fsfd, label, params->sectorsize,
113 LABELSECTOR * params->sectorsize)
114 != (ssize_t)params->sectorsize) {
115 warn("reading disklabel");
116 goto done;
117 }
118 /* And a quick validation - must be a big-endian label */
119 secsize = be32toh(label->d_secsize);
120 if (label->d_magic != htobe32(DISKMAGIC) ||
121 label->d_magic2 != htobe32(DISKMAGIC) ||
122 secsize == 0 || secsize & (secsize - 1) ||
123 be16toh(label->d_npartitions) > MAXMAXPARTITIONS) {
124 warnx("Invalid disklabel in %s", params->filesystem);
125 goto done;
126 }
127
128 i = be16toh(label->d_npartitions);
129 for (boot = label->d_partitions; ; boot++) {
130 if (--i < 0) {
131 warnx("No BOOT partition");
132 goto done;
133 }
134 if (boot->p_fstype == FS_BOOT)
135 break;
136 }
137 boot_size = be32toh(boot->p_size) * (uint64_t)secsize;
138 boot_offset = be32toh(boot->p_offset) * (uint64_t)secsize;
139
140 /*
141 * We put the entire LIF file into the BOOT partition even when
142 * it doesn't start at the beginning of the disk.
143 *
144 * Maybe we ought to be able to take a binary file and add
145 * it to the LIF filesystem.
146 */
147 if (boot_size < (uint64_t)params->s1stat.st_size) {
148 warn("BOOT partition too small (%llu < %llu)",
149 (unsigned long long)boot_size,
150 (unsigned long long)params->s1stat.st_size);
151 goto done;
152 }
153 }
154
155 bootstrap = mmap(NULL, params->s1stat.st_size, PROT_READ | PROT_WRITE,
156 MAP_PRIVATE, params->s1fd, 0);
157 if (bootstrap == MAP_FAILED) {
158 warn("mmaping `%s'", params->stage1);
159 goto done;
160 }
161
162 /* Relocate files, sanity check LIF directory on the way */
163 lifdir = (void *)(bootstrap + HP300_SECTSIZE * 2);
164 for (i = 0; i < 8; lifdir++, i++) {
165 int32_t addr = be32toh(lifdir->dir_addr);
166 int32_t limit = (params->s1stat.st_size - 1) / HP300_SECTSIZE + 1;
167 int32_t end = addr + be32toh(lifdir->dir_length);
168 if (end > limit) {
169 warnx("LIF entry %d larger (%d %d) than LIF file",
170 i, end, limit);
171 goto done;
172 }
173 if (addr != 0 && boot_offset != 0)
174 lifdir->dir_addr = htobe32(addr + boot_offset
175 / HP300_SECTSIZE);
176 }
177
178 if (params->flags & IB_NOWRITE) {
179 retval = 1;
180 goto done;
181 }
182
183 /* Write LIF volume header and directory to sectors 0 and 1 */
184 rv = pwrite(params->fsfd, bootstrap, 1024, 0);
185 if (rv != 1024) {
186 if (rv == -1)
187 warn("Writing `%s'", params->filesystem);
188 else
189 warnx("Writing `%s': short write", params->filesystem);
190 goto done;
191 }
192
193 /* Write files to BOOT partition */
194 offset = boot_offset <= HP300_SECTSIZE * 16 ? HP300_SECTSIZE * 16 : 0;
195 i = roundup(params->s1stat.st_size, secsize) - offset;
196 rv = pwrite(params->fsfd, bootstrap + offset, i, boot_offset + offset);
197 if (rv != i) {
198 if (rv == -1)
199 warn("Writing boot filesystem of `%s'",
200 params->filesystem);
201 else
202 warnx("Writing boot filesystem of `%s': short write",
203 params->filesystem);
204 goto done;
205 }
206
207 retval = 1;
208
209 done:
210 if (label != NULL)
211 free(label);
212 if (bootstrap != MAP_FAILED)
213 munmap(bootstrap, params->s1stat.st_size);
214 return retval;
215 }
216