1 // Virtio block boot support.
3 // Copyright (C) 2010 Red Hat Inc.
6 // Gleb Natapov <gnatapov@redhat.com>
8 // This file may be distributed under the terms of the GNU LGPLv3 license.
10 #include "util.h" // dprintf
11 #include "pci.h" // foreachpci
12 #include "config.h" // CONFIG_*
13 #include "biosvar.h" // GET_GLOBAL
14 #include "pci_ids.h" // PCI_DEVICE_ID_VIRTIO_BLK
15 #include "pci_regs.h" // PCI_VENDOR_ID
16 #include "boot.h" // boot_add_hd
17 #include "virtio-pci.h"
18 #include "virtio-ring.h"
19 #include "virtio-blk.h"
22 struct virtiodrive_s {
24 struct vring_virtqueue *vq;
29 virtio_blk_op(struct disk_op_s *op, int write)
31 struct virtiodrive_s *vdrive_g =
32 container_of(op->drive_g, struct virtiodrive_s, drive);
33 struct vring_virtqueue *vq = GET_GLOBAL(vdrive_g->vq);
34 struct virtio_blk_outhdr hdr = {
35 .type = write ? VIRTIO_BLK_T_OUT : VIRTIO_BLK_T_IN,
39 u8 status = VIRTIO_BLK_S_UNSUPP;
40 struct vring_list sg[] = {
42 .addr = MAKE_FLATPTR(GET_SEG(SS), &hdr),
43 .length = sizeof(hdr),
47 .length = GET_GLOBAL(vdrive_g->drive.blksize) * op->count,
50 .addr = MAKE_FLATPTR(GET_SEG(SS), &status),
51 .length = sizeof(status),
55 /* Add to virtqueue and kick host */
57 vring_add_buf(vq, sg, 2, 1, 0, 0);
59 vring_add_buf(vq, sg, 1, 2, 0, 0);
60 vring_kick(GET_GLOBAL(vdrive_g->ioaddr), vq, 1);
63 while (!vring_more_used(vq))
66 /* Reclaim virtqueue element */
67 vring_get_buf(vq, NULL);
69 /* Clear interrupt status register. Avoid leaving interrupts stuck if
70 * VRING_AVAIL_F_NO_INTERRUPT was ignored and interrupts were raised.
72 vp_get_isr(GET_GLOBAL(vdrive_g->ioaddr));
74 return status == VIRTIO_BLK_S_OK ? DISK_RET_SUCCESS : DISK_RET_EBADTRACK;
78 process_virtio_op(struct disk_op_s *op)
80 if (! CONFIG_VIRTIO_BLK || CONFIG_COREBOOT)
82 switch (op->command) {
84 return virtio_blk_op(op, 0);
86 return virtio_blk_op(op, 1);
92 return DISK_RET_SUCCESS;
95 return DISK_RET_EPARAM;
100 init_virtio_blk(u16 bdf)
102 dprintf(1, "found virtio-blk at %x:%x\n", pci_bdf_to_bus(bdf),
103 pci_bdf_to_dev(bdf));
104 struct virtiodrive_s *vdrive_g = malloc_fseg(sizeof(*vdrive_g));
105 struct vring_virtqueue *vq = memalign_low(PAGE_SIZE, sizeof(*vq));
106 if (!vdrive_g || !vq) {
110 memset(vdrive_g, 0, sizeof(*vdrive_g));
111 memset(vq, 0, sizeof(*vq));
112 vdrive_g->drive.type = DTYPE_VIRTIO;
113 vdrive_g->drive.cntl_id = bdf;
116 u16 ioaddr = pci_config_readl(bdf, PCI_BASE_ADDRESS_0) &
117 PCI_BASE_ADDRESS_IO_MASK;
119 vdrive_g->ioaddr = ioaddr;
122 vp_set_status(ioaddr, VIRTIO_CONFIG_S_ACKNOWLEDGE |
123 VIRTIO_CONFIG_S_DRIVER );
125 if (vp_find_vq(ioaddr, 0, vdrive_g->vq) < 0 ) {
126 dprintf(1, "fail to find vq for virtio-blk %x:%x\n",
127 pci_bdf_to_bus(bdf), pci_bdf_to_dev(bdf));
131 struct virtio_blk_config cfg;
132 vp_get(ioaddr, 0, &cfg, sizeof(cfg));
134 u32 f = vp_get_features(ioaddr);
135 vdrive_g->drive.blksize = (f & (1 << VIRTIO_BLK_F_BLK_SIZE)) ?
136 cfg.blk_size : DISK_SECTOR_SIZE;
138 vdrive_g->drive.sectors = cfg.capacity;
139 dprintf(3, "virtio-blk %x:%x blksize=%d sectors=%u\n",
140 pci_bdf_to_bus(bdf), pci_bdf_to_dev(bdf),
141 vdrive_g->drive.blksize, (u32)vdrive_g->drive.sectors);
143 if (vdrive_g->drive.blksize != DISK_SECTOR_SIZE) {
144 dprintf(1, "virtio-blk %x:%x block size %d is unsupported\n",
145 pci_bdf_to_bus(bdf), pci_bdf_to_dev(bdf),
146 vdrive_g->drive.blksize);
150 vdrive_g->drive.pchs.cylinders = cfg.cylinders;
151 vdrive_g->drive.pchs.heads = cfg.heads;
152 vdrive_g->drive.pchs.spt = cfg.sectors;
153 char *desc = znprintf(MAXDESCSIZE, "Virtio disk PCI:%x:%x",
154 pci_bdf_to_bus(bdf), pci_bdf_to_dev(bdf));
156 boot_add_hd(&vdrive_g->drive, desc, bootprio_find_pci_device(bdf));
158 vp_set_status(ioaddr, VIRTIO_CONFIG_S_ACKNOWLEDGE |
159 VIRTIO_CONFIG_S_DRIVER | VIRTIO_CONFIG_S_DRIVER_OK);
168 virtio_blk_setup(void)
171 if (! CONFIG_VIRTIO_BLK || CONFIG_COREBOOT)
174 dprintf(3, "init virtio-blk\n");
176 struct pci_device *pci;
178 if (pci->vendor != PCI_VENDOR_ID_REDHAT_QUMRANET
179 || pci->device != PCI_DEVICE_ID_VIRTIO_BLK)
181 init_virtio_blk(pci->bdf);