/*      $NetBSD: landisk.c,v 1.9 2024/05/11 06:50:23 andvar 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.9 2024/05/11 06:50:23 andvar 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 = {
       .name           =       "landisk",
       .setboot        =       landisk_setboot,
       .clearboot      =       no_clearboot,
       .editboot       =       no_editboot,
       .valid_flags    =       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 bootstrap %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;
}

You are viewing proxied material from ftp.icm.edu.pl. The copyright of proxied material belongs to its original authors. Any comments or complaints in relation to proxied material should be directed to the original authors of the content concerned. Please see the disclaimer for more details.