Source-Changes-HG archive
[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index][Old Index]
[src/perseant-stdc-iso10646]: src/usr.sbin/installboot/arch 3356330
details: https://anonhg.NetBSD.org/src/rev/3970030f94ec
branches: perseant-stdc-iso10646
changeset: 850661:3970030f94ec
user: christos <christos%NetBSD.org@localhost>
date: Mon Jul 17 18:43:46 2017 +0000
description:
3356330
diffstat:
usr.sbin/installboot/arch/landisk.c | 255 ++++++++++++++++++++++++++++++++++++
1 files changed, 255 insertions(+), 0 deletions(-)
diffs (259 lines):
diff -r 9b0d0e86d40e -r 3970030f94ec usr.sbin/installboot/arch/landisk.c
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/usr.sbin/installboot/arch/landisk.c Mon Jul 17 18:43:46 2017 +0000
@@ -0,0 +1,255 @@
+/* $NetBSD: landisk.c,v 1.7.2.2 2017/07/17 18:43:46 christos Exp $ */
+
+/*-
+ * Copyright (c) 2003 The NetBSD Foundation, Inc.
+ * All rights reserved.
+ *
+ * This code is derived from software contributed to The NetBSD Foundation
+ * by David Laight.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
+ * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
+ * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
+ * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#if HAVE_NBTOOL_CONFIG_H
+#include "nbtool_config.h"
+#endif
+
+#include <sys/cdefs.h>
+#if !defined(__lint)
+__RCSID("$NetBSD: landisk.c,v 1.7.2.2 2017/07/17 18:43:46 christos Exp $");
+#endif /* !__lint */
+
+#include <sys/param.h>
+
+#include <assert.h>
+#include <err.h>
+#include <md5.h>
+#include <stddef.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "installboot.h"
+
+static int landisk_setboot(ib_params *);
+
+struct ib_mach ib_mach_landisk =
+ { "landisk", landisk_setboot, no_clearboot, no_editboot,
+ IB_TIMEOUT };
+
+static int
+landisk_setboot(ib_params *params)
+{
+ struct mbr_sector mbr;
+ struct landisk_boot_params bp, *bpp;
+ uint8_t *bootstrapbuf;
+ ssize_t rv;
+ uint32_t magic;
+ size_t bootstrapsize;
+ int retval, i;
+ uint32_t bplen;
+
+ assert(params != NULL);
+ assert(params->fsfd != -1);
+ assert(params->filesystem != NULL);
+ assert(params->s1fd != -1);
+ assert(params->stage1 != NULL);
+
+ retval = 0;
+ bootstrapbuf = NULL;
+
+ /*
+ * There is only 8k of space in a FFSv1 partition (and ustarfs)
+ * so ensure we don't splat over anything important.
+ */
+ if (params->s1stat.st_size > 8192) {
+ warnx("stage1 bootstrap `%s' is larger than 8192 bytes",
+ params->stage1);
+ goto done;
+ }
+
+ /*
+ * Read in the existing MBR.
+ */
+ rv = pread(params->fsfd, &mbr, sizeof(mbr), MBR_BBSECTOR);
+ if (rv == -1) {
+ warn("Reading `%s'", params->filesystem);
+ goto done;
+ } else if (rv != sizeof(mbr)) {
+ warnx("Reading `%s': short read", params->filesystem);
+ goto done;
+ }
+ if (mbr.mbr_magic != le16toh(MBR_MAGIC)) {
+ const char *p = (const char *)&mbr;
+ const char *e = p + sizeof(mbr);
+ while (p < e && !*p)
+ p++;
+ if (p != e) {
+ if (params->flags & IB_VERBOSE) {
+ printf(
+ "Ignoring MBR with invalid magic in sector 0 of `%s'\n",
+ params->filesystem);
+ }
+ memset(&mbr, 0, sizeof(mbr));
+ }
+ }
+
+ /*
+ * Allocate a buffer, with space to round up the input file
+ * to the next block size boundary, and with space for the boot
+ * block.
+ */
+ bootstrapsize = roundup(params->s1stat.st_size, 512);
+
+ bootstrapbuf = malloc(bootstrapsize);
+ if (bootstrapbuf == NULL) {
+ warn("Allocating %zu bytes", bootstrapsize);
+ goto done;
+ }
+ memset(bootstrapbuf, 0, bootstrapsize);
+
+ /*
+ * Read the file into the buffer.
+ */
+ rv = pread(params->s1fd, bootstrapbuf, params->s1stat.st_size, 0);
+ if (rv == -1) {
+ warn("Reading `%s'", params->stage1);
+ goto done;
+ } else if (rv != params->s1stat.st_size) {
+ warnx("Reading `%s': short read", params->stage1);
+ goto done;
+ }
+
+ magic = *(uint32_t *)(bootstrapbuf + 512 * 2 + 4);
+ if (magic != htole32(LANDISK_BOOT_MAGIC_1)) {
+ warnx("Invalid magic in stage1 boostrap %x != %x",
+ magic, htole32(LANDISK_BOOT_MAGIC_1));
+ goto done;
+ }
+
+ /*
+ * Determine size of BIOS Parameter Block (BPB) to copy from
+ * original MBR to the temporary buffer by examining the first
+ * few instruction in the new bootblock. Supported values:
+ * 2b a0 11 jmp ENDOF(mbr_bpbFAT32)+1, nop
+ * (anything else) ; don't preserve
+ */
+#if 0
+ int bpbsize;
+ if (bootstrapbuf[1] == 0xa0 && bootstrapbuf[2] == 0x11 &&
+ (bootstrapbuf[0] == 0x2b /*|| bootstrapbuf[0] == 0x1d*/)) {
+ bpbsize = bootstrapbuf[0] + 2 - MBR_BPB_OFFSET;
+ }
+#endif
+
+ /*
+ * Ensure bootxx hasn't got any code or data (i.e, non-zero bytes) in
+ * the partition table.
+ */
+ for (i = 0; i < (int)sizeof(mbr.mbr_parts); i++) {
+ if (*(uint8_t *)(bootstrapbuf + MBR_PART_OFFSET + i) != 0) {
+ warnx(
+ "Partition table has non-zero byte at offset %d in `%s'",
+ MBR_PART_OFFSET + i, params->stage1);
+ goto done;
+ }
+ }
+
+#if 0
+ /*
+ * Copy the BPB and the partition table from the original MBR to the
+ * temporary buffer so that they're written back to the fs.
+ */
+ if (bpbsize != 0) {
+ if (params->flags & IB_VERBOSE)
+ printf("Preserving %d (%#x) bytes of the BPB\n",
+ bpbsize, bpbsize);
+ (void)memcpy(bootstrapbuf + MBR_BPB_OFFSET, &mbr.mbr_bpb,
+ bpbsize);
+ }
+#endif
+ memcpy(bootstrapbuf + MBR_PART_OFFSET, &mbr.mbr_parts,
+ sizeof(mbr.mbr_parts));
+
+ /*
+ * Fill in any user-specified options into the
+ * struct landisk_boot_params
+ * that's 8 bytes in from the start of the third sector.
+ * See sys/arch/landisk/stand/bootxx/bootxx.S for more information.
+ */
+ bpp = (void *)(bootstrapbuf + 512 * 2 + 8);
+ bplen = le32toh(bpp->bp_length);
+ if (bplen > sizeof bp)
+ /* Ignore pad space in bootxx */
+ bplen = sizeof bp;
+ /* Take (and update) local copy so we handle size mismatches */
+ memset(&bp, 0, sizeof bp);
+ memcpy(&bp, bpp, bplen);
+ if (params->flags & IB_TIMEOUT)
+ bp.bp_timeout = htole32(params->timeout);
+ /* Check we aren't trying to set anything we can't save */
+ if (bplen < sizeof bp && memcmp((char *)&bp + bplen,
+ (char *)&bp + bplen + 1,
+ sizeof bp - bplen - 1) != 0) {
+ warnx("Patch area in stage1 bootstrap is too small");
+ goto done;
+ }
+ memcpy(bpp, &bp, bplen);
+
+ if (params->flags & IB_NOWRITE) {
+ retval = 1;
+ goto done;
+ }
+
+ /*
+ * Write MBR code to sector zero.
+ */
+ rv = pwrite(params->fsfd, bootstrapbuf, 512, 0);
+ if (rv == -1) {
+ warn("Writing `%s'", params->filesystem);
+ goto done;
+ } else if (rv != 512) {
+ warnx("Writing `%s': short write", params->filesystem);
+ goto done;
+ }
+
+ /*
+ * Skip disklabel in sector 1 and write bootxx to sectors 2..N.
+ */
+ rv = pwrite(params->fsfd, bootstrapbuf + 512 * 2,
+ bootstrapsize - 512 * 2, 512 * 2);
+ if (rv == -1) {
+ warn("Writing `%s'", params->filesystem);
+ goto done;
+ } else if ((size_t)rv != bootstrapsize - 512 * 2) {
+ warnx("Writing `%s': short write", params->filesystem);
+ goto done;
+ }
+
+ retval = 1;
+
+ done:
+ if (bootstrapbuf)
+ free(bootstrapbuf);
+ return retval;
+}
Home |
Main Index |
Thread Index |
Old Index