/*	$OpenBSD: mbr.c,v 1.98 2021/08/07 13:33:12 krw Exp $	*/

/*
 * Copyright (c) 1997 Tobias Weingartner
 *
 * 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/ioctl.h>
#include <sys/disklabel.h>
#include <sys/dkio.h>

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

#include "part.h"
#include "disk.h"
#include "misc.h"
#include "mbr.h"
#include "gpt.h"

struct dos_mbr		default_dmbr;

void		mbr_to_dos_mbr(const struct mbr *, struct dos_mbr *);
void		dos_mbr_to_mbr(const struct dos_mbr *, const uint64_t,
    const uint64_t, struct mbr *);

void
MBR_init(struct mbr *mbr)
{
	uint64_t		adj;
	daddr_t			daddr;

	memset(&gmbr, 0, sizeof(gmbr));
	memset(&gh, 0, sizeof(gh));
	memset(&gp, 0, sizeof(gp));

	if (mbr->mbr_lba_self != 0) {
		/* Extended MBR - save lba's, set sig, zap everything else. */
		memset(mbr->mbr_code, 0, sizeof(mbr->mbr_code));
		memset(mbr->mbr_prt, 0, sizeof(mbr->mbr_prt));
		mbr->mbr_signature = DOSMBR_SIGNATURE;
		return;
	}

	dos_mbr_to_mbr(&default_dmbr, 0, 0, mbr);

	/*
	 * XXX Do *NOT* zap all MBR parts! Some archs still read default mbr
	 * from disk! Just mark them inactive until -b goodness spreads
	 * further.
	 */

	mbr->mbr_prt[0].prt_flag = 0;
	mbr->mbr_prt[1].prt_flag = 0;
	mbr->mbr_prt[2].prt_flag = 0;

	mbr->mbr_prt[3].prt_flag = DOSACTIVE;
	mbr->mbr_signature = DOSMBR_SIGNATURE;

	/* Use whole disk. Reserve first track, or first cyl, if possible. */
	mbr->mbr_prt[3].prt_id = DOSPTYP_OPENBSD;
	if (disk.dk_heads > 1)
		mbr->mbr_prt[3].prt_shead = 1;
	else
		mbr->mbr_prt[3].prt_shead = 0;
	if (disk.dk_heads < 2 && disk.dk_cylinders > 1)
		mbr->mbr_prt[3].prt_scyl = 1;
	else
		mbr->mbr_prt[3].prt_scyl = 0;
	mbr->mbr_prt[3].prt_ssect = 1;

	/* Go right to the end */
	mbr->mbr_prt[3].prt_ecyl = disk.dk_cylinders - 1;
	mbr->mbr_prt[3].prt_ehead = disk.dk_heads - 1;
	mbr->mbr_prt[3].prt_esect = disk.dk_sectors;

	/* Fix up start/length fields */
	PRT_fix_BN(&mbr->mbr_prt[3], 3);

#if defined(__powerpc__) || defined(__mips__)
	/* Now fix up for the MS-DOS boot partition on PowerPC/MIPS. */
	mbr->mbr_prt[0].prt_flag = DOSACTIVE;	/* Boot from dos part */
	mbr->mbr_prt[3].prt_flag = 0;
	mbr->mbr_prt[3].prt_ns += mbr->mbr_prt[3].prt_bs;
	mbr->mbr_prt[3].prt_bs = mbr->mbr_prt[0].prt_bs + mbr->mbr_prt[0].prt_ns;
	mbr->mbr_prt[3].prt_ns -= mbr->mbr_prt[3].prt_bs;
	PRT_fix_CHS(&mbr->mbr_prt[3]);
	if ((mbr->mbr_prt[3].prt_shead != 1) || (mbr->mbr_prt[3].prt_ssect != 1)) {
		/* align the partition on a cylinder boundary */
		mbr->mbr_prt[3].prt_shead = 0;
		mbr->mbr_prt[3].prt_ssect = 1;
		mbr->mbr_prt[3].prt_scyl += 1;
	}
	/* Fix up start/length fields */
	PRT_fix_BN(&mbr->mbr_prt[3], 3);
#else
	if (disk.dk_bootprt.prt_ns > 0) {
		mbr->mbr_prt[0] = disk.dk_bootprt;
		PRT_fix_CHS(&mbr->mbr_prt[0]);
		mbr->mbr_prt[3].prt_ns += mbr->mbr_prt[3].prt_bs;
		mbr->mbr_prt[3].prt_bs = mbr->mbr_prt[0].prt_bs + mbr->mbr_prt[0].prt_ns;
		mbr->mbr_prt[3].prt_ns -= mbr->mbr_prt[3].prt_bs;
		PRT_fix_CHS(&mbr->mbr_prt[3]);
	}
#endif

	/* Start OpenBSD MBR partition on a power of 2 block number. */
	daddr = 1;
	while (daddr < DL_SECTOBLK(&dl, mbr->mbr_prt[3].prt_bs))
		daddr *= 2;
	adj = DL_BLKTOSEC(&dl, daddr) - mbr->mbr_prt[3].prt_bs;
	mbr->mbr_prt[3].prt_bs += adj;
	mbr->mbr_prt[3].prt_ns -= adj;
	PRT_fix_CHS(&mbr->mbr_prt[3]);
}

void
dos_mbr_to_mbr(const struct dos_mbr *dmbr, const uint64_t lba_self,
    const uint64_t lba_firstembr, struct mbr *mbr)
{
	struct dos_partition	dos_parts[NDOSPART];
	int			i;

	memcpy(mbr->mbr_code, dmbr->dmbr_boot, sizeof(mbr->mbr_code));
	mbr->mbr_lba_self = lba_self;
	mbr->mbr_lba_firstembr = lba_firstembr;
	mbr->mbr_signature = letoh16(dmbr->dmbr_sign);

	memcpy(dos_parts, dmbr->dmbr_parts, sizeof(dos_parts));

	for (i = 0; i < NDOSPART; i++)
		PRT_parse(&dos_parts[i], lba_self, lba_firstembr,
		    &mbr->mbr_prt[i]);
}

void
mbr_to_dos_mbr(const struct mbr *mbr, struct dos_mbr *dos_mbr)
{
	struct dos_partition	dos_partition;
	int			i;

	memcpy(dos_mbr->dmbr_boot, mbr->mbr_code, sizeof(dos_mbr->dmbr_boot));
	dos_mbr->dmbr_sign = htole16(DOSMBR_SIGNATURE);

	for (i = 0; i < NDOSPART; i++) {
		PRT_make(&mbr->mbr_prt[i], mbr->mbr_lba_self, mbr->mbr_lba_firstembr,
		    &dos_partition);
		memcpy(&dos_mbr->dmbr_parts[i], &dos_partition,
		    sizeof(dos_mbr->dmbr_parts[i]));
	}
}

void
MBR_print(const struct mbr *mbr, const char *units)
{
	int			i;

	DISK_printgeometry(NULL);

	printf("Offset: %lld\t", (long long)mbr->mbr_lba_self);
	printf("Signature: 0x%X\n", (int)mbr->mbr_signature);
	PRT_print(0, NULL, units);

	for (i = 0; i < NDOSPART; i++)
		PRT_print(i, &mbr->mbr_prt[i], units);
}

int
MBR_read(const uint64_t lba_self, const uint64_t lba_firstembr, struct mbr *mbr)
{
	struct dos_mbr		 dos_mbr;
	char			*secbuf;

	secbuf = DISK_readsectors(lba_self, 1);
	if (secbuf == NULL)
		return -1;

	memcpy(&dos_mbr, secbuf, sizeof(dos_mbr));
	free(secbuf);

	dos_mbr_to_mbr(&dos_mbr, lba_self, lba_firstembr, mbr);

	return 0;
}

int
MBR_write(const struct mbr *mbr)
{
	struct dos_mbr		 dos_mbr;
	char			*secbuf;
	int			 rslt;

	secbuf = DISK_readsectors(mbr->mbr_lba_self, 1);
	if (secbuf == NULL)
		return -1;

	mbr_to_dos_mbr(mbr, &dos_mbr);
	memcpy(secbuf, &dos_mbr, sizeof(dos_mbr));

	rslt = DISK_writesectors(secbuf, mbr->mbr_lba_self, 1);
	free(secbuf);
	if (rslt)
		return -1;

	/* Refresh in-kernel disklabel from the updated disk information. */
	if (ioctl(disk.dk_fd, DIOCRLDINFO, 0) == -1)
		warn("DIOCRLDINFO");

	return 0;
}