// Virtio block boot support.
//
// Copyright (C) 2010 Red Hat Inc.
//
// Authors:
//  Gleb Natapov <gnatapov@redhat.com>
//
// This file may be distributed under the terms of the GNU LGPLv3 license.

#include "util.h" // dprintf
#include "pci.h" // foreachpci
#include "config.h" // CONFIG_*
#include "biosvar.h" // GET_GLOBAL
#include "pci_ids.h" // PCI_DEVICE_ID_VIRTIO_BLK
#include "pci_regs.h" // PCI_VENDOR_ID
#include "boot.h" // add_bcv_internal
#include "virtio-pci.h"
#include "virtio-ring.h"
#include "virtio-blk.h"
#include "disk.h"

struct virtiodrive_s {
    struct drive_s drive;
    struct vring_virtqueue *vq;
    u16 ioaddr;
};

static int
virtio_blk_read(struct disk_op_s *op)
{
    struct virtiodrive_s *vdrive_g =
        container_of(op->drive_g, struct virtiodrive_s, drive);
    struct vring_virtqueue *vq = GET_GLOBAL(vdrive_g->vq);
    struct virtio_blk_outhdr hdr = {
        .type = VIRTIO_BLK_T_IN,
        .ioprio = 0,
        .sector = op->lba,
    };
    u8 status = VIRTIO_BLK_S_UNSUPP;
    struct vring_list sg[] = {
        {
            .addr	= MAKE_FLATPTR(GET_SEG(SS), &hdr),
            .length	= sizeof(hdr),
        },
        {
            .addr	= op->buf_fl,
            .length	= GET_GLOBAL(vdrive_g->drive.blksize) * op->count,
        },
        {
            .addr	= MAKE_FLATPTR(GET_SEG(SS), &status),
            .length	= sizeof(status),
        },
    };

    /* Add to virtqueue and kick host */
    vring_add_buf(vq, sg, 1, 2, 0, 0);
    vring_kick(GET_GLOBAL(vdrive_g->ioaddr), vq, 1);

    /* Wait for reply */
    while (!vring_more_used(vq))
        usleep(5);

    /* Reclaim virtqueue element */
    vring_get_buf(vq, NULL);
    return status == VIRTIO_BLK_S_OK ? DISK_RET_SUCCESS : DISK_RET_EBADTRACK;
}

int
process_virtio_op(struct disk_op_s *op)
{
    switch (op->command) {
    case CMD_READ:
        return virtio_blk_read(op);
    case CMD_FORMAT:
    case CMD_WRITE:
        return DISK_RET_EWRITEPROTECT;
    case CMD_RESET:
    case CMD_ISREADY:
    case CMD_VERIFY:
    case CMD_SEEK:
        return DISK_RET_SUCCESS;
    default:
        op->count = 0;
        return DISK_RET_EPARAM;
    }
}

void
virtio_blk_setup(void)
{
    ASSERT32FLAT();
    if (! CONFIG_VIRTIO_BLK)
        return;

    dprintf(3, "init virtio-blk\n");

    int bdf, max;
    u32 id = PCI_VENDOR_ID_REDHAT_QUMRANET | (PCI_DEVICE_ID_VIRTIO_BLK << 16);
    foreachpci(bdf, max) {
        u32 v = pci_config_readl(bdf, PCI_VENDOR_ID);
        if (v != id)
            continue;
        dprintf(3, "found virtio-blk at %x:%x\n", pci_bdf_to_bus(bdf),
                pci_bdf_to_dev(bdf));
        char *desc = malloc_tmphigh(MAXDESCSIZE);
        struct virtiodrive_s *vdrive_g = malloc_fseg(sizeof(*vdrive_g));
        struct vring_virtqueue *vq = memalign_low(PAGE_SIZE, sizeof(*vq));
        if (!vdrive_g || !desc || !vq) {
            free(vdrive_g);
            free(desc);
            free(vq);
            warn_noalloc();
            return;
        }
        memset(vdrive_g, 0, sizeof(*vdrive_g));
        vdrive_g->drive.type = DTYPE_VIRTIO;
        vdrive_g->drive.cntl_id = bdf;
        vdrive_g->vq = vq;

        u16 ioaddr = pci_config_readl(bdf, PCI_BASE_ADDRESS_0) &
            PCI_BASE_ADDRESS_IO_MASK;

        vdrive_g->ioaddr = ioaddr;

        vp_reset(ioaddr);
        vp_set_status(ioaddr, VIRTIO_CONFIG_S_ACKNOWLEDGE |
                      VIRTIO_CONFIG_S_DRIVER );

        if (vp_find_vq(ioaddr, 0, vdrive_g->vq) < 0 ) {
            free(vdrive_g);
            free(desc);
            free(vq);
            dprintf(1, "fail to find vq for virtio-blk %x:%x\n",
                    pci_bdf_to_bus(bdf), pci_bdf_to_dev(bdf));
            continue;
        }

        struct virtio_blk_config cfg;
        vp_get(ioaddr, 0, &cfg, sizeof(cfg));

        vdrive_g->drive.blksize = cfg.blk_size;
        vdrive_g->drive.sectors = cfg.capacity;
        dprintf(3, "virtio-blk %x:%x blksize=%d sectors=%u\n",
                pci_bdf_to_bus(bdf), pci_bdf_to_dev(bdf),
                vdrive_g->drive.blksize, (u32)vdrive_g->drive.sectors);

        vdrive_g->drive.pchs.cylinders = cfg.cylinders;
        vdrive_g->drive.pchs.heads = cfg.heads;
        vdrive_g->drive.pchs.spt = cfg.sectors;

        setup_translation(&vdrive_g->drive);
        add_bcv_internal(&vdrive_g->drive);

        snprintf(desc, MAXDESCSIZE, "Virtio disk PCI:%x:%x",
                 pci_bdf_to_bus(bdf), pci_bdf_to_dev(bdf));

        vdrive_g->drive.desc = desc;

        vp_set_status(ioaddr, VIRTIO_CONFIG_S_ACKNOWLEDGE |
                      VIRTIO_CONFIG_S_DRIVER | VIRTIO_CONFIG_S_DRIVER_OK);
    }
}