subr_disk_mbr.c revision 1.57.12.1 1 1.57.12.1 martin /* $NetBSD: subr_disk_mbr.c,v 1.57.12.1 2025/02/22 11:47:15 martin Exp $ */
2 1.1 dsl
3 1.1 dsl /*
4 1.1 dsl * Copyright (c) 1982, 1986, 1988 Regents of the University of California.
5 1.1 dsl * All rights reserved.
6 1.1 dsl *
7 1.1 dsl * Redistribution and use in source and binary forms, with or without
8 1.1 dsl * modification, are permitted provided that the following conditions
9 1.1 dsl * are met:
10 1.1 dsl * 1. Redistributions of source code must retain the above copyright
11 1.1 dsl * notice, this list of conditions and the following disclaimer.
12 1.1 dsl * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 dsl * notice, this list of conditions and the following disclaimer in the
14 1.1 dsl * documentation and/or other materials provided with the distribution.
15 1.2 agc * 3. Neither the name of the University nor the names of its contributors
16 1.1 dsl * may be used to endorse or promote products derived from this software
17 1.1 dsl * without specific prior written permission.
18 1.1 dsl *
19 1.1 dsl * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 1.1 dsl * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 1.1 dsl * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 1.1 dsl * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 1.1 dsl * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 1.1 dsl * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 1.1 dsl * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 1.1 dsl * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 1.1 dsl * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 1.1 dsl * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 1.1 dsl * SUCH DAMAGE.
30 1.1 dsl *
31 1.1 dsl * @(#)ufs_disksubr.c 7.16 (Berkeley) 5/4/91
32 1.1 dsl */
33 1.1 dsl
34 1.1 dsl /*
35 1.1 dsl * Code to find a NetBSD label on a disk that contains an i386 style MBR.
36 1.1 dsl * The first NetBSD label found in the 2nd sector of a NetBSD partition
37 1.3 dsl * is used.
38 1.3 dsl * If we don't find a label searching the MBR, we look at the start of the
39 1.3 dsl * disk, if that fails then a label is faked up from the MBR.
40 1.1 dsl *
41 1.32 reinoud * If there isn't a disklabel or anything in the MBR then the disc is searched
42 1.32 reinoud * for ecma-167/iso9660/udf style partition indicators.
43 1.32 reinoud * Useful for media or files that contain single filesystems (etc).
44 1.1 dsl *
45 1.1 dsl * This code will read host endian netbsd labels from little endian MBR.
46 1.1 dsl *
47 1.1 dsl * Based on the i386 disksubr.c
48 1.1 dsl *
49 1.3 dsl * Since the mbr only has 32bit fields for sector addresses, we do the same.
50 1.3 dsl *
51 1.1 dsl * XXX There are potential problems writing labels to disks where there
52 1.1 dsl * is only space for 8 netbsd partitions but this code has been compiled
53 1.1 dsl * with MAXPARTITIONS=16.
54 1.1 dsl */
55 1.1 dsl
56 1.1 dsl #include <sys/cdefs.h>
57 1.57.12.1 martin __KERNEL_RCSID(0, "$NetBSD: subr_disk_mbr.c,v 1.57.12.1 2025/02/22 11:47:15 martin Exp $");
58 1.57 mrg
59 1.57 mrg #ifdef _KERNEL_OPT
60 1.57 mrg #include "opt_mbr.h"
61 1.57 mrg #include "opt_disklabel.h"
62 1.57 mrg #endif /* _KERNEL_OPT */
63 1.1 dsl
64 1.1 dsl #include <sys/param.h>
65 1.1 dsl #include <sys/systm.h>
66 1.1 dsl #include <sys/buf.h>
67 1.27 dyoung #include <sys/bootblock.h>
68 1.1 dsl #include <sys/disklabel.h>
69 1.1 dsl #include <sys/disk.h>
70 1.1 dsl #include <sys/syslog.h>
71 1.32 reinoud #include <sys/vnode.h>
72 1.32 reinoud #include <sys/fcntl.h>
73 1.32 reinoud #include <sys/conf.h>
74 1.32 reinoud #include <sys/cdio.h>
75 1.37 pooka #include <sys/dkbad.h>
76 1.32 reinoud #include <fs/udf/ecma167-udf.h>
77 1.32 reinoud
78 1.32 reinoud #include <sys/kauth.h>
79 1.1 dsl
80 1.13 dsl typedef struct mbr_partition mbr_partition_t;
81 1.13 dsl
82 1.17 dsl /*
83 1.38 pooka * We allocate a buffer 3 sectors large, and look in all....
84 1.17 dsl * That means we find labels written by other ports with different offsets.
85 1.17 dsl * LABELSECTOR and LABELOFFSET are only used if the disk doesn't have a label.
86 1.17 dsl */
87 1.38 pooka #define SCANBLOCKS 3
88 1.38 pooka #define DISKLABEL_SIZE 404
89 1.38 pooka #if LABELSECTOR*DEV_BSIZE + LABELOFFSET > SCANBLOCKS*DEV_BSIZE - DISKLABEL_SIZE
90 1.41 pooka #if _MACHINE != ews4800mips /* XXX: fail silently, ews4800mips LABELSECTOR */
91 1.17 dsl #error Invalid LABELSECTOR or LABELOFFSET
92 1.17 dsl #endif
93 1.40 tsutsui #endif
94 1.17 dsl
95 1.1 dsl #define MBR_LABELSECTOR 1
96 1.1 dsl
97 1.1 dsl #define SCAN_CONTINUE 0
98 1.1 dsl #define SCAN_FOUND 1
99 1.1 dsl #define SCAN_ERROR 2
100 1.1 dsl
101 1.1 dsl typedef struct mbr_args {
102 1.1 dsl struct disklabel *lp;
103 1.1 dsl void (*strat)(struct buf *);
104 1.1 dsl struct buf *bp;
105 1.1 dsl const char *msg;
106 1.1 dsl int error;
107 1.3 dsl int written; /* number of times we wrote label */
108 1.17 dsl int found_mbr; /* set if disk has a valid mbr */
109 1.3 dsl uint label_sector; /* where we found the label */
110 1.17 dsl int action;
111 1.28 dyoung uint32_t secperunit;
112 1.1 dsl #define READ_LABEL 1
113 1.3 dsl #define UPDATE_LABEL 2
114 1.3 dsl #define WRITE_LABEL 3
115 1.17 dsl } mbr_args_t;
116 1.17 dsl
117 1.17 dsl static int validate_label(mbr_args_t *, uint);
118 1.13 dsl static int look_netbsd_part(mbr_args_t *, mbr_partition_t *, int, uint);
119 1.13 dsl static int write_netbsd_label(mbr_args_t *, mbr_partition_t *, int, uint);
120 1.1 dsl
121 1.1 dsl static int
122 1.17 dsl read_sector(mbr_args_t *a, uint sector, int count)
123 1.1 dsl {
124 1.1 dsl int error;
125 1.1 dsl
126 1.28 dyoung error = disk_read_sectors(a->strat, a->lp, a->bp, sector, count);
127 1.1 dsl if (error != 0)
128 1.1 dsl a->error = error;
129 1.1 dsl return error;
130 1.1 dsl }
131 1.1 dsl
132 1.6 perry /*
133 1.1 dsl * Scan MBR for partitions, call 'action' routine for each.
134 1.1 dsl */
135 1.1 dsl
136 1.1 dsl static int
137 1.13 dsl scan_mbr(mbr_args_t *a, int (*actn)(mbr_args_t *, mbr_partition_t *, int, uint))
138 1.1 dsl {
139 1.13 dsl mbr_partition_t ptns[MBR_PART_COUNT];
140 1.13 dsl mbr_partition_t *dp;
141 1.4 lukem struct mbr_sector *mbr;
142 1.1 dsl uint ext_base, this_ext, next_ext;
143 1.1 dsl int rval;
144 1.1 dsl int i;
145 1.18 christos int j;
146 1.1 dsl #ifdef COMPAT_386BSD_MBRPART
147 1.1 dsl int dp_386bsd = -1;
148 1.18 christos int ap_386bsd = -1;
149 1.1 dsl #endif
150 1.1 dsl
151 1.1 dsl ext_base = 0;
152 1.1 dsl this_ext = 0;
153 1.1 dsl for (;;) {
154 1.17 dsl if (read_sector(a, this_ext, 1)) {
155 1.1 dsl a->msg = "dos partition I/O error";
156 1.1 dsl return SCAN_ERROR;
157 1.1 dsl }
158 1.1 dsl
159 1.1 dsl /* Note: Magic number is little-endian. */
160 1.1 dsl mbr = (void *)a->bp->b_data;
161 1.4 lukem if (mbr->mbr_magic != htole16(MBR_MAGIC))
162 1.3 dsl return SCAN_CONTINUE;
163 1.1 dsl
164 1.46 matt /*
165 1.46 matt * If this is a protective MBR, bail now.
166 1.46 matt */
167 1.46 matt if (mbr->mbr_parts[0].mbrp_type == MBR_PTYPE_PMBR
168 1.46 matt && mbr->mbr_parts[1].mbrp_type == MBR_PTYPE_UNUSED
169 1.46 matt && mbr->mbr_parts[2].mbrp_type == MBR_PTYPE_UNUSED
170 1.46 matt && mbr->mbr_parts[3].mbrp_type == MBR_PTYPE_UNUSED)
171 1.46 matt return SCAN_CONTINUE;
172 1.46 matt
173 1.1 dsl /* Copy data out of buffer so action can use bp */
174 1.1 dsl memcpy(ptns, &mbr->mbr_parts, sizeof ptns);
175 1.1 dsl
176 1.14 christos /* Look for drivers and skip them */
177 1.17 dsl if (ext_base == 0 && ptns[0].mbrp_type == MBR_PTYPE_DM6_DDO) {
178 1.16 jmmv /* We've found a DM6 DDO partition type (used by
179 1.16 jmmv * the Ontrack Disk Manager drivers).
180 1.16 jmmv *
181 1.16 jmmv * Ensure that there are no other partitions in the
182 1.16 jmmv * MBR and jump to the real partition table (stored
183 1.16 jmmv * in the first sector of the second track). */
184 1.25 thorpej bool ok = true;
185 1.14 christos
186 1.14 christos for (i = 1; i < MBR_PART_COUNT; i++)
187 1.14 christos if (ptns[i].mbrp_type != MBR_PTYPE_UNUSED)
188 1.25 thorpej ok = false;
189 1.14 christos
190 1.14 christos if (ok) {
191 1.15 christos this_ext = le32toh(a->lp->d_secpercyl /
192 1.15 christos a->lp->d_ntracks);
193 1.14 christos continue;
194 1.14 christos }
195 1.14 christos }
196 1.14 christos
197 1.1 dsl /* look for NetBSD partition */
198 1.1 dsl next_ext = 0;
199 1.1 dsl dp = ptns;
200 1.18 christos j = 0;
201 1.4 lukem for (i = 0; i < MBR_PART_COUNT; i++, dp++) {
202 1.18 christos if (dp->mbrp_type == MBR_PTYPE_UNUSED)
203 1.1 dsl continue;
204 1.17 dsl /* Check end of partition is inside disk limits */
205 1.17 dsl if ((uint64_t)ext_base + le32toh(dp->mbrp_start) +
206 1.17 dsl le32toh(dp->mbrp_size) > a->lp->d_secperunit) {
207 1.17 dsl /* This mbr doesn't look good.... */
208 1.17 dsl a->msg = "mbr partition exceeds disk size";
209 1.17 dsl /* ...but don't report this as an error (yet) */
210 1.17 dsl return SCAN_CONTINUE;
211 1.17 dsl }
212 1.17 dsl a->found_mbr = 1;
213 1.4 lukem if (MBR_IS_EXTENDED(dp->mbrp_type)) {
214 1.1 dsl next_ext = le32toh(dp->mbrp_start);
215 1.1 dsl continue;
216 1.1 dsl }
217 1.1 dsl #ifdef COMPAT_386BSD_MBRPART
218 1.4 lukem if (dp->mbrp_type == MBR_PTYPE_386BSD) {
219 1.1 dsl /*
220 1.1 dsl * If more than one matches, take last,
221 1.1 dsl * as NetBSD install tool does.
222 1.1 dsl */
223 1.18 christos if (this_ext == 0) {
224 1.1 dsl dp_386bsd = i;
225 1.18 christos ap_386bsd = j;
226 1.18 christos }
227 1.1 dsl continue;
228 1.1 dsl }
229 1.1 dsl #endif
230 1.18 christos rval = (*actn)(a, dp, j, this_ext);
231 1.1 dsl if (rval != SCAN_CONTINUE)
232 1.1 dsl return rval;
233 1.18 christos j++;
234 1.1 dsl }
235 1.1 dsl if (next_ext == 0)
236 1.1 dsl break;
237 1.1 dsl if (ext_base == 0) {
238 1.1 dsl ext_base = next_ext;
239 1.1 dsl next_ext = 0;
240 1.1 dsl }
241 1.1 dsl next_ext += ext_base;
242 1.1 dsl if (next_ext <= this_ext)
243 1.1 dsl break;
244 1.1 dsl this_ext = next_ext;
245 1.1 dsl }
246 1.1 dsl #ifdef COMPAT_386BSD_MBRPART
247 1.1 dsl if (this_ext == 0 && dp_386bsd != -1)
248 1.18 christos return (*actn)(a, &ptns[dp_386bsd], ap_386bsd, 0);
249 1.1 dsl #endif
250 1.1 dsl return SCAN_CONTINUE;
251 1.1 dsl }
252 1.1 dsl
253 1.32 reinoud
254 1.32 reinoud static void
255 1.32 reinoud scan_iso_vrs_session(mbr_args_t *a, uint32_t first_sector,
256 1.32 reinoud int *is_iso9660, int *is_udf)
257 1.32 reinoud {
258 1.32 reinoud struct vrs_desc *vrsd;
259 1.32 reinoud uint64_t vrs;
260 1.32 reinoud int sector_size;
261 1.57.12.1 martin int inc;
262 1.32 reinoud
263 1.32 reinoud sector_size = a->lp->d_secsize;
264 1.33 reinoud inc = MAX(1, 2048 / sector_size);
265 1.32 reinoud
266 1.32 reinoud /* by definition */
267 1.32 reinoud vrs = ((32*1024 + sector_size - 1) / sector_size)
268 1.32 reinoud + first_sector;
269 1.32 reinoud
270 1.32 reinoud /* read first vrs sector */
271 1.57.12.1 martin if (read_sector(a, vrs, 1))
272 1.32 reinoud return;
273 1.32 reinoud
274 1.32 reinoud /* skip all CD001 records */
275 1.32 reinoud vrsd = a->bp->b_data;
276 1.33 reinoud /* printf("vrsd->identifier = `%s`\n", vrsd->identifier); */
277 1.32 reinoud while (memcmp(vrsd->identifier, "CD001", 5) == 0) {
278 1.32 reinoud /* for sure */
279 1.32 reinoud *is_iso9660 = first_sector;
280 1.32 reinoud
281 1.33 reinoud vrs += inc;
282 1.57.12.1 martin if (read_sector(a, vrs, 1))
283 1.32 reinoud return;
284 1.32 reinoud }
285 1.32 reinoud
286 1.32 reinoud /* search for BEA01 */
287 1.32 reinoud vrsd = a->bp->b_data;
288 1.32 reinoud /* printf("vrsd->identifier = `%s`\n", vrsd->identifier); */
289 1.32 reinoud if (memcmp(vrsd->identifier, "BEA01", 5))
290 1.32 reinoud return;
291 1.32 reinoud
292 1.32 reinoud /* read successor */
293 1.33 reinoud vrs += inc;
294 1.57.12.1 martin if (read_sector(a, vrs, 1))
295 1.32 reinoud return;
296 1.32 reinoud
297 1.32 reinoud /* check for NSR[23] */
298 1.32 reinoud vrsd = a->bp->b_data;
299 1.33 reinoud /* printf("vrsd->identifier = `%s`\n", vrsd->identifier); */
300 1.32 reinoud if (memcmp(vrsd->identifier, "NSR0", 4))
301 1.32 reinoud return;
302 1.32 reinoud
303 1.32 reinoud *is_udf = first_sector;
304 1.32 reinoud }
305 1.32 reinoud
306 1.32 reinoud
307 1.32 reinoud /*
308 1.32 reinoud * Scan for ISO Volume Recognition Sequences
309 1.32 reinoud */
310 1.32 reinoud
311 1.32 reinoud static int
312 1.32 reinoud scan_iso_vrs(mbr_args_t *a)
313 1.32 reinoud {
314 1.32 reinoud struct mmc_discinfo di;
315 1.32 reinoud struct mmc_trackinfo ti;
316 1.32 reinoud dev_t dev;
317 1.32 reinoud uint64_t sector;
318 1.32 reinoud int is_iso9660, is_udf;
319 1.32 reinoud int tracknr, sessionnr;
320 1.32 reinoud int new_session, error;
321 1.32 reinoud
322 1.32 reinoud is_iso9660 = is_udf = -1;
323 1.32 reinoud
324 1.32 reinoud /* parse all sessions of disc if we're on a SCSI MMC device */
325 1.32 reinoud if (a->lp->d_flags & D_SCSI_MMC) {
326 1.32 reinoud /* get disc info */
327 1.32 reinoud dev = a->bp->b_dev;
328 1.32 reinoud error = bdev_ioctl(dev, MMCGETDISCINFO, &di, FKIOCTL, curlwp);
329 1.32 reinoud if (error)
330 1.57.12.1 martin goto notmmc;
331 1.32 reinoud
332 1.32 reinoud /* go trough all (data) tracks */
333 1.32 reinoud sessionnr = -1;
334 1.32 reinoud for (tracknr = di.first_track;
335 1.32 reinoud tracknr <= di.first_track_last_session; tracknr++)
336 1.32 reinoud {
337 1.32 reinoud ti.tracknr = tracknr;
338 1.32 reinoud error = bdev_ioctl(dev, MMCGETTRACKINFO, &ti,
339 1.32 reinoud FKIOCTL, curlwp);
340 1.32 reinoud if (error)
341 1.57.12.1 martin goto notmmc;
342 1.32 reinoud new_session = (ti.sessionnr != sessionnr);
343 1.32 reinoud sessionnr = ti.sessionnr;
344 1.32 reinoud if (new_session) {
345 1.32 reinoud if (ti.flags & MMC_TRACKINFO_BLANK)
346 1.32 reinoud continue;
347 1.32 reinoud if (!(ti.flags & MMC_TRACKINFO_DATA))
348 1.32 reinoud continue;
349 1.32 reinoud sector = ti.track_start;
350 1.32 reinoud scan_iso_vrs_session(a, sector,
351 1.32 reinoud &is_iso9660, &is_udf);
352 1.32 reinoud }
353 1.32 reinoud }
354 1.32 reinoud } else {
355 1.57.12.1 martin notmmc:
356 1.32 reinoud /* try start of disc */
357 1.32 reinoud sector = 0;
358 1.32 reinoud scan_iso_vrs_session(a, sector, &is_iso9660, &is_udf);
359 1.32 reinoud }
360 1.32 reinoud
361 1.32 reinoud if ((is_iso9660 < 0) && (is_udf < 0))
362 1.32 reinoud return SCAN_CONTINUE;
363 1.32 reinoud
364 1.35 reinoud strncpy(a->lp->d_typename, "iso partition", 16);
365 1.35 reinoud
366 1.45 mlelstv /* adjust session information for iso9660 partition */
367 1.32 reinoud if (is_iso9660 >= 0) {
368 1.32 reinoud /* set 'a' partition to iso9660 */
369 1.32 reinoud a->lp->d_partitions[0].p_offset = 0;
370 1.32 reinoud a->lp->d_partitions[0].p_size = a->lp->d_secperunit;
371 1.32 reinoud a->lp->d_partitions[0].p_cdsession = is_iso9660;
372 1.32 reinoud a->lp->d_partitions[0].p_fstype = FS_ISO9660;
373 1.32 reinoud }
374 1.32 reinoud
375 1.45 mlelstv /* UDF doesn't care about the cd session specified here */
376 1.32 reinoud
377 1.32 reinoud return SCAN_FOUND;
378 1.32 reinoud }
379 1.32 reinoud
380 1.32 reinoud
381 1.1 dsl /*
382 1.1 dsl * Attempt to read a disk label from a device
383 1.1 dsl * using the indicated strategy routine.
384 1.1 dsl * The label must be partly set up before this:
385 1.1 dsl * secpercyl, secsize and anything required for a block i/o read
386 1.1 dsl * operation in the driver's strategy/start routines
387 1.1 dsl * must be filled in before calling us.
388 1.1 dsl *
389 1.1 dsl * If dos partition table requested, attempt to load it and
390 1.1 dsl * find disklabel inside a DOS partition. Also, if bad block
391 1.1 dsl * table needed, attempt to extract it as well. Return buffer
392 1.1 dsl * for use in signalling errors if requested.
393 1.1 dsl *
394 1.1 dsl * Returns null on success and an error string on failure.
395 1.1 dsl */
396 1.1 dsl const char *
397 1.13 dsl readdisklabel(dev_t dev, void (*strat)(struct buf *), struct disklabel *lp,
398 1.13 dsl struct cpu_disklabel *osdep)
399 1.1 dsl {
400 1.1 dsl int rval;
401 1.1 dsl int i;
402 1.1 dsl mbr_args_t a;
403 1.1 dsl
404 1.3 dsl memset(&a, 0, sizeof a);
405 1.1 dsl a.lp = lp;
406 1.1 dsl a.strat = strat;
407 1.17 dsl a.action = READ_LABEL;
408 1.1 dsl
409 1.1 dsl /* minimal requirements for architypal disk label */
410 1.1 dsl if (lp->d_secsize == 0)
411 1.1 dsl lp->d_secsize = DEV_BSIZE;
412 1.1 dsl if (lp->d_secperunit == 0)
413 1.1 dsl lp->d_secperunit = 0x1fffffff;
414 1.28 dyoung a.secperunit = lp->d_secperunit;
415 1.1 dsl lp->d_npartitions = RAW_PART + 1;
416 1.1 dsl for (i = 0; i < RAW_PART; i++) {
417 1.1 dsl lp->d_partitions[i].p_size = 0;
418 1.1 dsl lp->d_partitions[i].p_offset = 0;
419 1.1 dsl }
420 1.1 dsl if (lp->d_partitions[RAW_PART].p_size == 0)
421 1.17 dsl lp->d_partitions[RAW_PART].p_size = lp->d_secperunit;
422 1.1 dsl lp->d_partitions[RAW_PART].p_offset = 0;
423 1.1 dsl
424 1.1 dsl /*
425 1.1 dsl * Set partition 'a' to be the whole disk.
426 1.1 dsl * Cleared if we find an mbr or a netbsd label.
427 1.1 dsl */
428 1.1 dsl lp->d_partitions[0].p_size = lp->d_partitions[RAW_PART].p_size;
429 1.1 dsl lp->d_partitions[0].p_fstype = FS_BSDFFS;
430 1.1 dsl
431 1.32 reinoud /*
432 1.34 reinoud * Get a buffer big enough to read a disklabel in and initialize it
433 1.38 pooka * make it three sectors long for the validate_label(); see comment at
434 1.34 reinoud * start of file.
435 1.32 reinoud */
436 1.38 pooka a.bp = geteblk(SCANBLOCKS * (int)lp->d_secsize);
437 1.1 dsl a.bp->b_dev = dev;
438 1.1 dsl
439 1.1 dsl if (osdep)
440 1.1 dsl /*
441 1.1 dsl * Scan mbr searching for netbsd partition and saving
442 1.1 dsl * bios partition information to use if the netbsd one
443 1.1 dsl * is absent.
444 1.1 dsl */
445 1.1 dsl rval = scan_mbr(&a, look_netbsd_part);
446 1.1 dsl else
447 1.1 dsl rval = SCAN_CONTINUE;
448 1.1 dsl
449 1.3 dsl if (rval == SCAN_CONTINUE) {
450 1.1 dsl /* Look at start of disk */
451 1.17 dsl rval = validate_label(&a, 0);
452 1.1 dsl }
453 1.1 dsl
454 1.32 reinoud if (rval == SCAN_CONTINUE) {
455 1.32 reinoud rval = scan_iso_vrs(&a);
456 1.32 reinoud }
457 1.3 dsl #if 0
458 1.3 dsl /*
459 1.3 dsl * Save sector where we found the label for the 'don't overwrite
460 1.3 dsl * the label' check in bounds_check_with_label.
461 1.3 dsl */
462 1.3 dsl if (rval == SCAN_FOUND)
463 1.3 dsl xxx->label_sector = a.label_sector;
464 1.3 dsl #endif
465 1.3 dsl
466 1.1 dsl /* Obtain bad sector table if requested and present */
467 1.37 pooka #ifdef __HAVE_DISKLABEL_DKBAD
468 1.1 dsl if (rval == SCAN_FOUND && osdep && (lp->d_flags & D_BADSECT)) {
469 1.37 pooka struct dkbad *bdp, *db;
470 1.1 dsl int blkno;
471 1.1 dsl
472 1.1 dsl bdp = &osdep->bad;
473 1.1 dsl i = 0;
474 1.1 dsl rval = SCAN_ERROR;
475 1.1 dsl do {
476 1.1 dsl /* read a bad sector table */
477 1.1 dsl blkno = lp->d_secperunit - lp->d_nsectors + i;
478 1.1 dsl if (lp->d_secsize > DEV_BSIZE)
479 1.1 dsl blkno *= lp->d_secsize / DEV_BSIZE;
480 1.1 dsl else
481 1.1 dsl blkno /= DEV_BSIZE / lp->d_secsize;
482 1.1 dsl /* if successful, validate, otherwise try another */
483 1.17 dsl if (read_sector(&a, blkno, 1)) {
484 1.1 dsl a.msg = "bad sector table I/O error";
485 1.1 dsl continue;
486 1.1 dsl }
487 1.1 dsl db = (struct dkbad *)(a.bp->b_data);
488 1.1 dsl #define DKBAD_MAGIC 0x4321
489 1.1 dsl if (db->bt_mbz != 0 || db->bt_flag != DKBAD_MAGIC) {
490 1.1 dsl a.msg = "bad sector table corrupted";
491 1.1 dsl continue;
492 1.1 dsl }
493 1.1 dsl rval = SCAN_FOUND;
494 1.1 dsl *bdp = *db;
495 1.1 dsl break;
496 1.29 ad } while (a.bp->b_error && (i += 2) < 10 &&
497 1.1 dsl i < lp->d_nsectors);
498 1.1 dsl }
499 1.37 pooka #endif /* __HAVE_DISKLABEL_DKBAD */
500 1.1 dsl
501 1.30 ad brelse(a.bp, 0);
502 1.17 dsl if (rval == SCAN_ERROR || rval == SCAN_CONTINUE)
503 1.10 cube return a.msg;
504 1.8 reinoud return NULL;
505 1.1 dsl }
506 1.1 dsl
507 1.1 dsl static int
508 1.13 dsl look_netbsd_part(mbr_args_t *a, mbr_partition_t *dp, int slot, uint ext_base)
509 1.1 dsl {
510 1.1 dsl struct partition *pp;
511 1.1 dsl int ptn_base = ext_base + le32toh(dp->mbrp_start);
512 1.1 dsl int rval;
513 1.1 dsl
514 1.1 dsl if (
515 1.1 dsl #ifdef COMPAT_386BSD_MBRPART
516 1.4 lukem dp->mbrp_type == MBR_PTYPE_386BSD ||
517 1.1 dsl #endif
518 1.4 lukem dp->mbrp_type == MBR_PTYPE_NETBSD) {
519 1.17 dsl rval = validate_label(a, ptn_base);
520 1.1 dsl
521 1.12 dsl #if RAW_PART == 3
522 1.1 dsl /* Put actual location where we found the label into ptn 2 */
523 1.12 dsl if (rval == SCAN_FOUND || a->lp->d_partitions[2].p_size == 0) {
524 1.1 dsl a->lp->d_partitions[2].p_size = le32toh(dp->mbrp_size);
525 1.1 dsl a->lp->d_partitions[2].p_offset = ptn_base;
526 1.1 dsl }
527 1.12 dsl #endif
528 1.1 dsl
529 1.1 dsl /* If we got a netbsd label look no further */
530 1.1 dsl if (rval == SCAN_FOUND)
531 1.1 dsl return rval;
532 1.1 dsl }
533 1.1 dsl
534 1.1 dsl /* Install main partitions into e..h and extended into i+ */
535 1.1 dsl if (ext_base == 0)
536 1.1 dsl slot += 4;
537 1.1 dsl else {
538 1.4 lukem slot = 4 + MBR_PART_COUNT;
539 1.1 dsl pp = &a->lp->d_partitions[slot];
540 1.1 dsl for (; slot < MAXPARTITIONS; pp++, slot++) {
541 1.1 dsl /* This gets called twice - avoid duplicates */
542 1.1 dsl if (pp->p_offset == ptn_base &&
543 1.1 dsl pp->p_size == le32toh(dp->mbrp_size))
544 1.1 dsl break;
545 1.1 dsl if (pp->p_size == 0)
546 1.1 dsl break;
547 1.1 dsl }
548 1.1 dsl }
549 1.1 dsl
550 1.1 dsl if (slot < MAXPARTITIONS) {
551 1.1 dsl /* Stop 'a' being the entire disk */
552 1.1 dsl a->lp->d_partitions[0].p_size = 0;
553 1.1 dsl a->lp->d_partitions[0].p_fstype = 0;
554 1.1 dsl
555 1.1 dsl /* save partition info */
556 1.1 dsl pp = &a->lp->d_partitions[slot];
557 1.1 dsl pp->p_offset = ptn_base;
558 1.1 dsl pp->p_size = le32toh(dp->mbrp_size);
559 1.4 lukem pp->p_fstype = xlat_mbr_fstype(dp->mbrp_type);
560 1.1 dsl
561 1.1 dsl if (slot >= a->lp->d_npartitions)
562 1.1 dsl a->lp->d_npartitions = slot + 1;
563 1.1 dsl }
564 1.1 dsl
565 1.1 dsl return SCAN_CONTINUE;
566 1.1 dsl }
567 1.1 dsl
568 1.55 kamil __noubsan
569 1.51 christos static bool
570 1.51 christos check_label_magic(const struct disklabel *dlp, uint32_t diskmagic)
571 1.51 christos {
572 1.54 kamil return memcmp(&dlp->d_magic, &diskmagic, sizeof(diskmagic)) == 0 &&
573 1.54 kamil memcmp(&dlp->d_magic2, &diskmagic, sizeof(diskmagic)) == 0;
574 1.51 christos }
575 1.1 dsl
576 1.47 rin #ifdef DISKLABEL_EI
577 1.47 rin /*
578 1.47 rin * - For read, convert a label to the native byte order.
579 1.47 rin * - For update or write, if a label already exists, keep its byte order.
580 1.47 rin * Otherwise, write a new label in the native byte order.
581 1.47 rin */
582 1.47 rin #endif
583 1.1 dsl static int
584 1.17 dsl validate_label(mbr_args_t *a, uint label_sector)
585 1.1 dsl {
586 1.56 kamil struct disklabel *dlp;
587 1.17 dsl char *dlp_lim, *dlp_byte;
588 1.1 dsl int error;
589 1.47 rin #ifdef DISKLABEL_EI
590 1.47 rin int swapped = 0;
591 1.47 rin uint16_t npartitions;
592 1.47 rin #endif
593 1.1 dsl
594 1.1 dsl /* Next, dig out disk label */
595 1.38 pooka if (read_sector(a, label_sector, SCANBLOCKS)) {
596 1.3 dsl a->msg = "disk label read failed";
597 1.1 dsl return SCAN_ERROR;
598 1.1 dsl }
599 1.1 dsl
600 1.1 dsl /* Locate disk label within block and validate */
601 1.1 dsl /*
602 1.1 dsl * XXX (dsl) This search may be a waste of time, a lot of other i386
603 1.1 dsl * code assumes the label is at offset LABELOFFSET (=0) in the sector.
604 1.1 dsl *
605 1.3 dsl * If we want to support disks from other netbsd ports, then the
606 1.1 dsl * code should also allow for a shorter label nearer the end of
607 1.1 dsl * the disk sector, and (IIRC) labels within 8k of the disk start.
608 1.1 dsl */
609 1.3 dsl dlp = (void *)a->bp->b_data;
610 1.26 christos dlp_lim = (char *)a->bp->b_data + a->bp->b_bcount - sizeof *dlp;
611 1.48 christos for (;; dlp = (void *)((char *)dlp + sizeof(uint32_t))) {
612 1.17 dsl if ((char *)dlp > dlp_lim) {
613 1.17 dsl if (a->action != WRITE_LABEL)
614 1.3 dsl return SCAN_CONTINUE;
615 1.42 wiz /* Write at arch. dependent default location */
616 1.26 christos dlp_byte = (char *)a->bp->b_data + LABELOFFSET;
617 1.17 dsl if (label_sector)
618 1.17 dsl dlp_byte += MBR_LABELSECTOR * a->lp->d_secsize;
619 1.17 dsl else
620 1.17 dsl dlp_byte += LABELSECTOR * a->lp->d_secsize;
621 1.17 dsl dlp = (void *)dlp_byte;
622 1.1 dsl break;
623 1.1 dsl }
624 1.56 kamil if (!check_label_magic(dlp, DISKMAGIC))
625 1.47 rin #ifdef DISKLABEL_EI
626 1.47 rin {
627 1.56 kamil if (!check_label_magic(dlp, bswap32(DISKMAGIC)))
628 1.47 rin continue;
629 1.47 rin
630 1.47 rin /*
631 1.47 rin * The label is in the other byte order. We need to
632 1.47 rin * checksum before swapping the byte order.
633 1.47 rin */
634 1.56 kamil npartitions = bswap16(dlp->d_npartitions);
635 1.47 rin if (npartitions > MAXPARTITIONS ||
636 1.56 kamil dkcksum_sized(dlp, npartitions) != 0)
637 1.47 rin goto corrupted;
638 1.47 rin
639 1.47 rin swapped = 1;
640 1.47 rin }
641 1.47 rin #else
642 1.17 dsl continue;
643 1.47 rin #endif
644 1.56 kamil else if (dlp->d_npartitions > MAXPARTITIONS ||
645 1.56 kamil dkcksum(dlp) != 0) {
646 1.47 rin #ifdef DISKLABEL_EI
647 1.47 rin corrupted:
648 1.47 rin #endif
649 1.17 dsl a->msg = "disk label corrupted";
650 1.17 dsl continue;
651 1.17 dsl }
652 1.17 dsl break;
653 1.3 dsl }
654 1.3 dsl
655 1.17 dsl switch (a->action) {
656 1.3 dsl case READ_LABEL:
657 1.47 rin #ifdef DISKLABEL_EI
658 1.47 rin if (swapped)
659 1.57 mrg disklabel_swap(a->lp, dlp);
660 1.47 rin else
661 1.56 kamil *a->lp = *dlp;
662 1.47 rin #else
663 1.56 kamil *a->lp = *dlp;
664 1.47 rin #endif
665 1.28 dyoung if ((a->msg = convertdisklabel(a->lp, a->strat, a->bp,
666 1.28 dyoung a->secperunit)) != NULL)
667 1.28 dyoung return SCAN_ERROR;
668 1.3 dsl a->label_sector = label_sector;
669 1.1 dsl return SCAN_FOUND;
670 1.3 dsl case UPDATE_LABEL:
671 1.3 dsl case WRITE_LABEL:
672 1.47 rin #ifdef DISKLABEL_EI
673 1.47 rin /* DO NOT swap a->lp itself for later references. */
674 1.47 rin if (swapped)
675 1.57 mrg disklabel_swap(dlp, a->lp);
676 1.47 rin else
677 1.56 kamil *dlp = *a->lp;
678 1.47 rin #else
679 1.56 kamil *dlp = *a->lp;
680 1.47 rin #endif
681 1.31 ad a->bp->b_oflags &= ~BO_DONE;
682 1.31 ad a->bp->b_flags &= ~B_READ;
683 1.3 dsl a->bp->b_flags |= B_WRITE;
684 1.3 dsl (*a->strat)(a->bp);
685 1.3 dsl error = biowait(a->bp);
686 1.3 dsl if (error != 0) {
687 1.3 dsl a->error = error;
688 1.3 dsl a->msg = "disk label write failed";
689 1.3 dsl return SCAN_ERROR;
690 1.3 dsl }
691 1.3 dsl a->written++;
692 1.3 dsl /* Write label to all mbr partitions */
693 1.3 dsl return SCAN_CONTINUE;
694 1.3 dsl default:
695 1.3 dsl return SCAN_ERROR;
696 1.1 dsl }
697 1.1 dsl }
698 1.1 dsl
699 1.1 dsl /*
700 1.1 dsl * Write disk label back to device after modification.
701 1.1 dsl */
702 1.1 dsl int
703 1.13 dsl writedisklabel(dev_t dev, void (*strat)(struct buf *), struct disklabel *lp,
704 1.13 dsl struct cpu_disklabel *osdep)
705 1.1 dsl {
706 1.1 dsl mbr_args_t a;
707 1.1 dsl
708 1.3 dsl memset(&a, 0, sizeof a);
709 1.1 dsl a.lp = lp;
710 1.1 dsl a.strat = strat;
711 1.1 dsl
712 1.1 dsl /* get a buffer and initialize it */
713 1.38 pooka a.bp = geteblk(SCANBLOCKS * (int)lp->d_secsize);
714 1.1 dsl a.bp->b_dev = dev;
715 1.1 dsl
716 1.17 dsl /* osdep => we expect an mbr with label in netbsd ptn */
717 1.17 dsl a.action = osdep != NULL ? WRITE_LABEL : UPDATE_LABEL;
718 1.17 dsl
719 1.17 dsl /* Write/update the label to every netbsd mbr partition */
720 1.17 dsl scan_mbr(&a, write_netbsd_label);
721 1.3 dsl
722 1.17 dsl /* Old write the label at the start of the volume on disks that
723 1.17 dsl * don't have a valid mbr (always update an existing one) */
724 1.17 dsl a.action = a.found_mbr ? UPDATE_LABEL : WRITE_LABEL;
725 1.17 dsl validate_label(&a, 0);
726 1.1 dsl
727 1.3 dsl if (a.written == 0 && a.error == 0)
728 1.1 dsl a.error = ESRCH;
729 1.1 dsl
730 1.30 ad brelse(a.bp, 0);
731 1.1 dsl return a.error;
732 1.1 dsl }
733 1.1 dsl
734 1.1 dsl static int
735 1.21 yamt write_netbsd_label(mbr_args_t *a, mbr_partition_t *dp, int slot, uint ext_base)
736 1.1 dsl {
737 1.1 dsl int ptn_base = ext_base + le32toh(dp->mbrp_start);
738 1.1 dsl
739 1.4 lukem if (dp->mbrp_type != MBR_PTYPE_NETBSD)
740 1.1 dsl return SCAN_CONTINUE;
741 1.1 dsl
742 1.17 dsl return validate_label(a, ptn_base);
743 1.1 dsl }
744