/*	$NetBSD: bootinfo_biosgeom.c,v 1.24 2019/08/01 13:11:03 manu Exp $	*/

/*
 * Copyright (c) 1997
 *	Matthias Drochner.  All rights reserved.
 *
 * 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 AUTHOR ``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 AUTHOR 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.
 *
 */

#include <sys/types.h>
#include <machine/disklabel.h>
#include <machine/cpu.h>

#include <lib/libkern/libkern.h>
#include <lib/libsa/stand.h>

#include "libi386.h"
#include "biosdisk_ll.h"
#include "bootinfo.h"

#ifdef BIOSDISK_EXTINFO_V3
static struct {
	char	*name;
	int	flag;
} bus_names[] = { {"ISA", BI_GEOM_BUS_ISA},
		  {"PCI", BI_GEOM_BUS_PCI},
		  {NULL, BI_GEOM_BUS_OTHER} };
static struct {
	char	*name;
	int	flag;
} iface_names[] = { {"ATA", BI_GEOM_IFACE_ATA},
		    {"ATAPI", BI_GEOM_IFACE_ATAPI},
		    {"SCSI", BI_GEOM_IFACE_SCSI},
		    {"USB", BI_GEOM_IFACE_USB},
		    {"1394", BI_GEOM_IFACE_1394},
		    {"FIBRE", BI_GEOM_IFACE_FIBRE},
		    {NULL, BI_GEOM_IFACE_OTHER} };
#endif

void
bi_getbiosgeom(void)
{
	struct btinfo_biosgeom *bibg;
	size_t bibg_len = sizeof(*bibg);
	int i, j, nvalid;
	int nhd;
	unsigned int cksum;
	struct biosdisk_ll d;
	struct biosdisk_extinfo ed;
	char buf[BIOSDISK_DEFAULT_SECSIZE];

	nhd = get_harddrives();
#ifdef GEOM_DEBUG
	printf("nhd %d\n", nhd);
#endif

	bibg_len += nhd * sizeof(struct bi_biosgeom_entry); 
	bibg = alloc(bibg_len);
	if (bibg == NULL)
		return;

	for (i = nvalid = 0; i < MAX_BIOSDISKS && nvalid < nhd; i++) {
		d.dev = 0x80 + i;

		if (set_geometry(&d, &ed))
			continue;
		memset(&bibg->disk[nvalid], 0, sizeof(bibg->disk[nvalid]));

		bibg->disk[nvalid].sec = d.sec;
		bibg->disk[nvalid].head = d.head;
		bibg->disk[nvalid].cyl = d.cyl;
		bibg->disk[nvalid].dev = d.dev;

		if (readsects(&d, 0, 1, buf, 0)) {
			bibg->disk[nvalid].flags |= BI_GEOM_INVALID;
			nvalid++;
			continue;
		}

#ifdef GEOM_DEBUG
		printf("#%d: %x: C %d H %d S %d\n", nvalid,
		       d.dev, d.cyl, d.head, d.sec);
		printf("   sz %d fl %x cyl %d head %d sec %d totsec %"PRId64" sbytes %d\n",
		       ed.size, ed.flags, ed.cyl, ed.head, ed.sec,
		       ed.totsec, ed.sbytes);
#endif

		if (d.flags & BIOSDISK_INT13EXT) {
			bibg->disk[nvalid].totsec = ed.totsec;
			bibg->disk[nvalid].flags |= BI_GEOM_EXTINT13;
		}
#ifdef BIOSDISK_EXTINFO_V3
#ifdef GEOM_DEBUG
		printf("   edd_cfg %x, sig %x, len %x, bus %s type %s\n",
		       ed.edd_cfg, ed.devpath_sig, ed.devpath_len,
		       ed.host_bus, ed.iface_type);
#endif

		/* The v3.0 stuff will help identify the disks */
		if (ed.size >= offsetof(struct biosdisk_ext13info, checksum)
		    && ed.devpath_sig == EXTINFO_DEVPATH_SIGNATURE) {
			char *cp;

			for (cp = (void *)&ed.devpath_sig, cksum = 0;
			     cp <= (char *)&ed.checksum; cp++) {
				cksum += *cp;
			}
			if ((cksum & 0xff) != 0)
				bibg->disk[nvalid].flags |= BI_GEOM_BADCKSUM;
#ifdef GEOM_DEBUG
			printf("checksum %x\n", cksum & 0xff);
#endif
			for (j = 0; ; j++) {
				cp = bus_names[j].name;
				if (cp == NULL)
					break;
				if (strncmp(cp, ed.host_bus,
					    sizeof(ed.host_bus)) == 0)
					break;
			}
#ifdef GEOM_DEBUG
			printf("bus %s (%x)\n", cp ? cp : "null",
			       bus_names[j].flag);
#endif
			bibg->disk[nvalid].flags |= bus_names[j].flag;
			for (j = 0; ; j++) {
				cp = iface_names[j].name;
				if (cp == NULL)
					break;
				if (strncmp(cp, ed.iface_type,
					    sizeof(ed.iface_type)) == 0)
					break;
			}
			bibg->disk[nvalid].flags |= iface_names[j].flag;
			/* Dump raw interface path and device path */
			bibg->disk[nvalid].interface_path =
					ed.interface_path.ip_32[0];
			bibg->disk[nvalid].device_path =
					ed.device_path.dp_64[0];
#ifdef GEOM_DEBUG
			printf("device %s (%x) interface %x path %llx\n",
			       cp ? cp : "null",
			       iface_names[j].flag,
			       ed.interface_path.ip_32[0],
			       ed.device_path.dp_64[0]);
#endif
		}
#endif

		for (j = 0, cksum = 0; j < BIOSDISK_DEFAULT_SECSIZE; j++)
			cksum += buf[j];
		bibg->disk[nvalid].cksum = cksum;
		memcpy(bibg->disk[nvalid].mbrparts, &buf[MBR_PART_OFFSET],
		       sizeof(bibg->disk[nvalid].mbrparts));
		nvalid++;
	}

	bibg->num = nvalid;

	if (nvalid < nhd)
		bibg_len -= (nhd - nvalid) * sizeof(struct bi_biosgeom_entry);

	BI_ADD(bibg, BTINFO_BIOSGEOM, bibg_len);
}