tests/qtest/ufs-test.c

   1 /*
   2  * QTest testcase for UFS
   3  *
   4  * Copyright (c) 2023 Samsung Electronics Co., Ltd. All rights reserved.
   5  *
   6  * SPDX-License-Identifier: GPL-2.0-or-later
   7  */
   8
   9 #include "qemu/osdep.h"
  10 #include "qemu/module.h"
  11 #include "qemu/units.h"
  12 #include "libqtest.h"
  13 #include "libqos/qgraph.h"
  14 #include "libqos/pci.h"
  15 #include "scsi/constants.h"
  16 #include "include/block/ufs.h"
  17
  18 /* Test images sizes in Bytes */
  19 #define TEST_IMAGE_SIZE (64 * 1024 * 1024)
  20 /* Timeout for various operations, in seconds. */
  21 #define TIMEOUT_SECONDS 10
  22 /* Maximum PRD entry count */
  23 #define MAX_PRD_ENTRY_COUNT 10
  24 #define PRD_ENTRY_DATA_SIZE 4096
  25 /* Constants to build upiu */
  26 #define UTP_COMMAND_DESCRIPTOR_SIZE 4096
  27 #define UTP_RESPONSE_UPIU_OFFSET 1024
  28 #define UTP_PRDT_UPIU_OFFSET 2048
  29
  30 typedef struct QUfs QUfs;
  31
  32 struct QUfs {
  33     QOSGraphObject obj;
  34     QPCIDevice dev;
  35     QPCIBar bar;
  36
  37     uint64_t utrlba;
  38     uint64_t utmrlba;
  39     uint64_t cmd_desc_addr;
  40     uint64_t data_buffer_addr;
  41
  42     bool enabled;
  43 };
  44
  45 static inline uint32_t ufs_rreg(QUfs *ufs, size_t offset)
  46 {
  47     return qpci_io_readl(&ufs->dev, ufs->bar, offset);
  48 }
  49
  50 static inline void ufs_wreg(QUfs *ufs, size_t offset, uint32_t value)
  51 {
  52     qpci_io_writel(&ufs->dev, ufs->bar, offset, value);
  53 }
  54
  55 static void ufs_wait_for_irq(QUfs *ufs)
  56 {
  57     uint64_t end_time;
  58     uint32_t is;
  59     /* Wait for device to reset as the linux driver does. */
  60     end_time = g_get_monotonic_time() + TIMEOUT_SECONDS * G_TIME_SPAN_SECOND;
  61     do {
  62         qtest_clock_step(ufs->dev.bus->qts, 100);
  63         is = ufs_rreg(ufs, A_IS);
  64     } while (is == 0 && g_get_monotonic_time() < end_time);
  65 }
  66
  67 static UtpTransferReqDesc ufs_build_req_utrd(uint64_t cmd_desc_addr,
  68                                              uint8_t slot,
  69                                              uint32_t data_direction,
  70                                              uint16_t prd_table_length)
  71 {
  72     UtpTransferReqDesc req = { 0 };
  73     uint64_t command_desc_base_addr =
  74         cmd_desc_addr + slot * UTP_COMMAND_DESCRIPTOR_SIZE;
  75
  76     req.header.dword_0 =
  77         cpu_to_le32(1 << 28 | data_direction | UFS_UTP_REQ_DESC_INT_CMD);
  78     req.header.dword_2 = cpu_to_le32(UFS_OCS_INVALID_COMMAND_STATUS);
  79
  80     req.command_desc_base_addr_hi = cpu_to_le32(command_desc_base_addr >> 32);
  81     req.command_desc_base_addr_lo =
  82         cpu_to_le32(command_desc_base_addr & 0xffffffff);
  83     req.response_upiu_offset =
  84         cpu_to_le16(UTP_RESPONSE_UPIU_OFFSET / sizeof(uint32_t));
  85     req.response_upiu_length = cpu_to_le16(sizeof(UtpUpiuRsp));
  86     req.prd_table_offset = cpu_to_le16(UTP_PRDT_UPIU_OFFSET / sizeof(uint32_t));
  87     req.prd_table_length = cpu_to_le16(prd_table_length);
  88     return req;
  89 }
  90
  91 static void ufs_send_nop_out(QUfs *ufs, uint8_t slot,
  92                              UtpTransferReqDesc *utrd_out, UtpUpiuRsp *rsp_out)
  93 {
  94     /* Build up utp transfer request descriptor */
  95     UtpTransferReqDesc utrd = ufs_build_req_utrd(ufs->cmd_desc_addr, slot,
  96                                                  UFS_UTP_NO_DATA_TRANSFER, 0);
  97     uint64_t utrd_addr = ufs->utrlba + slot * sizeof(UtpTransferReqDesc);
  98     uint64_t req_upiu_addr =
  99         ufs->cmd_desc_addr + slot * UTP_COMMAND_DESCRIPTOR_SIZE;
 100     uint64_t rsp_upiu_addr = req_upiu_addr + UTP_RESPONSE_UPIU_OFFSET;
 101     qtest_memwrite(ufs->dev.bus->qts, utrd_addr, &utrd, sizeof(utrd));
 102
 103     /* Build up request upiu */
 104     UtpUpiuReq req_upiu = { 0 };
 105     req_upiu.header.trans_type = UFS_UPIU_TRANSACTION_NOP_OUT;
 106     req_upiu.header.task_tag = slot;
 107     qtest_memwrite(ufs->dev.bus->qts, req_upiu_addr, &req_upiu,
 108                    sizeof(req_upiu));
 109
 110     /* Ring Doorbell */
 111     ufs_wreg(ufs, A_UTRLDBR, 1);
 112     ufs_wait_for_irq(ufs);
 113     g_assert_true(FIELD_EX32(ufs_rreg(ufs, A_IS), IS, UTRCS));
 114     ufs_wreg(ufs, A_IS, FIELD_DP32(0, IS, UTRCS, 1));
 115
 116     qtest_memread(ufs->dev.bus->qts, utrd_addr, utrd_out, sizeof(*utrd_out));
 117     qtest_memread(ufs->dev.bus->qts, rsp_upiu_addr, rsp_out, sizeof(*rsp_out));
 118 }
 119
 120 static void ufs_send_query(QUfs *ufs, uint8_t slot, uint8_t query_function,
 121                            uint8_t query_opcode, uint8_t idn, uint8_t index,
 122                            UtpTransferReqDesc *utrd_out, UtpUpiuRsp *rsp_out)
 123 {
 124     /* Build up utp transfer request descriptor */
 125     UtpTransferReqDesc utrd = ufs_build_req_utrd(ufs->cmd_desc_addr, slot,
 126                                                  UFS_UTP_NO_DATA_TRANSFER, 0);
 127     uint64_t utrd_addr = ufs->utrlba + slot * sizeof(UtpTransferReqDesc);
 128     uint64_t req_upiu_addr =
 129         ufs->cmd_desc_addr + slot * UTP_COMMAND_DESCRIPTOR_SIZE;
 130     uint64_t rsp_upiu_addr = req_upiu_addr + UTP_RESPONSE_UPIU_OFFSET;
 131     qtest_memwrite(ufs->dev.bus->qts, utrd_addr, &utrd, sizeof(utrd));
 132
 133     /* Build up request upiu */
 134     UtpUpiuReq req_upiu = { 0 };
 135     req_upiu.header.trans_type = UFS_UPIU_TRANSACTION_QUERY_REQ;
 136     req_upiu.header.query_func = query_function;
 137     req_upiu.header.task_tag = slot;
 138     /*
 139      * QEMU UFS does not currently support Write descriptor and Write attribute,
 140      * so the value of data_segment_length is always 0.
 141      */
 142     req_upiu.header.data_segment_length = 0;
 143     req_upiu.qr.opcode = query_opcode;
 144     req_upiu.qr.idn = idn;
 145     req_upiu.qr.index = index;
 146     qtest_memwrite(ufs->dev.bus->qts, req_upiu_addr, &req_upiu,
 147                    sizeof(req_upiu));
 148
 149     /* Ring Doorbell */
 150     ufs_wreg(ufs, A_UTRLDBR, 1);
 151     ufs_wait_for_irq(ufs);
 152     g_assert_true(FIELD_EX32(ufs_rreg(ufs, A_IS), IS, UTRCS));
 153     ufs_wreg(ufs, A_IS, FIELD_DP32(0, IS, UTRCS, 1));
 154
 155     qtest_memread(ufs->dev.bus->qts, utrd_addr, utrd_out, sizeof(*utrd_out));
 156     qtest_memread(ufs->dev.bus->qts, rsp_upiu_addr, rsp_out, sizeof(*rsp_out));
 157 }
 158
 159 static void ufs_send_scsi_command(QUfs *ufs, uint8_t slot, uint8_t lun,
 160                                   const uint8_t *cdb, const uint8_t *data_in,
 161                                   size_t data_in_len, uint8_t *data_out,
 162                                   size_t data_out_len,
 163                                   UtpTransferReqDesc *utrd_out,
 164                                   UtpUpiuRsp *rsp_out)
 165
 166 {
 167     /* Build up PRDT */
 168     UfshcdSgEntry entries[MAX_PRD_ENTRY_COUNT] = {
 169         0,
 170     };
 171     uint8_t flags;
 172     uint16_t prd_table_length, i;
 173     uint32_t data_direction, data_len;
 174     uint64_t req_upiu_addr =
 175         ufs->cmd_desc_addr + slot * UTP_COMMAND_DESCRIPTOR_SIZE;
 176     uint64_t prdt_addr = req_upiu_addr + UTP_PRDT_UPIU_OFFSET;
 177
 178     g_assert_true(data_in_len < MAX_PRD_ENTRY_COUNT * PRD_ENTRY_DATA_SIZE);
 179     g_assert_true(data_out_len < MAX_PRD_ENTRY_COUNT * PRD_ENTRY_DATA_SIZE);
 180     if (data_in_len > 0) {
 181         g_assert_nonnull(data_in);
 182         data_direction = UFS_UTP_HOST_TO_DEVICE;
 183         data_len = data_in_len;
 184         flags = UFS_UPIU_CMD_FLAGS_WRITE;
 185     } else if (data_out_len > 0) {
 186         g_assert_nonnull(data_out);
 187         data_direction = UFS_UTP_DEVICE_TO_HOST;
 188         data_len = data_out_len;
 189         flags = UFS_UPIU_CMD_FLAGS_READ;
 190     } else {
 191         data_direction = UFS_UTP_NO_DATA_TRANSFER;
 192         data_len = 0;
 193         flags = UFS_UPIU_CMD_FLAGS_NONE;
 194     }
 195     prd_table_length = DIV_ROUND_UP(data_len, PRD_ENTRY_DATA_SIZE);
 196
 197     qtest_memset(ufs->dev.bus->qts, ufs->data_buffer_addr, 0,
 198                  MAX_PRD_ENTRY_COUNT * PRD_ENTRY_DATA_SIZE);
 199     if (data_in_len) {
 200         qtest_memwrite(ufs->dev.bus->qts, ufs->data_buffer_addr, data_in,
 201                        data_in_len);
 202     }
 203
 204     for (i = 0; i < prd_table_length; i++) {
 205         entries[i].addr =
 206             cpu_to_le64(ufs->data_buffer_addr + i * sizeof(UfshcdSgEntry));
 207         if (i + 1 != prd_table_length) {
 208             entries[i].size = cpu_to_le32(PRD_ENTRY_DATA_SIZE - 1);
 209         } else {
 210             entries[i].size = cpu_to_le32(
 211                 data_len - (PRD_ENTRY_DATA_SIZE * (prd_table_length - 1)) - 1);
 212         }
 213     }
 214     qtest_memwrite(ufs->dev.bus->qts, prdt_addr, entries,
 215                    prd_table_length * sizeof(UfshcdSgEntry));
 216
 217     /* Build up utp transfer request descriptor */
 218     UtpTransferReqDesc utrd = ufs_build_req_utrd(
 219         ufs->cmd_desc_addr, slot, data_direction, prd_table_length);
 220     uint64_t utrd_addr = ufs->utrlba + slot * sizeof(UtpTransferReqDesc);
 221     uint64_t rsp_upiu_addr = req_upiu_addr + UTP_RESPONSE_UPIU_OFFSET;
 222     qtest_memwrite(ufs->dev.bus->qts, utrd_addr, &utrd, sizeof(utrd));
 223
 224     /* Build up request upiu */
 225     UtpUpiuReq req_upiu = { 0 };
 226     req_upiu.header.trans_type = UFS_UPIU_TRANSACTION_COMMAND;
 227     req_upiu.header.flags = flags;
 228     req_upiu.header.lun = lun;
 229     req_upiu.header.task_tag = slot;
 230     req_upiu.sc.exp_data_transfer_len = cpu_to_be32(data_len);
 231     memcpy(req_upiu.sc.cdb, cdb, UFS_CDB_SIZE);
 232     qtest_memwrite(ufs->dev.bus->qts, req_upiu_addr, &req_upiu,
 233                    sizeof(req_upiu));
 234
 235     /* Ring Doorbell */
 236     ufs_wreg(ufs, A_UTRLDBR, 1);
 237     ufs_wait_for_irq(ufs);
 238     g_assert_true(FIELD_EX32(ufs_rreg(ufs, A_IS), IS, UTRCS));
 239     ufs_wreg(ufs, A_IS, FIELD_DP32(0, IS, UTRCS, 1));
 240
 241     qtest_memread(ufs->dev.bus->qts, utrd_addr, utrd_out, sizeof(*utrd_out));
 242     qtest_memread(ufs->dev.bus->qts, rsp_upiu_addr, rsp_out, sizeof(*rsp_out));
 243     if (data_out_len) {
 244         qtest_memread(ufs->dev.bus->qts, ufs->data_buffer_addr, data_out,
 245                       data_out_len);
 246     }
 247 }
 248
 249 /**
 250  * Initialize Ufs host controller and logical unit.
 251  * After running this function, you can make a transfer request to the UFS.
 252  */
 253 static void ufs_init(QUfs *ufs, QGuestAllocator *alloc)
 254 {
 255     uint64_t end_time;
 256     uint32_t nutrs, nutmrs;
 257     uint32_t hcs, is, ucmdarg2, cap;
 258     uint32_t hce = 0, ie = 0;
 259     UtpTransferReqDesc utrd;
 260     UtpUpiuRsp rsp_upiu;
 261
 262     ufs->bar = qpci_iomap(&ufs->dev, 0, NULL);
 263     qpci_device_enable(&ufs->dev);
 264
 265     /* Start host controller initialization */
 266     hce = FIELD_DP32(hce, HCE, HCE, 1);
 267     ufs_wreg(ufs, A_HCE, hce);
 268
 269     /* Wait for device to reset */
 270     end_time = g_get_monotonic_time() + TIMEOUT_SECONDS * G_TIME_SPAN_SECOND;
 271     do {
 272         qtest_clock_step(ufs->dev.bus->qts, 100);
 273         hce = FIELD_EX32(ufs_rreg(ufs, A_HCE), HCE, HCE);
 274     } while (hce == 0 && g_get_monotonic_time() < end_time);
 275     g_assert_cmpuint(hce, ==, 1);
 276
 277     /* Enable interrupt */
 278     ie = FIELD_DP32(ie, IE, UCCE, 1);
 279     ie = FIELD_DP32(ie, IE, UHESE, 1);
 280     ie = FIELD_DP32(ie, IE, UHXSE, 1);
 281     ie = FIELD_DP32(ie, IE, UPMSE, 1);
 282     ufs_wreg(ufs, A_IE, ie);
 283
 284     /* Send DME_LINK_STARTUP uic command */
 285     hcs = ufs_rreg(ufs, A_HCS);
 286     g_assert_true(FIELD_EX32(hcs, HCS, UCRDY));
 287
 288     ufs_wreg(ufs, A_UCMDARG1, 0);
 289     ufs_wreg(ufs, A_UCMDARG2, 0);
 290     ufs_wreg(ufs, A_UCMDARG3, 0);
 291     ufs_wreg(ufs, A_UICCMD, UFS_UIC_CMD_DME_LINK_STARTUP);
 292
 293     is = ufs_rreg(ufs, A_IS);
 294     g_assert_true(FIELD_EX32(is, IS, UCCS));
 295     ufs_wreg(ufs, A_IS, FIELD_DP32(0, IS, UCCS, 1));
 296
 297     ucmdarg2 = ufs_rreg(ufs, A_UCMDARG2);
 298     g_assert_cmpuint(ucmdarg2, ==, 0);
 299     is = ufs_rreg(ufs, A_IS);
 300     g_assert_cmpuint(is, ==, 0);
 301     hcs = ufs_rreg(ufs, A_HCS);
 302     g_assert_true(FIELD_EX32(hcs, HCS, DP));
 303     g_assert_true(FIELD_EX32(hcs, HCS, UTRLRDY));
 304     g_assert_true(FIELD_EX32(hcs, HCS, UTMRLRDY));
 305     g_assert_true(FIELD_EX32(hcs, HCS, UCRDY));
 306
 307     /* Enable all interrupt functions */
 308     ie = FIELD_DP32(ie, IE, UTRCE, 1);
 309     ie = FIELD_DP32(ie, IE, UEE, 1);
 310     ie = FIELD_DP32(ie, IE, UPMSE, 1);
 311     ie = FIELD_DP32(ie, IE, UHXSE, 1);
 312     ie = FIELD_DP32(ie, IE, UHESE, 1);
 313     ie = FIELD_DP32(ie, IE, UTMRCE, 1);
 314     ie = FIELD_DP32(ie, IE, UCCE, 1);
 315     ie = FIELD_DP32(ie, IE, DFEE, 1);
 316     ie = FIELD_DP32(ie, IE, HCFEE, 1);
 317     ie = FIELD_DP32(ie, IE, SBFEE, 1);
 318     ie = FIELD_DP32(ie, IE, CEFEE, 1);
 319     ufs_wreg(ufs, A_IE, ie);
 320     ufs_wreg(ufs, A_UTRIACR, 0);
 321
 322     /* Enable tranfer request and task management request */
 323     cap = ufs_rreg(ufs, A_CAP);
 324     nutrs = FIELD_EX32(cap, CAP, NUTRS) + 1;
 325     nutmrs = FIELD_EX32(cap, CAP, NUTMRS) + 1;
 326     ufs->cmd_desc_addr =
 327         guest_alloc(alloc, nutrs * UTP_COMMAND_DESCRIPTOR_SIZE);
 328     ufs->data_buffer_addr =
 329         guest_alloc(alloc, MAX_PRD_ENTRY_COUNT * PRD_ENTRY_DATA_SIZE);
 330     ufs->utrlba = guest_alloc(alloc, nutrs * sizeof(UtpTransferReqDesc));
 331     ufs->utmrlba = guest_alloc(alloc, nutmrs * sizeof(UtpTaskReqDesc));
 332
 333     ufs_wreg(ufs, A_UTRLBA, ufs->utrlba & 0xffffffff);
 334     ufs_wreg(ufs, A_UTRLBAU, ufs->utrlba >> 32);
 335     ufs_wreg(ufs, A_UTMRLBA, ufs->utmrlba & 0xffffffff);
 336     ufs_wreg(ufs, A_UTMRLBAU, ufs->utmrlba >> 32);
 337     ufs_wreg(ufs, A_UTRLRSR, 1);
 338     ufs_wreg(ufs, A_UTMRLRSR, 1);
 339
 340     /* Send nop out to test transfer request */
 341     ufs_send_nop_out(ufs, 0, &utrd, &rsp_upiu);
 342     g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
 343
 344     /* Set fDeviceInit flag via query request */
 345     ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
 346                    UFS_UPIU_QUERY_OPCODE_SET_FLAG,
 347                    UFS_QUERY_FLAG_IDN_FDEVICEINIT, 0, &utrd, &rsp_upiu);
 348     g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
 349
 350     /* Wait for device to reset */
 351     end_time = g_get_monotonic_time() + TIMEOUT_SECONDS * G_TIME_SPAN_SECOND;
 352     do {
 353         qtest_clock_step(ufs->dev.bus->qts, 100);
 354         ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
 355                        UFS_UPIU_QUERY_OPCODE_READ_FLAG,
 356                        UFS_QUERY_FLAG_IDN_FDEVICEINIT, 0, &utrd, &rsp_upiu);
 357     } while (be32_to_cpu(rsp_upiu.qr.value) != 0 &&
 358              g_get_monotonic_time() < end_time);
 359     g_assert_cmpuint(be32_to_cpu(rsp_upiu.qr.value), ==, 0);
 360
 361     ufs->enabled = true;
 362 }
 363
 364 static void ufs_exit(QUfs *ufs, QGuestAllocator *alloc)
 365 {
 366     if (ufs->enabled) {
 367         guest_free(alloc, ufs->utrlba);
 368         guest_free(alloc, ufs->utmrlba);
 369         guest_free(alloc, ufs->cmd_desc_addr);
 370         guest_free(alloc, ufs->data_buffer_addr);
 371     }
 372
 373     qpci_iounmap(&ufs->dev, ufs->bar);
 374 }
 375
 376 static void *ufs_get_driver(void *obj, const char *interface)
 377 {
 378     QUfs *ufs = obj;
 379
 380     if (!g_strcmp0(interface, "pci-device")) {
 381         return &ufs->dev;
 382     }
 383
 384     fprintf(stderr, "%s not present in ufs\n", interface);
 385     g_assert_not_reached();
 386 }
 387
 388 static void *ufs_create(void *pci_bus, QGuestAllocator *alloc, void *addr)
 389 {
 390     QUfs *ufs = g_new0(QUfs, 1);
 391     QPCIBus *bus = pci_bus;
 392
 393     qpci_device_init(&ufs->dev, bus, addr);
 394     ufs->obj.get_driver = ufs_get_driver;
 395
 396     return &ufs->obj;
 397 }
 398
 399 static void ufstest_reg_read(void *obj, void *data, QGuestAllocator *alloc)
 400 {
 401     QUfs *ufs = obj;
 402     uint32_t cap;
 403
 404     ufs->bar = qpci_iomap(&ufs->dev, 0, NULL);
 405     qpci_device_enable(&ufs->dev);
 406
 407     cap = ufs_rreg(ufs, A_CAP);
 408     g_assert_cmpuint(FIELD_EX32(cap, CAP, NUTRS), ==, 31);
 409     g_assert_cmpuint(FIELD_EX32(cap, CAP, NUTMRS), ==, 7);
 410     g_assert_cmpuint(FIELD_EX32(cap, CAP, 64AS), ==, 1);
 411
 412     qpci_iounmap(&ufs->dev, ufs->bar);
 413 }
 414
 415 static void ufstest_init(void *obj, void *data, QGuestAllocator *alloc)
 416 {
 417     QUfs *ufs = obj;
 418
 419     uint8_t buf[4096] = { 0 };
 420     const uint8_t report_luns_cdb[UFS_CDB_SIZE] = {
 421         /* allocation length 4096 */
 422         REPORT_LUNS, 0x00, 0x00, 0x00, 0x00, 0x00,
 423         0x00,        0x00, 0x10, 0x00, 0x00, 0x00
 424     };
 425     const uint8_t test_unit_ready_cdb[UFS_CDB_SIZE] = {
 426         TEST_UNIT_READY,
 427     };
 428     UtpTransferReqDesc utrd;
 429     UtpUpiuRsp rsp_upiu;
 430
 431     ufs_init(ufs, alloc);
 432
 433     /* Check REPORT_LUNS */
 434     ufs_send_scsi_command(ufs, 0, 0, report_luns_cdb, NULL, 0, buf, sizeof(buf),
 435                           &utrd, &rsp_upiu);
 436     g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
 437     g_assert_cmpuint(rsp_upiu.header.scsi_status, ==, GOOD);
 438     /* LUN LIST LENGTH should be 8, in big endian */
 439     g_assert_cmpuint(buf[3], ==, 8);
 440     /* There is one logical unit whose lun is 0 */
 441     g_assert_cmpuint(buf[9], ==, 0);
 442
 443     /* Check TEST_UNIT_READY */
 444     ufs_send_scsi_command(ufs, 0, 0, test_unit_ready_cdb, NULL, 0, NULL, 0,
 445                           &utrd, &rsp_upiu);
 446     g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
 447     g_assert_cmpuint(rsp_upiu.header.scsi_status, ==, GOOD);
 448
 449     ufs_exit(ufs, alloc);
 450 }
 451
 452 static void ufstest_read_write(void *obj, void *data, QGuestAllocator *alloc)
 453 {
 454     QUfs *ufs = obj;
 455     uint8_t read_buf[4096] = { 0 };
 456     uint8_t write_buf[4096] = { 0 };
 457     const uint8_t read_capacity_cdb[UFS_CDB_SIZE] = {
 458         /* allocation length 4096 */
 459         SERVICE_ACTION_IN_16,
 460         SAI_READ_CAPACITY_16,
 461         0x00,
 462         0x00,
 463         0x00,
 464         0x00,
 465         0x00,
 466         0x00,
 467         0x00,
 468         0x00,
 469         0x00,
 470         0x00,
 471         0x10,
 472         0x00,
 473         0x00,
 474         0x00
 475     };
 476     const uint8_t read_cdb[UFS_CDB_SIZE] = {
 477         /* READ(10) to LBA 0, transfer length 1 */
 478         READ_10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00
 479     };
 480     const uint8_t write_cdb[UFS_CDB_SIZE] = {
 481         /* WRITE(10) to LBA 0, transfer length 1 */
 482         WRITE_10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00
 483     };
 484     uint32_t block_size;
 485     UtpTransferReqDesc utrd;
 486     UtpUpiuRsp rsp_upiu;
 487
 488     ufs_init(ufs, alloc);
 489
 490     /* Read capacity */
 491     ufs_send_scsi_command(ufs, 0, 1, read_capacity_cdb, NULL, 0, read_buf,
 492                           sizeof(read_buf), &utrd, &rsp_upiu);
 493     g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
 494     g_assert_cmpuint(rsp_upiu.header.scsi_status, ==,
 495                      UFS_COMMAND_RESULT_SUCESS);
 496     block_size = ldl_be_p(&read_buf[8]);
 497     g_assert_cmpuint(block_size, ==, 4096);
 498
 499     /* Write data */
 500     memset(write_buf, rand() % 255 + 1, block_size);
 501     ufs_send_scsi_command(ufs, 0, 1, write_cdb, write_buf, block_size, NULL, 0,
 502                           &utrd, &rsp_upiu);
 503     g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
 504     g_assert_cmpuint(rsp_upiu.header.scsi_status, ==,
 505                      UFS_COMMAND_RESULT_SUCESS);
 506
 507     /* Read data and verify */
 508     ufs_send_scsi_command(ufs, 0, 1, read_cdb, NULL, 0, read_buf, block_size,
 509                           &utrd, &rsp_upiu);
 510     g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
 511     g_assert_cmpuint(rsp_upiu.header.scsi_status, ==,
 512                      UFS_COMMAND_RESULT_SUCESS);
 513     g_assert_cmpint(memcmp(read_buf, write_buf, block_size), ==, 0);
 514
 515     ufs_exit(ufs, alloc);
 516 }
 517
 518 static void drive_destroy(void *path)
 519 {
 520     unlink(path);
 521     g_free(path);
 522     qos_invalidate_command_line();
 523 }
 524
 525 static char *drive_create(void)
 526 {
 527     int fd, ret;
 528     char *t_path;
 529
 530     /* Create a temporary raw image */
 531     fd = g_file_open_tmp("qtest-ufs.XXXXXX", &t_path, NULL);
 532     g_assert_cmpint(fd, >=, 0);
 533     ret = ftruncate(fd, TEST_IMAGE_SIZE);
 534     g_assert_cmpint(ret, ==, 0);
 535     close(fd);
 536
 537     g_test_queue_destroy(drive_destroy, t_path);
 538     return t_path;
 539 }
 540
 541 static void *ufs_blk_test_setup(GString *cmd_line, void *arg)
 542 {
 543     char *tmp_path = drive_create();
 544
 545     g_string_append_printf(cmd_line,
 546                            " -blockdev file,filename=%s,node-name=drv1 "
 547                            "-device ufs-lu,bus=ufs0,drive=drv1,lun=1 ",
 548                            tmp_path);
 549
 550     return arg;
 551 }
 552
 553 static void ufs_register_nodes(void)
 554 {
 555     const char *arch;
 556     QOSGraphEdgeOptions edge_opts = {
 557         .before_cmd_line = "-blockdev null-co,node-name=drv0,read-zeroes=on",
 558         .after_cmd_line = "-device ufs-lu,bus=ufs0,drive=drv0,lun=0",
 559         .extra_device_opts = "addr=04.0,id=ufs0,nutrs=32,nutmrs=8"
 560     };
 561
 562     QOSGraphTestOptions io_test_opts = {
 563         .before = ufs_blk_test_setup,
 564     };
 565
 566     add_qpci_address(&edge_opts, &(QPCIAddress){ .devfn = QPCI_DEVFN(4, 0) });
 567
 568     qos_node_create_driver("ufs", ufs_create);
 569     qos_node_consumes("ufs", "pci-bus", &edge_opts);
 570     qos_node_produces("ufs", "pci-device");
 571
 572     qos_add_test("reg-read", "ufs", ufstest_reg_read, NULL);
 573
 574     /*
 575      * Check architecture
 576      * TODO: Enable ufs io tests for ppc64
 577      */
 578     arch = qtest_get_arch();
 579     if (!strcmp(arch, "ppc64")) {
 580         g_test_message("Skipping ufs io tests for ppc64");
 581         return;
 582     }
 583     qos_add_test("init", "ufs", ufstest_init, NULL);
 584     qos_add_test("read-write", "ufs", ufstest_read_write, &io_test_opts);
 585 }
 586
 587 libqos_init(ufs_register_nodes);