1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19#include <linux/module.h>
20#include <linux/errno.h>
21#include <linux/slab.h>
22#include <linux/init.h>
23#include <linux/timer.h>
24#include <linux/sched.h>
25#include <linux/list.h>
26#include <linux/file.h>
27#include <linux/mount.h>
28#include <linux/buffer_head.h>
29#include <linux/raid/md.h>
30#include <linux/raid/bitmap.h>
31
32
33
34#define DEBUG 0
35
36#if DEBUG
37
38
39
40#define INJECT_FAULTS_1 0
41#define INJECT_FAULTS_2 0
42#define INJECT_FAULTS_3 0
43#define INJECT_FAULTS_4 0
44#define INJECT_FAULTS_5 0
45#define INJECT_FAULTS_6 0
46
47
48#define INJECT_FATAL_FAULT_1 0
49#define INJECT_FATAL_FAULT_2 0
50#define INJECT_FATAL_FAULT_3 0
51#endif
52
53
54#define DPRINTK(x...) do { } while(0)
55
56#ifndef PRINTK
57# if DEBUG > 0
58# define PRINTK(x...) printk(KERN_DEBUG x)
59# else
60# define PRINTK(x...)
61# endif
62#endif
63
64static inline char * bmname(struct bitmap *bitmap)
65{
66 return bitmap->mddev ? mdname(bitmap->mddev) : "mdX";
67}
68
69
70
71
72
73static unsigned char *bitmap_alloc_page(struct bitmap *bitmap)
74{
75 unsigned char *page;
76
77#ifdef INJECT_FAULTS_1
78 page = NULL;
79#else
80 page = kmalloc(PAGE_SIZE, GFP_NOIO);
81#endif
82 if (!page)
83 printk("%s: bitmap_alloc_page FAILED\n", bmname(bitmap));
84 else
85 PRINTK("%s: bitmap_alloc_page: allocated page at %p\n",
86 bmname(bitmap), page);
87 return page;
88}
89
90
91
92
93static void bitmap_free_page(struct bitmap *bitmap, unsigned char *page)
94{
95 PRINTK("%s: bitmap_free_page: free page %p\n", bmname(bitmap), page);
96 kfree(page);
97}
98
99
100
101
102
103
104
105
106
107
108
109static int bitmap_checkpage(struct bitmap *bitmap, unsigned long page, int create)
110{
111 unsigned char *mappage;
112
113 if (page >= bitmap->pages) {
114 printk(KERN_ALERT
115 "%s: invalid bitmap page request: %lu (> %lu)\n",
116 bmname(bitmap), page, bitmap->pages-1);
117 return -EINVAL;
118 }
119
120
121 if (bitmap->bp[page].hijacked)
122 return 0;
123
124 if (bitmap->bp[page].map)
125 return 0;
126
127 if (!create)
128 return -ENOENT;
129
130 spin_unlock_irq(&bitmap->lock);
131
132
133
134 if ((mappage = bitmap_alloc_page(bitmap)) == NULL) {
135 PRINTK("%s: bitmap map page allocation failed, hijacking\n",
136 bmname(bitmap));
137
138
139 spin_lock_irq(&bitmap->lock);
140 if (!bitmap->bp[page].map)
141 bitmap->bp[page].hijacked = 1;
142 goto out;
143 }
144
145
146
147 spin_lock_irq(&bitmap->lock);
148
149
150
151 if (bitmap->bp[page].map || bitmap->bp[page].hijacked) {
152
153 bitmap_free_page(bitmap, mappage);
154 return 0;
155 }
156
157
158
159 memset(mappage, 0, PAGE_SIZE);
160 bitmap->bp[page].map = mappage;
161 bitmap->missing_pages--;
162out:
163 return 0;
164}
165
166
167
168
169
170static void bitmap_checkfree(struct bitmap *bitmap, unsigned long page)
171{
172 char *ptr;
173
174 if (bitmap->bp[page].count)
175 return;
176
177
178
179 if (bitmap->bp[page].hijacked) {
180 bitmap->bp[page].hijacked = 0;
181 bitmap->bp[page].map = NULL;
182 return;
183 }
184
185
186
187#if 0
188
189
190
191 return;
192#else
193 ptr = bitmap->bp[page].map;
194 bitmap->bp[page].map = NULL;
195 bitmap->missing_pages++;
196 bitmap_free_page(bitmap, ptr);
197 return;
198#endif
199}
200
201
202
203
204
205
206
207char *file_path(struct file *file, char *buf, int count)
208{
209 struct dentry *d;
210 struct vfsmount *v;
211
212 if (!buf)
213 return NULL;
214
215 d = file->f_path.dentry;
216 v = file->f_path.mnt;
217
218 buf = d_path(d, v, buf, count);
219
220 return IS_ERR(buf) ? NULL : buf;
221}
222
223
224
225
226
227
228static struct page *read_sb_page(mddev_t *mddev, long offset, unsigned long index)
229{
230
231
232 mdk_rdev_t *rdev;
233 struct list_head *tmp;
234 struct page *page = alloc_page(GFP_KERNEL);
235 sector_t target;
236
237 if (!page)
238 return ERR_PTR(-ENOMEM);
239
240 ITERATE_RDEV(mddev, rdev, tmp) {
241 if (! test_bit(In_sync, &rdev->flags)
242 || test_bit(Faulty, &rdev->flags))
243 continue;
244
245 target = (rdev->sb_offset << 1) + offset + index * (PAGE_SIZE/512);
246
247 if (sync_page_io(rdev->bdev, target, PAGE_SIZE, page, READ)) {
248 page->index = index;
249 attach_page_buffers(page, NULL);
250
251 return page;
252 }
253 }
254 return ERR_PTR(-EIO);
255
256}
257
258static int write_sb_page(struct bitmap *bitmap, struct page *page, int wait)
259{
260 mdk_rdev_t *rdev;
261 struct list_head *tmp;
262 mddev_t *mddev = bitmap->mddev;
263
264 ITERATE_RDEV(mddev, rdev, tmp)
265 if (test_bit(In_sync, &rdev->flags)
266 && !test_bit(Faulty, &rdev->flags)) {
267 int size = PAGE_SIZE;
268 if (page->index == bitmap->file_pages-1)
269 size = roundup(bitmap->last_page_size,
270 bdev_hardsect_size(rdev->bdev));
271
272
273
274 if (bitmap->offset < 0) {
275
276 if (bitmap->offset
277 + (long)(page->index * (PAGE_SIZE/512))
278 + size/512 > 0)
279
280 return -EINVAL;
281 if (rdev->data_offset + mddev->size*2
282 > rdev->sb_offset*2 + bitmap->offset)
283
284 return -EINVAL;
285 } else if (rdev->sb_offset*2 < rdev->data_offset) {
286
287 if (rdev->sb_offset*2
288 + bitmap->offset
289 + page->index*(PAGE_SIZE/512) + size/512
290 > rdev->data_offset)
291
292 return -EINVAL;
293 } else {
294
295 }
296 md_super_write(mddev, rdev,
297 (rdev->sb_offset<<1) + bitmap->offset
298 + page->index * (PAGE_SIZE/512),
299 size,
300 page);
301 }
302
303 if (wait)
304 md_super_wait(mddev);
305 return 0;
306}
307
308static void bitmap_file_kick(struct bitmap *bitmap);
309
310
311
312static void write_page(struct bitmap *bitmap, struct page *page, int wait)
313{
314 struct buffer_head *bh;
315
316 if (bitmap->file == NULL) {
317 switch (write_sb_page(bitmap, page, wait)) {
318 case -EINVAL:
319 bitmap->flags |= BITMAP_WRITE_ERROR;
320 }
321 } else {
322
323 bh = page_buffers(page);
324
325 while (bh && bh->b_blocknr) {
326 atomic_inc(&bitmap->pending_writes);
327 set_buffer_locked(bh);
328 set_buffer_mapped(bh);
329 submit_bh(WRITE, bh);
330 bh = bh->b_this_page;
331 }
332
333 if (wait) {
334 wait_event(bitmap->write_wait,
335 atomic_read(&bitmap->pending_writes)==0);
336 }
337 }
338 if (bitmap->flags & BITMAP_WRITE_ERROR)
339 bitmap_file_kick(bitmap);
340}
341
342static void end_bitmap_write(struct buffer_head *bh, int uptodate)
343{
344 struct bitmap *bitmap = bh->b_private;
345 unsigned long flags;
346
347 if (!uptodate) {
348 spin_lock_irqsave(&bitmap->lock, flags);
349 bitmap->flags |= BITMAP_WRITE_ERROR;
350 spin_unlock_irqrestore(&bitmap->lock, flags);
351 }
352 if (atomic_dec_and_test(&bitmap->pending_writes))
353 wake_up(&bitmap->write_wait);
354}
355
356
357static void
358__clear_page_buffers(struct page *page)
359{
360 ClearPagePrivate(page);
361 set_page_private(page, 0);
362 page_cache_release(page);
363}
364static void free_buffers(struct page *page)
365{
366 struct buffer_head *bh = page_buffers(page);
367
368 while (bh) {
369 struct buffer_head *next = bh->b_this_page;
370 free_buffer_head(bh);
371 bh = next;
372 }
373 __clear_page_buffers(page);
374 put_page(page);
375}
376
377
378
379
380
381
382
383
384static struct page *read_page(struct file *file, unsigned long index,
385 struct bitmap *bitmap,
386 unsigned long count)
387{
388 struct page *page = NULL;
389 struct inode *inode = file->f_path.dentry->d_inode;
390 struct buffer_head *bh;
391 sector_t block;
392
393 PRINTK("read bitmap file (%dB @ %Lu)\n", (int)PAGE_SIZE,
394 (unsigned long long)index << PAGE_SHIFT);
395
396 page = alloc_page(GFP_KERNEL);
397 if (!page)
398 page = ERR_PTR(-ENOMEM);
399 if (IS_ERR(page))
400 goto out;
401
402 bh = alloc_page_buffers(page, 1<<inode->i_blkbits, 0);
403 if (!bh) {
404 put_page(page);
405 page = ERR_PTR(-ENOMEM);
406 goto out;
407 }
408 attach_page_buffers(page, bh);
409 block = index << (PAGE_SHIFT - inode->i_blkbits);
410 while (bh) {
411 if (count == 0)
412 bh->b_blocknr = 0;
413 else {
414 bh->b_blocknr = bmap(inode, block);
415 if (bh->b_blocknr == 0) {
416
417 free_buffers(page);
418 page = ERR_PTR(-EINVAL);
419 goto out;
420 }
421 bh->b_bdev = inode->i_sb->s_bdev;
422 if (count < (1<<inode->i_blkbits))
423 count = 0;
424 else
425 count -= (1<<inode->i_blkbits);
426
427 bh->b_end_io = end_bitmap_write;
428 bh->b_private = bitmap;
429 atomic_inc(&bitmap->pending_writes);
430 set_buffer_locked(bh);
431 set_buffer_mapped(bh);
432 submit_bh(READ, bh);
433 }
434 block++;
435 bh = bh->b_this_page;
436 }
437 page->index = index;
438
439 wait_event(bitmap->write_wait,
440 atomic_read(&bitmap->pending_writes)==0);
441 if (bitmap->flags & BITMAP_WRITE_ERROR) {
442 free_buffers(page);
443 page = ERR_PTR(-EIO);
444 }
445out:
446 if (IS_ERR(page))
447 printk(KERN_ALERT "md: bitmap read error: (%dB @ %Lu): %ld\n",
448 (int)PAGE_SIZE,
449 (unsigned long long)index << PAGE_SHIFT,
450 PTR_ERR(page));
451 return page;
452}
453
454
455
456
457
458
459void bitmap_update_sb(struct bitmap *bitmap)
460{
461 bitmap_super_t *sb;
462 unsigned long flags;
463
464 if (!bitmap || !bitmap->mddev)
465 return;
466 spin_lock_irqsave(&bitmap->lock, flags);
467 if (!bitmap->sb_page) {
468 spin_unlock_irqrestore(&bitmap->lock, flags);
469 return;
470 }
471 spin_unlock_irqrestore(&bitmap->lock, flags);
472 sb = (bitmap_super_t *)kmap_atomic(bitmap->sb_page, KM_USER0);
473 sb->events = cpu_to_le64(bitmap->mddev->events);
474 if (!bitmap->mddev->degraded)
475 sb->events_cleared = cpu_to_le64(bitmap->mddev->events);
476 kunmap_atomic(sb, KM_USER0);
477 write_page(bitmap, bitmap->sb_page, 1);
478}
479
480
481void bitmap_print_sb(struct bitmap *bitmap)
482{
483 bitmap_super_t *sb;
484
485 if (!bitmap || !bitmap->sb_page)
486 return;
487 sb = (bitmap_super_t *)kmap_atomic(bitmap->sb_page, KM_USER0);
488 printk(KERN_DEBUG "%s: bitmap file superblock:\n", bmname(bitmap));
489 printk(KERN_DEBUG " magic: %08x\n", le32_to_cpu(sb->magic));
490 printk(KERN_DEBUG " version: %d\n", le32_to_cpu(sb->version));
491 printk(KERN_DEBUG " uuid: %08x.%08x.%08x.%08x\n",
492 *(__u32 *)(sb->uuid+0),
493 *(__u32 *)(sb->uuid+4),
494 *(__u32 *)(sb->uuid+8),
495 *(__u32 *)(sb->uuid+12));
496 printk(KERN_DEBUG " events: %llu\n",
497 (unsigned long long) le64_to_cpu(sb->events));
498 printk(KERN_DEBUG "events cleared: %llu\n",
499 (unsigned long long) le64_to_cpu(sb->events_cleared));
500 printk(KERN_DEBUG " state: %08x\n", le32_to_cpu(sb->state));
501 printk(KERN_DEBUG " chunksize: %d B\n", le32_to_cpu(sb->chunksize));
502 printk(KERN_DEBUG " daemon sleep: %ds\n", le32_to_cpu(sb->daemon_sleep));
503 printk(KERN_DEBUG " sync size: %llu KB\n",
504 (unsigned long long)le64_to_cpu(sb->sync_size)/2);
505 printk(KERN_DEBUG "max write behind: %d\n", le32_to_cpu(sb->write_behind));
506 kunmap_atomic(sb, KM_USER0);
507}
508
509
510static int bitmap_read_sb(struct bitmap *bitmap)
511{
512 char *reason = NULL;
513 bitmap_super_t *sb;
514 unsigned long chunksize, daemon_sleep, write_behind;
515 unsigned long long events;
516 int err = -EINVAL;
517
518
519 if (bitmap->file) {
520 loff_t isize = i_size_read(bitmap->file->f_mapping->host);
521 int bytes = isize > PAGE_SIZE ? PAGE_SIZE : isize;
522
523 bitmap->sb_page = read_page(bitmap->file, 0, bitmap, bytes);
524 } else {
525 bitmap->sb_page = read_sb_page(bitmap->mddev, bitmap->offset, 0);
526 }
527 if (IS_ERR(bitmap->sb_page)) {
528 err = PTR_ERR(bitmap->sb_page);
529 bitmap->sb_page = NULL;
530 return err;
531 }
532
533 sb = (bitmap_super_t *)kmap_atomic(bitmap->sb_page, KM_USER0);
534
535 chunksize = le32_to_cpu(sb->chunksize);
536 daemon_sleep = le32_to_cpu(sb->daemon_sleep);
537 write_behind = le32_to_cpu(sb->write_behind);
538
539
540 if (sb->magic != cpu_to_le32(BITMAP_MAGIC))
541 reason = "bad magic";
542 else if (le32_to_cpu(sb->version) < BITMAP_MAJOR_LO ||
543 le32_to_cpu(sb->version) > BITMAP_MAJOR_HI)
544 reason = "unrecognized superblock version";
545 else if (chunksize < PAGE_SIZE)
546 reason = "bitmap chunksize too small";
547 else if ((1 << ffz(~chunksize)) != chunksize)
548 reason = "bitmap chunksize not a power of 2";
549 else if (daemon_sleep < 1 || daemon_sleep > MAX_SCHEDULE_TIMEOUT / HZ)
550 reason = "daemon sleep period out of range";
551 else if (write_behind > COUNTER_MAX)
552 reason = "write-behind limit out of range (0 - 16383)";
553 if (reason) {
554 printk(KERN_INFO "%s: invalid bitmap file superblock: %s\n",
555 bmname(bitmap), reason);
556 goto out;
557 }
558
559
560 sb->sync_size = cpu_to_le64(bitmap->mddev->resync_max_sectors);
561
562 if (!bitmap->mddev->persistent)
563 goto success;
564
565
566
567
568
569 if (memcmp(sb->uuid, bitmap->mddev->uuid, 16)) {
570 printk(KERN_INFO "%s: bitmap superblock UUID mismatch\n",
571 bmname(bitmap));
572 goto out;
573 }
574 events = le64_to_cpu(sb->events);
575 if (events < bitmap->mddev->events) {
576 printk(KERN_INFO "%s: bitmap file is out of date (%llu < %llu) "
577 "-- forcing full recovery\n", bmname(bitmap), events,
578 (unsigned long long) bitmap->mddev->events);
579 sb->state |= cpu_to_le32(BITMAP_STALE);
580 }
581success:
582
583 bitmap->chunksize = chunksize;
584 bitmap->daemon_sleep = daemon_sleep;
585 bitmap->daemon_lastrun = jiffies;
586 bitmap->max_write_behind = write_behind;
587 bitmap->flags |= le32_to_cpu(sb->state);
588 if (le32_to_cpu(sb->version) == BITMAP_MAJOR_HOSTENDIAN)
589 bitmap->flags |= BITMAP_HOSTENDIAN;
590 bitmap->events_cleared = le64_to_cpu(sb->events_cleared);
591 if (sb->state & cpu_to_le32(BITMAP_STALE))
592 bitmap->events_cleared = bitmap->mddev->events;
593 err = 0;
594out:
595 kunmap_atomic(sb, KM_USER0);
596 if (err)
597 bitmap_print_sb(bitmap);
598 return err;
599}
600
601enum bitmap_mask_op {
602 MASK_SET,
603 MASK_UNSET
604};
605
606
607static int bitmap_mask_state(struct bitmap *bitmap, enum bitmap_state bits,
608 enum bitmap_mask_op op)
609{
610 bitmap_super_t *sb;
611 unsigned long flags;
612 int old;
613
614 spin_lock_irqsave(&bitmap->lock, flags);
615 if (!bitmap->sb_page) {
616 spin_unlock_irqrestore(&bitmap->lock, flags);
617 return 0;
618 }
619 spin_unlock_irqrestore(&bitmap->lock, flags);
620 sb = (bitmap_super_t *)kmap_atomic(bitmap->sb_page, KM_USER0);
621 old = le32_to_cpu(sb->state) & bits;
622 switch (op) {
623 case MASK_SET: sb->state |= cpu_to_le32(bits);
624 break;
625 case MASK_UNSET: sb->state &= cpu_to_le32(~bits);
626 break;
627 default: BUG();
628 }
629 kunmap_atomic(sb, KM_USER0);
630 return old;
631}
632
633
634
635
636
637
638static inline unsigned long file_page_index(unsigned long chunk)
639{
640 return CHUNK_BIT_OFFSET(chunk) >> PAGE_BIT_SHIFT;
641}
642
643
644static inline unsigned long file_page_offset(unsigned long chunk)
645{
646 return CHUNK_BIT_OFFSET(chunk) & (PAGE_BITS - 1);
647}
648
649
650
651
652
653
654
655
656static inline struct page *filemap_get_page(struct bitmap *bitmap,
657 unsigned long chunk)
658{
659 if (file_page_index(chunk) >= bitmap->file_pages) return NULL;
660 return bitmap->filemap[file_page_index(chunk) - file_page_index(0)];
661}
662
663
664static void bitmap_file_unmap(struct bitmap *bitmap)
665{
666 struct page **map, *sb_page;
667 unsigned long *attr;
668 int pages;
669 unsigned long flags;
670
671 spin_lock_irqsave(&bitmap->lock, flags);
672 map = bitmap->filemap;
673 bitmap->filemap = NULL;
674 attr = bitmap->filemap_attr;
675 bitmap->filemap_attr = NULL;
676 pages = bitmap->file_pages;
677 bitmap->file_pages = 0;
678 sb_page = bitmap->sb_page;
679 bitmap->sb_page = NULL;
680 spin_unlock_irqrestore(&bitmap->lock, flags);
681
682 while (pages--)
683 if (map[pages]->index != 0)
684 free_buffers(map[pages]);
685 kfree(map);
686 kfree(attr);
687
688 if (sb_page)
689 free_buffers(sb_page);
690}
691
692static void bitmap_file_put(struct bitmap *bitmap)
693{
694 struct file *file;
695 unsigned long flags;
696
697 spin_lock_irqsave(&bitmap->lock, flags);
698 file = bitmap->file;
699 bitmap->file = NULL;
700 spin_unlock_irqrestore(&bitmap->lock, flags);
701
702 if (file)
703 wait_event(bitmap->write_wait,
704 atomic_read(&bitmap->pending_writes)==0);
705 bitmap_file_unmap(bitmap);
706
707 if (file) {
708 struct inode *inode = file->f_path.dentry->d_inode;
709 invalidate_mapping_pages(inode->i_mapping, 0, -1);
710 fput(file);
711 }
712}
713
714
715
716
717
718
719
720static void bitmap_file_kick(struct bitmap *bitmap)
721{
722 char *path, *ptr = NULL;
723
724 if (bitmap_mask_state(bitmap, BITMAP_STALE, MASK_SET) == 0) {
725 bitmap_update_sb(bitmap);
726
727 if (bitmap->file) {
728 path = kmalloc(PAGE_SIZE, GFP_KERNEL);
729 if (path)
730 ptr = file_path(bitmap->file, path, PAGE_SIZE);
731
732 printk(KERN_ALERT
733 "%s: kicking failed bitmap file %s from array!\n",
734 bmname(bitmap), ptr ? ptr : "");
735
736 kfree(path);
737 } else
738 printk(KERN_ALERT
739 "%s: disabling internal bitmap due to errors\n",
740 bmname(bitmap));
741 }
742
743 bitmap_file_put(bitmap);
744
745 return;
746}
747
748enum bitmap_page_attr {
749 BITMAP_PAGE_DIRTY = 0,
750 BITMAP_PAGE_CLEAN = 1,
751 BITMAP_PAGE_NEEDWRITE=2,
752};
753
754static inline void set_page_attr(struct bitmap *bitmap, struct page *page,
755 enum bitmap_page_attr attr)
756{
757 __set_bit((page->index<<2) + attr, bitmap->filemap_attr);
758}
759
760static inline void clear_page_attr(struct bitmap *bitmap, struct page *page,
761 enum bitmap_page_attr attr)
762{
763 __clear_bit((page->index<<2) + attr, bitmap->filemap_attr);
764}
765
766static inline unsigned long test_page_attr(struct bitmap *bitmap, struct page *page,
767 enum bitmap_page_attr attr)
768{
769 return test_bit((page->index<<2) + attr, bitmap->filemap_attr);
770}
771
772
773
774
775
776
777
778
779static void bitmap_file_set_bit(struct bitmap *bitmap, sector_t block)
780{
781 unsigned long bit;
782 struct page *page;
783 void *kaddr;
784 unsigned long chunk = block >> CHUNK_BLOCK_SHIFT(bitmap);
785
786 if (!bitmap->filemap) {
787 return;
788 }
789
790 page = filemap_get_page(bitmap, chunk);
791 if (!page) return;
792 bit = file_page_offset(chunk);
793
794
795 kaddr = kmap_atomic(page, KM_USER0);
796 if (bitmap->flags & BITMAP_HOSTENDIAN)
797 set_bit(bit, kaddr);
798 else
799 ext2_set_bit(bit, kaddr);
800 kunmap_atomic(kaddr, KM_USER0);
801 PRINTK("set file bit %lu page %lu\n", bit, page->index);
802
803
804 set_page_attr(bitmap, page, BITMAP_PAGE_DIRTY);
805
806}
807
808
809
810
811void bitmap_unplug(struct bitmap *bitmap)
812{
813 unsigned long i, flags;
814 int dirty, need_write;
815 struct page *page;
816 int wait = 0;
817
818 if (!bitmap)
819 return;
820
821
822
823 for (i = 0; i < bitmap->file_pages; i++) {
824 spin_lock_irqsave(&bitmap->lock, flags);
825 if (!bitmap->filemap) {
826 spin_unlock_irqrestore(&bitmap->lock, flags);
827 return;
828 }
829 page = bitmap->filemap[i];
830 dirty = test_page_attr(bitmap, page, BITMAP_PAGE_DIRTY);
831 need_write = test_page_attr(bitmap, page, BITMAP_PAGE_NEEDWRITE);
832 clear_page_attr(bitmap, page, BITMAP_PAGE_DIRTY);
833 clear_page_attr(bitmap, page, BITMAP_PAGE_NEEDWRITE);
834 if (dirty)
835 wait = 1;
836 spin_unlock_irqrestore(&bitmap->lock, flags);
837
838 if (dirty | need_write)
839 write_page(bitmap, page, 0);
840 }
841 if (wait) {
842 if (bitmap->file)
843 wait_event(bitmap->write_wait,
844 atomic_read(&bitmap->pending_writes)==0);
845 else
846 md_super_wait(bitmap->mddev);
847 }
848 if (bitmap->flags & BITMAP_WRITE_ERROR)
849 bitmap_file_kick(bitmap);
850}
851
852static void bitmap_set_memory_bits(struct bitmap *bitmap, sector_t offset, int needed);
853
854
855
856
857
858
859
860
861
862
863
864static int bitmap_init_from_disk(struct bitmap *bitmap, sector_t start)
865{
866 unsigned long i, chunks, index, oldindex, bit;
867 struct page *page = NULL, *oldpage = NULL;
868 unsigned long num_pages, bit_cnt = 0;
869 struct file *file;
870 unsigned long bytes, offset;
871 int outofdate;
872 int ret = -ENOSPC;
873 void *paddr;
874
875 chunks = bitmap->chunks;
876 file = bitmap->file;
877
878 BUG_ON(!file && !bitmap->offset);
879
880#ifdef INJECT_FAULTS_3
881 outofdate = 1;
882#else
883 outofdate = bitmap->flags & BITMAP_STALE;
884#endif
885 if (outofdate)
886 printk(KERN_INFO "%s: bitmap file is out of date, doing full "
887 "recovery\n", bmname(bitmap));
888
889 bytes = (chunks + 7) / 8;
890
891 num_pages = (bytes + sizeof(bitmap_super_t) + PAGE_SIZE - 1) / PAGE_SIZE;
892
893 if (file && i_size_read(file->f_mapping->host) < bytes + sizeof(bitmap_super_t)) {
894 printk(KERN_INFO "%s: bitmap file too short %lu < %lu\n",
895 bmname(bitmap),
896 (unsigned long) i_size_read(file->f_mapping->host),
897 bytes + sizeof(bitmap_super_t));
898 goto err;
899 }
900
901 ret = -ENOMEM;
902
903 bitmap->filemap = kmalloc(sizeof(struct page *) * num_pages, GFP_KERNEL);
904 if (!bitmap->filemap)
905 goto err;
906
907
908 bitmap->filemap_attr = kzalloc(
909 roundup( DIV_ROUND_UP(num_pages*4, 8), sizeof(unsigned long)),
910 GFP_KERNEL);
911 if (!bitmap->filemap_attr)
912 goto err;
913
914 oldindex = ~0L;
915
916 for (i = 0; i < chunks; i++) {
917 int b;
918 index = file_page_index(i);
919 bit = file_page_offset(i);
920 if (index != oldindex) {
921 int count;
922
923 if (index == num_pages-1)
924 count = bytes + sizeof(bitmap_super_t)
925 - index * PAGE_SIZE;
926 else
927 count = PAGE_SIZE;
928 if (index == 0) {
929
930
931
932
933
934 page = bitmap->sb_page;
935 offset = sizeof(bitmap_super_t);
936 } else if (file) {
937 page = read_page(file, index, bitmap, count);
938 offset = 0;
939 } else {
940 page = read_sb_page(bitmap->mddev, bitmap->offset, index);
941 offset = 0;
942 }
943 if (IS_ERR(page)) {
944 ret = PTR_ERR(page);
945 goto err;
946 }
947
948 oldindex = index;
949 oldpage = page;
950
951 if (outofdate) {
952
953
954
955
956 paddr = kmap_atomic(page, KM_USER0);
957 memset(paddr + offset, 0xff,
958 PAGE_SIZE - offset);
959 kunmap_atomic(paddr, KM_USER0);
960 write_page(bitmap, page, 1);
961
962 ret = -EIO;
963 if (bitmap->flags & BITMAP_WRITE_ERROR) {
964
965 put_page(page);
966 goto err;
967 }
968 }
969
970 bitmap->filemap[bitmap->file_pages++] = page;
971 bitmap->last_page_size = count;
972 }
973 paddr = kmap_atomic(page, KM_USER0);
974 if (bitmap->flags & BITMAP_HOSTENDIAN)
975 b = test_bit(bit, paddr);
976 else
977 b = ext2_test_bit(bit, paddr);
978 kunmap_atomic(paddr, KM_USER0);
979 if (b) {
980
981 bitmap_set_memory_bits(bitmap, i << CHUNK_BLOCK_SHIFT(bitmap),
982 ((i+1) << (CHUNK_BLOCK_SHIFT(bitmap)) >= start)
983 );
984 bit_cnt++;
985 set_page_attr(bitmap, page, BITMAP_PAGE_CLEAN);
986 }
987 }
988
989
990 ret = 0;
991 bitmap_mask_state(bitmap, BITMAP_STALE, MASK_UNSET);
992
993 if (bit_cnt) {
994 set_bit(MD_RECOVERY_NEEDED, &bitmap->mddev->recovery);
995 md_wakeup_thread(bitmap->mddev->thread);
996 }
997
998 printk(KERN_INFO "%s: bitmap initialized from disk: "
999 "read %lu/%lu pages, set %lu bits\n",
1000 bmname(bitmap), bitmap->file_pages, num_pages, bit_cnt);
1001
1002 return 0;
1003
1004 err:
1005 printk(KERN_INFO "%s: bitmap initialisation failed: %d\n",
1006 bmname(bitmap), ret);
1007 return ret;
1008}
1009
1010void bitmap_write_all(struct bitmap *bitmap)
1011{
1012
1013
1014
1015 int i;
1016
1017 for (i=0; i < bitmap->file_pages; i++)
1018 set_page_attr(bitmap, bitmap->filemap[i],
1019 BITMAP_PAGE_NEEDWRITE);
1020}
1021
1022
1023static void bitmap_count_page(struct bitmap *bitmap, sector_t offset, int inc)
1024{
1025 sector_t chunk = offset >> CHUNK_BLOCK_SHIFT(bitmap);
1026 unsigned long page = chunk >> PAGE_COUNTER_SHIFT;
1027 bitmap->bp[page].count += inc;
1028
1029
1030
1031
1032 bitmap_checkfree(bitmap, page);
1033}
1034static bitmap_counter_t *bitmap_get_counter(struct bitmap *bitmap,
1035 sector_t offset, int *blocks,
1036 int create);
1037
1038
1039
1040
1041
1042
1043void bitmap_daemon_work(struct bitmap *bitmap)
1044{
1045 unsigned long j;
1046 unsigned long flags;
1047 struct page *page = NULL, *lastpage = NULL;
1048 int blocks;
1049 void *paddr;
1050
1051 if (bitmap == NULL)
1052 return;
1053 if (time_before(jiffies, bitmap->daemon_lastrun + bitmap->daemon_sleep*HZ))
1054 return;
1055 bitmap->daemon_lastrun = jiffies;
1056
1057 for (j = 0; j < bitmap->chunks; j++) {
1058 bitmap_counter_t *bmc;
1059 spin_lock_irqsave(&bitmap->lock, flags);
1060 if (!bitmap->filemap) {
1061
1062 spin_unlock_irqrestore(&bitmap->lock, flags);
1063 break;
1064 }
1065
1066 page = filemap_get_page(bitmap, j);
1067
1068 if (page != lastpage) {
1069
1070 if (!test_page_attr(bitmap, page, BITMAP_PAGE_CLEAN)) {
1071 int need_write = test_page_attr(bitmap, page,
1072 BITMAP_PAGE_NEEDWRITE);
1073 if (need_write)
1074 clear_page_attr(bitmap, page, BITMAP_PAGE_NEEDWRITE);
1075
1076 spin_unlock_irqrestore(&bitmap->lock, flags);
1077 if (need_write)
1078 write_page(bitmap, page, 0);
1079 continue;
1080 }
1081
1082
1083 if (lastpage != NULL) {
1084 if (test_page_attr(bitmap, lastpage, BITMAP_PAGE_NEEDWRITE)) {
1085 clear_page_attr(bitmap, lastpage, BITMAP_PAGE_NEEDWRITE);
1086 spin_unlock_irqrestore(&bitmap->lock, flags);
1087 write_page(bitmap, lastpage, 0);
1088 } else {
1089 set_page_attr(bitmap, lastpage, BITMAP_PAGE_NEEDWRITE);
1090 spin_unlock_irqrestore(&bitmap->lock, flags);
1091 }
1092 } else
1093 spin_unlock_irqrestore(&bitmap->lock, flags);
1094 lastpage = page;
1095
1096
1097
1098 spin_lock_irqsave(&bitmap->lock, flags);
1099 clear_page_attr(bitmap, page, BITMAP_PAGE_CLEAN);
1100 }
1101 bmc = bitmap_get_counter(bitmap, j << CHUNK_BLOCK_SHIFT(bitmap),
1102 &blocks, 0);
1103 if (bmc) {
1104
1105
1106
1107 if (*bmc == 2) {
1108 *bmc=1;
1109 set_page_attr(bitmap, page, BITMAP_PAGE_CLEAN);
1110 } else if (*bmc == 1) {
1111
1112 *bmc = 0;
1113 bitmap_count_page(bitmap, j << CHUNK_BLOCK_SHIFT(bitmap),
1114 -1);
1115
1116
1117 paddr = kmap_atomic(page, KM_USER0);
1118 if (bitmap->flags & BITMAP_HOSTENDIAN)
1119 clear_bit(file_page_offset(j), paddr);
1120 else
1121 ext2_clear_bit(file_page_offset(j), paddr);
1122 kunmap_atomic(paddr, KM_USER0);
1123 }
1124 }
1125 spin_unlock_irqrestore(&bitmap->lock, flags);
1126 }
1127
1128
1129 if (lastpage != NULL) {
1130 spin_lock_irqsave(&bitmap->lock, flags);
1131 if (test_page_attr(bitmap, lastpage, BITMAP_PAGE_NEEDWRITE)) {
1132 clear_page_attr(bitmap, lastpage, BITMAP_PAGE_NEEDWRITE);
1133 spin_unlock_irqrestore(&bitmap->lock, flags);
1134 write_page(bitmap, lastpage, 0);
1135 } else {
1136 set_page_attr(bitmap, lastpage, BITMAP_PAGE_NEEDWRITE);
1137 spin_unlock_irqrestore(&bitmap->lock, flags);
1138 }
1139 }
1140
1141}
1142
1143static bitmap_counter_t *bitmap_get_counter(struct bitmap *bitmap,
1144 sector_t offset, int *blocks,
1145 int create)
1146{
1147
1148
1149
1150
1151 sector_t chunk = offset >> CHUNK_BLOCK_SHIFT(bitmap);
1152 unsigned long page = chunk >> PAGE_COUNTER_SHIFT;
1153 unsigned long pageoff = (chunk & PAGE_COUNTER_MASK) << COUNTER_BYTE_SHIFT;
1154 sector_t csize;
1155
1156 if (bitmap_checkpage(bitmap, page, create) < 0) {
1157 csize = ((sector_t)1) << (CHUNK_BLOCK_SHIFT(bitmap));
1158 *blocks = csize - (offset & (csize- 1));
1159 return NULL;
1160 }
1161
1162
1163 if (bitmap->bp[page].hijacked) {
1164
1165
1166 int hi = (pageoff > PAGE_COUNTER_MASK);
1167 csize = ((sector_t)1) << (CHUNK_BLOCK_SHIFT(bitmap) +
1168 PAGE_COUNTER_SHIFT - 1);
1169 *blocks = csize - (offset & (csize- 1));
1170 return &((bitmap_counter_t *)
1171 &bitmap->bp[page].map)[hi];
1172 } else {
1173 csize = ((sector_t)1) << (CHUNK_BLOCK_SHIFT(bitmap));
1174 *blocks = csize - (offset & (csize- 1));
1175 return (bitmap_counter_t *)
1176 &(bitmap->bp[page].map[pageoff]);
1177 }
1178}
1179
1180int bitmap_startwrite(struct bitmap *bitmap, sector_t offset, unsigned long sectors, int behind)
1181{
1182 if (!bitmap) return 0;
1183
1184 if (behind) {
1185 atomic_inc(&bitmap->behind_writes);
1186 PRINTK(KERN_DEBUG "inc write-behind count %d/%d\n",
1187 atomic_read(&bitmap->behind_writes), bitmap->max_write_behind);
1188 }
1189
1190 while (sectors) {
1191 int blocks;
1192 bitmap_counter_t *bmc;
1193
1194 spin_lock_irq(&bitmap->lock);
1195 bmc = bitmap_get_counter(bitmap, offset, &blocks, 1);
1196 if (!bmc) {
1197 spin_unlock_irq(&bitmap->lock);
1198 return 0;
1199 }
1200
1201 if (unlikely((*bmc & COUNTER_MAX) == COUNTER_MAX)) {
1202 DEFINE_WAIT(__wait);
1203
1204
1205
1206
1207 prepare_to_wait(&bitmap->overflow_wait, &__wait,
1208 TASK_UNINTERRUPTIBLE);
1209 spin_unlock_irq(&bitmap->lock);
1210 blk_unplug(bitmap->mddev->queue);
1211 schedule();
1212 finish_wait(&bitmap->overflow_wait, &__wait);
1213 continue;
1214 }
1215
1216 switch(*bmc) {
1217 case 0:
1218 bitmap_file_set_bit(bitmap, offset);
1219 bitmap_count_page(bitmap,offset, 1);
1220 blk_plug_device(bitmap->mddev->queue);
1221
1222 case 1:
1223 *bmc = 2;
1224 }
1225
1226 (*bmc)++;
1227
1228 spin_unlock_irq(&bitmap->lock);
1229
1230 offset += blocks;
1231 if (sectors > blocks)
1232 sectors -= blocks;
1233 else sectors = 0;
1234 }
1235 return 0;
1236}
1237
1238void bitmap_endwrite(struct bitmap *bitmap, sector_t offset, unsigned long sectors,
1239 int success, int behind)
1240{
1241 if (!bitmap) return;
1242 if (behind) {
1243 atomic_dec(&bitmap->behind_writes);
1244 PRINTK(KERN_DEBUG "dec write-behind count %d/%d\n",
1245 atomic_read(&bitmap->behind_writes), bitmap->max_write_behind);
1246 }
1247
1248 while (sectors) {
1249 int blocks;
1250 unsigned long flags;
1251 bitmap_counter_t *bmc;
1252
1253 spin_lock_irqsave(&bitmap->lock, flags);
1254 bmc = bitmap_get_counter(bitmap, offset, &blocks, 0);
1255 if (!bmc) {
1256 spin_unlock_irqrestore(&bitmap->lock, flags);
1257 return;
1258 }
1259
1260 if (!success && ! (*bmc & NEEDED_MASK))
1261 *bmc |= NEEDED_MASK;
1262
1263 if ((*bmc & COUNTER_MAX) == COUNTER_MAX)
1264 wake_up(&bitmap->overflow_wait);
1265
1266 (*bmc)--;
1267 if (*bmc <= 2) {
1268 set_page_attr(bitmap,
1269 filemap_get_page(bitmap, offset >> CHUNK_BLOCK_SHIFT(bitmap)),
1270 BITMAP_PAGE_CLEAN);
1271 }
1272 spin_unlock_irqrestore(&bitmap->lock, flags);
1273 offset += blocks;
1274 if (sectors > blocks)
1275 sectors -= blocks;
1276 else sectors = 0;
1277 }
1278}
1279
1280int bitmap_start_sync(struct bitmap *bitmap, sector_t offset, int *blocks,
1281 int degraded)
1282{
1283 bitmap_counter_t *bmc;
1284 int rv;
1285 if (bitmap == NULL) {
1286 *blocks = 1024;
1287 return 1;
1288 }
1289 spin_lock_irq(&bitmap->lock);
1290 bmc = bitmap_get_counter(bitmap, offset, blocks, 0);
1291 rv = 0;
1292 if (bmc) {
1293
1294 if (RESYNC(*bmc))
1295 rv = 1;
1296 else if (NEEDED(*bmc)) {
1297 rv = 1;
1298 if (!degraded) {
1299 *bmc |= RESYNC_MASK;
1300 *bmc &= ~NEEDED_MASK;
1301 }
1302 }
1303 }
1304 spin_unlock_irq(&bitmap->lock);
1305 return rv;
1306}
1307
1308void bitmap_end_sync(struct bitmap *bitmap, sector_t offset, int *blocks, int aborted)
1309{
1310 bitmap_counter_t *bmc;
1311 unsigned long flags;
1312
1313
1314 if (bitmap == NULL) {
1315 *blocks = 1024;
1316 return;
1317 }
1318 spin_lock_irqsave(&bitmap->lock, flags);
1319 bmc = bitmap_get_counter(bitmap, offset, blocks, 0);
1320 if (bmc == NULL)
1321 goto unlock;
1322
1323
1324
1325
1326 if (RESYNC(*bmc)) {
1327 *bmc &= ~RESYNC_MASK;
1328
1329 if (!NEEDED(*bmc) && aborted)
1330 *bmc |= NEEDED_MASK;
1331 else {
1332 if (*bmc <= 2) {
1333 set_page_attr(bitmap,
1334 filemap_get_page(bitmap, offset >> CHUNK_BLOCK_SHIFT(bitmap)),
1335 BITMAP_PAGE_CLEAN);
1336 }
1337 }
1338 }
1339 unlock:
1340 spin_unlock_irqrestore(&bitmap->lock, flags);
1341}
1342
1343void bitmap_close_sync(struct bitmap *bitmap)
1344{
1345
1346
1347
1348
1349 sector_t sector = 0;
1350 int blocks;
1351 if (!bitmap) return;
1352 while (sector < bitmap->mddev->resync_max_sectors) {
1353 bitmap_end_sync(bitmap, sector, &blocks, 0);
1354
1355
1356
1357 sector += blocks;
1358 }
1359}
1360
1361static void bitmap_set_memory_bits(struct bitmap *bitmap, sector_t offset, int needed)
1362{
1363
1364
1365
1366
1367
1368 int secs;
1369 bitmap_counter_t *bmc;
1370 spin_lock_irq(&bitmap->lock);
1371 bmc = bitmap_get_counter(bitmap, offset, &secs, 1);
1372 if (!bmc) {
1373 spin_unlock_irq(&bitmap->lock);
1374 return;
1375 }
1376 if (! *bmc) {
1377 struct page *page;
1378 *bmc = 1 | (needed?NEEDED_MASK:0);
1379 bitmap_count_page(bitmap, offset, 1);
1380 page = filemap_get_page(bitmap, offset >> CHUNK_BLOCK_SHIFT(bitmap));
1381 set_page_attr(bitmap, page, BITMAP_PAGE_CLEAN);
1382 }
1383 spin_unlock_irq(&bitmap->lock);
1384
1385}
1386
1387
1388void bitmap_dirty_bits(struct bitmap *bitmap, unsigned long s, unsigned long e)
1389{
1390 unsigned long chunk;
1391
1392 for (chunk = s; chunk <= e; chunk++) {
1393 sector_t sec = chunk << CHUNK_BLOCK_SHIFT(bitmap);
1394 bitmap_set_memory_bits(bitmap, sec, 1);
1395 bitmap_file_set_bit(bitmap, sec);
1396 }
1397}
1398
1399
1400
1401
1402void bitmap_flush(mddev_t *mddev)
1403{
1404 struct bitmap *bitmap = mddev->bitmap;
1405 int sleep;
1406
1407 if (!bitmap)
1408 return;
1409
1410
1411
1412
1413 sleep = bitmap->daemon_sleep;
1414 bitmap->daemon_sleep = 0;
1415 bitmap_daemon_work(bitmap);
1416 bitmap_daemon_work(bitmap);
1417 bitmap_daemon_work(bitmap);
1418 bitmap->daemon_sleep = sleep;
1419 bitmap_update_sb(bitmap);
1420}
1421
1422
1423
1424
1425static void bitmap_free(struct bitmap *bitmap)
1426{
1427 unsigned long k, pages;
1428 struct bitmap_page *bp;
1429
1430 if (!bitmap)
1431 return;
1432
1433
1434 bitmap_file_put(bitmap);
1435
1436 bp = bitmap->bp;
1437 pages = bitmap->pages;
1438
1439
1440
1441 if (bp)
1442 for (k = 0; k < pages; k++)
1443 if (bp[k].map && !bp[k].hijacked)
1444 kfree(bp[k].map);
1445 kfree(bp);
1446 kfree(bitmap);
1447}
1448void bitmap_destroy(mddev_t *mddev)
1449{
1450 struct bitmap *bitmap = mddev->bitmap;
1451
1452 if (!bitmap)
1453 return;
1454
1455 mddev->bitmap = NULL;
1456 if (mddev->thread)
1457 mddev->thread->timeout = MAX_SCHEDULE_TIMEOUT;
1458
1459 bitmap_free(bitmap);
1460}
1461
1462
1463
1464
1465
1466int bitmap_create(mddev_t *mddev)
1467{
1468 struct bitmap *bitmap;
1469 unsigned long blocks = mddev->resync_max_sectors;
1470 unsigned long chunks;
1471 unsigned long pages;
1472 struct file *file = mddev->bitmap_file;
1473 int err;
1474 sector_t start;
1475
1476 BUILD_BUG_ON(sizeof(bitmap_super_t) != 256);
1477
1478 if (!file && !mddev->bitmap_offset)
1479 return 0;
1480
1481 BUG_ON(file && mddev->bitmap_offset);
1482
1483 bitmap = kzalloc(sizeof(*bitmap), GFP_KERNEL);
1484 if (!bitmap)
1485 return -ENOMEM;
1486
1487 spin_lock_init(&bitmap->lock);
1488 atomic_set(&bitmap->pending_writes, 0);
1489 init_waitqueue_head(&bitmap->write_wait);
1490 init_waitqueue_head(&bitmap->overflow_wait);
1491
1492 bitmap->mddev = mddev;
1493
1494 bitmap->file = file;
1495 bitmap->offset = mddev->bitmap_offset;
1496 if (file) {
1497 get_file(file);
1498 do_sync_mapping_range(file->f_mapping, 0, LLONG_MAX,
1499 SYNC_FILE_RANGE_WAIT_BEFORE |
1500 SYNC_FILE_RANGE_WRITE |
1501 SYNC_FILE_RANGE_WAIT_AFTER);
1502 }
1503
1504 err = bitmap_read_sb(bitmap);
1505 if (err)
1506 goto error;
1507
1508 bitmap->chunkshift = ffz(~bitmap->chunksize);
1509
1510
1511 chunks = (blocks + CHUNK_BLOCK_RATIO(bitmap) - 1) /
1512 CHUNK_BLOCK_RATIO(bitmap);
1513 pages = (chunks + PAGE_COUNTER_RATIO - 1) / PAGE_COUNTER_RATIO;
1514
1515 BUG_ON(!pages);
1516
1517 bitmap->chunks = chunks;
1518 bitmap->pages = pages;
1519 bitmap->missing_pages = pages;
1520 bitmap->counter_bits = COUNTER_BITS;
1521
1522 bitmap->syncchunk = ~0UL;
1523
1524#ifdef INJECT_FATAL_FAULT_1
1525 bitmap->bp = NULL;
1526#else
1527 bitmap->bp = kzalloc(pages * sizeof(*bitmap->bp), GFP_KERNEL);
1528#endif
1529 err = -ENOMEM;
1530 if (!bitmap->bp)
1531 goto error;
1532
1533
1534
1535 start = 0;
1536 if (mddev->degraded == 0
1537 || bitmap->events_cleared == mddev->events)
1538
1539 start = mddev->recovery_cp;
1540 err = bitmap_init_from_disk(bitmap, start);
1541
1542 if (err)
1543 goto error;
1544
1545 printk(KERN_INFO "created bitmap (%lu pages) for device %s\n",
1546 pages, bmname(bitmap));
1547
1548 mddev->bitmap = bitmap;
1549
1550 mddev->thread->timeout = bitmap->daemon_sleep * HZ;
1551
1552 bitmap_update_sb(bitmap);
1553
1554 return (bitmap->flags & BITMAP_WRITE_ERROR) ? -EIO : 0;
1555
1556 error:
1557 bitmap_free(bitmap);
1558 return err;
1559}
1560
1561
1562EXPORT_SYMBOL(bitmap_startwrite);
1563EXPORT_SYMBOL(bitmap_endwrite);
1564EXPORT_SYMBOL(bitmap_start_sync);
1565EXPORT_SYMBOL(bitmap_end_sync);
1566EXPORT_SYMBOL(bitmap_unplug);
1567EXPORT_SYMBOL(bitmap_close_sync);
1568