Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause 2 : : * Copyright (C) 2018 Intel Corporation. 3 : : * All rights reserved. 4 : : */ 5 : : 6 : : #include "spdk/ftl.h" 7 : : #include "ftl/ftl_core.h" 8 : : 9 : : #include "ftl/mngt/ftl_mngt_bdev.c" 10 : : 11 : : struct base_bdev_geometry { 12 : : size_t write_unit_size; 13 : : size_t zone_size; 14 : : size_t optimal_open_zones; 15 : : size_t blockcnt; 16 : : }; 17 : : 18 : : extern struct base_bdev_geometry g_geo; 19 : : 20 : : struct spdk_ftl_dev *test_init_ftl_dev(const struct base_bdev_geometry *geo); 21 : : struct ftl_band *test_init_ftl_band(struct spdk_ftl_dev *dev, size_t id, size_t zone_size); 22 : : void test_free_ftl_dev(struct spdk_ftl_dev *dev); 23 : : void test_free_ftl_band(struct ftl_band *band); 24 : : uint64_t test_offset_from_addr(ftl_addr addr, struct ftl_band *band); 25 : : 26 [ - + ]: 30 : DEFINE_STUB(spdk_bdev_desc_get_bdev, struct spdk_bdev *, (struct spdk_bdev_desc *desc), NULL); 27 : : 28 [ # # ]: 0 : DEFINE_STUB(ftl_nv_cache_device_get_type_by_bdev, const struct ftl_nv_cache_device_type *, 29 : : (struct spdk_ftl_dev *dev, struct spdk_bdev *bdev), NULL); 30 : : 31 : : uint64_t 32 : 0 : spdk_bdev_get_zone_size(const struct spdk_bdev *bdev) 33 : : { 34 : 0 : return g_geo.zone_size; 35 : : } 36 : : 37 : : uint32_t 38 : 0 : spdk_bdev_get_optimal_open_zones(const struct spdk_bdev *bdev) 39 : : { 40 : 0 : return g_geo.optimal_open_zones; 41 : : } 42 : : 43 [ # # ]: 0 : DEFINE_RETURN_MOCK(ftl_mempool_get, void *); 44 : : void * 45 : 60 : ftl_mempool_get(struct ftl_mempool *mpool) 46 : : { 47 : 60 : return spdk_mempool_get((struct spdk_mempool *)mpool); 48 : : } 49 : : 50 : : void 51 : 60 : ftl_mempool_put(struct ftl_mempool *mpool, void *element) 52 : : { 53 : 60 : spdk_mempool_put((struct spdk_mempool *)mpool, element); 54 : 60 : } 55 : : 56 : : ftl_df_obj_id 57 : 30 : ftl_mempool_get_df_obj_id(struct ftl_mempool *mpool, void *df_obj_ptr) 58 : : { 59 : 30 : return (ftl_df_obj_id)df_obj_ptr; 60 : : } 61 : : 62 : : struct spdk_ftl_dev * 63 : 30 : test_init_ftl_dev(const struct base_bdev_geometry *geo) 64 : : { 65 : : struct spdk_ftl_dev *dev; 66 : : 67 : 30 : dev = calloc(1, sizeof(*dev)); 68 [ - + ]: 30 : SPDK_CU_ASSERT_FATAL(dev != NULL); 69 : : 70 : 30 : dev->xfer_size = geo->write_unit_size; 71 : 30 : dev->core_thread = spdk_thread_create("unit_test_thread", NULL); 72 : 30 : spdk_set_thread(dev->core_thread); 73 : 30 : dev->ioch = calloc(1, SPDK_IO_CHANNEL_STRUCT_SIZE 74 : : + sizeof(struct ftl_io_channel *)); 75 [ - + ]: 30 : dev->num_bands = geo->blockcnt / (geo->zone_size * geo->optimal_open_zones); 76 : 30 : dev->bands = calloc(dev->num_bands, sizeof(*dev->bands)); 77 [ - + ]: 30 : SPDK_CU_ASSERT_FATAL(dev->bands != NULL); 78 : : 79 : 30 : dev->layout.base.total_blocks = UINT64_MAX; 80 : : 81 [ + + ]: 1320 : for (size_t i = 0; i < dev->num_bands; i++) { 82 : 1032 : struct ftl_band_md *md; 83 : : int ret; 84 : : 85 [ - + ]: 1290 : ret = posix_memalign((void **)&md, FTL_BLOCK_SIZE, sizeof(*md)); 86 [ - + ]: 1290 : SPDK_CU_ASSERT_FATAL(ret == 0); 87 [ - + ]: 1290 : memset(md, 0, sizeof(*md)); 88 : 1290 : dev->bands[i].md = md; 89 : : } 90 : : 91 [ + + ]: 570 : for (int i = 0; i < FTL_LAYOUT_REGION_TYPE_MAX; i++) { 92 : 540 : dev->layout.region[i].type = i; 93 : : } 94 : : 95 : 30 : dev->p2l_pool = (struct ftl_mempool *)spdk_mempool_create("ftl_ut", 2, 0x210200, 96 : : SPDK_MEMPOOL_DEFAULT_CACHE_SIZE, 97 : : SPDK_ENV_SOCKET_ID_ANY); 98 [ - + ]: 30 : SPDK_CU_ASSERT_FATAL(dev->p2l_pool != NULL); 99 : : 100 : 30 : TAILQ_INIT(&dev->free_bands); 101 : 30 : TAILQ_INIT(&dev->shut_bands); 102 : : 103 : : /* Cache frequently used values */ 104 : 30 : dev->num_blocks_in_band = ftl_calculate_num_blocks_in_band(dev->base_bdev_desc); 105 : 30 : dev->is_zoned = spdk_bdev_is_zoned(spdk_bdev_desc_get_bdev(dev->base_bdev_desc)); 106 : : 107 : 30 : return dev; 108 : : } 109 : : 110 : : struct ftl_band * 111 : 30 : test_init_ftl_band(struct spdk_ftl_dev *dev, size_t id, size_t zone_size) 112 : : { 113 : : struct ftl_band *band; 114 : : 115 [ - + ]: 30 : SPDK_CU_ASSERT_FATAL(dev != NULL); 116 [ - + ]: 30 : SPDK_CU_ASSERT_FATAL(id < dev->num_bands); 117 : : 118 : 30 : band = &dev->bands[id]; 119 : 30 : band->dev = dev; 120 : 30 : band->id = id; 121 : : 122 : 30 : band->md->state = FTL_BAND_STATE_CLOSED; 123 : 30 : band->md->df_p2l_map = FTL_DF_OBJ_ID_INVALID; 124 [ - + ]: 30 : TAILQ_INSERT_HEAD(&dev->shut_bands, band, queue_entry); 125 : : 126 : 30 : band->p2l_map.valid = (struct ftl_bitmap *)spdk_bit_array_create(ftl_get_num_blocks_in_band(dev)); 127 [ - + ]: 30 : SPDK_CU_ASSERT_FATAL(band->p2l_map.valid != NULL); 128 : : 129 : 30 : band->start_addr = zone_size * id; 130 : : 131 : 30 : return band; 132 : : } 133 : : 134 : : void 135 : 30 : test_free_ftl_dev(struct spdk_ftl_dev *dev) 136 : : { 137 : : struct spdk_thread *thread; 138 : : 139 [ - + ]: 30 : SPDK_CU_ASSERT_FATAL(dev != NULL); 140 : 30 : free(dev->ioch); 141 : : 142 : 30 : thread = dev->core_thread; 143 : : 144 : 30 : spdk_set_thread(thread); 145 : 30 : spdk_thread_exit(thread); 146 [ + + ]: 60 : while (!spdk_thread_is_exited(thread)) { 147 : 30 : spdk_thread_poll(thread, 0, 0); 148 : : } 149 : 30 : spdk_thread_destroy(thread); 150 : 30 : spdk_mempool_free((struct spdk_mempool *)dev->p2l_pool); 151 [ + + ]: 1320 : for (size_t i = 0; i < dev->num_bands; i++) { 152 : 1290 : free(dev->bands[i].md); 153 : : } 154 : 30 : free(dev->bands); 155 : 30 : free(dev); 156 : 30 : } 157 : : 158 : : void 159 : 30 : test_free_ftl_band(struct ftl_band *band) 160 : : { 161 [ - + ]: 30 : SPDK_CU_ASSERT_FATAL(band != NULL); 162 : 30 : spdk_bit_array_free((struct spdk_bit_array **)&band->p2l_map.valid); 163 : 30 : spdk_dma_free(band->p2l_map.band_dma_md); 164 : : 165 : 30 : band->p2l_map.band_dma_md = NULL; 166 : 30 : } 167 : : 168 : : uint64_t 169 : 50075 : test_offset_from_addr(ftl_addr addr, struct ftl_band *band) 170 : : { 171 : 50075 : struct spdk_ftl_dev *dev = band->dev; 172 : : 173 : 50075 : CU_ASSERT_EQUAL(ftl_addr_get_band(dev, addr), band->id); 174 : : 175 : 50075 : return addr - band->id * ftl_get_num_blocks_in_band(dev); 176 : : }