xref: /openbmc/qemu/tests/qtest/ufs-test.c (revision 48876bfc471ec02e5edcc935ba93e14426d891ab)
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 "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  
ufs_rreg(QUfs * ufs,size_t offset)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  
ufs_wreg(QUfs * ufs,size_t offset,uint32_t value)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  
ufs_wait_for_irq(QUfs * ufs)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  
ufs_build_req_utrd(uint64_t cmd_desc_addr,uint8_t slot,uint32_t data_direction,uint16_t prd_table_length)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  
ufs_send_nop_out(QUfs * ufs,uint8_t slot,UtpTransferReqDesc * utrd_out,UtpUpiuRsp * rsp_out)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  
ufs_send_query(QUfs * ufs,uint8_t slot,uint8_t query_function,uint8_t query_opcode,uint8_t idn,uint8_t index,uint8_t selector,uint32_t attr_value,UtpTransferReqDesc * utrd_out,UtpUpiuRsp * rsp_out)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                             uint8_t selector, uint32_t attr_value,
123                             UtpTransferReqDesc *utrd_out, UtpUpiuRsp *rsp_out)
124  {
125      /* Build up utp transfer request descriptor */
126      UtpTransferReqDesc utrd = ufs_build_req_utrd(ufs->cmd_desc_addr, slot,
127                                                   UFS_UTP_NO_DATA_TRANSFER, 0);
128      uint64_t utrd_addr = ufs->utrlba + slot * sizeof(UtpTransferReqDesc);
129      uint64_t req_upiu_addr =
130          ufs->cmd_desc_addr + slot * UTP_COMMAND_DESCRIPTOR_SIZE;
131      uint64_t rsp_upiu_addr = req_upiu_addr + UTP_RESPONSE_UPIU_OFFSET;
132      qtest_memwrite(ufs->dev.bus->qts, utrd_addr, &utrd, sizeof(utrd));
133  
134      /* Build up request upiu */
135      UtpUpiuReq req_upiu = { 0 };
136      req_upiu.header.trans_type = UFS_UPIU_TRANSACTION_QUERY_REQ;
137      req_upiu.header.query_func = query_function;
138      req_upiu.header.task_tag = slot;
139      /*
140       * QEMU UFS does not currently support Write descriptor,
141       * so the value of data_segment_length is always 0.
142       */
143      req_upiu.header.data_segment_length = 0;
144      req_upiu.qr.opcode = query_opcode;
145      req_upiu.qr.idn = idn;
146      req_upiu.qr.index = index;
147      req_upiu.qr.selector = selector;
148      req_upiu.qr.value = cpu_to_be32(attr_value);
149      req_upiu.qr.length = UFS_QUERY_DESC_MAX_SIZE;
150      qtest_memwrite(ufs->dev.bus->qts, req_upiu_addr, &req_upiu,
151                     sizeof(req_upiu));
152  
153      /* Ring Doorbell */
154      ufs_wreg(ufs, A_UTRLDBR, 1);
155      ufs_wait_for_irq(ufs);
156      g_assert_true(FIELD_EX32(ufs_rreg(ufs, A_IS), IS, UTRCS));
157      ufs_wreg(ufs, A_IS, FIELD_DP32(0, IS, UTRCS, 1));
158  
159      qtest_memread(ufs->dev.bus->qts, utrd_addr, utrd_out, sizeof(*utrd_out));
160      qtest_memread(ufs->dev.bus->qts, rsp_upiu_addr, rsp_out, sizeof(*rsp_out));
161  }
162  
ufs_send_scsi_command(QUfs * ufs,uint8_t slot,uint8_t lun,const uint8_t * cdb,const uint8_t * data_in,size_t data_in_len,uint8_t * data_out,size_t data_out_len,UtpTransferReqDesc * utrd_out,UtpUpiuRsp * rsp_out)163  static void ufs_send_scsi_command(QUfs *ufs, uint8_t slot, uint8_t lun,
164                                    const uint8_t *cdb, const uint8_t *data_in,
165                                    size_t data_in_len, uint8_t *data_out,
166                                    size_t data_out_len,
167                                    UtpTransferReqDesc *utrd_out,
168                                    UtpUpiuRsp *rsp_out)
169  
170  {
171      /* Build up PRDT */
172      UfshcdSgEntry entries[MAX_PRD_ENTRY_COUNT] = {
173          0,
174      };
175      uint8_t flags;
176      uint16_t prd_table_length, i;
177      uint32_t data_direction, data_len;
178      uint64_t req_upiu_addr =
179          ufs->cmd_desc_addr + slot * UTP_COMMAND_DESCRIPTOR_SIZE;
180      uint64_t prdt_addr = req_upiu_addr + UTP_PRDT_UPIU_OFFSET;
181  
182      g_assert_true(data_in_len < MAX_PRD_ENTRY_COUNT * PRD_ENTRY_DATA_SIZE);
183      g_assert_true(data_out_len < MAX_PRD_ENTRY_COUNT * PRD_ENTRY_DATA_SIZE);
184      if (data_in_len > 0) {
185          g_assert_nonnull(data_in);
186          data_direction = UFS_UTP_HOST_TO_DEVICE;
187          data_len = data_in_len;
188          flags = UFS_UPIU_CMD_FLAGS_WRITE;
189      } else if (data_out_len > 0) {
190          g_assert_nonnull(data_out);
191          data_direction = UFS_UTP_DEVICE_TO_HOST;
192          data_len = data_out_len;
193          flags = UFS_UPIU_CMD_FLAGS_READ;
194      } else {
195          data_direction = UFS_UTP_NO_DATA_TRANSFER;
196          data_len = 0;
197          flags = UFS_UPIU_CMD_FLAGS_NONE;
198      }
199      prd_table_length = DIV_ROUND_UP(data_len, PRD_ENTRY_DATA_SIZE);
200  
201      qtest_memset(ufs->dev.bus->qts, ufs->data_buffer_addr, 0,
202                   MAX_PRD_ENTRY_COUNT * PRD_ENTRY_DATA_SIZE);
203      if (data_in_len) {
204          qtest_memwrite(ufs->dev.bus->qts, ufs->data_buffer_addr, data_in,
205                         data_in_len);
206      }
207  
208      for (i = 0; i < prd_table_length; i++) {
209          entries[i].addr =
210              cpu_to_le64(ufs->data_buffer_addr + i * sizeof(UfshcdSgEntry));
211          if (i + 1 != prd_table_length) {
212              entries[i].size = cpu_to_le32(PRD_ENTRY_DATA_SIZE - 1);
213          } else {
214              entries[i].size = cpu_to_le32(
215                  data_len - (PRD_ENTRY_DATA_SIZE * (prd_table_length - 1)) - 1);
216          }
217      }
218      qtest_memwrite(ufs->dev.bus->qts, prdt_addr, entries,
219                     prd_table_length * sizeof(UfshcdSgEntry));
220  
221      /* Build up utp transfer request descriptor */
222      UtpTransferReqDesc utrd = ufs_build_req_utrd(
223          ufs->cmd_desc_addr, slot, data_direction, prd_table_length);
224      uint64_t utrd_addr = ufs->utrlba + slot * sizeof(UtpTransferReqDesc);
225      uint64_t rsp_upiu_addr = req_upiu_addr + UTP_RESPONSE_UPIU_OFFSET;
226      qtest_memwrite(ufs->dev.bus->qts, utrd_addr, &utrd, sizeof(utrd));
227  
228      /* Build up request upiu */
229      UtpUpiuReq req_upiu = { 0 };
230      req_upiu.header.trans_type = UFS_UPIU_TRANSACTION_COMMAND;
231      req_upiu.header.flags = flags;
232      req_upiu.header.lun = lun;
233      req_upiu.header.task_tag = slot;
234      req_upiu.sc.exp_data_transfer_len = cpu_to_be32(data_len);
235      memcpy(req_upiu.sc.cdb, cdb, UFS_CDB_SIZE);
236      qtest_memwrite(ufs->dev.bus->qts, req_upiu_addr, &req_upiu,
237                     sizeof(req_upiu));
238  
239      /* Ring Doorbell */
240      ufs_wreg(ufs, A_UTRLDBR, 1);
241      ufs_wait_for_irq(ufs);
242      g_assert_true(FIELD_EX32(ufs_rreg(ufs, A_IS), IS, UTRCS));
243      ufs_wreg(ufs, A_IS, FIELD_DP32(0, IS, UTRCS, 1));
244  
245      qtest_memread(ufs->dev.bus->qts, utrd_addr, utrd_out, sizeof(*utrd_out));
246      qtest_memread(ufs->dev.bus->qts, rsp_upiu_addr, rsp_out, sizeof(*rsp_out));
247      if (data_out_len) {
248          qtest_memread(ufs->dev.bus->qts, ufs->data_buffer_addr, data_out,
249                        data_out_len);
250      }
251  }
252  
253  /**
254   * Initialize Ufs host controller and logical unit.
255   * After running this function, you can make a transfer request to the UFS.
256   */
ufs_init(QUfs * ufs,QGuestAllocator * alloc)257  static void ufs_init(QUfs *ufs, QGuestAllocator *alloc)
258  {
259      uint64_t end_time;
260      uint32_t nutrs, nutmrs;
261      uint32_t hcs, is, ucmdarg2, cap;
262      uint32_t hce = 0, ie = 0;
263      UtpTransferReqDesc utrd;
264      UtpUpiuRsp rsp_upiu;
265  
266      ufs->bar = qpci_iomap(&ufs->dev, 0, NULL);
267      qpci_device_enable(&ufs->dev);
268  
269      /* Start host controller initialization */
270      hce = FIELD_DP32(hce, HCE, HCE, 1);
271      ufs_wreg(ufs, A_HCE, hce);
272  
273      /* Wait for device to reset */
274      end_time = g_get_monotonic_time() + TIMEOUT_SECONDS * G_TIME_SPAN_SECOND;
275      do {
276          qtest_clock_step(ufs->dev.bus->qts, 100);
277          hce = FIELD_EX32(ufs_rreg(ufs, A_HCE), HCE, HCE);
278      } while (hce == 0 && g_get_monotonic_time() < end_time);
279      g_assert_cmpuint(hce, ==, 1);
280  
281      /* Enable interrupt */
282      ie = FIELD_DP32(ie, IE, UCCE, 1);
283      ie = FIELD_DP32(ie, IE, UHESE, 1);
284      ie = FIELD_DP32(ie, IE, UHXSE, 1);
285      ie = FIELD_DP32(ie, IE, UPMSE, 1);
286      ufs_wreg(ufs, A_IE, ie);
287  
288      /* Send DME_LINK_STARTUP uic command */
289      hcs = ufs_rreg(ufs, A_HCS);
290      g_assert_true(FIELD_EX32(hcs, HCS, UCRDY));
291  
292      ufs_wreg(ufs, A_UCMDARG1, 0);
293      ufs_wreg(ufs, A_UCMDARG2, 0);
294      ufs_wreg(ufs, A_UCMDARG3, 0);
295      ufs_wreg(ufs, A_UICCMD, UFS_UIC_CMD_DME_LINK_STARTUP);
296  
297      is = ufs_rreg(ufs, A_IS);
298      g_assert_true(FIELD_EX32(is, IS, UCCS));
299      ufs_wreg(ufs, A_IS, FIELD_DP32(0, IS, UCCS, 1));
300  
301      ucmdarg2 = ufs_rreg(ufs, A_UCMDARG2);
302      g_assert_cmpuint(ucmdarg2, ==, 0);
303      is = ufs_rreg(ufs, A_IS);
304      g_assert_cmpuint(is, ==, 0);
305      hcs = ufs_rreg(ufs, A_HCS);
306      g_assert_true(FIELD_EX32(hcs, HCS, DP));
307      g_assert_true(FIELD_EX32(hcs, HCS, UTRLRDY));
308      g_assert_true(FIELD_EX32(hcs, HCS, UTMRLRDY));
309      g_assert_true(FIELD_EX32(hcs, HCS, UCRDY));
310  
311      /* Enable all interrupt functions */
312      ie = FIELD_DP32(ie, IE, UTRCE, 1);
313      ie = FIELD_DP32(ie, IE, UEE, 1);
314      ie = FIELD_DP32(ie, IE, UPMSE, 1);
315      ie = FIELD_DP32(ie, IE, UHXSE, 1);
316      ie = FIELD_DP32(ie, IE, UHESE, 1);
317      ie = FIELD_DP32(ie, IE, UTMRCE, 1);
318      ie = FIELD_DP32(ie, IE, UCCE, 1);
319      ie = FIELD_DP32(ie, IE, DFEE, 1);
320      ie = FIELD_DP32(ie, IE, HCFEE, 1);
321      ie = FIELD_DP32(ie, IE, SBFEE, 1);
322      ie = FIELD_DP32(ie, IE, CEFEE, 1);
323      ufs_wreg(ufs, A_IE, ie);
324      ufs_wreg(ufs, A_UTRIACR, 0);
325  
326      /* Enable transfer request and task management request */
327      cap = ufs_rreg(ufs, A_CAP);
328      nutrs = FIELD_EX32(cap, CAP, NUTRS) + 1;
329      nutmrs = FIELD_EX32(cap, CAP, NUTMRS) + 1;
330      ufs->cmd_desc_addr =
331          guest_alloc(alloc, nutrs * UTP_COMMAND_DESCRIPTOR_SIZE);
332      ufs->data_buffer_addr =
333          guest_alloc(alloc, MAX_PRD_ENTRY_COUNT * PRD_ENTRY_DATA_SIZE);
334      ufs->utrlba = guest_alloc(alloc, nutrs * sizeof(UtpTransferReqDesc));
335      ufs->utmrlba = guest_alloc(alloc, nutmrs * sizeof(UtpTaskReqDesc));
336  
337      ufs_wreg(ufs, A_UTRLBA, ufs->utrlba & 0xffffffff);
338      ufs_wreg(ufs, A_UTRLBAU, ufs->utrlba >> 32);
339      ufs_wreg(ufs, A_UTMRLBA, ufs->utmrlba & 0xffffffff);
340      ufs_wreg(ufs, A_UTMRLBAU, ufs->utmrlba >> 32);
341      ufs_wreg(ufs, A_UTRLRSR, 1);
342      ufs_wreg(ufs, A_UTMRLRSR, 1);
343  
344      /* Send nop out to test transfer request */
345      ufs_send_nop_out(ufs, 0, &utrd, &rsp_upiu);
346      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
347  
348      /* Set fDeviceInit flag via query request */
349      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
350                     UFS_UPIU_QUERY_OPCODE_SET_FLAG,
351                     UFS_QUERY_FLAG_IDN_FDEVICEINIT, 0, 0, 0, &utrd, &rsp_upiu);
352      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
353  
354      /* Wait for device to reset */
355      end_time = g_get_monotonic_time() + TIMEOUT_SECONDS * G_TIME_SPAN_SECOND;
356      do {
357          qtest_clock_step(ufs->dev.bus->qts, 100);
358          ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
359                         UFS_UPIU_QUERY_OPCODE_READ_FLAG,
360                         UFS_QUERY_FLAG_IDN_FDEVICEINIT, 0, 0, 0, &utrd,
361                         &rsp_upiu);
362      } while (be32_to_cpu(rsp_upiu.qr.value) != 0 &&
363               g_get_monotonic_time() < end_time);
364      g_assert_cmpuint(be32_to_cpu(rsp_upiu.qr.value), ==, 0);
365  
366      ufs->enabled = true;
367  }
368  
ufs_exit(QUfs * ufs,QGuestAllocator * alloc)369  static void ufs_exit(QUfs *ufs, QGuestAllocator *alloc)
370  {
371      if (ufs->enabled) {
372          guest_free(alloc, ufs->utrlba);
373          guest_free(alloc, ufs->utmrlba);
374          guest_free(alloc, ufs->cmd_desc_addr);
375          guest_free(alloc, ufs->data_buffer_addr);
376      }
377  
378      qpci_iounmap(&ufs->dev, ufs->bar);
379  }
380  
ufs_get_driver(void * obj,const char * interface)381  static void *ufs_get_driver(void *obj, const char *interface)
382  {
383      QUfs *ufs = obj;
384  
385      if (!g_strcmp0(interface, "pci-device")) {
386          return &ufs->dev;
387      }
388  
389      fprintf(stderr, "%s not present in ufs\n", interface);
390      g_assert_not_reached();
391  }
392  
ufs_create(void * pci_bus,QGuestAllocator * alloc,void * addr)393  static void *ufs_create(void *pci_bus, QGuestAllocator *alloc, void *addr)
394  {
395      QUfs *ufs = g_new0(QUfs, 1);
396      QPCIBus *bus = pci_bus;
397  
398      qpci_device_init(&ufs->dev, bus, addr);
399      ufs->obj.get_driver = ufs_get_driver;
400  
401      return &ufs->obj;
402  }
403  
ufstest_reg_read(void * obj,void * data,QGuestAllocator * alloc)404  static void ufstest_reg_read(void *obj, void *data, QGuestAllocator *alloc)
405  {
406      QUfs *ufs = obj;
407      uint32_t cap;
408  
409      ufs->bar = qpci_iomap(&ufs->dev, 0, NULL);
410      qpci_device_enable(&ufs->dev);
411  
412      cap = ufs_rreg(ufs, A_CAP);
413      g_assert_cmpuint(FIELD_EX32(cap, CAP, NUTRS), ==, 31);
414      g_assert_cmpuint(FIELD_EX32(cap, CAP, NUTMRS), ==, 7);
415      g_assert_cmpuint(FIELD_EX32(cap, CAP, 64AS), ==, 1);
416  
417      qpci_iounmap(&ufs->dev, ufs->bar);
418  }
419  
ufstest_init(void * obj,void * data,QGuestAllocator * alloc)420  static void ufstest_init(void *obj, void *data, QGuestAllocator *alloc)
421  {
422      QUfs *ufs = obj;
423  
424      uint8_t buf[4096] = { 0 };
425      const uint8_t report_luns_cdb[UFS_CDB_SIZE] = {
426          /* allocation length 4096 */
427          REPORT_LUNS, 0x00, 0x00, 0x00, 0x00, 0x00,
428          0x00,        0x00, 0x10, 0x00, 0x00, 0x00
429      };
430      const uint8_t test_unit_ready_cdb[UFS_CDB_SIZE] = {
431          TEST_UNIT_READY,
432      };
433      const uint8_t request_sense_cdb[UFS_CDB_SIZE] = {
434          REQUEST_SENSE,
435      };
436      UtpTransferReqDesc utrd;
437      UtpUpiuRsp rsp_upiu;
438  
439      ufs_init(ufs, alloc);
440  
441      /* Check REPORT_LUNS */
442      ufs_send_scsi_command(ufs, 0, 0, report_luns_cdb, NULL, 0, buf, sizeof(buf),
443                            &utrd, &rsp_upiu);
444      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
445      g_assert_cmpuint(rsp_upiu.header.scsi_status, ==, GOOD);
446      /* LUN LIST LENGTH should be 8, in big endian */
447      g_assert_cmpuint(buf[3], ==, 8);
448      /* There is one logical unit whose lun is 0 */
449      g_assert_cmpuint(buf[9], ==, 0);
450  
451      /* Clear Unit Attention */
452      ufs_send_scsi_command(ufs, 0, 0, request_sense_cdb, NULL, 0, buf,
453                            sizeof(buf), &utrd, &rsp_upiu);
454      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
455      g_assert_cmpuint(rsp_upiu.header.scsi_status, ==, CHECK_CONDITION);
456  
457      /* Check TEST_UNIT_READY */
458      ufs_send_scsi_command(ufs, 0, 0, test_unit_ready_cdb, NULL, 0, NULL, 0,
459                            &utrd, &rsp_upiu);
460      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
461      g_assert_cmpuint(rsp_upiu.header.scsi_status, ==, GOOD);
462  
463      ufs_exit(ufs, alloc);
464  }
465  
ufstest_read_write(void * obj,void * data,QGuestAllocator * alloc)466  static void ufstest_read_write(void *obj, void *data, QGuestAllocator *alloc)
467  {
468      QUfs *ufs = obj;
469      uint8_t read_buf[4096] = { 0 };
470      uint8_t write_buf[4096] = { 0 };
471      const uint8_t read_capacity_cdb[UFS_CDB_SIZE] = {
472          /* allocation length 4096 */
473          SERVICE_ACTION_IN_16,
474          SAI_READ_CAPACITY_16,
475          0x00,
476          0x00,
477          0x00,
478          0x00,
479          0x00,
480          0x00,
481          0x00,
482          0x00,
483          0x00,
484          0x00,
485          0x10,
486          0x00,
487          0x00,
488          0x00
489      };
490      const uint8_t request_sense_cdb[UFS_CDB_SIZE] = {
491          REQUEST_SENSE,
492      };
493      const uint8_t read_cdb[UFS_CDB_SIZE] = {
494          /* READ(10) to LBA 0, transfer length 1 */
495          READ_10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00
496      };
497      const uint8_t write_cdb[UFS_CDB_SIZE] = {
498          /* WRITE(10) to LBA 0, transfer length 1 */
499          WRITE_10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00
500      };
501      uint32_t block_size;
502      UtpTransferReqDesc utrd;
503      UtpUpiuRsp rsp_upiu;
504      const int test_lun = 1;
505  
506      ufs_init(ufs, alloc);
507  
508      /* Clear Unit Attention */
509      ufs_send_scsi_command(ufs, 0, test_lun, request_sense_cdb, NULL, 0,
510                            read_buf, sizeof(read_buf), &utrd, &rsp_upiu);
511      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
512      g_assert_cmpuint(rsp_upiu.header.scsi_status, ==, CHECK_CONDITION);
513  
514      /* Read capacity */
515      ufs_send_scsi_command(ufs, 0, test_lun, read_capacity_cdb, NULL, 0,
516                            read_buf, sizeof(read_buf), &utrd, &rsp_upiu);
517      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
518      g_assert_cmpuint(rsp_upiu.header.scsi_status, ==,
519                       UFS_COMMAND_RESULT_SUCCESS);
520      block_size = ldl_be_p(&read_buf[8]);
521      g_assert_cmpuint(block_size, ==, 4096);
522  
523      /* Write data */
524      memset(write_buf, 0xab, block_size);
525      ufs_send_scsi_command(ufs, 0, test_lun, write_cdb, write_buf, block_size,
526                            NULL, 0, &utrd, &rsp_upiu);
527      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
528      g_assert_cmpuint(rsp_upiu.header.scsi_status, ==,
529                       UFS_COMMAND_RESULT_SUCCESS);
530  
531      /* Read data and verify */
532      ufs_send_scsi_command(ufs, 0, test_lun, read_cdb, NULL, 0, read_buf,
533                            block_size, &utrd, &rsp_upiu);
534      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
535      g_assert_cmpuint(rsp_upiu.header.scsi_status, ==,
536                       UFS_COMMAND_RESULT_SUCCESS);
537      g_assert_cmpint(memcmp(read_buf, write_buf, block_size), ==, 0);
538  
539      ufs_exit(ufs, alloc);
540  }
541  
ufstest_query_flag_request(void * obj,void * data,QGuestAllocator * alloc)542  static void ufstest_query_flag_request(void *obj, void *data,
543                                         QGuestAllocator *alloc)
544  {
545      QUfs *ufs = obj;
546  
547      UtpTransferReqDesc utrd;
548      UtpUpiuRsp rsp_upiu;
549      ufs_init(ufs, alloc);
550  
551      /* Read read-only flag */
552      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
553                     UFS_UPIU_QUERY_OPCODE_READ_FLAG,
554                     UFS_QUERY_FLAG_IDN_FDEVICEINIT, 0, 0, 0, &utrd, &rsp_upiu);
555      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
556      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
557      g_assert_cmpuint(rsp_upiu.qr.opcode, ==, UFS_UPIU_QUERY_OPCODE_READ_FLAG);
558      g_assert_cmpuint(rsp_upiu.qr.idn, ==, UFS_QUERY_FLAG_IDN_FDEVICEINIT);
559      g_assert_cmpuint(rsp_upiu.qr.value, ==, be32_to_cpu(0));
560  
561      /* Flag Set, Clear, Toggle Test with fDeviceLifeSpanModeEn */
562      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
563                     UFS_UPIU_QUERY_OPCODE_READ_FLAG,
564                     UFS_QUERY_FLAG_IDN_LIFE_SPAN_MODE_ENABLE, 0, 0, 0, &utrd,
565                     &rsp_upiu);
566      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
567      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
568      g_assert_cmpuint(rsp_upiu.qr.value, ==, be32_to_cpu(0));
569  
570      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
571                     UFS_UPIU_QUERY_OPCODE_SET_FLAG,
572                     UFS_QUERY_FLAG_IDN_LIFE_SPAN_MODE_ENABLE, 0, 0, 0, &utrd,
573                     &rsp_upiu);
574      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
575      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
576      g_assert_cmpuint(rsp_upiu.qr.value, ==, be32_to_cpu(1));
577  
578      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
579                     UFS_UPIU_QUERY_OPCODE_CLEAR_FLAG,
580                     UFS_QUERY_FLAG_IDN_LIFE_SPAN_MODE_ENABLE, 0, 0, 0, &utrd,
581                     &rsp_upiu);
582      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
583      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
584      g_assert_cmpuint(rsp_upiu.qr.value, ==, be32_to_cpu(0));
585  
586      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
587                     UFS_UPIU_QUERY_OPCODE_TOGGLE_FLAG,
588                     UFS_QUERY_FLAG_IDN_LIFE_SPAN_MODE_ENABLE, 0, 0, 0, &utrd,
589                     &rsp_upiu);
590      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
591      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
592      g_assert_cmpuint(rsp_upiu.qr.value, ==, be32_to_cpu(1));
593  
594      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
595                     UFS_UPIU_QUERY_OPCODE_TOGGLE_FLAG,
596                     UFS_QUERY_FLAG_IDN_LIFE_SPAN_MODE_ENABLE, 0, 0, 0, &utrd,
597                     &rsp_upiu);
598      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
599      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
600      g_assert_cmpuint(rsp_upiu.qr.value, ==, be32_to_cpu(0));
601  
602      /* Read Write-only Flag (Intended Failure) */
603      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
604                     UFS_UPIU_QUERY_OPCODE_READ_FLAG,
605                     UFS_QUERY_FLAG_IDN_PURGE_ENABLE, 0, 0, 0, &utrd, &rsp_upiu);
606      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
607                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
608      g_assert_cmpuint(rsp_upiu.header.response, ==,
609                       UFS_QUERY_RESULT_NOT_READABLE);
610  
611      /* Write Read-Only Flag (Intended Failure) */
612      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
613                     UFS_UPIU_QUERY_OPCODE_SET_FLAG, UFS_QUERY_FLAG_IDN_BUSY_RTC,
614                     0, 0, 0, &utrd, &rsp_upiu);
615      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
616                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
617      g_assert_cmpuint(rsp_upiu.header.response, ==,
618                       UFS_QUERY_RESULT_NOT_WRITEABLE);
619  
620      ufs_exit(ufs, alloc);
621  }
622  
ufstest_query_attr_request(void * obj,void * data,QGuestAllocator * alloc)623  static void ufstest_query_attr_request(void *obj, void *data,
624                                         QGuestAllocator *alloc)
625  {
626      QUfs *ufs = obj;
627  
628      UtpTransferReqDesc utrd;
629      UtpUpiuRsp rsp_upiu;
630      ufs_init(ufs, alloc);
631  
632      /* Read Readable Attributes*/
633      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
634                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
635                     UFS_QUERY_ATTR_IDN_BOOT_LU_EN, 0, 0, 0, &utrd, &rsp_upiu);
636      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
637      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
638      g_assert_cmpuint(rsp_upiu.qr.opcode, ==, UFS_UPIU_QUERY_OPCODE_READ_ATTR);
639      g_assert_cmpuint(rsp_upiu.qr.idn, ==, UFS_QUERY_ATTR_IDN_BOOT_LU_EN);
640      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x00));
641  
642      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
643                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
644                     UFS_QUERY_ATTR_IDN_BKOPS_STATUS, 0, 0, 0, &utrd, &rsp_upiu);
645      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
646      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
647      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x00));
648  
649      /* Write Writable Attributes & Read Again */
650      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
651                     UFS_UPIU_QUERY_OPCODE_WRITE_ATTR,
652                     UFS_QUERY_ATTR_IDN_ACTIVE_ICC_LVL, 0, 0, 0x03, &utrd,
653                     &rsp_upiu);
654      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
655      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
656      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x03));
657  
658      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
659                     UFS_UPIU_QUERY_OPCODE_WRITE_ATTR,
660                     UFS_QUERY_ATTR_IDN_EE_CONTROL, 0, 0, 0x07, &utrd, &rsp_upiu);
661      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
662      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
663      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x07));
664  
665      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
666                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
667                     UFS_QUERY_ATTR_IDN_ACTIVE_ICC_LVL, 0, 0, 0, &utrd,
668                     &rsp_upiu);
669      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
670      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
671      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x03));
672  
673      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
674                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
675                     UFS_QUERY_ATTR_IDN_EE_CONTROL, 0, 0, 0, &utrd, &rsp_upiu);
676      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
677      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
678      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x07));
679  
680      /* Write Invalid Value (Intended Error) */
681      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
682                     UFS_UPIU_QUERY_OPCODE_WRITE_ATTR,
683                     UFS_QUERY_ATTR_IDN_ACTIVE_ICC_LVL, 0, 0, 0x10, &utrd,
684                     &rsp_upiu);
685      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
686                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
687      g_assert_cmpuint(rsp_upiu.header.response, ==,
688                       UFS_QUERY_RESULT_INVALID_VALUE);
689  
690      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
691                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
692                     UFS_QUERY_ATTR_IDN_ACTIVE_ICC_LVL, 0, 0, 0, &utrd,
693                     &rsp_upiu);
694      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
695      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
696      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x03));
697  
698      /* Read Write-Only Attribute (Intended Error) */
699      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
700                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
701                     UFS_QUERY_ATTR_IDN_SECONDS_PASSED, 0, 0, 0, &utrd,
702                     &rsp_upiu);
703      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
704                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
705      g_assert_cmpuint(rsp_upiu.header.response, ==,
706                       UFS_QUERY_RESULT_NOT_READABLE);
707  
708      /* Write Read-Only Attribute (Intended Error) */
709      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
710                     UFS_UPIU_QUERY_OPCODE_WRITE_ATTR,
711                     UFS_QUERY_ATTR_IDN_POWER_MODE, 0, 0, 0x01, &utrd, &rsp_upiu);
712      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
713                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
714      g_assert_cmpuint(rsp_upiu.header.response, ==,
715                       UFS_QUERY_RESULT_NOT_WRITEABLE);
716  
717      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
718                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
719                     UFS_QUERY_ATTR_IDN_POWER_MODE, 0, 0, 0, &utrd, &rsp_upiu);
720      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
721      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
722      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x00));
723  
724      /* Reset Written Attributes */
725      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
726                     UFS_UPIU_QUERY_OPCODE_WRITE_ATTR,
727                     UFS_QUERY_ATTR_IDN_ACTIVE_ICC_LVL, 0, 0, 0, &utrd,
728                     &rsp_upiu);
729      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
730      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
731      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x00));
732  
733      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_WRITE_REQUEST,
734                     UFS_UPIU_QUERY_OPCODE_WRITE_ATTR,
735                     UFS_QUERY_ATTR_IDN_EE_CONTROL, 0, 0, 0, &utrd, &rsp_upiu);
736      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
737      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
738      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x00));
739  
740      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
741                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
742                     UFS_QUERY_ATTR_IDN_ACTIVE_ICC_LVL, 0, 0, 0, &utrd,
743                     &rsp_upiu);
744      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
745      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
746      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x00));
747  
748      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
749                     UFS_UPIU_QUERY_OPCODE_READ_ATTR,
750                     UFS_QUERY_ATTR_IDN_EE_CONTROL, 0, 0, 0, &utrd, &rsp_upiu);
751      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
752      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
753      g_assert_cmpuint(rsp_upiu.qr.value, ==, cpu_to_be32(0x00));
754  
755      ufs_exit(ufs, alloc);
756  }
757  
ufstest_query_desc_request(void * obj,void * data,QGuestAllocator * alloc)758  static void ufstest_query_desc_request(void *obj, void *data,
759                                         QGuestAllocator *alloc)
760  {
761      QUfs *ufs = obj;
762  
763      UtpTransferReqDesc utrd;
764      UtpUpiuRsp rsp_upiu;
765      ufs_init(ufs, alloc);
766  
767      /* Write Descriptor is not supported yet */
768  
769      /* Read Device Descriptor */
770      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
771                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_DEVICE,
772                     0, 0, 0, &utrd, &rsp_upiu);
773      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
774      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
775      g_assert_cmpuint(rsp_upiu.qr.opcode, ==, UFS_UPIU_QUERY_OPCODE_READ_DESC);
776      g_assert_cmpuint(rsp_upiu.qr.idn, ==, UFS_QUERY_DESC_IDN_DEVICE);
777      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, sizeof(DeviceDescriptor));
778      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_DEVICE);
779  
780      /* Read Configuration Descriptor is not supported yet*/
781  
782      /* Read Unit Descriptor */
783      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
784                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_UNIT, 0,
785                     0, 0, &utrd, &rsp_upiu);
786      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
787      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
788      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, sizeof(UnitDescriptor));
789      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_UNIT);
790      g_assert_cmpuint(rsp_upiu.qr.data[2], ==, 0);
791  
792      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
793                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_UNIT, 1,
794                     0, 0, &utrd, &rsp_upiu);
795      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
796      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
797      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, sizeof(UnitDescriptor));
798      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_UNIT);
799      g_assert_cmpuint(rsp_upiu.qr.data[2], ==, 1);
800  
801      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
802                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_UNIT,
803                     UFS_UPIU_RPMB_WLUN, 0, 0, &utrd, &rsp_upiu);
804      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
805      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
806      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, sizeof(RpmbUnitDescriptor));
807      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_UNIT);
808      g_assert_cmpuint(rsp_upiu.qr.data[2], ==, UFS_UPIU_RPMB_WLUN);
809  
810      /* Read Interconnect Descriptor */
811      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
812                     UFS_UPIU_QUERY_OPCODE_READ_DESC,
813                     UFS_QUERY_DESC_IDN_INTERCONNECT, 0, 0, 0, &utrd, &rsp_upiu);
814      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
815      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
816      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, sizeof(InterconnectDescriptor));
817      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_INTERCONNECT);
818  
819      /* Read String Descriptor */
820      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
821                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_STRING,
822                     0, 0, 0, &utrd, &rsp_upiu);
823      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
824      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
825      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, 0x12);
826      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_STRING);
827  
828      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
829                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_STRING,
830                     1, 0, 0, &utrd, &rsp_upiu);
831      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
832      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
833      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, 0x22);
834      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_STRING);
835  
836      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
837                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_STRING,
838                     4, 0, 0, &utrd, &rsp_upiu);
839      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
840      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
841      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, 0x0a);
842      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_STRING);
843  
844      /* Read Geometry Descriptor */
845      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
846                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_GEOMETRY,
847                     0, 0, 0, &utrd, &rsp_upiu);
848      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
849      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
850      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, sizeof(GeometryDescriptor));
851      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_GEOMETRY);
852  
853      /* Read Power Descriptor */
854      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
855                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_POWER, 0,
856                     0, 0, &utrd, &rsp_upiu);
857      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
858      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
859      g_assert_cmpuint(rsp_upiu.qr.data[0], ==,
860                       sizeof(PowerParametersDescriptor));
861      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_POWER);
862  
863      /* Read Health Descriptor */
864      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
865                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_HEALTH,
866                     0, 0, 0, &utrd, &rsp_upiu);
867      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==, UFS_OCS_SUCCESS);
868      g_assert_cmpuint(rsp_upiu.header.response, ==, UFS_COMMAND_RESULT_SUCCESS);
869      g_assert_cmpuint(rsp_upiu.qr.data[0], ==, sizeof(DeviceHealthDescriptor));
870      g_assert_cmpuint(rsp_upiu.qr.data[1], ==, UFS_QUERY_DESC_IDN_HEALTH);
871  
872      /* Invalid Index (Intended Failure) */
873      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
874                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_UNIT, 4,
875                     0, 0, &utrd, &rsp_upiu);
876      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
877                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
878      g_assert_cmpuint(rsp_upiu.header.response, ==,
879                       UFS_QUERY_RESULT_INVALID_INDEX);
880  
881      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
882                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_STRING,
883                     5, 0, 0, &utrd, &rsp_upiu);
884      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
885                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
886      g_assert_cmpuint(rsp_upiu.header.response, ==,
887                       UFS_QUERY_RESULT_INVALID_INDEX);
888  
889      /* Invalid Selector (Intended Failure) */
890      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
891                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_DEVICE,
892                     0, 1, 0, &utrd, &rsp_upiu);
893      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
894                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
895      g_assert_cmpuint(rsp_upiu.header.response, ==,
896                       UFS_QUERY_RESULT_INVALID_SELECTOR);
897  
898      ufs_send_query(ufs, 0, UFS_UPIU_QUERY_FUNC_STANDARD_READ_REQUEST,
899                     UFS_UPIU_QUERY_OPCODE_READ_DESC, UFS_QUERY_DESC_IDN_STRING,
900                     0, 1, 0, &utrd, &rsp_upiu);
901      g_assert_cmpuint(le32_to_cpu(utrd.header.dword_2), ==,
902                       UFS_OCS_INVALID_CMD_TABLE_ATTR);
903      g_assert_cmpuint(rsp_upiu.header.response, ==,
904                       UFS_QUERY_RESULT_INVALID_SELECTOR);
905  
906      ufs_exit(ufs, alloc);
907  }
908  
drive_destroy(void * path)909  static void drive_destroy(void *path)
910  {
911      unlink(path);
912      g_free(path);
913      qos_invalidate_command_line();
914  }
915  
drive_create(void)916  static char *drive_create(void)
917  {
918      int fd, ret;
919      char *t_path;
920  
921      /* Create a temporary raw image */
922      fd = g_file_open_tmp("qtest-ufs.XXXXXX", &t_path, NULL);
923      g_assert_cmpint(fd, >=, 0);
924      ret = ftruncate(fd, TEST_IMAGE_SIZE);
925      g_assert_cmpint(ret, ==, 0);
926      close(fd);
927  
928      g_test_queue_destroy(drive_destroy, t_path);
929      return t_path;
930  }
931  
ufs_blk_test_setup(GString * cmd_line,void * arg)932  static void *ufs_blk_test_setup(GString *cmd_line, void *arg)
933  {
934      char *tmp_path = drive_create();
935  
936      g_string_append_printf(cmd_line,
937                             " -blockdev file,filename=%s,node-name=drv1 "
938                             "-device ufs-lu,bus=ufs0,drive=drv1,lun=1 ",
939                             tmp_path);
940  
941      return arg;
942  }
943  
ufs_register_nodes(void)944  static void ufs_register_nodes(void)
945  {
946      const char *arch;
947      QOSGraphEdgeOptions edge_opts = {
948          .before_cmd_line = "-blockdev null-co,node-name=drv0,read-zeroes=on",
949          .after_cmd_line = "-device ufs-lu,bus=ufs0,drive=drv0,lun=0",
950          .extra_device_opts = "addr=04.0,id=ufs0,nutrs=32,nutmrs=8"
951      };
952  
953      QOSGraphTestOptions io_test_opts = {
954          .before = ufs_blk_test_setup,
955      };
956  
957      add_qpci_address(&edge_opts, &(QPCIAddress){ .devfn = QPCI_DEVFN(4, 0) });
958  
959      qos_node_create_driver("ufs", ufs_create);
960      qos_node_consumes("ufs", "pci-bus", &edge_opts);
961      qos_node_produces("ufs", "pci-device");
962  
963      qos_add_test("reg-read", "ufs", ufstest_reg_read, NULL);
964  
965      /*
966       * Check architecture
967       * TODO: Enable ufs io tests for ppc64
968       */
969      arch = qtest_get_arch();
970      if (!strcmp(arch, "ppc64")) {
971          g_test_message("Skipping ufs io tests for ppc64");
972          return;
973      }
974      qos_add_test("init", "ufs", ufstest_init, NULL);
975      qos_add_test("read-write", "ufs", ufstest_read_write, &io_test_opts);
976      qos_add_test("flag read-write", "ufs",
977                   ufstest_query_flag_request, &io_test_opts);
978      qos_add_test("attr read-write", "ufs",
979                   ufstest_query_attr_request, &io_test_opts);
980      qos_add_test("desc read-write", "ufs",
981                   ufstest_query_desc_request, &io_test_opts);
982  }
983  
984  libqos_init(ufs_register_nodes);
985