xref: /linux/drivers/md/md-bitmap.c (revision c532de5a67a70f8533d495f8f2aaa9a0491c3ad0)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * bitmap.c two-level bitmap (C) Peter T. Breuer (ptb@ot.uc3m.es) 2003
4  *
5  * bitmap_create  - sets up the bitmap structure
6  * bitmap_destroy - destroys the bitmap structure
7  *
8  * additions, Copyright (C) 2003-2004, Paul Clements, SteelEye Technology, Inc.:
9  * - added disk storage for bitmap
10  * - changes to allow various bitmap chunk sizes
11  */
12 
13 /*
14  * Still to do:
15  *
16  * flush after percent set rather than just time based. (maybe both).
17  */
18 
19 #include <linux/blkdev.h>
20 #include <linux/module.h>
21 #include <linux/errno.h>
22 #include <linux/slab.h>
23 #include <linux/init.h>
24 #include <linux/timer.h>
25 #include <linux/sched.h>
26 #include <linux/list.h>
27 #include <linux/file.h>
28 #include <linux/mount.h>
29 #include <linux/buffer_head.h>
30 #include <linux/seq_file.h>
31 #include <trace/events/block.h>
32 #include "md.h"
33 #include "md-bitmap.h"
34 
35 #define BITMAP_MAJOR_LO 3
36 /* version 4 insists the bitmap is in little-endian order
37  * with version 3, it is host-endian which is non-portable
38  * Version 5 is currently set only for clustered devices
39  */
40 #define BITMAP_MAJOR_HI 4
41 #define BITMAP_MAJOR_CLUSTERED 5
42 #define	BITMAP_MAJOR_HOSTENDIAN 3
43 
44 /*
45  * in-memory bitmap:
46  *
47  * Use 16 bit block counters to track pending writes to each "chunk".
48  * The 2 high order bits are special-purpose, the first is a flag indicating
49  * whether a resync is needed.  The second is a flag indicating whether a
50  * resync is active.
51  * This means that the counter is actually 14 bits:
52  *
53  * +--------+--------+------------------------------------------------+
54  * | resync | resync |               counter                          |
55  * | needed | active |                                                |
56  * |  (0-1) |  (0-1) |              (0-16383)                         |
57  * +--------+--------+------------------------------------------------+
58  *
59  * The "resync needed" bit is set when:
60  *    a '1' bit is read from storage at startup.
61  *    a write request fails on some drives
62  *    a resync is aborted on a chunk with 'resync active' set
63  * It is cleared (and resync-active set) when a resync starts across all drives
64  * of the chunk.
65  *
66  *
67  * The "resync active" bit is set when:
68  *    a resync is started on all drives, and resync_needed is set.
69  *       resync_needed will be cleared (as long as resync_active wasn't already set).
70  * It is cleared when a resync completes.
71  *
72  * The counter counts pending write requests, plus the on-disk bit.
73  * When the counter is '1' and the resync bits are clear, the on-disk
74  * bit can be cleared as well, thus setting the counter to 0.
75  * When we set a bit, or in the counter (to start a write), if the fields is
76  * 0, we first set the disk bit and set the counter to 1.
77  *
78  * If the counter is 0, the on-disk bit is clear and the stripe is clean
79  * Anything that dirties the stripe pushes the counter to 2 (at least)
80  * and sets the on-disk bit (lazily).
81  * If a periodic sweep find the counter at 2, it is decremented to 1.
82  * If the sweep find the counter at 1, the on-disk bit is cleared and the
83  * counter goes to zero.
84  *
85  * Also, we'll hijack the "map" pointer itself and use it as two 16 bit block
86  * counters as a fallback when "page" memory cannot be allocated:
87  *
88  * Normal case (page memory allocated):
89  *
90  *     page pointer (32-bit)
91  *
92  *     [ ] ------+
93  *               |
94  *               +-------> [   ][   ]..[   ] (4096 byte page == 2048 counters)
95  *                          c1   c2    c2048
96  *
97  * Hijacked case (page memory allocation failed):
98  *
99  *     hijacked page pointer (32-bit)
100  *
101  *     [		  ][		  ] (no page memory allocated)
102  *      counter #1 (16-bit) counter #2 (16-bit)
103  *
104  */
105 
106 #define PAGE_BITS (PAGE_SIZE << 3)
107 #define PAGE_BIT_SHIFT (PAGE_SHIFT + 3)
108 
109 #define NEEDED(x) (((bitmap_counter_t) x) & NEEDED_MASK)
110 #define RESYNC(x) (((bitmap_counter_t) x) & RESYNC_MASK)
111 #define COUNTER(x) (((bitmap_counter_t) x) & COUNTER_MAX)
112 
113 /* how many counters per page? */
114 #define PAGE_COUNTER_RATIO (PAGE_BITS / COUNTER_BITS)
115 /* same, except a shift value for more efficient bitops */
116 #define PAGE_COUNTER_SHIFT (PAGE_BIT_SHIFT - COUNTER_BIT_SHIFT)
117 /* same, except a mask value for more efficient bitops */
118 #define PAGE_COUNTER_MASK  (PAGE_COUNTER_RATIO - 1)
119 
120 #define BITMAP_BLOCK_SHIFT 9
121 
122 /*
123  * bitmap structures:
124  */
125 
126 /* the in-memory bitmap is represented by bitmap_pages */
127 struct bitmap_page {
128 	/*
129 	 * map points to the actual memory page
130 	 */
131 	char *map;
132 	/*
133 	 * in emergencies (when map cannot be alloced), hijack the map
134 	 * pointer and use it as two counters itself
135 	 */
136 	unsigned int hijacked:1;
137 	/*
138 	 * If any counter in this page is '1' or '2' - and so could be
139 	 * cleared then that page is marked as 'pending'
140 	 */
141 	unsigned int pending:1;
142 	/*
143 	 * count of dirty bits on the page
144 	 */
145 	unsigned int  count:30;
146 };
147 
148 /* the main bitmap structure - one per mddev */
149 struct bitmap {
150 
151 	struct bitmap_counts {
152 		spinlock_t lock;
153 		struct bitmap_page *bp;
154 		/* total number of pages in the bitmap */
155 		unsigned long pages;
156 		/* number of pages not yet allocated */
157 		unsigned long missing_pages;
158 		/* chunksize = 2^chunkshift (for bitops) */
159 		unsigned long chunkshift;
160 		/* total number of data chunks for the array */
161 		unsigned long chunks;
162 	} counts;
163 
164 	struct mddev *mddev; /* the md device that the bitmap is for */
165 
166 	__u64	events_cleared;
167 	int need_sync;
168 
169 	struct bitmap_storage {
170 		/* backing disk file */
171 		struct file *file;
172 		/* cached copy of the bitmap file superblock */
173 		struct page *sb_page;
174 		unsigned long sb_index;
175 		/* list of cache pages for the file */
176 		struct page **filemap;
177 		/* attributes associated filemap pages */
178 		unsigned long *filemap_attr;
179 		/* number of pages in the file */
180 		unsigned long file_pages;
181 		/* total bytes in the bitmap */
182 		unsigned long bytes;
183 	} storage;
184 
185 	unsigned long flags;
186 
187 	int allclean;
188 
189 	atomic_t behind_writes;
190 	/* highest actual value at runtime */
191 	unsigned long behind_writes_used;
192 
193 	/*
194 	 * the bitmap daemon - periodically wakes up and sweeps the bitmap
195 	 * file, cleaning up bits and flushing out pages to disk as necessary
196 	 */
197 	unsigned long daemon_lastrun; /* jiffies of last run */
198 	/*
199 	 * when we lasted called end_sync to update bitmap with resync
200 	 * progress.
201 	 */
202 	unsigned long last_end_sync;
203 
204 	/* pending writes to the bitmap file */
205 	atomic_t pending_writes;
206 	wait_queue_head_t write_wait;
207 	wait_queue_head_t overflow_wait;
208 	wait_queue_head_t behind_wait;
209 
210 	struct kernfs_node *sysfs_can_clear;
211 	/* slot offset for clustered env */
212 	int cluster_slot;
213 };
214 
215 static int __bitmap_resize(struct bitmap *bitmap, sector_t blocks,
216 			   int chunksize, bool init);
217 
218 static inline char *bmname(struct bitmap *bitmap)
219 {
220 	return bitmap->mddev ? mdname(bitmap->mddev) : "mdX";
221 }
222 
223 static bool __bitmap_enabled(struct bitmap *bitmap)
224 {
225 	return bitmap->storage.filemap &&
226 	       !test_bit(BITMAP_STALE, &bitmap->flags);
227 }
228 
229 static bool bitmap_enabled(struct mddev *mddev)
230 {
231 	struct bitmap *bitmap = mddev->bitmap;
232 
233 	if (!bitmap)
234 		return false;
235 
236 	return __bitmap_enabled(bitmap);
237 }
238 
239 /*
240  * check a page and, if necessary, allocate it (or hijack it if the alloc fails)
241  *
242  * 1) check to see if this page is allocated, if it's not then try to alloc
243  * 2) if the alloc fails, set the page's hijacked flag so we'll use the
244  *    page pointer directly as a counter
245  *
246  * if we find our page, we increment the page's refcount so that it stays
247  * allocated while we're using it
248  */
249 static int md_bitmap_checkpage(struct bitmap_counts *bitmap,
250 			       unsigned long page, int create, int no_hijack)
251 __releases(bitmap->lock)
252 __acquires(bitmap->lock)
253 {
254 	unsigned char *mappage;
255 
256 	WARN_ON_ONCE(page >= bitmap->pages);
257 	if (bitmap->bp[page].hijacked) /* it's hijacked, don't try to alloc */
258 		return 0;
259 
260 	if (bitmap->bp[page].map) /* page is already allocated, just return */
261 		return 0;
262 
263 	if (!create)
264 		return -ENOENT;
265 
266 	/* this page has not been allocated yet */
267 
268 	spin_unlock_irq(&bitmap->lock);
269 	/* It is possible that this is being called inside a
270 	 * prepare_to_wait/finish_wait loop from raid5c:make_request().
271 	 * In general it is not permitted to sleep in that context as it
272 	 * can cause the loop to spin freely.
273 	 * That doesn't apply here as we can only reach this point
274 	 * once with any loop.
275 	 * When this function completes, either bp[page].map or
276 	 * bp[page].hijacked.  In either case, this function will
277 	 * abort before getting to this point again.  So there is
278 	 * no risk of a free-spin, and so it is safe to assert
279 	 * that sleeping here is allowed.
280 	 */
281 	sched_annotate_sleep();
282 	mappage = kzalloc(PAGE_SIZE, GFP_NOIO);
283 	spin_lock_irq(&bitmap->lock);
284 
285 	if (mappage == NULL) {
286 		pr_debug("md/bitmap: map page allocation failed, hijacking\n");
287 		/* We don't support hijack for cluster raid */
288 		if (no_hijack)
289 			return -ENOMEM;
290 		/* failed - set the hijacked flag so that we can use the
291 		 * pointer as a counter */
292 		if (!bitmap->bp[page].map)
293 			bitmap->bp[page].hijacked = 1;
294 	} else if (bitmap->bp[page].map ||
295 		   bitmap->bp[page].hijacked) {
296 		/* somebody beat us to getting the page */
297 		kfree(mappage);
298 	} else {
299 
300 		/* no page was in place and we have one, so install it */
301 
302 		bitmap->bp[page].map = mappage;
303 		bitmap->missing_pages--;
304 	}
305 	return 0;
306 }
307 
308 /* if page is completely empty, put it back on the free list, or dealloc it */
309 /* if page was hijacked, unmark the flag so it might get alloced next time */
310 /* Note: lock should be held when calling this */
311 static void md_bitmap_checkfree(struct bitmap_counts *bitmap, unsigned long page)
312 {
313 	char *ptr;
314 
315 	if (bitmap->bp[page].count) /* page is still busy */
316 		return;
317 
318 	/* page is no longer in use, it can be released */
319 
320 	if (bitmap->bp[page].hijacked) { /* page was hijacked, undo this now */
321 		bitmap->bp[page].hijacked = 0;
322 		bitmap->bp[page].map = NULL;
323 	} else {
324 		/* normal case, free the page */
325 		ptr = bitmap->bp[page].map;
326 		bitmap->bp[page].map = NULL;
327 		bitmap->missing_pages++;
328 		kfree(ptr);
329 	}
330 }
331 
332 /*
333  * bitmap file handling - read and write the bitmap file and its superblock
334  */
335 
336 /*
337  * basic page I/O operations
338  */
339 
340 /* IO operations when bitmap is stored near all superblocks */
341 
342 /* choose a good rdev and read the page from there */
343 static int read_sb_page(struct mddev *mddev, loff_t offset,
344 		struct page *page, unsigned long index, int size)
345 {
346 
347 	sector_t sector = mddev->bitmap_info.offset + offset +
348 		index * (PAGE_SIZE / SECTOR_SIZE);
349 	struct md_rdev *rdev;
350 
351 	rdev_for_each(rdev, mddev) {
352 		u32 iosize = roundup(size, bdev_logical_block_size(rdev->bdev));
353 
354 		if (!test_bit(In_sync, &rdev->flags) ||
355 		    test_bit(Faulty, &rdev->flags) ||
356 		    test_bit(Bitmap_sync, &rdev->flags))
357 			continue;
358 
359 		if (sync_page_io(rdev, sector, iosize, page, REQ_OP_READ, true))
360 			return 0;
361 	}
362 	return -EIO;
363 }
364 
365 static struct md_rdev *next_active_rdev(struct md_rdev *rdev, struct mddev *mddev)
366 {
367 	/* Iterate the disks of an mddev, using rcu to protect access to the
368 	 * linked list, and raising the refcount of devices we return to ensure
369 	 * they don't disappear while in use.
370 	 * As devices are only added or removed when raid_disk is < 0 and
371 	 * nr_pending is 0 and In_sync is clear, the entries we return will
372 	 * still be in the same position on the list when we re-enter
373 	 * list_for_each_entry_continue_rcu.
374 	 *
375 	 * Note that if entered with 'rdev == NULL' to start at the
376 	 * beginning, we temporarily assign 'rdev' to an address which
377 	 * isn't really an rdev, but which can be used by
378 	 * list_for_each_entry_continue_rcu() to find the first entry.
379 	 */
380 	rcu_read_lock();
381 	if (rdev == NULL)
382 		/* start at the beginning */
383 		rdev = list_entry(&mddev->disks, struct md_rdev, same_set);
384 	else {
385 		/* release the previous rdev and start from there. */
386 		rdev_dec_pending(rdev, mddev);
387 	}
388 	list_for_each_entry_continue_rcu(rdev, &mddev->disks, same_set) {
389 		if (rdev->raid_disk >= 0 &&
390 		    !test_bit(Faulty, &rdev->flags)) {
391 			/* this is a usable devices */
392 			atomic_inc(&rdev->nr_pending);
393 			rcu_read_unlock();
394 			return rdev;
395 		}
396 	}
397 	rcu_read_unlock();
398 	return NULL;
399 }
400 
401 static unsigned int optimal_io_size(struct block_device *bdev,
402 				    unsigned int last_page_size,
403 				    unsigned int io_size)
404 {
405 	if (bdev_io_opt(bdev) > bdev_logical_block_size(bdev))
406 		return roundup(last_page_size, bdev_io_opt(bdev));
407 	return io_size;
408 }
409 
410 static unsigned int bitmap_io_size(unsigned int io_size, unsigned int opt_size,
411 				   loff_t start, loff_t boundary)
412 {
413 	if (io_size != opt_size &&
414 	    start + opt_size / SECTOR_SIZE <= boundary)
415 		return opt_size;
416 	if (start + io_size / SECTOR_SIZE <= boundary)
417 		return io_size;
418 
419 	/* Overflows boundary */
420 	return 0;
421 }
422 
423 static int __write_sb_page(struct md_rdev *rdev, struct bitmap *bitmap,
424 			   unsigned long pg_index, struct page *page)
425 {
426 	struct block_device *bdev;
427 	struct mddev *mddev = bitmap->mddev;
428 	struct bitmap_storage *store = &bitmap->storage;
429 	unsigned int bitmap_limit = (bitmap->storage.file_pages - pg_index) <<
430 		PAGE_SHIFT;
431 	loff_t sboff, offset = mddev->bitmap_info.offset;
432 	sector_t ps = pg_index * PAGE_SIZE / SECTOR_SIZE;
433 	unsigned int size = PAGE_SIZE;
434 	unsigned int opt_size = PAGE_SIZE;
435 	sector_t doff;
436 
437 	bdev = (rdev->meta_bdev) ? rdev->meta_bdev : rdev->bdev;
438 	/* we compare length (page numbers), not page offset. */
439 	if ((pg_index - store->sb_index) == store->file_pages - 1) {
440 		unsigned int last_page_size = store->bytes & (PAGE_SIZE - 1);
441 
442 		if (last_page_size == 0)
443 			last_page_size = PAGE_SIZE;
444 		size = roundup(last_page_size, bdev_logical_block_size(bdev));
445 		opt_size = optimal_io_size(bdev, last_page_size, size);
446 	}
447 
448 	sboff = rdev->sb_start + offset;
449 	doff = rdev->data_offset;
450 
451 	/* Just make sure we aren't corrupting data or metadata */
452 	if (mddev->external) {
453 		/* Bitmap could be anywhere. */
454 		if (sboff + ps > doff &&
455 		    sboff < (doff + mddev->dev_sectors + PAGE_SIZE / SECTOR_SIZE))
456 			return -EINVAL;
457 	} else if (offset < 0) {
458 		/* DATA  BITMAP METADATA  */
459 		size = bitmap_io_size(size, opt_size, offset + ps, 0);
460 		if (size == 0)
461 			/* bitmap runs in to metadata */
462 			return -EINVAL;
463 
464 		if (doff + mddev->dev_sectors > sboff)
465 			/* data runs in to bitmap */
466 			return -EINVAL;
467 	} else if (rdev->sb_start < rdev->data_offset) {
468 		/* METADATA BITMAP DATA */
469 		size = bitmap_io_size(size, opt_size, sboff + ps, doff);
470 		if (size == 0)
471 			/* bitmap runs in to data */
472 			return -EINVAL;
473 	}
474 
475 	md_super_write(mddev, rdev, sboff + ps, (int)min(size, bitmap_limit), page);
476 	return 0;
477 }
478 
479 static void write_sb_page(struct bitmap *bitmap, unsigned long pg_index,
480 			  struct page *page, bool wait)
481 {
482 	struct mddev *mddev = bitmap->mddev;
483 
484 	do {
485 		struct md_rdev *rdev = NULL;
486 
487 		while ((rdev = next_active_rdev(rdev, mddev)) != NULL) {
488 			if (__write_sb_page(rdev, bitmap, pg_index, page) < 0) {
489 				set_bit(BITMAP_WRITE_ERROR, &bitmap->flags);
490 				return;
491 			}
492 		}
493 	} while (wait && md_super_wait(mddev) < 0);
494 }
495 
496 static void md_bitmap_file_kick(struct bitmap *bitmap);
497 
498 #ifdef CONFIG_MD_BITMAP_FILE
499 static void write_file_page(struct bitmap *bitmap, struct page *page, int wait)
500 {
501 	struct buffer_head *bh = page_buffers(page);
502 
503 	while (bh && bh->b_blocknr) {
504 		atomic_inc(&bitmap->pending_writes);
505 		set_buffer_locked(bh);
506 		set_buffer_mapped(bh);
507 		submit_bh(REQ_OP_WRITE | REQ_SYNC, bh);
508 		bh = bh->b_this_page;
509 	}
510 
511 	if (wait)
512 		wait_event(bitmap->write_wait,
513 			   atomic_read(&bitmap->pending_writes) == 0);
514 }
515 
516 static void end_bitmap_write(struct buffer_head *bh, int uptodate)
517 {
518 	struct bitmap *bitmap = bh->b_private;
519 
520 	if (!uptodate)
521 		set_bit(BITMAP_WRITE_ERROR, &bitmap->flags);
522 	if (atomic_dec_and_test(&bitmap->pending_writes))
523 		wake_up(&bitmap->write_wait);
524 }
525 
526 static void free_buffers(struct page *page)
527 {
528 	struct buffer_head *bh;
529 
530 	if (!PagePrivate(page))
531 		return;
532 
533 	bh = page_buffers(page);
534 	while (bh) {
535 		struct buffer_head *next = bh->b_this_page;
536 		free_buffer_head(bh);
537 		bh = next;
538 	}
539 	detach_page_private(page);
540 	put_page(page);
541 }
542 
543 /* read a page from a file.
544  * We both read the page, and attach buffers to the page to record the
545  * address of each block (using bmap).  These addresses will be used
546  * to write the block later, completely bypassing the filesystem.
547  * This usage is similar to how swap files are handled, and allows us
548  * to write to a file with no concerns of memory allocation failing.
549  */
550 static int read_file_page(struct file *file, unsigned long index,
551 		struct bitmap *bitmap, unsigned long count, struct page *page)
552 {
553 	int ret = 0;
554 	struct inode *inode = file_inode(file);
555 	struct buffer_head *bh;
556 	sector_t block, blk_cur;
557 	unsigned long blocksize = i_blocksize(inode);
558 
559 	pr_debug("read bitmap file (%dB @ %llu)\n", (int)PAGE_SIZE,
560 		 (unsigned long long)index << PAGE_SHIFT);
561 
562 	bh = alloc_page_buffers(page, blocksize);
563 	if (!bh) {
564 		ret = -ENOMEM;
565 		goto out;
566 	}
567 	attach_page_private(page, bh);
568 	blk_cur = index << (PAGE_SHIFT - inode->i_blkbits);
569 	while (bh) {
570 		block = blk_cur;
571 
572 		if (count == 0)
573 			bh->b_blocknr = 0;
574 		else {
575 			ret = bmap(inode, &block);
576 			if (ret || !block) {
577 				ret = -EINVAL;
578 				bh->b_blocknr = 0;
579 				goto out;
580 			}
581 
582 			bh->b_blocknr = block;
583 			bh->b_bdev = inode->i_sb->s_bdev;
584 			if (count < blocksize)
585 				count = 0;
586 			else
587 				count -= blocksize;
588 
589 			bh->b_end_io = end_bitmap_write;
590 			bh->b_private = bitmap;
591 			atomic_inc(&bitmap->pending_writes);
592 			set_buffer_locked(bh);
593 			set_buffer_mapped(bh);
594 			submit_bh(REQ_OP_READ, bh);
595 		}
596 		blk_cur++;
597 		bh = bh->b_this_page;
598 	}
599 
600 	wait_event(bitmap->write_wait,
601 		   atomic_read(&bitmap->pending_writes)==0);
602 	if (test_bit(BITMAP_WRITE_ERROR, &bitmap->flags))
603 		ret = -EIO;
604 out:
605 	if (ret)
606 		pr_err("md: bitmap read error: (%dB @ %llu): %d\n",
607 		       (int)PAGE_SIZE,
608 		       (unsigned long long)index << PAGE_SHIFT,
609 		       ret);
610 	return ret;
611 }
612 #else /* CONFIG_MD_BITMAP_FILE */
613 static void write_file_page(struct bitmap *bitmap, struct page *page, int wait)
614 {
615 }
616 static int read_file_page(struct file *file, unsigned long index,
617 		struct bitmap *bitmap, unsigned long count, struct page *page)
618 {
619 	return -EIO;
620 }
621 static void free_buffers(struct page *page)
622 {
623 	put_page(page);
624 }
625 #endif /* CONFIG_MD_BITMAP_FILE */
626 
627 /*
628  * bitmap file superblock operations
629  */
630 
631 /*
632  * write out a page to a file
633  */
634 static void filemap_write_page(struct bitmap *bitmap, unsigned long pg_index,
635 			       bool wait)
636 {
637 	struct bitmap_storage *store = &bitmap->storage;
638 	struct page *page = store->filemap[pg_index];
639 
640 	if (mddev_is_clustered(bitmap->mddev)) {
641 		/* go to node bitmap area starting point */
642 		pg_index += store->sb_index;
643 	}
644 
645 	if (store->file)
646 		write_file_page(bitmap, page, wait);
647 	else
648 		write_sb_page(bitmap, pg_index, page, wait);
649 }
650 
651 /*
652  * md_bitmap_wait_writes() should be called before writing any bitmap
653  * blocks, to ensure previous writes, particularly from
654  * md_bitmap_daemon_work(), have completed.
655  */
656 static void md_bitmap_wait_writes(struct bitmap *bitmap)
657 {
658 	if (bitmap->storage.file)
659 		wait_event(bitmap->write_wait,
660 			   atomic_read(&bitmap->pending_writes)==0);
661 	else
662 		/* Note that we ignore the return value.  The writes
663 		 * might have failed, but that would just mean that
664 		 * some bits which should be cleared haven't been,
665 		 * which is safe.  The relevant bitmap blocks will
666 		 * probably get written again, but there is no great
667 		 * loss if they aren't.
668 		 */
669 		md_super_wait(bitmap->mddev);
670 }
671 
672 
673 /* update the event counter and sync the superblock to disk */
674 static void bitmap_update_sb(void *data)
675 {
676 	bitmap_super_t *sb;
677 	struct bitmap *bitmap = data;
678 
679 	if (!bitmap || !bitmap->mddev) /* no bitmap for this array */
680 		return;
681 	if (bitmap->mddev->bitmap_info.external)
682 		return;
683 	if (!bitmap->storage.sb_page) /* no superblock */
684 		return;
685 	sb = kmap_atomic(bitmap->storage.sb_page);
686 	sb->events = cpu_to_le64(bitmap->mddev->events);
687 	if (bitmap->mddev->events < bitmap->events_cleared)
688 		/* rocking back to read-only */
689 		bitmap->events_cleared = bitmap->mddev->events;
690 	sb->events_cleared = cpu_to_le64(bitmap->events_cleared);
691 	/*
692 	 * clear BITMAP_WRITE_ERROR bit to protect against the case that
693 	 * a bitmap write error occurred but the later writes succeeded.
694 	 */
695 	sb->state = cpu_to_le32(bitmap->flags & ~BIT(BITMAP_WRITE_ERROR));
696 	/* Just in case these have been changed via sysfs: */
697 	sb->daemon_sleep = cpu_to_le32(bitmap->mddev->bitmap_info.daemon_sleep/HZ);
698 	sb->write_behind = cpu_to_le32(bitmap->mddev->bitmap_info.max_write_behind);
699 	/* This might have been changed by a reshape */
700 	sb->sync_size = cpu_to_le64(bitmap->mddev->resync_max_sectors);
701 	sb->chunksize = cpu_to_le32(bitmap->mddev->bitmap_info.chunksize);
702 	sb->nodes = cpu_to_le32(bitmap->mddev->bitmap_info.nodes);
703 	sb->sectors_reserved = cpu_to_le32(bitmap->mddev->
704 					   bitmap_info.space);
705 	kunmap_atomic(sb);
706 
707 	if (bitmap->storage.file)
708 		write_file_page(bitmap, bitmap->storage.sb_page, 1);
709 	else
710 		write_sb_page(bitmap, bitmap->storage.sb_index,
711 			      bitmap->storage.sb_page, 1);
712 }
713 
714 static void bitmap_print_sb(struct bitmap *bitmap)
715 {
716 	bitmap_super_t *sb;
717 
718 	if (!bitmap || !bitmap->storage.sb_page)
719 		return;
720 	sb = kmap_atomic(bitmap->storage.sb_page);
721 	pr_debug("%s: bitmap file superblock:\n", bmname(bitmap));
722 	pr_debug("         magic: %08x\n", le32_to_cpu(sb->magic));
723 	pr_debug("       version: %u\n", le32_to_cpu(sb->version));
724 	pr_debug("          uuid: %08x.%08x.%08x.%08x\n",
725 		 le32_to_cpu(*(__le32 *)(sb->uuid+0)),
726 		 le32_to_cpu(*(__le32 *)(sb->uuid+4)),
727 		 le32_to_cpu(*(__le32 *)(sb->uuid+8)),
728 		 le32_to_cpu(*(__le32 *)(sb->uuid+12)));
729 	pr_debug("        events: %llu\n",
730 		 (unsigned long long) le64_to_cpu(sb->events));
731 	pr_debug("events cleared: %llu\n",
732 		 (unsigned long long) le64_to_cpu(sb->events_cleared));
733 	pr_debug("         state: %08x\n", le32_to_cpu(sb->state));
734 	pr_debug("     chunksize: %u B\n", le32_to_cpu(sb->chunksize));
735 	pr_debug("  daemon sleep: %us\n", le32_to_cpu(sb->daemon_sleep));
736 	pr_debug("     sync size: %llu KB\n",
737 		 (unsigned long long)le64_to_cpu(sb->sync_size)/2);
738 	pr_debug("max write behind: %u\n", le32_to_cpu(sb->write_behind));
739 	kunmap_atomic(sb);
740 }
741 
742 /*
743  * bitmap_new_disk_sb
744  * @bitmap
745  *
746  * This function is somewhat the reverse of bitmap_read_sb.  bitmap_read_sb
747  * reads and verifies the on-disk bitmap superblock and populates bitmap_info.
748  * This function verifies 'bitmap_info' and populates the on-disk bitmap
749  * structure, which is to be written to disk.
750  *
751  * Returns: 0 on success, -Exxx on error
752  */
753 static int md_bitmap_new_disk_sb(struct bitmap *bitmap)
754 {
755 	bitmap_super_t *sb;
756 	unsigned long chunksize, daemon_sleep, write_behind;
757 
758 	bitmap->storage.sb_page = alloc_page(GFP_KERNEL | __GFP_ZERO);
759 	if (bitmap->storage.sb_page == NULL)
760 		return -ENOMEM;
761 	bitmap->storage.sb_index = 0;
762 
763 	sb = kmap_atomic(bitmap->storage.sb_page);
764 
765 	sb->magic = cpu_to_le32(BITMAP_MAGIC);
766 	sb->version = cpu_to_le32(BITMAP_MAJOR_HI);
767 
768 	chunksize = bitmap->mddev->bitmap_info.chunksize;
769 	BUG_ON(!chunksize);
770 	if (!is_power_of_2(chunksize)) {
771 		kunmap_atomic(sb);
772 		pr_warn("bitmap chunksize not a power of 2\n");
773 		return -EINVAL;
774 	}
775 	sb->chunksize = cpu_to_le32(chunksize);
776 
777 	daemon_sleep = bitmap->mddev->bitmap_info.daemon_sleep;
778 	if (!daemon_sleep || (daemon_sleep > MAX_SCHEDULE_TIMEOUT)) {
779 		pr_debug("Choosing daemon_sleep default (5 sec)\n");
780 		daemon_sleep = 5 * HZ;
781 	}
782 	sb->daemon_sleep = cpu_to_le32(daemon_sleep);
783 	bitmap->mddev->bitmap_info.daemon_sleep = daemon_sleep;
784 
785 	/*
786 	 * FIXME: write_behind for RAID1.  If not specified, what
787 	 * is a good choice?  We choose COUNTER_MAX / 2 arbitrarily.
788 	 */
789 	write_behind = bitmap->mddev->bitmap_info.max_write_behind;
790 	if (write_behind > COUNTER_MAX)
791 		write_behind = COUNTER_MAX / 2;
792 	sb->write_behind = cpu_to_le32(write_behind);
793 	bitmap->mddev->bitmap_info.max_write_behind = write_behind;
794 
795 	/* keep the array size field of the bitmap superblock up to date */
796 	sb->sync_size = cpu_to_le64(bitmap->mddev->resync_max_sectors);
797 
798 	memcpy(sb->uuid, bitmap->mddev->uuid, 16);
799 
800 	set_bit(BITMAP_STALE, &bitmap->flags);
801 	sb->state = cpu_to_le32(bitmap->flags);
802 	bitmap->events_cleared = bitmap->mddev->events;
803 	sb->events_cleared = cpu_to_le64(bitmap->mddev->events);
804 	bitmap->mddev->bitmap_info.nodes = 0;
805 
806 	kunmap_atomic(sb);
807 
808 	return 0;
809 }
810 
811 /* read the superblock from the bitmap file and initialize some bitmap fields */
812 static int md_bitmap_read_sb(struct bitmap *bitmap)
813 {
814 	char *reason = NULL;
815 	bitmap_super_t *sb;
816 	unsigned long chunksize, daemon_sleep, write_behind;
817 	unsigned long long events;
818 	int nodes = 0;
819 	unsigned long sectors_reserved = 0;
820 	int err = -EINVAL;
821 	struct page *sb_page;
822 	loff_t offset = 0;
823 
824 	if (!bitmap->storage.file && !bitmap->mddev->bitmap_info.offset) {
825 		chunksize = 128 * 1024 * 1024;
826 		daemon_sleep = 5 * HZ;
827 		write_behind = 0;
828 		set_bit(BITMAP_STALE, &bitmap->flags);
829 		err = 0;
830 		goto out_no_sb;
831 	}
832 	/* page 0 is the superblock, read it... */
833 	sb_page = alloc_page(GFP_KERNEL);
834 	if (!sb_page)
835 		return -ENOMEM;
836 	bitmap->storage.sb_page = sb_page;
837 
838 re_read:
839 	/* If cluster_slot is set, the cluster is setup */
840 	if (bitmap->cluster_slot >= 0) {
841 		sector_t bm_blocks = bitmap->mddev->resync_max_sectors;
842 
843 		bm_blocks = DIV_ROUND_UP_SECTOR_T(bm_blocks,
844 			   (bitmap->mddev->bitmap_info.chunksize >> 9));
845 		/* bits to bytes */
846 		bm_blocks = ((bm_blocks+7) >> 3) + sizeof(bitmap_super_t);
847 		/* to 4k blocks */
848 		bm_blocks = DIV_ROUND_UP_SECTOR_T(bm_blocks, 4096);
849 		offset = bitmap->cluster_slot * (bm_blocks << 3);
850 		pr_debug("%s:%d bm slot: %d offset: %llu\n", __func__, __LINE__,
851 			bitmap->cluster_slot, offset);
852 	}
853 
854 	if (bitmap->storage.file) {
855 		loff_t isize = i_size_read(bitmap->storage.file->f_mapping->host);
856 		int bytes = isize > PAGE_SIZE ? PAGE_SIZE : isize;
857 
858 		err = read_file_page(bitmap->storage.file, 0,
859 				bitmap, bytes, sb_page);
860 	} else {
861 		err = read_sb_page(bitmap->mddev, offset, sb_page, 0,
862 				   sizeof(bitmap_super_t));
863 	}
864 	if (err)
865 		return err;
866 
867 	err = -EINVAL;
868 	sb = kmap_atomic(sb_page);
869 
870 	chunksize = le32_to_cpu(sb->chunksize);
871 	daemon_sleep = le32_to_cpu(sb->daemon_sleep) * HZ;
872 	write_behind = le32_to_cpu(sb->write_behind);
873 	sectors_reserved = le32_to_cpu(sb->sectors_reserved);
874 
875 	/* verify that the bitmap-specific fields are valid */
876 	if (sb->magic != cpu_to_le32(BITMAP_MAGIC))
877 		reason = "bad magic";
878 	else if (le32_to_cpu(sb->version) < BITMAP_MAJOR_LO ||
879 		 le32_to_cpu(sb->version) > BITMAP_MAJOR_CLUSTERED)
880 		reason = "unrecognized superblock version";
881 	else if (chunksize < 512)
882 		reason = "bitmap chunksize too small";
883 	else if (!is_power_of_2(chunksize))
884 		reason = "bitmap chunksize not a power of 2";
885 	else if (daemon_sleep < 1 || daemon_sleep > MAX_SCHEDULE_TIMEOUT)
886 		reason = "daemon sleep period out of range";
887 	else if (write_behind > COUNTER_MAX)
888 		reason = "write-behind limit out of range (0 - 16383)";
889 	if (reason) {
890 		pr_warn("%s: invalid bitmap file superblock: %s\n",
891 			bmname(bitmap), reason);
892 		goto out;
893 	}
894 
895 	/*
896 	 * Setup nodes/clustername only if bitmap version is
897 	 * cluster-compatible
898 	 */
899 	if (sb->version == cpu_to_le32(BITMAP_MAJOR_CLUSTERED)) {
900 		nodes = le32_to_cpu(sb->nodes);
901 		strscpy(bitmap->mddev->bitmap_info.cluster_name,
902 				sb->cluster_name, 64);
903 	}
904 
905 	/* keep the array size field of the bitmap superblock up to date */
906 	sb->sync_size = cpu_to_le64(bitmap->mddev->resync_max_sectors);
907 
908 	if (bitmap->mddev->persistent) {
909 		/*
910 		 * We have a persistent array superblock, so compare the
911 		 * bitmap's UUID and event counter to the mddev's
912 		 */
913 		if (memcmp(sb->uuid, bitmap->mddev->uuid, 16)) {
914 			pr_warn("%s: bitmap superblock UUID mismatch\n",
915 				bmname(bitmap));
916 			goto out;
917 		}
918 		events = le64_to_cpu(sb->events);
919 		if (!nodes && (events < bitmap->mddev->events)) {
920 			pr_warn("%s: bitmap file is out of date (%llu < %llu) -- forcing full recovery\n",
921 				bmname(bitmap), events,
922 				(unsigned long long) bitmap->mddev->events);
923 			set_bit(BITMAP_STALE, &bitmap->flags);
924 		}
925 	}
926 
927 	/* assign fields using values from superblock */
928 	bitmap->flags |= le32_to_cpu(sb->state);
929 	if (le32_to_cpu(sb->version) == BITMAP_MAJOR_HOSTENDIAN)
930 		set_bit(BITMAP_HOSTENDIAN, &bitmap->flags);
931 	bitmap->events_cleared = le64_to_cpu(sb->events_cleared);
932 	err = 0;
933 
934 out:
935 	kunmap_atomic(sb);
936 	if (err == 0 && nodes && (bitmap->cluster_slot < 0)) {
937 		/* Assigning chunksize is required for "re_read" */
938 		bitmap->mddev->bitmap_info.chunksize = chunksize;
939 		err = md_setup_cluster(bitmap->mddev, nodes);
940 		if (err) {
941 			pr_warn("%s: Could not setup cluster service (%d)\n",
942 				bmname(bitmap), err);
943 			goto out_no_sb;
944 		}
945 		bitmap->cluster_slot = md_cluster_ops->slot_number(bitmap->mddev);
946 		goto re_read;
947 	}
948 
949 out_no_sb:
950 	if (err == 0) {
951 		if (test_bit(BITMAP_STALE, &bitmap->flags))
952 			bitmap->events_cleared = bitmap->mddev->events;
953 		bitmap->mddev->bitmap_info.chunksize = chunksize;
954 		bitmap->mddev->bitmap_info.daemon_sleep = daemon_sleep;
955 		bitmap->mddev->bitmap_info.max_write_behind = write_behind;
956 		bitmap->mddev->bitmap_info.nodes = nodes;
957 		if (bitmap->mddev->bitmap_info.space == 0 ||
958 			bitmap->mddev->bitmap_info.space > sectors_reserved)
959 			bitmap->mddev->bitmap_info.space = sectors_reserved;
960 	} else {
961 		bitmap_print_sb(bitmap);
962 		if (bitmap->cluster_slot < 0)
963 			md_cluster_stop(bitmap->mddev);
964 	}
965 	return err;
966 }
967 
968 /*
969  * general bitmap file operations
970  */
971 
972 /*
973  * on-disk bitmap:
974  *
975  * Use one bit per "chunk" (block set). We do the disk I/O on the bitmap
976  * file a page at a time. There's a superblock at the start of the file.
977  */
978 /* calculate the index of the page that contains this bit */
979 static inline unsigned long file_page_index(struct bitmap_storage *store,
980 					    unsigned long chunk)
981 {
982 	if (store->sb_page)
983 		chunk += sizeof(bitmap_super_t) << 3;
984 	return chunk >> PAGE_BIT_SHIFT;
985 }
986 
987 /* calculate the (bit) offset of this bit within a page */
988 static inline unsigned long file_page_offset(struct bitmap_storage *store,
989 					     unsigned long chunk)
990 {
991 	if (store->sb_page)
992 		chunk += sizeof(bitmap_super_t) << 3;
993 	return chunk & (PAGE_BITS - 1);
994 }
995 
996 /*
997  * return a pointer to the page in the filemap that contains the given bit
998  *
999  */
1000 static inline struct page *filemap_get_page(struct bitmap_storage *store,
1001 					    unsigned long chunk)
1002 {
1003 	if (file_page_index(store, chunk) >= store->file_pages)
1004 		return NULL;
1005 	return store->filemap[file_page_index(store, chunk)];
1006 }
1007 
1008 static int md_bitmap_storage_alloc(struct bitmap_storage *store,
1009 				   unsigned long chunks, int with_super,
1010 				   int slot_number)
1011 {
1012 	int pnum, offset = 0;
1013 	unsigned long num_pages;
1014 	unsigned long bytes;
1015 
1016 	bytes = DIV_ROUND_UP(chunks, 8);
1017 	if (with_super)
1018 		bytes += sizeof(bitmap_super_t);
1019 
1020 	num_pages = DIV_ROUND_UP(bytes, PAGE_SIZE);
1021 	offset = slot_number * num_pages;
1022 
1023 	store->filemap = kmalloc_array(num_pages, sizeof(struct page *),
1024 				       GFP_KERNEL);
1025 	if (!store->filemap)
1026 		return -ENOMEM;
1027 
1028 	if (with_super && !store->sb_page) {
1029 		store->sb_page = alloc_page(GFP_KERNEL|__GFP_ZERO);
1030 		if (store->sb_page == NULL)
1031 			return -ENOMEM;
1032 	}
1033 
1034 	pnum = 0;
1035 	if (store->sb_page) {
1036 		store->filemap[0] = store->sb_page;
1037 		pnum = 1;
1038 		store->sb_index = offset;
1039 	}
1040 
1041 	for ( ; pnum < num_pages; pnum++) {
1042 		store->filemap[pnum] = alloc_page(GFP_KERNEL|__GFP_ZERO);
1043 		if (!store->filemap[pnum]) {
1044 			store->file_pages = pnum;
1045 			return -ENOMEM;
1046 		}
1047 	}
1048 	store->file_pages = pnum;
1049 
1050 	/* We need 4 bits per page, rounded up to a multiple
1051 	 * of sizeof(unsigned long) */
1052 	store->filemap_attr = kzalloc(
1053 		roundup(DIV_ROUND_UP(num_pages*4, 8), sizeof(unsigned long)),
1054 		GFP_KERNEL);
1055 	if (!store->filemap_attr)
1056 		return -ENOMEM;
1057 
1058 	store->bytes = bytes;
1059 
1060 	return 0;
1061 }
1062 
1063 static void md_bitmap_file_unmap(struct bitmap_storage *store)
1064 {
1065 	struct file *file = store->file;
1066 	struct page *sb_page = store->sb_page;
1067 	struct page **map = store->filemap;
1068 	int pages = store->file_pages;
1069 
1070 	while (pages--)
1071 		if (map[pages] != sb_page) /* 0 is sb_page, release it below */
1072 			free_buffers(map[pages]);
1073 	kfree(map);
1074 	kfree(store->filemap_attr);
1075 
1076 	if (sb_page)
1077 		free_buffers(sb_page);
1078 
1079 	if (file) {
1080 		struct inode *inode = file_inode(file);
1081 		invalidate_mapping_pages(inode->i_mapping, 0, -1);
1082 		fput(file);
1083 	}
1084 }
1085 
1086 /*
1087  * bitmap_file_kick - if an error occurs while manipulating the bitmap file
1088  * then it is no longer reliable, so we stop using it and we mark the file
1089  * as failed in the superblock
1090  */
1091 static void md_bitmap_file_kick(struct bitmap *bitmap)
1092 {
1093 	if (!test_and_set_bit(BITMAP_STALE, &bitmap->flags)) {
1094 		bitmap_update_sb(bitmap);
1095 
1096 		if (bitmap->storage.file) {
1097 			pr_warn("%s: kicking failed bitmap file %pD4 from array!\n",
1098 				bmname(bitmap), bitmap->storage.file);
1099 
1100 		} else
1101 			pr_warn("%s: disabling internal bitmap due to errors\n",
1102 				bmname(bitmap));
1103 	}
1104 }
1105 
1106 enum bitmap_page_attr {
1107 	BITMAP_PAGE_DIRTY = 0,     /* there are set bits that need to be synced */
1108 	BITMAP_PAGE_PENDING = 1,   /* there are bits that are being cleaned.
1109 				    * i.e. counter is 1 or 2. */
1110 	BITMAP_PAGE_NEEDWRITE = 2, /* there are cleared bits that need to be synced */
1111 };
1112 
1113 static inline void set_page_attr(struct bitmap *bitmap, int pnum,
1114 				 enum bitmap_page_attr attr)
1115 {
1116 	set_bit((pnum<<2) + attr, bitmap->storage.filemap_attr);
1117 }
1118 
1119 static inline void clear_page_attr(struct bitmap *bitmap, int pnum,
1120 				   enum bitmap_page_attr attr)
1121 {
1122 	clear_bit((pnum<<2) + attr, bitmap->storage.filemap_attr);
1123 }
1124 
1125 static inline int test_page_attr(struct bitmap *bitmap, int pnum,
1126 				 enum bitmap_page_attr attr)
1127 {
1128 	return test_bit((pnum<<2) + attr, bitmap->storage.filemap_attr);
1129 }
1130 
1131 static inline int test_and_clear_page_attr(struct bitmap *bitmap, int pnum,
1132 					   enum bitmap_page_attr attr)
1133 {
1134 	return test_and_clear_bit((pnum<<2) + attr,
1135 				  bitmap->storage.filemap_attr);
1136 }
1137 /*
1138  * bitmap_file_set_bit -- called before performing a write to the md device
1139  * to set (and eventually sync) a particular bit in the bitmap file
1140  *
1141  * we set the bit immediately, then we record the page number so that
1142  * when an unplug occurs, we can flush the dirty pages out to disk
1143  */
1144 static void md_bitmap_file_set_bit(struct bitmap *bitmap, sector_t block)
1145 {
1146 	unsigned long bit;
1147 	struct page *page;
1148 	void *kaddr;
1149 	unsigned long chunk = block >> bitmap->counts.chunkshift;
1150 	struct bitmap_storage *store = &bitmap->storage;
1151 	unsigned long index = file_page_index(store, chunk);
1152 	unsigned long node_offset = 0;
1153 
1154 	index += store->sb_index;
1155 	if (mddev_is_clustered(bitmap->mddev))
1156 		node_offset = bitmap->cluster_slot * store->file_pages;
1157 
1158 	page = filemap_get_page(&bitmap->storage, chunk);
1159 	if (!page)
1160 		return;
1161 	bit = file_page_offset(&bitmap->storage, chunk);
1162 
1163 	/* set the bit */
1164 	kaddr = kmap_atomic(page);
1165 	if (test_bit(BITMAP_HOSTENDIAN, &bitmap->flags))
1166 		set_bit(bit, kaddr);
1167 	else
1168 		set_bit_le(bit, kaddr);
1169 	kunmap_atomic(kaddr);
1170 	pr_debug("set file bit %lu page %lu\n", bit, index);
1171 	/* record page number so it gets flushed to disk when unplug occurs */
1172 	set_page_attr(bitmap, index - node_offset, BITMAP_PAGE_DIRTY);
1173 }
1174 
1175 static void md_bitmap_file_clear_bit(struct bitmap *bitmap, sector_t block)
1176 {
1177 	unsigned long bit;
1178 	struct page *page;
1179 	void *paddr;
1180 	unsigned long chunk = block >> bitmap->counts.chunkshift;
1181 	struct bitmap_storage *store = &bitmap->storage;
1182 	unsigned long index = file_page_index(store, chunk);
1183 	unsigned long node_offset = 0;
1184 
1185 	index += store->sb_index;
1186 	if (mddev_is_clustered(bitmap->mddev))
1187 		node_offset = bitmap->cluster_slot * store->file_pages;
1188 
1189 	page = filemap_get_page(&bitmap->storage, chunk);
1190 	if (!page)
1191 		return;
1192 	bit = file_page_offset(&bitmap->storage, chunk);
1193 	paddr = kmap_atomic(page);
1194 	if (test_bit(BITMAP_HOSTENDIAN, &bitmap->flags))
1195 		clear_bit(bit, paddr);
1196 	else
1197 		clear_bit_le(bit, paddr);
1198 	kunmap_atomic(paddr);
1199 	if (!test_page_attr(bitmap, index - node_offset, BITMAP_PAGE_NEEDWRITE)) {
1200 		set_page_attr(bitmap, index - node_offset, BITMAP_PAGE_PENDING);
1201 		bitmap->allclean = 0;
1202 	}
1203 }
1204 
1205 static int md_bitmap_file_test_bit(struct bitmap *bitmap, sector_t block)
1206 {
1207 	unsigned long bit;
1208 	struct page *page;
1209 	void *paddr;
1210 	unsigned long chunk = block >> bitmap->counts.chunkshift;
1211 	int set = 0;
1212 
1213 	page = filemap_get_page(&bitmap->storage, chunk);
1214 	if (!page)
1215 		return -EINVAL;
1216 	bit = file_page_offset(&bitmap->storage, chunk);
1217 	paddr = kmap_atomic(page);
1218 	if (test_bit(BITMAP_HOSTENDIAN, &bitmap->flags))
1219 		set = test_bit(bit, paddr);
1220 	else
1221 		set = test_bit_le(bit, paddr);
1222 	kunmap_atomic(paddr);
1223 	return set;
1224 }
1225 
1226 /* this gets called when the md device is ready to unplug its underlying
1227  * (slave) device queues -- before we let any writes go down, we need to
1228  * sync the dirty pages of the bitmap file to disk */
1229 static void __bitmap_unplug(struct bitmap *bitmap)
1230 {
1231 	unsigned long i;
1232 	int dirty, need_write;
1233 	int writing = 0;
1234 
1235 	if (!__bitmap_enabled(bitmap))
1236 		return;
1237 
1238 	/* look at each page to see if there are any set bits that need to be
1239 	 * flushed out to disk */
1240 	for (i = 0; i < bitmap->storage.file_pages; i++) {
1241 		dirty = test_and_clear_page_attr(bitmap, i, BITMAP_PAGE_DIRTY);
1242 		need_write = test_and_clear_page_attr(bitmap, i,
1243 						      BITMAP_PAGE_NEEDWRITE);
1244 		if (dirty || need_write) {
1245 			if (!writing) {
1246 				md_bitmap_wait_writes(bitmap);
1247 				mddev_add_trace_msg(bitmap->mddev,
1248 					"md bitmap_unplug");
1249 			}
1250 			clear_page_attr(bitmap, i, BITMAP_PAGE_PENDING);
1251 			filemap_write_page(bitmap, i, false);
1252 			writing = 1;
1253 		}
1254 	}
1255 	if (writing)
1256 		md_bitmap_wait_writes(bitmap);
1257 
1258 	if (test_bit(BITMAP_WRITE_ERROR, &bitmap->flags))
1259 		md_bitmap_file_kick(bitmap);
1260 }
1261 
1262 struct bitmap_unplug_work {
1263 	struct work_struct work;
1264 	struct bitmap *bitmap;
1265 	struct completion *done;
1266 };
1267 
1268 static void md_bitmap_unplug_fn(struct work_struct *work)
1269 {
1270 	struct bitmap_unplug_work *unplug_work =
1271 		container_of(work, struct bitmap_unplug_work, work);
1272 
1273 	__bitmap_unplug(unplug_work->bitmap);
1274 	complete(unplug_work->done);
1275 }
1276 
1277 static void bitmap_unplug_async(struct bitmap *bitmap)
1278 {
1279 	DECLARE_COMPLETION_ONSTACK(done);
1280 	struct bitmap_unplug_work unplug_work;
1281 
1282 	INIT_WORK_ONSTACK(&unplug_work.work, md_bitmap_unplug_fn);
1283 	unplug_work.bitmap = bitmap;
1284 	unplug_work.done = &done;
1285 
1286 	queue_work(md_bitmap_wq, &unplug_work.work);
1287 	wait_for_completion(&done);
1288 }
1289 
1290 static void bitmap_unplug(struct mddev *mddev, bool sync)
1291 {
1292 	struct bitmap *bitmap = mddev->bitmap;
1293 
1294 	if (!bitmap)
1295 		return;
1296 
1297 	if (sync)
1298 		__bitmap_unplug(bitmap);
1299 	else
1300 		bitmap_unplug_async(bitmap);
1301 }
1302 
1303 static void md_bitmap_set_memory_bits(struct bitmap *bitmap, sector_t offset, int needed);
1304 
1305 /*
1306  * Initialize the in-memory bitmap from the on-disk bitmap and set up the memory
1307  * mapping of the bitmap file.
1308  *
1309  * Special case: If there's no bitmap file, or if the bitmap file had been
1310  * previously kicked from the array, we mark all the bits as 1's in order to
1311  * cause a full resync.
1312  *
1313  * We ignore all bits for sectors that end earlier than 'start'.
1314  * This is used when reading an out-of-date bitmap.
1315  */
1316 static int md_bitmap_init_from_disk(struct bitmap *bitmap, sector_t start)
1317 {
1318 	bool outofdate = test_bit(BITMAP_STALE, &bitmap->flags);
1319 	struct mddev *mddev = bitmap->mddev;
1320 	unsigned long chunks = bitmap->counts.chunks;
1321 	struct bitmap_storage *store = &bitmap->storage;
1322 	struct file *file = store->file;
1323 	unsigned long node_offset = 0;
1324 	unsigned long bit_cnt = 0;
1325 	unsigned long i;
1326 	int ret;
1327 
1328 	if (!file && !mddev->bitmap_info.offset) {
1329 		/* No permanent bitmap - fill with '1s'. */
1330 		store->filemap = NULL;
1331 		store->file_pages = 0;
1332 		for (i = 0; i < chunks ; i++) {
1333 			/* if the disk bit is set, set the memory bit */
1334 			int needed = ((sector_t)(i+1) << (bitmap->counts.chunkshift)
1335 				      >= start);
1336 			md_bitmap_set_memory_bits(bitmap,
1337 						  (sector_t)i << bitmap->counts.chunkshift,
1338 						  needed);
1339 		}
1340 		return 0;
1341 	}
1342 
1343 	if (file && i_size_read(file->f_mapping->host) < store->bytes) {
1344 		pr_warn("%s: bitmap file too short %lu < %lu\n",
1345 			bmname(bitmap),
1346 			(unsigned long) i_size_read(file->f_mapping->host),
1347 			store->bytes);
1348 		ret = -ENOSPC;
1349 		goto err;
1350 	}
1351 
1352 	if (mddev_is_clustered(mddev))
1353 		node_offset = bitmap->cluster_slot * (DIV_ROUND_UP(store->bytes, PAGE_SIZE));
1354 
1355 	for (i = 0; i < store->file_pages; i++) {
1356 		struct page *page = store->filemap[i];
1357 		int count;
1358 
1359 		/* unmap the old page, we're done with it */
1360 		if (i == store->file_pages - 1)
1361 			count = store->bytes - i * PAGE_SIZE;
1362 		else
1363 			count = PAGE_SIZE;
1364 
1365 		if (file)
1366 			ret = read_file_page(file, i, bitmap, count, page);
1367 		else
1368 			ret = read_sb_page(mddev, 0, page, i + node_offset,
1369 					   count);
1370 		if (ret)
1371 			goto err;
1372 	}
1373 
1374 	if (outofdate) {
1375 		pr_warn("%s: bitmap file is out of date, doing full recovery\n",
1376 			bmname(bitmap));
1377 
1378 		for (i = 0; i < store->file_pages; i++) {
1379 			struct page *page = store->filemap[i];
1380 			unsigned long offset = 0;
1381 			void *paddr;
1382 
1383 			if (i == 0 && !mddev->bitmap_info.external)
1384 				offset = sizeof(bitmap_super_t);
1385 
1386 			/*
1387 			 * If the bitmap is out of date, dirty the whole page
1388 			 * and write it out
1389 			 */
1390 			paddr = kmap_atomic(page);
1391 			memset(paddr + offset, 0xff, PAGE_SIZE - offset);
1392 			kunmap_atomic(paddr);
1393 
1394 			filemap_write_page(bitmap, i, true);
1395 			if (test_bit(BITMAP_WRITE_ERROR, &bitmap->flags)) {
1396 				ret = -EIO;
1397 				goto err;
1398 			}
1399 		}
1400 	}
1401 
1402 	for (i = 0; i < chunks; i++) {
1403 		struct page *page = filemap_get_page(&bitmap->storage, i);
1404 		unsigned long bit = file_page_offset(&bitmap->storage, i);
1405 		void *paddr;
1406 		bool was_set;
1407 
1408 		paddr = kmap_atomic(page);
1409 		if (test_bit(BITMAP_HOSTENDIAN, &bitmap->flags))
1410 			was_set = test_bit(bit, paddr);
1411 		else
1412 			was_set = test_bit_le(bit, paddr);
1413 		kunmap_atomic(paddr);
1414 
1415 		if (was_set) {
1416 			/* if the disk bit is set, set the memory bit */
1417 			int needed = ((sector_t)(i+1) << bitmap->counts.chunkshift
1418 				      >= start);
1419 			md_bitmap_set_memory_bits(bitmap,
1420 						  (sector_t)i << bitmap->counts.chunkshift,
1421 						  needed);
1422 			bit_cnt++;
1423 		}
1424 	}
1425 
1426 	pr_debug("%s: bitmap initialized from disk: read %lu pages, set %lu of %lu bits\n",
1427 		 bmname(bitmap), store->file_pages,
1428 		 bit_cnt, chunks);
1429 
1430 	return 0;
1431 
1432  err:
1433 	pr_warn("%s: bitmap initialisation failed: %d\n",
1434 		bmname(bitmap), ret);
1435 	return ret;
1436 }
1437 
1438 /* just flag bitmap pages as needing to be written. */
1439 static void bitmap_write_all(struct mddev *mddev)
1440 {
1441 	int i;
1442 	struct bitmap *bitmap = mddev->bitmap;
1443 
1444 	if (!bitmap || !bitmap->storage.filemap)
1445 		return;
1446 
1447 	/* Only one copy, so nothing needed */
1448 	if (bitmap->storage.file)
1449 		return;
1450 
1451 	for (i = 0; i < bitmap->storage.file_pages; i++)
1452 		set_page_attr(bitmap, i, BITMAP_PAGE_NEEDWRITE);
1453 	bitmap->allclean = 0;
1454 }
1455 
1456 static void md_bitmap_count_page(struct bitmap_counts *bitmap,
1457 				 sector_t offset, int inc)
1458 {
1459 	sector_t chunk = offset >> bitmap->chunkshift;
1460 	unsigned long page = chunk >> PAGE_COUNTER_SHIFT;
1461 	bitmap->bp[page].count += inc;
1462 	md_bitmap_checkfree(bitmap, page);
1463 }
1464 
1465 static void md_bitmap_set_pending(struct bitmap_counts *bitmap, sector_t offset)
1466 {
1467 	sector_t chunk = offset >> bitmap->chunkshift;
1468 	unsigned long page = chunk >> PAGE_COUNTER_SHIFT;
1469 	struct bitmap_page *bp = &bitmap->bp[page];
1470 
1471 	if (!bp->pending)
1472 		bp->pending = 1;
1473 }
1474 
1475 static bitmap_counter_t *md_bitmap_get_counter(struct bitmap_counts *bitmap,
1476 					       sector_t offset, sector_t *blocks,
1477 					       int create);
1478 
1479 static void mddev_set_timeout(struct mddev *mddev, unsigned long timeout,
1480 			      bool force)
1481 {
1482 	struct md_thread *thread;
1483 
1484 	rcu_read_lock();
1485 	thread = rcu_dereference(mddev->thread);
1486 
1487 	if (!thread)
1488 		goto out;
1489 
1490 	if (force || thread->timeout < MAX_SCHEDULE_TIMEOUT)
1491 		thread->timeout = timeout;
1492 
1493 out:
1494 	rcu_read_unlock();
1495 }
1496 
1497 /*
1498  * bitmap daemon -- periodically wakes up to clean bits and flush pages
1499  *			out to disk
1500  */
1501 static void bitmap_daemon_work(struct mddev *mddev)
1502 {
1503 	struct bitmap *bitmap;
1504 	unsigned long j;
1505 	unsigned long nextpage;
1506 	sector_t blocks;
1507 	struct bitmap_counts *counts;
1508 
1509 	/* Use a mutex to guard daemon_work against
1510 	 * bitmap_destroy.
1511 	 */
1512 	mutex_lock(&mddev->bitmap_info.mutex);
1513 	bitmap = mddev->bitmap;
1514 	if (bitmap == NULL) {
1515 		mutex_unlock(&mddev->bitmap_info.mutex);
1516 		return;
1517 	}
1518 	if (time_before(jiffies, bitmap->daemon_lastrun
1519 			+ mddev->bitmap_info.daemon_sleep))
1520 		goto done;
1521 
1522 	bitmap->daemon_lastrun = jiffies;
1523 	if (bitmap->allclean) {
1524 		mddev_set_timeout(mddev, MAX_SCHEDULE_TIMEOUT, true);
1525 		goto done;
1526 	}
1527 	bitmap->allclean = 1;
1528 
1529 	mddev_add_trace_msg(bitmap->mddev, "md bitmap_daemon_work");
1530 
1531 	/* Any file-page which is PENDING now needs to be written.
1532 	 * So set NEEDWRITE now, then after we make any last-minute changes
1533 	 * we will write it.
1534 	 */
1535 	for (j = 0; j < bitmap->storage.file_pages; j++)
1536 		if (test_and_clear_page_attr(bitmap, j,
1537 					     BITMAP_PAGE_PENDING))
1538 			set_page_attr(bitmap, j,
1539 				      BITMAP_PAGE_NEEDWRITE);
1540 
1541 	if (bitmap->need_sync &&
1542 	    mddev->bitmap_info.external == 0) {
1543 		/* Arrange for superblock update as well as
1544 		 * other changes */
1545 		bitmap_super_t *sb;
1546 		bitmap->need_sync = 0;
1547 		if (bitmap->storage.filemap) {
1548 			sb = kmap_atomic(bitmap->storage.sb_page);
1549 			sb->events_cleared =
1550 				cpu_to_le64(bitmap->events_cleared);
1551 			kunmap_atomic(sb);
1552 			set_page_attr(bitmap, 0,
1553 				      BITMAP_PAGE_NEEDWRITE);
1554 		}
1555 	}
1556 	/* Now look at the bitmap counters and if any are '2' or '1',
1557 	 * decrement and handle accordingly.
1558 	 */
1559 	counts = &bitmap->counts;
1560 	spin_lock_irq(&counts->lock);
1561 	nextpage = 0;
1562 	for (j = 0; j < counts->chunks; j++) {
1563 		bitmap_counter_t *bmc;
1564 		sector_t  block = (sector_t)j << counts->chunkshift;
1565 
1566 		if (j == nextpage) {
1567 			nextpage += PAGE_COUNTER_RATIO;
1568 			if (!counts->bp[j >> PAGE_COUNTER_SHIFT].pending) {
1569 				j |= PAGE_COUNTER_MASK;
1570 				continue;
1571 			}
1572 			counts->bp[j >> PAGE_COUNTER_SHIFT].pending = 0;
1573 		}
1574 
1575 		bmc = md_bitmap_get_counter(counts, block, &blocks, 0);
1576 		if (!bmc) {
1577 			j |= PAGE_COUNTER_MASK;
1578 			continue;
1579 		}
1580 		if (*bmc == 1 && !bitmap->need_sync) {
1581 			/* We can clear the bit */
1582 			*bmc = 0;
1583 			md_bitmap_count_page(counts, block, -1);
1584 			md_bitmap_file_clear_bit(bitmap, block);
1585 		} else if (*bmc && *bmc <= 2) {
1586 			*bmc = 1;
1587 			md_bitmap_set_pending(counts, block);
1588 			bitmap->allclean = 0;
1589 		}
1590 	}
1591 	spin_unlock_irq(&counts->lock);
1592 
1593 	md_bitmap_wait_writes(bitmap);
1594 	/* Now start writeout on any page in NEEDWRITE that isn't DIRTY.
1595 	 * DIRTY pages need to be written by bitmap_unplug so it can wait
1596 	 * for them.
1597 	 * If we find any DIRTY page we stop there and let bitmap_unplug
1598 	 * handle all the rest.  This is important in the case where
1599 	 * the first blocking holds the superblock and it has been updated.
1600 	 * We mustn't write any other blocks before the superblock.
1601 	 */
1602 	for (j = 0;
1603 	     j < bitmap->storage.file_pages
1604 		     && !test_bit(BITMAP_STALE, &bitmap->flags);
1605 	     j++) {
1606 		if (test_page_attr(bitmap, j,
1607 				   BITMAP_PAGE_DIRTY))
1608 			/* bitmap_unplug will handle the rest */
1609 			break;
1610 		if (bitmap->storage.filemap &&
1611 		    test_and_clear_page_attr(bitmap, j,
1612 					     BITMAP_PAGE_NEEDWRITE))
1613 			filemap_write_page(bitmap, j, false);
1614 	}
1615 
1616  done:
1617 	if (bitmap->allclean == 0)
1618 		mddev_set_timeout(mddev, mddev->bitmap_info.daemon_sleep, true);
1619 	mutex_unlock(&mddev->bitmap_info.mutex);
1620 }
1621 
1622 static bitmap_counter_t *md_bitmap_get_counter(struct bitmap_counts *bitmap,
1623 					       sector_t offset, sector_t *blocks,
1624 					       int create)
1625 __releases(bitmap->lock)
1626 __acquires(bitmap->lock)
1627 {
1628 	/* If 'create', we might release the lock and reclaim it.
1629 	 * The lock must have been taken with interrupts enabled.
1630 	 * If !create, we don't release the lock.
1631 	 */
1632 	sector_t chunk = offset >> bitmap->chunkshift;
1633 	unsigned long page = chunk >> PAGE_COUNTER_SHIFT;
1634 	unsigned long pageoff = (chunk & PAGE_COUNTER_MASK) << COUNTER_BYTE_SHIFT;
1635 	sector_t csize = ((sector_t)1) << bitmap->chunkshift;
1636 	int err;
1637 
1638 	if (page >= bitmap->pages) {
1639 		/*
1640 		 * This can happen if bitmap_start_sync goes beyond
1641 		 * End-of-device while looking for a whole page or
1642 		 * user set a huge number to sysfs bitmap_set_bits.
1643 		 */
1644 		*blocks = csize - (offset & (csize - 1));
1645 		return NULL;
1646 	}
1647 	err = md_bitmap_checkpage(bitmap, page, create, 0);
1648 
1649 	if (bitmap->bp[page].hijacked ||
1650 	    bitmap->bp[page].map == NULL)
1651 		csize = ((sector_t)1) << (bitmap->chunkshift +
1652 					  PAGE_COUNTER_SHIFT);
1653 
1654 	*blocks = csize - (offset & (csize - 1));
1655 
1656 	if (err < 0)
1657 		return NULL;
1658 
1659 	/* now locked ... */
1660 
1661 	if (bitmap->bp[page].hijacked) { /* hijacked pointer */
1662 		/* should we use the first or second counter field
1663 		 * of the hijacked pointer? */
1664 		int hi = (pageoff > PAGE_COUNTER_MASK);
1665 		return  &((bitmap_counter_t *)
1666 			  &bitmap->bp[page].map)[hi];
1667 	} else /* page is allocated */
1668 		return (bitmap_counter_t *)
1669 			&(bitmap->bp[page].map[pageoff]);
1670 }
1671 
1672 static int bitmap_startwrite(struct mddev *mddev, sector_t offset,
1673 			     unsigned long sectors, bool behind)
1674 {
1675 	struct bitmap *bitmap = mddev->bitmap;
1676 
1677 	if (!bitmap)
1678 		return 0;
1679 
1680 	if (behind) {
1681 		int bw;
1682 		atomic_inc(&bitmap->behind_writes);
1683 		bw = atomic_read(&bitmap->behind_writes);
1684 		if (bw > bitmap->behind_writes_used)
1685 			bitmap->behind_writes_used = bw;
1686 
1687 		pr_debug("inc write-behind count %d/%lu\n",
1688 			 bw, bitmap->mddev->bitmap_info.max_write_behind);
1689 	}
1690 
1691 	while (sectors) {
1692 		sector_t blocks;
1693 		bitmap_counter_t *bmc;
1694 
1695 		spin_lock_irq(&bitmap->counts.lock);
1696 		bmc = md_bitmap_get_counter(&bitmap->counts, offset, &blocks, 1);
1697 		if (!bmc) {
1698 			spin_unlock_irq(&bitmap->counts.lock);
1699 			return 0;
1700 		}
1701 
1702 		if (unlikely(COUNTER(*bmc) == COUNTER_MAX)) {
1703 			DEFINE_WAIT(__wait);
1704 			/* note that it is safe to do the prepare_to_wait
1705 			 * after the test as long as we do it before dropping
1706 			 * the spinlock.
1707 			 */
1708 			prepare_to_wait(&bitmap->overflow_wait, &__wait,
1709 					TASK_UNINTERRUPTIBLE);
1710 			spin_unlock_irq(&bitmap->counts.lock);
1711 			schedule();
1712 			finish_wait(&bitmap->overflow_wait, &__wait);
1713 			continue;
1714 		}
1715 
1716 		switch (*bmc) {
1717 		case 0:
1718 			md_bitmap_file_set_bit(bitmap, offset);
1719 			md_bitmap_count_page(&bitmap->counts, offset, 1);
1720 			fallthrough;
1721 		case 1:
1722 			*bmc = 2;
1723 		}
1724 
1725 		(*bmc)++;
1726 
1727 		spin_unlock_irq(&bitmap->counts.lock);
1728 
1729 		offset += blocks;
1730 		if (sectors > blocks)
1731 			sectors -= blocks;
1732 		else
1733 			sectors = 0;
1734 	}
1735 	return 0;
1736 }
1737 
1738 static void bitmap_endwrite(struct mddev *mddev, sector_t offset,
1739 			    unsigned long sectors, bool success, bool behind)
1740 {
1741 	struct bitmap *bitmap = mddev->bitmap;
1742 
1743 	if (!bitmap)
1744 		return;
1745 
1746 	if (behind) {
1747 		if (atomic_dec_and_test(&bitmap->behind_writes))
1748 			wake_up(&bitmap->behind_wait);
1749 		pr_debug("dec write-behind count %d/%lu\n",
1750 			 atomic_read(&bitmap->behind_writes),
1751 			 bitmap->mddev->bitmap_info.max_write_behind);
1752 	}
1753 
1754 	while (sectors) {
1755 		sector_t blocks;
1756 		unsigned long flags;
1757 		bitmap_counter_t *bmc;
1758 
1759 		spin_lock_irqsave(&bitmap->counts.lock, flags);
1760 		bmc = md_bitmap_get_counter(&bitmap->counts, offset, &blocks, 0);
1761 		if (!bmc) {
1762 			spin_unlock_irqrestore(&bitmap->counts.lock, flags);
1763 			return;
1764 		}
1765 
1766 		if (success && !bitmap->mddev->degraded &&
1767 		    bitmap->events_cleared < bitmap->mddev->events) {
1768 			bitmap->events_cleared = bitmap->mddev->events;
1769 			bitmap->need_sync = 1;
1770 			sysfs_notify_dirent_safe(bitmap->sysfs_can_clear);
1771 		}
1772 
1773 		if (!success && !NEEDED(*bmc))
1774 			*bmc |= NEEDED_MASK;
1775 
1776 		if (COUNTER(*bmc) == COUNTER_MAX)
1777 			wake_up(&bitmap->overflow_wait);
1778 
1779 		(*bmc)--;
1780 		if (*bmc <= 2) {
1781 			md_bitmap_set_pending(&bitmap->counts, offset);
1782 			bitmap->allclean = 0;
1783 		}
1784 		spin_unlock_irqrestore(&bitmap->counts.lock, flags);
1785 		offset += blocks;
1786 		if (sectors > blocks)
1787 			sectors -= blocks;
1788 		else
1789 			sectors = 0;
1790 	}
1791 }
1792 
1793 static bool __bitmap_start_sync(struct bitmap *bitmap, sector_t offset,
1794 				sector_t *blocks, bool degraded)
1795 {
1796 	bitmap_counter_t *bmc;
1797 	bool rv;
1798 
1799 	if (bitmap == NULL) {/* FIXME or bitmap set as 'failed' */
1800 		*blocks = 1024;
1801 		return true; /* always resync if no bitmap */
1802 	}
1803 	spin_lock_irq(&bitmap->counts.lock);
1804 
1805 	rv = false;
1806 	bmc = md_bitmap_get_counter(&bitmap->counts, offset, blocks, 0);
1807 	if (bmc) {
1808 		/* locked */
1809 		if (RESYNC(*bmc)) {
1810 			rv = true;
1811 		} else if (NEEDED(*bmc)) {
1812 			rv = true;
1813 			if (!degraded) { /* don't set/clear bits if degraded */
1814 				*bmc |= RESYNC_MASK;
1815 				*bmc &= ~NEEDED_MASK;
1816 			}
1817 		}
1818 	}
1819 	spin_unlock_irq(&bitmap->counts.lock);
1820 
1821 	return rv;
1822 }
1823 
1824 static bool bitmap_start_sync(struct mddev *mddev, sector_t offset,
1825 			      sector_t *blocks, bool degraded)
1826 {
1827 	/* bitmap_start_sync must always report on multiples of whole
1828 	 * pages, otherwise resync (which is very PAGE_SIZE based) will
1829 	 * get confused.
1830 	 * So call __bitmap_start_sync repeatedly (if needed) until
1831 	 * At least PAGE_SIZE>>9 blocks are covered.
1832 	 * Return the 'or' of the result.
1833 	 */
1834 	bool rv = false;
1835 	sector_t blocks1;
1836 
1837 	*blocks = 0;
1838 	while (*blocks < (PAGE_SIZE>>9)) {
1839 		rv |= __bitmap_start_sync(mddev->bitmap, offset,
1840 					  &blocks1, degraded);
1841 		offset += blocks1;
1842 		*blocks += blocks1;
1843 	}
1844 
1845 	return rv;
1846 }
1847 
1848 static void __bitmap_end_sync(struct bitmap *bitmap, sector_t offset,
1849 			      sector_t *blocks, bool aborted)
1850 {
1851 	bitmap_counter_t *bmc;
1852 	unsigned long flags;
1853 
1854 	if (bitmap == NULL) {
1855 		*blocks = 1024;
1856 		return;
1857 	}
1858 	spin_lock_irqsave(&bitmap->counts.lock, flags);
1859 	bmc = md_bitmap_get_counter(&bitmap->counts, offset, blocks, 0);
1860 	if (bmc == NULL)
1861 		goto unlock;
1862 	/* locked */
1863 	if (RESYNC(*bmc)) {
1864 		*bmc &= ~RESYNC_MASK;
1865 
1866 		if (!NEEDED(*bmc) && aborted)
1867 			*bmc |= NEEDED_MASK;
1868 		else {
1869 			if (*bmc <= 2) {
1870 				md_bitmap_set_pending(&bitmap->counts, offset);
1871 				bitmap->allclean = 0;
1872 			}
1873 		}
1874 	}
1875  unlock:
1876 	spin_unlock_irqrestore(&bitmap->counts.lock, flags);
1877 }
1878 
1879 static void bitmap_end_sync(struct mddev *mddev, sector_t offset,
1880 			    sector_t *blocks)
1881 {
1882 	__bitmap_end_sync(mddev->bitmap, offset, blocks, true);
1883 }
1884 
1885 static void bitmap_close_sync(struct mddev *mddev)
1886 {
1887 	/* Sync has finished, and any bitmap chunks that weren't synced
1888 	 * properly have been aborted.  It remains to us to clear the
1889 	 * RESYNC bit wherever it is still on
1890 	 */
1891 	sector_t sector = 0;
1892 	sector_t blocks;
1893 	struct bitmap *bitmap = mddev->bitmap;
1894 
1895 	if (!bitmap)
1896 		return;
1897 
1898 	while (sector < bitmap->mddev->resync_max_sectors) {
1899 		__bitmap_end_sync(bitmap, sector, &blocks, false);
1900 		sector += blocks;
1901 	}
1902 }
1903 
1904 static void bitmap_cond_end_sync(struct mddev *mddev, sector_t sector,
1905 				 bool force)
1906 {
1907 	sector_t s = 0;
1908 	sector_t blocks;
1909 	struct bitmap *bitmap = mddev->bitmap;
1910 
1911 	if (!bitmap)
1912 		return;
1913 	if (sector == 0) {
1914 		bitmap->last_end_sync = jiffies;
1915 		return;
1916 	}
1917 	if (!force && time_before(jiffies, (bitmap->last_end_sync
1918 				  + bitmap->mddev->bitmap_info.daemon_sleep)))
1919 		return;
1920 	wait_event(bitmap->mddev->recovery_wait,
1921 		   atomic_read(&bitmap->mddev->recovery_active) == 0);
1922 
1923 	bitmap->mddev->curr_resync_completed = sector;
1924 	set_bit(MD_SB_CHANGE_CLEAN, &bitmap->mddev->sb_flags);
1925 	sector &= ~((1ULL << bitmap->counts.chunkshift) - 1);
1926 	s = 0;
1927 	while (s < sector && s < bitmap->mddev->resync_max_sectors) {
1928 		__bitmap_end_sync(bitmap, s, &blocks, false);
1929 		s += blocks;
1930 	}
1931 	bitmap->last_end_sync = jiffies;
1932 	sysfs_notify_dirent_safe(bitmap->mddev->sysfs_completed);
1933 }
1934 
1935 static void bitmap_sync_with_cluster(struct mddev *mddev,
1936 				     sector_t old_lo, sector_t old_hi,
1937 				     sector_t new_lo, sector_t new_hi)
1938 {
1939 	struct bitmap *bitmap = mddev->bitmap;
1940 	sector_t sector, blocks = 0;
1941 
1942 	for (sector = old_lo; sector < new_lo; ) {
1943 		__bitmap_end_sync(bitmap, sector, &blocks, false);
1944 		sector += blocks;
1945 	}
1946 	WARN((blocks > new_lo) && old_lo, "alignment is not correct for lo\n");
1947 
1948 	for (sector = old_hi; sector < new_hi; ) {
1949 		bitmap_start_sync(mddev, sector, &blocks, false);
1950 		sector += blocks;
1951 	}
1952 	WARN((blocks > new_hi) && old_hi, "alignment is not correct for hi\n");
1953 }
1954 
1955 static void md_bitmap_set_memory_bits(struct bitmap *bitmap, sector_t offset, int needed)
1956 {
1957 	/* For each chunk covered by any of these sectors, set the
1958 	 * counter to 2 and possibly set resync_needed.  They should all
1959 	 * be 0 at this point
1960 	 */
1961 
1962 	sector_t secs;
1963 	bitmap_counter_t *bmc;
1964 	spin_lock_irq(&bitmap->counts.lock);
1965 	bmc = md_bitmap_get_counter(&bitmap->counts, offset, &secs, 1);
1966 	if (!bmc) {
1967 		spin_unlock_irq(&bitmap->counts.lock);
1968 		return;
1969 	}
1970 	if (!*bmc) {
1971 		*bmc = 2;
1972 		md_bitmap_count_page(&bitmap->counts, offset, 1);
1973 		md_bitmap_set_pending(&bitmap->counts, offset);
1974 		bitmap->allclean = 0;
1975 	}
1976 	if (needed)
1977 		*bmc |= NEEDED_MASK;
1978 	spin_unlock_irq(&bitmap->counts.lock);
1979 }
1980 
1981 /* dirty the memory and file bits for bitmap chunks "s" to "e" */
1982 static void bitmap_dirty_bits(struct mddev *mddev, unsigned long s,
1983 			      unsigned long e)
1984 {
1985 	unsigned long chunk;
1986 	struct bitmap *bitmap = mddev->bitmap;
1987 
1988 	if (!bitmap)
1989 		return;
1990 
1991 	for (chunk = s; chunk <= e; chunk++) {
1992 		sector_t sec = (sector_t)chunk << bitmap->counts.chunkshift;
1993 
1994 		md_bitmap_set_memory_bits(bitmap, sec, 1);
1995 		md_bitmap_file_set_bit(bitmap, sec);
1996 		if (sec < bitmap->mddev->recovery_cp)
1997 			/* We are asserting that the array is dirty,
1998 			 * so move the recovery_cp address back so
1999 			 * that it is obvious that it is dirty
2000 			 */
2001 			bitmap->mddev->recovery_cp = sec;
2002 	}
2003 }
2004 
2005 static void bitmap_flush(struct mddev *mddev)
2006 {
2007 	struct bitmap *bitmap = mddev->bitmap;
2008 	long sleep;
2009 
2010 	if (!bitmap) /* there was no bitmap */
2011 		return;
2012 
2013 	/* run the daemon_work three time to ensure everything is flushed
2014 	 * that can be
2015 	 */
2016 	sleep = mddev->bitmap_info.daemon_sleep * 2;
2017 	bitmap->daemon_lastrun -= sleep;
2018 	bitmap_daemon_work(mddev);
2019 	bitmap->daemon_lastrun -= sleep;
2020 	bitmap_daemon_work(mddev);
2021 	bitmap->daemon_lastrun -= sleep;
2022 	bitmap_daemon_work(mddev);
2023 	if (mddev->bitmap_info.external)
2024 		md_super_wait(mddev);
2025 	bitmap_update_sb(bitmap);
2026 }
2027 
2028 static void md_bitmap_free(void *data)
2029 {
2030 	unsigned long k, pages;
2031 	struct bitmap_page *bp;
2032 	struct bitmap *bitmap = data;
2033 
2034 	if (!bitmap) /* there was no bitmap */
2035 		return;
2036 
2037 	if (bitmap->sysfs_can_clear)
2038 		sysfs_put(bitmap->sysfs_can_clear);
2039 
2040 	if (mddev_is_clustered(bitmap->mddev) && bitmap->mddev->cluster_info &&
2041 		bitmap->cluster_slot == md_cluster_ops->slot_number(bitmap->mddev))
2042 		md_cluster_stop(bitmap->mddev);
2043 
2044 	/* Shouldn't be needed - but just in case.... */
2045 	wait_event(bitmap->write_wait,
2046 		   atomic_read(&bitmap->pending_writes) == 0);
2047 
2048 	/* release the bitmap file  */
2049 	md_bitmap_file_unmap(&bitmap->storage);
2050 
2051 	bp = bitmap->counts.bp;
2052 	pages = bitmap->counts.pages;
2053 
2054 	/* free all allocated memory */
2055 
2056 	if (bp) /* deallocate the page memory */
2057 		for (k = 0; k < pages; k++)
2058 			if (bp[k].map && !bp[k].hijacked)
2059 				kfree(bp[k].map);
2060 	kfree(bp);
2061 	kfree(bitmap);
2062 }
2063 
2064 static void bitmap_wait_behind_writes(struct mddev *mddev)
2065 {
2066 	struct bitmap *bitmap = mddev->bitmap;
2067 
2068 	/* wait for behind writes to complete */
2069 	if (bitmap && atomic_read(&bitmap->behind_writes) > 0) {
2070 		pr_debug("md:%s: behind writes in progress - waiting to stop.\n",
2071 			 mdname(mddev));
2072 		/* need to kick something here to make sure I/O goes? */
2073 		wait_event(bitmap->behind_wait,
2074 			   atomic_read(&bitmap->behind_writes) == 0);
2075 	}
2076 }
2077 
2078 static void bitmap_destroy(struct mddev *mddev)
2079 {
2080 	struct bitmap *bitmap = mddev->bitmap;
2081 
2082 	if (!bitmap) /* there was no bitmap */
2083 		return;
2084 
2085 	bitmap_wait_behind_writes(mddev);
2086 	if (!mddev->serialize_policy)
2087 		mddev_destroy_serial_pool(mddev, NULL);
2088 
2089 	mutex_lock(&mddev->bitmap_info.mutex);
2090 	spin_lock(&mddev->lock);
2091 	mddev->bitmap = NULL; /* disconnect from the md device */
2092 	spin_unlock(&mddev->lock);
2093 	mutex_unlock(&mddev->bitmap_info.mutex);
2094 	mddev_set_timeout(mddev, MAX_SCHEDULE_TIMEOUT, true);
2095 
2096 	md_bitmap_free(bitmap);
2097 }
2098 
2099 /*
2100  * initialize the bitmap structure
2101  * if this returns an error, bitmap_destroy must be called to do clean up
2102  * once mddev->bitmap is set
2103  */
2104 static struct bitmap *__bitmap_create(struct mddev *mddev, int slot)
2105 {
2106 	struct bitmap *bitmap;
2107 	sector_t blocks = mddev->resync_max_sectors;
2108 	struct file *file = mddev->bitmap_info.file;
2109 	int err;
2110 	struct kernfs_node *bm = NULL;
2111 
2112 	BUILD_BUG_ON(sizeof(bitmap_super_t) != 256);
2113 
2114 	BUG_ON(file && mddev->bitmap_info.offset);
2115 
2116 	if (test_bit(MD_HAS_JOURNAL, &mddev->flags)) {
2117 		pr_notice("md/raid:%s: array with journal cannot have bitmap\n",
2118 			  mdname(mddev));
2119 		return ERR_PTR(-EBUSY);
2120 	}
2121 
2122 	bitmap = kzalloc(sizeof(*bitmap), GFP_KERNEL);
2123 	if (!bitmap)
2124 		return ERR_PTR(-ENOMEM);
2125 
2126 	spin_lock_init(&bitmap->counts.lock);
2127 	atomic_set(&bitmap->pending_writes, 0);
2128 	init_waitqueue_head(&bitmap->write_wait);
2129 	init_waitqueue_head(&bitmap->overflow_wait);
2130 	init_waitqueue_head(&bitmap->behind_wait);
2131 
2132 	bitmap->mddev = mddev;
2133 	bitmap->cluster_slot = slot;
2134 
2135 	if (mddev->kobj.sd)
2136 		bm = sysfs_get_dirent(mddev->kobj.sd, "bitmap");
2137 	if (bm) {
2138 		bitmap->sysfs_can_clear = sysfs_get_dirent(bm, "can_clear");
2139 		sysfs_put(bm);
2140 	} else
2141 		bitmap->sysfs_can_clear = NULL;
2142 
2143 	bitmap->storage.file = file;
2144 	if (file) {
2145 		get_file(file);
2146 		/* As future accesses to this file will use bmap,
2147 		 * and bypass the page cache, we must sync the file
2148 		 * first.
2149 		 */
2150 		vfs_fsync(file, 1);
2151 	}
2152 	/* read superblock from bitmap file (this sets mddev->bitmap_info.chunksize) */
2153 	if (!mddev->bitmap_info.external) {
2154 		/*
2155 		 * If 'MD_ARRAY_FIRST_USE' is set, then device-mapper is
2156 		 * instructing us to create a new on-disk bitmap instance.
2157 		 */
2158 		if (test_and_clear_bit(MD_ARRAY_FIRST_USE, &mddev->flags))
2159 			err = md_bitmap_new_disk_sb(bitmap);
2160 		else
2161 			err = md_bitmap_read_sb(bitmap);
2162 	} else {
2163 		err = 0;
2164 		if (mddev->bitmap_info.chunksize == 0 ||
2165 		    mddev->bitmap_info.daemon_sleep == 0)
2166 			/* chunksize and time_base need to be
2167 			 * set first. */
2168 			err = -EINVAL;
2169 	}
2170 	if (err)
2171 		goto error;
2172 
2173 	bitmap->daemon_lastrun = jiffies;
2174 	err = __bitmap_resize(bitmap, blocks, mddev->bitmap_info.chunksize,
2175 			      true);
2176 	if (err)
2177 		goto error;
2178 
2179 	pr_debug("created bitmap (%lu pages) for device %s\n",
2180 		 bitmap->counts.pages, bmname(bitmap));
2181 
2182 	err = test_bit(BITMAP_WRITE_ERROR, &bitmap->flags) ? -EIO : 0;
2183 	if (err)
2184 		goto error;
2185 
2186 	return bitmap;
2187  error:
2188 	md_bitmap_free(bitmap);
2189 	return ERR_PTR(err);
2190 }
2191 
2192 static int bitmap_create(struct mddev *mddev, int slot)
2193 {
2194 	struct bitmap *bitmap = __bitmap_create(mddev, slot);
2195 
2196 	if (IS_ERR(bitmap))
2197 		return PTR_ERR(bitmap);
2198 
2199 	mddev->bitmap = bitmap;
2200 	return 0;
2201 }
2202 
2203 static int bitmap_load(struct mddev *mddev)
2204 {
2205 	int err = 0;
2206 	sector_t start = 0;
2207 	sector_t sector = 0;
2208 	struct bitmap *bitmap = mddev->bitmap;
2209 	struct md_rdev *rdev;
2210 
2211 	if (!bitmap)
2212 		goto out;
2213 
2214 	rdev_for_each(rdev, mddev)
2215 		mddev_create_serial_pool(mddev, rdev);
2216 
2217 	if (mddev_is_clustered(mddev))
2218 		md_cluster_ops->load_bitmaps(mddev, mddev->bitmap_info.nodes);
2219 
2220 	/* Clear out old bitmap info first:  Either there is none, or we
2221 	 * are resuming after someone else has possibly changed things,
2222 	 * so we should forget old cached info.
2223 	 * All chunks should be clean, but some might need_sync.
2224 	 */
2225 	while (sector < mddev->resync_max_sectors) {
2226 		sector_t blocks;
2227 		bitmap_start_sync(mddev, sector, &blocks, false);
2228 		sector += blocks;
2229 	}
2230 	bitmap_close_sync(mddev);
2231 
2232 	if (mddev->degraded == 0
2233 	    || bitmap->events_cleared == mddev->events)
2234 		/* no need to keep dirty bits to optimise a
2235 		 * re-add of a missing device */
2236 		start = mddev->recovery_cp;
2237 
2238 	mutex_lock(&mddev->bitmap_info.mutex);
2239 	err = md_bitmap_init_from_disk(bitmap, start);
2240 	mutex_unlock(&mddev->bitmap_info.mutex);
2241 
2242 	if (err)
2243 		goto out;
2244 	clear_bit(BITMAP_STALE, &bitmap->flags);
2245 
2246 	/* Kick recovery in case any bits were set */
2247 	set_bit(MD_RECOVERY_NEEDED, &bitmap->mddev->recovery);
2248 
2249 	mddev_set_timeout(mddev, mddev->bitmap_info.daemon_sleep, true);
2250 	md_wakeup_thread(mddev->thread);
2251 
2252 	bitmap_update_sb(bitmap);
2253 
2254 	if (test_bit(BITMAP_WRITE_ERROR, &bitmap->flags))
2255 		err = -EIO;
2256 out:
2257 	return err;
2258 }
2259 
2260 /* caller need to free returned bitmap with md_bitmap_free() */
2261 static void *bitmap_get_from_slot(struct mddev *mddev, int slot)
2262 {
2263 	int rv = 0;
2264 	struct bitmap *bitmap;
2265 
2266 	bitmap = __bitmap_create(mddev, slot);
2267 	if (IS_ERR(bitmap)) {
2268 		rv = PTR_ERR(bitmap);
2269 		return ERR_PTR(rv);
2270 	}
2271 
2272 	rv = md_bitmap_init_from_disk(bitmap, 0);
2273 	if (rv) {
2274 		md_bitmap_free(bitmap);
2275 		return ERR_PTR(rv);
2276 	}
2277 
2278 	return bitmap;
2279 }
2280 
2281 /* Loads the bitmap associated with slot and copies the resync information
2282  * to our bitmap
2283  */
2284 static int bitmap_copy_from_slot(struct mddev *mddev, int slot, sector_t *low,
2285 				 sector_t *high, bool clear_bits)
2286 {
2287 	int rv = 0, i, j;
2288 	sector_t block, lo = 0, hi = 0;
2289 	struct bitmap_counts *counts;
2290 	struct bitmap *bitmap;
2291 
2292 	bitmap = bitmap_get_from_slot(mddev, slot);
2293 	if (IS_ERR(bitmap)) {
2294 		pr_err("%s can't get bitmap from slot %d\n", __func__, slot);
2295 		return -1;
2296 	}
2297 
2298 	counts = &bitmap->counts;
2299 	for (j = 0; j < counts->chunks; j++) {
2300 		block = (sector_t)j << counts->chunkshift;
2301 		if (md_bitmap_file_test_bit(bitmap, block)) {
2302 			if (!lo)
2303 				lo = block;
2304 			hi = block;
2305 			md_bitmap_file_clear_bit(bitmap, block);
2306 			md_bitmap_set_memory_bits(mddev->bitmap, block, 1);
2307 			md_bitmap_file_set_bit(mddev->bitmap, block);
2308 		}
2309 	}
2310 
2311 	if (clear_bits) {
2312 		bitmap_update_sb(bitmap);
2313 		/* BITMAP_PAGE_PENDING is set, but bitmap_unplug needs
2314 		 * BITMAP_PAGE_DIRTY or _NEEDWRITE to write ... */
2315 		for (i = 0; i < bitmap->storage.file_pages; i++)
2316 			if (test_page_attr(bitmap, i, BITMAP_PAGE_PENDING))
2317 				set_page_attr(bitmap, i, BITMAP_PAGE_NEEDWRITE);
2318 		__bitmap_unplug(bitmap);
2319 	}
2320 	__bitmap_unplug(mddev->bitmap);
2321 	*low = lo;
2322 	*high = hi;
2323 	md_bitmap_free(bitmap);
2324 
2325 	return rv;
2326 }
2327 
2328 static void bitmap_set_pages(void *data, unsigned long pages)
2329 {
2330 	struct bitmap *bitmap = data;
2331 
2332 	bitmap->counts.pages = pages;
2333 }
2334 
2335 static int bitmap_get_stats(void *data, struct md_bitmap_stats *stats)
2336 {
2337 	struct bitmap_storage *storage;
2338 	struct bitmap_counts *counts;
2339 	struct bitmap *bitmap = data;
2340 	bitmap_super_t *sb;
2341 
2342 	if (!bitmap)
2343 		return -ENOENT;
2344 
2345 	sb = kmap_local_page(bitmap->storage.sb_page);
2346 	stats->sync_size = le64_to_cpu(sb->sync_size);
2347 	kunmap_local(sb);
2348 
2349 	counts = &bitmap->counts;
2350 	stats->missing_pages = counts->missing_pages;
2351 	stats->pages = counts->pages;
2352 
2353 	storage = &bitmap->storage;
2354 	stats->file_pages = storage->file_pages;
2355 	stats->file = storage->file;
2356 
2357 	stats->behind_writes = atomic_read(&bitmap->behind_writes);
2358 	stats->behind_wait = wq_has_sleeper(&bitmap->behind_wait);
2359 	stats->events_cleared = bitmap->events_cleared;
2360 	return 0;
2361 }
2362 
2363 static int __bitmap_resize(struct bitmap *bitmap, sector_t blocks,
2364 			   int chunksize, bool init)
2365 {
2366 	/* If chunk_size is 0, choose an appropriate chunk size.
2367 	 * Then possibly allocate new storage space.
2368 	 * Then quiesce, copy bits, replace bitmap, and re-start
2369 	 *
2370 	 * This function is called both to set up the initial bitmap
2371 	 * and to resize the bitmap while the array is active.
2372 	 * If this happens as a result of the array being resized,
2373 	 * chunksize will be zero, and we need to choose a suitable
2374 	 * chunksize, otherwise we use what we are given.
2375 	 */
2376 	struct bitmap_storage store;
2377 	struct bitmap_counts old_counts;
2378 	unsigned long chunks;
2379 	sector_t block;
2380 	sector_t old_blocks, new_blocks;
2381 	int chunkshift;
2382 	int ret = 0;
2383 	long pages;
2384 	struct bitmap_page *new_bp;
2385 
2386 	if (bitmap->storage.file && !init) {
2387 		pr_info("md: cannot resize file-based bitmap\n");
2388 		return -EINVAL;
2389 	}
2390 
2391 	if (chunksize == 0) {
2392 		/* If there is enough space, leave the chunk size unchanged,
2393 		 * else increase by factor of two until there is enough space.
2394 		 */
2395 		long bytes;
2396 		long space = bitmap->mddev->bitmap_info.space;
2397 
2398 		if (space == 0) {
2399 			/* We don't know how much space there is, so limit
2400 			 * to current size - in sectors.
2401 			 */
2402 			bytes = DIV_ROUND_UP(bitmap->counts.chunks, 8);
2403 			if (!bitmap->mddev->bitmap_info.external)
2404 				bytes += sizeof(bitmap_super_t);
2405 			space = DIV_ROUND_UP(bytes, 512);
2406 			bitmap->mddev->bitmap_info.space = space;
2407 		}
2408 		chunkshift = bitmap->counts.chunkshift;
2409 		chunkshift--;
2410 		do {
2411 			/* 'chunkshift' is shift from block size to chunk size */
2412 			chunkshift++;
2413 			chunks = DIV_ROUND_UP_SECTOR_T(blocks, 1 << chunkshift);
2414 			bytes = DIV_ROUND_UP(chunks, 8);
2415 			if (!bitmap->mddev->bitmap_info.external)
2416 				bytes += sizeof(bitmap_super_t);
2417 		} while (bytes > (space << 9) && (chunkshift + BITMAP_BLOCK_SHIFT) <
2418 			(BITS_PER_BYTE * sizeof(((bitmap_super_t *)0)->chunksize) - 1));
2419 	} else
2420 		chunkshift = ffz(~chunksize) - BITMAP_BLOCK_SHIFT;
2421 
2422 	chunks = DIV_ROUND_UP_SECTOR_T(blocks, 1 << chunkshift);
2423 	memset(&store, 0, sizeof(store));
2424 	if (bitmap->mddev->bitmap_info.offset || bitmap->mddev->bitmap_info.file)
2425 		ret = md_bitmap_storage_alloc(&store, chunks,
2426 					      !bitmap->mddev->bitmap_info.external,
2427 					      mddev_is_clustered(bitmap->mddev)
2428 					      ? bitmap->cluster_slot : 0);
2429 	if (ret) {
2430 		md_bitmap_file_unmap(&store);
2431 		goto err;
2432 	}
2433 
2434 	pages = DIV_ROUND_UP(chunks, PAGE_COUNTER_RATIO);
2435 
2436 	new_bp = kcalloc(pages, sizeof(*new_bp), GFP_KERNEL);
2437 	ret = -ENOMEM;
2438 	if (!new_bp) {
2439 		md_bitmap_file_unmap(&store);
2440 		goto err;
2441 	}
2442 
2443 	if (!init)
2444 		bitmap->mddev->pers->quiesce(bitmap->mddev, 1);
2445 
2446 	store.file = bitmap->storage.file;
2447 	bitmap->storage.file = NULL;
2448 
2449 	if (store.sb_page && bitmap->storage.sb_page)
2450 		memcpy(page_address(store.sb_page),
2451 		       page_address(bitmap->storage.sb_page),
2452 		       sizeof(bitmap_super_t));
2453 	spin_lock_irq(&bitmap->counts.lock);
2454 	md_bitmap_file_unmap(&bitmap->storage);
2455 	bitmap->storage = store;
2456 
2457 	old_counts = bitmap->counts;
2458 	bitmap->counts.bp = new_bp;
2459 	bitmap->counts.pages = pages;
2460 	bitmap->counts.missing_pages = pages;
2461 	bitmap->counts.chunkshift = chunkshift;
2462 	bitmap->counts.chunks = chunks;
2463 	bitmap->mddev->bitmap_info.chunksize = 1UL << (chunkshift +
2464 						     BITMAP_BLOCK_SHIFT);
2465 
2466 	blocks = min(old_counts.chunks << old_counts.chunkshift,
2467 		     chunks << chunkshift);
2468 
2469 	/* For cluster raid, need to pre-allocate bitmap */
2470 	if (mddev_is_clustered(bitmap->mddev)) {
2471 		unsigned long page;
2472 		for (page = 0; page < pages; page++) {
2473 			ret = md_bitmap_checkpage(&bitmap->counts, page, 1, 1);
2474 			if (ret) {
2475 				unsigned long k;
2476 
2477 				/* deallocate the page memory */
2478 				for (k = 0; k < page; k++) {
2479 					kfree(new_bp[k].map);
2480 				}
2481 				kfree(new_bp);
2482 
2483 				/* restore some fields from old_counts */
2484 				bitmap->counts.bp = old_counts.bp;
2485 				bitmap->counts.pages = old_counts.pages;
2486 				bitmap->counts.missing_pages = old_counts.pages;
2487 				bitmap->counts.chunkshift = old_counts.chunkshift;
2488 				bitmap->counts.chunks = old_counts.chunks;
2489 				bitmap->mddev->bitmap_info.chunksize =
2490 					1UL << (old_counts.chunkshift + BITMAP_BLOCK_SHIFT);
2491 				blocks = old_counts.chunks << old_counts.chunkshift;
2492 				pr_warn("Could not pre-allocate in-memory bitmap for cluster raid\n");
2493 				break;
2494 			} else
2495 				bitmap->counts.bp[page].count += 1;
2496 		}
2497 	}
2498 
2499 	for (block = 0; block < blocks; ) {
2500 		bitmap_counter_t *bmc_old, *bmc_new;
2501 		int set;
2502 
2503 		bmc_old = md_bitmap_get_counter(&old_counts, block, &old_blocks, 0);
2504 		set = bmc_old && NEEDED(*bmc_old);
2505 
2506 		if (set) {
2507 			bmc_new = md_bitmap_get_counter(&bitmap->counts, block, &new_blocks, 1);
2508 			if (bmc_new) {
2509 				if (*bmc_new == 0) {
2510 					/* need to set on-disk bits too. */
2511 					sector_t end = block + new_blocks;
2512 					sector_t start = block >> chunkshift;
2513 
2514 					start <<= chunkshift;
2515 					while (start < end) {
2516 						md_bitmap_file_set_bit(bitmap, block);
2517 						start += 1 << chunkshift;
2518 					}
2519 					*bmc_new = 2;
2520 					md_bitmap_count_page(&bitmap->counts, block, 1);
2521 					md_bitmap_set_pending(&bitmap->counts, block);
2522 				}
2523 				*bmc_new |= NEEDED_MASK;
2524 			}
2525 			if (new_blocks < old_blocks)
2526 				old_blocks = new_blocks;
2527 		}
2528 		block += old_blocks;
2529 	}
2530 
2531 	if (bitmap->counts.bp != old_counts.bp) {
2532 		unsigned long k;
2533 		for (k = 0; k < old_counts.pages; k++)
2534 			if (!old_counts.bp[k].hijacked)
2535 				kfree(old_counts.bp[k].map);
2536 		kfree(old_counts.bp);
2537 	}
2538 
2539 	if (!init) {
2540 		int i;
2541 		while (block < (chunks << chunkshift)) {
2542 			bitmap_counter_t *bmc;
2543 			bmc = md_bitmap_get_counter(&bitmap->counts, block, &new_blocks, 1);
2544 			if (bmc) {
2545 				/* new space.  It needs to be resynced, so
2546 				 * we set NEEDED_MASK.
2547 				 */
2548 				if (*bmc == 0) {
2549 					*bmc = NEEDED_MASK | 2;
2550 					md_bitmap_count_page(&bitmap->counts, block, 1);
2551 					md_bitmap_set_pending(&bitmap->counts, block);
2552 				}
2553 			}
2554 			block += new_blocks;
2555 		}
2556 		for (i = 0; i < bitmap->storage.file_pages; i++)
2557 			set_page_attr(bitmap, i, BITMAP_PAGE_DIRTY);
2558 	}
2559 	spin_unlock_irq(&bitmap->counts.lock);
2560 
2561 	if (!init) {
2562 		__bitmap_unplug(bitmap);
2563 		bitmap->mddev->pers->quiesce(bitmap->mddev, 0);
2564 	}
2565 	ret = 0;
2566 err:
2567 	return ret;
2568 }
2569 
2570 static int bitmap_resize(struct mddev *mddev, sector_t blocks, int chunksize,
2571 			 bool init)
2572 {
2573 	struct bitmap *bitmap = mddev->bitmap;
2574 
2575 	if (!bitmap)
2576 		return 0;
2577 
2578 	return __bitmap_resize(bitmap, blocks, chunksize, init);
2579 }
2580 
2581 static ssize_t
2582 location_show(struct mddev *mddev, char *page)
2583 {
2584 	ssize_t len;
2585 	if (mddev->bitmap_info.file)
2586 		len = sprintf(page, "file");
2587 	else if (mddev->bitmap_info.offset)
2588 		len = sprintf(page, "%+lld", (long long)mddev->bitmap_info.offset);
2589 	else
2590 		len = sprintf(page, "none");
2591 	len += sprintf(page+len, "\n");
2592 	return len;
2593 }
2594 
2595 static ssize_t
2596 location_store(struct mddev *mddev, const char *buf, size_t len)
2597 {
2598 	int rv;
2599 
2600 	rv = mddev_suspend_and_lock(mddev);
2601 	if (rv)
2602 		return rv;
2603 
2604 	if (mddev->pers) {
2605 		if (mddev->recovery || mddev->sync_thread) {
2606 			rv = -EBUSY;
2607 			goto out;
2608 		}
2609 	}
2610 
2611 	if (mddev->bitmap || mddev->bitmap_info.file ||
2612 	    mddev->bitmap_info.offset) {
2613 		/* bitmap already configured.  Only option is to clear it */
2614 		if (strncmp(buf, "none", 4) != 0) {
2615 			rv = -EBUSY;
2616 			goto out;
2617 		}
2618 
2619 		bitmap_destroy(mddev);
2620 		mddev->bitmap_info.offset = 0;
2621 		if (mddev->bitmap_info.file) {
2622 			struct file *f = mddev->bitmap_info.file;
2623 			mddev->bitmap_info.file = NULL;
2624 			fput(f);
2625 		}
2626 	} else {
2627 		/* No bitmap, OK to set a location */
2628 		long long offset;
2629 
2630 		if (strncmp(buf, "none", 4) == 0)
2631 			/* nothing to be done */;
2632 		else if (strncmp(buf, "file:", 5) == 0) {
2633 			/* Not supported yet */
2634 			rv = -EINVAL;
2635 			goto out;
2636 		} else {
2637 			if (buf[0] == '+')
2638 				rv = kstrtoll(buf+1, 10, &offset);
2639 			else
2640 				rv = kstrtoll(buf, 10, &offset);
2641 			if (rv)
2642 				goto out;
2643 			if (offset == 0) {
2644 				rv = -EINVAL;
2645 				goto out;
2646 			}
2647 			if (mddev->bitmap_info.external == 0 &&
2648 			    mddev->major_version == 0 &&
2649 			    offset != mddev->bitmap_info.default_offset) {
2650 				rv = -EINVAL;
2651 				goto out;
2652 			}
2653 
2654 			mddev->bitmap_info.offset = offset;
2655 			rv = bitmap_create(mddev, -1);
2656 			if (rv)
2657 				goto out;
2658 
2659 			rv = bitmap_load(mddev);
2660 			if (rv) {
2661 				mddev->bitmap_info.offset = 0;
2662 				bitmap_destroy(mddev);
2663 				goto out;
2664 			}
2665 		}
2666 	}
2667 	if (!mddev->external) {
2668 		/* Ensure new bitmap info is stored in
2669 		 * metadata promptly.
2670 		 */
2671 		set_bit(MD_SB_CHANGE_DEVS, &mddev->sb_flags);
2672 		md_wakeup_thread(mddev->thread);
2673 	}
2674 	rv = 0;
2675 out:
2676 	mddev_unlock_and_resume(mddev);
2677 	if (rv)
2678 		return rv;
2679 	return len;
2680 }
2681 
2682 static struct md_sysfs_entry bitmap_location =
2683 __ATTR(location, S_IRUGO|S_IWUSR, location_show, location_store);
2684 
2685 /* 'bitmap/space' is the space available at 'location' for the
2686  * bitmap.  This allows the kernel to know when it is safe to
2687  * resize the bitmap to match a resized array.
2688  */
2689 static ssize_t
2690 space_show(struct mddev *mddev, char *page)
2691 {
2692 	return sprintf(page, "%lu\n", mddev->bitmap_info.space);
2693 }
2694 
2695 static ssize_t
2696 space_store(struct mddev *mddev, const char *buf, size_t len)
2697 {
2698 	struct bitmap *bitmap;
2699 	unsigned long sectors;
2700 	int rv;
2701 
2702 	rv = kstrtoul(buf, 10, &sectors);
2703 	if (rv)
2704 		return rv;
2705 
2706 	if (sectors == 0)
2707 		return -EINVAL;
2708 
2709 	bitmap = mddev->bitmap;
2710 	if (bitmap && sectors < (bitmap->storage.bytes + 511) >> 9)
2711 		return -EFBIG; /* Bitmap is too big for this small space */
2712 
2713 	/* could make sure it isn't too big, but that isn't really
2714 	 * needed - user-space should be careful.
2715 	 */
2716 	mddev->bitmap_info.space = sectors;
2717 	return len;
2718 }
2719 
2720 static struct md_sysfs_entry bitmap_space =
2721 __ATTR(space, S_IRUGO|S_IWUSR, space_show, space_store);
2722 
2723 static ssize_t
2724 timeout_show(struct mddev *mddev, char *page)
2725 {
2726 	ssize_t len;
2727 	unsigned long secs = mddev->bitmap_info.daemon_sleep / HZ;
2728 	unsigned long jifs = mddev->bitmap_info.daemon_sleep % HZ;
2729 
2730 	len = sprintf(page, "%lu", secs);
2731 	if (jifs)
2732 		len += sprintf(page+len, ".%03u", jiffies_to_msecs(jifs));
2733 	len += sprintf(page+len, "\n");
2734 	return len;
2735 }
2736 
2737 static ssize_t
2738 timeout_store(struct mddev *mddev, const char *buf, size_t len)
2739 {
2740 	/* timeout can be set at any time */
2741 	unsigned long timeout;
2742 	int rv = strict_strtoul_scaled(buf, &timeout, 4);
2743 	if (rv)
2744 		return rv;
2745 
2746 	/* just to make sure we don't overflow... */
2747 	if (timeout >= LONG_MAX / HZ)
2748 		return -EINVAL;
2749 
2750 	timeout = timeout * HZ / 10000;
2751 
2752 	if (timeout >= MAX_SCHEDULE_TIMEOUT)
2753 		timeout = MAX_SCHEDULE_TIMEOUT-1;
2754 	if (timeout < 1)
2755 		timeout = 1;
2756 
2757 	mddev->bitmap_info.daemon_sleep = timeout;
2758 	mddev_set_timeout(mddev, timeout, false);
2759 	md_wakeup_thread(mddev->thread);
2760 
2761 	return len;
2762 }
2763 
2764 static struct md_sysfs_entry bitmap_timeout =
2765 __ATTR(time_base, S_IRUGO|S_IWUSR, timeout_show, timeout_store);
2766 
2767 static ssize_t
2768 backlog_show(struct mddev *mddev, char *page)
2769 {
2770 	return sprintf(page, "%lu\n", mddev->bitmap_info.max_write_behind);
2771 }
2772 
2773 static ssize_t
2774 backlog_store(struct mddev *mddev, const char *buf, size_t len)
2775 {
2776 	unsigned long backlog;
2777 	unsigned long old_mwb = mddev->bitmap_info.max_write_behind;
2778 	struct md_rdev *rdev;
2779 	bool has_write_mostly = false;
2780 	int rv = kstrtoul(buf, 10, &backlog);
2781 	if (rv)
2782 		return rv;
2783 	if (backlog > COUNTER_MAX)
2784 		return -EINVAL;
2785 
2786 	rv = mddev_suspend_and_lock(mddev);
2787 	if (rv)
2788 		return rv;
2789 
2790 	/*
2791 	 * Without write mostly device, it doesn't make sense to set
2792 	 * backlog for max_write_behind.
2793 	 */
2794 	rdev_for_each(rdev, mddev) {
2795 		if (test_bit(WriteMostly, &rdev->flags)) {
2796 			has_write_mostly = true;
2797 			break;
2798 		}
2799 	}
2800 	if (!has_write_mostly) {
2801 		pr_warn_ratelimited("%s: can't set backlog, no write mostly device available\n",
2802 				    mdname(mddev));
2803 		mddev_unlock(mddev);
2804 		return -EINVAL;
2805 	}
2806 
2807 	mddev->bitmap_info.max_write_behind = backlog;
2808 	if (!backlog && mddev->serial_info_pool) {
2809 		/* serial_info_pool is not needed if backlog is zero */
2810 		if (!mddev->serialize_policy)
2811 			mddev_destroy_serial_pool(mddev, NULL);
2812 	} else if (backlog && !mddev->serial_info_pool) {
2813 		/* serial_info_pool is needed since backlog is not zero */
2814 		rdev_for_each(rdev, mddev)
2815 			mddev_create_serial_pool(mddev, rdev);
2816 	}
2817 	if (old_mwb != backlog)
2818 		bitmap_update_sb(mddev->bitmap);
2819 
2820 	mddev_unlock_and_resume(mddev);
2821 	return len;
2822 }
2823 
2824 static struct md_sysfs_entry bitmap_backlog =
2825 __ATTR(backlog, S_IRUGO|S_IWUSR, backlog_show, backlog_store);
2826 
2827 static ssize_t
2828 chunksize_show(struct mddev *mddev, char *page)
2829 {
2830 	return sprintf(page, "%lu\n", mddev->bitmap_info.chunksize);
2831 }
2832 
2833 static ssize_t
2834 chunksize_store(struct mddev *mddev, const char *buf, size_t len)
2835 {
2836 	/* Can only be changed when no bitmap is active */
2837 	int rv;
2838 	unsigned long csize;
2839 	if (mddev->bitmap)
2840 		return -EBUSY;
2841 	rv = kstrtoul(buf, 10, &csize);
2842 	if (rv)
2843 		return rv;
2844 	if (csize < 512 ||
2845 	    !is_power_of_2(csize))
2846 		return -EINVAL;
2847 	if (BITS_PER_LONG > 32 && csize >= (1ULL << (BITS_PER_BYTE *
2848 		sizeof(((bitmap_super_t *)0)->chunksize))))
2849 		return -EOVERFLOW;
2850 	mddev->bitmap_info.chunksize = csize;
2851 	return len;
2852 }
2853 
2854 static struct md_sysfs_entry bitmap_chunksize =
2855 __ATTR(chunksize, S_IRUGO|S_IWUSR, chunksize_show, chunksize_store);
2856 
2857 static ssize_t metadata_show(struct mddev *mddev, char *page)
2858 {
2859 	if (mddev_is_clustered(mddev))
2860 		return sprintf(page, "clustered\n");
2861 	return sprintf(page, "%s\n", (mddev->bitmap_info.external
2862 				      ? "external" : "internal"));
2863 }
2864 
2865 static ssize_t metadata_store(struct mddev *mddev, const char *buf, size_t len)
2866 {
2867 	if (mddev->bitmap ||
2868 	    mddev->bitmap_info.file ||
2869 	    mddev->bitmap_info.offset)
2870 		return -EBUSY;
2871 	if (strncmp(buf, "external", 8) == 0)
2872 		mddev->bitmap_info.external = 1;
2873 	else if ((strncmp(buf, "internal", 8) == 0) ||
2874 			(strncmp(buf, "clustered", 9) == 0))
2875 		mddev->bitmap_info.external = 0;
2876 	else
2877 		return -EINVAL;
2878 	return len;
2879 }
2880 
2881 static struct md_sysfs_entry bitmap_metadata =
2882 __ATTR(metadata, S_IRUGO|S_IWUSR, metadata_show, metadata_store);
2883 
2884 static ssize_t can_clear_show(struct mddev *mddev, char *page)
2885 {
2886 	int len;
2887 	struct bitmap *bitmap;
2888 
2889 	spin_lock(&mddev->lock);
2890 	bitmap = mddev->bitmap;
2891 	if (bitmap)
2892 		len = sprintf(page, "%s\n", (bitmap->need_sync ? "false" :
2893 								 "true"));
2894 	else
2895 		len = sprintf(page, "\n");
2896 	spin_unlock(&mddev->lock);
2897 	return len;
2898 }
2899 
2900 static ssize_t can_clear_store(struct mddev *mddev, const char *buf, size_t len)
2901 {
2902 	struct bitmap *bitmap = mddev->bitmap;
2903 
2904 	if (!bitmap)
2905 		return -ENOENT;
2906 
2907 	if (strncmp(buf, "false", 5) == 0) {
2908 		bitmap->need_sync = 1;
2909 		return len;
2910 	}
2911 
2912 	if (strncmp(buf, "true", 4) == 0) {
2913 		if (mddev->degraded)
2914 			return -EBUSY;
2915 		bitmap->need_sync = 0;
2916 		return len;
2917 	}
2918 
2919 	return -EINVAL;
2920 }
2921 
2922 static struct md_sysfs_entry bitmap_can_clear =
2923 __ATTR(can_clear, S_IRUGO|S_IWUSR, can_clear_show, can_clear_store);
2924 
2925 static ssize_t
2926 behind_writes_used_show(struct mddev *mddev, char *page)
2927 {
2928 	ssize_t ret;
2929 	struct bitmap *bitmap;
2930 
2931 	spin_lock(&mddev->lock);
2932 	bitmap = mddev->bitmap;
2933 	if (!bitmap)
2934 		ret = sprintf(page, "0\n");
2935 	else
2936 		ret = sprintf(page, "%lu\n", bitmap->behind_writes_used);
2937 	spin_unlock(&mddev->lock);
2938 
2939 	return ret;
2940 }
2941 
2942 static ssize_t
2943 behind_writes_used_reset(struct mddev *mddev, const char *buf, size_t len)
2944 {
2945 	struct bitmap *bitmap = mddev->bitmap;
2946 
2947 	if (bitmap)
2948 		bitmap->behind_writes_used = 0;
2949 	return len;
2950 }
2951 
2952 static struct md_sysfs_entry max_backlog_used =
2953 __ATTR(max_backlog_used, S_IRUGO | S_IWUSR,
2954        behind_writes_used_show, behind_writes_used_reset);
2955 
2956 static struct attribute *md_bitmap_attrs[] = {
2957 	&bitmap_location.attr,
2958 	&bitmap_space.attr,
2959 	&bitmap_timeout.attr,
2960 	&bitmap_backlog.attr,
2961 	&bitmap_chunksize.attr,
2962 	&bitmap_metadata.attr,
2963 	&bitmap_can_clear.attr,
2964 	&max_backlog_used.attr,
2965 	NULL
2966 };
2967 const struct attribute_group md_bitmap_group = {
2968 	.name = "bitmap",
2969 	.attrs = md_bitmap_attrs,
2970 };
2971 
2972 static struct bitmap_operations bitmap_ops = {
2973 	.enabled		= bitmap_enabled,
2974 	.create			= bitmap_create,
2975 	.resize			= bitmap_resize,
2976 	.load			= bitmap_load,
2977 	.destroy		= bitmap_destroy,
2978 	.flush			= bitmap_flush,
2979 	.write_all		= bitmap_write_all,
2980 	.dirty_bits		= bitmap_dirty_bits,
2981 	.unplug			= bitmap_unplug,
2982 	.daemon_work		= bitmap_daemon_work,
2983 	.wait_behind_writes	= bitmap_wait_behind_writes,
2984 
2985 	.startwrite		= bitmap_startwrite,
2986 	.endwrite		= bitmap_endwrite,
2987 	.start_sync		= bitmap_start_sync,
2988 	.end_sync		= bitmap_end_sync,
2989 	.cond_end_sync		= bitmap_cond_end_sync,
2990 	.close_sync		= bitmap_close_sync,
2991 
2992 	.update_sb		= bitmap_update_sb,
2993 	.get_stats		= bitmap_get_stats,
2994 
2995 	.sync_with_cluster	= bitmap_sync_with_cluster,
2996 	.get_from_slot		= bitmap_get_from_slot,
2997 	.copy_from_slot		= bitmap_copy_from_slot,
2998 	.set_pages		= bitmap_set_pages,
2999 	.free			= md_bitmap_free,
3000 };
3001 
3002 void mddev_set_bitmap_ops(struct mddev *mddev)
3003 {
3004 	mddev->bitmap_ops = &bitmap_ops;
3005 }
3006