214 lines
6.4 KiB
C
214 lines
6.4 KiB
C
// 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 "biosvar.h" // GET_GLOBALFLAT
|
|
#include "config.h" // CONFIG_*
|
|
#include "block.h" // struct drive_s
|
|
#include "malloc.h" // free
|
|
#include "output.h" // dprintf
|
|
#include "pcidevice.h" // foreachpci
|
|
#include "pci_ids.h" // PCI_DEVICE_ID_VIRTIO_BLK
|
|
#include "pci_regs.h" // PCI_VENDOR_ID
|
|
#include "stacks.h" // run_thread
|
|
#include "std/disk.h" // DISK_RET_SUCCESS
|
|
#include "string.h" // memset
|
|
#include "util.h" // usleep
|
|
#include "virtio-pci.h"
|
|
#include "virtio-ring.h"
|
|
#include "virtio-blk.h"
|
|
|
|
struct virtiodrive_s {
|
|
struct drive_s drive;
|
|
struct vring_virtqueue *vq;
|
|
struct vp_device vp;
|
|
};
|
|
|
|
static int
|
|
virtio_blk_op(struct disk_op_s *op, int write)
|
|
{
|
|
struct virtiodrive_s *vdrive =
|
|
container_of(op->drive_fl, struct virtiodrive_s, drive);
|
|
struct vring_virtqueue *vq = vdrive->vq;
|
|
struct virtio_blk_outhdr hdr = {
|
|
.type = write ? VIRTIO_BLK_T_OUT : VIRTIO_BLK_T_IN,
|
|
.ioprio = 0,
|
|
.sector = op->lba,
|
|
};
|
|
u8 status = VIRTIO_BLK_S_UNSUPP;
|
|
struct vring_list sg[] = {
|
|
{
|
|
.addr = (void*)(&hdr),
|
|
.length = sizeof(hdr),
|
|
},
|
|
{
|
|
.addr = op->buf_fl,
|
|
.length = vdrive->drive.blksize * op->count,
|
|
},
|
|
{
|
|
.addr = (void*)(&status),
|
|
.length = sizeof(status),
|
|
},
|
|
};
|
|
|
|
/* Add to virtqueue and kick host */
|
|
if (write)
|
|
vring_add_buf(vq, sg, 2, 1, 0, 0);
|
|
else
|
|
vring_add_buf(vq, sg, 1, 2, 0, 0);
|
|
vring_kick(&vdrive->vp, vq, 1);
|
|
|
|
/* Wait for reply */
|
|
while (!vring_more_used(vq))
|
|
usleep(5);
|
|
|
|
/* Reclaim virtqueue element */
|
|
vring_get_buf(vq, NULL);
|
|
|
|
/* Clear interrupt status register. Avoid leaving interrupts stuck if
|
|
* VRING_AVAIL_F_NO_INTERRUPT was ignored and interrupts were raised.
|
|
*/
|
|
vp_get_isr(&vdrive->vp);
|
|
|
|
return status == VIRTIO_BLK_S_OK ? DISK_RET_SUCCESS : DISK_RET_EBADTRACK;
|
|
}
|
|
|
|
int
|
|
virtio_blk_process_op(struct disk_op_s *op)
|
|
{
|
|
if (! CONFIG_VIRTIO_BLK)
|
|
return 0;
|
|
switch (op->command) {
|
|
case CMD_READ:
|
|
return virtio_blk_op(op, 0);
|
|
case CMD_WRITE:
|
|
return virtio_blk_op(op, 1);
|
|
default:
|
|
return default_process_op(op);
|
|
}
|
|
}
|
|
|
|
static void
|
|
init_virtio_blk(void *data)
|
|
{
|
|
struct pci_device *pci = data;
|
|
u8 status = VIRTIO_CONFIG_S_ACKNOWLEDGE | VIRTIO_CONFIG_S_DRIVER;
|
|
dprintf(1, "found virtio-blk at %pP\n", pci);
|
|
struct virtiodrive_s *vdrive = malloc_low(sizeof(*vdrive));
|
|
if (!vdrive) {
|
|
warn_noalloc();
|
|
return;
|
|
}
|
|
memset(vdrive, 0, sizeof(*vdrive));
|
|
vdrive->drive.type = DTYPE_VIRTIO_BLK;
|
|
vdrive->drive.cntl_id = pci->bdf;
|
|
|
|
vp_init_simple(&vdrive->vp, pci);
|
|
if (vp_find_vq(&vdrive->vp, 0, &vdrive->vq) < 0 ) {
|
|
dprintf(1, "fail to find vq for virtio-blk %pP\n", pci);
|
|
goto fail;
|
|
}
|
|
|
|
if (vdrive->vp.use_modern) {
|
|
struct vp_device *vp = &vdrive->vp;
|
|
u64 features = vp_get_features(vp);
|
|
u64 version1 = 1ull << VIRTIO_F_VERSION_1;
|
|
u64 iommu_platform = 1ull << VIRTIO_F_IOMMU_PLATFORM;
|
|
u64 blk_size = 1ull << VIRTIO_BLK_F_BLK_SIZE;
|
|
if (!(features & version1)) {
|
|
dprintf(1, "modern device without virtio_1 feature bit: %pP\n", pci);
|
|
goto fail;
|
|
}
|
|
|
|
features = features & (version1 | iommu_platform | blk_size);
|
|
vp_set_features(vp, features);
|
|
status |= VIRTIO_CONFIG_S_FEATURES_OK;
|
|
vp_set_status(vp, status);
|
|
if (!(vp_get_status(vp) & VIRTIO_CONFIG_S_FEATURES_OK)) {
|
|
dprintf(1, "device didn't accept features: %pP\n", pci);
|
|
goto fail;
|
|
}
|
|
|
|
vdrive->drive.sectors =
|
|
vp_read(&vp->device, struct virtio_blk_config, capacity);
|
|
if (features & blk_size) {
|
|
vdrive->drive.blksize =
|
|
vp_read(&vp->device, struct virtio_blk_config, blk_size);
|
|
} else {
|
|
vdrive->drive.blksize = DISK_SECTOR_SIZE;
|
|
}
|
|
if (vdrive->drive.blksize != DISK_SECTOR_SIZE) {
|
|
dprintf(1, "virtio-blk %pP block size %d is unsupported\n",
|
|
pci, vdrive->drive.blksize);
|
|
goto fail;
|
|
}
|
|
dprintf(3, "virtio-blk %pP blksize=%d sectors=%u\n",
|
|
pci, vdrive->drive.blksize, (u32)vdrive->drive.sectors);
|
|
|
|
vdrive->drive.pchs.cylinder =
|
|
vp_read(&vp->device, struct virtio_blk_config, cylinders);
|
|
vdrive->drive.pchs.head =
|
|
vp_read(&vp->device, struct virtio_blk_config, heads);
|
|
vdrive->drive.pchs.sector =
|
|
vp_read(&vp->device, struct virtio_blk_config, sectors);
|
|
} else {
|
|
struct virtio_blk_config cfg;
|
|
vp_get_legacy(&vdrive->vp, 0, &cfg, sizeof(cfg));
|
|
|
|
u64 f = vp_get_features(&vdrive->vp);
|
|
vdrive->drive.blksize = (f & (1 << VIRTIO_BLK_F_BLK_SIZE)) ?
|
|
cfg.blk_size : DISK_SECTOR_SIZE;
|
|
|
|
vdrive->drive.sectors = cfg.capacity;
|
|
dprintf(3, "virtio-blk %pP blksize=%d sectors=%u\n",
|
|
pci, vdrive->drive.blksize, (u32)vdrive->drive.sectors);
|
|
|
|
if (vdrive->drive.blksize != DISK_SECTOR_SIZE) {
|
|
dprintf(1, "virtio-blk %pP block size %d is unsupported\n",
|
|
pci, vdrive->drive.blksize);
|
|
goto fail;
|
|
}
|
|
vdrive->drive.pchs.cylinder = cfg.cylinders;
|
|
vdrive->drive.pchs.head = cfg.heads;
|
|
vdrive->drive.pchs.sector = cfg.sectors;
|
|
}
|
|
|
|
char *desc = znprintf(MAXDESCSIZE, "Virtio disk PCI:%pP", pci);
|
|
boot_add_hd(&vdrive->drive, desc, bootprio_find_pci_device(pci));
|
|
|
|
status |= VIRTIO_CONFIG_S_DRIVER_OK;
|
|
vp_set_status(&vdrive->vp, status);
|
|
|
|
boot_lchs_find_pci_device(pci, &vdrive->drive.lchs);
|
|
return;
|
|
|
|
fail:
|
|
vp_reset(&vdrive->vp);
|
|
free(vdrive->vq);
|
|
free(vdrive);
|
|
}
|
|
|
|
void
|
|
virtio_blk_setup(void)
|
|
{
|
|
ASSERT32FLAT();
|
|
if (! CONFIG_VIRTIO_BLK)
|
|
return;
|
|
|
|
dprintf(3, "init virtio-blk\n");
|
|
|
|
struct pci_device *pci;
|
|
foreachpci(pci) {
|
|
if (pci->vendor != PCI_VENDOR_ID_REDHAT_QUMRANET ||
|
|
(pci->device != PCI_DEVICE_ID_VIRTIO_BLK_09 &&
|
|
pci->device != PCI_DEVICE_ID_VIRTIO_BLK_10))
|
|
continue;
|
|
run_thread(init_virtio_blk, pci);
|
|
}
|
|
}
|