tests/virtio-blk-test.c

   1 /*
   2  * QTest testcase for VirtIO Block Device
   3  *
   4  * Copyright (c) 2014 SUSE LINUX Products GmbH
   5  * Copyright (c) 2014 Marc Marí
   6  *
   7  * This work is licensed under the terms of the GNU GPL, version 2 or later.
   8  * See the COPYING file in the top-level directory.
   9  */
  10
  11 #include <glib.h>
  12 #include <string.h>
  13 #include <stdlib.h>
  14 #include <unistd.h>
  15 #include <stdio.h>
  16 #include "libqtest.h"
  17 #include "libqos/virtio.h"
  18 #include "libqos/virtio-pci.h"
  19 #include "libqos/pci-pc.h"
  20 #include "libqos/malloc.h"
  21 #include "libqos/malloc-pc.h"
  22 #include "qemu/bswap.h"
  23
  24 #define QVIRTIO_BLK_F_BARRIER       0x00000001
  25 #define QVIRTIO_BLK_F_SIZE_MAX      0x00000002
  26 #define QVIRTIO_BLK_F_SEG_MAX       0x00000004
  27 #define QVIRTIO_BLK_F_GEOMETRY      0x00000010
  28 #define QVIRTIO_BLK_F_RO            0x00000020
  29 #define QVIRTIO_BLK_F_BLK_SIZE      0x00000040
  30 #define QVIRTIO_BLK_F_SCSI          0x00000080
  31 #define QVIRTIO_BLK_F_WCE           0x00000200
  32 #define QVIRTIO_BLK_F_TOPOLOGY      0x00000400
  33 #define QVIRTIO_BLK_F_CONFIG_WCE    0x00000800
  34
  35 #define QVIRTIO_BLK_T_IN            0
  36 #define QVIRTIO_BLK_T_OUT           1
  37 #define QVIRTIO_BLK_T_SCSI_CMD      2
  38 #define QVIRTIO_BLK_T_SCSI_CMD_OUT  3
  39 #define QVIRTIO_BLK_T_FLUSH         4
  40 #define QVIRTIO_BLK_T_FLUSH_OUT     5
  41 #define QVIRTIO_BLK_T_GET_ID        8
  42
  43 #define TEST_IMAGE_SIZE         (64 * 1024 * 1024)
  44 #define QVIRTIO_BLK_TIMEOUT     100
  45 #define PCI_SLOT                0x04
  46 #define PCI_FN                  0x00
  47
  48 typedef struct QVirtioBlkReq {
  49     uint32_t type;
  50     uint32_t ioprio;
  51     uint64_t sector;
  52     char *data;
  53     uint8_t status;
  54 } QVirtioBlkReq;
  55
  56 static QPCIBus *test_start(void)
  57 {
  58     char cmdline[100];
  59     char tmp_path[] = "/tmp/qtest.XXXXXX";
  60     int fd, ret;
  61
  62     /* Create a temporary raw image */
  63     fd = mkstemp(tmp_path);
  64     g_assert_cmpint(fd, >=, 0);
  65     ret = ftruncate(fd, TEST_IMAGE_SIZE);
  66     g_assert_cmpint(ret, ==, 0);
  67     close(fd);
  68
  69     snprintf(cmdline, 100, "-drive if=none,id=drive0,file=%s "
  70                             "-device virtio-blk-pci,drive=drive0,addr=%x.%x",
  71                             tmp_path, PCI_SLOT, PCI_FN);
  72     qtest_start(cmdline);
  73     unlink(tmp_path);
  74
  75     return qpci_init_pc();
  76 }
  77
  78 static void test_end(void)
  79 {
  80     qtest_end();
  81 }
  82
  83 static QVirtioPCIDevice *virtio_blk_init(QPCIBus *bus)
  84 {
  85     QVirtioPCIDevice *dev;
  86
  87     dev = qvirtio_pci_device_find(bus, QVIRTIO_BLK_DEVICE_ID);
  88     g_assert(dev != NULL);
  89     g_assert_cmphex(dev->vdev.device_type, ==, QVIRTIO_BLK_DEVICE_ID);
  90     g_assert_cmphex(dev->pdev->devfn, ==, ((PCI_SLOT << 3) | PCI_FN));
  91
  92     qvirtio_pci_device_enable(dev);
  93     qvirtio_reset(&qvirtio_pci, &dev->vdev);
  94     qvirtio_set_acknowledge(&qvirtio_pci, &dev->vdev);
  95     qvirtio_set_driver(&qvirtio_pci, &dev->vdev);
  96
  97     return dev;
  98 }
  99
 100 static inline void virtio_blk_fix_request(QVirtioBlkReq *req)
 101 {
 102 #ifdef HOST_WORDS_BIGENDIAN
 103     bool host_endian = true;
 104 #else
 105     bool host_endian = false;
 106 #endif
 107
 108     if (qtest_big_endian() != host_endian) {
 109         req->type = bswap32(req->type);
 110         req->ioprio = bswap32(req->ioprio);
 111         req->sector = bswap64(req->sector);
 112     }
 113 }
 114
 115 static uint64_t virtio_blk_request(QGuestAllocator *alloc, QVirtioBlkReq *req,
 116                                                             uint64_t data_size)
 117 {
 118     uint64_t addr;
 119     uint8_t status = 0xFF;
 120
 121     g_assert_cmpuint(data_size % 512, ==, 0);
 122     addr = guest_alloc(alloc, sizeof(*req) + data_size);
 123
 124     virtio_blk_fix_request(req);
 125
 126     memwrite(addr, req, 16);
 127     memwrite(addr + 16, req->data, data_size);
 128     memwrite(addr + 16 + data_size, &status, sizeof(status));
 129
 130     return addr;
 131 }
 132
 133 static void pci_basic(void)
 134 {
 135     QVirtioPCIDevice *dev;
 136     QPCIBus *bus;
 137     QVirtQueuePCI *vqpci;
 138     QGuestAllocator *alloc;
 139     QVirtioBlkReq req;
 140     void *addr;
 141     uint64_t req_addr;
 142     uint64_t capacity;
 143     uint32_t features;
 144     uint32_t free_head;
 145     uint8_t status;
 146     char *data;
 147
 148     bus = test_start();
 149
 150     dev = virtio_blk_init(bus);
 151
 152     /* MSI-X is not enabled */
 153     addr = dev->addr + QVIRTIO_DEVICE_SPECIFIC_NO_MSIX;
 154
 155     capacity = qvirtio_config_readq(&qvirtio_pci, &dev->vdev, addr);
 156     g_assert_cmpint(capacity, ==, TEST_IMAGE_SIZE / 512);
 157
 158     features = qvirtio_get_features(&qvirtio_pci, &dev->vdev);
 159     features = features & ~(QVIRTIO_F_BAD_FEATURE |
 160                     QVIRTIO_F_RING_INDIRECT_DESC | QVIRTIO_F_RING_EVENT_IDX |
 161                             QVIRTIO_BLK_F_SCSI);
 162     qvirtio_set_features(&qvirtio_pci, &dev->vdev, features);
 163
 164     alloc = pc_alloc_init();
 165     vqpci = (QVirtQueuePCI *)qvirtqueue_setup(&qvirtio_pci, &dev->vdev,
 166                                                                     alloc, 0);
 167
 168     qvirtio_set_driver_ok(&qvirtio_pci, &dev->vdev);
 169
 170     /* Write and read with 2 descriptor layout */
 171     /* Write request */
 172     req.type = QVIRTIO_BLK_T_OUT;
 173     req.ioprio = 1;
 174     req.sector = 0;
 175     req.data = g_malloc0(512);
 176     strcpy(req.data, "TEST");
 177
 178     req_addr = virtio_blk_request(alloc, &req, 512);
 179
 180     g_free(req.data);
 181
 182     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 528, false, true);
 183     qvirtqueue_add(&vqpci->vq, req_addr + 528, 1, true, false);
 184     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 185
 186     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 187                                                         QVIRTIO_BLK_TIMEOUT));
 188     status = readb(req_addr + 528);
 189     g_assert_cmpint(status, ==, 0);
 190
 191     guest_free(alloc, req_addr);
 192
 193     /* Read request */
 194     req.type = QVIRTIO_BLK_T_IN;
 195     req.ioprio = 1;
 196     req.sector = 0;
 197     req.data = g_malloc0(512);
 198
 199     req_addr = virtio_blk_request(alloc, &req, 512);
 200
 201     g_free(req.data);
 202
 203     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 16, false, true);
 204     qvirtqueue_add(&vqpci->vq, req_addr + 16, 513, true, false);
 205
 206     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 207
 208     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 209                                                         QVIRTIO_BLK_TIMEOUT));
 210     status = readb(req_addr + 528);
 211     g_assert_cmpint(status, ==, 0);
 212
 213     data = g_malloc0(512);
 214     memread(req_addr + 16, data, 512);
 215     g_assert_cmpstr(data, ==, "TEST");
 216     g_free(data);
 217
 218     guest_free(alloc, req_addr);
 219
 220     /* Write and read with 3 descriptor layout */
 221     /* Write request */
 222     req.type = QVIRTIO_BLK_T_OUT;
 223     req.ioprio = 1;
 224     req.sector = 1;
 225     req.data = g_malloc0(512);
 226     strcpy(req.data, "TEST");
 227
 228     req_addr = virtio_blk_request(alloc, &req, 512);
 229
 230     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 16, false, true);
 231     qvirtqueue_add(&vqpci->vq, req_addr + 16, 512, false, true);
 232     qvirtqueue_add(&vqpci->vq, req_addr + 528, 1, true, false);
 233
 234     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 235
 236     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 237                                                         QVIRTIO_BLK_TIMEOUT));
 238     status = readb(req_addr + 528);
 239     g_assert_cmpint(status, ==, 0);
 240
 241     guest_free(alloc, req_addr);
 242
 243     /* Read request */
 244     req.type = QVIRTIO_BLK_T_IN;
 245     req.ioprio = 1;
 246     req.sector = 1;
 247     req.data = g_malloc0(512);
 248
 249     req_addr = virtio_blk_request(alloc, &req, 512);
 250
 251     g_free(req.data);
 252
 253     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 16, false, true);
 254     qvirtqueue_add(&vqpci->vq, req_addr + 16, 512, true, true);
 255     qvirtqueue_add(&vqpci->vq, req_addr + 528, 1, true, false);
 256
 257     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 258
 259     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 260                                                         QVIRTIO_BLK_TIMEOUT));
 261     status = readb(req_addr + 528);
 262     g_assert_cmpint(status, ==, 0);
 263
 264     data = g_malloc0(512);
 265     memread(req_addr + 16, data, 512);
 266     g_assert_cmpstr(data, ==, "TEST");
 267     g_free(data);
 268
 269     guest_free(alloc, req_addr);
 270
 271     /* End test */
 272     guest_free(alloc, vqpci->vq.desc);
 273     qvirtio_pci_device_disable(dev);
 274     g_free(dev);
 275     test_end();
 276 }
 277
 278 static void pci_indirect(void)
 279 {
 280     QVirtioPCIDevice *dev;
 281     QPCIBus *bus;
 282     QVirtQueuePCI *vqpci;
 283     QGuestAllocator *alloc;
 284     QVirtioBlkReq req;
 285     QVRingIndirectDesc *indirect;
 286     void *addr;
 287     uint64_t req_addr;
 288     uint64_t capacity;
 289     uint32_t features;
 290     uint32_t free_head;
 291     uint8_t status;
 292     char *data;
 293
 294     bus = test_start();
 295
 296     dev = virtio_blk_init(bus);
 297
 298     /* MSI-X is not enabled */
 299     addr = dev->addr + QVIRTIO_DEVICE_SPECIFIC_NO_MSIX;
 300
 301     capacity = qvirtio_config_readq(&qvirtio_pci, &dev->vdev, addr);
 302     g_assert_cmpint(capacity, ==, TEST_IMAGE_SIZE / 512);
 303
 304     features = qvirtio_get_features(&qvirtio_pci, &dev->vdev);
 305     g_assert_cmphex(features & QVIRTIO_F_RING_INDIRECT_DESC, !=, 0);
 306     features = features & ~(QVIRTIO_F_BAD_FEATURE | QVIRTIO_F_RING_EVENT_IDX |
 307                                                             QVIRTIO_BLK_F_SCSI);
 308     qvirtio_set_features(&qvirtio_pci, &dev->vdev, features);
 309
 310     alloc = pc_alloc_init();
 311     vqpci = (QVirtQueuePCI *)qvirtqueue_setup(&qvirtio_pci, &dev->vdev,
 312                                                                     alloc, 0);
 313     qvirtio_set_driver_ok(&qvirtio_pci, &dev->vdev);
 314
 315     /* Write request */
 316     req.type = QVIRTIO_BLK_T_OUT;
 317     req.ioprio = 1;
 318     req.sector = 0;
 319     req.data = g_malloc0(512);
 320     strcpy(req.data, "TEST");
 321
 322     req_addr = virtio_blk_request(alloc, &req, 512);
 323
 324     g_free(req.data);
 325
 326     indirect = qvring_indirect_desc_setup(&dev->vdev, alloc, 2);
 327     qvring_indirect_desc_add(indirect, req_addr, 528, false);
 328     qvring_indirect_desc_add(indirect, req_addr + 528, 1, true);
 329     free_head = qvirtqueue_add_indirect(&vqpci->vq, indirect);
 330     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 331
 332     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 333                                                         QVIRTIO_BLK_TIMEOUT));
 334     status = readb(req_addr + 528);
 335     g_assert_cmpint(status, ==, 0);
 336
 337     g_free(indirect);
 338     guest_free(alloc, req_addr);
 339
 340     /* Read request */
 341     req.type = QVIRTIO_BLK_T_IN;
 342     req.ioprio = 1;
 343     req.sector = 0;
 344     req.data = g_malloc0(512);
 345     strcpy(req.data, "TEST");
 346
 347     req_addr = virtio_blk_request(alloc, &req, 512);
 348
 349     g_free(req.data);
 350
 351     indirect = qvring_indirect_desc_setup(&dev->vdev, alloc, 2);
 352     qvring_indirect_desc_add(indirect, req_addr, 16, false);
 353     qvring_indirect_desc_add(indirect, req_addr + 16, 513, true);
 354     free_head = qvirtqueue_add_indirect(&vqpci->vq, indirect);
 355     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 356
 357     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 358                                                         QVIRTIO_BLK_TIMEOUT));
 359     status = readb(req_addr + 528);
 360     g_assert_cmpint(status, ==, 0);
 361
 362     data = g_malloc0(512);
 363     memread(req_addr + 16, data, 512);
 364     g_assert_cmpstr(data, ==, "TEST");
 365     g_free(data);
 366
 367     g_free(indirect);
 368     guest_free(alloc, req_addr);
 369
 370     /* End test */
 371     guest_free(alloc, vqpci->vq.desc);
 372     qvirtio_pci_device_disable(dev);
 373     g_free(dev);
 374     test_end();
 375 }
 376
 377 static void pci_config(void)
 378 {
 379     QVirtioPCIDevice *dev;
 380     QPCIBus *bus;
 381     int n_size = TEST_IMAGE_SIZE / 2;
 382     void *addr;
 383     uint64_t capacity;
 384
 385     bus = test_start();
 386
 387     dev = virtio_blk_init(bus);
 388
 389     /* MSI-X is not enabled */
 390     addr = dev->addr + QVIRTIO_DEVICE_SPECIFIC_NO_MSIX;
 391
 392     capacity = qvirtio_config_readq(&qvirtio_pci, &dev->vdev, addr);
 393     g_assert_cmpint(capacity, ==, TEST_IMAGE_SIZE / 512);
 394
 395     qvirtio_set_driver_ok(&qvirtio_pci, &dev->vdev);
 396
 397     qmp("{ 'execute': 'block_resize', 'arguments': { 'device': 'drive0', "
 398                                                     " 'size': %d } }", n_size);
 399     g_assert(qvirtio_wait_config_isr(&qvirtio_pci, &dev->vdev,
 400                                                         QVIRTIO_BLK_TIMEOUT));
 401
 402     capacity = qvirtio_config_readq(&qvirtio_pci, &dev->vdev, addr);
 403     g_assert_cmpint(capacity, ==, n_size / 512);
 404
 405     qvirtio_pci_device_disable(dev);
 406     g_free(dev);
 407     test_end();
 408 }
 409
 410 static void pci_msix(void)
 411 {
 412     QVirtioPCIDevice *dev;
 413     QPCIBus *bus;
 414     QVirtQueuePCI *vqpci;
 415     QGuestAllocator *alloc;
 416     QVirtioBlkReq req;
 417     int n_size = TEST_IMAGE_SIZE / 2;
 418     void *addr;
 419     uint64_t req_addr;
 420     uint64_t capacity;
 421     uint32_t features;
 422     uint32_t free_head;
 423     uint8_t status;
 424     char *data;
 425
 426     bus = test_start();
 427     alloc = pc_alloc_init();
 428
 429     dev = virtio_blk_init(bus);
 430     qpci_msix_enable(dev->pdev);
 431
 432     qvirtio_pci_set_msix_configuration_vector(dev, alloc, 0);
 433
 434     /* MSI-X is enabled */
 435     addr = dev->addr + QVIRTIO_DEVICE_SPECIFIC_MSIX;
 436
 437     capacity = qvirtio_config_readq(&qvirtio_pci, &dev->vdev, addr);
 438     g_assert_cmpint(capacity, ==, TEST_IMAGE_SIZE / 512);
 439
 440     features = qvirtio_get_features(&qvirtio_pci, &dev->vdev);
 441     features = features & ~(QVIRTIO_F_BAD_FEATURE |
 442                             QVIRTIO_F_RING_INDIRECT_DESC |
 443                             QVIRTIO_F_RING_EVENT_IDX | QVIRTIO_BLK_F_SCSI);
 444     qvirtio_set_features(&qvirtio_pci, &dev->vdev, features);
 445
 446     vqpci = (QVirtQueuePCI *)qvirtqueue_setup(&qvirtio_pci, &dev->vdev,
 447                                                                     alloc, 0);
 448     qvirtqueue_pci_msix_setup(dev, vqpci, alloc, 1);
 449
 450     qvirtio_set_driver_ok(&qvirtio_pci, &dev->vdev);
 451
 452     qmp("{ 'execute': 'block_resize', 'arguments': { 'device': 'drive0', "
 453                                                     " 'size': %d } }", n_size);
 454
 455     g_assert(qvirtio_wait_config_isr(&qvirtio_pci, &dev->vdev,
 456                                                         QVIRTIO_BLK_TIMEOUT));
 457
 458     capacity = qvirtio_config_readq(&qvirtio_pci, &dev->vdev, addr);
 459     g_assert_cmpint(capacity, ==, n_size / 512);
 460
 461     /* Write request */
 462     req.type = QVIRTIO_BLK_T_OUT;
 463     req.ioprio = 1;
 464     req.sector = 0;
 465     req.data = g_malloc0(512);
 466     strcpy(req.data, "TEST");
 467
 468     req_addr = virtio_blk_request(alloc, &req, 512);
 469
 470     g_free(req.data);
 471
 472     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 528, false, true);
 473     qvirtqueue_add(&vqpci->vq, req_addr + 528, 1, true, false);
 474     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 475
 476     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 477                                                         QVIRTIO_BLK_TIMEOUT));
 478
 479     status = readb(req_addr + 528);
 480     g_assert_cmpint(status, ==, 0);
 481
 482     guest_free(alloc, req_addr);
 483
 484     /* Read request */
 485     req.type = QVIRTIO_BLK_T_IN;
 486     req.ioprio = 1;
 487     req.sector = 0;
 488     req.data = g_malloc0(512);
 489
 490     req_addr = virtio_blk_request(alloc, &req, 512);
 491
 492     g_free(req.data);
 493
 494     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 16, false, true);
 495     qvirtqueue_add(&vqpci->vq, req_addr + 16, 513, true, false);
 496
 497     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 498
 499
 500     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 501                                                         QVIRTIO_BLK_TIMEOUT));
 502
 503     status = readb(req_addr + 528);
 504     g_assert_cmpint(status, ==, 0);
 505
 506     data = g_malloc0(512);
 507     memread(req_addr + 16, data, 512);
 508     g_assert_cmpstr(data, ==, "TEST");
 509     g_free(data);
 510
 511     guest_free(alloc, req_addr);
 512
 513     /* End test */
 514     guest_free(alloc, (uint64_t)vqpci->vq.desc);
 515     qpci_msix_disable(dev->pdev);
 516     qvirtio_pci_device_disable(dev);
 517     g_free(dev);
 518     test_end();
 519 }
 520
 521 static void pci_idx(void)
 522 {
 523     QVirtioPCIDevice *dev;
 524     QPCIBus *bus;
 525     QVirtQueuePCI *vqpci;
 526     QGuestAllocator *alloc;
 527     QVirtioBlkReq req;
 528     void *addr;
 529     uint64_t req_addr;
 530     uint64_t capacity;
 531     uint32_t features;
 532     uint32_t free_head;
 533     uint8_t status;
 534     char *data;
 535
 536     bus = test_start();
 537     alloc = pc_alloc_init();
 538
 539     dev = virtio_blk_init(bus);
 540     qpci_msix_enable(dev->pdev);
 541
 542     qvirtio_pci_set_msix_configuration_vector(dev, alloc, 0);
 543
 544     /* MSI-X is enabled */
 545     addr = dev->addr + QVIRTIO_DEVICE_SPECIFIC_MSIX;
 546
 547     capacity = qvirtio_config_readq(&qvirtio_pci, &dev->vdev, addr);
 548     g_assert_cmpint(capacity, ==, TEST_IMAGE_SIZE / 512);
 549
 550     features = qvirtio_get_features(&qvirtio_pci, &dev->vdev);
 551     features = features & ~(QVIRTIO_F_BAD_FEATURE |
 552                             QVIRTIO_F_RING_INDIRECT_DESC |
 553                             QVIRTIO_F_NOTIFY_ON_EMPTY | QVIRTIO_BLK_F_SCSI);
 554     qvirtio_set_features(&qvirtio_pci, &dev->vdev, features);
 555
 556     vqpci = (QVirtQueuePCI *)qvirtqueue_setup(&qvirtio_pci, &dev->vdev,
 557                                                                     alloc, 0);
 558     qvirtqueue_pci_msix_setup(dev, vqpci, alloc, 1);
 559
 560     qvirtio_set_driver_ok(&qvirtio_pci, &dev->vdev);
 561
 562     /* Write request */
 563     req.type = QVIRTIO_BLK_T_OUT;
 564     req.ioprio = 1;
 565     req.sector = 0;
 566     req.data = g_malloc0(512);
 567     strcpy(req.data, "TEST");
 568
 569     req_addr = virtio_blk_request(alloc, &req, 512);
 570
 571     g_free(req.data);
 572
 573     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 528, false, true);
 574     qvirtqueue_add(&vqpci->vq, req_addr + 528, 1, true, false);
 575     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 576
 577     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 578                                                         QVIRTIO_BLK_TIMEOUT));
 579
 580     /* Write request */
 581     req.type = QVIRTIO_BLK_T_OUT;
 582     req.ioprio = 1;
 583     req.sector = 1;
 584     req.data = g_malloc0(512);
 585     strcpy(req.data, "TEST");
 586
 587     req_addr = virtio_blk_request(alloc, &req, 512);
 588
 589     g_free(req.data);
 590
 591     /* Notify after processing the third request */
 592     qvirtqueue_set_used_event(&vqpci->vq, 2);
 593     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 528, false, true);
 594     qvirtqueue_add(&vqpci->vq, req_addr + 528, 1, true, false);
 595     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 596
 597     /* No notification expected */
 598     g_assert(!qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 599                                                         QVIRTIO_BLK_TIMEOUT));
 600
 601     status = readb(req_addr + 528);
 602     g_assert_cmpint(status, ==, 0);
 603
 604     guest_free(alloc, req_addr);
 605
 606     /* Read request */
 607     req.type = QVIRTIO_BLK_T_IN;
 608     req.ioprio = 1;
 609     req.sector = 1;
 610     req.data = g_malloc0(512);
 611
 612     req_addr = virtio_blk_request(alloc, &req, 512);
 613
 614     g_free(req.data);
 615
 616     free_head = qvirtqueue_add(&vqpci->vq, req_addr, 16, false, true);
 617     qvirtqueue_add(&vqpci->vq, req_addr + 16, 513, true, false);
 618
 619     qvirtqueue_kick(&qvirtio_pci, &dev->vdev, &vqpci->vq, free_head);
 620
 621
 622     g_assert(qvirtio_wait_queue_isr(&qvirtio_pci, &dev->vdev, &vqpci->vq,
 623                                                         QVIRTIO_BLK_TIMEOUT));
 624
 625     status = readb(req_addr + 528);
 626     g_assert_cmpint(status, ==, 0);
 627
 628     data = g_malloc0(512);
 629     memread(req_addr + 16, data, 512);
 630     g_assert_cmpstr(data, ==, "TEST");
 631     g_free(data);
 632
 633     guest_free(alloc, req_addr);
 634
 635     /* End test */
 636     guest_free(alloc, vqpci->vq.desc);
 637     qpci_msix_disable(dev->pdev);
 638     qvirtio_pci_device_disable(dev);
 639     g_free(dev);
 640     test_end();
 641 }
 642
 643 int main(int argc, char **argv)
 644 {
 645     int ret;
 646
 647     g_test_init(&argc, &argv, NULL);
 648
 649     g_test_add_func("/virtio/blk/pci/basic", pci_basic);
 650     g_test_add_func("/virtio/blk/pci/indirect", pci_indirect);
 651     g_test_add_func("/virtio/blk/pci/config", pci_config);
 652     g_test_add_func("/virtio/blk/pci/msix", pci_msix);
 653     g_test_add_func("/virtio/blk/pci/idx", pci_idx);
 654
 655     ret = g_test_run();
 656
 657     return ret;
 658 }