/*      $OpenBSD: i386_softraid.c,v 1.23 2025/09/17 16:12:10 deraadt Exp $      */
/*
 * Copyright (c) 2012 Joel Sing <jsing@openbsd.org>
 * Copyright (c) 2010 Otto Moerbeek <otto@drijf.net>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/param.h>  /* DEV_BSIZE */
#include <sys/disklabel.h>
#include <sys/dkio.h>
#include <sys/ioctl.h>
#include <sys/stat.h>

#include <dev/biovar.h>
#include <dev/softraidvar.h>
#include <ufs/ufs/dinode.h>

#include <err.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "installboot.h"
#include "i386_installboot.h"

void    sr_install_bootblk(int, int, int);
void    sr_install_bootldr(int, char *);

void
sr_install_bootblk(int devfd, int vol, int disk)
{
        struct bioc_disk bd;
        struct disklabel dl;
        struct gpt_partition gp;
        struct partition *pp;
        uint32_t poffset;
        char *dev;
        char part;
        int gpart, efipart;
        int diskfd;

        diskfd = sr_open_chunk(devfd, vol, disk, &bd, &dev, &part);
        if (diskfd == -1)
                return;

        /* Get and check disklabel. */
        if (ioctl(diskfd, DIOCGDINFO, &dl) == -1)
                err(1, "disklabel: %s", dev);
        if (dl.d_magic != DISKMAGIC)
                err(1, "bad disklabel magic=0x%08x", dl.d_magic);

        /* Warn on unknown disklabel types. */
        if (dl.d_type == 0)
                warnx("disklabel type unknown");

        efipart = findgptefisys(diskfd, &dl, &gpart, &gp);
        if (efipart != -1) {
                write_filesystem(&dl, (char)efipart, gpart, &gp);
                return;
        }

        /* Determine poffset and set symbol value. */
        pp = &dl.d_partitions[DL_PARTNAME2NUM(part)];
        if (pp->p_offseth != 0)
                errx(1, "partition offset too high");
        poffset = pp->p_offset;                 /* Offset of RAID partition. */
        poffset += SR_BOOT_LOADER_OFFSET;       /* SR boot loader area. */
        sym_set_value(pbr_symbols, "_p_offset", poffset);

        if (verbose)
                fprintf(stderr, "%s%c: %s boot blocks on %s, part offset %u\n",
                    bd.bd_vendor, part,
                    (nowrite ? "would install" : "installing"), dev, poffset);

        /* Write boot blocks to device. */
        write_bootblocks(diskfd, dev, &dl);

        close(diskfd);
}

void
sr_install_bootldr(int devfd, char *dev)
{
        struct bioc_installboot bb;
        struct stat sb;
        struct ufs1_dinode *ino_p;
        uint32_t bootsize, inodeblk, inodedbl;
        uint16_t bsize = SR_FS_BLOCKSIZE;
        uint16_t nblocks;
        uint8_t bshift = 5;             /* fragsize == blocksize */
        int fd, i;
        u_char *p;

        /*
         * Install boot loader into softraid boot loader storage area.
         *
         * In order to allow us to reuse the existing biosboot we construct
         * a fake FFS filesystem with a single inode, which points to the
         * boot loader.
         */

        nblocks = howmany(SR_BOOT_LOADER_SIZE, SR_FS_BLOCKSIZE / DEV_BSIZE);
        inodeblk = nblocks - 1;
        bootsize = nblocks * SR_FS_BLOCKSIZE;

        p = calloc(1, bootsize);
        if (p == NULL)
                err(1, NULL);

        fd = open(stage2, O_RDONLY);
        if (fd == -1)
                err(1, NULL);

        if (fstat(fd, &sb) == -1)
                err(1, NULL);

        nblocks = howmany(sb.st_blocks, SR_FS_BLOCKSIZE / DEV_BSIZE);
        if (sb.st_blocks * S_BLKSIZE > bootsize -
            (int)(sizeof(struct ufs1_dinode)))
                errx(1, "boot code will not fit");

        /* We only need to fill the direct block array. */
        ino_p = (struct ufs1_dinode *)&p[bootsize - sizeof(struct ufs1_dinode)];

        ino_p->di_mode = sb.st_mode;
        ino_p->di_nlink = 1;
        ino_p->di_inumber = 0xfeebfaab;
        ino_p->di_size = read(fd, p, sb.st_blocks * S_BLKSIZE);
        ino_p->di_blocks = nblocks;
        for (i = 0; i < nblocks; i++)
                ino_p->di_db[i] = i;

        inodedbl = ((u_char*)&ino_p->di_db[0] -
            &p[bootsize - SR_FS_BLOCKSIZE]) + INODEOFF;

        memset(&bb, 0, sizeof(bb));
        bb.bb_bootldr = p;
        bb.bb_bootldr_size = bootsize;
        bb.bb_bootblk = "XXX";
        bb.bb_bootblk_size = sizeof("XXX");
        strncpy(bb.bb_dev, dev, sizeof(bb.bb_dev));
        if (verbose)
                fprintf(stderr, "%s: %s boot loader on softraid volume\n", dev,
                    (nowrite ? "would install" : "installing"));
        if (!nowrite) {
                if (ioctl(devfd, BIOCINSTALLBOOT, &bb) == -1)
                        errx(1, "softraid installboot failed");
                sr_status(&bb.bb_bio.bio_status);
        }

        /*
         * Set the values that will need to go into biosboot
         * (the partition boot record, a.k.a. the PBR).
         */
        sym_set_value(pbr_symbols, "_fs_bsize_p", (bsize / 16));
        sym_set_value(pbr_symbols, "_fs_bsize_s", (bsize / 512));
        sym_set_value(pbr_symbols, "_fsbtodb", bshift);
        sym_set_value(pbr_symbols, "_inodeblk", inodeblk);
        sym_set_value(pbr_symbols, "_inodedbl", inodedbl);
        sym_set_value(pbr_symbols, "_nblocks", nblocks);
        sym_set_value(pbr_symbols, "_blkincr", 0);

        if (verbose)
                fprintf(stderr, "%s is %d blocks x %d bytes\n",
                    stage2, nblocks, bsize);

        free(p);
        close(fd);
}