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 DEFINE_STUB(spdk_bdev_desc_get_bdev, struct spdk_bdev *, (struct spdk_bdev_desc *desc), NULL); 27 28 uint64_t 29 spdk_bdev_get_zone_size(const struct spdk_bdev *bdev) 30 { 31 return g_geo.zone_size; 32 } 33 34 uint32_t 35 spdk_bdev_get_optimal_open_zones(const struct spdk_bdev *bdev) 36 { 37 return g_geo.optimal_open_zones; 38 } 39 40 DEFINE_RETURN_MOCK(ftl_mempool_get, void *); 41 void * 42 ftl_mempool_get(struct ftl_mempool *mpool) 43 { 44 return spdk_mempool_get((struct spdk_mempool *)mpool); 45 } 46 47 void 48 ftl_mempool_put(struct ftl_mempool *mpool, void *element) 49 { 50 spdk_mempool_put((struct spdk_mempool *)mpool, element); 51 } 52 53 ftl_df_obj_id 54 ftl_mempool_get_df_obj_id(struct ftl_mempool *mpool, void *df_obj_ptr) 55 { 56 return (ftl_df_obj_id)df_obj_ptr; 57 } 58 59 struct spdk_ftl_dev * 60 test_init_ftl_dev(const struct base_bdev_geometry *geo) 61 { 62 struct spdk_ftl_dev *dev; 63 64 dev = calloc(1, sizeof(*dev)); 65 SPDK_CU_ASSERT_FATAL(dev != NULL); 66 67 dev->xfer_size = geo->write_unit_size; 68 dev->core_thread = spdk_thread_create("unit_test_thread", NULL); 69 spdk_set_thread(dev->core_thread); 70 dev->ioch = calloc(1, SPDK_IO_CHANNEL_STRUCT_SIZE 71 + sizeof(struct ftl_io_channel *)); 72 dev->num_bands = geo->blockcnt / (geo->zone_size * geo->optimal_open_zones); 73 dev->bands = calloc(dev->num_bands, sizeof(*dev->bands)); 74 SPDK_CU_ASSERT_FATAL(dev->bands != NULL); 75 76 dev->layout.base.total_blocks = UINT64_MAX; 77 78 for (size_t i = 0; i < dev->num_bands; i++) { 79 struct ftl_band_md *md; 80 int ret; 81 82 ret = posix_memalign((void **)&md, FTL_BLOCK_SIZE, sizeof(*md)); 83 SPDK_CU_ASSERT_FATAL(ret == 0); 84 memset(md, 0, sizeof(*md)); 85 dev->bands[i].md = md; 86 } 87 88 dev->p2l_pool = (struct ftl_mempool *)spdk_mempool_create("ftl_ut", 2, 0x210200, 89 SPDK_MEMPOOL_DEFAULT_CACHE_SIZE, 90 SPDK_ENV_SOCKET_ID_ANY); 91 SPDK_CU_ASSERT_FATAL(dev->p2l_pool != NULL); 92 93 TAILQ_INIT(&dev->free_bands); 94 TAILQ_INIT(&dev->shut_bands); 95 96 /* Cache frequently used values */ 97 dev->num_blocks_in_band = ftl_calculate_num_blocks_in_band(dev->base_bdev_desc); 98 dev->is_zoned = spdk_bdev_is_zoned(spdk_bdev_desc_get_bdev(dev->base_bdev_desc)); 99 100 return dev; 101 } 102 103 struct ftl_band * 104 test_init_ftl_band(struct spdk_ftl_dev *dev, size_t id, size_t zone_size) 105 { 106 struct ftl_band *band; 107 108 SPDK_CU_ASSERT_FATAL(dev != NULL); 109 SPDK_CU_ASSERT_FATAL(id < dev->num_bands); 110 111 band = &dev->bands[id]; 112 band->dev = dev; 113 band->id = id; 114 115 band->md->state = FTL_BAND_STATE_CLOSED; 116 band->md->df_p2l_map = FTL_DF_OBJ_ID_INVALID; 117 TAILQ_INSERT_HEAD(&dev->shut_bands, band, queue_entry); 118 119 band->p2l_map.valid = (struct ftl_bitmap *)spdk_bit_array_create(ftl_get_num_blocks_in_band(dev)); 120 SPDK_CU_ASSERT_FATAL(band->p2l_map.valid != NULL); 121 122 band->start_addr = zone_size * id; 123 124 return band; 125 } 126 127 void 128 test_free_ftl_dev(struct spdk_ftl_dev *dev) 129 { 130 struct spdk_thread *thread; 131 132 SPDK_CU_ASSERT_FATAL(dev != NULL); 133 free(dev->ioch); 134 135 thread = dev->core_thread; 136 137 spdk_set_thread(thread); 138 spdk_thread_exit(thread); 139 while (!spdk_thread_is_exited(thread)) { 140 spdk_thread_poll(thread, 0, 0); 141 } 142 spdk_thread_destroy(thread); 143 spdk_mempool_free((struct spdk_mempool *)dev->p2l_pool); 144 for (size_t i = 0; i < dev->num_bands; i++) { 145 free(dev->bands[i].md); 146 } 147 free(dev->bands); 148 free(dev); 149 } 150 151 void 152 test_free_ftl_band(struct ftl_band *band) 153 { 154 SPDK_CU_ASSERT_FATAL(band != NULL); 155 spdk_bit_array_free((struct spdk_bit_array **)&band->p2l_map.valid); 156 spdk_dma_free(band->p2l_map.band_dma_md); 157 158 band->p2l_map.band_dma_md = NULL; 159 } 160 161 uint64_t 162 test_offset_from_addr(ftl_addr addr, struct ftl_band *band) 163 { 164 struct spdk_ftl_dev *dev = band->dev; 165 166 CU_ASSERT_EQUAL(ftl_addr_get_band(dev, addr), band->id); 167 168 return addr - band->id * ftl_get_num_blocks_in_band(dev); 169 } 170