LCOV - code coverage report
Current view: top level - spdk/test/nvme/cuse - cuse.c (source / functions) Hit Total Coverage
Test: Combined Lines: 75 93 80.6 %
Date: 2024-07-13 21:41:39 Functions: 10 36 27.8 %
Legend: Lines: hit not hit | Branches: + taken - not taken # not executed Branches: 20 50 40.0 %

           Branch data     Line data    Source code
       1                 :            : /*   SPDX-License-Identifier: BSD-3-Clause
       2                 :            :  *
       3                 :            :  *   Copyright (C) 2020 Intel Corporation.
       4                 :            :  *   All rights reserved.
       5                 :            :  */
       6                 :            : 
       7                 :            : #include "spdk_internal/cunit.h"
       8                 :            : 
       9                 :            : #include "common/lib/test_env.c"
      10                 :            : #include "nvme/nvme_cuse.c"
      11                 :            : 
      12         [ #  # ]:          0 : DEFINE_STUB(nvme_io_msg_send, int, (struct spdk_nvme_ctrlr *ctrlr, uint32_t nsid,
      13                 :            :                                     spdk_nvme_io_msg_fn fn, void *arg), 0);
      14                 :            : 
      15         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ctrlr_cmd_admin_raw, int, (struct spdk_nvme_ctrlr *ctrlr,
      16                 :            :                 struct spdk_nvme_cmd *cmd, void *buf, uint32_t len,
      17                 :            :                 spdk_nvme_cmd_cb cb_fn, void *cb_arg), 0);
      18                 :            : 
      19         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ctrlr_cmd_io_raw_with_md, int, (struct spdk_nvme_ctrlr *ctrlr,
      20                 :            :                 struct spdk_nvme_qpair *qpair, struct spdk_nvme_cmd *cmd, void *buf, uint32_t len, void *md_buf,
      21                 :            :                 spdk_nvme_cmd_cb cb_fn, void *cb_arg), 0);
      22                 :            : 
      23         [ -  + ]:          8 : DEFINE_STUB(spdk_nvme_ctrlr_get_num_ns, uint32_t, (struct spdk_nvme_ctrlr *ctrlr), 128);
      24                 :            : 
      25                 :            : static uint32_t g_active_num_ns = 4;
      26                 :            : static uint32_t g_active_nsid_min = 1;
      27                 :            : 
      28                 :            : bool
      29                 :        962 : spdk_nvme_ctrlr_is_active_ns(struct spdk_nvme_ctrlr *ctrlr, uint32_t nsid)
      30                 :            : {
      31   [ +  +  +  + ]:        962 :         return nsid >= g_active_nsid_min && nsid < g_active_num_ns + g_active_nsid_min;
      32                 :            : }
      33                 :            : 
      34                 :            : uint32_t
      35                 :          8 : spdk_nvme_ctrlr_get_first_active_ns(struct spdk_nvme_ctrlr *ctrlr)
      36                 :            : {
      37                 :          8 :         return g_active_nsid_min;
      38                 :            : }
      39                 :            : 
      40                 :            : uint32_t
      41                 :         36 : spdk_nvme_ctrlr_get_next_active_ns(struct spdk_nvme_ctrlr *ctrlr, uint32_t nsid)
      42                 :            : {
      43                 :         36 :         nsid = nsid + 1;
      44                 :            : 
      45         [ +  + ]:         36 :         if (spdk_nvme_ctrlr_is_active_ns(ctrlr, nsid)) {
      46                 :         28 :                 return nsid;
      47                 :            :         }
      48                 :            : 
      49                 :          8 :         return 0;
      50                 :            : }
      51                 :            : 
      52         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ctrlr_reset, int, (struct spdk_nvme_ctrlr *ctrlr), 0);
      53                 :            : 
      54         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ctrlr_reset_subsystem, int, (struct spdk_nvme_ctrlr *ctrlr), 0);
      55                 :            : 
      56         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ns_cmd_read_with_md, int, (struct spdk_nvme_ns *ns,
      57                 :            :                 struct spdk_nvme_qpair *qpair,
      58                 :            :                 void *payload, void *metadata,
      59                 :            :                 uint64_t lba, uint32_t lba_count, spdk_nvme_cmd_cb cb_fn, void *cb_arg,
      60                 :            :                 uint32_t io_flags, uint16_t apptag_mask, uint16_t apptag), 0);
      61                 :            : 
      62         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ns_cmd_write_with_md, int, (struct spdk_nvme_ns *ns,
      63                 :            :                 struct spdk_nvme_qpair *qpair,
      64                 :            :                 void *payload, void *metadata,
      65                 :            :                 uint64_t lba, uint32_t lba_count, spdk_nvme_cmd_cb cb_fn, void *cb_arg,
      66                 :            :                 uint32_t io_flags, uint16_t apptag_mask, uint16_t apptag), 0);
      67                 :            : 
      68         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ns_get_num_sectors, uint64_t, (struct spdk_nvme_ns *ns), 0);
      69                 :            : 
      70         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ns_get_sector_size, uint32_t, (struct spdk_nvme_ns *ns), 0);
      71                 :            : 
      72         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ns_get_md_size, uint32_t, (struct spdk_nvme_ns *ns), 0);
      73                 :            : 
      74                 :          1 : DEFINE_STUB_V(spdk_unaffinitize_thread, (void));
      75                 :            : 
      76         [ #  # ]:          0 : DEFINE_STUB(spdk_nvme_ctrlr_get_ns, struct spdk_nvme_ns *, (struct spdk_nvme_ctrlr *ctrlr,
      77                 :            :                 uint32_t nsid), NULL);
      78                 :            : 
      79         [ -  + ]:          1 : DEFINE_STUB(nvme_io_msg_ctrlr_register, int,
      80                 :            :             (struct spdk_nvme_ctrlr *ctrlr, struct nvme_io_msg_producer *io_msg_producer), 0);
      81                 :            : 
      82                 :          0 : DEFINE_STUB_V(nvme_io_msg_ctrlr_unregister,
      83                 :            :               (struct spdk_nvme_ctrlr *ctrlr, struct nvme_io_msg_producer *io_msg_producer));
      84                 :            : 
      85                 :          0 : DEFINE_STUB_V(nvme_ctrlr_update_namespaces, (struct spdk_nvme_ctrlr *ctrlr));
      86                 :            : 
      87                 :            : static bool
      88                 :        910 : wait_for_file(char *filename, bool exists)
      89                 :            : {
      90                 :            :         int i;
      91                 :            : 
      92         [ +  - ]:        921 :         for (i = 0; i < 1000; i++) {
      93   [ -  +  +  + ]:        921 :                 if ((access(filename, F_OK) != -1) ^ (!exists)) {
      94                 :        910 :                         return true;
      95                 :            :                 }
      96                 :         11 :                 usleep(100);
      97                 :            :         }
      98                 :          0 :         return false;
      99                 :            : }
     100                 :            : 
     101                 :            : static void
     102                 :          7 : verify_devices(struct spdk_nvme_ctrlr *ctrlr)
     103                 :            : {
     104                 :          0 :         char ctrlr_name[256];
     105                 :          0 :         size_t ctrlr_name_size;
     106                 :          0 :         char ctrlr_dev[256];
     107                 :          0 :         char ns_dev[256 + 1 + 10]; /* sizeof ctrl_dev + 'n' + string size of UINT32_MAX */
     108                 :            :         uint32_t nsid, num_ns;
     109                 :            :         int rv;
     110                 :            : 
     111                 :          7 :         ctrlr_name_size = sizeof(ctrlr_name);
     112                 :          7 :         rv = spdk_nvme_cuse_get_ctrlr_name(ctrlr, ctrlr_name, &ctrlr_name_size);
     113         [ -  + ]:          7 :         SPDK_CU_ASSERT_FATAL(rv == 0);
     114                 :            : 
     115         [ -  + ]:          7 :         rv = snprintf(ctrlr_dev, sizeof(ctrlr_dev), "/dev/%s", ctrlr_name);
     116                 :          7 :         CU_ASSERT(rv > 0);
     117                 :          7 :         CU_ASSERT(wait_for_file(ctrlr_dev, true));
     118                 :            : 
     119                 :          7 :         num_ns = spdk_nvme_ctrlr_get_num_ns(ctrlr);
     120                 :            : 
     121         [ +  + ]:        903 :         for (nsid = 1; nsid <= num_ns; nsid++) {
     122         [ -  + ]:        896 :                 snprintf(ns_dev, sizeof(ns_dev), "%sn%" PRIu32, ctrlr_dev, nsid);
     123         [ +  + ]:        896 :                 if (spdk_nvme_ctrlr_is_active_ns(ctrlr, nsid)) {
     124                 :         31 :                         CU_ASSERT(wait_for_file(ns_dev, true));
     125                 :            :                 } else {
     126                 :        865 :                         CU_ASSERT(wait_for_file(ns_dev, false));
     127                 :            :                 }
     128                 :            :         }
     129                 :            : 
     130                 :            :         /* Next one should never exist */
     131         [ -  + ]:          7 :         snprintf(ns_dev, sizeof(ns_dev), "%sn%" PRIu32, ctrlr_dev, nsid);
     132                 :          7 :         CU_ASSERT(wait_for_file(ns_dev, false));
     133                 :          7 : }
     134                 :            : 
     135                 :            : static void
     136                 :          1 : test_cuse_update(void)
     137                 :            : {
     138                 :            :         int rc;
     139                 :          1 :         struct spdk_nvme_ctrlr  ctrlr = {};
     140                 :            : 
     141                 :          1 :         rc = spdk_nvme_cuse_register(&ctrlr);
     142                 :          1 :         CU_ASSERT(rc == 0);
     143                 :            : 
     144                 :          1 :         g_active_num_ns = 4;
     145                 :          1 :         g_active_nsid_min = 1;
     146                 :          1 :         nvme_cuse_update(&ctrlr);
     147                 :          1 :         verify_devices(&ctrlr);
     148                 :            : 
     149                 :          1 :         g_active_num_ns = 0;
     150                 :          1 :         g_active_nsid_min = 1;
     151                 :          1 :         nvme_cuse_update(&ctrlr);
     152                 :          1 :         verify_devices(&ctrlr);
     153                 :            : 
     154                 :          1 :         g_active_num_ns = 4;
     155                 :          1 :         g_active_nsid_min = spdk_nvme_ctrlr_get_num_ns(&ctrlr) - g_active_num_ns;
     156                 :          1 :         nvme_cuse_update(&ctrlr);
     157                 :          1 :         verify_devices(&ctrlr);
     158                 :            : 
     159                 :          1 :         g_active_num_ns = 2;
     160                 :          1 :         g_active_nsid_min = 2;
     161                 :          1 :         nvme_cuse_update(&ctrlr);
     162                 :          1 :         verify_devices(&ctrlr);
     163                 :            : 
     164                 :          1 :         g_active_num_ns = 10;
     165                 :          1 :         g_active_nsid_min = 5;
     166                 :          1 :         nvme_cuse_update(&ctrlr);
     167                 :          1 :         verify_devices(&ctrlr);
     168                 :            : 
     169                 :          1 :         g_active_num_ns = 5;
     170                 :          1 :         g_active_nsid_min = 3;
     171                 :          1 :         nvme_cuse_update(&ctrlr);
     172                 :          1 :         verify_devices(&ctrlr);
     173                 :            : 
     174                 :          1 :         g_active_num_ns = 6;
     175                 :          1 :         g_active_nsid_min = 1;
     176                 :          1 :         nvme_cuse_update(&ctrlr);
     177                 :          1 :         verify_devices(&ctrlr);
     178                 :            : 
     179                 :          1 :         nvme_cuse_stop(&ctrlr);
     180                 :          1 : }
     181                 :            : 
     182                 :            : int
     183                 :          1 : main(int argc, char **argv)
     184                 :            : {
     185                 :          1 :         CU_pSuite       suite = NULL;
     186                 :            :         unsigned int    num_failures;
     187                 :            : 
     188                 :          1 :         CU_initialize_registry();
     189                 :          1 :         suite = CU_add_suite("nvme_cuse", NULL, NULL);
     190                 :          1 :         CU_ADD_TEST(suite, test_cuse_update);
     191                 :            : 
     192                 :          1 :         num_failures = spdk_ut_run_tests(argc, argv, NULL);
     193                 :          1 :         CU_cleanup_registry();
     194                 :          1 :         return num_failures;
     195                 :            : }

Generated by: LCOV version 1.14