/*      $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);
}