1bae9a0aaSDmitry Fomichev // SPDX-License-Identifier: GPL-2.0-only 23b1a94c8SDamien Le Moal /* 33b1a94c8SDamien Le Moal * Copyright (C) 2017 Western Digital Corporation or its affiliates. 43b1a94c8SDamien Le Moal * 53b1a94c8SDamien Le Moal * This file is released under the GPL. 63b1a94c8SDamien Le Moal */ 73b1a94c8SDamien Le Moal 83b1a94c8SDamien Le Moal #include "dm-zoned.h" 93b1a94c8SDamien Le Moal 103b1a94c8SDamien Le Moal #include <linux/module.h> 113b1a94c8SDamien Le Moal #include <linux/crc32.h> 12bd976e52SDamien Le Moal #include <linux/sched/mm.h> 133b1a94c8SDamien Le Moal 143b1a94c8SDamien Le Moal #define DM_MSG_PREFIX "zoned metadata" 153b1a94c8SDamien Le Moal 163b1a94c8SDamien Le Moal /* 173b1a94c8SDamien Le Moal * Metadata version. 183b1a94c8SDamien Le Moal */ 193b1a94c8SDamien Le Moal #define DMZ_META_VER 1 203b1a94c8SDamien Le Moal 213b1a94c8SDamien Le Moal /* 223b1a94c8SDamien Le Moal * On-disk super block magic. 233b1a94c8SDamien Le Moal */ 243b1a94c8SDamien Le Moal #define DMZ_MAGIC ((((unsigned int)('D')) << 24) | \ 253b1a94c8SDamien Le Moal (((unsigned int)('Z')) << 16) | \ 263b1a94c8SDamien Le Moal (((unsigned int)('B')) << 8) | \ 273b1a94c8SDamien Le Moal ((unsigned int)('D'))) 283b1a94c8SDamien Le Moal 293b1a94c8SDamien Le Moal /* 303b1a94c8SDamien Le Moal * On disk super block. 313b1a94c8SDamien Le Moal * This uses only 512 B but uses on disk a full 4KB block. This block is 323b1a94c8SDamien Le Moal * followed on disk by the mapping table of chunks to zones and the bitmap 333b1a94c8SDamien Le Moal * blocks indicating zone block validity. 343b1a94c8SDamien Le Moal * The overall resulting metadata format is: 353b1a94c8SDamien Le Moal * (1) Super block (1 block) 363b1a94c8SDamien Le Moal * (2) Chunk mapping table (nr_map_blocks) 373b1a94c8SDamien Le Moal * (3) Bitmap blocks (nr_bitmap_blocks) 38ad1bd578SDmitry Fomichev * All metadata blocks are stored in conventional zones, starting from 393b1a94c8SDamien Le Moal * the first conventional zone found on disk. 403b1a94c8SDamien Le Moal */ 413b1a94c8SDamien Le Moal struct dmz_super { 423b1a94c8SDamien Le Moal /* Magic number */ 433b1a94c8SDamien Le Moal __le32 magic; /* 4 */ 443b1a94c8SDamien Le Moal 453b1a94c8SDamien Le Moal /* Metadata version number */ 463b1a94c8SDamien Le Moal __le32 version; /* 8 */ 473b1a94c8SDamien Le Moal 483b1a94c8SDamien Le Moal /* Generation number */ 493b1a94c8SDamien Le Moal __le64 gen; /* 16 */ 503b1a94c8SDamien Le Moal 513b1a94c8SDamien Le Moal /* This block number */ 523b1a94c8SDamien Le Moal __le64 sb_block; /* 24 */ 533b1a94c8SDamien Le Moal 543b1a94c8SDamien Le Moal /* The number of metadata blocks, including this super block */ 553b1a94c8SDamien Le Moal __le32 nr_meta_blocks; /* 28 */ 563b1a94c8SDamien Le Moal 573b1a94c8SDamien Le Moal /* The number of sequential zones reserved for reclaim */ 583b1a94c8SDamien Le Moal __le32 nr_reserved_seq; /* 32 */ 593b1a94c8SDamien Le Moal 603b1a94c8SDamien Le Moal /* The number of entries in the mapping table */ 613b1a94c8SDamien Le Moal __le32 nr_chunks; /* 36 */ 623b1a94c8SDamien Le Moal 633b1a94c8SDamien Le Moal /* The number of blocks used for the chunk mapping table */ 643b1a94c8SDamien Le Moal __le32 nr_map_blocks; /* 40 */ 653b1a94c8SDamien Le Moal 663b1a94c8SDamien Le Moal /* The number of blocks used for the block bitmaps */ 673b1a94c8SDamien Le Moal __le32 nr_bitmap_blocks; /* 44 */ 683b1a94c8SDamien Le Moal 693b1a94c8SDamien Le Moal /* Checksum */ 703b1a94c8SDamien Le Moal __le32 crc; /* 48 */ 713b1a94c8SDamien Le Moal 723b1a94c8SDamien Le Moal /* Padding to full 512B sector */ 733b1a94c8SDamien Le Moal u8 reserved[464]; /* 512 */ 743b1a94c8SDamien Le Moal }; 753b1a94c8SDamien Le Moal 763b1a94c8SDamien Le Moal /* 773b1a94c8SDamien Le Moal * Chunk mapping entry: entries are indexed by chunk number 783b1a94c8SDamien Le Moal * and give the zone ID (dzone_id) mapping the chunk on disk. 793b1a94c8SDamien Le Moal * This zone may be sequential or random. If it is a sequential 803b1a94c8SDamien Le Moal * zone, a second zone (bzone_id) used as a write buffer may 813b1a94c8SDamien Le Moal * also be specified. This second zone will always be a randomly 823b1a94c8SDamien Le Moal * writeable zone. 833b1a94c8SDamien Le Moal */ 843b1a94c8SDamien Le Moal struct dmz_map { 853b1a94c8SDamien Le Moal __le32 dzone_id; 863b1a94c8SDamien Le Moal __le32 bzone_id; 873b1a94c8SDamien Le Moal }; 883b1a94c8SDamien Le Moal 893b1a94c8SDamien Le Moal /* 903b1a94c8SDamien Le Moal * Chunk mapping table metadata: 512 8-bytes entries per 4KB block. 913b1a94c8SDamien Le Moal */ 923b1a94c8SDamien Le Moal #define DMZ_MAP_ENTRIES (DMZ_BLOCK_SIZE / sizeof(struct dmz_map)) 933b1a94c8SDamien Le Moal #define DMZ_MAP_ENTRIES_SHIFT (ilog2(DMZ_MAP_ENTRIES)) 943b1a94c8SDamien Le Moal #define DMZ_MAP_ENTRIES_MASK (DMZ_MAP_ENTRIES - 1) 953b1a94c8SDamien Le Moal #define DMZ_MAP_UNMAPPED UINT_MAX 963b1a94c8SDamien Le Moal 973b1a94c8SDamien Le Moal /* 983b1a94c8SDamien Le Moal * Meta data block descriptor (for cached metadata blocks). 993b1a94c8SDamien Le Moal */ 1003b1a94c8SDamien Le Moal struct dmz_mblock { 1013b1a94c8SDamien Le Moal struct rb_node node; 1023b1a94c8SDamien Le Moal struct list_head link; 1033b1a94c8SDamien Le Moal sector_t no; 10433c2865fSDamien Le Moal unsigned int ref; 1053b1a94c8SDamien Le Moal unsigned long state; 1063b1a94c8SDamien Le Moal struct page *page; 1073b1a94c8SDamien Le Moal void *data; 1083b1a94c8SDamien Le Moal }; 1093b1a94c8SDamien Le Moal 1103b1a94c8SDamien Le Moal /* 1113b1a94c8SDamien Le Moal * Metadata block state flags. 1123b1a94c8SDamien Le Moal */ 1133b1a94c8SDamien Le Moal enum { 1143b1a94c8SDamien Le Moal DMZ_META_DIRTY, 1153b1a94c8SDamien Le Moal DMZ_META_READING, 1163b1a94c8SDamien Le Moal DMZ_META_WRITING, 1173b1a94c8SDamien Le Moal DMZ_META_ERROR, 1183b1a94c8SDamien Le Moal }; 1193b1a94c8SDamien Le Moal 1203b1a94c8SDamien Le Moal /* 1213b1a94c8SDamien Le Moal * Super block information (one per metadata set). 1223b1a94c8SDamien Le Moal */ 1233b1a94c8SDamien Le Moal struct dmz_sb { 1243b1a94c8SDamien Le Moal sector_t block; 1253b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 1263b1a94c8SDamien Le Moal struct dmz_super *sb; 1273b1a94c8SDamien Le Moal }; 1283b1a94c8SDamien Le Moal 1293b1a94c8SDamien Le Moal /* 1303b1a94c8SDamien Le Moal * In-memory metadata. 1313b1a94c8SDamien Le Moal */ 1323b1a94c8SDamien Le Moal struct dmz_metadata { 1333b1a94c8SDamien Le Moal struct dmz_dev *dev; 1343b1a94c8SDamien Le Moal 1353b1a94c8SDamien Le Moal sector_t zone_bitmap_size; 1363b1a94c8SDamien Le Moal unsigned int zone_nr_bitmap_blocks; 1373b1a94c8SDamien Le Moal 1383b1a94c8SDamien Le Moal unsigned int nr_bitmap_blocks; 1393b1a94c8SDamien Le Moal unsigned int nr_map_blocks; 1403b1a94c8SDamien Le Moal 1413b1a94c8SDamien Le Moal unsigned int nr_useable_zones; 1423b1a94c8SDamien Le Moal unsigned int nr_meta_blocks; 1433b1a94c8SDamien Le Moal unsigned int nr_meta_zones; 1443b1a94c8SDamien Le Moal unsigned int nr_data_zones; 1453b1a94c8SDamien Le Moal unsigned int nr_rnd_zones; 1463b1a94c8SDamien Le Moal unsigned int nr_reserved_seq; 1473b1a94c8SDamien Le Moal unsigned int nr_chunks; 1483b1a94c8SDamien Le Moal 1493b1a94c8SDamien Le Moal /* Zone information array */ 1503b1a94c8SDamien Le Moal struct dm_zone *zones; 1513b1a94c8SDamien Le Moal 1523b1a94c8SDamien Le Moal struct dm_zone *sb_zone; 1533b1a94c8SDamien Le Moal struct dmz_sb sb[2]; 1543b1a94c8SDamien Le Moal unsigned int mblk_primary; 1553b1a94c8SDamien Le Moal u64 sb_gen; 1563b1a94c8SDamien Le Moal unsigned int min_nr_mblks; 1573b1a94c8SDamien Le Moal unsigned int max_nr_mblks; 1583b1a94c8SDamien Le Moal atomic_t nr_mblks; 1593b1a94c8SDamien Le Moal struct rw_semaphore mblk_sem; 1603b1a94c8SDamien Le Moal struct mutex mblk_flush_lock; 1613b1a94c8SDamien Le Moal spinlock_t mblk_lock; 1623b1a94c8SDamien Le Moal struct rb_root mblk_rbtree; 1633b1a94c8SDamien Le Moal struct list_head mblk_lru_list; 1643b1a94c8SDamien Le Moal struct list_head mblk_dirty_list; 1653b1a94c8SDamien Le Moal struct shrinker mblk_shrinker; 1663b1a94c8SDamien Le Moal 1673b1a94c8SDamien Le Moal /* Zone allocation management */ 1683b1a94c8SDamien Le Moal struct mutex map_lock; 1693b1a94c8SDamien Le Moal struct dmz_mblock **map_mblk; 1703b1a94c8SDamien Le Moal unsigned int nr_rnd; 1713b1a94c8SDamien Le Moal atomic_t unmap_nr_rnd; 1723b1a94c8SDamien Le Moal struct list_head unmap_rnd_list; 1733b1a94c8SDamien Le Moal struct list_head map_rnd_list; 1743b1a94c8SDamien Le Moal 1753b1a94c8SDamien Le Moal unsigned int nr_seq; 1763b1a94c8SDamien Le Moal atomic_t unmap_nr_seq; 1773b1a94c8SDamien Le Moal struct list_head unmap_seq_list; 1783b1a94c8SDamien Le Moal struct list_head map_seq_list; 1793b1a94c8SDamien Le Moal 1803b1a94c8SDamien Le Moal atomic_t nr_reserved_seq_zones; 1813b1a94c8SDamien Le Moal struct list_head reserved_seq_zones_list; 1823b1a94c8SDamien Le Moal 1833b1a94c8SDamien Le Moal wait_queue_head_t free_wq; 1843b1a94c8SDamien Le Moal }; 1853b1a94c8SDamien Le Moal 1863b1a94c8SDamien Le Moal /* 1873b1a94c8SDamien Le Moal * Various accessors 1883b1a94c8SDamien Le Moal */ 1893b1a94c8SDamien Le Moal unsigned int dmz_id(struct dmz_metadata *zmd, struct dm_zone *zone) 1903b1a94c8SDamien Le Moal { 1913b1a94c8SDamien Le Moal return ((unsigned int)(zone - zmd->zones)); 1923b1a94c8SDamien Le Moal } 1933b1a94c8SDamien Le Moal 1943b1a94c8SDamien Le Moal sector_t dmz_start_sect(struct dmz_metadata *zmd, struct dm_zone *zone) 1953b1a94c8SDamien Le Moal { 1963908c983SDamien Le Moal return (sector_t)dmz_id(zmd, zone) << zmd->dev->zone_nr_sectors_shift; 1973b1a94c8SDamien Le Moal } 1983b1a94c8SDamien Le Moal 1993b1a94c8SDamien Le Moal sector_t dmz_start_block(struct dmz_metadata *zmd, struct dm_zone *zone) 2003b1a94c8SDamien Le Moal { 2013908c983SDamien Le Moal return (sector_t)dmz_id(zmd, zone) << zmd->dev->zone_nr_blocks_shift; 2023b1a94c8SDamien Le Moal } 2033b1a94c8SDamien Le Moal 2043b1a94c8SDamien Le Moal unsigned int dmz_nr_chunks(struct dmz_metadata *zmd) 2053b1a94c8SDamien Le Moal { 2063b1a94c8SDamien Le Moal return zmd->nr_chunks; 2073b1a94c8SDamien Le Moal } 2083b1a94c8SDamien Le Moal 2093b1a94c8SDamien Le Moal unsigned int dmz_nr_rnd_zones(struct dmz_metadata *zmd) 2103b1a94c8SDamien Le Moal { 2113b1a94c8SDamien Le Moal return zmd->nr_rnd; 2123b1a94c8SDamien Le Moal } 2133b1a94c8SDamien Le Moal 2143b1a94c8SDamien Le Moal unsigned int dmz_nr_unmap_rnd_zones(struct dmz_metadata *zmd) 2153b1a94c8SDamien Le Moal { 2163b1a94c8SDamien Le Moal return atomic_read(&zmd->unmap_nr_rnd); 2173b1a94c8SDamien Le Moal } 2183b1a94c8SDamien Le Moal 2193b1a94c8SDamien Le Moal /* 2203b1a94c8SDamien Le Moal * Lock/unlock mapping table. 2213b1a94c8SDamien Le Moal * The map lock also protects all the zone lists. 2223b1a94c8SDamien Le Moal */ 2233b1a94c8SDamien Le Moal void dmz_lock_map(struct dmz_metadata *zmd) 2243b1a94c8SDamien Le Moal { 2253b1a94c8SDamien Le Moal mutex_lock(&zmd->map_lock); 2263b1a94c8SDamien Le Moal } 2273b1a94c8SDamien Le Moal 2283b1a94c8SDamien Le Moal void dmz_unlock_map(struct dmz_metadata *zmd) 2293b1a94c8SDamien Le Moal { 2303b1a94c8SDamien Le Moal mutex_unlock(&zmd->map_lock); 2313b1a94c8SDamien Le Moal } 2323b1a94c8SDamien Le Moal 2333b1a94c8SDamien Le Moal /* 2343b1a94c8SDamien Le Moal * Lock/unlock metadata access. This is a "read" lock on a semaphore 2353b1a94c8SDamien Le Moal * that prevents metadata flush from running while metadata are being 2363b1a94c8SDamien Le Moal * modified. The actual metadata write mutual exclusion is achieved with 237ad1bd578SDmitry Fomichev * the map lock and zone state management (active and reclaim state are 2383b1a94c8SDamien Le Moal * mutually exclusive). 2393b1a94c8SDamien Le Moal */ 2403b1a94c8SDamien Le Moal void dmz_lock_metadata(struct dmz_metadata *zmd) 2413b1a94c8SDamien Le Moal { 2423b1a94c8SDamien Le Moal down_read(&zmd->mblk_sem); 2433b1a94c8SDamien Le Moal } 2443b1a94c8SDamien Le Moal 2453b1a94c8SDamien Le Moal void dmz_unlock_metadata(struct dmz_metadata *zmd) 2463b1a94c8SDamien Le Moal { 2473b1a94c8SDamien Le Moal up_read(&zmd->mblk_sem); 2483b1a94c8SDamien Le Moal } 2493b1a94c8SDamien Le Moal 2503b1a94c8SDamien Le Moal /* 2513b1a94c8SDamien Le Moal * Lock/unlock flush: prevent concurrent executions 2523b1a94c8SDamien Le Moal * of dmz_flush_metadata as well as metadata modification in reclaim 2533b1a94c8SDamien Le Moal * while flush is being executed. 2543b1a94c8SDamien Le Moal */ 2553b1a94c8SDamien Le Moal void dmz_lock_flush(struct dmz_metadata *zmd) 2563b1a94c8SDamien Le Moal { 2573b1a94c8SDamien Le Moal mutex_lock(&zmd->mblk_flush_lock); 2583b1a94c8SDamien Le Moal } 2593b1a94c8SDamien Le Moal 2603b1a94c8SDamien Le Moal void dmz_unlock_flush(struct dmz_metadata *zmd) 2613b1a94c8SDamien Le Moal { 2623b1a94c8SDamien Le Moal mutex_unlock(&zmd->mblk_flush_lock); 2633b1a94c8SDamien Le Moal } 2643b1a94c8SDamien Le Moal 2653b1a94c8SDamien Le Moal /* 2663b1a94c8SDamien Le Moal * Allocate a metadata block. 2673b1a94c8SDamien Le Moal */ 2683b1a94c8SDamien Le Moal static struct dmz_mblock *dmz_alloc_mblock(struct dmz_metadata *zmd, 2693b1a94c8SDamien Le Moal sector_t mblk_no) 2703b1a94c8SDamien Le Moal { 2713b1a94c8SDamien Le Moal struct dmz_mblock *mblk = NULL; 2723b1a94c8SDamien Le Moal 2733b1a94c8SDamien Le Moal /* See if we can reuse cached blocks */ 2743b1a94c8SDamien Le Moal if (zmd->max_nr_mblks && atomic_read(&zmd->nr_mblks) > zmd->max_nr_mblks) { 2753b1a94c8SDamien Le Moal spin_lock(&zmd->mblk_lock); 2763b1a94c8SDamien Le Moal mblk = list_first_entry_or_null(&zmd->mblk_lru_list, 2773b1a94c8SDamien Le Moal struct dmz_mblock, link); 2783b1a94c8SDamien Le Moal if (mblk) { 2793b1a94c8SDamien Le Moal list_del_init(&mblk->link); 2803b1a94c8SDamien Le Moal rb_erase(&mblk->node, &zmd->mblk_rbtree); 2813b1a94c8SDamien Le Moal mblk->no = mblk_no; 2823b1a94c8SDamien Le Moal } 2833b1a94c8SDamien Le Moal spin_unlock(&zmd->mblk_lock); 2843b1a94c8SDamien Le Moal if (mblk) 2853b1a94c8SDamien Le Moal return mblk; 2863b1a94c8SDamien Le Moal } 2873b1a94c8SDamien Le Moal 2883b1a94c8SDamien Le Moal /* Allocate a new block */ 2893b1a94c8SDamien Le Moal mblk = kmalloc(sizeof(struct dmz_mblock), GFP_NOIO); 2903b1a94c8SDamien Le Moal if (!mblk) 2913b1a94c8SDamien Le Moal return NULL; 2923b1a94c8SDamien Le Moal 2933b1a94c8SDamien Le Moal mblk->page = alloc_page(GFP_NOIO); 2943b1a94c8SDamien Le Moal if (!mblk->page) { 2953b1a94c8SDamien Le Moal kfree(mblk); 2963b1a94c8SDamien Le Moal return NULL; 2973b1a94c8SDamien Le Moal } 2983b1a94c8SDamien Le Moal 2993b1a94c8SDamien Le Moal RB_CLEAR_NODE(&mblk->node); 3003b1a94c8SDamien Le Moal INIT_LIST_HEAD(&mblk->link); 30133c2865fSDamien Le Moal mblk->ref = 0; 3023b1a94c8SDamien Le Moal mblk->state = 0; 3033b1a94c8SDamien Le Moal mblk->no = mblk_no; 3043b1a94c8SDamien Le Moal mblk->data = page_address(mblk->page); 3053b1a94c8SDamien Le Moal 3063b1a94c8SDamien Le Moal atomic_inc(&zmd->nr_mblks); 3073b1a94c8SDamien Le Moal 3083b1a94c8SDamien Le Moal return mblk; 3093b1a94c8SDamien Le Moal } 3103b1a94c8SDamien Le Moal 3113b1a94c8SDamien Le Moal /* 3123b1a94c8SDamien Le Moal * Free a metadata block. 3133b1a94c8SDamien Le Moal */ 3143b1a94c8SDamien Le Moal static void dmz_free_mblock(struct dmz_metadata *zmd, struct dmz_mblock *mblk) 3153b1a94c8SDamien Le Moal { 3163b1a94c8SDamien Le Moal __free_pages(mblk->page, 0); 3173b1a94c8SDamien Le Moal kfree(mblk); 3183b1a94c8SDamien Le Moal 3193b1a94c8SDamien Le Moal atomic_dec(&zmd->nr_mblks); 3203b1a94c8SDamien Le Moal } 3213b1a94c8SDamien Le Moal 3223b1a94c8SDamien Le Moal /* 3233b1a94c8SDamien Le Moal * Insert a metadata block in the rbtree. 3243b1a94c8SDamien Le Moal */ 3253b1a94c8SDamien Le Moal static void dmz_insert_mblock(struct dmz_metadata *zmd, struct dmz_mblock *mblk) 3263b1a94c8SDamien Le Moal { 3273b1a94c8SDamien Le Moal struct rb_root *root = &zmd->mblk_rbtree; 3283b1a94c8SDamien Le Moal struct rb_node **new = &(root->rb_node), *parent = NULL; 3293b1a94c8SDamien Le Moal struct dmz_mblock *b; 3303b1a94c8SDamien Le Moal 3313b1a94c8SDamien Le Moal /* Figure out where to put the new node */ 3323b1a94c8SDamien Le Moal while (*new) { 3333b1a94c8SDamien Le Moal b = container_of(*new, struct dmz_mblock, node); 3343b1a94c8SDamien Le Moal parent = *new; 3353b1a94c8SDamien Le Moal new = (b->no < mblk->no) ? &((*new)->rb_left) : &((*new)->rb_right); 3363b1a94c8SDamien Le Moal } 3373b1a94c8SDamien Le Moal 3383b1a94c8SDamien Le Moal /* Add new node and rebalance tree */ 3393b1a94c8SDamien Le Moal rb_link_node(&mblk->node, parent, new); 3403b1a94c8SDamien Le Moal rb_insert_color(&mblk->node, root); 3413b1a94c8SDamien Le Moal } 3423b1a94c8SDamien Le Moal 3433b1a94c8SDamien Le Moal /* 3443d4e7383SDamien Le Moal * Lookup a metadata block in the rbtree. If the block is found, increment 3453d4e7383SDamien Le Moal * its reference count. 3463b1a94c8SDamien Le Moal */ 3473d4e7383SDamien Le Moal static struct dmz_mblock *dmz_get_mblock_fast(struct dmz_metadata *zmd, 3483b1a94c8SDamien Le Moal sector_t mblk_no) 3493b1a94c8SDamien Le Moal { 3503b1a94c8SDamien Le Moal struct rb_root *root = &zmd->mblk_rbtree; 3513b1a94c8SDamien Le Moal struct rb_node *node = root->rb_node; 3523b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 3533b1a94c8SDamien Le Moal 3543b1a94c8SDamien Le Moal while (node) { 3553b1a94c8SDamien Le Moal mblk = container_of(node, struct dmz_mblock, node); 3563d4e7383SDamien Le Moal if (mblk->no == mblk_no) { 3573d4e7383SDamien Le Moal /* 3583d4e7383SDamien Le Moal * If this is the first reference to the block, 3593d4e7383SDamien Le Moal * remove it from the LRU list. 3603d4e7383SDamien Le Moal */ 3613d4e7383SDamien Le Moal mblk->ref++; 3623d4e7383SDamien Le Moal if (mblk->ref == 1 && 3633d4e7383SDamien Le Moal !test_bit(DMZ_META_DIRTY, &mblk->state)) 3643d4e7383SDamien Le Moal list_del_init(&mblk->link); 3653b1a94c8SDamien Le Moal return mblk; 3663d4e7383SDamien Le Moal } 3673b1a94c8SDamien Le Moal node = (mblk->no < mblk_no) ? node->rb_left : node->rb_right; 3683b1a94c8SDamien Le Moal } 3693b1a94c8SDamien Le Moal 3703b1a94c8SDamien Le Moal return NULL; 3713b1a94c8SDamien Le Moal } 3723b1a94c8SDamien Le Moal 3733b1a94c8SDamien Le Moal /* 3743b1a94c8SDamien Le Moal * Metadata block BIO end callback. 3753b1a94c8SDamien Le Moal */ 3763b1a94c8SDamien Le Moal static void dmz_mblock_bio_end_io(struct bio *bio) 3773b1a94c8SDamien Le Moal { 3783b1a94c8SDamien Le Moal struct dmz_mblock *mblk = bio->bi_private; 3793b1a94c8SDamien Le Moal int flag; 3803b1a94c8SDamien Le Moal 3813b1a94c8SDamien Le Moal if (bio->bi_status) 3823b1a94c8SDamien Le Moal set_bit(DMZ_META_ERROR, &mblk->state); 3833b1a94c8SDamien Le Moal 3843b1a94c8SDamien Le Moal if (bio_op(bio) == REQ_OP_WRITE) 3853b1a94c8SDamien Le Moal flag = DMZ_META_WRITING; 3863b1a94c8SDamien Le Moal else 3873b1a94c8SDamien Le Moal flag = DMZ_META_READING; 3883b1a94c8SDamien Le Moal 3893b1a94c8SDamien Le Moal clear_bit_unlock(flag, &mblk->state); 3903b1a94c8SDamien Le Moal smp_mb__after_atomic(); 3913b1a94c8SDamien Le Moal wake_up_bit(&mblk->state, flag); 3923b1a94c8SDamien Le Moal 3933b1a94c8SDamien Le Moal bio_put(bio); 3943b1a94c8SDamien Le Moal } 3953b1a94c8SDamien Le Moal 3963b1a94c8SDamien Le Moal /* 3973d4e7383SDamien Le Moal * Read an uncached metadata block from disk and add it to the cache. 3983b1a94c8SDamien Le Moal */ 3993d4e7383SDamien Le Moal static struct dmz_mblock *dmz_get_mblock_slow(struct dmz_metadata *zmd, 4003b1a94c8SDamien Le Moal sector_t mblk_no) 4013b1a94c8SDamien Le Moal { 4023d4e7383SDamien Le Moal struct dmz_mblock *mblk, *m; 4033b1a94c8SDamien Le Moal sector_t block = zmd->sb[zmd->mblk_primary].block + mblk_no; 4043b1a94c8SDamien Le Moal struct bio *bio; 4053b1a94c8SDamien Le Moal 40675d66ffbSDmitry Fomichev if (dmz_bdev_is_dying(zmd->dev)) 40775d66ffbSDmitry Fomichev return ERR_PTR(-EIO); 40875d66ffbSDmitry Fomichev 4093d4e7383SDamien Le Moal /* Get a new block and a BIO to read it */ 4103b1a94c8SDamien Le Moal mblk = dmz_alloc_mblock(zmd, mblk_no); 4113b1a94c8SDamien Le Moal if (!mblk) 41275d66ffbSDmitry Fomichev return ERR_PTR(-ENOMEM); 4133b1a94c8SDamien Le Moal 4143b1a94c8SDamien Le Moal bio = bio_alloc(GFP_NOIO, 1); 4153b1a94c8SDamien Le Moal if (!bio) { 4163b1a94c8SDamien Le Moal dmz_free_mblock(zmd, mblk); 41775d66ffbSDmitry Fomichev return ERR_PTR(-ENOMEM); 4183b1a94c8SDamien Le Moal } 4193b1a94c8SDamien Le Moal 4203d4e7383SDamien Le Moal spin_lock(&zmd->mblk_lock); 4213d4e7383SDamien Le Moal 4223d4e7383SDamien Le Moal /* 4233d4e7383SDamien Le Moal * Make sure that another context did not start reading 4243d4e7383SDamien Le Moal * the block already. 4253d4e7383SDamien Le Moal */ 4263d4e7383SDamien Le Moal m = dmz_get_mblock_fast(zmd, mblk_no); 4273d4e7383SDamien Le Moal if (m) { 4283d4e7383SDamien Le Moal spin_unlock(&zmd->mblk_lock); 4293d4e7383SDamien Le Moal dmz_free_mblock(zmd, mblk); 4303d4e7383SDamien Le Moal bio_put(bio); 4313d4e7383SDamien Le Moal return m; 4323d4e7383SDamien Le Moal } 4333d4e7383SDamien Le Moal 4343d4e7383SDamien Le Moal mblk->ref++; 4353d4e7383SDamien Le Moal set_bit(DMZ_META_READING, &mblk->state); 4363d4e7383SDamien Le Moal dmz_insert_mblock(zmd, mblk); 4373d4e7383SDamien Le Moal 4383d4e7383SDamien Le Moal spin_unlock(&zmd->mblk_lock); 4393d4e7383SDamien Le Moal 4403d4e7383SDamien Le Moal /* Submit read BIO */ 4413b1a94c8SDamien Le Moal bio->bi_iter.bi_sector = dmz_blk2sect(block); 44274d46992SChristoph Hellwig bio_set_dev(bio, zmd->dev->bdev); 4433b1a94c8SDamien Le Moal bio->bi_private = mblk; 4443b1a94c8SDamien Le Moal bio->bi_end_io = dmz_mblock_bio_end_io; 4453b1a94c8SDamien Le Moal bio_set_op_attrs(bio, REQ_OP_READ, REQ_META | REQ_PRIO); 4463b1a94c8SDamien Le Moal bio_add_page(bio, mblk->page, DMZ_BLOCK_SIZE, 0); 4473b1a94c8SDamien Le Moal submit_bio(bio); 4483b1a94c8SDamien Le Moal 4493b1a94c8SDamien Le Moal return mblk; 4503b1a94c8SDamien Le Moal } 4513b1a94c8SDamien Le Moal 4523b1a94c8SDamien Le Moal /* 4533b1a94c8SDamien Le Moal * Free metadata blocks. 4543b1a94c8SDamien Le Moal */ 4553b1a94c8SDamien Le Moal static unsigned long dmz_shrink_mblock_cache(struct dmz_metadata *zmd, 4563b1a94c8SDamien Le Moal unsigned long limit) 4573b1a94c8SDamien Le Moal { 4583b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 4593b1a94c8SDamien Le Moal unsigned long count = 0; 4603b1a94c8SDamien Le Moal 4613b1a94c8SDamien Le Moal if (!zmd->max_nr_mblks) 4623b1a94c8SDamien Le Moal return 0; 4633b1a94c8SDamien Le Moal 4643b1a94c8SDamien Le Moal while (!list_empty(&zmd->mblk_lru_list) && 4653b1a94c8SDamien Le Moal atomic_read(&zmd->nr_mblks) > zmd->min_nr_mblks && 4663b1a94c8SDamien Le Moal count < limit) { 4673b1a94c8SDamien Le Moal mblk = list_first_entry(&zmd->mblk_lru_list, 4683b1a94c8SDamien Le Moal struct dmz_mblock, link); 4693b1a94c8SDamien Le Moal list_del_init(&mblk->link); 4703b1a94c8SDamien Le Moal rb_erase(&mblk->node, &zmd->mblk_rbtree); 4713b1a94c8SDamien Le Moal dmz_free_mblock(zmd, mblk); 4723b1a94c8SDamien Le Moal count++; 4733b1a94c8SDamien Le Moal } 4743b1a94c8SDamien Le Moal 4753b1a94c8SDamien Le Moal return count; 4763b1a94c8SDamien Le Moal } 4773b1a94c8SDamien Le Moal 4783b1a94c8SDamien Le Moal /* 4793b1a94c8SDamien Le Moal * For mblock shrinker: get the number of unused metadata blocks in the cache. 4803b1a94c8SDamien Le Moal */ 4813b1a94c8SDamien Le Moal static unsigned long dmz_mblock_shrinker_count(struct shrinker *shrink, 4823b1a94c8SDamien Le Moal struct shrink_control *sc) 4833b1a94c8SDamien Le Moal { 4843b1a94c8SDamien Le Moal struct dmz_metadata *zmd = container_of(shrink, struct dmz_metadata, mblk_shrinker); 4853b1a94c8SDamien Le Moal 4863b1a94c8SDamien Le Moal return atomic_read(&zmd->nr_mblks); 4873b1a94c8SDamien Le Moal } 4883b1a94c8SDamien Le Moal 4893b1a94c8SDamien Le Moal /* 4903b1a94c8SDamien Le Moal * For mblock shrinker: scan unused metadata blocks and shrink the cache. 4913b1a94c8SDamien Le Moal */ 4923b1a94c8SDamien Le Moal static unsigned long dmz_mblock_shrinker_scan(struct shrinker *shrink, 4933b1a94c8SDamien Le Moal struct shrink_control *sc) 4943b1a94c8SDamien Le Moal { 4953b1a94c8SDamien Le Moal struct dmz_metadata *zmd = container_of(shrink, struct dmz_metadata, mblk_shrinker); 4963b1a94c8SDamien Le Moal unsigned long count; 4973b1a94c8SDamien Le Moal 4983b1a94c8SDamien Le Moal spin_lock(&zmd->mblk_lock); 4993b1a94c8SDamien Le Moal count = dmz_shrink_mblock_cache(zmd, sc->nr_to_scan); 5003b1a94c8SDamien Le Moal spin_unlock(&zmd->mblk_lock); 5013b1a94c8SDamien Le Moal 5023b1a94c8SDamien Le Moal return count ? count : SHRINK_STOP; 5033b1a94c8SDamien Le Moal } 5043b1a94c8SDamien Le Moal 5053b1a94c8SDamien Le Moal /* 5063b1a94c8SDamien Le Moal * Release a metadata block. 5073b1a94c8SDamien Le Moal */ 5083b1a94c8SDamien Le Moal static void dmz_release_mblock(struct dmz_metadata *zmd, 5093b1a94c8SDamien Le Moal struct dmz_mblock *mblk) 5103b1a94c8SDamien Le Moal { 5113b1a94c8SDamien Le Moal 5123b1a94c8SDamien Le Moal if (!mblk) 5133b1a94c8SDamien Le Moal return; 5143b1a94c8SDamien Le Moal 5153b1a94c8SDamien Le Moal spin_lock(&zmd->mblk_lock); 5163b1a94c8SDamien Le Moal 51733c2865fSDamien Le Moal mblk->ref--; 51833c2865fSDamien Le Moal if (mblk->ref == 0) { 5193b1a94c8SDamien Le Moal if (test_bit(DMZ_META_ERROR, &mblk->state)) { 5203b1a94c8SDamien Le Moal rb_erase(&mblk->node, &zmd->mblk_rbtree); 5213b1a94c8SDamien Le Moal dmz_free_mblock(zmd, mblk); 5223b1a94c8SDamien Le Moal } else if (!test_bit(DMZ_META_DIRTY, &mblk->state)) { 5233b1a94c8SDamien Le Moal list_add_tail(&mblk->link, &zmd->mblk_lru_list); 5243b1a94c8SDamien Le Moal dmz_shrink_mblock_cache(zmd, 1); 5253b1a94c8SDamien Le Moal } 5263b1a94c8SDamien Le Moal } 5273b1a94c8SDamien Le Moal 5283b1a94c8SDamien Le Moal spin_unlock(&zmd->mblk_lock); 5293b1a94c8SDamien Le Moal } 5303b1a94c8SDamien Le Moal 5313b1a94c8SDamien Le Moal /* 5323b1a94c8SDamien Le Moal * Get a metadata block from the rbtree. If the block 5333b1a94c8SDamien Le Moal * is not present, read it from disk. 5343b1a94c8SDamien Le Moal */ 5353b1a94c8SDamien Le Moal static struct dmz_mblock *dmz_get_mblock(struct dmz_metadata *zmd, 5363b1a94c8SDamien Le Moal sector_t mblk_no) 5373b1a94c8SDamien Le Moal { 5383b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 5393b1a94c8SDamien Le Moal 5403b1a94c8SDamien Le Moal /* Check rbtree */ 5413b1a94c8SDamien Le Moal spin_lock(&zmd->mblk_lock); 5423d4e7383SDamien Le Moal mblk = dmz_get_mblock_fast(zmd, mblk_no); 5433b1a94c8SDamien Le Moal spin_unlock(&zmd->mblk_lock); 5443b1a94c8SDamien Le Moal 5453b1a94c8SDamien Le Moal if (!mblk) { 5463b1a94c8SDamien Le Moal /* Cache miss: read the block from disk */ 5473d4e7383SDamien Le Moal mblk = dmz_get_mblock_slow(zmd, mblk_no); 54875d66ffbSDmitry Fomichev if (IS_ERR(mblk)) 54975d66ffbSDmitry Fomichev return mblk; 5503b1a94c8SDamien Le Moal } 5513b1a94c8SDamien Le Moal 5523b1a94c8SDamien Le Moal /* Wait for on-going read I/O and check for error */ 5533b1a94c8SDamien Le Moal wait_on_bit_io(&mblk->state, DMZ_META_READING, 5543b1a94c8SDamien Le Moal TASK_UNINTERRUPTIBLE); 5553b1a94c8SDamien Le Moal if (test_bit(DMZ_META_ERROR, &mblk->state)) { 5563b1a94c8SDamien Le Moal dmz_release_mblock(zmd, mblk); 5573b1a94c8SDamien Le Moal return ERR_PTR(-EIO); 5583b1a94c8SDamien Le Moal } 5593b1a94c8SDamien Le Moal 5603b1a94c8SDamien Le Moal return mblk; 5613b1a94c8SDamien Le Moal } 5623b1a94c8SDamien Le Moal 5633b1a94c8SDamien Le Moal /* 5643b1a94c8SDamien Le Moal * Mark a metadata block dirty. 5653b1a94c8SDamien Le Moal */ 5663b1a94c8SDamien Le Moal static void dmz_dirty_mblock(struct dmz_metadata *zmd, struct dmz_mblock *mblk) 5673b1a94c8SDamien Le Moal { 5683b1a94c8SDamien Le Moal spin_lock(&zmd->mblk_lock); 5693b1a94c8SDamien Le Moal if (!test_and_set_bit(DMZ_META_DIRTY, &mblk->state)) 5703b1a94c8SDamien Le Moal list_add_tail(&mblk->link, &zmd->mblk_dirty_list); 5713b1a94c8SDamien Le Moal spin_unlock(&zmd->mblk_lock); 5723b1a94c8SDamien Le Moal } 5733b1a94c8SDamien Le Moal 5743b1a94c8SDamien Le Moal /* 5753b1a94c8SDamien Le Moal * Issue a metadata block write BIO. 5763b1a94c8SDamien Le Moal */ 57775d66ffbSDmitry Fomichev static int dmz_write_mblock(struct dmz_metadata *zmd, struct dmz_mblock *mblk, 5783b1a94c8SDamien Le Moal unsigned int set) 5793b1a94c8SDamien Le Moal { 5803b1a94c8SDamien Le Moal sector_t block = zmd->sb[set].block + mblk->no; 5813b1a94c8SDamien Le Moal struct bio *bio; 5823b1a94c8SDamien Le Moal 58375d66ffbSDmitry Fomichev if (dmz_bdev_is_dying(zmd->dev)) 58475d66ffbSDmitry Fomichev return -EIO; 58575d66ffbSDmitry Fomichev 5863b1a94c8SDamien Le Moal bio = bio_alloc(GFP_NOIO, 1); 5873b1a94c8SDamien Le Moal if (!bio) { 5883b1a94c8SDamien Le Moal set_bit(DMZ_META_ERROR, &mblk->state); 58975d66ffbSDmitry Fomichev return -ENOMEM; 5903b1a94c8SDamien Le Moal } 5913b1a94c8SDamien Le Moal 5923b1a94c8SDamien Le Moal set_bit(DMZ_META_WRITING, &mblk->state); 5933b1a94c8SDamien Le Moal 5943b1a94c8SDamien Le Moal bio->bi_iter.bi_sector = dmz_blk2sect(block); 59574d46992SChristoph Hellwig bio_set_dev(bio, zmd->dev->bdev); 5963b1a94c8SDamien Le Moal bio->bi_private = mblk; 5973b1a94c8SDamien Le Moal bio->bi_end_io = dmz_mblock_bio_end_io; 5983b1a94c8SDamien Le Moal bio_set_op_attrs(bio, REQ_OP_WRITE, REQ_META | REQ_PRIO); 5993b1a94c8SDamien Le Moal bio_add_page(bio, mblk->page, DMZ_BLOCK_SIZE, 0); 6003b1a94c8SDamien Le Moal submit_bio(bio); 60175d66ffbSDmitry Fomichev 60275d66ffbSDmitry Fomichev return 0; 6033b1a94c8SDamien Le Moal } 6043b1a94c8SDamien Le Moal 6053b1a94c8SDamien Le Moal /* 6063b1a94c8SDamien Le Moal * Read/write a metadata block. 6073b1a94c8SDamien Le Moal */ 6083b1a94c8SDamien Le Moal static int dmz_rdwr_block(struct dmz_metadata *zmd, int op, sector_t block, 6093b1a94c8SDamien Le Moal struct page *page) 6103b1a94c8SDamien Le Moal { 6113b1a94c8SDamien Le Moal struct bio *bio; 6123b1a94c8SDamien Le Moal int ret; 6133b1a94c8SDamien Le Moal 61475d66ffbSDmitry Fomichev if (dmz_bdev_is_dying(zmd->dev)) 61575d66ffbSDmitry Fomichev return -EIO; 61675d66ffbSDmitry Fomichev 6173b1a94c8SDamien Le Moal bio = bio_alloc(GFP_NOIO, 1); 6183b1a94c8SDamien Le Moal if (!bio) 6193b1a94c8SDamien Le Moal return -ENOMEM; 6203b1a94c8SDamien Le Moal 6213b1a94c8SDamien Le Moal bio->bi_iter.bi_sector = dmz_blk2sect(block); 62274d46992SChristoph Hellwig bio_set_dev(bio, zmd->dev->bdev); 6233b1a94c8SDamien Le Moal bio_set_op_attrs(bio, op, REQ_SYNC | REQ_META | REQ_PRIO); 6243b1a94c8SDamien Le Moal bio_add_page(bio, page, DMZ_BLOCK_SIZE, 0); 6253b1a94c8SDamien Le Moal ret = submit_bio_wait(bio); 6263b1a94c8SDamien Le Moal bio_put(bio); 6273b1a94c8SDamien Le Moal 6283b1a94c8SDamien Le Moal return ret; 6293b1a94c8SDamien Le Moal } 6303b1a94c8SDamien Le Moal 6313b1a94c8SDamien Le Moal /* 6323b1a94c8SDamien Le Moal * Write super block of the specified metadata set. 6333b1a94c8SDamien Le Moal */ 6343b1a94c8SDamien Le Moal static int dmz_write_sb(struct dmz_metadata *zmd, unsigned int set) 6353b1a94c8SDamien Le Moal { 6363b1a94c8SDamien Le Moal sector_t block = zmd->sb[set].block; 6373b1a94c8SDamien Le Moal struct dmz_mblock *mblk = zmd->sb[set].mblk; 6383b1a94c8SDamien Le Moal struct dmz_super *sb = zmd->sb[set].sb; 6393b1a94c8SDamien Le Moal u64 sb_gen = zmd->sb_gen + 1; 6403b1a94c8SDamien Le Moal int ret; 6413b1a94c8SDamien Le Moal 6423b1a94c8SDamien Le Moal sb->magic = cpu_to_le32(DMZ_MAGIC); 6433b1a94c8SDamien Le Moal sb->version = cpu_to_le32(DMZ_META_VER); 6443b1a94c8SDamien Le Moal 6453b1a94c8SDamien Le Moal sb->gen = cpu_to_le64(sb_gen); 6463b1a94c8SDamien Le Moal 6473b1a94c8SDamien Le Moal sb->sb_block = cpu_to_le64(block); 6483b1a94c8SDamien Le Moal sb->nr_meta_blocks = cpu_to_le32(zmd->nr_meta_blocks); 6493b1a94c8SDamien Le Moal sb->nr_reserved_seq = cpu_to_le32(zmd->nr_reserved_seq); 6503b1a94c8SDamien Le Moal sb->nr_chunks = cpu_to_le32(zmd->nr_chunks); 6513b1a94c8SDamien Le Moal 6523b1a94c8SDamien Le Moal sb->nr_map_blocks = cpu_to_le32(zmd->nr_map_blocks); 6533b1a94c8SDamien Le Moal sb->nr_bitmap_blocks = cpu_to_le32(zmd->nr_bitmap_blocks); 6543b1a94c8SDamien Le Moal 6553b1a94c8SDamien Le Moal sb->crc = 0; 6563b1a94c8SDamien Le Moal sb->crc = cpu_to_le32(crc32_le(sb_gen, (unsigned char *)sb, DMZ_BLOCK_SIZE)); 6573b1a94c8SDamien Le Moal 6583b1a94c8SDamien Le Moal ret = dmz_rdwr_block(zmd, REQ_OP_WRITE, block, mblk->page); 6593b1a94c8SDamien Le Moal if (ret == 0) 6604218a955SDamien Le Moal ret = blkdev_issue_flush(zmd->dev->bdev, GFP_NOIO, NULL); 6613b1a94c8SDamien Le Moal 6623b1a94c8SDamien Le Moal return ret; 6633b1a94c8SDamien Le Moal } 6643b1a94c8SDamien Le Moal 6653b1a94c8SDamien Le Moal /* 6663b1a94c8SDamien Le Moal * Write dirty metadata blocks to the specified set. 6673b1a94c8SDamien Le Moal */ 6683b1a94c8SDamien Le Moal static int dmz_write_dirty_mblocks(struct dmz_metadata *zmd, 6693b1a94c8SDamien Le Moal struct list_head *write_list, 6703b1a94c8SDamien Le Moal unsigned int set) 6713b1a94c8SDamien Le Moal { 6723b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 6733b1a94c8SDamien Le Moal struct blk_plug plug; 67475d66ffbSDmitry Fomichev int ret = 0, nr_mblks_submitted = 0; 6753b1a94c8SDamien Le Moal 6763b1a94c8SDamien Le Moal /* Issue writes */ 6773b1a94c8SDamien Le Moal blk_start_plug(&plug); 67875d66ffbSDmitry Fomichev list_for_each_entry(mblk, write_list, link) { 67975d66ffbSDmitry Fomichev ret = dmz_write_mblock(zmd, mblk, set); 68075d66ffbSDmitry Fomichev if (ret) 68175d66ffbSDmitry Fomichev break; 68275d66ffbSDmitry Fomichev nr_mblks_submitted++; 68375d66ffbSDmitry Fomichev } 6843b1a94c8SDamien Le Moal blk_finish_plug(&plug); 6853b1a94c8SDamien Le Moal 6863b1a94c8SDamien Le Moal /* Wait for completion */ 6873b1a94c8SDamien Le Moal list_for_each_entry(mblk, write_list, link) { 68875d66ffbSDmitry Fomichev if (!nr_mblks_submitted) 68975d66ffbSDmitry Fomichev break; 6903b1a94c8SDamien Le Moal wait_on_bit_io(&mblk->state, DMZ_META_WRITING, 6913b1a94c8SDamien Le Moal TASK_UNINTERRUPTIBLE); 6923b1a94c8SDamien Le Moal if (test_bit(DMZ_META_ERROR, &mblk->state)) { 6933b1a94c8SDamien Le Moal clear_bit(DMZ_META_ERROR, &mblk->state); 6943b1a94c8SDamien Le Moal ret = -EIO; 6953b1a94c8SDamien Le Moal } 69675d66ffbSDmitry Fomichev nr_mblks_submitted--; 6973b1a94c8SDamien Le Moal } 6983b1a94c8SDamien Le Moal 6993b1a94c8SDamien Le Moal /* Flush drive cache (this will also sync data) */ 7003b1a94c8SDamien Le Moal if (ret == 0) 7014218a955SDamien Le Moal ret = blkdev_issue_flush(zmd->dev->bdev, GFP_NOIO, NULL); 7023b1a94c8SDamien Le Moal 7033b1a94c8SDamien Le Moal return ret; 7043b1a94c8SDamien Le Moal } 7053b1a94c8SDamien Le Moal 7063b1a94c8SDamien Le Moal /* 7073b1a94c8SDamien Le Moal * Log dirty metadata blocks. 7083b1a94c8SDamien Le Moal */ 7093b1a94c8SDamien Le Moal static int dmz_log_dirty_mblocks(struct dmz_metadata *zmd, 7103b1a94c8SDamien Le Moal struct list_head *write_list) 7113b1a94c8SDamien Le Moal { 7123b1a94c8SDamien Le Moal unsigned int log_set = zmd->mblk_primary ^ 0x1; 7133b1a94c8SDamien Le Moal int ret; 7143b1a94c8SDamien Le Moal 7153b1a94c8SDamien Le Moal /* Write dirty blocks to the log */ 7163b1a94c8SDamien Le Moal ret = dmz_write_dirty_mblocks(zmd, write_list, log_set); 7173b1a94c8SDamien Le Moal if (ret) 7183b1a94c8SDamien Le Moal return ret; 7193b1a94c8SDamien Le Moal 7203b1a94c8SDamien Le Moal /* 7213b1a94c8SDamien Le Moal * No error so far: now validate the log by updating the 7223b1a94c8SDamien Le Moal * log index super block generation. 7233b1a94c8SDamien Le Moal */ 7243b1a94c8SDamien Le Moal ret = dmz_write_sb(zmd, log_set); 7253b1a94c8SDamien Le Moal if (ret) 7263b1a94c8SDamien Le Moal return ret; 7273b1a94c8SDamien Le Moal 7283b1a94c8SDamien Le Moal return 0; 7293b1a94c8SDamien Le Moal } 7303b1a94c8SDamien Le Moal 7313b1a94c8SDamien Le Moal /* 7323b1a94c8SDamien Le Moal * Flush dirty metadata blocks. 7333b1a94c8SDamien Le Moal */ 7343b1a94c8SDamien Le Moal int dmz_flush_metadata(struct dmz_metadata *zmd) 7353b1a94c8SDamien Le Moal { 7363b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 7373b1a94c8SDamien Le Moal struct list_head write_list; 7383b1a94c8SDamien Le Moal int ret; 7393b1a94c8SDamien Le Moal 7403b1a94c8SDamien Le Moal if (WARN_ON(!zmd)) 7413b1a94c8SDamien Le Moal return 0; 7423b1a94c8SDamien Le Moal 7433b1a94c8SDamien Le Moal INIT_LIST_HEAD(&write_list); 7443b1a94c8SDamien Le Moal 7453b1a94c8SDamien Le Moal /* 7463b1a94c8SDamien Le Moal * Make sure that metadata blocks are stable before logging: take 7473b1a94c8SDamien Le Moal * the write lock on the metadata semaphore to prevent target BIOs 7483b1a94c8SDamien Le Moal * from modifying metadata. 7493b1a94c8SDamien Le Moal */ 7503b1a94c8SDamien Le Moal down_write(&zmd->mblk_sem); 7513b1a94c8SDamien Le Moal 7523b1a94c8SDamien Le Moal /* 7533b1a94c8SDamien Le Moal * This is called from the target flush work and reclaim work. 7543b1a94c8SDamien Le Moal * Concurrent execution is not allowed. 7553b1a94c8SDamien Le Moal */ 7563b1a94c8SDamien Le Moal dmz_lock_flush(zmd); 7573b1a94c8SDamien Le Moal 75875d66ffbSDmitry Fomichev if (dmz_bdev_is_dying(zmd->dev)) { 75975d66ffbSDmitry Fomichev ret = -EIO; 76075d66ffbSDmitry Fomichev goto out; 76175d66ffbSDmitry Fomichev } 76275d66ffbSDmitry Fomichev 7633b1a94c8SDamien Le Moal /* Get dirty blocks */ 7643b1a94c8SDamien Le Moal spin_lock(&zmd->mblk_lock); 7653b1a94c8SDamien Le Moal list_splice_init(&zmd->mblk_dirty_list, &write_list); 7663b1a94c8SDamien Le Moal spin_unlock(&zmd->mblk_lock); 7673b1a94c8SDamien Le Moal 7683b1a94c8SDamien Le Moal /* If there are no dirty metadata blocks, just flush the device cache */ 7693b1a94c8SDamien Le Moal if (list_empty(&write_list)) { 7704218a955SDamien Le Moal ret = blkdev_issue_flush(zmd->dev->bdev, GFP_NOIO, NULL); 7713b1a94c8SDamien Le Moal goto out; 7723b1a94c8SDamien Le Moal } 7733b1a94c8SDamien Le Moal 7743b1a94c8SDamien Le Moal /* 7753b1a94c8SDamien Le Moal * The primary metadata set is still clean. Keep it this way until 7763b1a94c8SDamien Le Moal * all updates are successful in the secondary set. That is, use 7773b1a94c8SDamien Le Moal * the secondary set as a log. 7783b1a94c8SDamien Le Moal */ 7793b1a94c8SDamien Le Moal ret = dmz_log_dirty_mblocks(zmd, &write_list); 7803b1a94c8SDamien Le Moal if (ret) 7813b1a94c8SDamien Le Moal goto out; 7823b1a94c8SDamien Le Moal 7833b1a94c8SDamien Le Moal /* 7843b1a94c8SDamien Le Moal * The log is on disk. It is now safe to update in place 7853b1a94c8SDamien Le Moal * in the primary metadata set. 7863b1a94c8SDamien Le Moal */ 7873b1a94c8SDamien Le Moal ret = dmz_write_dirty_mblocks(zmd, &write_list, zmd->mblk_primary); 7883b1a94c8SDamien Le Moal if (ret) 7893b1a94c8SDamien Le Moal goto out; 7903b1a94c8SDamien Le Moal 7913b1a94c8SDamien Le Moal ret = dmz_write_sb(zmd, zmd->mblk_primary); 7923b1a94c8SDamien Le Moal if (ret) 7933b1a94c8SDamien Le Moal goto out; 7943b1a94c8SDamien Le Moal 7953b1a94c8SDamien Le Moal while (!list_empty(&write_list)) { 7963b1a94c8SDamien Le Moal mblk = list_first_entry(&write_list, struct dmz_mblock, link); 7973b1a94c8SDamien Le Moal list_del_init(&mblk->link); 7983b1a94c8SDamien Le Moal 7993b1a94c8SDamien Le Moal spin_lock(&zmd->mblk_lock); 8003b1a94c8SDamien Le Moal clear_bit(DMZ_META_DIRTY, &mblk->state); 80133c2865fSDamien Le Moal if (mblk->ref == 0) 8023b1a94c8SDamien Le Moal list_add_tail(&mblk->link, &zmd->mblk_lru_list); 8033b1a94c8SDamien Le Moal spin_unlock(&zmd->mblk_lock); 8043b1a94c8SDamien Le Moal } 8053b1a94c8SDamien Le Moal 8063b1a94c8SDamien Le Moal zmd->sb_gen++; 8073b1a94c8SDamien Le Moal out: 8083b1a94c8SDamien Le Moal if (ret && !list_empty(&write_list)) { 8093b1a94c8SDamien Le Moal spin_lock(&zmd->mblk_lock); 8103b1a94c8SDamien Le Moal list_splice(&write_list, &zmd->mblk_dirty_list); 8113b1a94c8SDamien Le Moal spin_unlock(&zmd->mblk_lock); 8123b1a94c8SDamien Le Moal } 8133b1a94c8SDamien Le Moal 8143b1a94c8SDamien Le Moal dmz_unlock_flush(zmd); 8153b1a94c8SDamien Le Moal up_write(&zmd->mblk_sem); 8163b1a94c8SDamien Le Moal 8173b1a94c8SDamien Le Moal return ret; 8183b1a94c8SDamien Le Moal } 8193b1a94c8SDamien Le Moal 8203b1a94c8SDamien Le Moal /* 8213b1a94c8SDamien Le Moal * Check super block. 8223b1a94c8SDamien Le Moal */ 8233b1a94c8SDamien Le Moal static int dmz_check_sb(struct dmz_metadata *zmd, struct dmz_super *sb) 8243b1a94c8SDamien Le Moal { 8253b1a94c8SDamien Le Moal unsigned int nr_meta_zones, nr_data_zones; 8263b1a94c8SDamien Le Moal struct dmz_dev *dev = zmd->dev; 8273b1a94c8SDamien Le Moal u32 crc, stored_crc; 8283b1a94c8SDamien Le Moal u64 gen; 8293b1a94c8SDamien Le Moal 8303b1a94c8SDamien Le Moal gen = le64_to_cpu(sb->gen); 8313b1a94c8SDamien Le Moal stored_crc = le32_to_cpu(sb->crc); 8323b1a94c8SDamien Le Moal sb->crc = 0; 8333b1a94c8SDamien Le Moal crc = crc32_le(gen, (unsigned char *)sb, DMZ_BLOCK_SIZE); 8343b1a94c8SDamien Le Moal if (crc != stored_crc) { 8353b1a94c8SDamien Le Moal dmz_dev_err(dev, "Invalid checksum (needed 0x%08x, got 0x%08x)", 8363b1a94c8SDamien Le Moal crc, stored_crc); 8373b1a94c8SDamien Le Moal return -ENXIO; 8383b1a94c8SDamien Le Moal } 8393b1a94c8SDamien Le Moal 8403b1a94c8SDamien Le Moal if (le32_to_cpu(sb->magic) != DMZ_MAGIC) { 8413b1a94c8SDamien Le Moal dmz_dev_err(dev, "Invalid meta magic (needed 0x%08x, got 0x%08x)", 8423b1a94c8SDamien Le Moal DMZ_MAGIC, le32_to_cpu(sb->magic)); 8433b1a94c8SDamien Le Moal return -ENXIO; 8443b1a94c8SDamien Le Moal } 8453b1a94c8SDamien Le Moal 8463b1a94c8SDamien Le Moal if (le32_to_cpu(sb->version) != DMZ_META_VER) { 8473b1a94c8SDamien Le Moal dmz_dev_err(dev, "Invalid meta version (needed %d, got %d)", 8483b1a94c8SDamien Le Moal DMZ_META_VER, le32_to_cpu(sb->version)); 8493b1a94c8SDamien Le Moal return -ENXIO; 8503b1a94c8SDamien Le Moal } 8513b1a94c8SDamien Le Moal 8523b1a94c8SDamien Le Moal nr_meta_zones = (le32_to_cpu(sb->nr_meta_blocks) + dev->zone_nr_blocks - 1) 8533b1a94c8SDamien Le Moal >> dev->zone_nr_blocks_shift; 8543b1a94c8SDamien Le Moal if (!nr_meta_zones || 8553b1a94c8SDamien Le Moal nr_meta_zones >= zmd->nr_rnd_zones) { 8563b1a94c8SDamien Le Moal dmz_dev_err(dev, "Invalid number of metadata blocks"); 8573b1a94c8SDamien Le Moal return -ENXIO; 8583b1a94c8SDamien Le Moal } 8593b1a94c8SDamien Le Moal 8603b1a94c8SDamien Le Moal if (!le32_to_cpu(sb->nr_reserved_seq) || 8613b1a94c8SDamien Le Moal le32_to_cpu(sb->nr_reserved_seq) >= (zmd->nr_useable_zones - nr_meta_zones)) { 8623b1a94c8SDamien Le Moal dmz_dev_err(dev, "Invalid number of reserved sequential zones"); 8633b1a94c8SDamien Le Moal return -ENXIO; 8643b1a94c8SDamien Le Moal } 8653b1a94c8SDamien Le Moal 8663b1a94c8SDamien Le Moal nr_data_zones = zmd->nr_useable_zones - 8673b1a94c8SDamien Le Moal (nr_meta_zones * 2 + le32_to_cpu(sb->nr_reserved_seq)); 8683b1a94c8SDamien Le Moal if (le32_to_cpu(sb->nr_chunks) > nr_data_zones) { 8693b1a94c8SDamien Le Moal dmz_dev_err(dev, "Invalid number of chunks %u / %u", 8703b1a94c8SDamien Le Moal le32_to_cpu(sb->nr_chunks), nr_data_zones); 8713b1a94c8SDamien Le Moal return -ENXIO; 8723b1a94c8SDamien Le Moal } 8733b1a94c8SDamien Le Moal 8743b1a94c8SDamien Le Moal /* OK */ 8753b1a94c8SDamien Le Moal zmd->nr_meta_blocks = le32_to_cpu(sb->nr_meta_blocks); 8763b1a94c8SDamien Le Moal zmd->nr_reserved_seq = le32_to_cpu(sb->nr_reserved_seq); 8773b1a94c8SDamien Le Moal zmd->nr_chunks = le32_to_cpu(sb->nr_chunks); 8783b1a94c8SDamien Le Moal zmd->nr_map_blocks = le32_to_cpu(sb->nr_map_blocks); 8793b1a94c8SDamien Le Moal zmd->nr_bitmap_blocks = le32_to_cpu(sb->nr_bitmap_blocks); 8803b1a94c8SDamien Le Moal zmd->nr_meta_zones = nr_meta_zones; 8813b1a94c8SDamien Le Moal zmd->nr_data_zones = nr_data_zones; 8823b1a94c8SDamien Le Moal 8833b1a94c8SDamien Le Moal return 0; 8843b1a94c8SDamien Le Moal } 8853b1a94c8SDamien Le Moal 8863b1a94c8SDamien Le Moal /* 8873b1a94c8SDamien Le Moal * Read the first or second super block from disk. 8883b1a94c8SDamien Le Moal */ 8893b1a94c8SDamien Le Moal static int dmz_read_sb(struct dmz_metadata *zmd, unsigned int set) 8903b1a94c8SDamien Le Moal { 8913b1a94c8SDamien Le Moal return dmz_rdwr_block(zmd, REQ_OP_READ, zmd->sb[set].block, 8923b1a94c8SDamien Le Moal zmd->sb[set].mblk->page); 8933b1a94c8SDamien Le Moal } 8943b1a94c8SDamien Le Moal 8953b1a94c8SDamien Le Moal /* 8963b1a94c8SDamien Le Moal * Determine the position of the secondary super blocks on disk. 8973b1a94c8SDamien Le Moal * This is used only if a corruption of the primary super block 8983b1a94c8SDamien Le Moal * is detected. 8993b1a94c8SDamien Le Moal */ 9003b1a94c8SDamien Le Moal static int dmz_lookup_secondary_sb(struct dmz_metadata *zmd) 9013b1a94c8SDamien Le Moal { 9023b1a94c8SDamien Le Moal unsigned int zone_nr_blocks = zmd->dev->zone_nr_blocks; 9033b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 9043b1a94c8SDamien Le Moal int i; 9053b1a94c8SDamien Le Moal 9063b1a94c8SDamien Le Moal /* Allocate a block */ 9073b1a94c8SDamien Le Moal mblk = dmz_alloc_mblock(zmd, 0); 9083b1a94c8SDamien Le Moal if (!mblk) 9093b1a94c8SDamien Le Moal return -ENOMEM; 9103b1a94c8SDamien Le Moal 9113b1a94c8SDamien Le Moal zmd->sb[1].mblk = mblk; 9123b1a94c8SDamien Le Moal zmd->sb[1].sb = mblk->data; 9133b1a94c8SDamien Le Moal 9143b1a94c8SDamien Le Moal /* Bad first super block: search for the second one */ 9153b1a94c8SDamien Le Moal zmd->sb[1].block = zmd->sb[0].block + zone_nr_blocks; 9163b1a94c8SDamien Le Moal for (i = 0; i < zmd->nr_rnd_zones - 1; i++) { 9173b1a94c8SDamien Le Moal if (dmz_read_sb(zmd, 1) != 0) 9183b1a94c8SDamien Le Moal break; 9193b1a94c8SDamien Le Moal if (le32_to_cpu(zmd->sb[1].sb->magic) == DMZ_MAGIC) 9203b1a94c8SDamien Le Moal return 0; 9213b1a94c8SDamien Le Moal zmd->sb[1].block += zone_nr_blocks; 9223b1a94c8SDamien Le Moal } 9233b1a94c8SDamien Le Moal 9243b1a94c8SDamien Le Moal dmz_free_mblock(zmd, mblk); 9253b1a94c8SDamien Le Moal zmd->sb[1].mblk = NULL; 9263b1a94c8SDamien Le Moal 9273b1a94c8SDamien Le Moal return -EIO; 9283b1a94c8SDamien Le Moal } 9293b1a94c8SDamien Le Moal 9303b1a94c8SDamien Le Moal /* 9313b1a94c8SDamien Le Moal * Read the first or second super block from disk. 9323b1a94c8SDamien Le Moal */ 9333b1a94c8SDamien Le Moal static int dmz_get_sb(struct dmz_metadata *zmd, unsigned int set) 9343b1a94c8SDamien Le Moal { 9353b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 9363b1a94c8SDamien Le Moal int ret; 9373b1a94c8SDamien Le Moal 9383b1a94c8SDamien Le Moal /* Allocate a block */ 9393b1a94c8SDamien Le Moal mblk = dmz_alloc_mblock(zmd, 0); 9403b1a94c8SDamien Le Moal if (!mblk) 9413b1a94c8SDamien Le Moal return -ENOMEM; 9423b1a94c8SDamien Le Moal 9433b1a94c8SDamien Le Moal zmd->sb[set].mblk = mblk; 9443b1a94c8SDamien Le Moal zmd->sb[set].sb = mblk->data; 9453b1a94c8SDamien Le Moal 9463b1a94c8SDamien Le Moal /* Read super block */ 9473b1a94c8SDamien Le Moal ret = dmz_read_sb(zmd, set); 9483b1a94c8SDamien Le Moal if (ret) { 9493b1a94c8SDamien Le Moal dmz_free_mblock(zmd, mblk); 9503b1a94c8SDamien Le Moal zmd->sb[set].mblk = NULL; 9513b1a94c8SDamien Le Moal return ret; 9523b1a94c8SDamien Le Moal } 9533b1a94c8SDamien Le Moal 9543b1a94c8SDamien Le Moal return 0; 9553b1a94c8SDamien Le Moal } 9563b1a94c8SDamien Le Moal 9573b1a94c8SDamien Le Moal /* 9583b1a94c8SDamien Le Moal * Recover a metadata set. 9593b1a94c8SDamien Le Moal */ 9603b1a94c8SDamien Le Moal static int dmz_recover_mblocks(struct dmz_metadata *zmd, unsigned int dst_set) 9613b1a94c8SDamien Le Moal { 9623b1a94c8SDamien Le Moal unsigned int src_set = dst_set ^ 0x1; 9633b1a94c8SDamien Le Moal struct page *page; 9643b1a94c8SDamien Le Moal int i, ret; 9653b1a94c8SDamien Le Moal 9663b1a94c8SDamien Le Moal dmz_dev_warn(zmd->dev, "Metadata set %u invalid: recovering", dst_set); 9673b1a94c8SDamien Le Moal 9683b1a94c8SDamien Le Moal if (dst_set == 0) 9693b1a94c8SDamien Le Moal zmd->sb[0].block = dmz_start_block(zmd, zmd->sb_zone); 9703b1a94c8SDamien Le Moal else { 9713b1a94c8SDamien Le Moal zmd->sb[1].block = zmd->sb[0].block + 9723b1a94c8SDamien Le Moal (zmd->nr_meta_zones << zmd->dev->zone_nr_blocks_shift); 9733b1a94c8SDamien Le Moal } 9743b1a94c8SDamien Le Moal 9754218a955SDamien Le Moal page = alloc_page(GFP_NOIO); 9763b1a94c8SDamien Le Moal if (!page) 9773b1a94c8SDamien Le Moal return -ENOMEM; 9783b1a94c8SDamien Le Moal 9793b1a94c8SDamien Le Moal /* Copy metadata blocks */ 9803b1a94c8SDamien Le Moal for (i = 1; i < zmd->nr_meta_blocks; i++) { 9813b1a94c8SDamien Le Moal ret = dmz_rdwr_block(zmd, REQ_OP_READ, 9823b1a94c8SDamien Le Moal zmd->sb[src_set].block + i, page); 9833b1a94c8SDamien Le Moal if (ret) 9843b1a94c8SDamien Le Moal goto out; 9853b1a94c8SDamien Le Moal ret = dmz_rdwr_block(zmd, REQ_OP_WRITE, 9863b1a94c8SDamien Le Moal zmd->sb[dst_set].block + i, page); 9873b1a94c8SDamien Le Moal if (ret) 9883b1a94c8SDamien Le Moal goto out; 9893b1a94c8SDamien Le Moal } 9903b1a94c8SDamien Le Moal 9913b1a94c8SDamien Le Moal /* Finalize with the super block */ 9923b1a94c8SDamien Le Moal if (!zmd->sb[dst_set].mblk) { 9933b1a94c8SDamien Le Moal zmd->sb[dst_set].mblk = dmz_alloc_mblock(zmd, 0); 9943b1a94c8SDamien Le Moal if (!zmd->sb[dst_set].mblk) { 9953b1a94c8SDamien Le Moal ret = -ENOMEM; 9963b1a94c8SDamien Le Moal goto out; 9973b1a94c8SDamien Le Moal } 9983b1a94c8SDamien Le Moal zmd->sb[dst_set].sb = zmd->sb[dst_set].mblk->data; 9993b1a94c8SDamien Le Moal } 10003b1a94c8SDamien Le Moal 10013b1a94c8SDamien Le Moal ret = dmz_write_sb(zmd, dst_set); 10023b1a94c8SDamien Le Moal out: 10033b1a94c8SDamien Le Moal __free_pages(page, 0); 10043b1a94c8SDamien Le Moal 10053b1a94c8SDamien Le Moal return ret; 10063b1a94c8SDamien Le Moal } 10073b1a94c8SDamien Le Moal 10083b1a94c8SDamien Le Moal /* 10093b1a94c8SDamien Le Moal * Get super block from disk. 10103b1a94c8SDamien Le Moal */ 10113b1a94c8SDamien Le Moal static int dmz_load_sb(struct dmz_metadata *zmd) 10123b1a94c8SDamien Le Moal { 10133b1a94c8SDamien Le Moal bool sb_good[2] = {false, false}; 10143b1a94c8SDamien Le Moal u64 sb_gen[2] = {0, 0}; 10153b1a94c8SDamien Le Moal int ret; 10163b1a94c8SDamien Le Moal 10173b1a94c8SDamien Le Moal /* Read and check the primary super block */ 10183b1a94c8SDamien Le Moal zmd->sb[0].block = dmz_start_block(zmd, zmd->sb_zone); 10193b1a94c8SDamien Le Moal ret = dmz_get_sb(zmd, 0); 10203b1a94c8SDamien Le Moal if (ret) { 10213b1a94c8SDamien Le Moal dmz_dev_err(zmd->dev, "Read primary super block failed"); 10223b1a94c8SDamien Le Moal return ret; 10233b1a94c8SDamien Le Moal } 10243b1a94c8SDamien Le Moal 10253b1a94c8SDamien Le Moal ret = dmz_check_sb(zmd, zmd->sb[0].sb); 10263b1a94c8SDamien Le Moal 10273b1a94c8SDamien Le Moal /* Read and check secondary super block */ 10283b1a94c8SDamien Le Moal if (ret == 0) { 10293b1a94c8SDamien Le Moal sb_good[0] = true; 10303b1a94c8SDamien Le Moal zmd->sb[1].block = zmd->sb[0].block + 10313b1a94c8SDamien Le Moal (zmd->nr_meta_zones << zmd->dev->zone_nr_blocks_shift); 10323b1a94c8SDamien Le Moal ret = dmz_get_sb(zmd, 1); 10333b1a94c8SDamien Le Moal } else 10343b1a94c8SDamien Le Moal ret = dmz_lookup_secondary_sb(zmd); 10353b1a94c8SDamien Le Moal 10363b1a94c8SDamien Le Moal if (ret) { 10373b1a94c8SDamien Le Moal dmz_dev_err(zmd->dev, "Read secondary super block failed"); 10383b1a94c8SDamien Le Moal return ret; 10393b1a94c8SDamien Le Moal } 10403b1a94c8SDamien Le Moal 10413b1a94c8SDamien Le Moal ret = dmz_check_sb(zmd, zmd->sb[1].sb); 10423b1a94c8SDamien Le Moal if (ret == 0) 10433b1a94c8SDamien Le Moal sb_good[1] = true; 10443b1a94c8SDamien Le Moal 10453b1a94c8SDamien Le Moal /* Use highest generation sb first */ 10463b1a94c8SDamien Le Moal if (!sb_good[0] && !sb_good[1]) { 10473b1a94c8SDamien Le Moal dmz_dev_err(zmd->dev, "No valid super block found"); 10483b1a94c8SDamien Le Moal return -EIO; 10493b1a94c8SDamien Le Moal } 10503b1a94c8SDamien Le Moal 10513b1a94c8SDamien Le Moal if (sb_good[0]) 10523b1a94c8SDamien Le Moal sb_gen[0] = le64_to_cpu(zmd->sb[0].sb->gen); 10533b1a94c8SDamien Le Moal else 10543b1a94c8SDamien Le Moal ret = dmz_recover_mblocks(zmd, 0); 10553b1a94c8SDamien Le Moal 10563b1a94c8SDamien Le Moal if (sb_good[1]) 10573b1a94c8SDamien Le Moal sb_gen[1] = le64_to_cpu(zmd->sb[1].sb->gen); 10583b1a94c8SDamien Le Moal else 10593b1a94c8SDamien Le Moal ret = dmz_recover_mblocks(zmd, 1); 10603b1a94c8SDamien Le Moal 10613b1a94c8SDamien Le Moal if (ret) { 10623b1a94c8SDamien Le Moal dmz_dev_err(zmd->dev, "Recovery failed"); 10633b1a94c8SDamien Le Moal return -EIO; 10643b1a94c8SDamien Le Moal } 10653b1a94c8SDamien Le Moal 10663b1a94c8SDamien Le Moal if (sb_gen[0] >= sb_gen[1]) { 10673b1a94c8SDamien Le Moal zmd->sb_gen = sb_gen[0]; 10683b1a94c8SDamien Le Moal zmd->mblk_primary = 0; 10693b1a94c8SDamien Le Moal } else { 10703b1a94c8SDamien Le Moal zmd->sb_gen = sb_gen[1]; 10713b1a94c8SDamien Le Moal zmd->mblk_primary = 1; 10723b1a94c8SDamien Le Moal } 10733b1a94c8SDamien Le Moal 10743b1a94c8SDamien Le Moal dmz_dev_debug(zmd->dev, "Using super block %u (gen %llu)", 10753b1a94c8SDamien Le Moal zmd->mblk_primary, zmd->sb_gen); 10763b1a94c8SDamien Le Moal 10773b1a94c8SDamien Le Moal return 0; 10783b1a94c8SDamien Le Moal } 10793b1a94c8SDamien Le Moal 10803b1a94c8SDamien Le Moal /* 10813b1a94c8SDamien Le Moal * Initialize a zone descriptor. 10823b1a94c8SDamien Le Moal */ 10833b1a94c8SDamien Le Moal static int dmz_init_zone(struct dmz_metadata *zmd, struct dm_zone *zone, 10843b1a94c8SDamien Le Moal struct blk_zone *blkz) 10853b1a94c8SDamien Le Moal { 10863b1a94c8SDamien Le Moal struct dmz_dev *dev = zmd->dev; 10873b1a94c8SDamien Le Moal 10883b1a94c8SDamien Le Moal /* Ignore the eventual last runt (smaller) zone */ 10893b1a94c8SDamien Le Moal if (blkz->len != dev->zone_nr_sectors) { 10903b1a94c8SDamien Le Moal if (blkz->start + blkz->len == dev->capacity) 10913b1a94c8SDamien Le Moal return 0; 10923b1a94c8SDamien Le Moal return -ENXIO; 10933b1a94c8SDamien Le Moal } 10943b1a94c8SDamien Le Moal 10953b1a94c8SDamien Le Moal INIT_LIST_HEAD(&zone->link); 10963b1a94c8SDamien Le Moal atomic_set(&zone->refcount, 0); 10973b1a94c8SDamien Le Moal zone->chunk = DMZ_MAP_UNMAPPED; 10983b1a94c8SDamien Le Moal 10993b1a94c8SDamien Le Moal if (blkz->type == BLK_ZONE_TYPE_CONVENTIONAL) { 11003b1a94c8SDamien Le Moal set_bit(DMZ_RND, &zone->flags); 11013b1a94c8SDamien Le Moal zmd->nr_rnd_zones++; 11023b1a94c8SDamien Le Moal } else if (blkz->type == BLK_ZONE_TYPE_SEQWRITE_REQ || 11033b1a94c8SDamien Le Moal blkz->type == BLK_ZONE_TYPE_SEQWRITE_PREF) { 11043b1a94c8SDamien Le Moal set_bit(DMZ_SEQ, &zone->flags); 11053b1a94c8SDamien Le Moal } else 11063b1a94c8SDamien Le Moal return -ENXIO; 11073b1a94c8SDamien Le Moal 11083b1a94c8SDamien Le Moal if (blkz->cond == BLK_ZONE_COND_OFFLINE) 11093b1a94c8SDamien Le Moal set_bit(DMZ_OFFLINE, &zone->flags); 11103b1a94c8SDamien Le Moal else if (blkz->cond == BLK_ZONE_COND_READONLY) 11113b1a94c8SDamien Le Moal set_bit(DMZ_READ_ONLY, &zone->flags); 11123b1a94c8SDamien Le Moal 11133b1a94c8SDamien Le Moal if (dmz_is_rnd(zone)) 11143b1a94c8SDamien Le Moal zone->wp_block = 0; 11153b1a94c8SDamien Le Moal else 11163b1a94c8SDamien Le Moal zone->wp_block = dmz_sect2blk(blkz->wp - blkz->start); 11173b1a94c8SDamien Le Moal 11183b1a94c8SDamien Le Moal if (!dmz_is_offline(zone) && !dmz_is_readonly(zone)) { 11193b1a94c8SDamien Le Moal zmd->nr_useable_zones++; 11203b1a94c8SDamien Le Moal if (dmz_is_rnd(zone)) { 11213b1a94c8SDamien Le Moal zmd->nr_rnd_zones++; 11223b1a94c8SDamien Le Moal if (!zmd->sb_zone) { 11233b1a94c8SDamien Le Moal /* Super block zone */ 11243b1a94c8SDamien Le Moal zmd->sb_zone = zone; 11253b1a94c8SDamien Le Moal } 11263b1a94c8SDamien Le Moal } 11273b1a94c8SDamien Le Moal } 11283b1a94c8SDamien Le Moal 11293b1a94c8SDamien Le Moal return 0; 11303b1a94c8SDamien Le Moal } 11313b1a94c8SDamien Le Moal 11323b1a94c8SDamien Le Moal /* 11333b1a94c8SDamien Le Moal * Free zones descriptors. 11343b1a94c8SDamien Le Moal */ 11353b1a94c8SDamien Le Moal static void dmz_drop_zones(struct dmz_metadata *zmd) 11363b1a94c8SDamien Le Moal { 11373b1a94c8SDamien Le Moal kfree(zmd->zones); 11383b1a94c8SDamien Le Moal zmd->zones = NULL; 11393b1a94c8SDamien Le Moal } 11403b1a94c8SDamien Le Moal 11413b1a94c8SDamien Le Moal /* 11423b1a94c8SDamien Le Moal * The size of a zone report in number of zones. 11433b1a94c8SDamien Le Moal * This results in 4096*64B=256KB report zones commands. 11443b1a94c8SDamien Le Moal */ 11453b1a94c8SDamien Le Moal #define DMZ_REPORT_NR_ZONES 4096 11463b1a94c8SDamien Le Moal 11473b1a94c8SDamien Le Moal /* 11483b1a94c8SDamien Le Moal * Allocate and initialize zone descriptors using the zone 11493b1a94c8SDamien Le Moal * information from disk. 11503b1a94c8SDamien Le Moal */ 11513b1a94c8SDamien Le Moal static int dmz_init_zones(struct dmz_metadata *zmd) 11523b1a94c8SDamien Le Moal { 11533b1a94c8SDamien Le Moal struct dmz_dev *dev = zmd->dev; 11543b1a94c8SDamien Le Moal struct dm_zone *zone; 11553b1a94c8SDamien Le Moal struct blk_zone *blkz; 11563b1a94c8SDamien Le Moal unsigned int nr_blkz; 11573b1a94c8SDamien Le Moal sector_t sector = 0; 11583b1a94c8SDamien Le Moal int i, ret = 0; 11593b1a94c8SDamien Le Moal 11603b1a94c8SDamien Le Moal /* Init */ 11613b1a94c8SDamien Le Moal zmd->zone_bitmap_size = dev->zone_nr_blocks >> 3; 11623b1a94c8SDamien Le Moal zmd->zone_nr_bitmap_blocks = zmd->zone_bitmap_size >> DMZ_BLOCK_SHIFT; 11633b1a94c8SDamien Le Moal 11643b1a94c8SDamien Le Moal /* Allocate zone array */ 11653b1a94c8SDamien Le Moal zmd->zones = kcalloc(dev->nr_zones, sizeof(struct dm_zone), GFP_KERNEL); 11663b1a94c8SDamien Le Moal if (!zmd->zones) 11673b1a94c8SDamien Le Moal return -ENOMEM; 11683b1a94c8SDamien Le Moal 11693b1a94c8SDamien Le Moal dmz_dev_info(dev, "Using %zu B for zone information", 11703b1a94c8SDamien Le Moal sizeof(struct dm_zone) * dev->nr_zones); 11713b1a94c8SDamien Le Moal 11723b1a94c8SDamien Le Moal /* Get zone information */ 11733b1a94c8SDamien Le Moal nr_blkz = DMZ_REPORT_NR_ZONES; 11743b1a94c8SDamien Le Moal blkz = kcalloc(nr_blkz, sizeof(struct blk_zone), GFP_KERNEL); 11753b1a94c8SDamien Le Moal if (!blkz) { 11763b1a94c8SDamien Le Moal ret = -ENOMEM; 11773b1a94c8SDamien Le Moal goto out; 11783b1a94c8SDamien Le Moal } 11793b1a94c8SDamien Le Moal 11803b1a94c8SDamien Le Moal /* 11813b1a94c8SDamien Le Moal * Get zone information and initialize zone descriptors. 11823b1a94c8SDamien Le Moal * At the same time, determine where the super block 11833b1a94c8SDamien Le Moal * should be: first block of the first randomly writable 11843b1a94c8SDamien Le Moal * zone. 11853b1a94c8SDamien Le Moal */ 11863b1a94c8SDamien Le Moal zone = zmd->zones; 11873b1a94c8SDamien Le Moal while (sector < dev->capacity) { 11883b1a94c8SDamien Le Moal /* Get zone information */ 11893b1a94c8SDamien Le Moal nr_blkz = DMZ_REPORT_NR_ZONES; 1190bd976e52SDamien Le Moal ret = blkdev_report_zones(dev->bdev, sector, blkz, &nr_blkz); 11913b1a94c8SDamien Le Moal if (ret) { 11923b1a94c8SDamien Le Moal dmz_dev_err(dev, "Report zones failed %d", ret); 11933b1a94c8SDamien Le Moal goto out; 11943b1a94c8SDamien Le Moal } 11953b1a94c8SDamien Le Moal 11967aedf75fSDamien Le Moal if (!nr_blkz) 11977aedf75fSDamien Le Moal break; 11987aedf75fSDamien Le Moal 11993b1a94c8SDamien Le Moal /* Process report */ 12003b1a94c8SDamien Le Moal for (i = 0; i < nr_blkz; i++) { 12013b1a94c8SDamien Le Moal ret = dmz_init_zone(zmd, zone, &blkz[i]); 12023b1a94c8SDamien Le Moal if (ret) 12033b1a94c8SDamien Le Moal goto out; 12043b1a94c8SDamien Le Moal sector += dev->zone_nr_sectors; 12053b1a94c8SDamien Le Moal zone++; 12063b1a94c8SDamien Le Moal } 12073b1a94c8SDamien Le Moal } 12083b1a94c8SDamien Le Moal 12093b1a94c8SDamien Le Moal /* The entire zone configuration of the disk should now be known */ 12103b1a94c8SDamien Le Moal if (sector < dev->capacity) { 12113b1a94c8SDamien Le Moal dmz_dev_err(dev, "Failed to get correct zone information"); 12123b1a94c8SDamien Le Moal ret = -ENXIO; 12133b1a94c8SDamien Le Moal } 12143b1a94c8SDamien Le Moal out: 12153b1a94c8SDamien Le Moal kfree(blkz); 12163b1a94c8SDamien Le Moal if (ret) 12173b1a94c8SDamien Le Moal dmz_drop_zones(zmd); 12183b1a94c8SDamien Le Moal 12193b1a94c8SDamien Le Moal return ret; 12203b1a94c8SDamien Le Moal } 12213b1a94c8SDamien Le Moal 12223b1a94c8SDamien Le Moal /* 12233b1a94c8SDamien Le Moal * Update a zone information. 12243b1a94c8SDamien Le Moal */ 12253b1a94c8SDamien Le Moal static int dmz_update_zone(struct dmz_metadata *zmd, struct dm_zone *zone) 12263b1a94c8SDamien Le Moal { 12273b1a94c8SDamien Le Moal unsigned int nr_blkz = 1; 1228bd976e52SDamien Le Moal unsigned int noio_flag; 12293b1a94c8SDamien Le Moal struct blk_zone blkz; 12303b1a94c8SDamien Le Moal int ret; 12313b1a94c8SDamien Le Moal 1232bd976e52SDamien Le Moal /* 1233bd976e52SDamien Le Moal * Get zone information from disk. Since blkdev_report_zones() uses 1234bd976e52SDamien Le Moal * GFP_KERNEL by default for memory allocations, set the per-task 1235bd976e52SDamien Le Moal * PF_MEMALLOC_NOIO flag so that all allocations are done as if 1236bd976e52SDamien Le Moal * GFP_NOIO was specified. 1237bd976e52SDamien Le Moal */ 1238bd976e52SDamien Le Moal noio_flag = memalloc_noio_save(); 12393b1a94c8SDamien Le Moal ret = blkdev_report_zones(zmd->dev->bdev, dmz_start_sect(zmd, zone), 1240bd976e52SDamien Le Moal &blkz, &nr_blkz); 1241bd976e52SDamien Le Moal memalloc_noio_restore(noio_flag); 12427aedf75fSDamien Le Moal if (!nr_blkz) 12437aedf75fSDamien Le Moal ret = -EIO; 12443b1a94c8SDamien Le Moal if (ret) { 12453b1a94c8SDamien Le Moal dmz_dev_err(zmd->dev, "Get zone %u report failed", 12463b1a94c8SDamien Le Moal dmz_id(zmd, zone)); 12473b1a94c8SDamien Le Moal return ret; 12483b1a94c8SDamien Le Moal } 12493b1a94c8SDamien Le Moal 12503b1a94c8SDamien Le Moal clear_bit(DMZ_OFFLINE, &zone->flags); 12513b1a94c8SDamien Le Moal clear_bit(DMZ_READ_ONLY, &zone->flags); 12523b1a94c8SDamien Le Moal if (blkz.cond == BLK_ZONE_COND_OFFLINE) 12533b1a94c8SDamien Le Moal set_bit(DMZ_OFFLINE, &zone->flags); 12543b1a94c8SDamien Le Moal else if (blkz.cond == BLK_ZONE_COND_READONLY) 12553b1a94c8SDamien Le Moal set_bit(DMZ_READ_ONLY, &zone->flags); 12563b1a94c8SDamien Le Moal 12573b1a94c8SDamien Le Moal if (dmz_is_seq(zone)) 12583b1a94c8SDamien Le Moal zone->wp_block = dmz_sect2blk(blkz.wp - blkz.start); 12593b1a94c8SDamien Le Moal else 12603b1a94c8SDamien Le Moal zone->wp_block = 0; 12613b1a94c8SDamien Le Moal 12623b1a94c8SDamien Le Moal return 0; 12633b1a94c8SDamien Le Moal } 12643b1a94c8SDamien Le Moal 12653b1a94c8SDamien Le Moal /* 12663b1a94c8SDamien Le Moal * Check a zone write pointer position when the zone is marked 12673b1a94c8SDamien Le Moal * with the sequential write error flag. 12683b1a94c8SDamien Le Moal */ 12693b1a94c8SDamien Le Moal static int dmz_handle_seq_write_err(struct dmz_metadata *zmd, 12703b1a94c8SDamien Le Moal struct dm_zone *zone) 12713b1a94c8SDamien Le Moal { 12723b1a94c8SDamien Le Moal unsigned int wp = 0; 12733b1a94c8SDamien Le Moal int ret; 12743b1a94c8SDamien Le Moal 12753b1a94c8SDamien Le Moal wp = zone->wp_block; 12763b1a94c8SDamien Le Moal ret = dmz_update_zone(zmd, zone); 12773b1a94c8SDamien Le Moal if (ret) 12783b1a94c8SDamien Le Moal return ret; 12793b1a94c8SDamien Le Moal 12803b1a94c8SDamien Le Moal dmz_dev_warn(zmd->dev, "Processing zone %u write error (zone wp %u/%u)", 12813b1a94c8SDamien Le Moal dmz_id(zmd, zone), zone->wp_block, wp); 12823b1a94c8SDamien Le Moal 12833b1a94c8SDamien Le Moal if (zone->wp_block < wp) { 12843b1a94c8SDamien Le Moal dmz_invalidate_blocks(zmd, zone, zone->wp_block, 12853b1a94c8SDamien Le Moal wp - zone->wp_block); 12863b1a94c8SDamien Le Moal } 12873b1a94c8SDamien Le Moal 12883b1a94c8SDamien Le Moal return 0; 12893b1a94c8SDamien Le Moal } 12903b1a94c8SDamien Le Moal 12913b1a94c8SDamien Le Moal static struct dm_zone *dmz_get(struct dmz_metadata *zmd, unsigned int zone_id) 12923b1a94c8SDamien Le Moal { 12933b1a94c8SDamien Le Moal return &zmd->zones[zone_id]; 12943b1a94c8SDamien Le Moal } 12953b1a94c8SDamien Le Moal 12963b1a94c8SDamien Le Moal /* 12973b1a94c8SDamien Le Moal * Reset a zone write pointer. 12983b1a94c8SDamien Le Moal */ 12993b1a94c8SDamien Le Moal static int dmz_reset_zone(struct dmz_metadata *zmd, struct dm_zone *zone) 13003b1a94c8SDamien Le Moal { 13013b1a94c8SDamien Le Moal int ret; 13023b1a94c8SDamien Le Moal 13033b1a94c8SDamien Le Moal /* 13043b1a94c8SDamien Le Moal * Ignore offline zones, read only zones, 13053b1a94c8SDamien Le Moal * and conventional zones. 13063b1a94c8SDamien Le Moal */ 13073b1a94c8SDamien Le Moal if (dmz_is_offline(zone) || 13083b1a94c8SDamien Le Moal dmz_is_readonly(zone) || 13093b1a94c8SDamien Le Moal dmz_is_rnd(zone)) 13103b1a94c8SDamien Le Moal return 0; 13113b1a94c8SDamien Le Moal 13123b1a94c8SDamien Le Moal if (!dmz_is_empty(zone) || dmz_seq_write_err(zone)) { 13133b1a94c8SDamien Le Moal struct dmz_dev *dev = zmd->dev; 13143b1a94c8SDamien Le Moal 13153b1a94c8SDamien Le Moal ret = blkdev_reset_zones(dev->bdev, 13163b1a94c8SDamien Le Moal dmz_start_sect(zmd, zone), 13174218a955SDamien Le Moal dev->zone_nr_sectors, GFP_NOIO); 13183b1a94c8SDamien Le Moal if (ret) { 13193b1a94c8SDamien Le Moal dmz_dev_err(dev, "Reset zone %u failed %d", 13203b1a94c8SDamien Le Moal dmz_id(zmd, zone), ret); 13213b1a94c8SDamien Le Moal return ret; 13223b1a94c8SDamien Le Moal } 13233b1a94c8SDamien Le Moal } 13243b1a94c8SDamien Le Moal 13253b1a94c8SDamien Le Moal /* Clear write error bit and rewind write pointer position */ 13263b1a94c8SDamien Le Moal clear_bit(DMZ_SEQ_WRITE_ERR, &zone->flags); 13273b1a94c8SDamien Le Moal zone->wp_block = 0; 13283b1a94c8SDamien Le Moal 13293b1a94c8SDamien Le Moal return 0; 13303b1a94c8SDamien Le Moal } 13313b1a94c8SDamien Le Moal 13323b1a94c8SDamien Le Moal static void dmz_get_zone_weight(struct dmz_metadata *zmd, struct dm_zone *zone); 13333b1a94c8SDamien Le Moal 13343b1a94c8SDamien Le Moal /* 13353b1a94c8SDamien Le Moal * Initialize chunk mapping. 13363b1a94c8SDamien Le Moal */ 13373b1a94c8SDamien Le Moal static int dmz_load_mapping(struct dmz_metadata *zmd) 13383b1a94c8SDamien Le Moal { 13393b1a94c8SDamien Le Moal struct dmz_dev *dev = zmd->dev; 13403b1a94c8SDamien Le Moal struct dm_zone *dzone, *bzone; 13413b1a94c8SDamien Le Moal struct dmz_mblock *dmap_mblk = NULL; 13423b1a94c8SDamien Le Moal struct dmz_map *dmap; 13433b1a94c8SDamien Le Moal unsigned int i = 0, e = 0, chunk = 0; 13443b1a94c8SDamien Le Moal unsigned int dzone_id; 13453b1a94c8SDamien Le Moal unsigned int bzone_id; 13463b1a94c8SDamien Le Moal 13473b1a94c8SDamien Le Moal /* Metadata block array for the chunk mapping table */ 13483b1a94c8SDamien Le Moal zmd->map_mblk = kcalloc(zmd->nr_map_blocks, 13493b1a94c8SDamien Le Moal sizeof(struct dmz_mblk *), GFP_KERNEL); 13503b1a94c8SDamien Le Moal if (!zmd->map_mblk) 13513b1a94c8SDamien Le Moal return -ENOMEM; 13523b1a94c8SDamien Le Moal 13533b1a94c8SDamien Le Moal /* Get chunk mapping table blocks and initialize zone mapping */ 13543b1a94c8SDamien Le Moal while (chunk < zmd->nr_chunks) { 13553b1a94c8SDamien Le Moal if (!dmap_mblk) { 13563b1a94c8SDamien Le Moal /* Get mapping block */ 13573b1a94c8SDamien Le Moal dmap_mblk = dmz_get_mblock(zmd, i + 1); 13583b1a94c8SDamien Le Moal if (IS_ERR(dmap_mblk)) 13593b1a94c8SDamien Le Moal return PTR_ERR(dmap_mblk); 13603b1a94c8SDamien Le Moal zmd->map_mblk[i] = dmap_mblk; 13613b1a94c8SDamien Le Moal dmap = (struct dmz_map *) dmap_mblk->data; 13623b1a94c8SDamien Le Moal i++; 13633b1a94c8SDamien Le Moal e = 0; 13643b1a94c8SDamien Le Moal } 13653b1a94c8SDamien Le Moal 13663b1a94c8SDamien Le Moal /* Check data zone */ 13673b1a94c8SDamien Le Moal dzone_id = le32_to_cpu(dmap[e].dzone_id); 13683b1a94c8SDamien Le Moal if (dzone_id == DMZ_MAP_UNMAPPED) 13693b1a94c8SDamien Le Moal goto next; 13703b1a94c8SDamien Le Moal 13713b1a94c8SDamien Le Moal if (dzone_id >= dev->nr_zones) { 13723b1a94c8SDamien Le Moal dmz_dev_err(dev, "Chunk %u mapping: invalid data zone ID %u", 13733b1a94c8SDamien Le Moal chunk, dzone_id); 13743b1a94c8SDamien Le Moal return -EIO; 13753b1a94c8SDamien Le Moal } 13763b1a94c8SDamien Le Moal 13773b1a94c8SDamien Le Moal dzone = dmz_get(zmd, dzone_id); 13783b1a94c8SDamien Le Moal set_bit(DMZ_DATA, &dzone->flags); 13793b1a94c8SDamien Le Moal dzone->chunk = chunk; 13803b1a94c8SDamien Le Moal dmz_get_zone_weight(zmd, dzone); 13813b1a94c8SDamien Le Moal 13823b1a94c8SDamien Le Moal if (dmz_is_rnd(dzone)) 13833b1a94c8SDamien Le Moal list_add_tail(&dzone->link, &zmd->map_rnd_list); 13843b1a94c8SDamien Le Moal else 13853b1a94c8SDamien Le Moal list_add_tail(&dzone->link, &zmd->map_seq_list); 13863b1a94c8SDamien Le Moal 13873b1a94c8SDamien Le Moal /* Check buffer zone */ 13883b1a94c8SDamien Le Moal bzone_id = le32_to_cpu(dmap[e].bzone_id); 13893b1a94c8SDamien Le Moal if (bzone_id == DMZ_MAP_UNMAPPED) 13903b1a94c8SDamien Le Moal goto next; 13913b1a94c8SDamien Le Moal 13923b1a94c8SDamien Le Moal if (bzone_id >= dev->nr_zones) { 13933b1a94c8SDamien Le Moal dmz_dev_err(dev, "Chunk %u mapping: invalid buffer zone ID %u", 13943b1a94c8SDamien Le Moal chunk, bzone_id); 13953b1a94c8SDamien Le Moal return -EIO; 13963b1a94c8SDamien Le Moal } 13973b1a94c8SDamien Le Moal 13983b1a94c8SDamien Le Moal bzone = dmz_get(zmd, bzone_id); 13993b1a94c8SDamien Le Moal if (!dmz_is_rnd(bzone)) { 14003b1a94c8SDamien Le Moal dmz_dev_err(dev, "Chunk %u mapping: invalid buffer zone %u", 14013b1a94c8SDamien Le Moal chunk, bzone_id); 14023b1a94c8SDamien Le Moal return -EIO; 14033b1a94c8SDamien Le Moal } 14043b1a94c8SDamien Le Moal 14053b1a94c8SDamien Le Moal set_bit(DMZ_DATA, &bzone->flags); 14063b1a94c8SDamien Le Moal set_bit(DMZ_BUF, &bzone->flags); 14073b1a94c8SDamien Le Moal bzone->chunk = chunk; 14083b1a94c8SDamien Le Moal bzone->bzone = dzone; 14093b1a94c8SDamien Le Moal dzone->bzone = bzone; 14103b1a94c8SDamien Le Moal dmz_get_zone_weight(zmd, bzone); 14113b1a94c8SDamien Le Moal list_add_tail(&bzone->link, &zmd->map_rnd_list); 14123b1a94c8SDamien Le Moal next: 14133b1a94c8SDamien Le Moal chunk++; 14143b1a94c8SDamien Le Moal e++; 14153b1a94c8SDamien Le Moal if (e >= DMZ_MAP_ENTRIES) 14163b1a94c8SDamien Le Moal dmap_mblk = NULL; 14173b1a94c8SDamien Le Moal } 14183b1a94c8SDamien Le Moal 14193b1a94c8SDamien Le Moal /* 14203b1a94c8SDamien Le Moal * At this point, only meta zones and mapped data zones were 14213b1a94c8SDamien Le Moal * fully initialized. All remaining zones are unmapped data 14223b1a94c8SDamien Le Moal * zones. Finish initializing those here. 14233b1a94c8SDamien Le Moal */ 14243b1a94c8SDamien Le Moal for (i = 0; i < dev->nr_zones; i++) { 14253b1a94c8SDamien Le Moal dzone = dmz_get(zmd, i); 14263b1a94c8SDamien Le Moal if (dmz_is_meta(dzone)) 14273b1a94c8SDamien Le Moal continue; 14283b1a94c8SDamien Le Moal 14293b1a94c8SDamien Le Moal if (dmz_is_rnd(dzone)) 14303b1a94c8SDamien Le Moal zmd->nr_rnd++; 14313b1a94c8SDamien Le Moal else 14323b1a94c8SDamien Le Moal zmd->nr_seq++; 14333b1a94c8SDamien Le Moal 14343b1a94c8SDamien Le Moal if (dmz_is_data(dzone)) { 14353b1a94c8SDamien Le Moal /* Already initialized */ 14363b1a94c8SDamien Le Moal continue; 14373b1a94c8SDamien Le Moal } 14383b1a94c8SDamien Le Moal 14393b1a94c8SDamien Le Moal /* Unmapped data zone */ 14403b1a94c8SDamien Le Moal set_bit(DMZ_DATA, &dzone->flags); 14413b1a94c8SDamien Le Moal dzone->chunk = DMZ_MAP_UNMAPPED; 14423b1a94c8SDamien Le Moal if (dmz_is_rnd(dzone)) { 14433b1a94c8SDamien Le Moal list_add_tail(&dzone->link, &zmd->unmap_rnd_list); 14443b1a94c8SDamien Le Moal atomic_inc(&zmd->unmap_nr_rnd); 14453b1a94c8SDamien Le Moal } else if (atomic_read(&zmd->nr_reserved_seq_zones) < zmd->nr_reserved_seq) { 14463b1a94c8SDamien Le Moal list_add_tail(&dzone->link, &zmd->reserved_seq_zones_list); 14473b1a94c8SDamien Le Moal atomic_inc(&zmd->nr_reserved_seq_zones); 14483b1a94c8SDamien Le Moal zmd->nr_seq--; 14493b1a94c8SDamien Le Moal } else { 14503b1a94c8SDamien Le Moal list_add_tail(&dzone->link, &zmd->unmap_seq_list); 14513b1a94c8SDamien Le Moal atomic_inc(&zmd->unmap_nr_seq); 14523b1a94c8SDamien Le Moal } 14533b1a94c8SDamien Le Moal } 14543b1a94c8SDamien Le Moal 14553b1a94c8SDamien Le Moal return 0; 14563b1a94c8SDamien Le Moal } 14573b1a94c8SDamien Le Moal 14583b1a94c8SDamien Le Moal /* 14593b1a94c8SDamien Le Moal * Set a data chunk mapping. 14603b1a94c8SDamien Le Moal */ 14613b1a94c8SDamien Le Moal static void dmz_set_chunk_mapping(struct dmz_metadata *zmd, unsigned int chunk, 14623b1a94c8SDamien Le Moal unsigned int dzone_id, unsigned int bzone_id) 14633b1a94c8SDamien Le Moal { 14643b1a94c8SDamien Le Moal struct dmz_mblock *dmap_mblk = zmd->map_mblk[chunk >> DMZ_MAP_ENTRIES_SHIFT]; 14653b1a94c8SDamien Le Moal struct dmz_map *dmap = (struct dmz_map *) dmap_mblk->data; 14663b1a94c8SDamien Le Moal int map_idx = chunk & DMZ_MAP_ENTRIES_MASK; 14673b1a94c8SDamien Le Moal 14683b1a94c8SDamien Le Moal dmap[map_idx].dzone_id = cpu_to_le32(dzone_id); 14693b1a94c8SDamien Le Moal dmap[map_idx].bzone_id = cpu_to_le32(bzone_id); 14703b1a94c8SDamien Le Moal dmz_dirty_mblock(zmd, dmap_mblk); 14713b1a94c8SDamien Le Moal } 14723b1a94c8SDamien Le Moal 14733b1a94c8SDamien Le Moal /* 14743b1a94c8SDamien Le Moal * The list of mapped zones is maintained in LRU order. 14753b1a94c8SDamien Le Moal * This rotates a zone at the end of its map list. 14763b1a94c8SDamien Le Moal */ 14773b1a94c8SDamien Le Moal static void __dmz_lru_zone(struct dmz_metadata *zmd, struct dm_zone *zone) 14783b1a94c8SDamien Le Moal { 14793b1a94c8SDamien Le Moal if (list_empty(&zone->link)) 14803b1a94c8SDamien Le Moal return; 14813b1a94c8SDamien Le Moal 14823b1a94c8SDamien Le Moal list_del_init(&zone->link); 14833b1a94c8SDamien Le Moal if (dmz_is_seq(zone)) { 14843b1a94c8SDamien Le Moal /* LRU rotate sequential zone */ 14853b1a94c8SDamien Le Moal list_add_tail(&zone->link, &zmd->map_seq_list); 14863b1a94c8SDamien Le Moal } else { 14873b1a94c8SDamien Le Moal /* LRU rotate random zone */ 14883b1a94c8SDamien Le Moal list_add_tail(&zone->link, &zmd->map_rnd_list); 14893b1a94c8SDamien Le Moal } 14903b1a94c8SDamien Le Moal } 14913b1a94c8SDamien Le Moal 14923b1a94c8SDamien Le Moal /* 14933b1a94c8SDamien Le Moal * The list of mapped random zones is maintained 14943b1a94c8SDamien Le Moal * in LRU order. This rotates a zone at the end of the list. 14953b1a94c8SDamien Le Moal */ 14963b1a94c8SDamien Le Moal static void dmz_lru_zone(struct dmz_metadata *zmd, struct dm_zone *zone) 14973b1a94c8SDamien Le Moal { 14983b1a94c8SDamien Le Moal __dmz_lru_zone(zmd, zone); 14993b1a94c8SDamien Le Moal if (zone->bzone) 15003b1a94c8SDamien Le Moal __dmz_lru_zone(zmd, zone->bzone); 15013b1a94c8SDamien Le Moal } 15023b1a94c8SDamien Le Moal 15033b1a94c8SDamien Le Moal /* 15043b1a94c8SDamien Le Moal * Wait for any zone to be freed. 15053b1a94c8SDamien Le Moal */ 15063b1a94c8SDamien Le Moal static void dmz_wait_for_free_zones(struct dmz_metadata *zmd) 15073b1a94c8SDamien Le Moal { 15083b1a94c8SDamien Le Moal DEFINE_WAIT(wait); 15093b1a94c8SDamien Le Moal 15103b1a94c8SDamien Le Moal prepare_to_wait(&zmd->free_wq, &wait, TASK_UNINTERRUPTIBLE); 15113b1a94c8SDamien Le Moal dmz_unlock_map(zmd); 15123b1a94c8SDamien Le Moal dmz_unlock_metadata(zmd); 15133b1a94c8SDamien Le Moal 15143b1a94c8SDamien Le Moal io_schedule_timeout(HZ); 15153b1a94c8SDamien Le Moal 15163b1a94c8SDamien Le Moal dmz_lock_metadata(zmd); 15173b1a94c8SDamien Le Moal dmz_lock_map(zmd); 15183b1a94c8SDamien Le Moal finish_wait(&zmd->free_wq, &wait); 15193b1a94c8SDamien Le Moal } 15203b1a94c8SDamien Le Moal 15213b1a94c8SDamien Le Moal /* 15223b1a94c8SDamien Le Moal * Lock a zone for reclaim (set the zone RECLAIM bit). 15233b1a94c8SDamien Le Moal * Returns false if the zone cannot be locked or if it is already locked 15243b1a94c8SDamien Le Moal * and 1 otherwise. 15253b1a94c8SDamien Le Moal */ 15263b1a94c8SDamien Le Moal int dmz_lock_zone_reclaim(struct dm_zone *zone) 15273b1a94c8SDamien Le Moal { 15283b1a94c8SDamien Le Moal /* Active zones cannot be reclaimed */ 15293b1a94c8SDamien Le Moal if (dmz_is_active(zone)) 15303b1a94c8SDamien Le Moal return 0; 15313b1a94c8SDamien Le Moal 15323b1a94c8SDamien Le Moal return !test_and_set_bit(DMZ_RECLAIM, &zone->flags); 15333b1a94c8SDamien Le Moal } 15343b1a94c8SDamien Le Moal 15353b1a94c8SDamien Le Moal /* 15363b1a94c8SDamien Le Moal * Clear a zone reclaim flag. 15373b1a94c8SDamien Le Moal */ 15383b1a94c8SDamien Le Moal void dmz_unlock_zone_reclaim(struct dm_zone *zone) 15393b1a94c8SDamien Le Moal { 15403b1a94c8SDamien Le Moal WARN_ON(dmz_is_active(zone)); 15413b1a94c8SDamien Le Moal WARN_ON(!dmz_in_reclaim(zone)); 15423b1a94c8SDamien Le Moal 15433b1a94c8SDamien Le Moal clear_bit_unlock(DMZ_RECLAIM, &zone->flags); 15443b1a94c8SDamien Le Moal smp_mb__after_atomic(); 15453b1a94c8SDamien Le Moal wake_up_bit(&zone->flags, DMZ_RECLAIM); 15463b1a94c8SDamien Le Moal } 15473b1a94c8SDamien Le Moal 15483b1a94c8SDamien Le Moal /* 15493b1a94c8SDamien Le Moal * Wait for a zone reclaim to complete. 15503b1a94c8SDamien Le Moal */ 15513b1a94c8SDamien Le Moal static void dmz_wait_for_reclaim(struct dmz_metadata *zmd, struct dm_zone *zone) 15523b1a94c8SDamien Le Moal { 15533b1a94c8SDamien Le Moal dmz_unlock_map(zmd); 15543b1a94c8SDamien Le Moal dmz_unlock_metadata(zmd); 15553b1a94c8SDamien Le Moal wait_on_bit_timeout(&zone->flags, DMZ_RECLAIM, TASK_UNINTERRUPTIBLE, HZ); 15563b1a94c8SDamien Le Moal dmz_lock_metadata(zmd); 15573b1a94c8SDamien Le Moal dmz_lock_map(zmd); 15583b1a94c8SDamien Le Moal } 15593b1a94c8SDamien Le Moal 15603b1a94c8SDamien Le Moal /* 15613b1a94c8SDamien Le Moal * Select a random write zone for reclaim. 15623b1a94c8SDamien Le Moal */ 15633b1a94c8SDamien Le Moal static struct dm_zone *dmz_get_rnd_zone_for_reclaim(struct dmz_metadata *zmd) 15643b1a94c8SDamien Le Moal { 15653b1a94c8SDamien Le Moal struct dm_zone *dzone = NULL; 15663b1a94c8SDamien Le Moal struct dm_zone *zone; 15673b1a94c8SDamien Le Moal 15683b1a94c8SDamien Le Moal if (list_empty(&zmd->map_rnd_list)) 1569b234c6d7SDmitry Fomichev return ERR_PTR(-EBUSY); 15703b1a94c8SDamien Le Moal 15713b1a94c8SDamien Le Moal list_for_each_entry(zone, &zmd->map_rnd_list, link) { 15723b1a94c8SDamien Le Moal if (dmz_is_buf(zone)) 15733b1a94c8SDamien Le Moal dzone = zone->bzone; 15743b1a94c8SDamien Le Moal else 15753b1a94c8SDamien Le Moal dzone = zone; 15763b1a94c8SDamien Le Moal if (dmz_lock_zone_reclaim(dzone)) 15773b1a94c8SDamien Le Moal return dzone; 15783b1a94c8SDamien Le Moal } 15793b1a94c8SDamien Le Moal 1580b234c6d7SDmitry Fomichev return ERR_PTR(-EBUSY); 15813b1a94c8SDamien Le Moal } 15823b1a94c8SDamien Le Moal 15833b1a94c8SDamien Le Moal /* 15843b1a94c8SDamien Le Moal * Select a buffered sequential zone for reclaim. 15853b1a94c8SDamien Le Moal */ 15863b1a94c8SDamien Le Moal static struct dm_zone *dmz_get_seq_zone_for_reclaim(struct dmz_metadata *zmd) 15873b1a94c8SDamien Le Moal { 15883b1a94c8SDamien Le Moal struct dm_zone *zone; 15893b1a94c8SDamien Le Moal 15903b1a94c8SDamien Le Moal if (list_empty(&zmd->map_seq_list)) 15913b1a94c8SDamien Le Moal return NULL; 15923b1a94c8SDamien Le Moal 15933b1a94c8SDamien Le Moal list_for_each_entry(zone, &zmd->map_seq_list, link) { 15943b1a94c8SDamien Le Moal if (!zone->bzone) 15953b1a94c8SDamien Le Moal continue; 15963b1a94c8SDamien Le Moal if (dmz_lock_zone_reclaim(zone)) 15973b1a94c8SDamien Le Moal return zone; 15983b1a94c8SDamien Le Moal } 15993b1a94c8SDamien Le Moal 16003b1a94c8SDamien Le Moal return NULL; 16013b1a94c8SDamien Le Moal } 16023b1a94c8SDamien Le Moal 16033b1a94c8SDamien Le Moal /* 16043b1a94c8SDamien Le Moal * Select a zone for reclaim. 16053b1a94c8SDamien Le Moal */ 16063b1a94c8SDamien Le Moal struct dm_zone *dmz_get_zone_for_reclaim(struct dmz_metadata *zmd) 16073b1a94c8SDamien Le Moal { 16083b1a94c8SDamien Le Moal struct dm_zone *zone; 16093b1a94c8SDamien Le Moal 16103b1a94c8SDamien Le Moal /* 16113b1a94c8SDamien Le Moal * Search for a zone candidate to reclaim: 2 cases are possible. 16123b1a94c8SDamien Le Moal * (1) There is no free sequential zones. Then a random data zone 16133b1a94c8SDamien Le Moal * cannot be reclaimed. So choose a sequential zone to reclaim so 16143b1a94c8SDamien Le Moal * that afterward a random zone can be reclaimed. 16153b1a94c8SDamien Le Moal * (2) At least one free sequential zone is available, then choose 16163b1a94c8SDamien Le Moal * the oldest random zone (data or buffer) that can be locked. 16173b1a94c8SDamien Le Moal */ 16183b1a94c8SDamien Le Moal dmz_lock_map(zmd); 16193b1a94c8SDamien Le Moal if (list_empty(&zmd->reserved_seq_zones_list)) 16203b1a94c8SDamien Le Moal zone = dmz_get_seq_zone_for_reclaim(zmd); 16213b1a94c8SDamien Le Moal else 16223b1a94c8SDamien Le Moal zone = dmz_get_rnd_zone_for_reclaim(zmd); 16233b1a94c8SDamien Le Moal dmz_unlock_map(zmd); 16243b1a94c8SDamien Le Moal 16253b1a94c8SDamien Le Moal return zone; 16263b1a94c8SDamien Le Moal } 16273b1a94c8SDamien Le Moal 16283b1a94c8SDamien Le Moal /* 16293b1a94c8SDamien Le Moal * Get the zone mapping a chunk, if the chunk is mapped already. 16303b1a94c8SDamien Le Moal * If no mapping exist and the operation is WRITE, a zone is 16313b1a94c8SDamien Le Moal * allocated and used to map the chunk. 16323b1a94c8SDamien Le Moal * The zone returned will be set to the active state. 16333b1a94c8SDamien Le Moal */ 16343b1a94c8SDamien Le Moal struct dm_zone *dmz_get_chunk_mapping(struct dmz_metadata *zmd, unsigned int chunk, int op) 16353b1a94c8SDamien Le Moal { 16363b1a94c8SDamien Le Moal struct dmz_mblock *dmap_mblk = zmd->map_mblk[chunk >> DMZ_MAP_ENTRIES_SHIFT]; 16373b1a94c8SDamien Le Moal struct dmz_map *dmap = (struct dmz_map *) dmap_mblk->data; 16383b1a94c8SDamien Le Moal int dmap_idx = chunk & DMZ_MAP_ENTRIES_MASK; 16393b1a94c8SDamien Le Moal unsigned int dzone_id; 16403b1a94c8SDamien Le Moal struct dm_zone *dzone = NULL; 16413b1a94c8SDamien Le Moal int ret = 0; 16423b1a94c8SDamien Le Moal 16433b1a94c8SDamien Le Moal dmz_lock_map(zmd); 16443b1a94c8SDamien Le Moal again: 16453b1a94c8SDamien Le Moal /* Get the chunk mapping */ 16463b1a94c8SDamien Le Moal dzone_id = le32_to_cpu(dmap[dmap_idx].dzone_id); 16473b1a94c8SDamien Le Moal if (dzone_id == DMZ_MAP_UNMAPPED) { 16483b1a94c8SDamien Le Moal /* 16493b1a94c8SDamien Le Moal * Read or discard in unmapped chunks are fine. But for 16503b1a94c8SDamien Le Moal * writes, we need a mapping, so get one. 16513b1a94c8SDamien Le Moal */ 16523b1a94c8SDamien Le Moal if (op != REQ_OP_WRITE) 16533b1a94c8SDamien Le Moal goto out; 16543b1a94c8SDamien Le Moal 1655ad1bd578SDmitry Fomichev /* Allocate a random zone */ 16563b1a94c8SDamien Le Moal dzone = dmz_alloc_zone(zmd, DMZ_ALLOC_RND); 16573b1a94c8SDamien Le Moal if (!dzone) { 165875d66ffbSDmitry Fomichev if (dmz_bdev_is_dying(zmd->dev)) { 165975d66ffbSDmitry Fomichev dzone = ERR_PTR(-EIO); 166075d66ffbSDmitry Fomichev goto out; 166175d66ffbSDmitry Fomichev } 16623b1a94c8SDamien Le Moal dmz_wait_for_free_zones(zmd); 16633b1a94c8SDamien Le Moal goto again; 16643b1a94c8SDamien Le Moal } 16653b1a94c8SDamien Le Moal 16663b1a94c8SDamien Le Moal dmz_map_zone(zmd, dzone, chunk); 16673b1a94c8SDamien Le Moal 16683b1a94c8SDamien Le Moal } else { 16693b1a94c8SDamien Le Moal /* The chunk is already mapped: get the mapping zone */ 16703b1a94c8SDamien Le Moal dzone = dmz_get(zmd, dzone_id); 16713b1a94c8SDamien Le Moal if (dzone->chunk != chunk) { 16723b1a94c8SDamien Le Moal dzone = ERR_PTR(-EIO); 16733b1a94c8SDamien Le Moal goto out; 16743b1a94c8SDamien Le Moal } 16753b1a94c8SDamien Le Moal 16763b1a94c8SDamien Le Moal /* Repair write pointer if the sequential dzone has error */ 16773b1a94c8SDamien Le Moal if (dmz_seq_write_err(dzone)) { 16783b1a94c8SDamien Le Moal ret = dmz_handle_seq_write_err(zmd, dzone); 16793b1a94c8SDamien Le Moal if (ret) { 16803b1a94c8SDamien Le Moal dzone = ERR_PTR(-EIO); 16813b1a94c8SDamien Le Moal goto out; 16823b1a94c8SDamien Le Moal } 16833b1a94c8SDamien Le Moal clear_bit(DMZ_SEQ_WRITE_ERR, &dzone->flags); 16843b1a94c8SDamien Le Moal } 16853b1a94c8SDamien Le Moal } 16863b1a94c8SDamien Le Moal 16873b1a94c8SDamien Le Moal /* 16883b1a94c8SDamien Le Moal * If the zone is being reclaimed, the chunk mapping may change 16893b1a94c8SDamien Le Moal * to a different zone. So wait for reclaim and retry. Otherwise, 16903b1a94c8SDamien Le Moal * activate the zone (this will prevent reclaim from touching it). 16913b1a94c8SDamien Le Moal */ 16923b1a94c8SDamien Le Moal if (dmz_in_reclaim(dzone)) { 16933b1a94c8SDamien Le Moal dmz_wait_for_reclaim(zmd, dzone); 16943b1a94c8SDamien Le Moal goto again; 16953b1a94c8SDamien Le Moal } 16963b1a94c8SDamien Le Moal dmz_activate_zone(dzone); 16973b1a94c8SDamien Le Moal dmz_lru_zone(zmd, dzone); 16983b1a94c8SDamien Le Moal out: 16993b1a94c8SDamien Le Moal dmz_unlock_map(zmd); 17003b1a94c8SDamien Le Moal 17013b1a94c8SDamien Le Moal return dzone; 17023b1a94c8SDamien Le Moal } 17033b1a94c8SDamien Le Moal 17043b1a94c8SDamien Le Moal /* 17053b1a94c8SDamien Le Moal * Write and discard change the block validity of data zones and their buffer 17063b1a94c8SDamien Le Moal * zones. Check here that valid blocks are still present. If all blocks are 17073b1a94c8SDamien Le Moal * invalid, the zones can be unmapped on the fly without waiting for reclaim 17083b1a94c8SDamien Le Moal * to do it. 17093b1a94c8SDamien Le Moal */ 17103b1a94c8SDamien Le Moal void dmz_put_chunk_mapping(struct dmz_metadata *zmd, struct dm_zone *dzone) 17113b1a94c8SDamien Le Moal { 17123b1a94c8SDamien Le Moal struct dm_zone *bzone; 17133b1a94c8SDamien Le Moal 17143b1a94c8SDamien Le Moal dmz_lock_map(zmd); 17153b1a94c8SDamien Le Moal 17163b1a94c8SDamien Le Moal bzone = dzone->bzone; 17173b1a94c8SDamien Le Moal if (bzone) { 17183b1a94c8SDamien Le Moal if (dmz_weight(bzone)) 17193b1a94c8SDamien Le Moal dmz_lru_zone(zmd, bzone); 17203b1a94c8SDamien Le Moal else { 17213b1a94c8SDamien Le Moal /* Empty buffer zone: reclaim it */ 17223b1a94c8SDamien Le Moal dmz_unmap_zone(zmd, bzone); 17233b1a94c8SDamien Le Moal dmz_free_zone(zmd, bzone); 17243b1a94c8SDamien Le Moal bzone = NULL; 17253b1a94c8SDamien Le Moal } 17263b1a94c8SDamien Le Moal } 17273b1a94c8SDamien Le Moal 17283b1a94c8SDamien Le Moal /* Deactivate the data zone */ 17293b1a94c8SDamien Le Moal dmz_deactivate_zone(dzone); 17303b1a94c8SDamien Le Moal if (dmz_is_active(dzone) || bzone || dmz_weight(dzone)) 17313b1a94c8SDamien Le Moal dmz_lru_zone(zmd, dzone); 17323b1a94c8SDamien Le Moal else { 17333b1a94c8SDamien Le Moal /* Unbuffered inactive empty data zone: reclaim it */ 17343b1a94c8SDamien Le Moal dmz_unmap_zone(zmd, dzone); 17353b1a94c8SDamien Le Moal dmz_free_zone(zmd, dzone); 17363b1a94c8SDamien Le Moal } 17373b1a94c8SDamien Le Moal 17383b1a94c8SDamien Le Moal dmz_unlock_map(zmd); 17393b1a94c8SDamien Le Moal } 17403b1a94c8SDamien Le Moal 17413b1a94c8SDamien Le Moal /* 17423b1a94c8SDamien Le Moal * Allocate and map a random zone to buffer a chunk 17433b1a94c8SDamien Le Moal * already mapped to a sequential zone. 17443b1a94c8SDamien Le Moal */ 17453b1a94c8SDamien Le Moal struct dm_zone *dmz_get_chunk_buffer(struct dmz_metadata *zmd, 17463b1a94c8SDamien Le Moal struct dm_zone *dzone) 17473b1a94c8SDamien Le Moal { 17483b1a94c8SDamien Le Moal struct dm_zone *bzone; 17493b1a94c8SDamien Le Moal 17503b1a94c8SDamien Le Moal dmz_lock_map(zmd); 17513b1a94c8SDamien Le Moal again: 17523b1a94c8SDamien Le Moal bzone = dzone->bzone; 17533b1a94c8SDamien Le Moal if (bzone) 17543b1a94c8SDamien Le Moal goto out; 17553b1a94c8SDamien Le Moal 1756ad1bd578SDmitry Fomichev /* Allocate a random zone */ 17573b1a94c8SDamien Le Moal bzone = dmz_alloc_zone(zmd, DMZ_ALLOC_RND); 17583b1a94c8SDamien Le Moal if (!bzone) { 175975d66ffbSDmitry Fomichev if (dmz_bdev_is_dying(zmd->dev)) { 176075d66ffbSDmitry Fomichev bzone = ERR_PTR(-EIO); 176175d66ffbSDmitry Fomichev goto out; 176275d66ffbSDmitry Fomichev } 17633b1a94c8SDamien Le Moal dmz_wait_for_free_zones(zmd); 17643b1a94c8SDamien Le Moal goto again; 17653b1a94c8SDamien Le Moal } 17663b1a94c8SDamien Le Moal 17673b1a94c8SDamien Le Moal /* Update the chunk mapping */ 17683b1a94c8SDamien Le Moal dmz_set_chunk_mapping(zmd, dzone->chunk, dmz_id(zmd, dzone), 17693b1a94c8SDamien Le Moal dmz_id(zmd, bzone)); 17703b1a94c8SDamien Le Moal 17713b1a94c8SDamien Le Moal set_bit(DMZ_BUF, &bzone->flags); 17723b1a94c8SDamien Le Moal bzone->chunk = dzone->chunk; 17733b1a94c8SDamien Le Moal bzone->bzone = dzone; 17743b1a94c8SDamien Le Moal dzone->bzone = bzone; 17753b1a94c8SDamien Le Moal list_add_tail(&bzone->link, &zmd->map_rnd_list); 17763b1a94c8SDamien Le Moal out: 17773b1a94c8SDamien Le Moal dmz_unlock_map(zmd); 17783b1a94c8SDamien Le Moal 17793b1a94c8SDamien Le Moal return bzone; 17803b1a94c8SDamien Le Moal } 17813b1a94c8SDamien Le Moal 17823b1a94c8SDamien Le Moal /* 17833b1a94c8SDamien Le Moal * Get an unmapped (free) zone. 17843b1a94c8SDamien Le Moal * This must be called with the mapping lock held. 17853b1a94c8SDamien Le Moal */ 17863b1a94c8SDamien Le Moal struct dm_zone *dmz_alloc_zone(struct dmz_metadata *zmd, unsigned long flags) 17873b1a94c8SDamien Le Moal { 17883b1a94c8SDamien Le Moal struct list_head *list; 17893b1a94c8SDamien Le Moal struct dm_zone *zone; 17903b1a94c8SDamien Le Moal 17913b1a94c8SDamien Le Moal if (flags & DMZ_ALLOC_RND) 17923b1a94c8SDamien Le Moal list = &zmd->unmap_rnd_list; 17933b1a94c8SDamien Le Moal else 17943b1a94c8SDamien Le Moal list = &zmd->unmap_seq_list; 17953b1a94c8SDamien Le Moal again: 17963b1a94c8SDamien Le Moal if (list_empty(list)) { 17973b1a94c8SDamien Le Moal /* 17983b1a94c8SDamien Le Moal * No free zone: if this is for reclaim, allow using the 17993b1a94c8SDamien Le Moal * reserved sequential zones. 18003b1a94c8SDamien Le Moal */ 18013b1a94c8SDamien Le Moal if (!(flags & DMZ_ALLOC_RECLAIM) || 18023b1a94c8SDamien Le Moal list_empty(&zmd->reserved_seq_zones_list)) 18033b1a94c8SDamien Le Moal return NULL; 18043b1a94c8SDamien Le Moal 18053b1a94c8SDamien Le Moal zone = list_first_entry(&zmd->reserved_seq_zones_list, 18063b1a94c8SDamien Le Moal struct dm_zone, link); 18073b1a94c8SDamien Le Moal list_del_init(&zone->link); 18083b1a94c8SDamien Le Moal atomic_dec(&zmd->nr_reserved_seq_zones); 18093b1a94c8SDamien Le Moal return zone; 18103b1a94c8SDamien Le Moal } 18113b1a94c8SDamien Le Moal 18123b1a94c8SDamien Le Moal zone = list_first_entry(list, struct dm_zone, link); 18133b1a94c8SDamien Le Moal list_del_init(&zone->link); 18143b1a94c8SDamien Le Moal 18153b1a94c8SDamien Le Moal if (dmz_is_rnd(zone)) 18163b1a94c8SDamien Le Moal atomic_dec(&zmd->unmap_nr_rnd); 18173b1a94c8SDamien Le Moal else 18183b1a94c8SDamien Le Moal atomic_dec(&zmd->unmap_nr_seq); 18193b1a94c8SDamien Le Moal 18203b1a94c8SDamien Le Moal if (dmz_is_offline(zone)) { 18213b1a94c8SDamien Le Moal dmz_dev_warn(zmd->dev, "Zone %u is offline", dmz_id(zmd, zone)); 18223b1a94c8SDamien Le Moal zone = NULL; 18233b1a94c8SDamien Le Moal goto again; 18243b1a94c8SDamien Le Moal } 18253b1a94c8SDamien Le Moal 18263b1a94c8SDamien Le Moal return zone; 18273b1a94c8SDamien Le Moal } 18283b1a94c8SDamien Le Moal 18293b1a94c8SDamien Le Moal /* 18303b1a94c8SDamien Le Moal * Free a zone. 18313b1a94c8SDamien Le Moal * This must be called with the mapping lock held. 18323b1a94c8SDamien Le Moal */ 18333b1a94c8SDamien Le Moal void dmz_free_zone(struct dmz_metadata *zmd, struct dm_zone *zone) 18343b1a94c8SDamien Le Moal { 18353b1a94c8SDamien Le Moal /* If this is a sequential zone, reset it */ 18363b1a94c8SDamien Le Moal if (dmz_is_seq(zone)) 18373b1a94c8SDamien Le Moal dmz_reset_zone(zmd, zone); 18383b1a94c8SDamien Le Moal 18393b1a94c8SDamien Le Moal /* Return the zone to its type unmap list */ 18403b1a94c8SDamien Le Moal if (dmz_is_rnd(zone)) { 18413b1a94c8SDamien Le Moal list_add_tail(&zone->link, &zmd->unmap_rnd_list); 18423b1a94c8SDamien Le Moal atomic_inc(&zmd->unmap_nr_rnd); 18433b1a94c8SDamien Le Moal } else if (atomic_read(&zmd->nr_reserved_seq_zones) < 18443b1a94c8SDamien Le Moal zmd->nr_reserved_seq) { 18453b1a94c8SDamien Le Moal list_add_tail(&zone->link, &zmd->reserved_seq_zones_list); 18463b1a94c8SDamien Le Moal atomic_inc(&zmd->nr_reserved_seq_zones); 18473b1a94c8SDamien Le Moal } else { 18483b1a94c8SDamien Le Moal list_add_tail(&zone->link, &zmd->unmap_seq_list); 18493b1a94c8SDamien Le Moal atomic_inc(&zmd->unmap_nr_seq); 18503b1a94c8SDamien Le Moal } 18513b1a94c8SDamien Le Moal 18523b1a94c8SDamien Le Moal wake_up_all(&zmd->free_wq); 18533b1a94c8SDamien Le Moal } 18543b1a94c8SDamien Le Moal 18553b1a94c8SDamien Le Moal /* 18563b1a94c8SDamien Le Moal * Map a chunk to a zone. 18573b1a94c8SDamien Le Moal * This must be called with the mapping lock held. 18583b1a94c8SDamien Le Moal */ 18593b1a94c8SDamien Le Moal void dmz_map_zone(struct dmz_metadata *zmd, struct dm_zone *dzone, 18603b1a94c8SDamien Le Moal unsigned int chunk) 18613b1a94c8SDamien Le Moal { 18623b1a94c8SDamien Le Moal /* Set the chunk mapping */ 18633b1a94c8SDamien Le Moal dmz_set_chunk_mapping(zmd, chunk, dmz_id(zmd, dzone), 18643b1a94c8SDamien Le Moal DMZ_MAP_UNMAPPED); 18653b1a94c8SDamien Le Moal dzone->chunk = chunk; 18663b1a94c8SDamien Le Moal if (dmz_is_rnd(dzone)) 18673b1a94c8SDamien Le Moal list_add_tail(&dzone->link, &zmd->map_rnd_list); 18683b1a94c8SDamien Le Moal else 18693b1a94c8SDamien Le Moal list_add_tail(&dzone->link, &zmd->map_seq_list); 18703b1a94c8SDamien Le Moal } 18713b1a94c8SDamien Le Moal 18723b1a94c8SDamien Le Moal /* 18733b1a94c8SDamien Le Moal * Unmap a zone. 18743b1a94c8SDamien Le Moal * This must be called with the mapping lock held. 18753b1a94c8SDamien Le Moal */ 18763b1a94c8SDamien Le Moal void dmz_unmap_zone(struct dmz_metadata *zmd, struct dm_zone *zone) 18773b1a94c8SDamien Le Moal { 18783b1a94c8SDamien Le Moal unsigned int chunk = zone->chunk; 18793b1a94c8SDamien Le Moal unsigned int dzone_id; 18803b1a94c8SDamien Le Moal 18813b1a94c8SDamien Le Moal if (chunk == DMZ_MAP_UNMAPPED) { 18823b1a94c8SDamien Le Moal /* Already unmapped */ 18833b1a94c8SDamien Le Moal return; 18843b1a94c8SDamien Le Moal } 18853b1a94c8SDamien Le Moal 18863b1a94c8SDamien Le Moal if (test_and_clear_bit(DMZ_BUF, &zone->flags)) { 18873b1a94c8SDamien Le Moal /* 18883b1a94c8SDamien Le Moal * Unmapping the chunk buffer zone: clear only 18893b1a94c8SDamien Le Moal * the chunk buffer mapping 18903b1a94c8SDamien Le Moal */ 18913b1a94c8SDamien Le Moal dzone_id = dmz_id(zmd, zone->bzone); 18923b1a94c8SDamien Le Moal zone->bzone->bzone = NULL; 18933b1a94c8SDamien Le Moal zone->bzone = NULL; 18943b1a94c8SDamien Le Moal 18953b1a94c8SDamien Le Moal } else { 18963b1a94c8SDamien Le Moal /* 18973b1a94c8SDamien Le Moal * Unmapping the chunk data zone: the zone must 18983b1a94c8SDamien Le Moal * not be buffered. 18993b1a94c8SDamien Le Moal */ 19003b1a94c8SDamien Le Moal if (WARN_ON(zone->bzone)) { 19013b1a94c8SDamien Le Moal zone->bzone->bzone = NULL; 19023b1a94c8SDamien Le Moal zone->bzone = NULL; 19033b1a94c8SDamien Le Moal } 19043b1a94c8SDamien Le Moal dzone_id = DMZ_MAP_UNMAPPED; 19053b1a94c8SDamien Le Moal } 19063b1a94c8SDamien Le Moal 19073b1a94c8SDamien Le Moal dmz_set_chunk_mapping(zmd, chunk, dzone_id, DMZ_MAP_UNMAPPED); 19083b1a94c8SDamien Le Moal 19093b1a94c8SDamien Le Moal zone->chunk = DMZ_MAP_UNMAPPED; 19103b1a94c8SDamien Le Moal list_del_init(&zone->link); 19113b1a94c8SDamien Le Moal } 19123b1a94c8SDamien Le Moal 19133b1a94c8SDamien Le Moal /* 19143b1a94c8SDamien Le Moal * Set @nr_bits bits in @bitmap starting from @bit. 19153b1a94c8SDamien Le Moal * Return the number of bits changed from 0 to 1. 19163b1a94c8SDamien Le Moal */ 19173b1a94c8SDamien Le Moal static unsigned int dmz_set_bits(unsigned long *bitmap, 19183b1a94c8SDamien Le Moal unsigned int bit, unsigned int nr_bits) 19193b1a94c8SDamien Le Moal { 19203b1a94c8SDamien Le Moal unsigned long *addr; 19213b1a94c8SDamien Le Moal unsigned int end = bit + nr_bits; 19223b1a94c8SDamien Le Moal unsigned int n = 0; 19233b1a94c8SDamien Le Moal 19243b1a94c8SDamien Le Moal while (bit < end) { 19253b1a94c8SDamien Le Moal if (((bit & (BITS_PER_LONG - 1)) == 0) && 19263b1a94c8SDamien Le Moal ((end - bit) >= BITS_PER_LONG)) { 19273b1a94c8SDamien Le Moal /* Try to set the whole word at once */ 19283b1a94c8SDamien Le Moal addr = bitmap + BIT_WORD(bit); 19293b1a94c8SDamien Le Moal if (*addr == 0) { 19303b1a94c8SDamien Le Moal *addr = ULONG_MAX; 19313b1a94c8SDamien Le Moal n += BITS_PER_LONG; 19323b1a94c8SDamien Le Moal bit += BITS_PER_LONG; 19333b1a94c8SDamien Le Moal continue; 19343b1a94c8SDamien Le Moal } 19353b1a94c8SDamien Le Moal } 19363b1a94c8SDamien Le Moal 19373b1a94c8SDamien Le Moal if (!test_and_set_bit(bit, bitmap)) 19383b1a94c8SDamien Le Moal n++; 19393b1a94c8SDamien Le Moal bit++; 19403b1a94c8SDamien Le Moal } 19413b1a94c8SDamien Le Moal 19423b1a94c8SDamien Le Moal return n; 19433b1a94c8SDamien Le Moal } 19443b1a94c8SDamien Le Moal 19453b1a94c8SDamien Le Moal /* 19463b1a94c8SDamien Le Moal * Get the bitmap block storing the bit for chunk_block in zone. 19473b1a94c8SDamien Le Moal */ 19483b1a94c8SDamien Le Moal static struct dmz_mblock *dmz_get_bitmap(struct dmz_metadata *zmd, 19493b1a94c8SDamien Le Moal struct dm_zone *zone, 19503b1a94c8SDamien Le Moal sector_t chunk_block) 19513b1a94c8SDamien Le Moal { 19523b1a94c8SDamien Le Moal sector_t bitmap_block = 1 + zmd->nr_map_blocks + 19533b1a94c8SDamien Le Moal (sector_t)(dmz_id(zmd, zone) * zmd->zone_nr_bitmap_blocks) + 19543b1a94c8SDamien Le Moal (chunk_block >> DMZ_BLOCK_SHIFT_BITS); 19553b1a94c8SDamien Le Moal 19563b1a94c8SDamien Le Moal return dmz_get_mblock(zmd, bitmap_block); 19573b1a94c8SDamien Le Moal } 19583b1a94c8SDamien Le Moal 19593b1a94c8SDamien Le Moal /* 19603b1a94c8SDamien Le Moal * Copy the valid blocks bitmap of from_zone to the bitmap of to_zone. 19613b1a94c8SDamien Le Moal */ 19623b1a94c8SDamien Le Moal int dmz_copy_valid_blocks(struct dmz_metadata *zmd, struct dm_zone *from_zone, 19633b1a94c8SDamien Le Moal struct dm_zone *to_zone) 19643b1a94c8SDamien Le Moal { 19653b1a94c8SDamien Le Moal struct dmz_mblock *from_mblk, *to_mblk; 19663b1a94c8SDamien Le Moal sector_t chunk_block = 0; 19673b1a94c8SDamien Le Moal 19683b1a94c8SDamien Le Moal /* Get the zones bitmap blocks */ 19693b1a94c8SDamien Le Moal while (chunk_block < zmd->dev->zone_nr_blocks) { 19703b1a94c8SDamien Le Moal from_mblk = dmz_get_bitmap(zmd, from_zone, chunk_block); 19713b1a94c8SDamien Le Moal if (IS_ERR(from_mblk)) 19723b1a94c8SDamien Le Moal return PTR_ERR(from_mblk); 19733b1a94c8SDamien Le Moal to_mblk = dmz_get_bitmap(zmd, to_zone, chunk_block); 19743b1a94c8SDamien Le Moal if (IS_ERR(to_mblk)) { 19753b1a94c8SDamien Le Moal dmz_release_mblock(zmd, from_mblk); 19763b1a94c8SDamien Le Moal return PTR_ERR(to_mblk); 19773b1a94c8SDamien Le Moal } 19783b1a94c8SDamien Le Moal 19793b1a94c8SDamien Le Moal memcpy(to_mblk->data, from_mblk->data, DMZ_BLOCK_SIZE); 19803b1a94c8SDamien Le Moal dmz_dirty_mblock(zmd, to_mblk); 19813b1a94c8SDamien Le Moal 19823b1a94c8SDamien Le Moal dmz_release_mblock(zmd, to_mblk); 19833b1a94c8SDamien Le Moal dmz_release_mblock(zmd, from_mblk); 19843b1a94c8SDamien Le Moal 19853b1a94c8SDamien Le Moal chunk_block += DMZ_BLOCK_SIZE_BITS; 19863b1a94c8SDamien Le Moal } 19873b1a94c8SDamien Le Moal 19883b1a94c8SDamien Le Moal to_zone->weight = from_zone->weight; 19893b1a94c8SDamien Le Moal 19903b1a94c8SDamien Le Moal return 0; 19913b1a94c8SDamien Le Moal } 19923b1a94c8SDamien Le Moal 19933b1a94c8SDamien Le Moal /* 19943b1a94c8SDamien Le Moal * Merge the valid blocks bitmap of from_zone into the bitmap of to_zone, 19953b1a94c8SDamien Le Moal * starting from chunk_block. 19963b1a94c8SDamien Le Moal */ 19973b1a94c8SDamien Le Moal int dmz_merge_valid_blocks(struct dmz_metadata *zmd, struct dm_zone *from_zone, 19983b1a94c8SDamien Le Moal struct dm_zone *to_zone, sector_t chunk_block) 19993b1a94c8SDamien Le Moal { 20003b1a94c8SDamien Le Moal unsigned int nr_blocks; 20013b1a94c8SDamien Le Moal int ret; 20023b1a94c8SDamien Le Moal 20033b1a94c8SDamien Le Moal /* Get the zones bitmap blocks */ 20043b1a94c8SDamien Le Moal while (chunk_block < zmd->dev->zone_nr_blocks) { 20053b1a94c8SDamien Le Moal /* Get a valid region from the source zone */ 20063b1a94c8SDamien Le Moal ret = dmz_first_valid_block(zmd, from_zone, &chunk_block); 20073b1a94c8SDamien Le Moal if (ret <= 0) 20083b1a94c8SDamien Le Moal return ret; 20093b1a94c8SDamien Le Moal 20103b1a94c8SDamien Le Moal nr_blocks = ret; 20113b1a94c8SDamien Le Moal ret = dmz_validate_blocks(zmd, to_zone, chunk_block, nr_blocks); 20123b1a94c8SDamien Le Moal if (ret) 20133b1a94c8SDamien Le Moal return ret; 20143b1a94c8SDamien Le Moal 20153b1a94c8SDamien Le Moal chunk_block += nr_blocks; 20163b1a94c8SDamien Le Moal } 20173b1a94c8SDamien Le Moal 20183b1a94c8SDamien Le Moal return 0; 20193b1a94c8SDamien Le Moal } 20203b1a94c8SDamien Le Moal 20213b1a94c8SDamien Le Moal /* 20223b1a94c8SDamien Le Moal * Validate all the blocks in the range [block..block+nr_blocks-1]. 20233b1a94c8SDamien Le Moal */ 20243b1a94c8SDamien Le Moal int dmz_validate_blocks(struct dmz_metadata *zmd, struct dm_zone *zone, 20253b1a94c8SDamien Le Moal sector_t chunk_block, unsigned int nr_blocks) 20263b1a94c8SDamien Le Moal { 20273b1a94c8SDamien Le Moal unsigned int count, bit, nr_bits; 20283b1a94c8SDamien Le Moal unsigned int zone_nr_blocks = zmd->dev->zone_nr_blocks; 20293b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 20303b1a94c8SDamien Le Moal unsigned int n = 0; 20313b1a94c8SDamien Le Moal 20323b1a94c8SDamien Le Moal dmz_dev_debug(zmd->dev, "=> VALIDATE zone %u, block %llu, %u blocks", 20333b1a94c8SDamien Le Moal dmz_id(zmd, zone), (unsigned long long)chunk_block, 20343b1a94c8SDamien Le Moal nr_blocks); 20353b1a94c8SDamien Le Moal 20363b1a94c8SDamien Le Moal WARN_ON(chunk_block + nr_blocks > zone_nr_blocks); 20373b1a94c8SDamien Le Moal 20383b1a94c8SDamien Le Moal while (nr_blocks) { 20393b1a94c8SDamien Le Moal /* Get bitmap block */ 20403b1a94c8SDamien Le Moal mblk = dmz_get_bitmap(zmd, zone, chunk_block); 20413b1a94c8SDamien Le Moal if (IS_ERR(mblk)) 20423b1a94c8SDamien Le Moal return PTR_ERR(mblk); 20433b1a94c8SDamien Le Moal 20443b1a94c8SDamien Le Moal /* Set bits */ 20453b1a94c8SDamien Le Moal bit = chunk_block & DMZ_BLOCK_MASK_BITS; 20463b1a94c8SDamien Le Moal nr_bits = min(nr_blocks, DMZ_BLOCK_SIZE_BITS - bit); 20473b1a94c8SDamien Le Moal 20483b1a94c8SDamien Le Moal count = dmz_set_bits((unsigned long *)mblk->data, bit, nr_bits); 20493b1a94c8SDamien Le Moal if (count) { 20503b1a94c8SDamien Le Moal dmz_dirty_mblock(zmd, mblk); 20513b1a94c8SDamien Le Moal n += count; 20523b1a94c8SDamien Le Moal } 20533b1a94c8SDamien Le Moal dmz_release_mblock(zmd, mblk); 20543b1a94c8SDamien Le Moal 20553b1a94c8SDamien Le Moal nr_blocks -= nr_bits; 20563b1a94c8SDamien Le Moal chunk_block += nr_bits; 20573b1a94c8SDamien Le Moal } 20583b1a94c8SDamien Le Moal 20593b1a94c8SDamien Le Moal if (likely(zone->weight + n <= zone_nr_blocks)) 20603b1a94c8SDamien Le Moal zone->weight += n; 20613b1a94c8SDamien Le Moal else { 20623b1a94c8SDamien Le Moal dmz_dev_warn(zmd->dev, "Zone %u: weight %u should be <= %u", 20633b1a94c8SDamien Le Moal dmz_id(zmd, zone), zone->weight, 20643b1a94c8SDamien Le Moal zone_nr_blocks - n); 20653b1a94c8SDamien Le Moal zone->weight = zone_nr_blocks; 20663b1a94c8SDamien Le Moal } 20673b1a94c8SDamien Le Moal 20683b1a94c8SDamien Le Moal return 0; 20693b1a94c8SDamien Le Moal } 20703b1a94c8SDamien Le Moal 20713b1a94c8SDamien Le Moal /* 20723b1a94c8SDamien Le Moal * Clear nr_bits bits in bitmap starting from bit. 20733b1a94c8SDamien Le Moal * Return the number of bits cleared. 20743b1a94c8SDamien Le Moal */ 20753b1a94c8SDamien Le Moal static int dmz_clear_bits(unsigned long *bitmap, int bit, int nr_bits) 20763b1a94c8SDamien Le Moal { 20773b1a94c8SDamien Le Moal unsigned long *addr; 20783b1a94c8SDamien Le Moal int end = bit + nr_bits; 20793b1a94c8SDamien Le Moal int n = 0; 20803b1a94c8SDamien Le Moal 20813b1a94c8SDamien Le Moal while (bit < end) { 20823b1a94c8SDamien Le Moal if (((bit & (BITS_PER_LONG - 1)) == 0) && 20833b1a94c8SDamien Le Moal ((end - bit) >= BITS_PER_LONG)) { 20843b1a94c8SDamien Le Moal /* Try to clear whole word at once */ 20853b1a94c8SDamien Le Moal addr = bitmap + BIT_WORD(bit); 20863b1a94c8SDamien Le Moal if (*addr == ULONG_MAX) { 20873b1a94c8SDamien Le Moal *addr = 0; 20883b1a94c8SDamien Le Moal n += BITS_PER_LONG; 20893b1a94c8SDamien Le Moal bit += BITS_PER_LONG; 20903b1a94c8SDamien Le Moal continue; 20913b1a94c8SDamien Le Moal } 20923b1a94c8SDamien Le Moal } 20933b1a94c8SDamien Le Moal 20943b1a94c8SDamien Le Moal if (test_and_clear_bit(bit, bitmap)) 20953b1a94c8SDamien Le Moal n++; 20963b1a94c8SDamien Le Moal bit++; 20973b1a94c8SDamien Le Moal } 20983b1a94c8SDamien Le Moal 20993b1a94c8SDamien Le Moal return n; 21003b1a94c8SDamien Le Moal } 21013b1a94c8SDamien Le Moal 21023b1a94c8SDamien Le Moal /* 21033b1a94c8SDamien Le Moal * Invalidate all the blocks in the range [block..block+nr_blocks-1]. 21043b1a94c8SDamien Le Moal */ 21053b1a94c8SDamien Le Moal int dmz_invalidate_blocks(struct dmz_metadata *zmd, struct dm_zone *zone, 21063b1a94c8SDamien Le Moal sector_t chunk_block, unsigned int nr_blocks) 21073b1a94c8SDamien Le Moal { 21083b1a94c8SDamien Le Moal unsigned int count, bit, nr_bits; 21093b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 21103b1a94c8SDamien Le Moal unsigned int n = 0; 21113b1a94c8SDamien Le Moal 21123b1a94c8SDamien Le Moal dmz_dev_debug(zmd->dev, "=> INVALIDATE zone %u, block %llu, %u blocks", 21133b1a94c8SDamien Le Moal dmz_id(zmd, zone), (u64)chunk_block, nr_blocks); 21143b1a94c8SDamien Le Moal 21153b1a94c8SDamien Le Moal WARN_ON(chunk_block + nr_blocks > zmd->dev->zone_nr_blocks); 21163b1a94c8SDamien Le Moal 21173b1a94c8SDamien Le Moal while (nr_blocks) { 21183b1a94c8SDamien Le Moal /* Get bitmap block */ 21193b1a94c8SDamien Le Moal mblk = dmz_get_bitmap(zmd, zone, chunk_block); 21203b1a94c8SDamien Le Moal if (IS_ERR(mblk)) 21213b1a94c8SDamien Le Moal return PTR_ERR(mblk); 21223b1a94c8SDamien Le Moal 21233b1a94c8SDamien Le Moal /* Clear bits */ 21243b1a94c8SDamien Le Moal bit = chunk_block & DMZ_BLOCK_MASK_BITS; 21253b1a94c8SDamien Le Moal nr_bits = min(nr_blocks, DMZ_BLOCK_SIZE_BITS - bit); 21263b1a94c8SDamien Le Moal 21273b1a94c8SDamien Le Moal count = dmz_clear_bits((unsigned long *)mblk->data, 21283b1a94c8SDamien Le Moal bit, nr_bits); 21293b1a94c8SDamien Le Moal if (count) { 21303b1a94c8SDamien Le Moal dmz_dirty_mblock(zmd, mblk); 21313b1a94c8SDamien Le Moal n += count; 21323b1a94c8SDamien Le Moal } 21333b1a94c8SDamien Le Moal dmz_release_mblock(zmd, mblk); 21343b1a94c8SDamien Le Moal 21353b1a94c8SDamien Le Moal nr_blocks -= nr_bits; 21363b1a94c8SDamien Le Moal chunk_block += nr_bits; 21373b1a94c8SDamien Le Moal } 21383b1a94c8SDamien Le Moal 21393b1a94c8SDamien Le Moal if (zone->weight >= n) 21403b1a94c8SDamien Le Moal zone->weight -= n; 21413b1a94c8SDamien Le Moal else { 21423b1a94c8SDamien Le Moal dmz_dev_warn(zmd->dev, "Zone %u: weight %u should be >= %u", 21433b1a94c8SDamien Le Moal dmz_id(zmd, zone), zone->weight, n); 21443b1a94c8SDamien Le Moal zone->weight = 0; 21453b1a94c8SDamien Le Moal } 21463b1a94c8SDamien Le Moal 21473b1a94c8SDamien Le Moal return 0; 21483b1a94c8SDamien Le Moal } 21493b1a94c8SDamien Le Moal 21503b1a94c8SDamien Le Moal /* 21513b1a94c8SDamien Le Moal * Get a block bit value. 21523b1a94c8SDamien Le Moal */ 21533b1a94c8SDamien Le Moal static int dmz_test_block(struct dmz_metadata *zmd, struct dm_zone *zone, 21543b1a94c8SDamien Le Moal sector_t chunk_block) 21553b1a94c8SDamien Le Moal { 21563b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 21573b1a94c8SDamien Le Moal int ret; 21583b1a94c8SDamien Le Moal 21593b1a94c8SDamien Le Moal WARN_ON(chunk_block >= zmd->dev->zone_nr_blocks); 21603b1a94c8SDamien Le Moal 21613b1a94c8SDamien Le Moal /* Get bitmap block */ 21623b1a94c8SDamien Le Moal mblk = dmz_get_bitmap(zmd, zone, chunk_block); 21633b1a94c8SDamien Le Moal if (IS_ERR(mblk)) 21643b1a94c8SDamien Le Moal return PTR_ERR(mblk); 21653b1a94c8SDamien Le Moal 21663b1a94c8SDamien Le Moal /* Get offset */ 21673b1a94c8SDamien Le Moal ret = test_bit(chunk_block & DMZ_BLOCK_MASK_BITS, 21683b1a94c8SDamien Le Moal (unsigned long *) mblk->data) != 0; 21693b1a94c8SDamien Le Moal 21703b1a94c8SDamien Le Moal dmz_release_mblock(zmd, mblk); 21713b1a94c8SDamien Le Moal 21723b1a94c8SDamien Le Moal return ret; 21733b1a94c8SDamien Le Moal } 21743b1a94c8SDamien Le Moal 21753b1a94c8SDamien Le Moal /* 21763b1a94c8SDamien Le Moal * Return the number of blocks from chunk_block to the first block with a bit 21773b1a94c8SDamien Le Moal * value specified by set. Search at most nr_blocks blocks from chunk_block. 21783b1a94c8SDamien Le Moal */ 21793b1a94c8SDamien Le Moal static int dmz_to_next_set_block(struct dmz_metadata *zmd, struct dm_zone *zone, 21803b1a94c8SDamien Le Moal sector_t chunk_block, unsigned int nr_blocks, 21813b1a94c8SDamien Le Moal int set) 21823b1a94c8SDamien Le Moal { 21833b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 21843b1a94c8SDamien Le Moal unsigned int bit, set_bit, nr_bits; 21853b1a94c8SDamien Le Moal unsigned long *bitmap; 21863b1a94c8SDamien Le Moal int n = 0; 21873b1a94c8SDamien Le Moal 21883b1a94c8SDamien Le Moal WARN_ON(chunk_block + nr_blocks > zmd->dev->zone_nr_blocks); 21893b1a94c8SDamien Le Moal 21903b1a94c8SDamien Le Moal while (nr_blocks) { 21913b1a94c8SDamien Le Moal /* Get bitmap block */ 21923b1a94c8SDamien Le Moal mblk = dmz_get_bitmap(zmd, zone, chunk_block); 21933b1a94c8SDamien Le Moal if (IS_ERR(mblk)) 21943b1a94c8SDamien Le Moal return PTR_ERR(mblk); 21953b1a94c8SDamien Le Moal 21963b1a94c8SDamien Le Moal /* Get offset */ 21973b1a94c8SDamien Le Moal bitmap = (unsigned long *) mblk->data; 21983b1a94c8SDamien Le Moal bit = chunk_block & DMZ_BLOCK_MASK_BITS; 21993b1a94c8SDamien Le Moal nr_bits = min(nr_blocks, DMZ_BLOCK_SIZE_BITS - bit); 22003b1a94c8SDamien Le Moal if (set) 22013b1a94c8SDamien Le Moal set_bit = find_next_bit(bitmap, DMZ_BLOCK_SIZE_BITS, bit); 22023b1a94c8SDamien Le Moal else 22033b1a94c8SDamien Le Moal set_bit = find_next_zero_bit(bitmap, DMZ_BLOCK_SIZE_BITS, bit); 22043b1a94c8SDamien Le Moal dmz_release_mblock(zmd, mblk); 22053b1a94c8SDamien Le Moal 22063b1a94c8SDamien Le Moal n += set_bit - bit; 22073b1a94c8SDamien Le Moal if (set_bit < DMZ_BLOCK_SIZE_BITS) 22083b1a94c8SDamien Le Moal break; 22093b1a94c8SDamien Le Moal 22103b1a94c8SDamien Le Moal nr_blocks -= nr_bits; 22113b1a94c8SDamien Le Moal chunk_block += nr_bits; 22123b1a94c8SDamien Le Moal } 22133b1a94c8SDamien Le Moal 22143b1a94c8SDamien Le Moal return n; 22153b1a94c8SDamien Le Moal } 22163b1a94c8SDamien Le Moal 22173b1a94c8SDamien Le Moal /* 22183b1a94c8SDamien Le Moal * Test if chunk_block is valid. If it is, the number of consecutive 22193b1a94c8SDamien Le Moal * valid blocks from chunk_block will be returned. 22203b1a94c8SDamien Le Moal */ 22213b1a94c8SDamien Le Moal int dmz_block_valid(struct dmz_metadata *zmd, struct dm_zone *zone, 22223b1a94c8SDamien Le Moal sector_t chunk_block) 22233b1a94c8SDamien Le Moal { 22243b1a94c8SDamien Le Moal int valid; 22253b1a94c8SDamien Le Moal 22263b1a94c8SDamien Le Moal valid = dmz_test_block(zmd, zone, chunk_block); 22273b1a94c8SDamien Le Moal if (valid <= 0) 22283b1a94c8SDamien Le Moal return valid; 22293b1a94c8SDamien Le Moal 22303b1a94c8SDamien Le Moal /* The block is valid: get the number of valid blocks from block */ 22313b1a94c8SDamien Le Moal return dmz_to_next_set_block(zmd, zone, chunk_block, 22323b1a94c8SDamien Le Moal zmd->dev->zone_nr_blocks - chunk_block, 0); 22333b1a94c8SDamien Le Moal } 22343b1a94c8SDamien Le Moal 22353b1a94c8SDamien Le Moal /* 22363b1a94c8SDamien Le Moal * Find the first valid block from @chunk_block in @zone. 22373b1a94c8SDamien Le Moal * If such a block is found, its number is returned using 22383b1a94c8SDamien Le Moal * @chunk_block and the total number of valid blocks from @chunk_block 22393b1a94c8SDamien Le Moal * is returned. 22403b1a94c8SDamien Le Moal */ 22413b1a94c8SDamien Le Moal int dmz_first_valid_block(struct dmz_metadata *zmd, struct dm_zone *zone, 22423b1a94c8SDamien Le Moal sector_t *chunk_block) 22433b1a94c8SDamien Le Moal { 22443b1a94c8SDamien Le Moal sector_t start_block = *chunk_block; 22453b1a94c8SDamien Le Moal int ret; 22463b1a94c8SDamien Le Moal 22473b1a94c8SDamien Le Moal ret = dmz_to_next_set_block(zmd, zone, start_block, 22483b1a94c8SDamien Le Moal zmd->dev->zone_nr_blocks - start_block, 1); 22493b1a94c8SDamien Le Moal if (ret < 0) 22503b1a94c8SDamien Le Moal return ret; 22513b1a94c8SDamien Le Moal 22523b1a94c8SDamien Le Moal start_block += ret; 22533b1a94c8SDamien Le Moal *chunk_block = start_block; 22543b1a94c8SDamien Le Moal 22553b1a94c8SDamien Le Moal return dmz_to_next_set_block(zmd, zone, start_block, 22563b1a94c8SDamien Le Moal zmd->dev->zone_nr_blocks - start_block, 0); 22573b1a94c8SDamien Le Moal } 22583b1a94c8SDamien Le Moal 22593b1a94c8SDamien Le Moal /* 22603b1a94c8SDamien Le Moal * Count the number of bits set starting from bit up to bit + nr_bits - 1. 22613b1a94c8SDamien Le Moal */ 22623b1a94c8SDamien Le Moal static int dmz_count_bits(void *bitmap, int bit, int nr_bits) 22633b1a94c8SDamien Le Moal { 22643b1a94c8SDamien Le Moal unsigned long *addr; 22653b1a94c8SDamien Le Moal int end = bit + nr_bits; 22663b1a94c8SDamien Le Moal int n = 0; 22673b1a94c8SDamien Le Moal 22683b1a94c8SDamien Le Moal while (bit < end) { 22693b1a94c8SDamien Le Moal if (((bit & (BITS_PER_LONG - 1)) == 0) && 22703b1a94c8SDamien Le Moal ((end - bit) >= BITS_PER_LONG)) { 22713b1a94c8SDamien Le Moal addr = (unsigned long *)bitmap + BIT_WORD(bit); 22723b1a94c8SDamien Le Moal if (*addr == ULONG_MAX) { 22733b1a94c8SDamien Le Moal n += BITS_PER_LONG; 22743b1a94c8SDamien Le Moal bit += BITS_PER_LONG; 22753b1a94c8SDamien Le Moal continue; 22763b1a94c8SDamien Le Moal } 22773b1a94c8SDamien Le Moal } 22783b1a94c8SDamien Le Moal 22793b1a94c8SDamien Le Moal if (test_bit(bit, bitmap)) 22803b1a94c8SDamien Le Moal n++; 22813b1a94c8SDamien Le Moal bit++; 22823b1a94c8SDamien Le Moal } 22833b1a94c8SDamien Le Moal 22843b1a94c8SDamien Le Moal return n; 22853b1a94c8SDamien Le Moal } 22863b1a94c8SDamien Le Moal 22873b1a94c8SDamien Le Moal /* 22883b1a94c8SDamien Le Moal * Get a zone weight. 22893b1a94c8SDamien Le Moal */ 22903b1a94c8SDamien Le Moal static void dmz_get_zone_weight(struct dmz_metadata *zmd, struct dm_zone *zone) 22913b1a94c8SDamien Le Moal { 22923b1a94c8SDamien Le Moal struct dmz_mblock *mblk; 22933b1a94c8SDamien Le Moal sector_t chunk_block = 0; 22943b1a94c8SDamien Le Moal unsigned int bit, nr_bits; 22953b1a94c8SDamien Le Moal unsigned int nr_blocks = zmd->dev->zone_nr_blocks; 22963b1a94c8SDamien Le Moal void *bitmap; 22973b1a94c8SDamien Le Moal int n = 0; 22983b1a94c8SDamien Le Moal 22993b1a94c8SDamien Le Moal while (nr_blocks) { 23003b1a94c8SDamien Le Moal /* Get bitmap block */ 23013b1a94c8SDamien Le Moal mblk = dmz_get_bitmap(zmd, zone, chunk_block); 23023b1a94c8SDamien Le Moal if (IS_ERR(mblk)) { 23033b1a94c8SDamien Le Moal n = 0; 23043b1a94c8SDamien Le Moal break; 23053b1a94c8SDamien Le Moal } 23063b1a94c8SDamien Le Moal 23073b1a94c8SDamien Le Moal /* Count bits in this block */ 23083b1a94c8SDamien Le Moal bitmap = mblk->data; 23093b1a94c8SDamien Le Moal bit = chunk_block & DMZ_BLOCK_MASK_BITS; 23103b1a94c8SDamien Le Moal nr_bits = min(nr_blocks, DMZ_BLOCK_SIZE_BITS - bit); 23113b1a94c8SDamien Le Moal n += dmz_count_bits(bitmap, bit, nr_bits); 23123b1a94c8SDamien Le Moal 23133b1a94c8SDamien Le Moal dmz_release_mblock(zmd, mblk); 23143b1a94c8SDamien Le Moal 23153b1a94c8SDamien Le Moal nr_blocks -= nr_bits; 23163b1a94c8SDamien Le Moal chunk_block += nr_bits; 23173b1a94c8SDamien Le Moal } 23183b1a94c8SDamien Le Moal 23193b1a94c8SDamien Le Moal zone->weight = n; 23203b1a94c8SDamien Le Moal } 23213b1a94c8SDamien Le Moal 23223b1a94c8SDamien Le Moal /* 23233b1a94c8SDamien Le Moal * Cleanup the zoned metadata resources. 23243b1a94c8SDamien Le Moal */ 23253b1a94c8SDamien Le Moal static void dmz_cleanup_metadata(struct dmz_metadata *zmd) 23263b1a94c8SDamien Le Moal { 23273b1a94c8SDamien Le Moal struct rb_root *root; 23283b1a94c8SDamien Le Moal struct dmz_mblock *mblk, *next; 23293b1a94c8SDamien Le Moal int i; 23303b1a94c8SDamien Le Moal 23313b1a94c8SDamien Le Moal /* Release zone mapping resources */ 23323b1a94c8SDamien Le Moal if (zmd->map_mblk) { 23333b1a94c8SDamien Le Moal for (i = 0; i < zmd->nr_map_blocks; i++) 23343b1a94c8SDamien Le Moal dmz_release_mblock(zmd, zmd->map_mblk[i]); 23353b1a94c8SDamien Le Moal kfree(zmd->map_mblk); 23363b1a94c8SDamien Le Moal zmd->map_mblk = NULL; 23373b1a94c8SDamien Le Moal } 23383b1a94c8SDamien Le Moal 23393b1a94c8SDamien Le Moal /* Release super blocks */ 23403b1a94c8SDamien Le Moal for (i = 0; i < 2; i++) { 23413b1a94c8SDamien Le Moal if (zmd->sb[i].mblk) { 23423b1a94c8SDamien Le Moal dmz_free_mblock(zmd, zmd->sb[i].mblk); 23433b1a94c8SDamien Le Moal zmd->sb[i].mblk = NULL; 23443b1a94c8SDamien Le Moal } 23453b1a94c8SDamien Le Moal } 23463b1a94c8SDamien Le Moal 23473b1a94c8SDamien Le Moal /* Free cached blocks */ 23483b1a94c8SDamien Le Moal while (!list_empty(&zmd->mblk_dirty_list)) { 23493b1a94c8SDamien Le Moal mblk = list_first_entry(&zmd->mblk_dirty_list, 23503b1a94c8SDamien Le Moal struct dmz_mblock, link); 23513b1a94c8SDamien Le Moal dmz_dev_warn(zmd->dev, "mblock %llu still in dirty list (ref %u)", 235233c2865fSDamien Le Moal (u64)mblk->no, mblk->ref); 23533b1a94c8SDamien Le Moal list_del_init(&mblk->link); 23543b1a94c8SDamien Le Moal rb_erase(&mblk->node, &zmd->mblk_rbtree); 23553b1a94c8SDamien Le Moal dmz_free_mblock(zmd, mblk); 23563b1a94c8SDamien Le Moal } 23573b1a94c8SDamien Le Moal 23583b1a94c8SDamien Le Moal while (!list_empty(&zmd->mblk_lru_list)) { 23593b1a94c8SDamien Le Moal mblk = list_first_entry(&zmd->mblk_lru_list, 23603b1a94c8SDamien Le Moal struct dmz_mblock, link); 23613b1a94c8SDamien Le Moal list_del_init(&mblk->link); 23623b1a94c8SDamien Le Moal rb_erase(&mblk->node, &zmd->mblk_rbtree); 23633b1a94c8SDamien Le Moal dmz_free_mblock(zmd, mblk); 23643b1a94c8SDamien Le Moal } 23653b1a94c8SDamien Le Moal 23663b1a94c8SDamien Le Moal /* Sanity checks: the mblock rbtree should now be empty */ 23673b1a94c8SDamien Le Moal root = &zmd->mblk_rbtree; 23683b1a94c8SDamien Le Moal rbtree_postorder_for_each_entry_safe(mblk, next, root, node) { 23693b1a94c8SDamien Le Moal dmz_dev_warn(zmd->dev, "mblock %llu ref %u still in rbtree", 237033c2865fSDamien Le Moal (u64)mblk->no, mblk->ref); 237133c2865fSDamien Le Moal mblk->ref = 0; 23723b1a94c8SDamien Le Moal dmz_free_mblock(zmd, mblk); 23733b1a94c8SDamien Le Moal } 23743b1a94c8SDamien Le Moal 23753b1a94c8SDamien Le Moal /* Free the zone descriptors */ 23763b1a94c8SDamien Le Moal dmz_drop_zones(zmd); 2377d5ffebddSMike Snitzer 2378d5ffebddSMike Snitzer mutex_destroy(&zmd->mblk_flush_lock); 2379d5ffebddSMike Snitzer mutex_destroy(&zmd->map_lock); 23803b1a94c8SDamien Le Moal } 23813b1a94c8SDamien Le Moal 23823b1a94c8SDamien Le Moal /* 23833b1a94c8SDamien Le Moal * Initialize the zoned metadata. 23843b1a94c8SDamien Le Moal */ 23853b1a94c8SDamien Le Moal int dmz_ctr_metadata(struct dmz_dev *dev, struct dmz_metadata **metadata) 23863b1a94c8SDamien Le Moal { 23873b1a94c8SDamien Le Moal struct dmz_metadata *zmd; 23883b1a94c8SDamien Le Moal unsigned int i, zid; 23893b1a94c8SDamien Le Moal struct dm_zone *zone; 23903b1a94c8SDamien Le Moal int ret; 23913b1a94c8SDamien Le Moal 23923b1a94c8SDamien Le Moal zmd = kzalloc(sizeof(struct dmz_metadata), GFP_KERNEL); 23933b1a94c8SDamien Le Moal if (!zmd) 23943b1a94c8SDamien Le Moal return -ENOMEM; 23953b1a94c8SDamien Le Moal 23963b1a94c8SDamien Le Moal zmd->dev = dev; 23973b1a94c8SDamien Le Moal zmd->mblk_rbtree = RB_ROOT; 23983b1a94c8SDamien Le Moal init_rwsem(&zmd->mblk_sem); 23993b1a94c8SDamien Le Moal mutex_init(&zmd->mblk_flush_lock); 24003b1a94c8SDamien Le Moal spin_lock_init(&zmd->mblk_lock); 24013b1a94c8SDamien Le Moal INIT_LIST_HEAD(&zmd->mblk_lru_list); 24023b1a94c8SDamien Le Moal INIT_LIST_HEAD(&zmd->mblk_dirty_list); 24033b1a94c8SDamien Le Moal 24043b1a94c8SDamien Le Moal mutex_init(&zmd->map_lock); 24053b1a94c8SDamien Le Moal atomic_set(&zmd->unmap_nr_rnd, 0); 24063b1a94c8SDamien Le Moal INIT_LIST_HEAD(&zmd->unmap_rnd_list); 24073b1a94c8SDamien Le Moal INIT_LIST_HEAD(&zmd->map_rnd_list); 24083b1a94c8SDamien Le Moal 24093b1a94c8SDamien Le Moal atomic_set(&zmd->unmap_nr_seq, 0); 24103b1a94c8SDamien Le Moal INIT_LIST_HEAD(&zmd->unmap_seq_list); 24113b1a94c8SDamien Le Moal INIT_LIST_HEAD(&zmd->map_seq_list); 24123b1a94c8SDamien Le Moal 24133b1a94c8SDamien Le Moal atomic_set(&zmd->nr_reserved_seq_zones, 0); 24143b1a94c8SDamien Le Moal INIT_LIST_HEAD(&zmd->reserved_seq_zones_list); 24153b1a94c8SDamien Le Moal 24163b1a94c8SDamien Le Moal init_waitqueue_head(&zmd->free_wq); 24173b1a94c8SDamien Le Moal 24183b1a94c8SDamien Le Moal /* Initialize zone descriptors */ 24193b1a94c8SDamien Le Moal ret = dmz_init_zones(zmd); 24203b1a94c8SDamien Le Moal if (ret) 24213b1a94c8SDamien Le Moal goto err; 24223b1a94c8SDamien Le Moal 24233b1a94c8SDamien Le Moal /* Get super block */ 24243b1a94c8SDamien Le Moal ret = dmz_load_sb(zmd); 24253b1a94c8SDamien Le Moal if (ret) 24263b1a94c8SDamien Le Moal goto err; 24273b1a94c8SDamien Le Moal 24283b1a94c8SDamien Le Moal /* Set metadata zones starting from sb_zone */ 24293b1a94c8SDamien Le Moal zid = dmz_id(zmd, zmd->sb_zone); 24303b1a94c8SDamien Le Moal for (i = 0; i < zmd->nr_meta_zones << 1; i++) { 24313b1a94c8SDamien Le Moal zone = dmz_get(zmd, zid + i); 24323b1a94c8SDamien Le Moal if (!dmz_is_rnd(zone)) 24333b1a94c8SDamien Le Moal goto err; 24343b1a94c8SDamien Le Moal set_bit(DMZ_META, &zone->flags); 24353b1a94c8SDamien Le Moal } 24363b1a94c8SDamien Le Moal 24373b1a94c8SDamien Le Moal /* Load mapping table */ 24383b1a94c8SDamien Le Moal ret = dmz_load_mapping(zmd); 24393b1a94c8SDamien Le Moal if (ret) 24403b1a94c8SDamien Le Moal goto err; 24413b1a94c8SDamien Le Moal 24423b1a94c8SDamien Le Moal /* 24433b1a94c8SDamien Le Moal * Cache size boundaries: allow at least 2 super blocks, the chunk map 24443b1a94c8SDamien Le Moal * blocks and enough blocks to be able to cache the bitmap blocks of 24453b1a94c8SDamien Le Moal * up to 16 zones when idle (min_nr_mblks). Otherwise, if busy, allow 24463b1a94c8SDamien Le Moal * the cache to add 512 more metadata blocks. 24473b1a94c8SDamien Le Moal */ 24483b1a94c8SDamien Le Moal zmd->min_nr_mblks = 2 + zmd->nr_map_blocks + zmd->zone_nr_bitmap_blocks * 16; 24493b1a94c8SDamien Le Moal zmd->max_nr_mblks = zmd->min_nr_mblks + 512; 24503b1a94c8SDamien Le Moal zmd->mblk_shrinker.count_objects = dmz_mblock_shrinker_count; 24513b1a94c8SDamien Le Moal zmd->mblk_shrinker.scan_objects = dmz_mblock_shrinker_scan; 24523b1a94c8SDamien Le Moal zmd->mblk_shrinker.seeks = DEFAULT_SEEKS; 24533b1a94c8SDamien Le Moal 24543b1a94c8SDamien Le Moal /* Metadata cache shrinker */ 24553b1a94c8SDamien Le Moal ret = register_shrinker(&zmd->mblk_shrinker); 24563b1a94c8SDamien Le Moal if (ret) { 24573b1a94c8SDamien Le Moal dmz_dev_err(dev, "Register metadata cache shrinker failed"); 24583b1a94c8SDamien Le Moal goto err; 24593b1a94c8SDamien Le Moal } 24603b1a94c8SDamien Le Moal 24613b1a94c8SDamien Le Moal dmz_dev_info(dev, "Host-%s zoned block device", 24623b1a94c8SDamien Le Moal bdev_zoned_model(dev->bdev) == BLK_ZONED_HA ? 24633b1a94c8SDamien Le Moal "aware" : "managed"); 24643b1a94c8SDamien Le Moal dmz_dev_info(dev, " %llu 512-byte logical sectors", 24653b1a94c8SDamien Le Moal (u64)dev->capacity); 24663b1a94c8SDamien Le Moal dmz_dev_info(dev, " %u zones of %llu 512-byte logical sectors", 24673b1a94c8SDamien Le Moal dev->nr_zones, (u64)dev->zone_nr_sectors); 24683b1a94c8SDamien Le Moal dmz_dev_info(dev, " %u metadata zones", 24693b1a94c8SDamien Le Moal zmd->nr_meta_zones * 2); 24703b1a94c8SDamien Le Moal dmz_dev_info(dev, " %u data zones for %u chunks", 24713b1a94c8SDamien Le Moal zmd->nr_data_zones, zmd->nr_chunks); 24723b1a94c8SDamien Le Moal dmz_dev_info(dev, " %u random zones (%u unmapped)", 24733b1a94c8SDamien Le Moal zmd->nr_rnd, atomic_read(&zmd->unmap_nr_rnd)); 24743b1a94c8SDamien Le Moal dmz_dev_info(dev, " %u sequential zones (%u unmapped)", 24753b1a94c8SDamien Le Moal zmd->nr_seq, atomic_read(&zmd->unmap_nr_seq)); 24763b1a94c8SDamien Le Moal dmz_dev_info(dev, " %u reserved sequential data zones", 24773b1a94c8SDamien Le Moal zmd->nr_reserved_seq); 24783b1a94c8SDamien Le Moal 24793b1a94c8SDamien Le Moal dmz_dev_debug(dev, "Format:"); 24803b1a94c8SDamien Le Moal dmz_dev_debug(dev, "%u metadata blocks per set (%u max cache)", 24813b1a94c8SDamien Le Moal zmd->nr_meta_blocks, zmd->max_nr_mblks); 24823b1a94c8SDamien Le Moal dmz_dev_debug(dev, " %u data zone mapping blocks", 24833b1a94c8SDamien Le Moal zmd->nr_map_blocks); 24843b1a94c8SDamien Le Moal dmz_dev_debug(dev, " %u bitmap blocks", 24853b1a94c8SDamien Le Moal zmd->nr_bitmap_blocks); 24863b1a94c8SDamien Le Moal 24873b1a94c8SDamien Le Moal *metadata = zmd; 24883b1a94c8SDamien Le Moal 24893b1a94c8SDamien Le Moal return 0; 24903b1a94c8SDamien Le Moal err: 24913b1a94c8SDamien Le Moal dmz_cleanup_metadata(zmd); 24923b1a94c8SDamien Le Moal kfree(zmd); 24933b1a94c8SDamien Le Moal *metadata = NULL; 24943b1a94c8SDamien Le Moal 24953b1a94c8SDamien Le Moal return ret; 24963b1a94c8SDamien Le Moal } 24973b1a94c8SDamien Le Moal 24983b1a94c8SDamien Le Moal /* 24993b1a94c8SDamien Le Moal * Cleanup the zoned metadata resources. 25003b1a94c8SDamien Le Moal */ 25013b1a94c8SDamien Le Moal void dmz_dtr_metadata(struct dmz_metadata *zmd) 25023b1a94c8SDamien Le Moal { 25033b1a94c8SDamien Le Moal unregister_shrinker(&zmd->mblk_shrinker); 25043b1a94c8SDamien Le Moal dmz_cleanup_metadata(zmd); 25053b1a94c8SDamien Le Moal kfree(zmd); 25063b1a94c8SDamien Le Moal } 25073b1a94c8SDamien Le Moal 25083b1a94c8SDamien Le Moal /* 25093b1a94c8SDamien Le Moal * Check zone information on resume. 25103b1a94c8SDamien Le Moal */ 25113b1a94c8SDamien Le Moal int dmz_resume_metadata(struct dmz_metadata *zmd) 25123b1a94c8SDamien Le Moal { 25133b1a94c8SDamien Le Moal struct dmz_dev *dev = zmd->dev; 25143b1a94c8SDamien Le Moal struct dm_zone *zone; 25153b1a94c8SDamien Le Moal sector_t wp_block; 25163b1a94c8SDamien Le Moal unsigned int i; 25173b1a94c8SDamien Le Moal int ret; 25183b1a94c8SDamien Le Moal 25193b1a94c8SDamien Le Moal /* Check zones */ 25203b1a94c8SDamien Le Moal for (i = 0; i < dev->nr_zones; i++) { 25213b1a94c8SDamien Le Moal zone = dmz_get(zmd, i); 25223b1a94c8SDamien Le Moal if (!zone) { 25233b1a94c8SDamien Le Moal dmz_dev_err(dev, "Unable to get zone %u", i); 25243b1a94c8SDamien Le Moal return -EIO; 25253b1a94c8SDamien Le Moal } 25263b1a94c8SDamien Le Moal 25273b1a94c8SDamien Le Moal wp_block = zone->wp_block; 25283b1a94c8SDamien Le Moal 25293b1a94c8SDamien Le Moal ret = dmz_update_zone(zmd, zone); 25303b1a94c8SDamien Le Moal if (ret) { 25313b1a94c8SDamien Le Moal dmz_dev_err(dev, "Broken zone %u", i); 25323b1a94c8SDamien Le Moal return ret; 25333b1a94c8SDamien Le Moal } 25343b1a94c8SDamien Le Moal 25353b1a94c8SDamien Le Moal if (dmz_is_offline(zone)) { 25363b1a94c8SDamien Le Moal dmz_dev_warn(dev, "Zone %u is offline", i); 25373b1a94c8SDamien Le Moal continue; 25383b1a94c8SDamien Le Moal } 25393b1a94c8SDamien Le Moal 25403b1a94c8SDamien Le Moal /* Check write pointer */ 25413b1a94c8SDamien Le Moal if (!dmz_is_seq(zone)) 25423b1a94c8SDamien Le Moal zone->wp_block = 0; 25433b1a94c8SDamien Le Moal else if (zone->wp_block != wp_block) { 25443b1a94c8SDamien Le Moal dmz_dev_err(dev, "Zone %u: Invalid wp (%llu / %llu)", 25453b1a94c8SDamien Le Moal i, (u64)zone->wp_block, (u64)wp_block); 25463b1a94c8SDamien Le Moal zone->wp_block = wp_block; 25473b1a94c8SDamien Le Moal dmz_invalidate_blocks(zmd, zone, zone->wp_block, 25483b1a94c8SDamien Le Moal dev->zone_nr_blocks - zone->wp_block); 25493b1a94c8SDamien Le Moal } 25503b1a94c8SDamien Le Moal } 25513b1a94c8SDamien Le Moal 25523b1a94c8SDamien Le Moal return 0; 25533b1a94c8SDamien Le Moal } 2554