Back to home page

OSCL-LXR

 
 

    


0001 // SPDX-License-Identifier: GPL-2.0-or-later
0002 /*
0003    drbd_bitmap.c
0004 
0005    This file is part of DRBD by Philipp Reisner and Lars Ellenberg.
0006 
0007    Copyright (C) 2004-2008, LINBIT Information Technologies GmbH.
0008    Copyright (C) 2004-2008, Philipp Reisner <philipp.reisner@linbit.com>.
0009    Copyright (C) 2004-2008, Lars Ellenberg <lars.ellenberg@linbit.com>.
0010 
0011  */
0012 
0013 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
0014 
0015 #include <linux/bitmap.h>
0016 #include <linux/vmalloc.h>
0017 #include <linux/string.h>
0018 #include <linux/drbd.h>
0019 #include <linux/slab.h>
0020 #include <linux/highmem.h>
0021 
0022 #include "drbd_int.h"
0023 
0024 
0025 /* OPAQUE outside this file!
0026  * interface defined in drbd_int.h
0027 
0028  * convention:
0029  * function name drbd_bm_... => used elsewhere, "public".
0030  * function name      bm_... => internal to implementation, "private".
0031  */
0032 
0033 
0034 /*
0035  * LIMITATIONS:
0036  * We want to support >= peta byte of backend storage, while for now still using
0037  * a granularity of one bit per 4KiB of storage.
0038  * 1 << 50      bytes backend storage (1 PiB)
0039  * 1 << (50 - 12)   bits needed
0040  *  38 --> we need u64 to index and count bits
0041  * 1 << (38 - 3)    bitmap bytes needed
0042  *  35 --> we still need u64 to index and count bytes
0043  *          (that's 32 GiB of bitmap for 1 PiB storage)
0044  * 1 << (35 - 2)    32bit longs needed
0045  *  33 --> we'd even need u64 to index and count 32bit long words.
0046  * 1 << (35 - 3)    64bit longs needed
0047  *  32 --> we could get away with a 32bit unsigned int to index and count
0048  *  64bit long words, but I rather stay with unsigned long for now.
0049  *  We probably should neither count nor point to bytes or long words
0050  *  directly, but either by bitnumber, or by page index and offset.
0051  * 1 << (35 - 12)
0052  *  22 --> we need that much 4KiB pages of bitmap.
0053  *  1 << (22 + 3) --> on a 64bit arch,
0054  *  we need 32 MiB to store the array of page pointers.
0055  *
0056  * Because I'm lazy, and because the resulting patch was too large, too ugly
0057  * and still incomplete, on 32bit we still "only" support 16 TiB (minus some),
0058  * (1 << 32) bits * 4k storage.
0059  *
0060 
0061  * bitmap storage and IO:
0062  *  Bitmap is stored little endian on disk, and is kept little endian in
0063  *  core memory. Currently we still hold the full bitmap in core as long
0064  *  as we are "attached" to a local disk, which at 32 GiB for 1PiB storage
0065  *  seems excessive.
0066  *
0067  *  We plan to reduce the amount of in-core bitmap pages by paging them in
0068  *  and out against their on-disk location as necessary, but need to make
0069  *  sure we don't cause too much meta data IO, and must not deadlock in
0070  *  tight memory situations. This needs some more work.
0071  */
0072 
0073 /*
0074  * NOTE
0075  *  Access to the *bm_pages is protected by bm_lock.
0076  *  It is safe to read the other members within the lock.
0077  *
0078  *  drbd_bm_set_bits is called from bio_endio callbacks,
0079  *  We may be called with irq already disabled,
0080  *  so we need spin_lock_irqsave().
0081  *  And we need the kmap_atomic.
0082  */
0083 struct drbd_bitmap {
0084     struct page **bm_pages;
0085     spinlock_t bm_lock;
0086 
0087     /* exclusively to be used by __al_write_transaction(),
0088      * drbd_bm_mark_for_writeout() and
0089      * and drbd_bm_write_hinted() -> bm_rw() called from there.
0090      */
0091     unsigned int n_bitmap_hints;
0092     unsigned int al_bitmap_hints[AL_UPDATES_PER_TRANSACTION];
0093 
0094     /* see LIMITATIONS: above */
0095 
0096     unsigned long bm_set;       /* nr of set bits; THINK maybe atomic_t? */
0097     unsigned long bm_bits;
0098     size_t   bm_words;
0099     size_t   bm_number_of_pages;
0100     sector_t bm_dev_capacity;
0101     struct mutex bm_change; /* serializes resize operations */
0102 
0103     wait_queue_head_t bm_io_wait; /* used to serialize IO of single pages */
0104 
0105     enum bm_flag bm_flags;
0106 
0107     /* debugging aid, in case we are still racy somewhere */
0108     char          *bm_why;
0109     struct task_struct *bm_task;
0110 };
0111 
0112 #define bm_print_lock_info(m) __bm_print_lock_info(m, __func__)
0113 static void __bm_print_lock_info(struct drbd_device *device, const char *func)
0114 {
0115     struct drbd_bitmap *b = device->bitmap;
0116     if (!__ratelimit(&drbd_ratelimit_state))
0117         return;
0118     drbd_err(device, "FIXME %s[%d] in %s, bitmap locked for '%s' by %s[%d]\n",
0119          current->comm, task_pid_nr(current),
0120          func, b->bm_why ?: "?",
0121          b->bm_task->comm, task_pid_nr(b->bm_task));
0122 }
0123 
0124 void drbd_bm_lock(struct drbd_device *device, char *why, enum bm_flag flags)
0125 {
0126     struct drbd_bitmap *b = device->bitmap;
0127     int trylock_failed;
0128 
0129     if (!b) {
0130         drbd_err(device, "FIXME no bitmap in drbd_bm_lock!?\n");
0131         return;
0132     }
0133 
0134     trylock_failed = !mutex_trylock(&b->bm_change);
0135 
0136     if (trylock_failed) {
0137         drbd_warn(device, "%s[%d] going to '%s' but bitmap already locked for '%s' by %s[%d]\n",
0138               current->comm, task_pid_nr(current),
0139               why, b->bm_why ?: "?",
0140               b->bm_task->comm, task_pid_nr(b->bm_task));
0141         mutex_lock(&b->bm_change);
0142     }
0143     if (BM_LOCKED_MASK & b->bm_flags)
0144         drbd_err(device, "FIXME bitmap already locked in bm_lock\n");
0145     b->bm_flags |= flags & BM_LOCKED_MASK;
0146 
0147     b->bm_why  = why;
0148     b->bm_task = current;
0149 }
0150 
0151 void drbd_bm_unlock(struct drbd_device *device)
0152 {
0153     struct drbd_bitmap *b = device->bitmap;
0154     if (!b) {
0155         drbd_err(device, "FIXME no bitmap in drbd_bm_unlock!?\n");
0156         return;
0157     }
0158 
0159     if (!(BM_LOCKED_MASK & device->bitmap->bm_flags))
0160         drbd_err(device, "FIXME bitmap not locked in bm_unlock\n");
0161 
0162     b->bm_flags &= ~BM_LOCKED_MASK;
0163     b->bm_why  = NULL;
0164     b->bm_task = NULL;
0165     mutex_unlock(&b->bm_change);
0166 }
0167 
0168 /* we store some "meta" info about our pages in page->private */
0169 /* at a granularity of 4k storage per bitmap bit:
0170  * one peta byte storage: 1<<50 byte, 1<<38 * 4k storage blocks
0171  *  1<<38 bits,
0172  *  1<<23 4k bitmap pages.
0173  * Use 24 bits as page index, covers 2 peta byte storage
0174  * at a granularity of 4k per bit.
0175  * Used to report the failed page idx on io error from the endio handlers.
0176  */
0177 #define BM_PAGE_IDX_MASK    ((1UL<<24)-1)
0178 /* this page is currently read in, or written back */
0179 #define BM_PAGE_IO_LOCK     31
0180 /* if there has been an IO error for this page */
0181 #define BM_PAGE_IO_ERROR    30
0182 /* this is to be able to intelligently skip disk IO,
0183  * set if bits have been set since last IO. */
0184 #define BM_PAGE_NEED_WRITEOUT   29
0185 /* to mark for lazy writeout once syncer cleared all clearable bits,
0186  * we if bits have been cleared since last IO. */
0187 #define BM_PAGE_LAZY_WRITEOUT   28
0188 /* pages marked with this "HINT" will be considered for writeout
0189  * on activity log transactions */
0190 #define BM_PAGE_HINT_WRITEOUT   27
0191 
0192 /* store_page_idx uses non-atomic assignment. It is only used directly after
0193  * allocating the page.  All other bm_set_page_* and bm_clear_page_* need to
0194  * use atomic bit manipulation, as set_out_of_sync (and therefore bitmap
0195  * changes) may happen from various contexts, and wait_on_bit/wake_up_bit
0196  * requires it all to be atomic as well. */
0197 static void bm_store_page_idx(struct page *page, unsigned long idx)
0198 {
0199     BUG_ON(0 != (idx & ~BM_PAGE_IDX_MASK));
0200     set_page_private(page, idx);
0201 }
0202 
0203 static unsigned long bm_page_to_idx(struct page *page)
0204 {
0205     return page_private(page) & BM_PAGE_IDX_MASK;
0206 }
0207 
0208 /* As is very unlikely that the same page is under IO from more than one
0209  * context, we can get away with a bit per page and one wait queue per bitmap.
0210  */
0211 static void bm_page_lock_io(struct drbd_device *device, int page_nr)
0212 {
0213     struct drbd_bitmap *b = device->bitmap;
0214     void *addr = &page_private(b->bm_pages[page_nr]);
0215     wait_event(b->bm_io_wait, !test_and_set_bit(BM_PAGE_IO_LOCK, addr));
0216 }
0217 
0218 static void bm_page_unlock_io(struct drbd_device *device, int page_nr)
0219 {
0220     struct drbd_bitmap *b = device->bitmap;
0221     void *addr = &page_private(b->bm_pages[page_nr]);
0222     clear_bit_unlock(BM_PAGE_IO_LOCK, addr);
0223     wake_up(&device->bitmap->bm_io_wait);
0224 }
0225 
0226 /* set _before_ submit_io, so it may be reset due to being changed
0227  * while this page is in flight... will get submitted later again */
0228 static void bm_set_page_unchanged(struct page *page)
0229 {
0230     /* use cmpxchg? */
0231     clear_bit(BM_PAGE_NEED_WRITEOUT, &page_private(page));
0232     clear_bit(BM_PAGE_LAZY_WRITEOUT, &page_private(page));
0233 }
0234 
0235 static void bm_set_page_need_writeout(struct page *page)
0236 {
0237     set_bit(BM_PAGE_NEED_WRITEOUT, &page_private(page));
0238 }
0239 
0240 void drbd_bm_reset_al_hints(struct drbd_device *device)
0241 {
0242     device->bitmap->n_bitmap_hints = 0;
0243 }
0244 
0245 /**
0246  * drbd_bm_mark_for_writeout() - mark a page with a "hint" to be considered for writeout
0247  * @device: DRBD device.
0248  * @page_nr:    the bitmap page to mark with the "hint" flag
0249  *
0250  * From within an activity log transaction, we mark a few pages with these
0251  * hints, then call drbd_bm_write_hinted(), which will only write out changed
0252  * pages which are flagged with this mark.
0253  */
0254 void drbd_bm_mark_for_writeout(struct drbd_device *device, int page_nr)
0255 {
0256     struct drbd_bitmap *b = device->bitmap;
0257     struct page *page;
0258     if (page_nr >= device->bitmap->bm_number_of_pages) {
0259         drbd_warn(device, "BAD: page_nr: %u, number_of_pages: %u\n",
0260              page_nr, (int)device->bitmap->bm_number_of_pages);
0261         return;
0262     }
0263     page = device->bitmap->bm_pages[page_nr];
0264     BUG_ON(b->n_bitmap_hints >= ARRAY_SIZE(b->al_bitmap_hints));
0265     if (!test_and_set_bit(BM_PAGE_HINT_WRITEOUT, &page_private(page)))
0266         b->al_bitmap_hints[b->n_bitmap_hints++] = page_nr;
0267 }
0268 
0269 static int bm_test_page_unchanged(struct page *page)
0270 {
0271     volatile const unsigned long *addr = &page_private(page);
0272     return (*addr & ((1UL<<BM_PAGE_NEED_WRITEOUT)|(1UL<<BM_PAGE_LAZY_WRITEOUT))) == 0;
0273 }
0274 
0275 static void bm_set_page_io_err(struct page *page)
0276 {
0277     set_bit(BM_PAGE_IO_ERROR, &page_private(page));
0278 }
0279 
0280 static void bm_clear_page_io_err(struct page *page)
0281 {
0282     clear_bit(BM_PAGE_IO_ERROR, &page_private(page));
0283 }
0284 
0285 static void bm_set_page_lazy_writeout(struct page *page)
0286 {
0287     set_bit(BM_PAGE_LAZY_WRITEOUT, &page_private(page));
0288 }
0289 
0290 static int bm_test_page_lazy_writeout(struct page *page)
0291 {
0292     return test_bit(BM_PAGE_LAZY_WRITEOUT, &page_private(page));
0293 }
0294 
0295 /* on a 32bit box, this would allow for exactly (2<<38) bits. */
0296 static unsigned int bm_word_to_page_idx(struct drbd_bitmap *b, unsigned long long_nr)
0297 {
0298     /* page_nr = (word*sizeof(long)) >> PAGE_SHIFT; */
0299     unsigned int page_nr = long_nr >> (PAGE_SHIFT - LN2_BPL + 3);
0300     BUG_ON(page_nr >= b->bm_number_of_pages);
0301     return page_nr;
0302 }
0303 
0304 static unsigned int bm_bit_to_page_idx(struct drbd_bitmap *b, u64 bitnr)
0305 {
0306     /* page_nr = (bitnr/8) >> PAGE_SHIFT; */
0307     unsigned int page_nr = bitnr >> (PAGE_SHIFT + 3);
0308     BUG_ON(page_nr >= b->bm_number_of_pages);
0309     return page_nr;
0310 }
0311 
0312 static unsigned long *__bm_map_pidx(struct drbd_bitmap *b, unsigned int idx)
0313 {
0314     struct page *page = b->bm_pages[idx];
0315     return (unsigned long *) kmap_atomic(page);
0316 }
0317 
0318 static unsigned long *bm_map_pidx(struct drbd_bitmap *b, unsigned int idx)
0319 {
0320     return __bm_map_pidx(b, idx);
0321 }
0322 
0323 static void __bm_unmap(unsigned long *p_addr)
0324 {
0325     kunmap_atomic(p_addr);
0326 };
0327 
0328 static void bm_unmap(unsigned long *p_addr)
0329 {
0330     return __bm_unmap(p_addr);
0331 }
0332 
0333 /* long word offset of _bitmap_ sector */
0334 #define S2W(s)  ((s)<<(BM_EXT_SHIFT-BM_BLOCK_SHIFT-LN2_BPL))
0335 /* word offset from start of bitmap to word number _in_page_
0336  * modulo longs per page
0337 #define MLPP(X) ((X) % (PAGE_SIZE/sizeof(long))
0338  hm, well, Philipp thinks gcc might not optimize the % into & (... - 1)
0339  so do it explicitly:
0340  */
0341 #define MLPP(X) ((X) & ((PAGE_SIZE/sizeof(long))-1))
0342 
0343 /* Long words per page */
0344 #define LWPP (PAGE_SIZE/sizeof(long))
0345 
0346 /*
0347  * actually most functions herein should take a struct drbd_bitmap*, not a
0348  * struct drbd_device*, but for the debug macros I like to have the device around
0349  * to be able to report device specific.
0350  */
0351 
0352 
0353 static void bm_free_pages(struct page **pages, unsigned long number)
0354 {
0355     unsigned long i;
0356     if (!pages)
0357         return;
0358 
0359     for (i = 0; i < number; i++) {
0360         if (!pages[i]) {
0361             pr_alert("bm_free_pages tried to free a NULL pointer; i=%lu n=%lu\n",
0362                  i, number);
0363             continue;
0364         }
0365         __free_page(pages[i]);
0366         pages[i] = NULL;
0367     }
0368 }
0369 
0370 static inline void bm_vk_free(void *ptr)
0371 {
0372     kvfree(ptr);
0373 }
0374 
0375 /*
0376  * "have" and "want" are NUMBER OF PAGES.
0377  */
0378 static struct page **bm_realloc_pages(struct drbd_bitmap *b, unsigned long want)
0379 {
0380     struct page **old_pages = b->bm_pages;
0381     struct page **new_pages, *page;
0382     unsigned int i, bytes;
0383     unsigned long have = b->bm_number_of_pages;
0384 
0385     BUG_ON(have == 0 && old_pages != NULL);
0386     BUG_ON(have != 0 && old_pages == NULL);
0387 
0388     if (have == want)
0389         return old_pages;
0390 
0391     /* Trying kmalloc first, falling back to vmalloc.
0392      * GFP_NOIO, as this is called while drbd IO is "suspended",
0393      * and during resize or attach on diskless Primary,
0394      * we must not block on IO to ourselves.
0395      * Context is receiver thread or dmsetup. */
0396     bytes = sizeof(struct page *)*want;
0397     new_pages = kzalloc(bytes, GFP_NOIO | __GFP_NOWARN);
0398     if (!new_pages) {
0399         new_pages = __vmalloc(bytes, GFP_NOIO | __GFP_ZERO);
0400         if (!new_pages)
0401             return NULL;
0402     }
0403 
0404     if (want >= have) {
0405         for (i = 0; i < have; i++)
0406             new_pages[i] = old_pages[i];
0407         for (; i < want; i++) {
0408             page = alloc_page(GFP_NOIO | __GFP_HIGHMEM);
0409             if (!page) {
0410                 bm_free_pages(new_pages + have, i - have);
0411                 bm_vk_free(new_pages);
0412                 return NULL;
0413             }
0414             /* we want to know which page it is
0415              * from the endio handlers */
0416             bm_store_page_idx(page, i);
0417             new_pages[i] = page;
0418         }
0419     } else {
0420         for (i = 0; i < want; i++)
0421             new_pages[i] = old_pages[i];
0422         /* NOT HERE, we are outside the spinlock!
0423         bm_free_pages(old_pages + want, have - want);
0424         */
0425     }
0426 
0427     return new_pages;
0428 }
0429 
0430 /*
0431  * allocates the drbd_bitmap and stores it in device->bitmap.
0432  */
0433 int drbd_bm_init(struct drbd_device *device)
0434 {
0435     struct drbd_bitmap *b = device->bitmap;
0436     WARN_ON(b != NULL);
0437     b = kzalloc(sizeof(struct drbd_bitmap), GFP_KERNEL);
0438     if (!b)
0439         return -ENOMEM;
0440     spin_lock_init(&b->bm_lock);
0441     mutex_init(&b->bm_change);
0442     init_waitqueue_head(&b->bm_io_wait);
0443 
0444     device->bitmap = b;
0445 
0446     return 0;
0447 }
0448 
0449 sector_t drbd_bm_capacity(struct drbd_device *device)
0450 {
0451     if (!expect(device->bitmap))
0452         return 0;
0453     return device->bitmap->bm_dev_capacity;
0454 }
0455 
0456 /* called on driver unload. TODO: call when a device is destroyed.
0457  */
0458 void drbd_bm_cleanup(struct drbd_device *device)
0459 {
0460     if (!expect(device->bitmap))
0461         return;
0462     bm_free_pages(device->bitmap->bm_pages, device->bitmap->bm_number_of_pages);
0463     bm_vk_free(device->bitmap->bm_pages);
0464     kfree(device->bitmap);
0465     device->bitmap = NULL;
0466 }
0467 
0468 /*
0469  * since (b->bm_bits % BITS_PER_LONG) != 0,
0470  * this masks out the remaining bits.
0471  * Returns the number of bits cleared.
0472  */
0473 #ifndef BITS_PER_PAGE
0474 #define BITS_PER_PAGE       (1UL << (PAGE_SHIFT + 3))
0475 #define BITS_PER_PAGE_MASK  (BITS_PER_PAGE - 1)
0476 #else
0477 # if BITS_PER_PAGE != (1UL << (PAGE_SHIFT + 3))
0478 #  error "ambiguous BITS_PER_PAGE"
0479 # endif
0480 #endif
0481 #define BITS_PER_LONG_MASK  (BITS_PER_LONG - 1)
0482 static int bm_clear_surplus(struct drbd_bitmap *b)
0483 {
0484     unsigned long mask;
0485     unsigned long *p_addr, *bm;
0486     int tmp;
0487     int cleared = 0;
0488 
0489     /* number of bits modulo bits per page */
0490     tmp = (b->bm_bits & BITS_PER_PAGE_MASK);
0491     /* mask the used bits of the word containing the last bit */
0492     mask = (1UL << (tmp & BITS_PER_LONG_MASK)) -1;
0493     /* bitmap is always stored little endian,
0494      * on disk and in core memory alike */
0495     mask = cpu_to_lel(mask);
0496 
0497     p_addr = bm_map_pidx(b, b->bm_number_of_pages - 1);
0498     bm = p_addr + (tmp/BITS_PER_LONG);
0499     if (mask) {
0500         /* If mask != 0, we are not exactly aligned, so bm now points
0501          * to the long containing the last bit.
0502          * If mask == 0, bm already points to the word immediately
0503          * after the last (long word aligned) bit. */
0504         cleared = hweight_long(*bm & ~mask);
0505         *bm &= mask;
0506         bm++;
0507     }
0508 
0509     if (BITS_PER_LONG == 32 && ((bm - p_addr) & 1) == 1) {
0510         /* on a 32bit arch, we may need to zero out
0511          * a padding long to align with a 64bit remote */
0512         cleared += hweight_long(*bm);
0513         *bm = 0;
0514     }
0515     bm_unmap(p_addr);
0516     return cleared;
0517 }
0518 
0519 static void bm_set_surplus(struct drbd_bitmap *b)
0520 {
0521     unsigned long mask;
0522     unsigned long *p_addr, *bm;
0523     int tmp;
0524 
0525     /* number of bits modulo bits per page */
0526     tmp = (b->bm_bits & BITS_PER_PAGE_MASK);
0527     /* mask the used bits of the word containing the last bit */
0528     mask = (1UL << (tmp & BITS_PER_LONG_MASK)) -1;
0529     /* bitmap is always stored little endian,
0530      * on disk and in core memory alike */
0531     mask = cpu_to_lel(mask);
0532 
0533     p_addr = bm_map_pidx(b, b->bm_number_of_pages - 1);
0534     bm = p_addr + (tmp/BITS_PER_LONG);
0535     if (mask) {
0536         /* If mask != 0, we are not exactly aligned, so bm now points
0537          * to the long containing the last bit.
0538          * If mask == 0, bm already points to the word immediately
0539          * after the last (long word aligned) bit. */
0540         *bm |= ~mask;
0541         bm++;
0542     }
0543 
0544     if (BITS_PER_LONG == 32 && ((bm - p_addr) & 1) == 1) {
0545         /* on a 32bit arch, we may need to zero out
0546          * a padding long to align with a 64bit remote */
0547         *bm = ~0UL;
0548     }
0549     bm_unmap(p_addr);
0550 }
0551 
0552 /* you better not modify the bitmap while this is running,
0553  * or its results will be stale */
0554 static unsigned long bm_count_bits(struct drbd_bitmap *b)
0555 {
0556     unsigned long *p_addr;
0557     unsigned long bits = 0;
0558     unsigned long mask = (1UL << (b->bm_bits & BITS_PER_LONG_MASK)) -1;
0559     int idx, last_word;
0560 
0561     /* all but last page */
0562     for (idx = 0; idx < b->bm_number_of_pages - 1; idx++) {
0563         p_addr = __bm_map_pidx(b, idx);
0564         bits += bitmap_weight(p_addr, BITS_PER_PAGE);
0565         __bm_unmap(p_addr);
0566         cond_resched();
0567     }
0568     /* last (or only) page */
0569     last_word = ((b->bm_bits - 1) & BITS_PER_PAGE_MASK) >> LN2_BPL;
0570     p_addr = __bm_map_pidx(b, idx);
0571     bits += bitmap_weight(p_addr, last_word * BITS_PER_LONG);
0572     p_addr[last_word] &= cpu_to_lel(mask);
0573     bits += hweight_long(p_addr[last_word]);
0574     /* 32bit arch, may have an unused padding long */
0575     if (BITS_PER_LONG == 32 && (last_word & 1) == 0)
0576         p_addr[last_word+1] = 0;
0577     __bm_unmap(p_addr);
0578     return bits;
0579 }
0580 
0581 /* offset and len in long words.*/
0582 static void bm_memset(struct drbd_bitmap *b, size_t offset, int c, size_t len)
0583 {
0584     unsigned long *p_addr, *bm;
0585     unsigned int idx;
0586     size_t do_now, end;
0587 
0588     end = offset + len;
0589 
0590     if (end > b->bm_words) {
0591         pr_alert("bm_memset end > bm_words\n");
0592         return;
0593     }
0594 
0595     while (offset < end) {
0596         do_now = min_t(size_t, ALIGN(offset + 1, LWPP), end) - offset;
0597         idx = bm_word_to_page_idx(b, offset);
0598         p_addr = bm_map_pidx(b, idx);
0599         bm = p_addr + MLPP(offset);
0600         if (bm+do_now > p_addr + LWPP) {
0601             pr_alert("BUG BUG BUG! p_addr:%p bm:%p do_now:%d\n",
0602                    p_addr, bm, (int)do_now);
0603         } else
0604             memset(bm, c, do_now * sizeof(long));
0605         bm_unmap(p_addr);
0606         bm_set_page_need_writeout(b->bm_pages[idx]);
0607         offset += do_now;
0608     }
0609 }
0610 
0611 /* For the layout, see comment above drbd_md_set_sector_offsets(). */
0612 static u64 drbd_md_on_disk_bits(struct drbd_backing_dev *ldev)
0613 {
0614     u64 bitmap_sectors;
0615     if (ldev->md.al_offset == 8)
0616         bitmap_sectors = ldev->md.md_size_sect - ldev->md.bm_offset;
0617     else
0618         bitmap_sectors = ldev->md.al_offset - ldev->md.bm_offset;
0619     return bitmap_sectors << (9 + 3);
0620 }
0621 
0622 /*
0623  * make sure the bitmap has enough room for the attached storage,
0624  * if necessary, resize.
0625  * called whenever we may have changed the device size.
0626  * returns -ENOMEM if we could not allocate enough memory, 0 on success.
0627  * In case this is actually a resize, we copy the old bitmap into the new one.
0628  * Otherwise, the bitmap is initialized to all bits set.
0629  */
0630 int drbd_bm_resize(struct drbd_device *device, sector_t capacity, int set_new_bits)
0631 {
0632     struct drbd_bitmap *b = device->bitmap;
0633     unsigned long bits, words, owords, obits;
0634     unsigned long want, have, onpages; /* number of pages */
0635     struct page **npages, **opages = NULL;
0636     int err = 0;
0637     bool growing;
0638 
0639     if (!expect(b))
0640         return -ENOMEM;
0641 
0642     drbd_bm_lock(device, "resize", BM_LOCKED_MASK);
0643 
0644     drbd_info(device, "drbd_bm_resize called with capacity == %llu\n",
0645             (unsigned long long)capacity);
0646 
0647     if (capacity == b->bm_dev_capacity)
0648         goto out;
0649 
0650     if (capacity == 0) {
0651         spin_lock_irq(&b->bm_lock);
0652         opages = b->bm_pages;
0653         onpages = b->bm_number_of_pages;
0654         owords = b->bm_words;
0655         b->bm_pages = NULL;
0656         b->bm_number_of_pages =
0657         b->bm_set   =
0658         b->bm_bits  =
0659         b->bm_words =
0660         b->bm_dev_capacity = 0;
0661         spin_unlock_irq(&b->bm_lock);
0662         bm_free_pages(opages, onpages);
0663         bm_vk_free(opages);
0664         goto out;
0665     }
0666     bits  = BM_SECT_TO_BIT(ALIGN(capacity, BM_SECT_PER_BIT));
0667 
0668     /* if we would use
0669        words = ALIGN(bits,BITS_PER_LONG) >> LN2_BPL;
0670        a 32bit host could present the wrong number of words
0671        to a 64bit host.
0672     */
0673     words = ALIGN(bits, 64) >> LN2_BPL;
0674 
0675     if (get_ldev(device)) {
0676         u64 bits_on_disk = drbd_md_on_disk_bits(device->ldev);
0677         put_ldev(device);
0678         if (bits > bits_on_disk) {
0679             drbd_info(device, "bits = %lu\n", bits);
0680             drbd_info(device, "bits_on_disk = %llu\n", bits_on_disk);
0681             err = -ENOSPC;
0682             goto out;
0683         }
0684     }
0685 
0686     want = PFN_UP(words*sizeof(long));
0687     have = b->bm_number_of_pages;
0688     if (want == have) {
0689         D_ASSERT(device, b->bm_pages != NULL);
0690         npages = b->bm_pages;
0691     } else {
0692         if (drbd_insert_fault(device, DRBD_FAULT_BM_ALLOC))
0693             npages = NULL;
0694         else
0695             npages = bm_realloc_pages(b, want);
0696     }
0697 
0698     if (!npages) {
0699         err = -ENOMEM;
0700         goto out;
0701     }
0702 
0703     spin_lock_irq(&b->bm_lock);
0704     opages = b->bm_pages;
0705     owords = b->bm_words;
0706     obits  = b->bm_bits;
0707 
0708     growing = bits > obits;
0709     if (opages && growing && set_new_bits)
0710         bm_set_surplus(b);
0711 
0712     b->bm_pages = npages;
0713     b->bm_number_of_pages = want;
0714     b->bm_bits  = bits;
0715     b->bm_words = words;
0716     b->bm_dev_capacity = capacity;
0717 
0718     if (growing) {
0719         if (set_new_bits) {
0720             bm_memset(b, owords, 0xff, words-owords);
0721             b->bm_set += bits - obits;
0722         } else
0723             bm_memset(b, owords, 0x00, words-owords);
0724 
0725     }
0726 
0727     if (want < have) {
0728         /* implicit: (opages != NULL) && (opages != npages) */
0729         bm_free_pages(opages + want, have - want);
0730     }
0731 
0732     (void)bm_clear_surplus(b);
0733 
0734     spin_unlock_irq(&b->bm_lock);
0735     if (opages != npages)
0736         bm_vk_free(opages);
0737     if (!growing)
0738         b->bm_set = bm_count_bits(b);
0739     drbd_info(device, "resync bitmap: bits=%lu words=%lu pages=%lu\n", bits, words, want);
0740 
0741  out:
0742     drbd_bm_unlock(device);
0743     return err;
0744 }
0745 
0746 /* inherently racy:
0747  * if not protected by other means, return value may be out of date when
0748  * leaving this function...
0749  * we still need to lock it, since it is important that this returns
0750  * bm_set == 0 precisely.
0751  *
0752  * maybe bm_set should be atomic_t ?
0753  */
0754 unsigned long _drbd_bm_total_weight(struct drbd_device *device)
0755 {
0756     struct drbd_bitmap *b = device->bitmap;
0757     unsigned long s;
0758     unsigned long flags;
0759 
0760     if (!expect(b))
0761         return 0;
0762     if (!expect(b->bm_pages))
0763         return 0;
0764 
0765     spin_lock_irqsave(&b->bm_lock, flags);
0766     s = b->bm_set;
0767     spin_unlock_irqrestore(&b->bm_lock, flags);
0768 
0769     return s;
0770 }
0771 
0772 unsigned long drbd_bm_total_weight(struct drbd_device *device)
0773 {
0774     unsigned long s;
0775     /* if I don't have a disk, I don't know about out-of-sync status */
0776     if (!get_ldev_if_state(device, D_NEGOTIATING))
0777         return 0;
0778     s = _drbd_bm_total_weight(device);
0779     put_ldev(device);
0780     return s;
0781 }
0782 
0783 size_t drbd_bm_words(struct drbd_device *device)
0784 {
0785     struct drbd_bitmap *b = device->bitmap;
0786     if (!expect(b))
0787         return 0;
0788     if (!expect(b->bm_pages))
0789         return 0;
0790 
0791     return b->bm_words;
0792 }
0793 
0794 unsigned long drbd_bm_bits(struct drbd_device *device)
0795 {
0796     struct drbd_bitmap *b = device->bitmap;
0797     if (!expect(b))
0798         return 0;
0799 
0800     return b->bm_bits;
0801 }
0802 
0803 /* merge number words from buffer into the bitmap starting at offset.
0804  * buffer[i] is expected to be little endian unsigned long.
0805  * bitmap must be locked by drbd_bm_lock.
0806  * currently only used from receive_bitmap.
0807  */
0808 void drbd_bm_merge_lel(struct drbd_device *device, size_t offset, size_t number,
0809             unsigned long *buffer)
0810 {
0811     struct drbd_bitmap *b = device->bitmap;
0812     unsigned long *p_addr, *bm;
0813     unsigned long word, bits;
0814     unsigned int idx;
0815     size_t end, do_now;
0816 
0817     end = offset + number;
0818 
0819     if (!expect(b))
0820         return;
0821     if (!expect(b->bm_pages))
0822         return;
0823     if (number == 0)
0824         return;
0825     WARN_ON(offset >= b->bm_words);
0826     WARN_ON(end    >  b->bm_words);
0827 
0828     spin_lock_irq(&b->bm_lock);
0829     while (offset < end) {
0830         do_now = min_t(size_t, ALIGN(offset+1, LWPP), end) - offset;
0831         idx = bm_word_to_page_idx(b, offset);
0832         p_addr = bm_map_pidx(b, idx);
0833         bm = p_addr + MLPP(offset);
0834         offset += do_now;
0835         while (do_now--) {
0836             bits = hweight_long(*bm);
0837             word = *bm | *buffer++;
0838             *bm++ = word;
0839             b->bm_set += hweight_long(word) - bits;
0840         }
0841         bm_unmap(p_addr);
0842         bm_set_page_need_writeout(b->bm_pages[idx]);
0843     }
0844     /* with 32bit <-> 64bit cross-platform connect
0845      * this is only correct for current usage,
0846      * where we _know_ that we are 64 bit aligned,
0847      * and know that this function is used in this way, too...
0848      */
0849     if (end == b->bm_words)
0850         b->bm_set -= bm_clear_surplus(b);
0851     spin_unlock_irq(&b->bm_lock);
0852 }
0853 
0854 /* copy number words from the bitmap starting at offset into the buffer.
0855  * buffer[i] will be little endian unsigned long.
0856  */
0857 void drbd_bm_get_lel(struct drbd_device *device, size_t offset, size_t number,
0858              unsigned long *buffer)
0859 {
0860     struct drbd_bitmap *b = device->bitmap;
0861     unsigned long *p_addr, *bm;
0862     size_t end, do_now;
0863 
0864     end = offset + number;
0865 
0866     if (!expect(b))
0867         return;
0868     if (!expect(b->bm_pages))
0869         return;
0870 
0871     spin_lock_irq(&b->bm_lock);
0872     if ((offset >= b->bm_words) ||
0873         (end    >  b->bm_words) ||
0874         (number <= 0))
0875         drbd_err(device, "offset=%lu number=%lu bm_words=%lu\n",
0876             (unsigned long) offset,
0877             (unsigned long) number,
0878             (unsigned long) b->bm_words);
0879     else {
0880         while (offset < end) {
0881             do_now = min_t(size_t, ALIGN(offset+1, LWPP), end) - offset;
0882             p_addr = bm_map_pidx(b, bm_word_to_page_idx(b, offset));
0883             bm = p_addr + MLPP(offset);
0884             offset += do_now;
0885             while (do_now--)
0886                 *buffer++ = *bm++;
0887             bm_unmap(p_addr);
0888         }
0889     }
0890     spin_unlock_irq(&b->bm_lock);
0891 }
0892 
0893 /* set all bits in the bitmap */
0894 void drbd_bm_set_all(struct drbd_device *device)
0895 {
0896     struct drbd_bitmap *b = device->bitmap;
0897     if (!expect(b))
0898         return;
0899     if (!expect(b->bm_pages))
0900         return;
0901 
0902     spin_lock_irq(&b->bm_lock);
0903     bm_memset(b, 0, 0xff, b->bm_words);
0904     (void)bm_clear_surplus(b);
0905     b->bm_set = b->bm_bits;
0906     spin_unlock_irq(&b->bm_lock);
0907 }
0908 
0909 /* clear all bits in the bitmap */
0910 void drbd_bm_clear_all(struct drbd_device *device)
0911 {
0912     struct drbd_bitmap *b = device->bitmap;
0913     if (!expect(b))
0914         return;
0915     if (!expect(b->bm_pages))
0916         return;
0917 
0918     spin_lock_irq(&b->bm_lock);
0919     bm_memset(b, 0, 0, b->bm_words);
0920     b->bm_set = 0;
0921     spin_unlock_irq(&b->bm_lock);
0922 }
0923 
0924 static void drbd_bm_aio_ctx_destroy(struct kref *kref)
0925 {
0926     struct drbd_bm_aio_ctx *ctx = container_of(kref, struct drbd_bm_aio_ctx, kref);
0927     unsigned long flags;
0928 
0929     spin_lock_irqsave(&ctx->device->resource->req_lock, flags);
0930     list_del(&ctx->list);
0931     spin_unlock_irqrestore(&ctx->device->resource->req_lock, flags);
0932     put_ldev(ctx->device);
0933     kfree(ctx);
0934 }
0935 
0936 /* bv_page may be a copy, or may be the original */
0937 static void drbd_bm_endio(struct bio *bio)
0938 {
0939     struct drbd_bm_aio_ctx *ctx = bio->bi_private;
0940     struct drbd_device *device = ctx->device;
0941     struct drbd_bitmap *b = device->bitmap;
0942     unsigned int idx = bm_page_to_idx(bio_first_page_all(bio));
0943 
0944     if ((ctx->flags & BM_AIO_COPY_PAGES) == 0 &&
0945         !bm_test_page_unchanged(b->bm_pages[idx]))
0946         drbd_warn(device, "bitmap page idx %u changed during IO!\n", idx);
0947 
0948     if (bio->bi_status) {
0949         /* ctx error will hold the completed-last non-zero error code,
0950          * in case error codes differ. */
0951         ctx->error = blk_status_to_errno(bio->bi_status);
0952         bm_set_page_io_err(b->bm_pages[idx]);
0953         /* Not identical to on disk version of it.
0954          * Is BM_PAGE_IO_ERROR enough? */
0955         if (__ratelimit(&drbd_ratelimit_state))
0956             drbd_err(device, "IO ERROR %d on bitmap page idx %u\n",
0957                     bio->bi_status, idx);
0958     } else {
0959         bm_clear_page_io_err(b->bm_pages[idx]);
0960         dynamic_drbd_dbg(device, "bitmap page idx %u completed\n", idx);
0961     }
0962 
0963     bm_page_unlock_io(device, idx);
0964 
0965     if (ctx->flags & BM_AIO_COPY_PAGES)
0966         mempool_free(bio->bi_io_vec[0].bv_page, &drbd_md_io_page_pool);
0967 
0968     bio_put(bio);
0969 
0970     if (atomic_dec_and_test(&ctx->in_flight)) {
0971         ctx->done = 1;
0972         wake_up(&device->misc_wait);
0973         kref_put(&ctx->kref, &drbd_bm_aio_ctx_destroy);
0974     }
0975 }
0976 
0977 /* For the layout, see comment above drbd_md_set_sector_offsets(). */
0978 static inline sector_t drbd_md_last_bitmap_sector(struct drbd_backing_dev *bdev)
0979 {
0980     switch (bdev->md.meta_dev_idx) {
0981     case DRBD_MD_INDEX_INTERNAL:
0982     case DRBD_MD_INDEX_FLEX_INT:
0983         return bdev->md.md_offset + bdev->md.al_offset -1;
0984     case DRBD_MD_INDEX_FLEX_EXT:
0985     default:
0986         return bdev->md.md_offset + bdev->md.md_size_sect -1;
0987     }
0988 }
0989 
0990 static void bm_page_io_async(struct drbd_bm_aio_ctx *ctx, int page_nr) __must_hold(local)
0991 {
0992     struct drbd_device *device = ctx->device;
0993     enum req_op op = ctx->flags & BM_AIO_READ ? REQ_OP_READ : REQ_OP_WRITE;
0994     struct drbd_bitmap *b = device->bitmap;
0995     struct bio *bio;
0996     struct page *page;
0997     sector_t last_bm_sect;
0998     sector_t first_bm_sect;
0999     sector_t on_disk_sector;
1000     unsigned int len;
1001 
1002     first_bm_sect = device->ldev->md.md_offset + device->ldev->md.bm_offset;
1003     on_disk_sector = first_bm_sect + (((sector_t)page_nr) << (PAGE_SHIFT-SECTOR_SHIFT));
1004 
1005     /* this might happen with very small
1006      * flexible external meta data device,
1007      * or with PAGE_SIZE > 4k */
1008     last_bm_sect = drbd_md_last_bitmap_sector(device->ldev);
1009     if (first_bm_sect <= on_disk_sector && last_bm_sect >= on_disk_sector) {
1010         sector_t len_sect = last_bm_sect - on_disk_sector + 1;
1011         if (len_sect < PAGE_SIZE/SECTOR_SIZE)
1012             len = (unsigned int)len_sect*SECTOR_SIZE;
1013         else
1014             len = PAGE_SIZE;
1015     } else {
1016         if (__ratelimit(&drbd_ratelimit_state)) {
1017             drbd_err(device, "Invalid offset during on-disk bitmap access: "
1018                  "page idx %u, sector %llu\n", page_nr, on_disk_sector);
1019         }
1020         ctx->error = -EIO;
1021         bm_set_page_io_err(b->bm_pages[page_nr]);
1022         if (atomic_dec_and_test(&ctx->in_flight)) {
1023             ctx->done = 1;
1024             wake_up(&device->misc_wait);
1025             kref_put(&ctx->kref, &drbd_bm_aio_ctx_destroy);
1026         }
1027         return;
1028     }
1029 
1030     /* serialize IO on this page */
1031     bm_page_lock_io(device, page_nr);
1032     /* before memcpy and submit,
1033      * so it can be redirtied any time */
1034     bm_set_page_unchanged(b->bm_pages[page_nr]);
1035 
1036     if (ctx->flags & BM_AIO_COPY_PAGES) {
1037         page = mempool_alloc(&drbd_md_io_page_pool,
1038                 GFP_NOIO | __GFP_HIGHMEM);
1039         copy_highpage(page, b->bm_pages[page_nr]);
1040         bm_store_page_idx(page, page_nr);
1041     } else
1042         page = b->bm_pages[page_nr];
1043     bio = bio_alloc_bioset(device->ldev->md_bdev, 1, op, GFP_NOIO,
1044             &drbd_md_io_bio_set);
1045     bio->bi_iter.bi_sector = on_disk_sector;
1046     /* bio_add_page of a single page to an empty bio will always succeed,
1047      * according to api.  Do we want to assert that? */
1048     bio_add_page(bio, page, len, 0);
1049     bio->bi_private = ctx;
1050     bio->bi_end_io = drbd_bm_endio;
1051 
1052     if (drbd_insert_fault(device, (op == REQ_OP_WRITE) ? DRBD_FAULT_MD_WR : DRBD_FAULT_MD_RD)) {
1053         bio_io_error(bio);
1054     } else {
1055         submit_bio(bio);
1056         /* this should not count as user activity and cause the
1057          * resync to throttle -- see drbd_rs_should_slow_down(). */
1058         atomic_add(len >> 9, &device->rs_sect_ev);
1059     }
1060 }
1061 
1062 /*
1063  * bm_rw: read/write the whole bitmap from/to its on disk location.
1064  */
1065 static int bm_rw(struct drbd_device *device, const unsigned int flags, unsigned lazy_writeout_upper_idx) __must_hold(local)
1066 {
1067     struct drbd_bm_aio_ctx *ctx;
1068     struct drbd_bitmap *b = device->bitmap;
1069     unsigned int num_pages, i, count = 0;
1070     unsigned long now;
1071     char ppb[10];
1072     int err = 0;
1073 
1074     /*
1075      * We are protected against bitmap disappearing/resizing by holding an
1076      * ldev reference (caller must have called get_ldev()).
1077      * For read/write, we are protected against changes to the bitmap by
1078      * the bitmap lock (see drbd_bitmap_io).
1079      * For lazy writeout, we don't care for ongoing changes to the bitmap,
1080      * as we submit copies of pages anyways.
1081      */
1082 
1083     ctx = kmalloc(sizeof(struct drbd_bm_aio_ctx), GFP_NOIO);
1084     if (!ctx)
1085         return -ENOMEM;
1086 
1087     *ctx = (struct drbd_bm_aio_ctx) {
1088         .device = device,
1089         .start_jif = jiffies,
1090         .in_flight = ATOMIC_INIT(1),
1091         .done = 0,
1092         .flags = flags,
1093         .error = 0,
1094         .kref = KREF_INIT(2),
1095     };
1096 
1097     if (!get_ldev_if_state(device, D_ATTACHING)) {  /* put is in drbd_bm_aio_ctx_destroy() */
1098         drbd_err(device, "ASSERT FAILED: get_ldev_if_state() == 1 in bm_rw()\n");
1099         kfree(ctx);
1100         return -ENODEV;
1101     }
1102     /* Here D_ATTACHING is sufficient since drbd_bm_read() is called only from
1103        drbd_adm_attach(), after device->ldev was assigned. */
1104 
1105     if (0 == (ctx->flags & ~BM_AIO_READ))
1106         WARN_ON(!(BM_LOCKED_MASK & b->bm_flags));
1107 
1108     spin_lock_irq(&device->resource->req_lock);
1109     list_add_tail(&ctx->list, &device->pending_bitmap_io);
1110     spin_unlock_irq(&device->resource->req_lock);
1111 
1112     num_pages = b->bm_number_of_pages;
1113 
1114     now = jiffies;
1115 
1116     /* let the layers below us try to merge these bios... */
1117 
1118     if (flags & BM_AIO_READ) {
1119         for (i = 0; i < num_pages; i++) {
1120             atomic_inc(&ctx->in_flight);
1121             bm_page_io_async(ctx, i);
1122             ++count;
1123             cond_resched();
1124         }
1125     } else if (flags & BM_AIO_WRITE_HINTED) {
1126         /* ASSERT: BM_AIO_WRITE_ALL_PAGES is not set. */
1127         unsigned int hint;
1128         for (hint = 0; hint < b->n_bitmap_hints; hint++) {
1129             i = b->al_bitmap_hints[hint];
1130             if (i >= num_pages) /* == -1U: no hint here. */
1131                 continue;
1132             /* Several AL-extents may point to the same page. */
1133             if (!test_and_clear_bit(BM_PAGE_HINT_WRITEOUT,
1134                 &page_private(b->bm_pages[i])))
1135                 continue;
1136             /* Has it even changed? */
1137             if (bm_test_page_unchanged(b->bm_pages[i]))
1138                 continue;
1139             atomic_inc(&ctx->in_flight);
1140             bm_page_io_async(ctx, i);
1141             ++count;
1142         }
1143     } else {
1144         for (i = 0; i < num_pages; i++) {
1145             /* ignore completely unchanged pages */
1146             if (lazy_writeout_upper_idx && i == lazy_writeout_upper_idx)
1147                 break;
1148             if (!(flags & BM_AIO_WRITE_ALL_PAGES) &&
1149                 bm_test_page_unchanged(b->bm_pages[i])) {
1150                 dynamic_drbd_dbg(device, "skipped bm write for idx %u\n", i);
1151                 continue;
1152             }
1153             /* during lazy writeout,
1154              * ignore those pages not marked for lazy writeout. */
1155             if (lazy_writeout_upper_idx &&
1156                 !bm_test_page_lazy_writeout(b->bm_pages[i])) {
1157                 dynamic_drbd_dbg(device, "skipped bm lazy write for idx %u\n", i);
1158                 continue;
1159             }
1160             atomic_inc(&ctx->in_flight);
1161             bm_page_io_async(ctx, i);
1162             ++count;
1163             cond_resched();
1164         }
1165     }
1166 
1167     /*
1168      * We initialize ctx->in_flight to one to make sure drbd_bm_endio
1169      * will not set ctx->done early, and decrement / test it here.  If there
1170      * are still some bios in flight, we need to wait for them here.
1171      * If all IO is done already (or nothing had been submitted), there is
1172      * no need to wait.  Still, we need to put the kref associated with the
1173      * "in_flight reached zero, all done" event.
1174      */
1175     if (!atomic_dec_and_test(&ctx->in_flight))
1176         wait_until_done_or_force_detached(device, device->ldev, &ctx->done);
1177     else
1178         kref_put(&ctx->kref, &drbd_bm_aio_ctx_destroy);
1179 
1180     /* summary for global bitmap IO */
1181     if (flags == 0) {
1182         unsigned int ms = jiffies_to_msecs(jiffies - now);
1183         if (ms > 5) {
1184             drbd_info(device, "bitmap %s of %u pages took %u ms\n",
1185                  (flags & BM_AIO_READ) ? "READ" : "WRITE",
1186                  count, ms);
1187         }
1188     }
1189 
1190     if (ctx->error) {
1191         drbd_alert(device, "we had at least one MD IO ERROR during bitmap IO\n");
1192         drbd_chk_io_error(device, 1, DRBD_META_IO_ERROR);
1193         err = -EIO; /* ctx->error ? */
1194     }
1195 
1196     if (atomic_read(&ctx->in_flight))
1197         err = -EIO; /* Disk timeout/force-detach during IO... */
1198 
1199     now = jiffies;
1200     if (flags & BM_AIO_READ) {
1201         b->bm_set = bm_count_bits(b);
1202         drbd_info(device, "recounting of set bits took additional %lu jiffies\n",
1203              jiffies - now);
1204     }
1205     now = b->bm_set;
1206 
1207     if ((flags & ~BM_AIO_READ) == 0)
1208         drbd_info(device, "%s (%lu bits) marked out-of-sync by on disk bit-map.\n",
1209              ppsize(ppb, now << (BM_BLOCK_SHIFT-10)), now);
1210 
1211     kref_put(&ctx->kref, &drbd_bm_aio_ctx_destroy);
1212     return err;
1213 }
1214 
1215 /**
1216  * drbd_bm_read() - Read the whole bitmap from its on disk location.
1217  * @device: DRBD device.
1218  */
1219 int drbd_bm_read(struct drbd_device *device) __must_hold(local)
1220 {
1221     return bm_rw(device, BM_AIO_READ, 0);
1222 }
1223 
1224 /**
1225  * drbd_bm_write() - Write the whole bitmap to its on disk location.
1226  * @device: DRBD device.
1227  *
1228  * Will only write pages that have changed since last IO.
1229  */
1230 int drbd_bm_write(struct drbd_device *device) __must_hold(local)
1231 {
1232     return bm_rw(device, 0, 0);
1233 }
1234 
1235 /**
1236  * drbd_bm_write_all() - Write the whole bitmap to its on disk location.
1237  * @device: DRBD device.
1238  *
1239  * Will write all pages.
1240  */
1241 int drbd_bm_write_all(struct drbd_device *device) __must_hold(local)
1242 {
1243     return bm_rw(device, BM_AIO_WRITE_ALL_PAGES, 0);
1244 }
1245 
1246 /**
1247  * drbd_bm_write_lazy() - Write bitmap pages 0 to @upper_idx-1, if they have changed.
1248  * @device: DRBD device.
1249  * @upper_idx:  0: write all changed pages; +ve: page index to stop scanning for changed pages
1250  */
1251 int drbd_bm_write_lazy(struct drbd_device *device, unsigned upper_idx) __must_hold(local)
1252 {
1253     return bm_rw(device, BM_AIO_COPY_PAGES, upper_idx);
1254 }
1255 
1256 /**
1257  * drbd_bm_write_copy_pages() - Write the whole bitmap to its on disk location.
1258  * @device: DRBD device.
1259  *
1260  * Will only write pages that have changed since last IO.
1261  * In contrast to drbd_bm_write(), this will copy the bitmap pages
1262  * to temporary writeout pages. It is intended to trigger a full write-out
1263  * while still allowing the bitmap to change, for example if a resync or online
1264  * verify is aborted due to a failed peer disk, while local IO continues, or
1265  * pending resync acks are still being processed.
1266  */
1267 int drbd_bm_write_copy_pages(struct drbd_device *device) __must_hold(local)
1268 {
1269     return bm_rw(device, BM_AIO_COPY_PAGES, 0);
1270 }
1271 
1272 /**
1273  * drbd_bm_write_hinted() - Write bitmap pages with "hint" marks, if they have changed.
1274  * @device: DRBD device.
1275  */
1276 int drbd_bm_write_hinted(struct drbd_device *device) __must_hold(local)
1277 {
1278     return bm_rw(device, BM_AIO_WRITE_HINTED | BM_AIO_COPY_PAGES, 0);
1279 }
1280 
1281 /* NOTE
1282  * find_first_bit returns int, we return unsigned long.
1283  * For this to work on 32bit arch with bitnumbers > (1<<32),
1284  * we'd need to return u64, and get a whole lot of other places
1285  * fixed where we still use unsigned long.
1286  *
1287  * this returns a bit number, NOT a sector!
1288  */
1289 static unsigned long __bm_find_next(struct drbd_device *device, unsigned long bm_fo,
1290     const int find_zero_bit)
1291 {
1292     struct drbd_bitmap *b = device->bitmap;
1293     unsigned long *p_addr;
1294     unsigned long bit_offset;
1295     unsigned i;
1296 
1297 
1298     if (bm_fo > b->bm_bits) {
1299         drbd_err(device, "bm_fo=%lu bm_bits=%lu\n", bm_fo, b->bm_bits);
1300         bm_fo = DRBD_END_OF_BITMAP;
1301     } else {
1302         while (bm_fo < b->bm_bits) {
1303             /* bit offset of the first bit in the page */
1304             bit_offset = bm_fo & ~BITS_PER_PAGE_MASK;
1305             p_addr = __bm_map_pidx(b, bm_bit_to_page_idx(b, bm_fo));
1306 
1307             if (find_zero_bit)
1308                 i = find_next_zero_bit_le(p_addr,
1309                         PAGE_SIZE*8, bm_fo & BITS_PER_PAGE_MASK);
1310             else
1311                 i = find_next_bit_le(p_addr,
1312                         PAGE_SIZE*8, bm_fo & BITS_PER_PAGE_MASK);
1313 
1314             __bm_unmap(p_addr);
1315             if (i < PAGE_SIZE*8) {
1316                 bm_fo = bit_offset + i;
1317                 if (bm_fo >= b->bm_bits)
1318                     break;
1319                 goto found;
1320             }
1321             bm_fo = bit_offset + PAGE_SIZE*8;
1322         }
1323         bm_fo = DRBD_END_OF_BITMAP;
1324     }
1325  found:
1326     return bm_fo;
1327 }
1328 
1329 static unsigned long bm_find_next(struct drbd_device *device,
1330     unsigned long bm_fo, const int find_zero_bit)
1331 {
1332     struct drbd_bitmap *b = device->bitmap;
1333     unsigned long i = DRBD_END_OF_BITMAP;
1334 
1335     if (!expect(b))
1336         return i;
1337     if (!expect(b->bm_pages))
1338         return i;
1339 
1340     spin_lock_irq(&b->bm_lock);
1341     if (BM_DONT_TEST & b->bm_flags)
1342         bm_print_lock_info(device);
1343 
1344     i = __bm_find_next(device, bm_fo, find_zero_bit);
1345 
1346     spin_unlock_irq(&b->bm_lock);
1347     return i;
1348 }
1349 
1350 unsigned long drbd_bm_find_next(struct drbd_device *device, unsigned long bm_fo)
1351 {
1352     return bm_find_next(device, bm_fo, 0);
1353 }
1354 
1355 #if 0
1356 /* not yet needed for anything. */
1357 unsigned long drbd_bm_find_next_zero(struct drbd_device *device, unsigned long bm_fo)
1358 {
1359     return bm_find_next(device, bm_fo, 1);
1360 }
1361 #endif
1362 
1363 /* does not spin_lock_irqsave.
1364  * you must take drbd_bm_lock() first */
1365 unsigned long _drbd_bm_find_next(struct drbd_device *device, unsigned long bm_fo)
1366 {
1367     /* WARN_ON(!(BM_DONT_SET & device->b->bm_flags)); */
1368     return __bm_find_next(device, bm_fo, 0);
1369 }
1370 
1371 unsigned long _drbd_bm_find_next_zero(struct drbd_device *device, unsigned long bm_fo)
1372 {
1373     /* WARN_ON(!(BM_DONT_SET & device->b->bm_flags)); */
1374     return __bm_find_next(device, bm_fo, 1);
1375 }
1376 
1377 /* returns number of bits actually changed.
1378  * for val != 0, we change 0 -> 1, return code positive
1379  * for val == 0, we change 1 -> 0, return code negative
1380  * wants bitnr, not sector.
1381  * expected to be called for only a few bits (e - s about BITS_PER_LONG).
1382  * Must hold bitmap lock already. */
1383 static int __bm_change_bits_to(struct drbd_device *device, const unsigned long s,
1384     unsigned long e, int val)
1385 {
1386     struct drbd_bitmap *b = device->bitmap;
1387     unsigned long *p_addr = NULL;
1388     unsigned long bitnr;
1389     unsigned int last_page_nr = -1U;
1390     int c = 0;
1391     int changed_total = 0;
1392 
1393     if (e >= b->bm_bits) {
1394         drbd_err(device, "ASSERT FAILED: bit_s=%lu bit_e=%lu bm_bits=%lu\n",
1395                 s, e, b->bm_bits);
1396         e = b->bm_bits ? b->bm_bits -1 : 0;
1397     }
1398     for (bitnr = s; bitnr <= e; bitnr++) {
1399         unsigned int page_nr = bm_bit_to_page_idx(b, bitnr);
1400         if (page_nr != last_page_nr) {
1401             if (p_addr)
1402                 __bm_unmap(p_addr);
1403             if (c < 0)
1404                 bm_set_page_lazy_writeout(b->bm_pages[last_page_nr]);
1405             else if (c > 0)
1406                 bm_set_page_need_writeout(b->bm_pages[last_page_nr]);
1407             changed_total += c;
1408             c = 0;
1409             p_addr = __bm_map_pidx(b, page_nr);
1410             last_page_nr = page_nr;
1411         }
1412         if (val)
1413             c += (0 == __test_and_set_bit_le(bitnr & BITS_PER_PAGE_MASK, p_addr));
1414         else
1415             c -= (0 != __test_and_clear_bit_le(bitnr & BITS_PER_PAGE_MASK, p_addr));
1416     }
1417     if (p_addr)
1418         __bm_unmap(p_addr);
1419     if (c < 0)
1420         bm_set_page_lazy_writeout(b->bm_pages[last_page_nr]);
1421     else if (c > 0)
1422         bm_set_page_need_writeout(b->bm_pages[last_page_nr]);
1423     changed_total += c;
1424     b->bm_set += changed_total;
1425     return changed_total;
1426 }
1427 
1428 /* returns number of bits actually changed.
1429  * for val != 0, we change 0 -> 1, return code positive
1430  * for val == 0, we change 1 -> 0, return code negative
1431  * wants bitnr, not sector */
1432 static int bm_change_bits_to(struct drbd_device *device, const unsigned long s,
1433     const unsigned long e, int val)
1434 {
1435     unsigned long flags;
1436     struct drbd_bitmap *b = device->bitmap;
1437     int c = 0;
1438 
1439     if (!expect(b))
1440         return 1;
1441     if (!expect(b->bm_pages))
1442         return 0;
1443 
1444     spin_lock_irqsave(&b->bm_lock, flags);
1445     if ((val ? BM_DONT_SET : BM_DONT_CLEAR) & b->bm_flags)
1446         bm_print_lock_info(device);
1447 
1448     c = __bm_change_bits_to(device, s, e, val);
1449 
1450     spin_unlock_irqrestore(&b->bm_lock, flags);
1451     return c;
1452 }
1453 
1454 /* returns number of bits changed 0 -> 1 */
1455 int drbd_bm_set_bits(struct drbd_device *device, const unsigned long s, const unsigned long e)
1456 {
1457     return bm_change_bits_to(device, s, e, 1);
1458 }
1459 
1460 /* returns number of bits changed 1 -> 0 */
1461 int drbd_bm_clear_bits(struct drbd_device *device, const unsigned long s, const unsigned long e)
1462 {
1463     return -bm_change_bits_to(device, s, e, 0);
1464 }
1465 
1466 /* sets all bits in full words,
1467  * from first_word up to, but not including, last_word */
1468 static inline void bm_set_full_words_within_one_page(struct drbd_bitmap *b,
1469         int page_nr, int first_word, int last_word)
1470 {
1471     int i;
1472     int bits;
1473     int changed = 0;
1474     unsigned long *paddr = kmap_atomic(b->bm_pages[page_nr]);
1475 
1476     /* I think it is more cache line friendly to hweight_long then set to ~0UL,
1477      * than to first bitmap_weight() all words, then bitmap_fill() all words */
1478     for (i = first_word; i < last_word; i++) {
1479         bits = hweight_long(paddr[i]);
1480         paddr[i] = ~0UL;
1481         changed += BITS_PER_LONG - bits;
1482     }
1483     kunmap_atomic(paddr);
1484     if (changed) {
1485         /* We only need lazy writeout, the information is still in the
1486          * remote bitmap as well, and is reconstructed during the next
1487          * bitmap exchange, if lost locally due to a crash. */
1488         bm_set_page_lazy_writeout(b->bm_pages[page_nr]);
1489         b->bm_set += changed;
1490     }
1491 }
1492 
1493 /* Same thing as drbd_bm_set_bits,
1494  * but more efficient for a large bit range.
1495  * You must first drbd_bm_lock().
1496  * Can be called to set the whole bitmap in one go.
1497  * Sets bits from s to e _inclusive_. */
1498 void _drbd_bm_set_bits(struct drbd_device *device, const unsigned long s, const unsigned long e)
1499 {
1500     /* First set_bit from the first bit (s)
1501      * up to the next long boundary (sl),
1502      * then assign full words up to the last long boundary (el),
1503      * then set_bit up to and including the last bit (e).
1504      *
1505      * Do not use memset, because we must account for changes,
1506      * so we need to loop over the words with hweight() anyways.
1507      */
1508     struct drbd_bitmap *b = device->bitmap;
1509     unsigned long sl = ALIGN(s,BITS_PER_LONG);
1510     unsigned long el = (e+1) & ~((unsigned long)BITS_PER_LONG-1);
1511     int first_page;
1512     int last_page;
1513     int page_nr;
1514     int first_word;
1515     int last_word;
1516 
1517     if (e - s <= 3*BITS_PER_LONG) {
1518         /* don't bother; el and sl may even be wrong. */
1519         spin_lock_irq(&b->bm_lock);
1520         __bm_change_bits_to(device, s, e, 1);
1521         spin_unlock_irq(&b->bm_lock);
1522         return;
1523     }
1524 
1525     /* difference is large enough that we can trust sl and el */
1526 
1527     spin_lock_irq(&b->bm_lock);
1528 
1529     /* bits filling the current long */
1530     if (sl)
1531         __bm_change_bits_to(device, s, sl-1, 1);
1532 
1533     first_page = sl >> (3 + PAGE_SHIFT);
1534     last_page = el >> (3 + PAGE_SHIFT);
1535 
1536     /* MLPP: modulo longs per page */
1537     /* LWPP: long words per page */
1538     first_word = MLPP(sl >> LN2_BPL);
1539     last_word = LWPP;
1540 
1541     /* first and full pages, unless first page == last page */
1542     for (page_nr = first_page; page_nr < last_page; page_nr++) {
1543         bm_set_full_words_within_one_page(device->bitmap, page_nr, first_word, last_word);
1544         spin_unlock_irq(&b->bm_lock);
1545         cond_resched();
1546         first_word = 0;
1547         spin_lock_irq(&b->bm_lock);
1548     }
1549     /* last page (respectively only page, for first page == last page) */
1550     last_word = MLPP(el >> LN2_BPL);
1551 
1552     /* consider bitmap->bm_bits = 32768, bitmap->bm_number_of_pages = 1. (or multiples).
1553      * ==> e = 32767, el = 32768, last_page = 2,
1554      * and now last_word = 0.
1555      * We do not want to touch last_page in this case,
1556      * as we did not allocate it, it is not present in bitmap->bm_pages.
1557      */
1558     if (last_word)
1559         bm_set_full_words_within_one_page(device->bitmap, last_page, first_word, last_word);
1560 
1561     /* possibly trailing bits.
1562      * example: (e & 63) == 63, el will be e+1.
1563      * if that even was the very last bit,
1564      * it would trigger an assert in __bm_change_bits_to()
1565      */
1566     if (el <= e)
1567         __bm_change_bits_to(device, el, e, 1);
1568     spin_unlock_irq(&b->bm_lock);
1569 }
1570 
1571 /* returns bit state
1572  * wants bitnr, NOT sector.
1573  * inherently racy... area needs to be locked by means of {al,rs}_lru
1574  *  1 ... bit set
1575  *  0 ... bit not set
1576  * -1 ... first out of bounds access, stop testing for bits!
1577  */
1578 int drbd_bm_test_bit(struct drbd_device *device, const unsigned long bitnr)
1579 {
1580     unsigned long flags;
1581     struct drbd_bitmap *b = device->bitmap;
1582     unsigned long *p_addr;
1583     int i;
1584 
1585     if (!expect(b))
1586         return 0;
1587     if (!expect(b->bm_pages))
1588         return 0;
1589 
1590     spin_lock_irqsave(&b->bm_lock, flags);
1591     if (BM_DONT_TEST & b->bm_flags)
1592         bm_print_lock_info(device);
1593     if (bitnr < b->bm_bits) {
1594         p_addr = bm_map_pidx(b, bm_bit_to_page_idx(b, bitnr));
1595         i = test_bit_le(bitnr & BITS_PER_PAGE_MASK, p_addr) ? 1 : 0;
1596         bm_unmap(p_addr);
1597     } else if (bitnr == b->bm_bits) {
1598         i = -1;
1599     } else { /* (bitnr > b->bm_bits) */
1600         drbd_err(device, "bitnr=%lu > bm_bits=%lu\n", bitnr, b->bm_bits);
1601         i = 0;
1602     }
1603 
1604     spin_unlock_irqrestore(&b->bm_lock, flags);
1605     return i;
1606 }
1607 
1608 /* returns number of bits set in the range [s, e] */
1609 int drbd_bm_count_bits(struct drbd_device *device, const unsigned long s, const unsigned long e)
1610 {
1611     unsigned long flags;
1612     struct drbd_bitmap *b = device->bitmap;
1613     unsigned long *p_addr = NULL;
1614     unsigned long bitnr;
1615     unsigned int page_nr = -1U;
1616     int c = 0;
1617 
1618     /* If this is called without a bitmap, that is a bug.  But just to be
1619      * robust in case we screwed up elsewhere, in that case pretend there
1620      * was one dirty bit in the requested area, so we won't try to do a
1621      * local read there (no bitmap probably implies no disk) */
1622     if (!expect(b))
1623         return 1;
1624     if (!expect(b->bm_pages))
1625         return 1;
1626 
1627     spin_lock_irqsave(&b->bm_lock, flags);
1628     if (BM_DONT_TEST & b->bm_flags)
1629         bm_print_lock_info(device);
1630     for (bitnr = s; bitnr <= e; bitnr++) {
1631         unsigned int idx = bm_bit_to_page_idx(b, bitnr);
1632         if (page_nr != idx) {
1633             page_nr = idx;
1634             if (p_addr)
1635                 bm_unmap(p_addr);
1636             p_addr = bm_map_pidx(b, idx);
1637         }
1638         if (expect(bitnr < b->bm_bits))
1639             c += (0 != test_bit_le(bitnr - (page_nr << (PAGE_SHIFT+3)), p_addr));
1640         else
1641             drbd_err(device, "bitnr=%lu bm_bits=%lu\n", bitnr, b->bm_bits);
1642     }
1643     if (p_addr)
1644         bm_unmap(p_addr);
1645     spin_unlock_irqrestore(&b->bm_lock, flags);
1646     return c;
1647 }
1648 
1649 
1650 /* inherently racy...
1651  * return value may be already out-of-date when this function returns.
1652  * but the general usage is that this is only use during a cstate when bits are
1653  * only cleared, not set, and typically only care for the case when the return
1654  * value is zero, or we already "locked" this "bitmap extent" by other means.
1655  *
1656  * enr is bm-extent number, since we chose to name one sector (512 bytes)
1657  * worth of the bitmap a "bitmap extent".
1658  *
1659  * TODO
1660  * I think since we use it like a reference count, we should use the real
1661  * reference count of some bitmap extent element from some lru instead...
1662  *
1663  */
1664 int drbd_bm_e_weight(struct drbd_device *device, unsigned long enr)
1665 {
1666     struct drbd_bitmap *b = device->bitmap;
1667     int count, s, e;
1668     unsigned long flags;
1669     unsigned long *p_addr, *bm;
1670 
1671     if (!expect(b))
1672         return 0;
1673     if (!expect(b->bm_pages))
1674         return 0;
1675 
1676     spin_lock_irqsave(&b->bm_lock, flags);
1677     if (BM_DONT_TEST & b->bm_flags)
1678         bm_print_lock_info(device);
1679 
1680     s = S2W(enr);
1681     e = min((size_t)S2W(enr+1), b->bm_words);
1682     count = 0;
1683     if (s < b->bm_words) {
1684         int n = e-s;
1685         p_addr = bm_map_pidx(b, bm_word_to_page_idx(b, s));
1686         bm = p_addr + MLPP(s);
1687         count += bitmap_weight(bm, n * BITS_PER_LONG);
1688         bm_unmap(p_addr);
1689     } else {
1690         drbd_err(device, "start offset (%d) too large in drbd_bm_e_weight\n", s);
1691     }
1692     spin_unlock_irqrestore(&b->bm_lock, flags);
1693     return count;
1694 }