md/raid456: distribute raid processing over multiple cores
[pandora-kernel.git] / drivers / md / raid5.c
1 /*
2  * raid5.c : Multiple Devices driver for Linux
3  *         Copyright (C) 1996, 1997 Ingo Molnar, Miguel de Icaza, Gadi Oxman
4  *         Copyright (C) 1999, 2000 Ingo Molnar
5  *         Copyright (C) 2002, 2003 H. Peter Anvin
6  *
7  * RAID-4/5/6 management functions.
8  * Thanks to Penguin Computing for making the RAID-6 development possible
9  * by donating a test server!
10  *
11  * This program is free software; you can redistribute it and/or modify
12  * it under the terms of the GNU General Public License as published by
13  * the Free Software Foundation; either version 2, or (at your option)
14  * any later version.
15  *
16  * You should have received a copy of the GNU General Public License
17  * (for example /usr/src/linux/COPYING); if not, write to the Free
18  * Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19  */
20
21 /*
22  * BITMAP UNPLUGGING:
23  *
24  * The sequencing for updating the bitmap reliably is a little
25  * subtle (and I got it wrong the first time) so it deserves some
26  * explanation.
27  *
28  * We group bitmap updates into batches.  Each batch has a number.
29  * We may write out several batches at once, but that isn't very important.
30  * conf->bm_write is the number of the last batch successfully written.
31  * conf->bm_flush is the number of the last batch that was closed to
32  *    new additions.
33  * When we discover that we will need to write to any block in a stripe
34  * (in add_stripe_bio) we update the in-memory bitmap and record in sh->bm_seq
35  * the number of the batch it will be in. This is bm_flush+1.
36  * When we are ready to do a write, if that batch hasn't been written yet,
37  *   we plug the array and queue the stripe for later.
38  * When an unplug happens, we increment bm_flush, thus closing the current
39  *   batch.
40  * When we notice that bm_flush > bm_write, we write out all pending updates
41  * to the bitmap, and advance bm_write to where bm_flush was.
42  * This may occasionally write a bit out twice, but is sure never to
43  * miss any bits.
44  */
45
46 #include <linux/blkdev.h>
47 #include <linux/kthread.h>
48 #include <linux/raid/pq.h>
49 #include <linux/async_tx.h>
50 #include <linux/async.h>
51 #include <linux/seq_file.h>
52 #include <linux/cpu.h>
53 #include "md.h"
54 #include "raid5.h"
55 #include "bitmap.h"
56
57 /*
58  * Stripe cache
59  */
60
61 #define NR_STRIPES              256
62 #define STRIPE_SIZE             PAGE_SIZE
63 #define STRIPE_SHIFT            (PAGE_SHIFT - 9)
64 #define STRIPE_SECTORS          (STRIPE_SIZE>>9)
65 #define IO_THRESHOLD            1
66 #define BYPASS_THRESHOLD        1
67 #define NR_HASH                 (PAGE_SIZE / sizeof(struct hlist_head))
68 #define HASH_MASK               (NR_HASH - 1)
69
70 #define stripe_hash(conf, sect) (&((conf)->stripe_hashtbl[((sect) >> STRIPE_SHIFT) & HASH_MASK]))
71
72 /* bio's attached to a stripe+device for I/O are linked together in bi_sector
73  * order without overlap.  There may be several bio's per stripe+device, and
74  * a bio could span several devices.
75  * When walking this list for a particular stripe+device, we must never proceed
76  * beyond a bio that extends past this device, as the next bio might no longer
77  * be valid.
78  * This macro is used to determine the 'next' bio in the list, given the sector
79  * of the current stripe+device
80  */
81 #define r5_next_bio(bio, sect) ( ( (bio)->bi_sector + ((bio)->bi_size>>9) < sect + STRIPE_SECTORS) ? (bio)->bi_next : NULL)
82 /*
83  * The following can be used to debug the driver
84  */
85 #define RAID5_PARANOIA  1
86 #if RAID5_PARANOIA && defined(CONFIG_SMP)
87 # define CHECK_DEVLOCK() assert_spin_locked(&conf->device_lock)
88 #else
89 # define CHECK_DEVLOCK()
90 #endif
91
92 #ifdef DEBUG
93 #define inline
94 #define __inline__
95 #endif
96
97 #define printk_rl(args...) ((void) (printk_ratelimit() && printk(args)))
98
99 /*
100  * We maintain a biased count of active stripes in the bottom 16 bits of
101  * bi_phys_segments, and a count of processed stripes in the upper 16 bits
102  */
103 static inline int raid5_bi_phys_segments(struct bio *bio)
104 {
105         return bio->bi_phys_segments & 0xffff;
106 }
107
108 static inline int raid5_bi_hw_segments(struct bio *bio)
109 {
110         return (bio->bi_phys_segments >> 16) & 0xffff;
111 }
112
113 static inline int raid5_dec_bi_phys_segments(struct bio *bio)
114 {
115         --bio->bi_phys_segments;
116         return raid5_bi_phys_segments(bio);
117 }
118
119 static inline int raid5_dec_bi_hw_segments(struct bio *bio)
120 {
121         unsigned short val = raid5_bi_hw_segments(bio);
122
123         --val;
124         bio->bi_phys_segments = (val << 16) | raid5_bi_phys_segments(bio);
125         return val;
126 }
127
128 static inline void raid5_set_bi_hw_segments(struct bio *bio, unsigned int cnt)
129 {
130         bio->bi_phys_segments = raid5_bi_phys_segments(bio) || (cnt << 16);
131 }
132
133 /* Find first data disk in a raid6 stripe */
134 static inline int raid6_d0(struct stripe_head *sh)
135 {
136         if (sh->ddf_layout)
137                 /* ddf always start from first device */
138                 return 0;
139         /* md starts just after Q block */
140         if (sh->qd_idx == sh->disks - 1)
141                 return 0;
142         else
143                 return sh->qd_idx + 1;
144 }
145 static inline int raid6_next_disk(int disk, int raid_disks)
146 {
147         disk++;
148         return (disk < raid_disks) ? disk : 0;
149 }
150
151 /* When walking through the disks in a raid5, starting at raid6_d0,
152  * We need to map each disk to a 'slot', where the data disks are slot
153  * 0 .. raid_disks-3, the parity disk is raid_disks-2 and the Q disk
154  * is raid_disks-1.  This help does that mapping.
155  */
156 static int raid6_idx_to_slot(int idx, struct stripe_head *sh,
157                              int *count, int syndrome_disks)
158 {
159         int slot;
160
161         if (idx == sh->pd_idx)
162                 return syndrome_disks;
163         if (idx == sh->qd_idx)
164                 return syndrome_disks + 1;
165         slot = (*count)++;
166         return slot;
167 }
168
169 static void return_io(struct bio *return_bi)
170 {
171         struct bio *bi = return_bi;
172         while (bi) {
173
174                 return_bi = bi->bi_next;
175                 bi->bi_next = NULL;
176                 bi->bi_size = 0;
177                 bio_endio(bi, 0);
178                 bi = return_bi;
179         }
180 }
181
182 static void print_raid5_conf (raid5_conf_t *conf);
183
184 static int stripe_operations_active(struct stripe_head *sh)
185 {
186         return sh->check_state || sh->reconstruct_state ||
187                test_bit(STRIPE_BIOFILL_RUN, &sh->state) ||
188                test_bit(STRIPE_COMPUTE_RUN, &sh->state);
189 }
190
191 static void __release_stripe(raid5_conf_t *conf, struct stripe_head *sh)
192 {
193         if (atomic_dec_and_test(&sh->count)) {
194                 BUG_ON(!list_empty(&sh->lru));
195                 BUG_ON(atomic_read(&conf->active_stripes)==0);
196                 if (test_bit(STRIPE_HANDLE, &sh->state)) {
197                         if (test_bit(STRIPE_DELAYED, &sh->state)) {
198                                 list_add_tail(&sh->lru, &conf->delayed_list);
199                                 blk_plug_device(conf->mddev->queue);
200                         } else if (test_bit(STRIPE_BIT_DELAY, &sh->state) &&
201                                    sh->bm_seq - conf->seq_write > 0) {
202                                 list_add_tail(&sh->lru, &conf->bitmap_list);
203                                 blk_plug_device(conf->mddev->queue);
204                         } else {
205                                 clear_bit(STRIPE_BIT_DELAY, &sh->state);
206                                 list_add_tail(&sh->lru, &conf->handle_list);
207                         }
208                         md_wakeup_thread(conf->mddev->thread);
209                 } else {
210                         BUG_ON(stripe_operations_active(sh));
211                         if (test_and_clear_bit(STRIPE_PREREAD_ACTIVE, &sh->state)) {
212                                 atomic_dec(&conf->preread_active_stripes);
213                                 if (atomic_read(&conf->preread_active_stripes) < IO_THRESHOLD)
214                                         md_wakeup_thread(conf->mddev->thread);
215                         }
216                         atomic_dec(&conf->active_stripes);
217                         if (!test_bit(STRIPE_EXPANDING, &sh->state)) {
218                                 list_add_tail(&sh->lru, &conf->inactive_list);
219                                 wake_up(&conf->wait_for_stripe);
220                                 if (conf->retry_read_aligned)
221                                         md_wakeup_thread(conf->mddev->thread);
222                         }
223                 }
224         }
225 }
226
227 static void release_stripe(struct stripe_head *sh)
228 {
229         raid5_conf_t *conf = sh->raid_conf;
230         unsigned long flags;
231
232         spin_lock_irqsave(&conf->device_lock, flags);
233         __release_stripe(conf, sh);
234         spin_unlock_irqrestore(&conf->device_lock, flags);
235 }
236
237 static inline void remove_hash(struct stripe_head *sh)
238 {
239         pr_debug("remove_hash(), stripe %llu\n",
240                 (unsigned long long)sh->sector);
241
242         hlist_del_init(&sh->hash);
243 }
244
245 static inline void insert_hash(raid5_conf_t *conf, struct stripe_head *sh)
246 {
247         struct hlist_head *hp = stripe_hash(conf, sh->sector);
248
249         pr_debug("insert_hash(), stripe %llu\n",
250                 (unsigned long long)sh->sector);
251
252         CHECK_DEVLOCK();
253         hlist_add_head(&sh->hash, hp);
254 }
255
256
257 /* find an idle stripe, make sure it is unhashed, and return it. */
258 static struct stripe_head *get_free_stripe(raid5_conf_t *conf)
259 {
260         struct stripe_head *sh = NULL;
261         struct list_head *first;
262
263         CHECK_DEVLOCK();
264         if (list_empty(&conf->inactive_list))
265                 goto out;
266         first = conf->inactive_list.next;
267         sh = list_entry(first, struct stripe_head, lru);
268         list_del_init(first);
269         remove_hash(sh);
270         atomic_inc(&conf->active_stripes);
271 out:
272         return sh;
273 }
274
275 static void shrink_buffers(struct stripe_head *sh, int num)
276 {
277         struct page *p;
278         int i;
279
280         for (i=0; i<num ; i++) {
281                 p = sh->dev[i].page;
282                 if (!p)
283                         continue;
284                 sh->dev[i].page = NULL;
285                 put_page(p);
286         }
287 }
288
289 static int grow_buffers(struct stripe_head *sh, int num)
290 {
291         int i;
292
293         for (i=0; i<num; i++) {
294                 struct page *page;
295
296                 if (!(page = alloc_page(GFP_KERNEL))) {
297                         return 1;
298                 }
299                 sh->dev[i].page = page;
300         }
301         return 0;
302 }
303
304 static void raid5_build_block(struct stripe_head *sh, int i, int previous);
305 static void stripe_set_idx(sector_t stripe, raid5_conf_t *conf, int previous,
306                             struct stripe_head *sh);
307
308 static void init_stripe(struct stripe_head *sh, sector_t sector, int previous)
309 {
310         raid5_conf_t *conf = sh->raid_conf;
311         int i;
312
313         BUG_ON(atomic_read(&sh->count) != 0);
314         BUG_ON(test_bit(STRIPE_HANDLE, &sh->state));
315         BUG_ON(stripe_operations_active(sh));
316
317         CHECK_DEVLOCK();
318         pr_debug("init_stripe called, stripe %llu\n",
319                 (unsigned long long)sh->sector);
320
321         remove_hash(sh);
322
323         sh->generation = conf->generation - previous;
324         sh->disks = previous ? conf->previous_raid_disks : conf->raid_disks;
325         sh->sector = sector;
326         stripe_set_idx(sector, conf, previous, sh);
327         sh->state = 0;
328
329
330         for (i = sh->disks; i--; ) {
331                 struct r5dev *dev = &sh->dev[i];
332
333                 if (dev->toread || dev->read || dev->towrite || dev->written ||
334                     test_bit(R5_LOCKED, &dev->flags)) {
335                         printk(KERN_ERR "sector=%llx i=%d %p %p %p %p %d\n",
336                                (unsigned long long)sh->sector, i, dev->toread,
337                                dev->read, dev->towrite, dev->written,
338                                test_bit(R5_LOCKED, &dev->flags));
339                         BUG();
340                 }
341                 dev->flags = 0;
342                 raid5_build_block(sh, i, previous);
343         }
344         insert_hash(conf, sh);
345 }
346
347 static struct stripe_head *__find_stripe(raid5_conf_t *conf, sector_t sector,
348                                          short generation)
349 {
350         struct stripe_head *sh;
351         struct hlist_node *hn;
352
353         CHECK_DEVLOCK();
354         pr_debug("__find_stripe, sector %llu\n", (unsigned long long)sector);
355         hlist_for_each_entry(sh, hn, stripe_hash(conf, sector), hash)
356                 if (sh->sector == sector && sh->generation == generation)
357                         return sh;
358         pr_debug("__stripe %llu not in cache\n", (unsigned long long)sector);
359         return NULL;
360 }
361
362 static void unplug_slaves(mddev_t *mddev);
363 static void raid5_unplug_device(struct request_queue *q);
364
365 static struct stripe_head *
366 get_active_stripe(raid5_conf_t *conf, sector_t sector,
367                   int previous, int noblock)
368 {
369         struct stripe_head *sh;
370
371         pr_debug("get_stripe, sector %llu\n", (unsigned long long)sector);
372
373         spin_lock_irq(&conf->device_lock);
374
375         do {
376                 wait_event_lock_irq(conf->wait_for_stripe,
377                                     conf->quiesce == 0,
378                                     conf->device_lock, /* nothing */);
379                 sh = __find_stripe(conf, sector, conf->generation - previous);
380                 if (!sh) {
381                         if (!conf->inactive_blocked)
382                                 sh = get_free_stripe(conf);
383                         if (noblock && sh == NULL)
384                                 break;
385                         if (!sh) {
386                                 conf->inactive_blocked = 1;
387                                 wait_event_lock_irq(conf->wait_for_stripe,
388                                                     !list_empty(&conf->inactive_list) &&
389                                                     (atomic_read(&conf->active_stripes)
390                                                      < (conf->max_nr_stripes *3/4)
391                                                      || !conf->inactive_blocked),
392                                                     conf->device_lock,
393                                                     raid5_unplug_device(conf->mddev->queue)
394                                         );
395                                 conf->inactive_blocked = 0;
396                         } else
397                                 init_stripe(sh, sector, previous);
398                 } else {
399                         if (atomic_read(&sh->count)) {
400                                 BUG_ON(!list_empty(&sh->lru)
401                                     && !test_bit(STRIPE_EXPANDING, &sh->state));
402                         } else {
403                                 if (!test_bit(STRIPE_HANDLE, &sh->state))
404                                         atomic_inc(&conf->active_stripes);
405                                 if (list_empty(&sh->lru) &&
406                                     !test_bit(STRIPE_EXPANDING, &sh->state))
407                                         BUG();
408                                 list_del_init(&sh->lru);
409                         }
410                 }
411         } while (sh == NULL);
412
413         if (sh)
414                 atomic_inc(&sh->count);
415
416         spin_unlock_irq(&conf->device_lock);
417         return sh;
418 }
419
420 static void
421 raid5_end_read_request(struct bio *bi, int error);
422 static void
423 raid5_end_write_request(struct bio *bi, int error);
424
425 static void ops_run_io(struct stripe_head *sh, struct stripe_head_state *s)
426 {
427         raid5_conf_t *conf = sh->raid_conf;
428         int i, disks = sh->disks;
429
430         might_sleep();
431
432         for (i = disks; i--; ) {
433                 int rw;
434                 struct bio *bi;
435                 mdk_rdev_t *rdev;
436                 if (test_and_clear_bit(R5_Wantwrite, &sh->dev[i].flags))
437                         rw = WRITE;
438                 else if (test_and_clear_bit(R5_Wantread, &sh->dev[i].flags))
439                         rw = READ;
440                 else
441                         continue;
442
443                 bi = &sh->dev[i].req;
444
445                 bi->bi_rw = rw;
446                 if (rw == WRITE)
447                         bi->bi_end_io = raid5_end_write_request;
448                 else
449                         bi->bi_end_io = raid5_end_read_request;
450
451                 rcu_read_lock();
452                 rdev = rcu_dereference(conf->disks[i].rdev);
453                 if (rdev && test_bit(Faulty, &rdev->flags))
454                         rdev = NULL;
455                 if (rdev)
456                         atomic_inc(&rdev->nr_pending);
457                 rcu_read_unlock();
458
459                 if (rdev) {
460                         if (s->syncing || s->expanding || s->expanded)
461                                 md_sync_acct(rdev->bdev, STRIPE_SECTORS);
462
463                         set_bit(STRIPE_IO_STARTED, &sh->state);
464
465                         bi->bi_bdev = rdev->bdev;
466                         pr_debug("%s: for %llu schedule op %ld on disc %d\n",
467                                 __func__, (unsigned long long)sh->sector,
468                                 bi->bi_rw, i);
469                         atomic_inc(&sh->count);
470                         bi->bi_sector = sh->sector + rdev->data_offset;
471                         bi->bi_flags = 1 << BIO_UPTODATE;
472                         bi->bi_vcnt = 1;
473                         bi->bi_max_vecs = 1;
474                         bi->bi_idx = 0;
475                         bi->bi_io_vec = &sh->dev[i].vec;
476                         bi->bi_io_vec[0].bv_len = STRIPE_SIZE;
477                         bi->bi_io_vec[0].bv_offset = 0;
478                         bi->bi_size = STRIPE_SIZE;
479                         bi->bi_next = NULL;
480                         if (rw == WRITE &&
481                             test_bit(R5_ReWrite, &sh->dev[i].flags))
482                                 atomic_add(STRIPE_SECTORS,
483                                         &rdev->corrected_errors);
484                         generic_make_request(bi);
485                 } else {
486                         if (rw == WRITE)
487                                 set_bit(STRIPE_DEGRADED, &sh->state);
488                         pr_debug("skip op %ld on disc %d for sector %llu\n",
489                                 bi->bi_rw, i, (unsigned long long)sh->sector);
490                         clear_bit(R5_LOCKED, &sh->dev[i].flags);
491                         set_bit(STRIPE_HANDLE, &sh->state);
492                 }
493         }
494 }
495
496 static struct dma_async_tx_descriptor *
497 async_copy_data(int frombio, struct bio *bio, struct page *page,
498         sector_t sector, struct dma_async_tx_descriptor *tx)
499 {
500         struct bio_vec *bvl;
501         struct page *bio_page;
502         int i;
503         int page_offset;
504         struct async_submit_ctl submit;
505
506         if (bio->bi_sector >= sector)
507                 page_offset = (signed)(bio->bi_sector - sector) * 512;
508         else
509                 page_offset = (signed)(sector - bio->bi_sector) * -512;
510
511         init_async_submit(&submit, 0, tx, NULL, NULL, NULL);
512         bio_for_each_segment(bvl, bio, i) {
513                 int len = bio_iovec_idx(bio, i)->bv_len;
514                 int clen;
515                 int b_offset = 0;
516
517                 if (page_offset < 0) {
518                         b_offset = -page_offset;
519                         page_offset += b_offset;
520                         len -= b_offset;
521                 }
522
523                 if (len > 0 && page_offset + len > STRIPE_SIZE)
524                         clen = STRIPE_SIZE - page_offset;
525                 else
526                         clen = len;
527
528                 if (clen > 0) {
529                         b_offset += bio_iovec_idx(bio, i)->bv_offset;
530                         bio_page = bio_iovec_idx(bio, i)->bv_page;
531                         if (frombio)
532                                 tx = async_memcpy(page, bio_page, page_offset,
533                                                   b_offset, clen, &submit);
534                         else
535                                 tx = async_memcpy(bio_page, page, b_offset,
536                                                   page_offset, clen, &submit);
537                 }
538                 /* chain the operations */
539                 submit.depend_tx = tx;
540
541                 if (clen < len) /* hit end of page */
542                         break;
543                 page_offset +=  len;
544         }
545
546         return tx;
547 }
548
549 static void ops_complete_biofill(void *stripe_head_ref)
550 {
551         struct stripe_head *sh = stripe_head_ref;
552         struct bio *return_bi = NULL;
553         raid5_conf_t *conf = sh->raid_conf;
554         int i;
555
556         pr_debug("%s: stripe %llu\n", __func__,
557                 (unsigned long long)sh->sector);
558
559         /* clear completed biofills */
560         spin_lock_irq(&conf->device_lock);
561         for (i = sh->disks; i--; ) {
562                 struct r5dev *dev = &sh->dev[i];
563
564                 /* acknowledge completion of a biofill operation */
565                 /* and check if we need to reply to a read request,
566                  * new R5_Wantfill requests are held off until
567                  * !STRIPE_BIOFILL_RUN
568                  */
569                 if (test_and_clear_bit(R5_Wantfill, &dev->flags)) {
570                         struct bio *rbi, *rbi2;
571
572                         BUG_ON(!dev->read);
573                         rbi = dev->read;
574                         dev->read = NULL;
575                         while (rbi && rbi->bi_sector <
576                                 dev->sector + STRIPE_SECTORS) {
577                                 rbi2 = r5_next_bio(rbi, dev->sector);
578                                 if (!raid5_dec_bi_phys_segments(rbi)) {
579                                         rbi->bi_next = return_bi;
580                                         return_bi = rbi;
581                                 }
582                                 rbi = rbi2;
583                         }
584                 }
585         }
586         spin_unlock_irq(&conf->device_lock);
587         clear_bit(STRIPE_BIOFILL_RUN, &sh->state);
588
589         return_io(return_bi);
590
591         set_bit(STRIPE_HANDLE, &sh->state);
592         release_stripe(sh);
593 }
594
595 static void ops_run_biofill(struct stripe_head *sh)
596 {
597         struct dma_async_tx_descriptor *tx = NULL;
598         raid5_conf_t *conf = sh->raid_conf;
599         struct async_submit_ctl submit;
600         int i;
601
602         pr_debug("%s: stripe %llu\n", __func__,
603                 (unsigned long long)sh->sector);
604
605         for (i = sh->disks; i--; ) {
606                 struct r5dev *dev = &sh->dev[i];
607                 if (test_bit(R5_Wantfill, &dev->flags)) {
608                         struct bio *rbi;
609                         spin_lock_irq(&conf->device_lock);
610                         dev->read = rbi = dev->toread;
611                         dev->toread = NULL;
612                         spin_unlock_irq(&conf->device_lock);
613                         while (rbi && rbi->bi_sector <
614                                 dev->sector + STRIPE_SECTORS) {
615                                 tx = async_copy_data(0, rbi, dev->page,
616                                         dev->sector, tx);
617                                 rbi = r5_next_bio(rbi, dev->sector);
618                         }
619                 }
620         }
621
622         atomic_inc(&sh->count);
623         init_async_submit(&submit, ASYNC_TX_ACK, tx, ops_complete_biofill, sh, NULL);
624         async_trigger_callback(&submit);
625 }
626
627 static void mark_target_uptodate(struct stripe_head *sh, int target)
628 {
629         struct r5dev *tgt;
630
631         if (target < 0)
632                 return;
633
634         tgt = &sh->dev[target];
635         set_bit(R5_UPTODATE, &tgt->flags);
636         BUG_ON(!test_bit(R5_Wantcompute, &tgt->flags));
637         clear_bit(R5_Wantcompute, &tgt->flags);
638 }
639
640 static void ops_complete_compute(void *stripe_head_ref)
641 {
642         struct stripe_head *sh = stripe_head_ref;
643
644         pr_debug("%s: stripe %llu\n", __func__,
645                 (unsigned long long)sh->sector);
646
647         /* mark the computed target(s) as uptodate */
648         mark_target_uptodate(sh, sh->ops.target);
649         mark_target_uptodate(sh, sh->ops.target2);
650
651         clear_bit(STRIPE_COMPUTE_RUN, &sh->state);
652         if (sh->check_state == check_state_compute_run)
653                 sh->check_state = check_state_compute_result;
654         set_bit(STRIPE_HANDLE, &sh->state);
655         release_stripe(sh);
656 }
657
658 /* return a pointer to the address conversion region of the scribble buffer */
659 static addr_conv_t *to_addr_conv(struct stripe_head *sh,
660                                  struct raid5_percpu *percpu)
661 {
662         return percpu->scribble + sizeof(struct page *) * (sh->disks + 2);
663 }
664
665 static struct dma_async_tx_descriptor *
666 ops_run_compute5(struct stripe_head *sh, struct raid5_percpu *percpu)
667 {
668         int disks = sh->disks;
669         struct page **xor_srcs = percpu->scribble;
670         int target = sh->ops.target;
671         struct r5dev *tgt = &sh->dev[target];
672         struct page *xor_dest = tgt->page;
673         int count = 0;
674         struct dma_async_tx_descriptor *tx;
675         struct async_submit_ctl submit;
676         int i;
677
678         pr_debug("%s: stripe %llu block: %d\n",
679                 __func__, (unsigned long long)sh->sector, target);
680         BUG_ON(!test_bit(R5_Wantcompute, &tgt->flags));
681
682         for (i = disks; i--; )
683                 if (i != target)
684                         xor_srcs[count++] = sh->dev[i].page;
685
686         atomic_inc(&sh->count);
687
688         init_async_submit(&submit, ASYNC_TX_XOR_ZERO_DST, NULL,
689                           ops_complete_compute, sh, to_addr_conv(sh, percpu));
690         if (unlikely(count == 1))
691                 tx = async_memcpy(xor_dest, xor_srcs[0], 0, 0, STRIPE_SIZE, &submit);
692         else
693                 tx = async_xor(xor_dest, xor_srcs, 0, count, STRIPE_SIZE, &submit);
694
695         return tx;
696 }
697
698 /* set_syndrome_sources - populate source buffers for gen_syndrome
699  * @srcs - (struct page *) array of size sh->disks
700  * @sh - stripe_head to parse
701  *
702  * Populates srcs in proper layout order for the stripe and returns the
703  * 'count' of sources to be used in a call to async_gen_syndrome.  The P
704  * destination buffer is recorded in srcs[count] and the Q destination
705  * is recorded in srcs[count+1]].
706  */
707 static int set_syndrome_sources(struct page **srcs, struct stripe_head *sh)
708 {
709         int disks = sh->disks;
710         int syndrome_disks = sh->ddf_layout ? disks : (disks - 2);
711         int d0_idx = raid6_d0(sh);
712         int count;
713         int i;
714
715         for (i = 0; i < disks; i++)
716                 srcs[i] = (void *)raid6_empty_zero_page;
717
718         count = 0;
719         i = d0_idx;
720         do {
721                 int slot = raid6_idx_to_slot(i, sh, &count, syndrome_disks);
722
723                 srcs[slot] = sh->dev[i].page;
724                 i = raid6_next_disk(i, disks);
725         } while (i != d0_idx);
726         BUG_ON(count != syndrome_disks);
727
728         return count;
729 }
730
731 static struct dma_async_tx_descriptor *
732 ops_run_compute6_1(struct stripe_head *sh, struct raid5_percpu *percpu)
733 {
734         int disks = sh->disks;
735         struct page **blocks = percpu->scribble;
736         int target;
737         int qd_idx = sh->qd_idx;
738         struct dma_async_tx_descriptor *tx;
739         struct async_submit_ctl submit;
740         struct r5dev *tgt;
741         struct page *dest;
742         int i;
743         int count;
744
745         if (sh->ops.target < 0)
746                 target = sh->ops.target2;
747         else if (sh->ops.target2 < 0)
748                 target = sh->ops.target;
749         else
750                 /* we should only have one valid target */
751                 BUG();
752         BUG_ON(target < 0);
753         pr_debug("%s: stripe %llu block: %d\n",
754                 __func__, (unsigned long long)sh->sector, target);
755
756         tgt = &sh->dev[target];
757         BUG_ON(!test_bit(R5_Wantcompute, &tgt->flags));
758         dest = tgt->page;
759
760         atomic_inc(&sh->count);
761
762         if (target == qd_idx) {
763                 count = set_syndrome_sources(blocks, sh);
764                 blocks[count] = NULL; /* regenerating p is not necessary */
765                 BUG_ON(blocks[count+1] != dest); /* q should already be set */
766                 init_async_submit(&submit, 0, NULL, ops_complete_compute, sh,
767                                   to_addr_conv(sh, percpu));
768                 tx = async_gen_syndrome(blocks, 0, count+2, STRIPE_SIZE, &submit);
769         } else {
770                 /* Compute any data- or p-drive using XOR */
771                 count = 0;
772                 for (i = disks; i-- ; ) {
773                         if (i == target || i == qd_idx)
774                                 continue;
775                         blocks[count++] = sh->dev[i].page;
776                 }
777
778                 init_async_submit(&submit, ASYNC_TX_XOR_ZERO_DST, NULL,
779                                   ops_complete_compute, sh,
780                                   to_addr_conv(sh, percpu));
781                 tx = async_xor(dest, blocks, 0, count, STRIPE_SIZE, &submit);
782         }
783
784         return tx;
785 }
786
787 static struct dma_async_tx_descriptor *
788 ops_run_compute6_2(struct stripe_head *sh, struct raid5_percpu *percpu)
789 {
790         int i, count, disks = sh->disks;
791         int syndrome_disks = sh->ddf_layout ? disks : disks-2;
792         int d0_idx = raid6_d0(sh);
793         int faila = -1, failb = -1;
794         int target = sh->ops.target;
795         int target2 = sh->ops.target2;
796         struct r5dev *tgt = &sh->dev[target];
797         struct r5dev *tgt2 = &sh->dev[target2];
798         struct dma_async_tx_descriptor *tx;
799         struct page **blocks = percpu->scribble;
800         struct async_submit_ctl submit;
801
802         pr_debug("%s: stripe %llu block1: %d block2: %d\n",
803                  __func__, (unsigned long long)sh->sector, target, target2);
804         BUG_ON(target < 0 || target2 < 0);
805         BUG_ON(!test_bit(R5_Wantcompute, &tgt->flags));
806         BUG_ON(!test_bit(R5_Wantcompute, &tgt2->flags));
807
808         /* we need to open-code set_syndrome_sources to handle to the
809          * slot number conversion for 'faila' and 'failb'
810          */
811         for (i = 0; i < disks ; i++)
812                 blocks[i] = (void *)raid6_empty_zero_page;
813         count = 0;
814         i = d0_idx;
815         do {
816                 int slot = raid6_idx_to_slot(i, sh, &count, syndrome_disks);
817
818                 blocks[slot] = sh->dev[i].page;
819
820                 if (i == target)
821                         faila = slot;
822                 if (i == target2)
823                         failb = slot;
824                 i = raid6_next_disk(i, disks);
825         } while (i != d0_idx);
826         BUG_ON(count != syndrome_disks);
827
828         BUG_ON(faila == failb);
829         if (failb < faila)
830                 swap(faila, failb);
831         pr_debug("%s: stripe: %llu faila: %d failb: %d\n",
832                  __func__, (unsigned long long)sh->sector, faila, failb);
833
834         atomic_inc(&sh->count);
835
836         if (failb == syndrome_disks+1) {
837                 /* Q disk is one of the missing disks */
838                 if (faila == syndrome_disks) {
839                         /* Missing P+Q, just recompute */
840                         init_async_submit(&submit, 0, NULL, ops_complete_compute,
841                                           sh, to_addr_conv(sh, percpu));
842                         return async_gen_syndrome(blocks, 0, count+2,
843                                                   STRIPE_SIZE, &submit);
844                 } else {
845                         struct page *dest;
846                         int data_target;
847                         int qd_idx = sh->qd_idx;
848
849                         /* Missing D+Q: recompute D from P, then recompute Q */
850                         if (target == qd_idx)
851                                 data_target = target2;
852                         else
853                                 data_target = target;
854
855                         count = 0;
856                         for (i = disks; i-- ; ) {
857                                 if (i == data_target || i == qd_idx)
858                                         continue;
859                                 blocks[count++] = sh->dev[i].page;
860                         }
861                         dest = sh->dev[data_target].page;
862                         init_async_submit(&submit, ASYNC_TX_XOR_ZERO_DST, NULL,
863                                           NULL, NULL, to_addr_conv(sh, percpu));
864                         tx = async_xor(dest, blocks, 0, count, STRIPE_SIZE,
865                                        &submit);
866
867                         count = set_syndrome_sources(blocks, sh);
868                         init_async_submit(&submit, 0, tx, ops_complete_compute,
869                                           sh, to_addr_conv(sh, percpu));
870                         return async_gen_syndrome(blocks, 0, count+2,
871                                                   STRIPE_SIZE, &submit);
872                 }
873         }
874
875         init_async_submit(&submit, 0, NULL, ops_complete_compute, sh,
876                           to_addr_conv(sh, percpu));
877         if (failb == syndrome_disks) {
878                 /* We're missing D+P. */
879                 return async_raid6_datap_recov(syndrome_disks+2, STRIPE_SIZE,
880                                                faila, blocks, &submit);
881         } else {
882                 /* We're missing D+D. */
883                 return async_raid6_2data_recov(syndrome_disks+2, STRIPE_SIZE,
884                                                faila, failb, blocks, &submit);
885         }
886 }
887
888
889 static void ops_complete_prexor(void *stripe_head_ref)
890 {
891         struct stripe_head *sh = stripe_head_ref;
892
893         pr_debug("%s: stripe %llu\n", __func__,
894                 (unsigned long long)sh->sector);
895 }
896
897 static struct dma_async_tx_descriptor *
898 ops_run_prexor(struct stripe_head *sh, struct raid5_percpu *percpu,
899                struct dma_async_tx_descriptor *tx)
900 {
901         int disks = sh->disks;
902         struct page **xor_srcs = percpu->scribble;
903         int count = 0, pd_idx = sh->pd_idx, i;
904         struct async_submit_ctl submit;
905
906         /* existing parity data subtracted */
907         struct page *xor_dest = xor_srcs[count++] = sh->dev[pd_idx].page;
908
909         pr_debug("%s: stripe %llu\n", __func__,
910                 (unsigned long long)sh->sector);
911
912         for (i = disks; i--; ) {
913                 struct r5dev *dev = &sh->dev[i];
914                 /* Only process blocks that are known to be uptodate */
915                 if (test_bit(R5_Wantdrain, &dev->flags))
916                         xor_srcs[count++] = dev->page;
917         }
918
919         init_async_submit(&submit, ASYNC_TX_XOR_DROP_DST, tx,
920                           ops_complete_prexor, sh, to_addr_conv(sh, percpu));
921         tx = async_xor(xor_dest, xor_srcs, 0, count, STRIPE_SIZE, &submit);
922
923         return tx;
924 }
925
926 static struct dma_async_tx_descriptor *
927 ops_run_biodrain(struct stripe_head *sh, struct dma_async_tx_descriptor *tx)
928 {
929         int disks = sh->disks;
930         int i;
931
932         pr_debug("%s: stripe %llu\n", __func__,
933                 (unsigned long long)sh->sector);
934
935         for (i = disks; i--; ) {
936                 struct r5dev *dev = &sh->dev[i];
937                 struct bio *chosen;
938
939                 if (test_and_clear_bit(R5_Wantdrain, &dev->flags)) {
940                         struct bio *wbi;
941
942                         spin_lock(&sh->lock);
943                         chosen = dev->towrite;
944                         dev->towrite = NULL;
945                         BUG_ON(dev->written);
946                         wbi = dev->written = chosen;
947                         spin_unlock(&sh->lock);
948
949                         while (wbi && wbi->bi_sector <
950                                 dev->sector + STRIPE_SECTORS) {
951                                 tx = async_copy_data(1, wbi, dev->page,
952                                         dev->sector, tx);
953                                 wbi = r5_next_bio(wbi, dev->sector);
954                         }
955                 }
956         }
957
958         return tx;
959 }
960
961 static void ops_complete_reconstruct(void *stripe_head_ref)
962 {
963         struct stripe_head *sh = stripe_head_ref;
964         int disks = sh->disks;
965         int pd_idx = sh->pd_idx;
966         int qd_idx = sh->qd_idx;
967         int i;
968
969         pr_debug("%s: stripe %llu\n", __func__,
970                 (unsigned long long)sh->sector);
971
972         for (i = disks; i--; ) {
973                 struct r5dev *dev = &sh->dev[i];
974
975                 if (dev->written || i == pd_idx || i == qd_idx)
976                         set_bit(R5_UPTODATE, &dev->flags);
977         }
978
979         if (sh->reconstruct_state == reconstruct_state_drain_run)
980                 sh->reconstruct_state = reconstruct_state_drain_result;
981         else if (sh->reconstruct_state == reconstruct_state_prexor_drain_run)
982                 sh->reconstruct_state = reconstruct_state_prexor_drain_result;
983         else {
984                 BUG_ON(sh->reconstruct_state != reconstruct_state_run);
985                 sh->reconstruct_state = reconstruct_state_result;
986         }
987
988         set_bit(STRIPE_HANDLE, &sh->state);
989         release_stripe(sh);
990 }
991
992 static void
993 ops_run_reconstruct5(struct stripe_head *sh, struct raid5_percpu *percpu,
994                      struct dma_async_tx_descriptor *tx)
995 {
996         int disks = sh->disks;
997         struct page **xor_srcs = percpu->scribble;
998         struct async_submit_ctl submit;
999         int count = 0, pd_idx = sh->pd_idx, i;
1000         struct page *xor_dest;
1001         int prexor = 0;
1002         unsigned long flags;
1003
1004         pr_debug("%s: stripe %llu\n", __func__,
1005                 (unsigned long long)sh->sector);
1006
1007         /* check if prexor is active which means only process blocks
1008          * that are part of a read-modify-write (written)
1009          */
1010         if (sh->reconstruct_state == reconstruct_state_prexor_drain_run) {
1011                 prexor = 1;
1012                 xor_dest = xor_srcs[count++] = sh->dev[pd_idx].page;
1013                 for (i = disks; i--; ) {
1014                         struct r5dev *dev = &sh->dev[i];
1015                         if (dev->written)
1016                                 xor_srcs[count++] = dev->page;
1017                 }
1018         } else {
1019                 xor_dest = sh->dev[pd_idx].page;
1020                 for (i = disks; i--; ) {
1021                         struct r5dev *dev = &sh->dev[i];
1022                         if (i != pd_idx)
1023                                 xor_srcs[count++] = dev->page;
1024                 }
1025         }
1026
1027         /* 1/ if we prexor'd then the dest is reused as a source
1028          * 2/ if we did not prexor then we are redoing the parity
1029          * set ASYNC_TX_XOR_DROP_DST and ASYNC_TX_XOR_ZERO_DST
1030          * for the synchronous xor case
1031          */
1032         flags = ASYNC_TX_ACK |
1033                 (prexor ? ASYNC_TX_XOR_DROP_DST : ASYNC_TX_XOR_ZERO_DST);
1034
1035         atomic_inc(&sh->count);
1036
1037         init_async_submit(&submit, flags, tx, ops_complete_reconstruct, sh,
1038                           to_addr_conv(sh, percpu));
1039         if (unlikely(count == 1))
1040                 tx = async_memcpy(xor_dest, xor_srcs[0], 0, 0, STRIPE_SIZE, &submit);
1041         else
1042                 tx = async_xor(xor_dest, xor_srcs, 0, count, STRIPE_SIZE, &submit);
1043 }
1044
1045 static void
1046 ops_run_reconstruct6(struct stripe_head *sh, struct raid5_percpu *percpu,
1047                      struct dma_async_tx_descriptor *tx)
1048 {
1049         struct async_submit_ctl submit;
1050         struct page **blocks = percpu->scribble;
1051         int count;
1052
1053         pr_debug("%s: stripe %llu\n", __func__, (unsigned long long)sh->sector);
1054
1055         count = set_syndrome_sources(blocks, sh);
1056
1057         atomic_inc(&sh->count);
1058
1059         init_async_submit(&submit, ASYNC_TX_ACK, tx, ops_complete_reconstruct,
1060                           sh, to_addr_conv(sh, percpu));
1061         async_gen_syndrome(blocks, 0, count+2, STRIPE_SIZE,  &submit);
1062 }
1063
1064 static void ops_complete_check(void *stripe_head_ref)
1065 {
1066         struct stripe_head *sh = stripe_head_ref;
1067
1068         pr_debug("%s: stripe %llu\n", __func__,
1069                 (unsigned long long)sh->sector);
1070
1071         sh->check_state = check_state_check_result;
1072         set_bit(STRIPE_HANDLE, &sh->state);
1073         release_stripe(sh);
1074 }
1075
1076 static void ops_run_check_p(struct stripe_head *sh, struct raid5_percpu *percpu)
1077 {
1078         int disks = sh->disks;
1079         int pd_idx = sh->pd_idx;
1080         int qd_idx = sh->qd_idx;
1081         struct page *xor_dest;
1082         struct page **xor_srcs = percpu->scribble;
1083         struct dma_async_tx_descriptor *tx;
1084         struct async_submit_ctl submit;
1085         int count;
1086         int i;
1087
1088         pr_debug("%s: stripe %llu\n", __func__,
1089                 (unsigned long long)sh->sector);
1090
1091         count = 0;
1092         xor_dest = sh->dev[pd_idx].page;
1093         xor_srcs[count++] = xor_dest;
1094         for (i = disks; i--; ) {
1095                 if (i == pd_idx || i == qd_idx)
1096                         continue;
1097                 xor_srcs[count++] = sh->dev[i].page;
1098         }
1099
1100         init_async_submit(&submit, 0, NULL, NULL, NULL,
1101                           to_addr_conv(sh, percpu));
1102         tx = async_xor_val(xor_dest, xor_srcs, 0, count, STRIPE_SIZE,
1103                            &sh->ops.zero_sum_result, &submit);
1104
1105         atomic_inc(&sh->count);
1106         init_async_submit(&submit, ASYNC_TX_ACK, tx, ops_complete_check, sh, NULL);
1107         tx = async_trigger_callback(&submit);
1108 }
1109
1110 static void ops_run_check_pq(struct stripe_head *sh, struct raid5_percpu *percpu, int checkp)
1111 {
1112         struct page **srcs = percpu->scribble;
1113         struct async_submit_ctl submit;
1114         int count;
1115
1116         pr_debug("%s: stripe %llu checkp: %d\n", __func__,
1117                 (unsigned long long)sh->sector, checkp);
1118
1119         count = set_syndrome_sources(srcs, sh);
1120         if (!checkp)
1121                 srcs[count] = NULL;
1122
1123         atomic_inc(&sh->count);
1124         init_async_submit(&submit, ASYNC_TX_ACK, NULL, ops_complete_check,
1125                           sh, to_addr_conv(sh, percpu));
1126         async_syndrome_val(srcs, 0, count+2, STRIPE_SIZE,
1127                            &sh->ops.zero_sum_result, percpu->spare_page, &submit);
1128 }
1129
1130 static void raid_run_ops(struct stripe_head *sh, unsigned long ops_request)
1131 {
1132         int overlap_clear = 0, i, disks = sh->disks;
1133         struct dma_async_tx_descriptor *tx = NULL;
1134         raid5_conf_t *conf = sh->raid_conf;
1135         int level = conf->level;
1136         struct raid5_percpu *percpu;
1137         unsigned long cpu;
1138
1139         cpu = get_cpu();
1140         percpu = per_cpu_ptr(conf->percpu, cpu);
1141         if (test_bit(STRIPE_OP_BIOFILL, &ops_request)) {
1142                 ops_run_biofill(sh);
1143                 overlap_clear++;
1144         }
1145
1146         if (test_bit(STRIPE_OP_COMPUTE_BLK, &ops_request)) {
1147                 if (level < 6)
1148                         tx = ops_run_compute5(sh, percpu);
1149                 else {
1150                         if (sh->ops.target2 < 0 || sh->ops.target < 0)
1151                                 tx = ops_run_compute6_1(sh, percpu);
1152                         else
1153                                 tx = ops_run_compute6_2(sh, percpu);
1154                 }
1155                 /* terminate the chain if reconstruct is not set to be run */
1156                 if (tx && !test_bit(STRIPE_OP_RECONSTRUCT, &ops_request))
1157                         async_tx_ack(tx);
1158         }
1159
1160         if (test_bit(STRIPE_OP_PREXOR, &ops_request))
1161                 tx = ops_run_prexor(sh, percpu, tx);
1162
1163         if (test_bit(STRIPE_OP_BIODRAIN, &ops_request)) {
1164                 tx = ops_run_biodrain(sh, tx);
1165                 overlap_clear++;
1166         }
1167
1168         if (test_bit(STRIPE_OP_RECONSTRUCT, &ops_request)) {
1169                 if (level < 6)
1170                         ops_run_reconstruct5(sh, percpu, tx);
1171                 else
1172                         ops_run_reconstruct6(sh, percpu, tx);
1173         }
1174
1175         if (test_bit(STRIPE_OP_CHECK, &ops_request)) {
1176                 if (sh->check_state == check_state_run)
1177                         ops_run_check_p(sh, percpu);
1178                 else if (sh->check_state == check_state_run_q)
1179                         ops_run_check_pq(sh, percpu, 0);
1180                 else if (sh->check_state == check_state_run_pq)
1181                         ops_run_check_pq(sh, percpu, 1);
1182                 else
1183                         BUG();
1184         }
1185
1186         if (overlap_clear)
1187                 for (i = disks; i--; ) {
1188                         struct r5dev *dev = &sh->dev[i];
1189                         if (test_and_clear_bit(R5_Overlap, &dev->flags))
1190                                 wake_up(&sh->raid_conf->wait_for_overlap);
1191                 }
1192         put_cpu();
1193 }
1194
1195 static int grow_one_stripe(raid5_conf_t *conf)
1196 {
1197         struct stripe_head *sh;
1198         sh = kmem_cache_alloc(conf->slab_cache, GFP_KERNEL);
1199         if (!sh)
1200                 return 0;
1201         memset(sh, 0, sizeof(*sh) + (conf->raid_disks-1)*sizeof(struct r5dev));
1202         sh->raid_conf = conf;
1203         spin_lock_init(&sh->lock);
1204
1205         if (grow_buffers(sh, conf->raid_disks)) {
1206                 shrink_buffers(sh, conf->raid_disks);
1207                 kmem_cache_free(conf->slab_cache, sh);
1208                 return 0;
1209         }
1210         sh->disks = conf->raid_disks;
1211         /* we just created an active stripe so... */
1212         atomic_set(&sh->count, 1);
1213         atomic_inc(&conf->active_stripes);
1214         INIT_LIST_HEAD(&sh->lru);
1215         release_stripe(sh);
1216         return 1;
1217 }
1218
1219 static int grow_stripes(raid5_conf_t *conf, int num)
1220 {
1221         struct kmem_cache *sc;
1222         int devs = conf->raid_disks;
1223
1224         sprintf(conf->cache_name[0],
1225                 "raid%d-%s", conf->level, mdname(conf->mddev));
1226         sprintf(conf->cache_name[1],
1227                 "raid%d-%s-alt", conf->level, mdname(conf->mddev));
1228         conf->active_name = 0;
1229         sc = kmem_cache_create(conf->cache_name[conf->active_name],
1230                                sizeof(struct stripe_head)+(devs-1)*sizeof(struct r5dev),
1231                                0, 0, NULL);
1232         if (!sc)
1233                 return 1;
1234         conf->slab_cache = sc;
1235         conf->pool_size = devs;
1236         while (num--)
1237                 if (!grow_one_stripe(conf))
1238                         return 1;
1239         return 0;
1240 }
1241
1242 /**
1243  * scribble_len - return the required size of the scribble region
1244  * @num - total number of disks in the array
1245  *
1246  * The size must be enough to contain:
1247  * 1/ a struct page pointer for each device in the array +2
1248  * 2/ room to convert each entry in (1) to its corresponding dma
1249  *    (dma_map_page()) or page (page_address()) address.
1250  *
1251  * Note: the +2 is for the destination buffers of the ddf/raid6 case where we
1252  * calculate over all devices (not just the data blocks), using zeros in place
1253  * of the P and Q blocks.
1254  */
1255 static size_t scribble_len(int num)
1256 {
1257         size_t len;
1258
1259         len = sizeof(struct page *) * (num+2) + sizeof(addr_conv_t) * (num+2);
1260
1261         return len;
1262 }
1263
1264 static int resize_stripes(raid5_conf_t *conf, int newsize)
1265 {
1266         /* Make all the stripes able to hold 'newsize' devices.
1267          * New slots in each stripe get 'page' set to a new page.
1268          *
1269          * This happens in stages:
1270          * 1/ create a new kmem_cache and allocate the required number of
1271          *    stripe_heads.
1272          * 2/ gather all the old stripe_heads and tranfer the pages across
1273          *    to the new stripe_heads.  This will have the side effect of
1274          *    freezing the array as once all stripe_heads have been collected,
1275          *    no IO will be possible.  Old stripe heads are freed once their
1276          *    pages have been transferred over, and the old kmem_cache is
1277          *    freed when all stripes are done.
1278          * 3/ reallocate conf->disks to be suitable bigger.  If this fails,
1279          *    we simple return a failre status - no need to clean anything up.
1280          * 4/ allocate new pages for the new slots in the new stripe_heads.
1281          *    If this fails, we don't bother trying the shrink the
1282          *    stripe_heads down again, we just leave them as they are.
1283          *    As each stripe_head is processed the new one is released into
1284          *    active service.
1285          *
1286          * Once step2 is started, we cannot afford to wait for a write,
1287          * so we use GFP_NOIO allocations.
1288          */
1289         struct stripe_head *osh, *nsh;
1290         LIST_HEAD(newstripes);
1291         struct disk_info *ndisks;
1292         unsigned long cpu;
1293         int err;
1294         struct kmem_cache *sc;
1295         int i;
1296
1297         if (newsize <= conf->pool_size)
1298                 return 0; /* never bother to shrink */
1299
1300         err = md_allow_write(conf->mddev);
1301         if (err)
1302                 return err;
1303
1304         /* Step 1 */
1305         sc = kmem_cache_create(conf->cache_name[1-conf->active_name],
1306                                sizeof(struct stripe_head)+(newsize-1)*sizeof(struct r5dev),
1307                                0, 0, NULL);
1308         if (!sc)
1309                 return -ENOMEM;
1310
1311         for (i = conf->max_nr_stripes; i; i--) {
1312                 nsh = kmem_cache_alloc(sc, GFP_KERNEL);
1313                 if (!nsh)
1314                         break;
1315
1316                 memset(nsh, 0, sizeof(*nsh) + (newsize-1)*sizeof(struct r5dev));
1317
1318                 nsh->raid_conf = conf;
1319                 spin_lock_init(&nsh->lock);
1320
1321                 list_add(&nsh->lru, &newstripes);
1322         }
1323         if (i) {
1324                 /* didn't get enough, give up */
1325                 while (!list_empty(&newstripes)) {
1326                         nsh = list_entry(newstripes.next, struct stripe_head, lru);
1327                         list_del(&nsh->lru);
1328                         kmem_cache_free(sc, nsh);
1329                 }
1330                 kmem_cache_destroy(sc);
1331                 return -ENOMEM;
1332         }
1333         /* Step 2 - Must use GFP_NOIO now.
1334          * OK, we have enough stripes, start collecting inactive
1335          * stripes and copying them over
1336          */
1337         list_for_each_entry(nsh, &newstripes, lru) {
1338                 spin_lock_irq(&conf->device_lock);
1339                 wait_event_lock_irq(conf->wait_for_stripe,
1340                                     !list_empty(&conf->inactive_list),
1341                                     conf->device_lock,
1342                                     unplug_slaves(conf->mddev)
1343                         );
1344                 osh = get_free_stripe(conf);
1345                 spin_unlock_irq(&conf->device_lock);
1346                 atomic_set(&nsh->count, 1);
1347                 for(i=0; i<conf->pool_size; i++)
1348                         nsh->dev[i].page = osh->dev[i].page;
1349                 for( ; i<newsize; i++)
1350                         nsh->dev[i].page = NULL;
1351                 kmem_cache_free(conf->slab_cache, osh);
1352         }
1353         kmem_cache_destroy(conf->slab_cache);
1354
1355         /* Step 3.
1356          * At this point, we are holding all the stripes so the array
1357          * is completely stalled, so now is a good time to resize
1358          * conf->disks and the scribble region
1359          */
1360         ndisks = kzalloc(newsize * sizeof(struct disk_info), GFP_NOIO);
1361         if (ndisks) {
1362                 for (i=0; i<conf->raid_disks; i++)
1363                         ndisks[i] = conf->disks[i];
1364                 kfree(conf->disks);
1365                 conf->disks = ndisks;
1366         } else
1367                 err = -ENOMEM;
1368
1369         get_online_cpus();
1370         conf->scribble_len = scribble_len(newsize);
1371         for_each_present_cpu(cpu) {
1372                 struct raid5_percpu *percpu;
1373                 void *scribble;
1374
1375                 percpu = per_cpu_ptr(conf->percpu, cpu);
1376                 scribble = kmalloc(conf->scribble_len, GFP_NOIO);
1377
1378                 if (scribble) {
1379                         kfree(percpu->scribble);
1380                         percpu->scribble = scribble;
1381                 } else {
1382                         err = -ENOMEM;
1383                         break;
1384                 }
1385         }
1386         put_online_cpus();
1387
1388         /* Step 4, return new stripes to service */
1389         while(!list_empty(&newstripes)) {
1390                 nsh = list_entry(newstripes.next, struct stripe_head, lru);
1391                 list_del_init(&nsh->lru);
1392
1393                 for (i=conf->raid_disks; i < newsize; i++)
1394                         if (nsh->dev[i].page == NULL) {
1395                                 struct page *p = alloc_page(GFP_NOIO);
1396                                 nsh->dev[i].page = p;
1397                                 if (!p)
1398                                         err = -ENOMEM;
1399                         }
1400                 release_stripe(nsh);
1401         }
1402         /* critical section pass, GFP_NOIO no longer needed */
1403
1404         conf->slab_cache = sc;
1405         conf->active_name = 1-conf->active_name;
1406         conf->pool_size = newsize;
1407         return err;
1408 }
1409
1410 static int drop_one_stripe(raid5_conf_t *conf)
1411 {
1412         struct stripe_head *sh;
1413
1414         spin_lock_irq(&conf->device_lock);
1415         sh = get_free_stripe(conf);
1416         spin_unlock_irq(&conf->device_lock);
1417         if (!sh)
1418                 return 0;
1419         BUG_ON(atomic_read(&sh->count));
1420         shrink_buffers(sh, conf->pool_size);
1421         kmem_cache_free(conf->slab_cache, sh);
1422         atomic_dec(&conf->active_stripes);
1423         return 1;
1424 }
1425
1426 static void shrink_stripes(raid5_conf_t *conf)
1427 {
1428         while (drop_one_stripe(conf))
1429                 ;
1430
1431         if (conf->slab_cache)
1432                 kmem_cache_destroy(conf->slab_cache);
1433         conf->slab_cache = NULL;
1434 }
1435
1436 static void raid5_end_read_request(struct bio * bi, int error)
1437 {
1438         struct stripe_head *sh = bi->bi_private;
1439         raid5_conf_t *conf = sh->raid_conf;
1440         int disks = sh->disks, i;
1441         int uptodate = test_bit(BIO_UPTODATE, &bi->bi_flags);
1442         char b[BDEVNAME_SIZE];
1443         mdk_rdev_t *rdev;
1444
1445
1446         for (i=0 ; i<disks; i++)
1447                 if (bi == &sh->dev[i].req)
1448                         break;
1449
1450         pr_debug("end_read_request %llu/%d, count: %d, uptodate %d.\n",
1451                 (unsigned long long)sh->sector, i, atomic_read(&sh->count),
1452                 uptodate);
1453         if (i == disks) {
1454                 BUG();
1455                 return;
1456         }
1457
1458         if (uptodate) {
1459                 set_bit(R5_UPTODATE, &sh->dev[i].flags);
1460                 if (test_bit(R5_ReadError, &sh->dev[i].flags)) {
1461                         rdev = conf->disks[i].rdev;
1462                         printk_rl(KERN_INFO "raid5:%s: read error corrected"
1463                                   " (%lu sectors at %llu on %s)\n",
1464                                   mdname(conf->mddev), STRIPE_SECTORS,
1465                                   (unsigned long long)(sh->sector
1466                                                        + rdev->data_offset),
1467                                   bdevname(rdev->bdev, b));
1468                         clear_bit(R5_ReadError, &sh->dev[i].flags);
1469                         clear_bit(R5_ReWrite, &sh->dev[i].flags);
1470                 }
1471                 if (atomic_read(&conf->disks[i].rdev->read_errors))
1472                         atomic_set(&conf->disks[i].rdev->read_errors, 0);
1473         } else {
1474                 const char *bdn = bdevname(conf->disks[i].rdev->bdev, b);
1475                 int retry = 0;
1476                 rdev = conf->disks[i].rdev;
1477
1478                 clear_bit(R5_UPTODATE, &sh->dev[i].flags);
1479                 atomic_inc(&rdev->read_errors);
1480                 if (conf->mddev->degraded)
1481                         printk_rl(KERN_WARNING
1482                                   "raid5:%s: read error not correctable "
1483                                   "(sector %llu on %s).\n",
1484                                   mdname(conf->mddev),
1485                                   (unsigned long long)(sh->sector
1486                                                        + rdev->data_offset),
1487                                   bdn);
1488                 else if (test_bit(R5_ReWrite, &sh->dev[i].flags))
1489                         /* Oh, no!!! */
1490                         printk_rl(KERN_WARNING
1491                                   "raid5:%s: read error NOT corrected!! "
1492                                   "(sector %llu on %s).\n",
1493                                   mdname(conf->mddev),
1494                                   (unsigned long long)(sh->sector
1495                                                        + rdev->data_offset),
1496                                   bdn);
1497                 else if (atomic_read(&rdev->read_errors)
1498                          > conf->max_nr_stripes)
1499                         printk(KERN_WARNING
1500                                "raid5:%s: Too many read errors, failing device %s.\n",
1501                                mdname(conf->mddev), bdn);
1502                 else
1503                         retry = 1;
1504                 if (retry)
1505                         set_bit(R5_ReadError, &sh->dev[i].flags);
1506                 else {
1507                         clear_bit(R5_ReadError, &sh->dev[i].flags);
1508                         clear_bit(R5_ReWrite, &sh->dev[i].flags);
1509                         md_error(conf->mddev, rdev);
1510                 }
1511         }
1512         rdev_dec_pending(conf->disks[i].rdev, conf->mddev);
1513         clear_bit(R5_LOCKED, &sh->dev[i].flags);
1514         set_bit(STRIPE_HANDLE, &sh->state);
1515         release_stripe(sh);
1516 }
1517
1518 static void raid5_end_write_request(struct bio *bi, int error)
1519 {
1520         struct stripe_head *sh = bi->bi_private;
1521         raid5_conf_t *conf = sh->raid_conf;
1522         int disks = sh->disks, i;
1523         int uptodate = test_bit(BIO_UPTODATE, &bi->bi_flags);
1524
1525         for (i=0 ; i<disks; i++)
1526                 if (bi == &sh->dev[i].req)
1527                         break;
1528
1529         pr_debug("end_write_request %llu/%d, count %d, uptodate: %d.\n",
1530                 (unsigned long long)sh->sector, i, atomic_read(&sh->count),
1531                 uptodate);
1532         if (i == disks) {
1533                 BUG();
1534                 return;
1535         }
1536
1537         if (!uptodate)
1538                 md_error(conf->mddev, conf->disks[i].rdev);
1539
1540         rdev_dec_pending(conf->disks[i].rdev, conf->mddev);
1541         
1542         clear_bit(R5_LOCKED, &sh->dev[i].flags);
1543         set_bit(STRIPE_HANDLE, &sh->state);
1544         release_stripe(sh);
1545 }
1546
1547
1548 static sector_t compute_blocknr(struct stripe_head *sh, int i, int previous);
1549         
1550 static void raid5_build_block(struct stripe_head *sh, int i, int previous)
1551 {
1552         struct r5dev *dev = &sh->dev[i];
1553
1554         bio_init(&dev->req);
1555         dev->req.bi_io_vec = &dev->vec;
1556         dev->req.bi_vcnt++;
1557         dev->req.bi_max_vecs++;
1558         dev->vec.bv_page = dev->page;
1559         dev->vec.bv_len = STRIPE_SIZE;
1560         dev->vec.bv_offset = 0;
1561
1562         dev->req.bi_sector = sh->sector;
1563         dev->req.bi_private = sh;
1564
1565         dev->flags = 0;
1566         dev->sector = compute_blocknr(sh, i, previous);
1567 }
1568
1569 static void error(mddev_t *mddev, mdk_rdev_t *rdev)
1570 {
1571         char b[BDEVNAME_SIZE];
1572         raid5_conf_t *conf = (raid5_conf_t *) mddev->private;
1573         pr_debug("raid5: error called\n");
1574
1575         if (!test_bit(Faulty, &rdev->flags)) {
1576                 set_bit(MD_CHANGE_DEVS, &mddev->flags);
1577                 if (test_and_clear_bit(In_sync, &rdev->flags)) {
1578                         unsigned long flags;
1579                         spin_lock_irqsave(&conf->device_lock, flags);
1580                         mddev->degraded++;
1581                         spin_unlock_irqrestore(&conf->device_lock, flags);
1582                         /*
1583                          * if recovery was running, make sure it aborts.
1584                          */
1585                         set_bit(MD_RECOVERY_INTR, &mddev->recovery);
1586                 }
1587                 set_bit(Faulty, &rdev->flags);
1588                 printk(KERN_ALERT
1589                        "raid5: Disk failure on %s, disabling device.\n"
1590                        "raid5: Operation continuing on %d devices.\n",
1591                        bdevname(rdev->bdev,b), conf->raid_disks - mddev->degraded);
1592         }
1593 }
1594
1595 /*
1596  * Input: a 'big' sector number,
1597  * Output: index of the data and parity disk, and the sector # in them.
1598  */
1599 static sector_t raid5_compute_sector(raid5_conf_t *conf, sector_t r_sector,
1600                                      int previous, int *dd_idx,
1601                                      struct stripe_head *sh)
1602 {
1603         long stripe;
1604         unsigned long chunk_number;
1605         unsigned int chunk_offset;
1606         int pd_idx, qd_idx;
1607         int ddf_layout = 0;
1608         sector_t new_sector;
1609         int algorithm = previous ? conf->prev_algo
1610                                  : conf->algorithm;
1611         int sectors_per_chunk = previous ? (conf->prev_chunk >> 9)
1612                                          : (conf->chunk_size >> 9);
1613         int raid_disks = previous ? conf->previous_raid_disks
1614                                   : conf->raid_disks;
1615         int data_disks = raid_disks - conf->max_degraded;
1616
1617         /* First compute the information on this sector */
1618
1619         /*
1620          * Compute the chunk number and the sector offset inside the chunk
1621          */
1622         chunk_offset = sector_div(r_sector, sectors_per_chunk);
1623         chunk_number = r_sector;
1624         BUG_ON(r_sector != chunk_number);
1625
1626         /*
1627          * Compute the stripe number
1628          */
1629         stripe = chunk_number / data_disks;
1630
1631         /*
1632          * Compute the data disk and parity disk indexes inside the stripe
1633          */
1634         *dd_idx = chunk_number % data_disks;
1635
1636         /*
1637          * Select the parity disk based on the user selected algorithm.
1638          */
1639         pd_idx = qd_idx = ~0;
1640         switch(conf->level) {
1641         case 4:
1642                 pd_idx = data_disks;
1643                 break;
1644         case 5:
1645                 switch (algorithm) {
1646                 case ALGORITHM_LEFT_ASYMMETRIC:
1647                         pd_idx = data_disks - stripe % raid_disks;
1648                         if (*dd_idx >= pd_idx)
1649                                 (*dd_idx)++;
1650                         break;
1651                 case ALGORITHM_RIGHT_ASYMMETRIC:
1652                         pd_idx = stripe % raid_disks;
1653                         if (*dd_idx >= pd_idx)
1654                                 (*dd_idx)++;
1655                         break;
1656                 case ALGORITHM_LEFT_SYMMETRIC:
1657                         pd_idx = data_disks - stripe % raid_disks;
1658                         *dd_idx = (pd_idx + 1 + *dd_idx) % raid_disks;
1659                         break;
1660                 case ALGORITHM_RIGHT_SYMMETRIC:
1661                         pd_idx = stripe % raid_disks;
1662                         *dd_idx = (pd_idx + 1 + *dd_idx) % raid_disks;
1663                         break;
1664                 case ALGORITHM_PARITY_0:
1665                         pd_idx = 0;
1666                         (*dd_idx)++;
1667                         break;
1668                 case ALGORITHM_PARITY_N:
1669                         pd_idx = data_disks;
1670                         break;
1671                 default:
1672                         printk(KERN_ERR "raid5: unsupported algorithm %d\n",
1673                                 algorithm);
1674                         BUG();
1675                 }
1676                 break;
1677         case 6:
1678
1679                 switch (algorithm) {
1680                 case ALGORITHM_LEFT_ASYMMETRIC:
1681                         pd_idx = raid_disks - 1 - (stripe % raid_disks);
1682                         qd_idx = pd_idx + 1;
1683                         if (pd_idx == raid_disks-1) {
1684                                 (*dd_idx)++;    /* Q D D D P */
1685                                 qd_idx = 0;
1686                         } else if (*dd_idx >= pd_idx)
1687                                 (*dd_idx) += 2; /* D D P Q D */
1688                         break;
1689                 case ALGORITHM_RIGHT_ASYMMETRIC:
1690                         pd_idx = stripe % raid_disks;
1691                         qd_idx = pd_idx + 1;
1692                         if (pd_idx == raid_disks-1) {
1693                                 (*dd_idx)++;    /* Q D D D P */
1694                                 qd_idx = 0;
1695                         } else if (*dd_idx >= pd_idx)
1696                                 (*dd_idx) += 2; /* D D P Q D */
1697                         break;
1698                 case ALGORITHM_LEFT_SYMMETRIC:
1699                         pd_idx = raid_disks - 1 - (stripe % raid_disks);
1700                         qd_idx = (pd_idx + 1) % raid_disks;
1701                         *dd_idx = (pd_idx + 2 + *dd_idx) % raid_disks;
1702                         break;
1703                 case ALGORITHM_RIGHT_SYMMETRIC:
1704                         pd_idx = stripe % raid_disks;
1705                         qd_idx = (pd_idx + 1) % raid_disks;
1706                         *dd_idx = (pd_idx + 2 + *dd_idx) % raid_disks;
1707                         break;
1708
1709                 case ALGORITHM_PARITY_0:
1710                         pd_idx = 0;
1711                         qd_idx = 1;
1712                         (*dd_idx) += 2;
1713                         break;
1714                 case ALGORITHM_PARITY_N:
1715                         pd_idx = data_disks;
1716                         qd_idx = data_disks + 1;
1717                         break;
1718
1719                 case ALGORITHM_ROTATING_ZERO_RESTART:
1720                         /* Exactly the same as RIGHT_ASYMMETRIC, but or
1721                          * of blocks for computing Q is different.
1722                          */
1723                         pd_idx = stripe % raid_disks;
1724                         qd_idx = pd_idx + 1;
1725                         if (pd_idx == raid_disks-1) {
1726                                 (*dd_idx)++;    /* Q D D D P */
1727                                 qd_idx = 0;
1728                         } else if (*dd_idx >= pd_idx)
1729                                 (*dd_idx) += 2; /* D D P Q D */
1730                         ddf_layout = 1;
1731                         break;
1732
1733                 case ALGORITHM_ROTATING_N_RESTART:
1734                         /* Same a left_asymmetric, by first stripe is
1735                          * D D D P Q  rather than
1736                          * Q D D D P
1737                          */
1738                         pd_idx = raid_disks - 1 - ((stripe + 1) % raid_disks);
1739                         qd_idx = pd_idx + 1;
1740                         if (pd_idx == raid_disks-1) {
1741                                 (*dd_idx)++;    /* Q D D D P */
1742                                 qd_idx = 0;
1743                         } else if (*dd_idx >= pd_idx)
1744                                 (*dd_idx) += 2; /* D D P Q D */
1745                         ddf_layout = 1;
1746                         break;
1747
1748                 case ALGORITHM_ROTATING_N_CONTINUE:
1749                         /* Same as left_symmetric but Q is before P */
1750                         pd_idx = raid_disks - 1 - (stripe % raid_disks);
1751                         qd_idx = (pd_idx + raid_disks - 1) % raid_disks;
1752                         *dd_idx = (pd_idx + 1 + *dd_idx) % raid_disks;
1753                         ddf_layout = 1;
1754                         break;
1755
1756                 case ALGORITHM_LEFT_ASYMMETRIC_6:
1757                         /* RAID5 left_asymmetric, with Q on last device */
1758                         pd_idx = data_disks - stripe % (raid_disks-1);
1759                         if (*dd_idx >= pd_idx)
1760                                 (*dd_idx)++;
1761                         qd_idx = raid_disks - 1;
1762                         break;
1763
1764                 case ALGORITHM_RIGHT_ASYMMETRIC_6:
1765                         pd_idx = stripe % (raid_disks-1);
1766                         if (*dd_idx >= pd_idx)
1767                                 (*dd_idx)++;
1768                         qd_idx = raid_disks - 1;
1769                         break;
1770
1771                 case ALGORITHM_LEFT_SYMMETRIC_6:
1772                         pd_idx = data_disks - stripe % (raid_disks-1);
1773                         *dd_idx = (pd_idx + 1 + *dd_idx) % (raid_disks-1);
1774                         qd_idx = raid_disks - 1;
1775                         break;
1776
1777                 case ALGORITHM_RIGHT_SYMMETRIC_6:
1778                         pd_idx = stripe % (raid_disks-1);
1779                         *dd_idx = (pd_idx + 1 + *dd_idx) % (raid_disks-1);
1780                         qd_idx = raid_disks - 1;
1781                         break;
1782
1783                 case ALGORITHM_PARITY_0_6:
1784                         pd_idx = 0;
1785                         (*dd_idx)++;
1786                         qd_idx = raid_disks - 1;
1787                         break;
1788
1789
1790                 default:
1791                         printk(KERN_CRIT "raid6: unsupported algorithm %d\n",
1792                                algorithm);
1793                         BUG();
1794                 }
1795                 break;
1796         }
1797
1798         if (sh) {
1799                 sh->pd_idx = pd_idx;
1800                 sh->qd_idx = qd_idx;
1801                 sh->ddf_layout = ddf_layout;
1802         }
1803         /*
1804          * Finally, compute the new sector number
1805          */
1806         new_sector = (sector_t)stripe * sectors_per_chunk + chunk_offset;
1807         return new_sector;
1808 }
1809
1810
1811 static sector_t compute_blocknr(struct stripe_head *sh, int i, int previous)
1812 {
1813         raid5_conf_t *conf = sh->raid_conf;
1814         int raid_disks = sh->disks;
1815         int data_disks = raid_disks - conf->max_degraded;
1816         sector_t new_sector = sh->sector, check;
1817         int sectors_per_chunk = previous ? (conf->prev_chunk >> 9)
1818                                          : (conf->chunk_size >> 9);
1819         int algorithm = previous ? conf->prev_algo
1820                                  : conf->algorithm;
1821         sector_t stripe;
1822         int chunk_offset;
1823         int chunk_number, dummy1, dd_idx = i;
1824         sector_t r_sector;
1825         struct stripe_head sh2;
1826
1827
1828         chunk_offset = sector_div(new_sector, sectors_per_chunk);
1829         stripe = new_sector;
1830         BUG_ON(new_sector != stripe);
1831
1832         if (i == sh->pd_idx)
1833                 return 0;
1834         switch(conf->level) {
1835         case 4: break;
1836         case 5:
1837                 switch (algorithm) {
1838                 case ALGORITHM_LEFT_ASYMMETRIC:
1839                 case ALGORITHM_RIGHT_ASYMMETRIC:
1840                         if (i > sh->pd_idx)
1841                                 i--;
1842                         break;
1843                 case ALGORITHM_LEFT_SYMMETRIC:
1844                 case ALGORITHM_RIGHT_SYMMETRIC:
1845                         if (i < sh->pd_idx)
1846                                 i += raid_disks;
1847                         i -= (sh->pd_idx + 1);
1848                         break;
1849                 case ALGORITHM_PARITY_0:
1850                         i -= 1;
1851                         break;
1852                 case ALGORITHM_PARITY_N:
1853                         break;
1854                 default:
1855                         printk(KERN_ERR "raid5: unsupported algorithm %d\n",
1856                                algorithm);
1857                         BUG();
1858                 }
1859                 break;
1860         case 6:
1861                 if (i == sh->qd_idx)
1862                         return 0; /* It is the Q disk */
1863                 switch (algorithm) {
1864                 case ALGORITHM_LEFT_ASYMMETRIC:
1865                 case ALGORITHM_RIGHT_ASYMMETRIC:
1866                 case ALGORITHM_ROTATING_ZERO_RESTART:
1867                 case ALGORITHM_ROTATING_N_RESTART:
1868                         if (sh->pd_idx == raid_disks-1)
1869                                 i--;    /* Q D D D P */
1870                         else if (i > sh->pd_idx)
1871                                 i -= 2; /* D D P Q D */
1872                         break;
1873                 case ALGORITHM_LEFT_SYMMETRIC:
1874                 case ALGORITHM_RIGHT_SYMMETRIC:
1875                         if (sh->pd_idx == raid_disks-1)
1876                                 i--; /* Q D D D P */
1877                         else {
1878                                 /* D D P Q D */
1879                                 if (i < sh->pd_idx)
1880                                         i += raid_disks;
1881                                 i -= (sh->pd_idx + 2);
1882                         }
1883                         break;
1884                 case ALGORITHM_PARITY_0:
1885                         i -= 2;
1886                         break;
1887                 case ALGORITHM_PARITY_N:
1888                         break;
1889                 case ALGORITHM_ROTATING_N_CONTINUE:
1890                         if (sh->pd_idx == 0)
1891                                 i--;    /* P D D D Q */
1892                         else if (i > sh->pd_idx)
1893                                 i -= 2; /* D D Q P D */
1894                         break;
1895                 case ALGORITHM_LEFT_ASYMMETRIC_6:
1896                 case ALGORITHM_RIGHT_ASYMMETRIC_6:
1897                         if (i > sh->pd_idx)
1898                                 i--;
1899                         break;
1900                 case ALGORITHM_LEFT_SYMMETRIC_6:
1901                 case ALGORITHM_RIGHT_SYMMETRIC_6:
1902                         if (i < sh->pd_idx)
1903                                 i += data_disks + 1;
1904                         i -= (sh->pd_idx + 1);
1905                         break;
1906                 case ALGORITHM_PARITY_0_6:
1907                         i -= 1;
1908                         break;
1909                 default:
1910                         printk(KERN_CRIT "raid6: unsupported algorithm %d\n",
1911                                algorithm);
1912                         BUG();
1913                 }
1914                 break;
1915         }
1916
1917         chunk_number = stripe * data_disks + i;
1918         r_sector = (sector_t)chunk_number * sectors_per_chunk + chunk_offset;
1919
1920         check = raid5_compute_sector(conf, r_sector,
1921                                      previous, &dummy1, &sh2);
1922         if (check != sh->sector || dummy1 != dd_idx || sh2.pd_idx != sh->pd_idx
1923                 || sh2.qd_idx != sh->qd_idx) {
1924                 printk(KERN_ERR "compute_blocknr: map not correct\n");
1925                 return 0;
1926         }
1927         return r_sector;
1928 }
1929
1930
1931 static void
1932 schedule_reconstruction(struct stripe_head *sh, struct stripe_head_state *s,
1933                          int rcw, int expand)
1934 {
1935         int i, pd_idx = sh->pd_idx, disks = sh->disks;
1936         raid5_conf_t *conf = sh->raid_conf;
1937         int level = conf->level;
1938
1939         if (rcw) {
1940                 /* if we are not expanding this is a proper write request, and
1941                  * there will be bios with new data to be drained into the
1942                  * stripe cache
1943                  */
1944                 if (!expand) {
1945                         sh->reconstruct_state = reconstruct_state_drain_run;
1946                         set_bit(STRIPE_OP_BIODRAIN, &s->ops_request);
1947                 } else
1948                         sh->reconstruct_state = reconstruct_state_run;
1949
1950                 set_bit(STRIPE_OP_RECONSTRUCT, &s->ops_request);
1951
1952                 for (i = disks; i--; ) {
1953                         struct r5dev *dev = &sh->dev[i];
1954
1955                         if (dev->towrite) {
1956                                 set_bit(R5_LOCKED, &dev->flags);
1957                                 set_bit(R5_Wantdrain, &dev->flags);
1958                                 if (!expand)
1959                                         clear_bit(R5_UPTODATE, &dev->flags);
1960                                 s->locked++;
1961                         }
1962                 }
1963                 if (s->locked + conf->max_degraded == disks)
1964                         if (!test_and_set_bit(STRIPE_FULL_WRITE, &sh->state))
1965                                 atomic_inc(&conf->pending_full_writes);
1966         } else {
1967                 BUG_ON(level == 6);
1968                 BUG_ON(!(test_bit(R5_UPTODATE, &sh->dev[pd_idx].flags) ||
1969                         test_bit(R5_Wantcompute, &sh->dev[pd_idx].flags)));
1970
1971                 sh->reconstruct_state = reconstruct_state_prexor_drain_run;
1972                 set_bit(STRIPE_OP_PREXOR, &s->ops_request);
1973                 set_bit(STRIPE_OP_BIODRAIN, &s->ops_request);
1974                 set_bit(STRIPE_OP_RECONSTRUCT, &s->ops_request);
1975
1976                 for (i = disks; i--; ) {
1977                         struct r5dev *dev = &sh->dev[i];
1978                         if (i == pd_idx)
1979                                 continue;
1980
1981                         if (dev->towrite &&
1982                             (test_bit(R5_UPTODATE, &dev->flags) ||
1983                              test_bit(R5_Wantcompute, &dev->flags))) {
1984                                 set_bit(R5_Wantdrain, &dev->flags);
1985                                 set_bit(R5_LOCKED, &dev->flags);
1986                                 clear_bit(R5_UPTODATE, &dev->flags);
1987                                 s->locked++;
1988                         }
1989                 }
1990         }
1991
1992         /* keep the parity disk(s) locked while asynchronous operations
1993          * are in flight
1994          */
1995         set_bit(R5_LOCKED, &sh->dev[pd_idx].flags);
1996         clear_bit(R5_UPTODATE, &sh->dev[pd_idx].flags);
1997         s->locked++;
1998
1999         if (level == 6) {
2000                 int qd_idx = sh->qd_idx;
2001                 struct r5dev *dev = &sh->dev[qd_idx];
2002
2003                 set_bit(R5_LOCKED, &dev->flags);
2004                 clear_bit(R5_UPTODATE, &dev->flags);
2005                 s->locked++;
2006         }
2007
2008         pr_debug("%s: stripe %llu locked: %d ops_request: %lx\n",
2009                 __func__, (unsigned long long)sh->sector,
2010                 s->locked, s->ops_request);
2011 }
2012
2013 /*
2014  * Each stripe/dev can have one or more bion attached.
2015  * toread/towrite point to the first in a chain.
2016  * The bi_next chain must be in order.
2017  */
2018 static int add_stripe_bio(struct stripe_head *sh, struct bio *bi, int dd_idx, int forwrite)
2019 {
2020         struct bio **bip;
2021         raid5_conf_t *conf = sh->raid_conf;
2022         int firstwrite=0;
2023
2024         pr_debug("adding bh b#%llu to stripe s#%llu\n",
2025                 (unsigned long long)bi->bi_sector,
2026                 (unsigned long long)sh->sector);
2027
2028
2029         spin_lock(&sh->lock);
2030         spin_lock_irq(&conf->device_lock);
2031         if (forwrite) {
2032                 bip = &sh->dev[dd_idx].towrite;
2033                 if (*bip == NULL && sh->dev[dd_idx].written == NULL)
2034                         firstwrite = 1;
2035         } else
2036                 bip = &sh->dev[dd_idx].toread;
2037         while (*bip && (*bip)->bi_sector < bi->bi_sector) {
2038                 if ((*bip)->bi_sector + ((*bip)->bi_size >> 9) > bi->bi_sector)
2039                         goto overlap;
2040                 bip = & (*bip)->bi_next;
2041         }
2042         if (*bip && (*bip)->bi_sector < bi->bi_sector + ((bi->bi_size)>>9))
2043                 goto overlap;
2044
2045         BUG_ON(*bip && bi->bi_next && (*bip) != bi->bi_next);
2046         if (*bip)
2047                 bi->bi_next = *bip;
2048         *bip = bi;
2049         bi->bi_phys_segments++;
2050         spin_unlock_irq(&conf->device_lock);
2051         spin_unlock(&sh->lock);
2052
2053         pr_debug("added bi b#%llu to stripe s#%llu, disk %d.\n",
2054                 (unsigned long long)bi->bi_sector,
2055                 (unsigned long long)sh->sector, dd_idx);
2056
2057         if (conf->mddev->bitmap && firstwrite) {
2058                 bitmap_startwrite(conf->mddev->bitmap, sh->sector,
2059                                   STRIPE_SECTORS, 0);
2060                 sh->bm_seq = conf->seq_flush+1;
2061                 set_bit(STRIPE_BIT_DELAY, &sh->state);
2062         }
2063
2064         if (forwrite) {
2065                 /* check if page is covered */
2066                 sector_t sector = sh->dev[dd_idx].sector;
2067                 for (bi=sh->dev[dd_idx].towrite;
2068                      sector < sh->dev[dd_idx].sector + STRIPE_SECTORS &&
2069                              bi && bi->bi_sector <= sector;
2070                      bi = r5_next_bio(bi, sh->dev[dd_idx].sector)) {
2071                         if (bi->bi_sector + (bi->bi_size>>9) >= sector)
2072                                 sector = bi->bi_sector + (bi->bi_size>>9);
2073                 }
2074                 if (sector >= sh->dev[dd_idx].sector + STRIPE_SECTORS)
2075                         set_bit(R5_OVERWRITE, &sh->dev[dd_idx].flags);
2076         }
2077         return 1;
2078
2079  overlap:
2080         set_bit(R5_Overlap, &sh->dev[dd_idx].flags);
2081         spin_unlock_irq(&conf->device_lock);
2082         spin_unlock(&sh->lock);
2083         return 0;
2084 }
2085
2086 static void end_reshape(raid5_conf_t *conf);
2087
2088 static void stripe_set_idx(sector_t stripe, raid5_conf_t *conf, int previous,
2089                             struct stripe_head *sh)
2090 {
2091         int sectors_per_chunk =
2092                 previous ? (conf->prev_chunk >> 9)
2093                          : (conf->chunk_size >> 9);
2094         int dd_idx;
2095         int chunk_offset = sector_div(stripe, sectors_per_chunk);
2096         int disks = previous ? conf->previous_raid_disks : conf->raid_disks;
2097
2098         raid5_compute_sector(conf,
2099                              stripe * (disks - conf->max_degraded)
2100                              *sectors_per_chunk + chunk_offset,
2101                              previous,
2102                              &dd_idx, sh);
2103 }
2104
2105 static void
2106 handle_failed_stripe(raid5_conf_t *conf, struct stripe_head *sh,
2107                                 struct stripe_head_state *s, int disks,
2108                                 struct bio **return_bi)
2109 {
2110         int i;
2111         for (i = disks; i--; ) {
2112                 struct bio *bi;
2113                 int bitmap_end = 0;
2114
2115                 if (test_bit(R5_ReadError, &sh->dev[i].flags)) {
2116                         mdk_rdev_t *rdev;
2117                         rcu_read_lock();
2118                         rdev = rcu_dereference(conf->disks[i].rdev);
2119                         if (rdev && test_bit(In_sync, &rdev->flags))
2120                                 /* multiple read failures in one stripe */
2121                                 md_error(conf->mddev, rdev);
2122                         rcu_read_unlock();
2123                 }
2124                 spin_lock_irq(&conf->device_lock);
2125                 /* fail all writes first */
2126                 bi = sh->dev[i].towrite;
2127                 sh->dev[i].towrite = NULL;
2128                 if (bi) {
2129                         s->to_write--;
2130                         bitmap_end = 1;
2131                 }
2132
2133                 if (test_and_clear_bit(R5_Overlap, &sh->dev[i].flags))
2134                         wake_up(&conf->wait_for_overlap);
2135
2136                 while (bi && bi->bi_sector <
2137                         sh->dev[i].sector + STRIPE_SECTORS) {
2138                         struct bio *nextbi = r5_next_bio(bi, sh->dev[i].sector);
2139                         clear_bit(BIO_UPTODATE, &bi->bi_flags);
2140                         if (!raid5_dec_bi_phys_segments(bi)) {
2141                                 md_write_end(conf->mddev);
2142                                 bi->bi_next = *return_bi;
2143                                 *return_bi = bi;
2144                         }
2145                         bi = nextbi;
2146                 }
2147                 /* and fail all 'written' */
2148                 bi = sh->dev[i].written;
2149                 sh->dev[i].written = NULL;
2150                 if (bi) bitmap_end = 1;
2151                 while (bi && bi->bi_sector <
2152                        sh->dev[i].sector + STRIPE_SECTORS) {
2153                         struct bio *bi2 = r5_next_bio(bi, sh->dev[i].sector);
2154                         clear_bit(BIO_UPTODATE, &bi->bi_flags);
2155                         if (!raid5_dec_bi_phys_segments(bi)) {
2156                                 md_write_end(conf->mddev);
2157                                 bi->bi_next = *return_bi;
2158                                 *return_bi = bi;
2159                         }
2160                         bi = bi2;
2161                 }
2162
2163                 /* fail any reads if this device is non-operational and
2164                  * the data has not reached the cache yet.
2165                  */
2166                 if (!test_bit(R5_Wantfill, &sh->dev[i].flags) &&
2167                     (!test_bit(R5_Insync, &sh->dev[i].flags) ||
2168                       test_bit(R5_ReadError, &sh->dev[i].flags))) {
2169                         bi = sh->dev[i].toread;
2170                         sh->dev[i].toread = NULL;
2171                         if (test_and_clear_bit(R5_Overlap, &sh->dev[i].flags))
2172                                 wake_up(&conf->wait_for_overlap);
2173                         if (bi) s->to_read--;
2174                         while (bi && bi->bi_sector <
2175                                sh->dev[i].sector + STRIPE_SECTORS) {
2176                                 struct bio *nextbi =
2177                                         r5_next_bio(bi, sh->dev[i].sector);
2178                                 clear_bit(BIO_UPTODATE, &bi->bi_flags);
2179                                 if (!raid5_dec_bi_phys_segments(bi)) {
2180                                         bi->bi_next = *return_bi;
2181                                         *return_bi = bi;
2182                                 }
2183                                 bi = nextbi;
2184                         }
2185                 }
2186                 spin_unlock_irq(&conf->device_lock);
2187                 if (bitmap_end)
2188                         bitmap_endwrite(conf->mddev->bitmap, sh->sector,
2189                                         STRIPE_SECTORS, 0, 0);
2190         }
2191
2192         if (test_and_clear_bit(STRIPE_FULL_WRITE, &sh->state))
2193                 if (atomic_dec_and_test(&conf->pending_full_writes))
2194                         md_wakeup_thread(conf->mddev->thread);
2195 }
2196
2197 /* fetch_block5 - checks the given member device to see if its data needs
2198  * to be read or computed to satisfy a request.
2199  *
2200  * Returns 1 when no more member devices need to be checked, otherwise returns
2201  * 0 to tell the loop in handle_stripe_fill5 to continue
2202  */
2203 static int fetch_block5(struct stripe_head *sh, struct stripe_head_state *s,
2204                         int disk_idx, int disks)
2205 {
2206         struct r5dev *dev = &sh->dev[disk_idx];
2207         struct r5dev *failed_dev = &sh->dev[s->failed_num];
2208
2209         /* is the data in this block needed, and can we get it? */
2210         if (!test_bit(R5_LOCKED, &dev->flags) &&
2211             !test_bit(R5_UPTODATE, &dev->flags) &&
2212             (dev->toread ||
2213              (dev->towrite && !test_bit(R5_OVERWRITE, &dev->flags)) ||
2214              s->syncing || s->expanding ||
2215              (s->failed &&
2216               (failed_dev->toread ||
2217                (failed_dev->towrite &&
2218                 !test_bit(R5_OVERWRITE, &failed_dev->flags)))))) {
2219                 /* We would like to get this block, possibly by computing it,
2220                  * otherwise read it if the backing disk is insync
2221                  */
2222                 if ((s->uptodate == disks - 1) &&
2223                     (s->failed && disk_idx == s->failed_num)) {
2224                         set_bit(STRIPE_COMPUTE_RUN, &sh->state);
2225                         set_bit(STRIPE_OP_COMPUTE_BLK, &s->ops_request);
2226                         set_bit(R5_Wantcompute, &dev->flags);
2227                         sh->ops.target = disk_idx;
2228                         sh->ops.target2 = -1;
2229                         s->req_compute = 1;
2230                         /* Careful: from this point on 'uptodate' is in the eye
2231                          * of raid_run_ops which services 'compute' operations
2232                          * before writes. R5_Wantcompute flags a block that will
2233                          * be R5_UPTODATE by the time it is needed for a
2234                          * subsequent operation.
2235                          */
2236                         s->uptodate++;
2237                         return 1; /* uptodate + compute == disks */
2238                 } else if (test_bit(R5_Insync, &dev->flags)) {
2239                         set_bit(R5_LOCKED, &dev->flags);
2240                         set_bit(R5_Wantread, &dev->flags);
2241                         s->locked++;
2242                         pr_debug("Reading block %d (sync=%d)\n", disk_idx,
2243                                 s->syncing);
2244                 }
2245         }
2246
2247         return 0;
2248 }
2249
2250 /**
2251  * handle_stripe_fill5 - read or compute data to satisfy pending requests.
2252  */
2253 static void handle_stripe_fill5(struct stripe_head *sh,
2254                         struct stripe_head_state *s, int disks)
2255 {
2256         int i;
2257
2258         /* look for blocks to read/compute, skip this if a compute
2259          * is already in flight, or if the stripe contents are in the
2260          * midst of changing due to a write
2261          */
2262         if (!test_bit(STRIPE_COMPUTE_RUN, &sh->state) && !sh->check_state &&
2263             !sh->reconstruct_state)
2264                 for (i = disks; i--; )
2265                         if (fetch_block5(sh, s, i, disks))
2266                                 break;
2267         set_bit(STRIPE_HANDLE, &sh->state);
2268 }
2269
2270 /* fetch_block6 - checks the given member device to see if its data needs
2271  * to be read or computed to satisfy a request.
2272  *
2273  * Returns 1 when no more member devices need to be checked, otherwise returns
2274  * 0 to tell the loop in handle_stripe_fill6 to continue
2275  */
2276 static int fetch_block6(struct stripe_head *sh, struct stripe_head_state *s,
2277                          struct r6_state *r6s, int disk_idx, int disks)
2278 {
2279         struct r5dev *dev = &sh->dev[disk_idx];
2280         struct r5dev *fdev[2] = { &sh->dev[r6s->failed_num[0]],
2281                                   &sh->dev[r6s->failed_num[1]] };
2282
2283         if (!test_bit(R5_LOCKED, &dev->flags) &&
2284             !test_bit(R5_UPTODATE, &dev->flags) &&
2285             (dev->toread ||
2286              (dev->towrite && !test_bit(R5_OVERWRITE, &dev->flags)) ||
2287              s->syncing || s->expanding ||
2288              (s->failed >= 1 &&
2289               (fdev[0]->toread || s->to_write)) ||
2290              (s->failed >= 2 &&
2291               (fdev[1]->toread || s->to_write)))) {
2292                 /* we would like to get this block, possibly by computing it,
2293                  * otherwise read it if the backing disk is insync
2294                  */
2295                 BUG_ON(test_bit(R5_Wantcompute, &dev->flags));
2296                 BUG_ON(test_bit(R5_Wantread, &dev->flags));
2297                 if ((s->uptodate == disks - 1) &&
2298                     (s->failed && (disk_idx == r6s->failed_num[0] ||
2299                                    disk_idx == r6s->failed_num[1]))) {
2300                         /* have disk failed, and we're requested to fetch it;
2301                          * do compute it
2302                          */
2303                         pr_debug("Computing stripe %llu block %d\n",
2304                                (unsigned long long)sh->sector, disk_idx);
2305                         set_bit(STRIPE_COMPUTE_RUN, &sh->state);
2306                         set_bit(STRIPE_OP_COMPUTE_BLK, &s->ops_request);
2307                         set_bit(R5_Wantcompute, &dev->flags);
2308                         sh->ops.target = disk_idx;
2309                         sh->ops.target2 = -1; /* no 2nd target */
2310                         s->req_compute = 1;
2311                         s->uptodate++;
2312                         return 1;
2313                 } else if (s->uptodate == disks-2 && s->failed >= 2) {
2314                         /* Computing 2-failure is *very* expensive; only
2315                          * do it if failed >= 2
2316                          */
2317                         int other;
2318                         for (other = disks; other--; ) {
2319                                 if (other == disk_idx)
2320                                         continue;
2321                                 if (!test_bit(R5_UPTODATE,
2322                                       &sh->dev[other].flags))
2323                                         break;
2324                         }
2325                         BUG_ON(other < 0);
2326                         pr_debug("Computing stripe %llu blocks %d,%d\n",
2327                                (unsigned long long)sh->sector,
2328                                disk_idx, other);
2329                         set_bit(STRIPE_COMPUTE_RUN, &sh->state);
2330                         set_bit(STRIPE_OP_COMPUTE_BLK, &s->ops_request);
2331                         set_bit(R5_Wantcompute, &sh->dev[disk_idx].flags);
2332                         set_bit(R5_Wantcompute, &sh->dev[other].flags);
2333                         sh->ops.target = disk_idx;
2334                         sh->ops.target2 = other;
2335                         s->uptodate += 2;
2336                         s->req_compute = 1;
2337                         return 1;
2338                 } else if (test_bit(R5_Insync, &dev->flags)) {
2339                         set_bit(R5_LOCKED, &dev->flags);
2340                         set_bit(R5_Wantread, &dev->flags);
2341                         s->locked++;
2342                         pr_debug("Reading block %d (sync=%d)\n",
2343                                 disk_idx, s->syncing);
2344                 }
2345         }
2346
2347         return 0;
2348 }
2349
2350 /**
2351  * handle_stripe_fill6 - read or compute data to satisfy pending requests.
2352  */
2353 static void handle_stripe_fill6(struct stripe_head *sh,
2354                         struct stripe_head_state *s, struct r6_state *r6s,
2355                         int disks)
2356 {
2357         int i;
2358
2359         /* look for blocks to read/compute, skip this if a compute
2360          * is already in flight, or if the stripe contents are in the
2361          * midst of changing due to a write
2362          */
2363         if (!test_bit(STRIPE_COMPUTE_RUN, &sh->state) && !sh->check_state &&
2364             !sh->reconstruct_state)
2365                 for (i = disks; i--; )
2366                         if (fetch_block6(sh, s, r6s, i, disks))
2367                                 break;
2368         set_bit(STRIPE_HANDLE, &sh->state);
2369 }
2370
2371
2372 /* handle_stripe_clean_event
2373  * any written block on an uptodate or failed drive can be returned.
2374  * Note that if we 'wrote' to a failed drive, it will be UPTODATE, but
2375  * never LOCKED, so we don't need to test 'failed' directly.
2376  */
2377 static void handle_stripe_clean_event(raid5_conf_t *conf,
2378         struct stripe_head *sh, int disks, struct bio **return_bi)
2379 {
2380         int i;
2381         struct r5dev *dev;
2382
2383         for (i = disks; i--; )
2384                 if (sh->dev[i].written) {
2385                         dev = &sh->dev[i];
2386                         if (!test_bit(R5_LOCKED, &dev->flags) &&
2387                                 test_bit(R5_UPTODATE, &dev->flags)) {
2388                                 /* We can return any write requests */
2389                                 struct bio *wbi, *wbi2;
2390                                 int bitmap_end = 0;
2391                                 pr_debug("Return write for disc %d\n", i);
2392                                 spin_lock_irq(&conf->device_lock);
2393                                 wbi = dev->written;
2394                                 dev->written = NULL;
2395                                 while (wbi && wbi->bi_sector <
2396                                         dev->sector + STRIPE_SECTORS) {
2397                                         wbi2 = r5_next_bio(wbi, dev->sector);
2398                                         if (!raid5_dec_bi_phys_segments(wbi)) {
2399                                                 md_write_end(conf->mddev);
2400                                                 wbi->bi_next = *return_bi;
2401                                                 *return_bi = wbi;
2402                                         }
2403                                         wbi = wbi2;
2404                                 }
2405                                 if (dev->towrite == NULL)
2406                                         bitmap_end = 1;
2407                                 spin_unlock_irq(&conf->device_lock);
2408                                 if (bitmap_end)
2409                                         bitmap_endwrite(conf->mddev->bitmap,
2410                                                         sh->sector,
2411                                                         STRIPE_SECTORS,
2412                                          !test_bit(STRIPE_DEGRADED, &sh->state),
2413                                                         0);
2414                         }
2415                 }
2416
2417         if (test_and_clear_bit(STRIPE_FULL_WRITE, &sh->state))
2418                 if (atomic_dec_and_test(&conf->pending_full_writes))
2419                         md_wakeup_thread(conf->mddev->thread);
2420 }
2421
2422 static void handle_stripe_dirtying5(raid5_conf_t *conf,
2423                 struct stripe_head *sh, struct stripe_head_state *s, int disks)
2424 {
2425         int rmw = 0, rcw = 0, i;
2426         for (i = disks; i--; ) {
2427                 /* would I have to read this buffer for read_modify_write */
2428                 struct r5dev *dev = &sh->dev[i];
2429                 if ((dev->towrite || i == sh->pd_idx) &&
2430                     !test_bit(R5_LOCKED, &dev->flags) &&
2431                     !(test_bit(R5_UPTODATE, &dev->flags) ||
2432                       test_bit(R5_Wantcompute, &dev->flags))) {
2433                         if (test_bit(R5_Insync, &dev->flags))
2434                                 rmw++;
2435                         else
2436                                 rmw += 2*disks;  /* cannot read it */
2437                 }
2438                 /* Would I have to read this buffer for reconstruct_write */
2439                 if (!test_bit(R5_OVERWRITE, &dev->flags) && i != sh->pd_idx &&
2440                     !test_bit(R5_LOCKED, &dev->flags) &&
2441                     !(test_bit(R5_UPTODATE, &dev->flags) ||
2442                     test_bit(R5_Wantcompute, &dev->flags))) {
2443                         if (test_bit(R5_Insync, &dev->flags)) rcw++;
2444                         else
2445                                 rcw += 2*disks;
2446                 }
2447         }
2448         pr_debug("for sector %llu, rmw=%d rcw=%d\n",
2449                 (unsigned long long)sh->sector, rmw, rcw);
2450         set_bit(STRIPE_HANDLE, &sh->state);
2451         if (rmw < rcw && rmw > 0)
2452                 /* prefer read-modify-write, but need to get some data */
2453                 for (i = disks; i--; ) {
2454                         struct r5dev *dev = &sh->dev[i];
2455                         if ((dev->towrite || i == sh->pd_idx) &&
2456                             !test_bit(R5_LOCKED, &dev->flags) &&
2457                             !(test_bit(R5_UPTODATE, &dev->flags) ||
2458                             test_bit(R5_Wantcompute, &dev->flags)) &&
2459                             test_bit(R5_Insync, &dev->flags)) {
2460                                 if (
2461                                   test_bit(STRIPE_PREREAD_ACTIVE, &sh->state)) {
2462                                         pr_debug("Read_old block "
2463                                                 "%d for r-m-w\n", i);
2464                                         set_bit(R5_LOCKED, &dev->flags);
2465                                         set_bit(R5_Wantread, &dev->flags);
2466                                         s->locked++;
2467                                 } else {
2468                                         set_bit(STRIPE_DELAYED, &sh->state);
2469                                         set_bit(STRIPE_HANDLE, &sh->state);
2470                                 }
2471                         }
2472                 }
2473         if (rcw <= rmw && rcw > 0)
2474                 /* want reconstruct write, but need to get some data */
2475                 for (i = disks; i--; ) {
2476                         struct r5dev *dev = &sh->dev[i];
2477                         if (!test_bit(R5_OVERWRITE, &dev->flags) &&
2478                             i != sh->pd_idx &&
2479                             !test_bit(R5_LOCKED, &dev->flags) &&
2480                             !(test_bit(R5_UPTODATE, &dev->flags) ||
2481                             test_bit(R5_Wantcompute, &dev->flags)) &&
2482                             test_bit(R5_Insync, &dev->flags)) {
2483                                 if (
2484                                   test_bit(STRIPE_PREREAD_ACTIVE, &sh->state)) {
2485                                         pr_debug("Read_old block "
2486                                                 "%d for Reconstruct\n", i);
2487                                         set_bit(R5_LOCKED, &dev->flags);
2488                                         set_bit(R5_Wantread, &dev->flags);
2489                                         s->locked++;
2490                                 } else {
2491                                         set_bit(STRIPE_DELAYED, &sh->state);
2492                                         set_bit(STRIPE_HANDLE, &sh->state);
2493                                 }
2494                         }
2495                 }
2496         /* now if nothing is locked, and if we have enough data,
2497          * we can start a write request
2498          */
2499         /* since handle_stripe can be called at any time we need to handle the
2500          * case where a compute block operation has been submitted and then a
2501          * subsequent call wants to start a write request.  raid_run_ops only
2502          * handles the case where compute block and reconstruct are requested
2503          * simultaneously.  If this is not the case then new writes need to be
2504          * held off until the compute completes.
2505          */
2506         if ((s->req_compute || !test_bit(STRIPE_COMPUTE_RUN, &sh->state)) &&
2507             (s->locked == 0 && (rcw == 0 || rmw == 0) &&
2508             !test_bit(STRIPE_BIT_DELAY, &sh->state)))
2509                 schedule_reconstruction(sh, s, rcw == 0, 0);
2510 }
2511
2512 static void handle_stripe_dirtying6(raid5_conf_t *conf,
2513                 struct stripe_head *sh, struct stripe_head_state *s,
2514                 struct r6_state *r6s, int disks)
2515 {
2516         int rcw = 0, pd_idx = sh->pd_idx, i;
2517         int qd_idx = sh->qd_idx;
2518
2519         set_bit(STRIPE_HANDLE, &sh->state);
2520         for (i = disks; i--; ) {
2521                 struct r5dev *dev = &sh->dev[i];
2522                 /* check if we haven't enough data */
2523                 if (!test_bit(R5_OVERWRITE, &dev->flags) &&
2524                     i != pd_idx && i != qd_idx &&
2525                     !test_bit(R5_LOCKED, &dev->flags) &&
2526                     !(test_bit(R5_UPTODATE, &dev->flags) ||
2527                       test_bit(R5_Wantcompute, &dev->flags))) {
2528                         rcw++;
2529                         if (!test_bit(R5_Insync, &dev->flags))
2530                                 continue; /* it's a failed drive */
2531
2532                         if (
2533                           test_bit(STRIPE_PREREAD_ACTIVE, &sh->state)) {
2534                                 pr_debug("Read_old stripe %llu "
2535                                         "block %d for Reconstruct\n",
2536                                      (unsigned long long)sh->sector, i);
2537                                 set_bit(R5_LOCKED, &dev->flags);
2538                                 set_bit(R5_Wantread, &dev->flags);
2539                                 s->locked++;
2540                         } else {
2541                                 pr_debug("Request delayed stripe %llu "
2542                                         "block %d for Reconstruct\n",
2543                                      (unsigned long long)sh->sector, i);
2544                                 set_bit(STRIPE_DELAYED, &sh->state);
2545                                 set_bit(STRIPE_HANDLE, &sh->state);
2546                         }
2547                 }
2548         }
2549         /* now if nothing is locked, and if we have enough data, we can start a
2550          * write request
2551          */
2552         if ((s->req_compute || !test_bit(STRIPE_COMPUTE_RUN, &sh->state)) &&
2553             s->locked == 0 && rcw == 0 &&
2554             !test_bit(STRIPE_BIT_DELAY, &sh->state)) {
2555                 schedule_reconstruction(sh, s, 1, 0);
2556         }
2557 }
2558
2559 static void handle_parity_checks5(raid5_conf_t *conf, struct stripe_head *sh,
2560                                 struct stripe_head_state *s, int disks)
2561 {
2562         struct r5dev *dev = NULL;
2563
2564         set_bit(STRIPE_HANDLE, &sh->state);
2565
2566         switch (sh->check_state) {
2567         case check_state_idle:
2568                 /* start a new check operation if there are no failures */
2569                 if (s->failed == 0) {
2570                         BUG_ON(s->uptodate != disks);
2571                         sh->check_state = check_state_run;
2572                         set_bit(STRIPE_OP_CHECK, &s->ops_request);
2573                         clear_bit(R5_UPTODATE, &sh->dev[sh->pd_idx].flags);
2574                         s->uptodate--;
2575                         break;
2576                 }
2577                 dev = &sh->dev[s->failed_num];
2578                 /* fall through */
2579         case check_state_compute_result:
2580                 sh->check_state = check_state_idle;
2581                 if (!dev)
2582                         dev = &sh->dev[sh->pd_idx];
2583
2584                 /* check that a write has not made the stripe insync */
2585                 if (test_bit(STRIPE_INSYNC, &sh->state))
2586                         break;
2587
2588                 /* either failed parity check, or recovery is happening */
2589                 BUG_ON(!test_bit(R5_UPTODATE, &dev->flags));
2590                 BUG_ON(s->uptodate != disks);
2591
2592                 set_bit(R5_LOCKED, &dev->flags);
2593                 s->locked++;
2594                 set_bit(R5_Wantwrite, &dev->flags);
2595
2596                 clear_bit(STRIPE_DEGRADED, &sh->state);
2597                 set_bit(STRIPE_INSYNC, &sh->state);
2598                 break;
2599         case check_state_run:
2600                 break; /* we will be called again upon completion */
2601         case check_state_check_result:
2602                 sh->check_state = check_state_idle;
2603
2604                 /* if a failure occurred during the check operation, leave
2605                  * STRIPE_INSYNC not set and let the stripe be handled again
2606                  */
2607                 if (s->failed)
2608                         break;
2609
2610                 /* handle a successful check operation, if parity is correct
2611                  * we are done.  Otherwise update the mismatch count and repair
2612                  * parity if !MD_RECOVERY_CHECK
2613                  */
2614                 if ((sh->ops.zero_sum_result & SUM_CHECK_P_RESULT) == 0)
2615                         /* parity is correct (on disc,
2616                          * not in buffer any more)
2617                          */
2618                         set_bit(STRIPE_INSYNC, &sh->state);
2619                 else {
2620                         conf->mddev->resync_mismatches += STRIPE_SECTORS;
2621                         if (test_bit(MD_RECOVERY_CHECK, &conf->mddev->recovery))
2622                                 /* don't try to repair!! */
2623                                 set_bit(STRIPE_INSYNC, &sh->state);
2624                         else {
2625                                 sh->check_state = check_state_compute_run;
2626                                 set_bit(STRIPE_COMPUTE_RUN, &sh->state);
2627                                 set_bit(STRIPE_OP_COMPUTE_BLK, &s->ops_request);
2628                                 set_bit(R5_Wantcompute,
2629                                         &sh->dev[sh->pd_idx].flags);
2630                                 sh->ops.target = sh->pd_idx;
2631                                 sh->ops.target2 = -1;
2632                                 s->uptodate++;
2633                         }
2634                 }
2635                 break;
2636         case check_state_compute_run:
2637                 break;
2638         default:
2639                 printk(KERN_ERR "%s: unknown check_state: %d sector: %llu\n",
2640                        __func__, sh->check_state,
2641                        (unsigned long long) sh->sector);
2642                 BUG();
2643         }
2644 }
2645
2646
2647 static void handle_parity_checks6(raid5_conf_t *conf, struct stripe_head *sh,
2648                                   struct stripe_head_state *s,
2649                                   struct r6_state *r6s, int disks)
2650 {
2651         int pd_idx = sh->pd_idx;
2652         int qd_idx = sh->qd_idx;
2653         struct r5dev *dev;
2654
2655         set_bit(STRIPE_HANDLE, &sh->state);
2656
2657         BUG_ON(s->failed > 2);
2658
2659         /* Want to check and possibly repair P and Q.
2660          * However there could be one 'failed' device, in which
2661          * case we can only check one of them, possibly using the
2662          * other to generate missing data
2663          */
2664
2665         switch (sh->check_state) {
2666         case check_state_idle:
2667                 /* start a new check operation if there are < 2 failures */
2668                 if (s->failed == r6s->q_failed) {
2669                         /* The only possible failed device holds Q, so it
2670                          * makes sense to check P (If anything else were failed,
2671                          * we would have used P to recreate it).
2672                          */
2673                         sh->check_state = check_state_run;
2674                 }
2675                 if (!r6s->q_failed && s->failed < 2) {
2676                         /* Q is not failed, and we didn't use it to generate
2677                          * anything, so it makes sense to check it
2678                          */
2679                         if (sh->check_state == check_state_run)
2680                                 sh->check_state = check_state_run_pq;
2681                         else
2682                                 sh->check_state = check_state_run_q;
2683                 }
2684
2685                 /* discard potentially stale zero_sum_result */
2686                 sh->ops.zero_sum_result = 0;
2687
2688                 if (sh->check_state == check_state_run) {
2689                         /* async_xor_zero_sum destroys the contents of P */
2690                         clear_bit(R5_UPTODATE, &sh->dev[pd_idx].flags);
2691                         s->uptodate--;
2692                 }
2693                 if (sh->check_state >= check_state_run &&
2694                     sh->check_state <= check_state_run_pq) {
2695                         /* async_syndrome_zero_sum preserves P and Q, so
2696                          * no need to mark them !uptodate here
2697                          */
2698                         set_bit(STRIPE_OP_CHECK, &s->ops_request);
2699                         break;
2700                 }
2701
2702                 /* we have 2-disk failure */
2703                 BUG_ON(s->failed != 2);
2704                 /* fall through */
2705         case check_state_compute_result:
2706                 sh->check_state = check_state_idle;
2707
2708                 /* check that a write has not made the stripe insync */
2709                 if (test_bit(STRIPE_INSYNC, &sh->state))
2710                         break;
2711
2712                 /* now write out any block on a failed drive,
2713                  * or P or Q if they were recomputed
2714                  */
2715                 BUG_ON(s->uptodate < disks - 1); /* We don't need Q to recover */
2716                 if (s->failed == 2) {
2717                         dev = &sh->dev[r6s->failed_num[1]];
2718                         s->locked++;
2719                         set_bit(R5_LOCKED, &dev->flags);
2720                         set_bit(R5_Wantwrite, &dev->flags);
2721                 }
2722                 if (s->failed >= 1) {
2723                         dev = &sh->dev[r6s->failed_num[0]];
2724                         s->locked++;
2725                         set_bit(R5_LOCKED, &dev->flags);
2726                         set_bit(R5_Wantwrite, &dev->flags);
2727                 }
2728                 if (sh->ops.zero_sum_result & SUM_CHECK_P_RESULT) {
2729                         dev = &sh->dev[pd_idx];
2730                         s->locked++;
2731                         set_bit(R5_LOCKED, &dev->flags);
2732                         set_bit(R5_Wantwrite, &dev->flags);
2733                 }
2734                 if (sh->ops.zero_sum_result & SUM_CHECK_Q_RESULT) {
2735                         dev = &sh->dev[qd_idx];
2736                         s->locked++;
2737                         set_bit(R5_LOCKED, &dev->flags);
2738                         set_bit(R5_Wantwrite, &dev->flags);
2739                 }
2740                 clear_bit(STRIPE_DEGRADED, &sh->state);
2741
2742                 set_bit(STRIPE_INSYNC, &sh->state);
2743                 break;
2744         case check_state_run:
2745         case check_state_run_q:
2746         case check_state_run_pq:
2747                 break; /* we will be called again upon completion */
2748         case check_state_check_result:
2749                 sh->check_state = check_state_idle;
2750
2751                 /* handle a successful check operation, if parity is correct
2752                  * we are done.  Otherwise update the mismatch count and repair
2753                  * parity if !MD_RECOVERY_CHECK
2754                  */
2755                 if (sh->ops.zero_sum_result == 0) {
2756                         /* both parities are correct */
2757                         if (!s->failed)
2758                                 set_bit(STRIPE_INSYNC, &sh->state);
2759                         else {
2760                                 /* in contrast to the raid5 case we can validate
2761                                  * parity, but still have a failure to write
2762                                  * back
2763                                  */
2764                                 sh->check_state = check_state_compute_result;
2765                                 /* Returning at this point means that we may go
2766                                  * off and bring p and/or q uptodate again so
2767                                  * we make sure to check zero_sum_result again
2768                                  * to verify if p or q need writeback
2769                                  */
2770                         }
2771                 } else {
2772                         conf->mddev->resync_mismatches += STRIPE_SECTORS;
2773                         if (test_bit(MD_RECOVERY_CHECK, &conf->mddev->recovery))
2774                                 /* don't try to repair!! */
2775                                 set_bit(STRIPE_INSYNC, &sh->state);
2776                         else {
2777                                 int *target = &sh->ops.target;
2778
2779                                 sh->ops.target = -1;
2780                                 sh->ops.target2 = -1;
2781                                 sh->check_state = check_state_compute_run;
2782                                 set_bit(STRIPE_COMPUTE_RUN, &sh->state);
2783                                 set_bit(STRIPE_OP_COMPUTE_BLK, &s->ops_request);
2784                                 if (sh->ops.zero_sum_result & SUM_CHECK_P_RESULT) {
2785                                         set_bit(R5_Wantcompute,
2786                                                 &sh->dev[pd_idx].flags);
2787                                         *target = pd_idx;
2788                                         target = &sh->ops.target2;
2789                                         s->uptodate++;
2790                                 }
2791                                 if (sh->ops.zero_sum_result & SUM_CHECK_Q_RESULT) {
2792                                         set_bit(R5_Wantcompute,
2793                                                 &sh->dev[qd_idx].flags);
2794                                         *target = qd_idx;
2795                                         s->uptodate++;
2796                                 }
2797                         }
2798                 }
2799                 break;
2800         case check_state_compute_run:
2801                 break;
2802         default:
2803                 printk(KERN_ERR "%s: unknown check_state: %d sector: %llu\n",
2804                        __func__, sh->check_state,
2805                        (unsigned long long) sh->sector);
2806                 BUG();
2807         }
2808 }
2809
2810 static void handle_stripe_expansion(raid5_conf_t *conf, struct stripe_head *sh,
2811                                 struct r6_state *r6s)
2812 {
2813         int i;
2814
2815         /* We have read all the blocks in this stripe and now we need to
2816          * copy some of them into a target stripe for expand.
2817          */
2818         struct dma_async_tx_descriptor *tx = NULL;
2819         clear_bit(STRIPE_EXPAND_SOURCE, &sh->state);
2820         for (i = 0; i < sh->disks; i++)
2821                 if (i != sh->pd_idx && i != sh->qd_idx) {
2822                         int dd_idx, j;
2823                         struct stripe_head *sh2;
2824                         struct async_submit_ctl submit;
2825
2826                         sector_t bn = compute_blocknr(sh, i, 1);
2827                         sector_t s = raid5_compute_sector(conf, bn, 0,
2828                                                           &dd_idx, NULL);
2829                         sh2 = get_active_stripe(conf, s, 0, 1);
2830                         if (sh2 == NULL)
2831                                 /* so far only the early blocks of this stripe
2832                                  * have been requested.  When later blocks
2833                                  * get requested, we will try again
2834                                  */
2835                                 continue;
2836                         if (!test_bit(STRIPE_EXPANDING, &sh2->state) ||
2837                            test_bit(R5_Expanded, &sh2->dev[dd_idx].flags)) {
2838                                 /* must have already done this block */
2839                                 release_stripe(sh2);
2840                                 continue;
2841                         }
2842
2843                         /* place all the copies on one channel */
2844                         init_async_submit(&submit, 0, tx, NULL, NULL, NULL);
2845                         tx = async_memcpy(sh2->dev[dd_idx].page,
2846                                           sh->dev[i].page, 0, 0, STRIPE_SIZE,
2847                                           &submit);
2848
2849                         set_bit(R5_Expanded, &sh2->dev[dd_idx].flags);
2850                         set_bit(R5_UPTODATE, &sh2->dev[dd_idx].flags);
2851                         for (j = 0; j < conf->raid_disks; j++)
2852                                 if (j != sh2->pd_idx &&
2853                                     (!r6s || j != sh2->qd_idx) &&
2854                                     !test_bit(R5_Expanded, &sh2->dev[j].flags))
2855                                         break;
2856                         if (j == conf->raid_disks) {
2857                                 set_bit(STRIPE_EXPAND_READY, &sh2->state);
2858                                 set_bit(STRIPE_HANDLE, &sh2->state);
2859                         }
2860                         release_stripe(sh2);
2861
2862                 }
2863         /* done submitting copies, wait for them to complete */
2864         if (tx) {
2865                 async_tx_ack(tx);
2866                 dma_wait_for_async_tx(tx);
2867         }
2868 }
2869
2870
2871 /*
2872  * handle_stripe - do things to a stripe.
2873  *
2874  * We lock the stripe and then examine the state of various bits
2875  * to see what needs to be done.
2876  * Possible results:
2877  *    return some read request which now have data
2878  *    return some write requests which are safely on disc
2879  *    schedule a read on some buffers
2880  *    schedule a write of some buffers
2881  *    return confirmation of parity correctness
2882  *
2883  * buffers are taken off read_list or write_list, and bh_cache buffers
2884  * get BH_Lock set before the stripe lock is released.
2885  *
2886  */
2887
2888 static bool handle_stripe5(struct stripe_head *sh)
2889 {
2890         raid5_conf_t *conf = sh->raid_conf;
2891         int disks = sh->disks, i;
2892         struct bio *return_bi = NULL;
2893         struct stripe_head_state s;
2894         struct r5dev *dev;
2895         mdk_rdev_t *blocked_rdev = NULL;
2896         int prexor;
2897
2898         memset(&s, 0, sizeof(s));
2899         pr_debug("handling stripe %llu, state=%#lx cnt=%d, pd_idx=%d check:%d "
2900                  "reconstruct:%d\n", (unsigned long long)sh->sector, sh->state,
2901                  atomic_read(&sh->count), sh->pd_idx, sh->check_state,
2902                  sh->reconstruct_state);
2903
2904         spin_lock(&sh->lock);
2905         clear_bit(STRIPE_HANDLE, &sh->state);
2906         clear_bit(STRIPE_DELAYED, &sh->state);
2907
2908         s.syncing = test_bit(STRIPE_SYNCING, &sh->state);
2909         s.expanding = test_bit(STRIPE_EXPAND_SOURCE, &sh->state);
2910         s.expanded = test_bit(STRIPE_EXPAND_READY, &sh->state);
2911
2912         /* Now to look around and see what can be done */
2913         rcu_read_lock();
2914         for (i=disks; i--; ) {
2915                 mdk_rdev_t *rdev;
2916                 struct r5dev *dev = &sh->dev[i];
2917                 clear_bit(R5_Insync, &dev->flags);
2918
2919                 pr_debug("check %d: state 0x%lx toread %p read %p write %p "
2920                         "written %p\n", i, dev->flags, dev->toread, dev->read,
2921                         dev->towrite, dev->written);
2922
2923                 /* maybe we can request a biofill operation
2924                  *
2925                  * new wantfill requests are only permitted while
2926                  * ops_complete_biofill is guaranteed to be inactive
2927                  */
2928                 if (test_bit(R5_UPTODATE, &dev->flags) && dev->toread &&
2929                     !test_bit(STRIPE_BIOFILL_RUN, &sh->state))
2930                         set_bit(R5_Wantfill, &dev->flags);
2931
2932                 /* now count some things */
2933                 if (test_bit(R5_LOCKED, &dev->flags)) s.locked++;
2934                 if (test_bit(R5_UPTODATE, &dev->flags)) s.uptodate++;
2935                 if (test_bit(R5_Wantcompute, &dev->flags)) s.compute++;
2936
2937                 if (test_bit(R5_Wantfill, &dev->flags))
2938                         s.to_fill++;
2939                 else if (dev->toread)
2940                         s.to_read++;
2941                 if (dev->towrite) {
2942                         s.to_write++;
2943                         if (!test_bit(R5_OVERWRITE, &dev->flags))
2944                                 s.non_overwrite++;
2945                 }
2946                 if (dev->written)
2947                         s.written++;
2948                 rdev = rcu_dereference(conf->disks[i].rdev);
2949                 if (blocked_rdev == NULL &&
2950                     rdev && unlikely(test_bit(Blocked, &rdev->flags))) {
2951                         blocked_rdev = rdev;
2952                         atomic_inc(&rdev->nr_pending);
2953                 }
2954                 if (!rdev || !test_bit(In_sync, &rdev->flags)) {
2955                         /* The ReadError flag will just be confusing now */
2956                         clear_bit(R5_ReadError, &dev->flags);
2957                         clear_bit(R5_ReWrite, &dev->flags);
2958                 }
2959                 if (!rdev || !test_bit(In_sync, &rdev->flags)
2960                     || test_bit(R5_ReadError, &dev->flags)) {
2961                         s.failed++;
2962                         s.failed_num = i;
2963                 } else
2964                         set_bit(R5_Insync, &dev->flags);
2965         }
2966         rcu_read_unlock();
2967
2968         if (unlikely(blocked_rdev)) {
2969                 if (s.syncing || s.expanding || s.expanded ||
2970                     s.to_write || s.written) {
2971                         set_bit(STRIPE_HANDLE, &sh->state);
2972                         goto unlock;
2973                 }
2974                 /* There is nothing for the blocked_rdev to block */
2975                 rdev_dec_pending(blocked_rdev, conf->mddev);
2976                 blocked_rdev = NULL;
2977         }
2978
2979         if (s.to_fill && !test_bit(STRIPE_BIOFILL_RUN, &sh->state)) {
2980                 set_bit(STRIPE_OP_BIOFILL, &s.ops_request);
2981                 set_bit(STRIPE_BIOFILL_RUN, &sh->state);
2982         }
2983
2984         pr_debug("locked=%d uptodate=%d to_read=%d"
2985                 " to_write=%d failed=%d failed_num=%d\n",
2986                 s.locked, s.uptodate, s.to_read, s.to_write,
2987                 s.failed, s.failed_num);
2988         /* check if the array has lost two devices and, if so, some requests might
2989          * need to be failed
2990          */
2991         if (s.failed > 1 && s.to_read+s.to_write+s.written)
2992                 handle_failed_stripe(conf, sh, &s, disks, &return_bi);
2993         if (s.failed > 1 && s.syncing) {
2994                 md_done_sync(conf->mddev, STRIPE_SECTORS,0);
2995                 clear_bit(STRIPE_SYNCING, &sh->state);
2996                 s.syncing = 0;
2997         }
2998
2999         /* might be able to return some write requests if the parity block
3000          * is safe, or on a failed drive
3001          */
3002         dev = &sh->dev[sh->pd_idx];
3003         if ( s.written &&
3004              ((test_bit(R5_Insync, &dev->flags) &&
3005                !test_bit(R5_LOCKED, &dev->flags) &&
3006                test_bit(R5_UPTODATE, &dev->flags)) ||
3007                (s.failed == 1 && s.failed_num == sh->pd_idx)))
3008                 handle_stripe_clean_event(conf, sh, disks, &return_bi);
3009
3010         /* Now we might consider reading some blocks, either to check/generate
3011          * parity, or to satisfy requests
3012          * or to load a block that is being partially written.
3013          */
3014         if (s.to_read || s.non_overwrite ||
3015             (s.syncing && (s.uptodate + s.compute < disks)) || s.expanding)
3016                 handle_stripe_fill5(sh, &s, disks);
3017
3018         /* Now we check to see if any write operations have recently
3019          * completed
3020          */
3021         prexor = 0;
3022         if (sh->reconstruct_state == reconstruct_state_prexor_drain_result)
3023                 prexor = 1;
3024         if (sh->reconstruct_state == reconstruct_state_drain_result ||
3025             sh->reconstruct_state == reconstruct_state_prexor_drain_result) {
3026                 sh->reconstruct_state = reconstruct_state_idle;
3027
3028                 /* All the 'written' buffers and the parity block are ready to
3029                  * be written back to disk
3030                  */
3031                 BUG_ON(!test_bit(R5_UPTODATE, &sh->dev[sh->pd_idx].flags));
3032                 for (i = disks; i--; ) {
3033                         dev = &sh->dev[i];
3034                         if (test_bit(R5_LOCKED, &dev->flags) &&
3035                                 (i == sh->pd_idx || dev->written)) {
3036                                 pr_debug("Writing block %d\n", i);
3037                                 set_bit(R5_Wantwrite, &dev->flags);
3038                                 if (prexor)
3039                                         continue;
3040                                 if (!test_bit(R5_Insync, &dev->flags) ||
3041                                     (i == sh->pd_idx && s.failed == 0))
3042                                         set_bit(STRIPE_INSYNC, &sh->state);
3043                         }
3044                 }
3045                 if (test_and_clear_bit(STRIPE_PREREAD_ACTIVE, &sh->state)) {
3046                         atomic_dec(&conf->preread_active_stripes);
3047                         if (atomic_read(&conf->preread_active_stripes) <
3048                                 IO_THRESHOLD)
3049                                 md_wakeup_thread(conf->mddev->thread);
3050                 }
3051         }
3052
3053         /* Now to consider new write requests and what else, if anything
3054          * should be read.  We do not handle new writes when:
3055          * 1/ A 'write' operation (copy+xor) is already in flight.
3056          * 2/ A 'check' operation is in flight, as it may clobber the parity
3057          *    block.
3058          */
3059         if (s.to_write && !sh->reconstruct_state && !sh->check_state)
3060                 handle_stripe_dirtying5(conf, sh, &s, disks);
3061
3062         /* maybe we need to check and possibly fix the parity for this stripe
3063          * Any reads will already have been scheduled, so we just see if enough
3064          * data is available.  The parity check is held off while parity
3065          * dependent operations are in flight.
3066          */
3067         if (sh->check_state ||
3068             (s.syncing && s.locked == 0 &&
3069              !test_bit(STRIPE_COMPUTE_RUN, &sh->state) &&
3070              !test_bit(STRIPE_INSYNC, &sh->state)))
3071                 handle_parity_checks5(conf, sh, &s, disks);
3072
3073         if (s.syncing && s.locked == 0 && test_bit(STRIPE_INSYNC, &sh->state)) {
3074                 md_done_sync(conf->mddev, STRIPE_SECTORS,1);
3075                 clear_bit(STRIPE_SYNCING, &sh->state);
3076         }
3077
3078         /* If the failed drive is just a ReadError, then we might need to progress
3079          * the repair/check process
3080          */
3081         if (s.failed == 1 && !conf->mddev->ro &&
3082             test_bit(R5_ReadError, &sh->dev[s.failed_num].flags)
3083             && !test_bit(R5_LOCKED, &sh->dev[s.failed_num].flags)
3084             && test_bit(R5_UPTODATE, &sh->dev[s.failed_num].flags)
3085                 ) {
3086                 dev = &sh->dev[s.failed_num];
3087                 if (!test_bit(R5_ReWrite, &dev->flags)) {
3088                         set_bit(R5_Wantwrite, &dev->flags);
3089                         set_bit(R5_ReWrite, &dev->flags);
3090                         set_bit(R5_LOCKED, &dev->flags);
3091                         s.locked++;
3092                 } else {
3093                         /* let's read it back */
3094                         set_bit(R5_Wantread, &dev->flags);
3095                         set_bit(R5_LOCKED, &dev->flags);
3096                         s.locked++;
3097                 }
3098         }
3099
3100         /* Finish reconstruct operations initiated by the expansion process */
3101         if (sh->reconstruct_state == reconstruct_state_result) {
3102                 struct stripe_head *sh2
3103                         = get_active_stripe(conf, sh->sector, 1, 1);
3104                 if (sh2 && test_bit(STRIPE_EXPAND_SOURCE, &sh2->state)) {
3105                         /* sh cannot be written until sh2 has been read.
3106                          * so arrange for sh to be delayed a little
3107                          */
3108                         set_bit(STRIPE_DELAYED, &sh->state);
3109                         set_bit(STRIPE_HANDLE, &sh->state);
3110                         if (!test_and_set_bit(STRIPE_PREREAD_ACTIVE,
3111                                               &sh2->state))
3112                                 atomic_inc(&conf->preread_active_stripes);
3113                         release_stripe(sh2);
3114                         goto unlock;
3115                 }
3116                 if (sh2)
3117                         release_stripe(sh2);
3118
3119                 sh->reconstruct_state = reconstruct_state_idle;
3120                 clear_bit(STRIPE_EXPANDING, &sh->state);
3121                 for (i = conf->raid_disks; i--; ) {
3122                         set_bit(R5_Wantwrite, &sh->dev[i].flags);
3123                         set_bit(R5_LOCKED, &sh->dev[i].flags);
3124                         s.locked++;
3125                 }
3126         }
3127
3128         if (s.expanded && test_bit(STRIPE_EXPANDING, &sh->state) &&
3129             !sh->reconstruct_state) {
3130                 /* Need to write out all blocks after computing parity */
3131                 sh->disks = conf->raid_disks;
3132                 stripe_set_idx(sh->sector, conf, 0, sh);
3133                 schedule_reconstruction(sh, &s, 1, 1);
3134         } else if (s.expanded && !sh->reconstruct_state && s.locked == 0) {
3135                 clear_bit(STRIPE_EXPAND_READY, &sh->state);
3136                 atomic_dec(&conf->reshape_stripes);
3137                 wake_up(&conf->wait_for_overlap);
3138                 md_done_sync(conf->mddev, STRIPE_SECTORS, 1);
3139         }
3140
3141         if (s.expanding && s.locked == 0 &&
3142             !test_bit(STRIPE_COMPUTE_RUN, &sh->state))
3143                 handle_stripe_expansion(conf, sh, NULL);
3144
3145  unlock:
3146         spin_unlock(&sh->lock);
3147
3148         /* wait for this device to become unblocked */
3149         if (unlikely(blocked_rdev))
3150                 md_wait_for_blocked_rdev(blocked_rdev, conf->mddev);
3151
3152         if (s.ops_request)
3153                 raid_run_ops(sh, s.ops_request);
3154
3155         ops_run_io(sh, &s);
3156
3157         return_io(return_bi);
3158
3159         return blocked_rdev == NULL;
3160 }
3161
3162 static bool handle_stripe6(struct stripe_head *sh)
3163 {
3164         raid5_conf_t *conf = sh->raid_conf;
3165         int disks = sh->disks;
3166         struct bio *return_bi = NULL;
3167         int i, pd_idx = sh->pd_idx, qd_idx = sh->qd_idx;
3168         struct stripe_head_state s;
3169         struct r6_state r6s;
3170         struct r5dev *dev, *pdev, *qdev;
3171         mdk_rdev_t *blocked_rdev = NULL;
3172
3173         pr_debug("handling stripe %llu, state=%#lx cnt=%d, "
3174                 "pd_idx=%d, qd_idx=%d\n, check:%d, reconstruct:%d\n",
3175                (unsigned long long)sh->sector, sh->state,
3176                atomic_read(&sh->count), pd_idx, qd_idx,
3177                sh->check_state, sh->reconstruct_state);
3178         memset(&s, 0, sizeof(s));
3179
3180         spin_lock(&sh->lock);
3181         clear_bit(STRIPE_HANDLE, &sh->state);
3182         clear_bit(STRIPE_DELAYED, &sh->state);
3183
3184         s.syncing = test_bit(STRIPE_SYNCING, &sh->state);
3185         s.expanding = test_bit(STRIPE_EXPAND_SOURCE, &sh->state);
3186         s.expanded = test_bit(STRIPE_EXPAND_READY, &sh->state);
3187         /* Now to look around and see what can be done */
3188
3189         rcu_read_lock();
3190         for (i=disks; i--; ) {
3191                 mdk_rdev_t *rdev;
3192                 dev = &sh->dev[i];
3193                 clear_bit(R5_Insync, &dev->flags);
3194
3195                 pr_debug("check %d: state 0x%lx read %p write %p written %p\n",
3196                         i, dev->flags, dev->toread, dev->towrite, dev->written);
3197                 /* maybe we can reply to a read
3198                  *
3199                  * new wantfill requests are only permitted while
3200                  * ops_complete_biofill is guaranteed to be inactive
3201                  */
3202                 if (test_bit(R5_UPTODATE, &dev->flags) && dev->toread &&
3203                     !test_bit(STRIPE_BIOFILL_RUN, &sh->state))
3204                         set_bit(R5_Wantfill, &dev->flags);
3205
3206                 /* now count some things */
3207                 if (test_bit(R5_LOCKED, &dev->flags)) s.locked++;
3208                 if (test_bit(R5_UPTODATE, &dev->flags)) s.uptodate++;
3209                 if (test_bit(R5_Wantcompute, &dev->flags))
3210                         BUG_ON(++s.compute > 2);
3211
3212                 if (test_bit(R5_Wantfill, &dev->flags)) {
3213                         s.to_fill++;
3214                 } else if (dev->toread)
3215                         s.to_read++;
3216                 if (dev->towrite) {
3217                         s.to_write++;
3218                         if (!test_bit(R5_OVERWRITE, &dev->flags))
3219                                 s.non_overwrite++;
3220                 }
3221                 if (dev->written)
3222                         s.written++;
3223                 rdev = rcu_dereference(conf->disks[i].rdev);
3224                 if (blocked_rdev == NULL &&
3225                     rdev && unlikely(test_bit(Blocked, &rdev->flags))) {
3226                         blocked_rdev = rdev;
3227                         atomic_inc(&rdev->nr_pending);
3228                 }
3229                 if (!rdev || !test_bit(In_sync, &rdev->flags)) {
3230                         /* The ReadError flag will just be confusing now */
3231                         clear_bit(R5_ReadError, &dev->flags);
3232                         clear_bit(R5_ReWrite, &dev->flags);
3233                 }
3234                 if (!rdev || !test_bit(In_sync, &rdev->flags)
3235                     || test_bit(R5_ReadError, &dev->flags)) {
3236                         if (s.failed < 2)
3237                                 r6s.failed_num[s.failed] = i;
3238                         s.failed++;
3239                 } else
3240                         set_bit(R5_Insync, &dev->flags);
3241         }
3242         rcu_read_unlock();
3243
3244         if (unlikely(blocked_rdev)) {
3245                 if (s.syncing || s.expanding || s.expanded ||
3246                     s.to_write || s.written) {
3247                         set_bit(STRIPE_HANDLE, &sh->state);
3248                         goto unlock;
3249                 }
3250                 /* There is nothing for the blocked_rdev to block */
3251                 rdev_dec_pending(blocked_rdev, conf->mddev);
3252                 blocked_rdev = NULL;
3253         }
3254
3255         if (s.to_fill && !test_bit(STRIPE_BIOFILL_RUN, &sh->state)) {
3256                 set_bit(STRIPE_OP_BIOFILL, &s.ops_request);
3257                 set_bit(STRIPE_BIOFILL_RUN, &sh->state);
3258         }
3259
3260         pr_debug("locked=%d uptodate=%d to_read=%d"
3261                " to_write=%d failed=%d failed_num=%d,%d\n",
3262                s.locked, s.uptodate, s.to_read, s.to_write, s.failed,
3263                r6s.failed_num[0], r6s.failed_num[1]);
3264         /* check if the array has lost >2 devices and, if so, some requests
3265          * might need to be failed
3266          */
3267         if (s.failed > 2 && s.to_read+s.to_write+s.written)
3268                 handle_failed_stripe(conf, sh, &s, disks, &return_bi);
3269         if (s.failed > 2 && s.syncing) {
3270                 md_done_sync(conf->mddev, STRIPE_SECTORS,0);
3271                 clear_bit(STRIPE_SYNCING, &sh->state);
3272                 s.syncing = 0;
3273         }
3274
3275         /*
3276          * might be able to return some write requests if the parity blocks
3277          * are safe, or on a failed drive
3278          */
3279         pdev = &sh->dev[pd_idx];
3280         r6s.p_failed = (s.failed >= 1 && r6s.failed_num[0] == pd_idx)
3281                 || (s.failed >= 2 && r6s.failed_num[1] == pd_idx);
3282         qdev = &sh->dev[qd_idx];
3283         r6s.q_failed = (s.failed >= 1 && r6s.failed_num[0] == qd_idx)
3284                 || (s.failed >= 2 && r6s.failed_num[1] == qd_idx);
3285
3286         if ( s.written &&
3287              ( r6s.p_failed || ((test_bit(R5_Insync, &pdev->flags)
3288                              && !test_bit(R5_LOCKED, &pdev->flags)
3289                              && test_bit(R5_UPTODATE, &pdev->flags)))) &&
3290              ( r6s.q_failed || ((test_bit(R5_Insync, &qdev->flags)
3291                              && !test_bit(R5_LOCKED, &qdev->flags)
3292                              && test_bit(R5_UPTODATE, &qdev->flags)))))
3293                 handle_stripe_clean_event(conf, sh, disks, &return_bi);
3294
3295         /* Now we might consider reading some blocks, either to check/generate
3296          * parity, or to satisfy requests
3297          * or to load a block that is being partially written.
3298          */
3299         if (s.to_read || s.non_overwrite || (s.to_write && s.failed) ||
3300             (s.syncing && (s.uptodate + s.compute < disks)) || s.expanding)
3301                 handle_stripe_fill6(sh, &s, &r6s, disks);
3302
3303         /* Now we check to see if any write operations have recently
3304          * completed
3305          */
3306         if (sh->reconstruct_state == reconstruct_state_drain_result) {
3307                 int qd_idx = sh->qd_idx;
3308
3309                 sh->reconstruct_state = reconstruct_state_idle;
3310                 /* All the 'written' buffers and the parity blocks are ready to
3311                  * be written back to disk
3312                  */
3313                 BUG_ON(!test_bit(R5_UPTODATE, &sh->dev[sh->pd_idx].flags));
3314                 BUG_ON(!test_bit(R5_UPTODATE, &sh->dev[qd_idx].flags));
3315                 for (i = disks; i--; ) {
3316                         dev = &sh->dev[i];
3317                         if (test_bit(R5_LOCKED, &dev->flags) &&
3318                             (i == sh->pd_idx || i == qd_idx ||
3319                              dev->written)) {
3320                                 pr_debug("Writing block %d\n", i);
3321                                 BUG_ON(!test_bit(R5_UPTODATE, &dev->flags));
3322                                 set_bit(R5_Wantwrite, &dev->flags);
3323                                 if (!test_bit(R5_Insync, &dev->flags) ||
3324                                     ((i == sh->pd_idx || i == qd_idx) &&
3325                                       s.failed == 0))
3326                                         set_bit(STRIPE_INSYNC, &sh->state);
3327                         }
3328                 }
3329                 if (test_and_clear_bit(STRIPE_PREREAD_ACTIVE, &sh->state)) {
3330                         atomic_dec(&conf->preread_active_stripes);
3331                         if (atomic_read(&conf->preread_active_stripes) <
3332                                 IO_THRESHOLD)
3333                                 md_wakeup_thread(conf->mddev->thread);
3334                 }
3335         }
3336
3337         /* Now to consider new write requests and what else, if anything
3338          * should be read.  We do not handle new writes when:
3339          * 1/ A 'write' operation (copy+gen_syndrome) is already in flight.
3340          * 2/ A 'check' operation is in flight, as it may clobber the parity
3341          *    block.
3342          */
3343         if (s.to_write && !sh->reconstruct_state && !sh->check_state)
3344                 handle_stripe_dirtying6(conf, sh, &s, &r6s, disks);
3345
3346         /* maybe we need to check and possibly fix the parity for this stripe
3347          * Any reads will already have been scheduled, so we just see if enough
3348          * data is available.  The parity check is held off while parity
3349          * dependent operations are in flight.
3350          */
3351         if (sh->check_state ||
3352             (s.syncing && s.locked == 0 &&
3353              !test_bit(STRIPE_COMPUTE_RUN, &sh->state) &&
3354              !test_bit(STRIPE_INSYNC, &sh->state)))
3355                 handle_parity_checks6(conf, sh, &s, &r6s, disks);
3356
3357         if (s.syncing && s.locked == 0 && test_bit(STRIPE_INSYNC, &sh->state)) {
3358                 md_done_sync(conf->mddev, STRIPE_SECTORS,1);
3359                 clear_bit(STRIPE_SYNCING, &sh->state);
3360         }
3361
3362         /* If the failed drives are just a ReadError, then we might need
3363          * to progress the repair/check process
3364          */
3365         if (s.failed <= 2 && !conf->mddev->ro)
3366                 for (i = 0; i < s.failed; i++) {
3367                         dev = &sh->dev[r6s.failed_num[i]];
3368                         if (test_bit(R5_ReadError, &dev->flags)
3369                             && !test_bit(R5_LOCKED, &dev->flags)
3370                             && test_bit(R5_UPTODATE, &dev->flags)
3371                                 ) {
3372                                 if (!test_bit(R5_ReWrite, &dev->flags)) {
3373                                         set_bit(R5_Wantwrite, &dev->flags);
3374                                         set_bit(R5_ReWrite, &dev->flags);
3375                                         set_bit(R5_LOCKED, &dev->flags);
3376                                         s.locked++;
3377                                 } else {
3378                                         /* let's read it back */
3379                                         set_bit(R5_Wantread, &dev->flags);
3380                                         set_bit(R5_LOCKED, &dev->flags);
3381                                         s.locked++;
3382                                 }
3383                         }
3384                 }
3385
3386         /* Finish reconstruct operations initiated by the expansion process */
3387         if (sh->reconstruct_state == reconstruct_state_result) {
3388                 sh->reconstruct_state = reconstruct_state_idle;
3389                 clear_bit(STRIPE_EXPANDING, &sh->state);
3390                 for (i = conf->raid_disks; i--; ) {
3391                         set_bit(R5_Wantwrite, &sh->dev[i].flags);
3392                         set_bit(R5_LOCKED, &sh->dev[i].flags);
3393                         s.locked++;
3394                 }
3395         }
3396
3397         if (s.expanded && test_bit(STRIPE_EXPANDING, &sh->state) &&
3398             !sh->reconstruct_state) {
3399                 struct stripe_head *sh2
3400                         = get_active_stripe(conf, sh->sector, 1, 1);
3401                 if (sh2 && test_bit(STRIPE_EXPAND_SOURCE, &sh2->state)) {
3402                         /* sh cannot be written until sh2 has been read.
3403                          * so arrange for sh to be delayed a little
3404                          */
3405                         set_bit(STRIPE_DELAYED, &sh->state);
3406                         set_bit(STRIPE_HANDLE, &sh->state);
3407                         if (!test_and_set_bit(STRIPE_PREREAD_ACTIVE,
3408                                               &sh2->state))
3409                                 atomic_inc(&conf->preread_active_stripes);
3410                         release_stripe(sh2);
3411                         goto unlock;
3412                 }
3413                 if (sh2)
3414                         release_stripe(sh2);
3415
3416                 /* Need to write out all blocks after computing P&Q */
3417                 sh->disks = conf->raid_disks;
3418                 stripe_set_idx(sh->sector, conf, 0, sh);
3419                 schedule_reconstruction(sh, &s, 1, 1);
3420         } else if (s.expanded && !sh->reconstruct_state && s.locked == 0) {
3421                 clear_bit(STRIPE_EXPAND_READY, &sh->state);
3422                 atomic_dec(&conf->reshape_stripes);
3423                 wake_up(&conf->wait_for_overlap);
3424                 md_done_sync(conf->mddev, STRIPE_SECTORS, 1);
3425         }
3426
3427         if (s.expanding && s.locked == 0 &&
3428             !test_bit(STRIPE_COMPUTE_RUN, &sh->state))
3429                 handle_stripe_expansion(conf, sh, &r6s);
3430
3431  unlock:
3432         spin_unlock(&sh->lock);
3433
3434         /* wait for this device to become unblocked */
3435         if (unlikely(blocked_rdev))
3436                 md_wait_for_blocked_rdev(blocked_rdev, conf->mddev);
3437
3438         if (s.ops_request)
3439                 raid_run_ops(sh, s.ops_request);
3440
3441         ops_run_io(sh, &s);
3442
3443         return_io(return_bi);
3444
3445         return blocked_rdev == NULL;
3446 }
3447
3448 /* returns true if the stripe was handled */
3449 static bool handle_stripe(struct stripe_head *sh)
3450 {
3451         if (sh->raid_conf->level == 6)
3452                 return handle_stripe6(sh);
3453         else
3454                 return handle_stripe5(sh);
3455 }
3456
3457 static void raid5_activate_delayed(raid5_conf_t *conf)
3458 {
3459         if (atomic_read(&conf->preread_active_stripes) < IO_THRESHOLD) {
3460                 while (!list_empty(&conf->delayed_list)) {
3461                         struct list_head *l = conf->delayed_list.next;
3462                         struct stripe_head *sh;
3463                         sh = list_entry(l, struct stripe_head, lru);
3464                         list_del_init(l);
3465                         clear_bit(STRIPE_DELAYED, &sh->state);
3466                         if (!test_and_set_bit(STRIPE_PREREAD_ACTIVE, &sh->state))
3467                                 atomic_inc(&conf->preread_active_stripes);
3468                         list_add_tail(&sh->lru, &conf->hold_list);
3469                 }
3470         } else
3471                 blk_plug_device(conf->mddev->queue);
3472 }
3473
3474 static void activate_bit_delay(raid5_conf_t *conf)
3475 {
3476         /* device_lock is held */
3477         struct list_head head;
3478         list_add(&head, &conf->bitmap_list);
3479         list_del_init(&conf->bitmap_list);
3480         while (!list_empty(&head)) {
3481                 struct stripe_head *sh = list_entry(head.next, struct stripe_head, lru);
3482                 list_del_init(&sh->lru);
3483                 atomic_inc(&sh->count);
3484                 __release_stripe(conf, sh);
3485         }
3486 }
3487
3488 static void unplug_slaves(mddev_t *mddev)
3489 {
3490         raid5_conf_t *conf = mddev_to_conf(mddev);
3491         int i;
3492
3493         rcu_read_lock();
3494         for (i=0; i<mddev->raid_disks; i++) {
3495                 mdk_rdev_t *rdev = rcu_dereference(conf->disks[i].rdev);
3496                 if (rdev && !test_bit(Faulty, &rdev->flags) && atomic_read(&rdev->nr_pending)) {
3497                         struct request_queue *r_queue = bdev_get_queue(rdev->bdev);
3498
3499                         atomic_inc(&rdev->nr_pending);
3500                         rcu_read_unlock();
3501
3502                         blk_unplug(r_queue);
3503
3504                         rdev_dec_pending(rdev, mddev);
3505                         rcu_read_lock();
3506                 }
3507         }
3508         rcu_read_unlock();
3509 }
3510
3511 static void raid5_unplug_device(struct request_queue *q)
3512 {
3513         mddev_t *mddev = q->queuedata;
3514         raid5_conf_t *conf = mddev_to_conf(mddev);
3515         unsigned long flags;
3516
3517         spin_lock_irqsave(&conf->device_lock, flags);
3518
3519         if (blk_remove_plug(q)) {
3520                 conf->seq_flush++;
3521                 raid5_activate_delayed(conf);
3522         }
3523         md_wakeup_thread(mddev->thread);
3524
3525         spin_unlock_irqrestore(&conf->device_lock, flags);
3526
3527         unplug_slaves(mddev);
3528 }
3529
3530 static int raid5_congested(void *data, int bits)
3531 {
3532         mddev_t *mddev = data;
3533         raid5_conf_t *conf = mddev_to_conf(mddev);
3534
3535         /* No difference between reads and writes.  Just check
3536          * how busy the stripe_cache is
3537          */
3538         if (conf->inactive_blocked)
3539                 return 1;
3540         if (conf->quiesce)
3541                 return 1;
3542         if (list_empty_careful(&conf->inactive_list))
3543                 return 1;
3544
3545         return 0;
3546 }
3547
3548 /* We want read requests to align with chunks where possible,
3549  * but write requests don't need to.
3550  */
3551 static int raid5_mergeable_bvec(struct request_queue *q,
3552                                 struct bvec_merge_data *bvm,
3553                                 struct bio_vec *biovec)
3554 {
3555         mddev_t *mddev = q->queuedata;
3556         sector_t sector = bvm->bi_sector + get_start_sect(bvm->bi_bdev);
3557         int max;
3558         unsigned int chunk_sectors = mddev->chunk_size >> 9;
3559         unsigned int bio_sectors = bvm->bi_size >> 9;
3560
3561         if ((bvm->bi_rw & 1) == WRITE)
3562                 return biovec->bv_len; /* always allow writes to be mergeable */
3563
3564         if (mddev->new_chunk < mddev->chunk_size)
3565                 chunk_sectors = mddev->new_chunk >> 9;
3566         max =  (chunk_sectors - ((sector & (chunk_sectors - 1)) + bio_sectors)) << 9;
3567         if (max < 0) max = 0;
3568         if (max <= biovec->bv_len && bio_sectors == 0)
3569                 return biovec->bv_len;
3570         else
3571                 return max;
3572 }
3573
3574
3575 static int in_chunk_boundary(mddev_t *mddev, struct bio *bio)
3576 {
3577         sector_t sector = bio->bi_sector + get_start_sect(bio->bi_bdev);
3578         unsigned int chunk_sectors = mddev->chunk_size >> 9;
3579         unsigned int bio_sectors = bio->bi_size >> 9;
3580
3581         if (mddev->new_chunk < mddev->chunk_size)
3582                 chunk_sectors = mddev->new_chunk >> 9;
3583         return  chunk_sectors >=
3584                 ((sector & (chunk_sectors - 1)) + bio_sectors);
3585 }
3586
3587 /*
3588  *  add bio to the retry LIFO  ( in O(1) ... we are in interrupt )
3589  *  later sampled by raid5d.
3590  */
3591 static void add_bio_to_retry(struct bio *bi,raid5_conf_t *conf)
3592 {
3593         unsigned long flags;
3594
3595         spin_lock_irqsave(&conf->device_lock, flags);
3596
3597         bi->bi_next = conf->retry_read_aligned_list;
3598         conf->retry_read_aligned_list = bi;
3599
3600         spin_unlock_irqrestore(&conf->device_lock, flags);
3601         md_wakeup_thread(conf->mddev->thread);
3602 }
3603
3604
3605 static struct bio *remove_bio_from_retry(raid5_conf_t *conf)
3606 {
3607         struct bio *bi;
3608
3609         bi = conf->retry_read_aligned;
3610         if (bi) {
3611                 conf->retry_read_aligned = NULL;
3612                 return bi;
3613         }
3614         bi = conf->retry_read_aligned_list;
3615         if(bi) {
3616                 conf->retry_read_aligned_list = bi->bi_next;
3617                 bi->bi_next = NULL;
3618                 /*
3619                  * this sets the active strip count to 1 and the processed
3620                  * strip count to zero (upper 8 bits)
3621                  */
3622                 bi->bi_phys_segments = 1; /* biased count of active stripes */
3623         }
3624
3625         return bi;
3626 }
3627
3628
3629 /*
3630  *  The "raid5_align_endio" should check if the read succeeded and if it
3631  *  did, call bio_endio on the original bio (having bio_put the new bio
3632  *  first).
3633  *  If the read failed..
3634  */
3635 static void raid5_align_endio(struct bio *bi, int error)
3636 {
3637         struct bio* raid_bi  = bi->bi_private;
3638         mddev_t *mddev;
3639         raid5_conf_t *conf;
3640         int uptodate = test_bit(BIO_UPTODATE, &bi->bi_flags);
3641         mdk_rdev_t *rdev;
3642
3643         bio_put(bi);
3644
3645         mddev = raid_bi->bi_bdev->bd_disk->queue->queuedata;
3646         conf = mddev_to_conf(mddev);
3647         rdev = (void*)raid_bi->bi_next;
3648         raid_bi->bi_next = NULL;
3649
3650         rdev_dec_pending(rdev, conf->mddev);
3651
3652         if (!error && uptodate) {
3653                 bio_endio(raid_bi, 0);
3654                 if (atomic_dec_and_test(&conf->active_aligned_reads))
3655                         wake_up(&conf->wait_for_stripe);
3656                 return;
3657         }
3658
3659
3660         pr_debug("raid5_align_endio : io error...handing IO for a retry\n");
3661
3662         add_bio_to_retry(raid_bi, conf);
3663 }
3664
3665 static int bio_fits_rdev(struct bio *bi)
3666 {
3667         struct request_queue *q = bdev_get_queue(bi->bi_bdev);
3668
3669         if ((bi->bi_size>>9) > q->max_sectors)
3670                 return 0;
3671         blk_recount_segments(q, bi);
3672         if (bi->bi_phys_segments > q->max_phys_segments)
3673                 return 0;
3674
3675         if (q->merge_bvec_fn)
3676                 /* it's too hard to apply the merge_bvec_fn at this stage,
3677                  * just just give up
3678                  */
3679                 return 0;
3680
3681         return 1;
3682 }
3683
3684
3685 static int chunk_aligned_read(struct request_queue *q, struct bio * raid_bio)
3686 {
3687         mddev_t *mddev = q->queuedata;
3688         raid5_conf_t *conf = mddev_to_conf(mddev);
3689         unsigned int dd_idx;
3690         struct bio* align_bi;
3691         mdk_rdev_t *rdev;
3692
3693         if (!in_chunk_boundary(mddev, raid_bio)) {
3694                 pr_debug("chunk_aligned_read : non aligned\n");
3695                 return 0;
3696         }
3697         /*
3698          * use bio_clone to make a copy of the bio
3699          */
3700         align_bi = bio_clone(raid_bio, GFP_NOIO);
3701         if (!align_bi)
3702                 return 0;
3703         /*
3704          *   set bi_end_io to a new function, and set bi_private to the
3705          *     original bio.
3706          */
3707         align_bi->bi_end_io  = raid5_align_endio;
3708         align_bi->bi_private = raid_bio;
3709         /*
3710          *      compute position
3711          */
3712         align_bi->bi_sector =  raid5_compute_sector(conf, raid_bio->bi_sector,
3713                                                     0,
3714                                                     &dd_idx, NULL);
3715
3716         rcu_read_lock();
3717         rdev = rcu_dereference(conf->disks[dd_idx].rdev);
3718         if (rdev && test_bit(In_sync, &rdev->flags)) {
3719                 atomic_inc(&rdev->nr_pending);
3720                 rcu_read_unlock();
3721                 raid_bio->bi_next = (void*)rdev;
3722                 align_bi->bi_bdev =  rdev->bdev;
3723                 align_bi->bi_flags &= ~(1 << BIO_SEG_VALID);
3724                 align_bi->bi_sector += rdev->data_offset;
3725
3726                 if (!bio_fits_rdev(align_bi)) {
3727                         /* too big in some way */
3728                         bio_put(align_bi);
3729                         rdev_dec_pending(rdev, mddev);
3730                         return 0;
3731                 }
3732
3733                 spin_lock_irq(&conf->device_lock);
3734                 wait_event_lock_irq(conf->wait_for_stripe,
3735                                     conf->quiesce == 0,
3736                                     conf->device_lock, /* nothing */);
3737                 atomic_inc(&conf->active_aligned_reads);
3738                 spin_unlock_irq(&conf->device_lock);
3739
3740                 generic_make_request(align_bi);
3741                 return 1;
3742         } else {
3743                 rcu_read_unlock();
3744                 bio_put(align_bi);
3745                 return 0;
3746         }
3747 }
3748
3749 /* __get_priority_stripe - get the next stripe to process
3750  *
3751  * Full stripe writes are allowed to pass preread active stripes up until
3752  * the bypass_threshold is exceeded.  In general the bypass_count
3753  * increments when the handle_list is handled before the hold_list; however, it
3754  * will not be incremented when STRIPE_IO_STARTED is sampled set signifying a
3755  * stripe with in flight i/o.  The bypass_count will be reset when the
3756  * head of the hold_list has changed, i.e. the head was promoted to the
3757  * handle_list.
3758  */
3759 static struct stripe_head *__get_priority_stripe(raid5_conf_t *conf)
3760 {
3761         struct stripe_head *sh;
3762
3763         pr_debug("%s: handle: %s hold: %s full_writes: %d bypass_count: %d\n",
3764                   __func__,
3765                   list_empty(&conf->handle_list) ? "empty" : "busy",
3766                   list_empty(&conf->hold_list) ? "empty" : "busy",
3767                   atomic_read(&conf->pending_full_writes), conf->bypass_count);
3768
3769         if (!list_empty(&conf->handle_list)) {
3770                 sh = list_entry(conf->handle_list.next, typeof(*sh), lru);
3771
3772                 if (list_empty(&conf->hold_list))
3773                         conf->bypass_count = 0;
3774                 else if (!test_bit(STRIPE_IO_STARTED, &sh->state)) {
3775                         if (conf->hold_list.next == conf->last_hold)
3776                                 conf->bypass_count++;
3777                         else {
3778                                 conf->last_hold = conf->hold_list.next;
3779                                 conf->bypass_count -= conf->bypass_threshold;
3780                                 if (conf->bypass_count < 0)
3781                                         conf->bypass_count = 0;
3782                         }
3783                 }
3784         } else if (!list_empty(&conf->hold_list) &&
3785                    ((conf->bypass_threshold &&
3786                      conf->bypass_count > conf->bypass_threshold) ||
3787                     atomic_read(&conf->pending_full_writes) == 0)) {
3788                 sh = list_entry(conf->hold_list.next,
3789                                 typeof(*sh), lru);
3790                 conf->bypass_count -= conf->bypass_threshold;
3791                 if (conf->bypass_count < 0)
3792                         conf->bypass_count = 0;
3793         } else
3794                 return NULL;
3795
3796         list_del_init(&sh->lru);
3797         atomic_inc(&sh->count);
3798         BUG_ON(atomic_read(&sh->count) != 1);
3799         return sh;
3800 }
3801
3802 static int make_request(struct request_queue *q, struct bio * bi)
3803 {
3804         mddev_t *mddev = q->queuedata;
3805         raid5_conf_t *conf = mddev_to_conf(mddev);
3806         int dd_idx;
3807         sector_t new_sector;
3808         sector_t logical_sector, last_sector;
3809         struct stripe_head *sh;
3810         const int rw = bio_data_dir(bi);
3811         int cpu, remaining;
3812
3813         if (unlikely(bio_barrier(bi))) {
3814                 bio_endio(bi, -EOPNOTSUPP);
3815                 return 0;
3816         }
3817
3818         md_write_start(mddev, bi);
3819
3820         cpu = part_stat_lock();
3821         part_stat_inc(cpu, &mddev->gendisk->part0, ios[rw]);
3822         part_stat_add(cpu, &mddev->gendisk->part0, sectors[rw],
3823                       bio_sectors(bi));
3824         part_stat_unlock();
3825
3826         if (rw == READ &&
3827              mddev->reshape_position == MaxSector &&
3828              chunk_aligned_read(q,bi))
3829                 return 0;
3830
3831         logical_sector = bi->bi_sector & ~((sector_t)STRIPE_SECTORS-1);
3832         last_sector = bi->bi_sector + (bi->bi_size>>9);
3833         bi->bi_next = NULL;
3834         bi->bi_phys_segments = 1;       /* over-loaded to count active stripes */
3835
3836         for (;logical_sector < last_sector; logical_sector += STRIPE_SECTORS) {
3837                 DEFINE_WAIT(w);
3838                 int disks, data_disks;
3839                 int previous;
3840
3841         retry:
3842                 previous = 0;
3843                 disks = conf->raid_disks;
3844                 prepare_to_wait(&conf->wait_for_overlap, &w, TASK_UNINTERRUPTIBLE);
3845                 if (unlikely(conf->reshape_progress != MaxSector)) {
3846                         /* spinlock is needed as reshape_progress may be
3847                          * 64bit on a 32bit platform, and so it might be
3848                          * possible to see a half-updated value
3849                          * Ofcourse reshape_progress could change after
3850                          * the lock is dropped, so once we get a reference
3851                          * to the stripe that we think it is, we will have
3852                          * to check again.
3853                          */
3854                         spin_lock_irq(&conf->device_lock);
3855                         if (mddev->delta_disks < 0
3856                             ? logical_sector < conf->reshape_progress
3857                             : logical_sector >= conf->reshape_progress) {
3858                                 disks = conf->previous_raid_disks;
3859                                 previous = 1;
3860                         } else {
3861                                 if (mddev->delta_disks < 0
3862                                     ? logical_sector < conf->reshape_safe
3863                                     : logical_sector >= conf->reshape_safe) {
3864                                         spin_unlock_irq(&conf->device_lock);
3865                                         schedule();
3866                                         goto retry;
3867                                 }
3868                         }
3869                         spin_unlock_irq(&conf->device_lock);
3870                 }
3871                 data_disks = disks - conf->max_degraded;
3872
3873                 new_sector = raid5_compute_sector(conf, logical_sector,
3874                                                   previous,
3875                                                   &dd_idx, NULL);
3876                 pr_debug("raid5: make_request, sector %llu logical %llu\n",
3877                         (unsigned long long)new_sector, 
3878                         (unsigned long long)logical_sector);
3879
3880                 sh = get_active_stripe(conf, new_sector, previous,
3881                                        (bi->bi_rw&RWA_MASK));
3882                 if (sh) {
3883                         if (unlikely(previous)) {
3884                                 /* expansion might have moved on while waiting for a
3885                                  * stripe, so we must do the range check again.
3886                                  * Expansion could still move past after this
3887                                  * test, but as we are holding a reference to
3888                                  * 'sh', we know that if that happens,
3889                                  *  STRIPE_EXPANDING will get set and the expansion
3890                                  * won't proceed until we finish with the stripe.
3891                                  */
3892                                 int must_retry = 0;
3893                                 spin_lock_irq(&conf->device_lock);
3894                                 if (mddev->delta_disks < 0
3895                                     ? logical_sector >= conf->reshape_progress
3896                                     : logical_sector < conf->reshape_progress)
3897                                         /* mismatch, need to try again */
3898                                         must_retry = 1;
3899                                 spin_unlock_irq(&conf->device_lock);
3900                                 if (must_retry) {
3901                                         release_stripe(sh);
3902                                         goto retry;
3903                                 }
3904                         }
3905                         /* FIXME what if we get a false positive because these
3906                          * are being updated.
3907                          */
3908                         if (logical_sector >= mddev->suspend_lo &&
3909                             logical_sector < mddev->suspend_hi) {
3910                                 release_stripe(sh);
3911                                 schedule();
3912                                 goto retry;
3913                         }
3914
3915                         if (test_bit(STRIPE_EXPANDING, &sh->state) ||
3916                             !add_stripe_bio(sh, bi, dd_idx, (bi->bi_rw&RW_MASK))) {
3917                                 /* Stripe is busy expanding or
3918                                  * add failed due to overlap.  Flush everything
3919                                  * and wait a while
3920                                  */
3921                                 raid5_unplug_device(mddev->queue);
3922                                 release_stripe(sh);
3923                                 schedule();
3924                                 goto retry;
3925                         }
3926                         finish_wait(&conf->wait_for_overlap, &w);
3927                         set_bit(STRIPE_HANDLE, &sh->state);
3928                         clear_bit(STRIPE_DELAYED, &sh->state);
3929                         release_stripe(sh);
3930                 } else {
3931                         /* cannot get stripe for read-ahead, just give-up */
3932                         clear_bit(BIO_UPTODATE, &bi->bi_flags);
3933                         finish_wait(&conf->wait_for_overlap, &w);
3934                         break;
3935                 }
3936                         
3937         }
3938         spin_lock_irq(&conf->device_lock);
3939         remaining = raid5_dec_bi_phys_segments(bi);
3940         spin_unlock_irq(&conf->device_lock);
3941         if (remaining == 0) {
3942
3943                 if ( rw == WRITE )
3944                         md_write_end(mddev);
3945
3946                 bio_endio(bi, 0);
3947         }
3948         return 0;
3949 }
3950
3951 static sector_t raid5_size(mddev_t *mddev, sector_t sectors, int raid_disks);
3952
3953 static sector_t reshape_request(mddev_t *mddev, sector_t sector_nr, int *skipped)
3954 {
3955         /* reshaping is quite different to recovery/resync so it is
3956          * handled quite separately ... here.
3957          *
3958          * On each call to sync_request, we gather one chunk worth of
3959          * destination stripes and flag them as expanding.
3960          * Then we find all the source stripes and request reads.
3961          * As the reads complete, handle_stripe will copy the data
3962          * into the destination stripe and release that stripe.
3963          */
3964         raid5_conf_t *conf = (raid5_conf_t *) mddev->private;
3965         struct stripe_head *sh;
3966         sector_t first_sector, last_sector;
3967         int raid_disks = conf->previous_raid_disks;
3968         int data_disks = raid_disks - conf->max_degraded;
3969         int new_data_disks = conf->raid_disks - conf->max_degraded;
3970         int i;
3971         int dd_idx;
3972         sector_t writepos, readpos, safepos;
3973         sector_t stripe_addr;
3974         int reshape_sectors;
3975         struct list_head stripes;
3976
3977         if (sector_nr == 0) {
3978                 /* If restarting in the middle, skip the initial sectors */
3979                 if (mddev->delta_disks < 0 &&
3980                     conf->reshape_progress < raid5_size(mddev, 0, 0)) {
3981                         sector_nr = raid5_size(mddev, 0, 0)
3982                                 - conf->reshape_progress;
3983                 } else if (mddev->delta_disks > 0 &&
3984                            conf->reshape_progress > 0)
3985                         sector_nr = conf->reshape_progress;
3986                 sector_div(sector_nr, new_data_disks);
3987                 if (sector_nr) {
3988                         *skipped = 1;
3989                         return sector_nr;
3990                 }
3991         }
3992
3993         /* We need to process a full chunk at a time.
3994          * If old and new chunk sizes differ, we need to process the
3995          * largest of these
3996          */
3997         if (mddev->new_chunk > mddev->chunk_size)
3998                 reshape_sectors = mddev->new_chunk / 512;
3999         else
4000                 reshape_sectors = mddev->chunk_size / 512;
4001
4002         /* we update the metadata when there is more than 3Meg
4003          * in the block range (that is rather arbitrary, should
4004          * probably be time based) or when the data about to be
4005          * copied would over-write the source of the data at
4006          * the front of the range.
4007          * i.e. one new_stripe along from reshape_progress new_maps
4008          * to after where reshape_safe old_maps to
4009          */
4010         writepos = conf->reshape_progress;
4011         sector_div(writepos, new_data_disks);
4012         readpos = conf->reshape_progress;
4013         sector_div(readpos, data_disks);
4014         safepos = conf->reshape_safe;
4015         sector_div(safepos, data_disks);
4016         if (mddev->delta_disks < 0) {
4017                 writepos -= reshape_sectors;
4018                 readpos += reshape_sectors;
4019                 safepos += reshape_sectors;
4020         } else {
4021                 writepos += reshape_sectors;
4022                 readpos -= reshape_sectors;
4023                 safepos -= reshape_sectors;
4024         }
4025
4026         /* 'writepos' is the most advanced device address we might write.
4027          * 'readpos' is the least advanced device address we might read.
4028          * 'safepos' is the least address recorded in the metadata as having
4029          *     been reshaped.
4030          * If 'readpos' is behind 'writepos', then there is no way that we can
4031          * ensure safety in the face of a crash - that must be done by userspace
4032          * making a backup of the data.  So in that case there is no particular
4033          * rush to update metadata.
4034          * Otherwise if 'safepos' is behind 'writepos', then we really need to
4035          * update the metadata to advance 'safepos' to match 'readpos' so that
4036          * we can be safe in the event of a crash.
4037          * So we insist on updating metadata if safepos is behind writepos and
4038          * readpos is beyond writepos.
4039          * In any case, update the metadata every 10 seconds.
4040          * Maybe that number should be configurable, but I'm not sure it is
4041          * worth it.... maybe it could be a multiple of safemode_delay???
4042          */
4043         if ((mddev->delta_disks < 0
4044              ? (safepos > writepos && readpos < writepos)
4045              : (safepos < writepos && readpos > writepos)) ||
4046             time_after(jiffies, conf->reshape_checkpoint + 10*HZ)) {
4047                 /* Cannot proceed until we've updated the superblock... */
4048                 wait_event(conf->wait_for_overlap,
4049                            atomic_read(&conf->reshape_stripes)==0);
4050                 mddev->reshape_position = conf->reshape_progress;
4051                 conf->reshape_checkpoint = jiffies;
4052                 set_bit(MD_CHANGE_DEVS, &mddev->flags);
4053                 md_wakeup_thread(mddev->thread);
4054                 wait_event(mddev->sb_wait, mddev->flags == 0 ||
4055                            kthread_should_stop());
4056                 spin_lock_irq(&conf->device_lock);
4057                 conf->reshape_safe = mddev->reshape_position;
4058                 spin_unlock_irq(&conf->device_lock);
4059                 wake_up(&conf->wait_for_overlap);
4060         }
4061
4062         if (mddev->delta_disks < 0) {
4063                 BUG_ON(conf->reshape_progress == 0);
4064                 stripe_addr = writepos;
4065                 BUG_ON((mddev->dev_sectors &
4066                         ~((sector_t)reshape_sectors - 1))
4067                        - reshape_sectors - stripe_addr
4068                        != sector_nr);
4069         } else {
4070                 BUG_ON(writepos != sector_nr + reshape_sectors);
4071                 stripe_addr = sector_nr;
4072         }
4073         INIT_LIST_HEAD(&stripes);
4074         for (i = 0; i < reshape_sectors; i += STRIPE_SECTORS) {
4075                 int j;
4076                 int skipped = 0;
4077                 sh = get_active_stripe(conf, stripe_addr+i, 0, 0);
4078                 set_bit(STRIPE_EXPANDING, &sh->state);
4079                 atomic_inc(&conf->reshape_stripes);
4080                 /* If any of this stripe is beyond the end of the old
4081                  * array, then we need to zero those blocks
4082                  */
4083                 for (j=sh->disks; j--;) {
4084                         sector_t s;
4085                         if (j == sh->pd_idx)
4086                                 continue;
4087                         if (conf->level == 6 &&
4088                             j == sh->qd_idx)
4089                                 continue;
4090                         s = compute_blocknr(sh, j, 0);
4091                         if (s < raid5_size(mddev, 0, 0)) {
4092                                 skipped = 1;
4093                                 continue;
4094                         }
4095                         memset(page_address(sh->dev[j].page), 0, STRIPE_SIZE);
4096                         set_bit(R5_Expanded, &sh->dev[j].flags);
4097                         set_bit(R5_UPTODATE, &sh->dev[j].flags);
4098                 }
4099                 if (!skipped) {
4100                         set_bit(STRIPE_EXPAND_READY, &sh->state);
4101                         set_bit(STRIPE_HANDLE, &sh->state);
4102                 }
4103                 list_add(&sh->lru, &stripes);
4104         }
4105         spin_lock_irq(&conf->device_lock);
4106         if (mddev->delta_disks < 0)
4107                 conf->reshape_progress -= reshape_sectors * new_data_disks;
4108         else
4109                 conf->reshape_progress += reshape_sectors * new_data_disks;
4110         spin_unlock_irq(&conf->device_lock);
4111         /* Ok, those stripe are ready. We can start scheduling
4112          * reads on the source stripes.
4113          * The source stripes are determined by mapping the first and last
4114          * block on the destination stripes.
4115          */
4116         first_sector =
4117                 raid5_compute_sector(conf, stripe_addr*(new_data_disks),
4118                                      1, &dd_idx, NULL);
4119         last_sector =
4120                 raid5_compute_sector(conf, ((stripe_addr+conf->chunk_size/512)
4121                                             *(new_data_disks) - 1),
4122                                      1, &dd_idx, NULL);
4123         if (last_sector >= mddev->dev_sectors)
4124                 last_sector = mddev->dev_sectors - 1;
4125         while (first_sector <= last_sector) {
4126                 sh = get_active_stripe(conf, first_sector, 1, 0);
4127                 set_bit(STRIPE_EXPAND_SOURCE, &sh->state);
4128                 set_bit(STRIPE_HANDLE, &sh->state);
4129                 release_stripe(sh);
4130                 first_sector += STRIPE_SECTORS;
4131         }
4132         /* Now that the sources are clearly marked, we can release
4133          * the destination stripes
4134          */
4135         while (!list_empty(&stripes)) {
4136                 sh = list_entry(stripes.next, struct stripe_head, lru);
4137                 list_del_init(&sh->lru);
4138                 release_stripe(sh);
4139         }
4140         /* If this takes us to the resync_max point where we have to pause,
4141          * then we need to write out the superblock.
4142          */
4143         sector_nr += reshape_sectors;
4144         if (sector_nr >= mddev->resync_max) {
4145                 /* Cannot proceed until we've updated the superblock... */
4146                 wait_event(conf->wait_for_overlap,
4147                            atomic_read(&conf->reshape_stripes) == 0);
4148                 mddev->reshape_position = conf->reshape_progress;
4149                 conf->reshape_checkpoint = jiffies;
4150                 set_bit(MD_CHANGE_DEVS, &mddev->flags);
4151                 md_wakeup_thread(mddev->thread);
4152                 wait_event(mddev->sb_wait,
4153                            !test_bit(MD_CHANGE_DEVS, &mddev->flags)
4154                            || kthread_should_stop());
4155                 spin_lock_irq(&conf->device_lock);
4156                 conf->reshape_safe = mddev->reshape_position;
4157                 spin_unlock_irq(&conf->device_lock);
4158                 wake_up(&conf->wait_for_overlap);
4159         }
4160         return reshape_sectors;
4161 }
4162
4163 /* FIXME go_faster isn't used */
4164 static inline sector_t sync_request(mddev_t *mddev, sector_t sector_nr, int *skipped, int go_faster)
4165 {
4166         raid5_conf_t *conf = (raid5_conf_t *) mddev->private;
4167         struct stripe_head *sh;
4168         sector_t max_sector = mddev->dev_sectors;
4169         int sync_blocks;
4170         int still_degraded = 0;
4171         int i;
4172
4173         if (sector_nr >= max_sector) {
4174                 /* just being told to finish up .. nothing much to do */
4175                 unplug_slaves(mddev);
4176
4177                 if (test_bit(MD_RECOVERY_RESHAPE, &mddev->recovery)) {
4178                         end_reshape(conf);
4179                         return 0;
4180                 }
4181
4182                 if (mddev->curr_resync < max_sector) /* aborted */
4183                         bitmap_end_sync(mddev->bitmap, mddev->curr_resync,
4184                                         &sync_blocks, 1);
4185                 else /* completed sync */
4186                         conf->fullsync = 0;
4187                 bitmap_close_sync(mddev->bitmap);
4188
4189                 return 0;
4190         }
4191
4192         if (test_bit(MD_RECOVERY_RESHAPE, &mddev->recovery))
4193                 return reshape_request(mddev, sector_nr, skipped);
4194
4195         /* No need to check resync_max as we never do more than one
4196          * stripe, and as resync_max will always be on a chunk boundary,
4197          * if the check in md_do_sync didn't fire, there is no chance
4198          * of overstepping resync_max here
4199          */
4200
4201         /* if there is too many failed drives and we are trying
4202          * to resync, then assert that we are finished, because there is
4203          * nothing we can do.
4204          */
4205         if (mddev->degraded >= conf->max_degraded &&
4206             test_bit(MD_RECOVERY_SYNC, &mddev->recovery)) {
4207                 sector_t rv = mddev->dev_sectors - sector_nr;
4208                 *skipped = 1;
4209                 return rv;
4210         }
4211         if (!bitmap_start_sync(mddev->bitmap, sector_nr, &sync_blocks, 1) &&
4212             !test_bit(MD_RECOVERY_REQUESTED, &mddev->recovery) &&
4213             !conf->fullsync && sync_blocks >= STRIPE_SECTORS) {
4214                 /* we can skip this block, and probably more */
4215                 sync_blocks /= STRIPE_SECTORS;
4216                 *skipped = 1;
4217                 return sync_blocks * STRIPE_SECTORS; /* keep things rounded to whole stripes */
4218         }
4219
4220
4221         bitmap_cond_end_sync(mddev->bitmap, sector_nr);
4222
4223         sh = get_active_stripe(conf, sector_nr, 0, 1);
4224         if (sh == NULL) {
4225                 sh = get_active_stripe(conf, sector_nr, 0, 0);
4226                 /* make sure we don't swamp the stripe cache if someone else
4227                  * is trying to get access
4228                  */
4229                 schedule_timeout_uninterruptible(1);
4230         }
4231         /* Need to check if array will still be degraded after recovery/resync
4232          * We don't need to check the 'failed' flag as when that gets set,
4233          * recovery aborts.
4234          */
4235         for (i=0; i<mddev->raid_disks; i++)
4236                 if (conf->disks[i].rdev == NULL)
4237                         still_degraded = 1;
4238
4239         bitmap_start_sync(mddev->bitmap, sector_nr, &sync_blocks, still_degraded);
4240
4241         spin_lock(&sh->lock);
4242         set_bit(STRIPE_SYNCING, &sh->state);
4243         clear_bit(STRIPE_INSYNC, &sh->state);
4244         spin_unlock(&sh->lock);
4245
4246         /* wait for any blocked device to be handled */
4247         while (unlikely(!handle_stripe(sh)))
4248                 ;
4249         release_stripe(sh);
4250
4251         return STRIPE_SECTORS;
4252 }
4253
4254 static int  retry_aligned_read(raid5_conf_t *conf, struct bio *raid_bio)
4255 {
4256         /* We may not be able to submit a whole bio at once as there
4257          * may not be enough stripe_heads available.
4258          * We cannot pre-allocate enough stripe_heads as we may need
4259          * more than exist in the cache (if we allow ever large chunks).
4260          * So we do one stripe head at a time and record in
4261          * ->bi_hw_segments how many have been done.
4262          *
4263          * We *know* that this entire raid_bio is in one chunk, so
4264          * it will be only one 'dd_idx' and only need one call to raid5_compute_sector.
4265          */
4266         struct stripe_head *sh;
4267         int dd_idx;
4268         sector_t sector, logical_sector, last_sector;
4269         int scnt = 0;
4270         int remaining;
4271         int handled = 0;
4272
4273         logical_sector = raid_bio->bi_sector & ~((sector_t)STRIPE_SECTORS-1);
4274         sector = raid5_compute_sector(conf, logical_sector,
4275                                       0, &dd_idx, NULL);
4276         last_sector = raid_bio->bi_sector + (raid_bio->bi_size>>9);
4277
4278         for (; logical_sector < last_sector;
4279              logical_sector += STRIPE_SECTORS,
4280                      sector += STRIPE_SECTORS,
4281                      scnt++) {
4282
4283                 if (scnt < raid5_bi_hw_segments(raid_bio))
4284                         /* already done this stripe */
4285                         continue;
4286
4287                 sh = get_active_stripe(conf, sector, 0, 1);
4288
4289                 if (!sh) {
4290                         /* failed to get a stripe - must wait */
4291                         raid5_set_bi_hw_segments(raid_bio, scnt);
4292                         conf->retry_read_aligned = raid_bio;
4293                         return handled;
4294                 }
4295
4296                 set_bit(R5_ReadError, &sh->dev[dd_idx].flags);
4297                 if (!add_stripe_bio(sh, raid_bio, dd_idx, 0)) {
4298                         release_stripe(sh);
4299                         raid5_set_bi_hw_segments(raid_bio, scnt);
4300                         conf->retry_read_aligned = raid_bio;
4301                         return handled;
4302                 }
4303
4304                 handle_stripe(sh);
4305                 release_stripe(sh);
4306                 handled++;
4307         }
4308         spin_lock_irq(&conf->device_lock);
4309         remaining = raid5_dec_bi_phys_segments(raid_bio);
4310         spin_unlock_irq(&conf->device_lock);
4311         if (remaining == 0)
4312                 bio_endio(raid_bio, 0);
4313         if (atomic_dec_and_test(&conf->active_aligned_reads))
4314                 wake_up(&conf->wait_for_stripe);
4315         return handled;
4316 }
4317
4318 #ifdef CONFIG_MULTICORE_RAID456
4319 static void __process_stripe(void *param, async_cookie_t cookie)
4320 {
4321         struct stripe_head *sh = param;
4322
4323         handle_stripe(sh);
4324         release_stripe(sh);
4325 }
4326
4327 static void process_stripe(struct stripe_head *sh, struct list_head *domain)
4328 {
4329         async_schedule_domain(__process_stripe, sh, domain);
4330 }
4331
4332 static void synchronize_stripe_processing(struct list_head *domain)
4333 {
4334         async_synchronize_full_domain(domain);
4335 }
4336 #else
4337 static void process_stripe(struct stripe_head *sh, struct list_head *domain)
4338 {
4339         handle_stripe(sh);
4340         release_stripe(sh);
4341         cond_resched();
4342 }
4343
4344 static void synchronize_stripe_processing(struct list_head *domain)
4345 {
4346 }
4347 #endif
4348
4349
4350 /*
4351  * This is our raid5 kernel thread.
4352  *
4353  * We scan the hash table for stripes which can be handled now.
4354  * During the scan, completed stripes are saved for us by the interrupt
4355  * handler, so that they will not have to wait for our next wakeup.
4356  */
4357 static void raid5d(mddev_t *mddev)
4358 {
4359         struct stripe_head *sh;
4360         raid5_conf_t *conf = mddev_to_conf(mddev);
4361         int handled;
4362         LIST_HEAD(raid_domain);
4363
4364         pr_debug("+++ raid5d active\n");
4365
4366         md_check_recovery(mddev);
4367
4368         handled = 0;
4369         spin_lock_irq(&conf->device_lock);
4370         while (1) {
4371                 struct bio *bio;
4372
4373                 if (conf->seq_flush != conf->seq_write) {
4374                         int seq = conf->seq_flush;
4375                         spin_unlock_irq(&conf->device_lock);
4376                         bitmap_unplug(mddev->bitmap);
4377                         spin_lock_irq(&conf->device_lock);
4378                         conf->seq_write = seq;
4379                         activate_bit_delay(conf);
4380                 }
4381
4382                 while ((bio = remove_bio_from_retry(conf))) {
4383                         int ok;
4384                         spin_unlock_irq(&conf->device_lock);
4385                         ok = retry_aligned_read(conf, bio);
4386                         spin_lock_irq(&conf->device_lock);
4387                         if (!ok)
4388                                 break;
4389                         handled++;
4390                 }
4391
4392                 sh = __get_priority_stripe(conf);
4393
4394                 if (!sh)
4395                         break;
4396                 spin_unlock_irq(&conf->device_lock);
4397                 
4398                 handled++;
4399                 process_stripe(sh, &raid_domain);
4400
4401                 spin_lock_irq(&conf->device_lock);
4402         }
4403         pr_debug("%d stripes handled\n", handled);
4404
4405         spin_unlock_irq(&conf->device_lock);
4406
4407         synchronize_stripe_processing(&raid_domain);
4408         async_tx_issue_pending_all();
4409         unplug_slaves(mddev);
4410
4411         pr_debug("--- raid5d inactive\n");
4412 }
4413
4414 static ssize_t
4415 raid5_show_stripe_cache_size(mddev_t *mddev, char *page)
4416 {
4417         raid5_conf_t *conf = mddev_to_conf(mddev);
4418         if (conf)
4419                 return sprintf(page, "%d\n", conf->max_nr_stripes);
4420         else
4421                 return 0;
4422 }
4423
4424 static ssize_t
4425 raid5_store_stripe_cache_size(mddev_t *mddev, const char *page, size_t len)
4426 {
4427         raid5_conf_t *conf = mddev_to_conf(mddev);
4428         unsigned long new;
4429         int err;
4430
4431         if (len >= PAGE_SIZE)
4432                 return -EINVAL;
4433         if (!conf)
4434                 return -ENODEV;
4435
4436         if (strict_strtoul(page, 10, &new))
4437                 return -EINVAL;
4438         if (new <= 16 || new > 32768)
4439                 return -EINVAL;
4440         while (new < conf->max_nr_stripes) {
4441                 if (drop_one_stripe(conf))
4442                         conf->max_nr_stripes--;
4443                 else
4444                         break;
4445         }
4446         err = md_allow_write(mddev);
4447         if (err)
4448                 return err;
4449         while (new > conf->max_nr_stripes) {
4450                 if (grow_one_stripe(conf))
4451                         conf->max_nr_stripes++;
4452                 else break;
4453         }
4454         return len;
4455 }
4456
4457 static struct md_sysfs_entry
4458 raid5_stripecache_size = __ATTR(stripe_cache_size, S_IRUGO | S_IWUSR,
4459                                 raid5_show_stripe_cache_size,
4460                                 raid5_store_stripe_cache_size);
4461
4462 static ssize_t
4463 raid5_show_preread_threshold(mddev_t *mddev, char *page)
4464 {
4465         raid5_conf_t *conf = mddev_to_conf(mddev);
4466         if (conf)
4467                 return sprintf(page, "%d\n", conf->bypass_threshold);
4468         else
4469                 return 0;
4470 }
4471
4472 static ssize_t
4473 raid5_store_preread_threshold(mddev_t *mddev, const char *page, size_t len)
4474 {
4475         raid5_conf_t *conf = mddev_to_conf(mddev);
4476         unsigned long new;
4477         if (len >= PAGE_SIZE)
4478                 return -EINVAL;
4479         if (!conf)
4480                 return -ENODEV;
4481
4482         if (strict_strtoul(page, 10, &new))
4483                 return -EINVAL;
4484         if (new > conf->max_nr_stripes)
4485                 return -EINVAL;
4486         conf->bypass_threshold = new;
4487         return len;
4488 }
4489
4490 static struct md_sysfs_entry
4491 raid5_preread_bypass_threshold = __ATTR(preread_bypass_threshold,
4492                                         S_IRUGO | S_IWUSR,
4493                                         raid5_show_preread_threshold,
4494                                         raid5_store_preread_threshold);
4495
4496 static ssize_t
4497 stripe_cache_active_show(mddev_t *mddev, char *page)
4498 {
4499         raid5_conf_t *conf = mddev_to_conf(mddev);
4500         if (conf)
4501                 return sprintf(page, "%d\n", atomic_read(&conf->active_stripes));
4502         else
4503                 return 0;
4504 }
4505
4506 static struct md_sysfs_entry
4507 raid5_stripecache_active = __ATTR_RO(stripe_cache_active);
4508
4509 static struct attribute *raid5_attrs[] =  {
4510         &raid5_stripecache_size.attr,
4511         &raid5_stripecache_active.attr,
4512         &raid5_preread_bypass_threshold.attr,
4513         NULL,
4514 };
4515 static struct attribute_group raid5_attrs_group = {
4516         .name = NULL,
4517         .attrs = raid5_attrs,
4518 };
4519
4520 static sector_t
4521 raid5_size(mddev_t *mddev, sector_t sectors, int raid_disks)
4522 {
4523         raid5_conf_t *conf = mddev_to_conf(mddev);
4524
4525         if (!sectors)
4526                 sectors = mddev->dev_sectors;
4527         if (!raid_disks) {
4528                 /* size is defined by the smallest of previous and new size */
4529                 if (conf->raid_disks < conf->previous_raid_disks)
4530                         raid_disks = conf->raid_disks;
4531                 else
4532                         raid_disks = conf->previous_raid_disks;
4533         }
4534
4535         sectors &= ~((sector_t)mddev->chunk_size/512 - 1);
4536         sectors &= ~((sector_t)mddev->new_chunk/512 - 1);
4537         return sectors * (raid_disks - conf->max_degraded);
4538 }
4539
4540 static void raid5_free_percpu(raid5_conf_t *conf)
4541 {
4542         struct raid5_percpu *percpu;
4543         unsigned long cpu;
4544
4545         if (!conf->percpu)
4546                 return;
4547
4548         get_online_cpus();
4549         for_each_possible_cpu(cpu) {
4550                 percpu = per_cpu_ptr(conf->percpu, cpu);
4551                 safe_put_page(percpu->spare_page);
4552                 kfree(percpu->scribble);
4553         }
4554 #ifdef CONFIG_HOTPLUG_CPU
4555         unregister_cpu_notifier(&conf->cpu_notify);
4556 #endif
4557         put_online_cpus();
4558
4559         free_percpu(conf->percpu);
4560 }
4561
4562 static void free_conf(raid5_conf_t *conf)
4563 {
4564         shrink_stripes(conf);
4565         raid5_free_percpu(conf);
4566         kfree(conf->disks);
4567         kfree(conf->stripe_hashtbl);
4568         kfree(conf);
4569 }
4570
4571 #ifdef CONFIG_HOTPLUG_CPU
4572 static int raid456_cpu_notify(struct notifier_block *nfb, unsigned long action,
4573                               void *hcpu)
4574 {
4575         raid5_conf_t *conf = container_of(nfb, raid5_conf_t, cpu_notify);
4576         long cpu = (long)hcpu;
4577         struct raid5_percpu *percpu = per_cpu_ptr(conf->percpu, cpu);
4578
4579         switch (action) {
4580         case CPU_UP_PREPARE:
4581         case CPU_UP_PREPARE_FROZEN:
4582                 if (conf->level == 6 && !percpu->spare_page)
4583                         percpu->spare_page = alloc_page(GFP_KERNEL);
4584                 if (!percpu->scribble)
4585                         percpu->scribble = kmalloc(conf->scribble_len, GFP_KERNEL);
4586
4587                 if (!percpu->scribble ||
4588                     (conf->level == 6 && !percpu->spare_page)) {
4589                         safe_put_page(percpu->spare_page);
4590                         kfree(percpu->scribble);
4591                         pr_err("%s: failed memory allocation for cpu%ld\n",
4592                                __func__, cpu);
4593                         return NOTIFY_BAD;
4594                 }
4595                 break;
4596         case CPU_DEAD:
4597         case CPU_DEAD_FROZEN:
4598                 safe_put_page(percpu->spare_page);
4599                 kfree(percpu->scribble);
4600                 percpu->spare_page = NULL;
4601                 percpu->scribble = NULL;
4602                 break;
4603         default:
4604                 break;
4605         }
4606         return NOTIFY_OK;
4607 }
4608 #endif
4609
4610 static int raid5_alloc_percpu(raid5_conf_t *conf)
4611 {
4612         unsigned long cpu;
4613         struct page *spare_page;
4614         struct raid5_percpu *allcpus;
4615         void *scribble;
4616         int err;
4617
4618         allcpus = alloc_percpu(struct raid5_percpu);
4619         if (!allcpus)
4620                 return -ENOMEM;
4621         conf->percpu = allcpus;
4622
4623         get_online_cpus();
4624         err = 0;
4625         for_each_present_cpu(cpu) {
4626                 if (conf->level == 6) {
4627                         spare_page = alloc_page(GFP_KERNEL);
4628                         if (!spare_page) {
4629                                 err = -ENOMEM;
4630                                 break;
4631                         }
4632                         per_cpu_ptr(conf->percpu, cpu)->spare_page = spare_page;
4633                 }
4634                 scribble = kmalloc(scribble_len(conf->raid_disks), GFP_KERNEL);
4635                 if (!scribble) {
4636                         err = -ENOMEM;
4637                         break;
4638                 }
4639                 per_cpu_ptr(conf->percpu, cpu)->scribble = scribble;
4640         }
4641 #ifdef CONFIG_HOTPLUG_CPU
4642         conf->cpu_notify.notifier_call = raid456_cpu_notify;
4643         conf->cpu_notify.priority = 0;
4644         if (err == 0)
4645                 err = register_cpu_notifier(&conf->cpu_notify);
4646 #endif
4647         put_online_cpus();
4648
4649         return err;
4650 }
4651
4652 static raid5_conf_t *setup_conf(mddev_t *mddev)
4653 {
4654         raid5_conf_t *conf;
4655         int raid_disk, memory;
4656         mdk_rdev_t *rdev;
4657         struct disk_info *disk;
4658
4659         if (mddev->new_level != 5
4660             && mddev->new_level != 4
4661             && mddev->new_level != 6) {
4662                 printk(KERN_ERR "raid5: %s: raid level not set to 4/5/6 (%d)\n",
4663                        mdname(mddev), mddev->new_level);
4664                 return ERR_PTR(-EIO);
4665         }
4666         if ((mddev->new_level == 5
4667              && !algorithm_valid_raid5(mddev->new_layout)) ||
4668             (mddev->new_level == 6
4669              && !algorithm_valid_raid6(mddev->new_layout))) {
4670                 printk(KERN_ERR "raid5: %s: layout %d not supported\n",
4671                        mdname(mddev), mddev->new_layout);
4672                 return ERR_PTR(-EIO);
4673         }
4674         if (mddev->new_level == 6 && mddev->raid_disks < 4) {
4675                 printk(KERN_ERR "raid6: not enough configured devices for %s (%d, minimum 4)\n",
4676                        mdname(mddev), mddev->raid_disks);
4677                 return ERR_PTR(-EINVAL);
4678         }
4679
4680         if (!mddev->new_chunk || mddev->new_chunk % PAGE_SIZE) {
4681                 printk(KERN_ERR "raid5: invalid chunk size %d for %s\n",
4682                         mddev->new_chunk, mdname(mddev));
4683                 return ERR_PTR(-EINVAL);
4684         }
4685
4686         conf = kzalloc(sizeof(raid5_conf_t), GFP_KERNEL);
4687         if (conf == NULL)
4688                 goto abort;
4689
4690         conf->raid_disks = mddev->raid_disks;
4691         conf->scribble_len = scribble_len(conf->raid_disks);
4692         if (mddev->reshape_position == MaxSector)
4693                 conf->previous_raid_disks = mddev->raid_disks;
4694         else
4695                 conf->previous_raid_disks = mddev->raid_disks - mddev->delta_disks;
4696
4697         conf->disks = kzalloc(conf->raid_disks * sizeof(struct disk_info),
4698                               GFP_KERNEL);
4699         if (!conf->disks)
4700                 goto abort;
4701
4702         conf->mddev = mddev;
4703
4704         if ((conf->stripe_hashtbl = kzalloc(PAGE_SIZE, GFP_KERNEL)) == NULL)
4705                 goto abort;
4706
4707         conf->level = mddev->new_level;
4708         if (raid5_alloc_percpu(conf) != 0)
4709                 goto abort;
4710
4711         spin_lock_init(&conf->device_lock);
4712         init_waitqueue_head(&conf->wait_for_stripe);
4713         init_waitqueue_head(&conf->wait_for_overlap);
4714         INIT_LIST_HEAD(&conf->handle_list);
4715         INIT_LIST_HEAD(&conf->hold_list);
4716         INIT_LIST_HEAD(&conf->delayed_list);
4717         INIT_LIST_HEAD(&conf->bitmap_list);
4718         INIT_LIST_HEAD(&conf->inactive_list);
4719         atomic_set(&conf->active_stripes, 0);
4720         atomic_set(&conf->preread_active_stripes, 0);
4721         atomic_set(&conf->active_aligned_reads, 0);
4722         conf->bypass_threshold = BYPASS_THRESHOLD;
4723
4724         pr_debug("raid5: run(%s) called.\n", mdname(mddev));
4725
4726         list_for_each_entry(rdev, &mddev->disks, same_set) {
4727                 raid_disk = rdev->raid_disk;
4728                 if (raid_disk >= conf->raid_disks
4729                     || raid_disk < 0)
4730                         continue;
4731                 disk = conf->disks + raid_disk;
4732
4733                 disk->rdev = rdev;
4734
4735                 if (test_bit(In_sync, &rdev->flags)) {
4736                         char b[BDEVNAME_SIZE];
4737                         printk(KERN_INFO "raid5: device %s operational as raid"
4738                                 " disk %d\n", bdevname(rdev->bdev,b),
4739                                 raid_disk);
4740                 } else
4741                         /* Cannot rely on bitmap to complete recovery */
4742                         conf->fullsync = 1;
4743         }
4744
4745         conf->chunk_size = mddev->new_chunk;
4746         if (conf->level == 6)
4747                 conf->max_degraded = 2;
4748         else
4749                 conf->max_degraded = 1;
4750         conf->algorithm = mddev->new_layout;
4751         conf->max_nr_stripes = NR_STRIPES;
4752         conf->reshape_progress = mddev->reshape_position;
4753         if (conf->reshape_progress != MaxSector) {
4754                 conf->prev_chunk = mddev->chunk_size;
4755                 conf->prev_algo = mddev->layout;
4756         }
4757
4758         memory = conf->max_nr_stripes * (sizeof(struct stripe_head) +
4759                  conf->raid_disks * ((sizeof(struct bio) + PAGE_SIZE))) / 1024;
4760         if (grow_stripes(conf, conf->max_nr_stripes)) {
4761                 printk(KERN_ERR
4762                         "raid5: couldn't allocate %dkB for buffers\n", memory);
4763                 goto abort;
4764         } else
4765                 printk(KERN_INFO "raid5: allocated %dkB for %s\n",
4766                         memory, mdname(mddev));
4767
4768         conf->thread = md_register_thread(raid5d, mddev, "%s_raid5");
4769         if (!conf->thread) {
4770                 printk(KERN_ERR
4771                        "raid5: couldn't allocate thread for %s\n",
4772                        mdname(mddev));
4773                 goto abort;
4774         }
4775
4776         return conf;
4777
4778  abort:
4779         if (conf) {
4780                 free_conf(conf);
4781                 return ERR_PTR(-EIO);
4782         } else
4783                 return ERR_PTR(-ENOMEM);
4784 }
4785
4786 static int run(mddev_t *mddev)
4787 {
4788         raid5_conf_t *conf;
4789         int working_disks = 0;
4790         mdk_rdev_t *rdev;
4791
4792         if (mddev->reshape_position != MaxSector) {
4793                 /* Check that we can continue the reshape.
4794                  * Currently only disks can change, it must
4795                  * increase, and we must be past the point where
4796                  * a stripe over-writes itself
4797                  */
4798                 sector_t here_new, here_old;
4799                 int old_disks;
4800                 int max_degraded = (mddev->level == 6 ? 2 : 1);
4801
4802                 if (mddev->new_level != mddev->level) {
4803                         printk(KERN_ERR "raid5: %s: unsupported reshape "
4804                                "required - aborting.\n",
4805                                mdname(mddev));
4806                         return -EINVAL;
4807                 }
4808                 old_disks = mddev->raid_disks - mddev->delta_disks;
4809                 /* reshape_position must be on a new-stripe boundary, and one
4810                  * further up in new geometry must map after here in old
4811                  * geometry.
4812                  */
4813                 here_new = mddev->reshape_position;
4814                 if (sector_div(here_new, (mddev->new_chunk>>9)*
4815                                (mddev->raid_disks - max_degraded))) {
4816                         printk(KERN_ERR "raid5: reshape_position not "
4817                                "on a stripe boundary\n");
4818                         return -EINVAL;
4819                 }
4820                 /* here_new is the stripe we will write to */
4821                 here_old = mddev->reshape_position;
4822                 sector_div(here_old, (mddev->chunk_size>>9)*
4823                            (old_disks-max_degraded));
4824                 /* here_old is the first stripe that we might need to read
4825                  * from */
4826                 if (here_new >= here_old) {
4827                         /* Reading from the same stripe as writing to - bad */
4828                         printk(KERN_ERR "raid5: reshape_position too early for "
4829                                "auto-recovery - aborting.\n");
4830                         return -EINVAL;
4831                 }
4832                 printk(KERN_INFO "raid5: reshape will continue\n");
4833                 /* OK, we should be able to continue; */
4834         } else {
4835                 BUG_ON(mddev->level != mddev->new_level);
4836                 BUG_ON(mddev->layout != mddev->new_layout);
4837                 BUG_ON(mddev->chunk_size != mddev->new_chunk);
4838                 BUG_ON(mddev->delta_disks != 0);
4839         }
4840
4841         if (mddev->private == NULL)
4842                 conf = setup_conf(mddev);
4843         else
4844                 conf = mddev->private;
4845
4846         if (IS_ERR(conf))
4847                 return PTR_ERR(conf);
4848
4849         mddev->thread = conf->thread;
4850         conf->thread = NULL;
4851         mddev->private = conf;
4852
4853         /*
4854          * 0 for a fully functional array, 1 or 2 for a degraded array.
4855          */
4856         list_for_each_entry(rdev, &mddev->disks, same_set)
4857                 if (rdev->raid_disk >= 0 &&
4858                     test_bit(In_sync, &rdev->flags))
4859                         working_disks++;
4860
4861         mddev->degraded = conf->raid_disks - working_disks;
4862
4863         if (mddev->degraded > conf->max_degraded) {
4864                 printk(KERN_ERR "raid5: not enough operational devices for %s"
4865                         " (%d/%d failed)\n",
4866                         mdname(mddev), mddev->degraded, conf->raid_disks);
4867                 goto abort;
4868         }
4869
4870         /* device size must be a multiple of chunk size */
4871         mddev->dev_sectors &= ~(mddev->chunk_size / 512 - 1);
4872         mddev->resync_max_sectors = mddev->dev_sectors;
4873
4874         if (mddev->degraded > 0 &&
4875             mddev->recovery_cp != MaxSector) {
4876                 if (mddev->ok_start_degraded)
4877                         printk(KERN_WARNING
4878                                "raid5: starting dirty degraded array: %s"
4879                                "- data corruption possible.\n",
4880                                mdname(mddev));
4881                 else {
4882                         printk(KERN_ERR
4883                                "raid5: cannot start dirty degraded array for %s\n",
4884                                mdname(mddev));
4885                         goto abort;
4886                 }
4887         }
4888
4889         if (mddev->degraded == 0)
4890                 printk("raid5: raid level %d set %s active with %d out of %d"
4891                        " devices, algorithm %d\n", conf->level, mdname(mddev),
4892                        mddev->raid_disks-mddev->degraded, mddev->raid_disks,
4893                        mddev->new_layout);
4894         else
4895                 printk(KERN_ALERT "raid5: raid level %d set %s active with %d"
4896                         " out of %d devices, algorithm %d\n", conf->level,
4897                         mdname(mddev), mddev->raid_disks - mddev->degraded,
4898                         mddev->raid_disks, mddev->new_layout);
4899
4900         print_raid5_conf(conf);
4901
4902         if (conf->reshape_progress != MaxSector) {
4903                 printk("...ok start reshape thread\n");
4904                 conf->reshape_safe = conf->reshape_progress;
4905                 atomic_set(&conf->reshape_stripes, 0);
4906                 clear_bit(MD_RECOVERY_SYNC, &mddev->recovery);
4907                 clear_bit(MD_RECOVERY_CHECK, &mddev->recovery);
4908                 set_bit(MD_RECOVERY_RESHAPE, &mddev->recovery);
4909                 set_bit(MD_RECOVERY_RUNNING, &mddev->recovery);
4910                 mddev->sync_thread = md_register_thread(md_do_sync, mddev,
4911                                                         "%s_reshape");
4912         }
4913
4914         /* read-ahead size must cover two whole stripes, which is
4915          * 2 * (datadisks) * chunksize where 'n' is the number of raid devices
4916          */
4917         {
4918                 int data_disks = conf->previous_raid_disks - conf->max_degraded;
4919                 int stripe = data_disks *
4920                         (mddev->chunk_size / PAGE_SIZE);
4921                 if (mddev->queue->backing_dev_info.ra_pages < 2 * stripe)
4922                         mddev->queue->backing_dev_info.ra_pages = 2 * stripe;
4923         }
4924
4925         /* Ok, everything is just fine now */
4926         if (sysfs_create_group(&mddev->kobj, &raid5_attrs_group))
4927                 printk(KERN_WARNING
4928                        "raid5: failed to create sysfs attributes for %s\n",
4929                        mdname(mddev));
4930
4931         mddev->queue->queue_lock = &conf->device_lock;
4932
4933         mddev->queue->unplug_fn = raid5_unplug_device;
4934         mddev->queue->backing_dev_info.congested_data = mddev;
4935         mddev->queue->backing_dev_info.congested_fn = raid5_congested;
4936
4937         md_set_array_sectors(mddev, raid5_size(mddev, 0, 0));
4938
4939         blk_queue_merge_bvec(mddev->queue, raid5_mergeable_bvec);
4940
4941         return 0;
4942 abort:
4943         md_unregister_thread(mddev->thread);
4944         mddev->thread = NULL;
4945         if (conf) {
4946                 print_raid5_conf(conf);
4947                 free_conf(conf);
4948         }
4949         mddev->private = NULL;
4950         printk(KERN_ALERT "raid5: failed to run raid set %s\n", mdname(mddev));
4951         return -EIO;
4952 }
4953
4954
4955
4956 static int stop(mddev_t *mddev)
4957 {
4958         raid5_conf_t *conf = (raid5_conf_t *) mddev->private;
4959
4960         md_unregister_thread(mddev->thread);
4961         mddev->thread = NULL;
4962         mddev->queue->backing_dev_info.congested_fn = NULL;
4963         blk_sync_queue(mddev->queue); /* the unplug fn references 'conf'*/
4964         sysfs_remove_group(&mddev->kobj, &raid5_attrs_group);
4965         free_conf(conf);
4966         mddev->private = NULL;
4967         return 0;
4968 }
4969
4970 #ifdef DEBUG
4971 static void print_sh(struct seq_file *seq, struct stripe_head *sh)
4972 {
4973         int i;
4974
4975         seq_printf(seq, "sh %llu, pd_idx %d, state %ld.\n",
4976                    (unsigned long long)sh->sector, sh->pd_idx, sh->state);
4977         seq_printf(seq, "sh %llu,  count %d.\n",
4978                    (unsigned long long)sh->sector, atomic_read(&sh->count));
4979         seq_printf(seq, "sh %llu, ", (unsigned long long)sh->sector);
4980         for (i = 0; i < sh->disks; i++) {
4981                 seq_printf(seq, "(cache%d: %p %ld) ",
4982                            i, sh->dev[i].page, sh->dev[i].flags);
4983         }
4984         seq_printf(seq, "\n");
4985 }
4986
4987 static void printall(struct seq_file *seq, raid5_conf_t *conf)
4988 {
4989         struct stripe_head *sh;
4990         struct hlist_node *hn;
4991         int i;
4992
4993         spin_lock_irq(&conf->device_lock);
4994         for (i = 0; i < NR_HASH; i++) {
4995                 hlist_for_each_entry(sh, hn, &conf->stripe_hashtbl[i], hash) {
4996                         if (sh->raid_conf != conf)
4997                                 continue;
4998                         print_sh(seq, sh);
4999                 }
5000         }
5001         spin_unlock_irq(&conf->device_lock);
5002 }
5003 #endif
5004
5005 static void status(struct seq_file *seq, mddev_t *mddev)
5006 {
5007         raid5_conf_t *conf = (raid5_conf_t *) mddev->private;
5008         int i;
5009
5010         seq_printf (seq, " level %d, %dk chunk, algorithm %d", mddev->level, mddev->chunk_size >> 10, mddev->layout);
5011         seq_printf (seq, " [%d/%d] [", conf->raid_disks, conf->raid_disks - mddev->degraded);
5012         for (i = 0; i < conf->raid_disks; i++)
5013                 seq_printf (seq, "%s",
5014                                conf->disks[i].rdev &&
5015                                test_bit(In_sync, &conf->disks[i].rdev->flags) ? "U" : "_");
5016         seq_printf (seq, "]");
5017 #ifdef DEBUG
5018         seq_printf (seq, "\n");
5019         printall(seq, conf);
5020 #endif
5021 }
5022
5023 static void print_raid5_conf (raid5_conf_t *conf)
5024 {
5025         int i;
5026         struct disk_info *tmp;
5027
5028         printk("RAID5 conf printout:\n");
5029         if (!conf) {
5030                 printk("(conf==NULL)\n");
5031                 return;
5032         }
5033         printk(" --- rd:%d wd:%d\n", conf->raid_disks,
5034                  conf->raid_disks - conf->mddev->degraded);
5035
5036         for (i = 0; i < conf->raid_disks; i++) {
5037                 char b[BDEVNAME_SIZE];
5038                 tmp = conf->disks + i;
5039                 if (tmp->rdev)
5040                 printk(" disk %d, o:%d, dev:%s\n",
5041                         i, !test_bit(Faulty, &tmp->rdev->flags),
5042                         bdevname(tmp->rdev->bdev,b));
5043         }
5044 }
5045
5046 static int raid5_spare_active(mddev_t *mddev)
5047 {
5048         int i;
5049         raid5_conf_t *conf = mddev->private;
5050         struct disk_info *tmp;
5051
5052         for (i = 0; i < conf->raid_disks; i++) {
5053                 tmp = conf->disks + i;
5054                 if (tmp->rdev
5055                     && !test_bit(Faulty, &tmp->rdev->flags)
5056                     && !test_and_set_bit(In_sync, &tmp->rdev->flags)) {
5057                         unsigned long flags;
5058                         spin_lock_irqsave(&conf->device_lock, flags);
5059                         mddev->degraded--;
5060                         spin_unlock_irqrestore(&conf->device_lock, flags);
5061                 }
5062         }
5063         print_raid5_conf(conf);
5064         return 0;
5065 }
5066
5067 static int raid5_remove_disk(mddev_t *mddev, int number)
5068 {
5069         raid5_conf_t *conf = mddev->private;
5070         int err = 0;
5071         mdk_rdev_t *rdev;
5072         struct disk_info *p = conf->disks + number;
5073
5074         print_raid5_conf(conf);
5075         rdev = p->rdev;
5076         if (rdev) {
5077                 if (number >= conf->raid_disks &&
5078                     conf->reshape_progress == MaxSector)
5079                         clear_bit(In_sync, &rdev->flags);
5080
5081                 if (test_bit(In_sync, &rdev->flags) ||
5082                     atomic_read(&rdev->nr_pending)) {
5083                         err = -EBUSY;
5084                         goto abort;
5085                 }
5086                 /* Only remove non-faulty devices if recovery
5087                  * isn't possible.
5088                  */
5089                 if (!test_bit(Faulty, &rdev->flags) &&
5090                     mddev->degraded <= conf->max_degraded &&
5091                     number < conf->raid_disks) {
5092                         err = -EBUSY;
5093                         goto abort;
5094                 }
5095                 p->rdev = NULL;
5096                 synchronize_rcu();
5097                 if (atomic_read(&rdev->nr_pending)) {
5098                         /* lost the race, try later */
5099                         err = -EBUSY;
5100                         p->rdev = rdev;
5101                 }
5102         }
5103 abort:
5104
5105         print_raid5_conf(conf);
5106         return err;
5107 }
5108
5109 static int raid5_add_disk(mddev_t *mddev, mdk_rdev_t *rdev)
5110 {
5111         raid5_conf_t *conf = mddev->private;
5112         int err = -EEXIST;
5113         int disk;
5114         struct disk_info *p;
5115         int first = 0;
5116         int last = conf->raid_disks - 1;
5117
5118         if (mddev->degraded > conf->max_degraded)
5119                 /* no point adding a device */
5120                 return -EINVAL;
5121
5122         if (rdev->raid_disk >= 0)
5123                 first = last = rdev->raid_disk;
5124
5125         /*
5126          * find the disk ... but prefer rdev->saved_raid_disk
5127          * if possible.
5128          */
5129         if (rdev->saved_raid_disk >= 0 &&
5130             rdev->saved_raid_disk >= first &&
5131             conf->disks[rdev->saved_raid_disk].rdev == NULL)
5132                 disk = rdev->saved_raid_disk;
5133         else
5134                 disk = first;
5135         for ( ; disk <= last ; disk++)
5136                 if ((p=conf->disks + disk)->rdev == NULL) {
5137                         clear_bit(In_sync, &rdev->flags);
5138                         rdev->raid_disk = disk;
5139                         err = 0;
5140                         if (rdev->saved_raid_disk != disk)
5141                                 conf->fullsync = 1;
5142                         rcu_assign_pointer(p->rdev, rdev);
5143                         break;
5144                 }
5145         print_raid5_conf(conf);
5146         return err;
5147 }
5148
5149 static int raid5_resize(mddev_t *mddev, sector_t sectors)
5150 {
5151         /* no resync is happening, and there is enough space
5152          * on all devices, so we can resize.
5153          * We need to make sure resync covers any new space.
5154          * If the array is shrinking we should possibly wait until
5155          * any io in the removed space completes, but it hardly seems
5156          * worth it.
5157          */
5158         sectors &= ~((sector_t)mddev->chunk_size/512 - 1);
5159         md_set_array_sectors(mddev, raid5_size(mddev, sectors,
5160                                                mddev->raid_disks));
5161         if (mddev->array_sectors >
5162             raid5_size(mddev, sectors, mddev->raid_disks))
5163                 return -EINVAL;
5164         set_capacity(mddev->gendisk, mddev->array_sectors);
5165         mddev->changed = 1;
5166         if (sectors > mddev->dev_sectors && mddev->recovery_cp == MaxSector) {
5167                 mddev->recovery_cp = mddev->dev_sectors;
5168                 set_bit(MD_RECOVERY_NEEDED, &mddev->recovery);
5169         }
5170         mddev->dev_sectors = sectors;
5171         mddev->resync_max_sectors = sectors;
5172         return 0;
5173 }
5174
5175 static int raid5_check_reshape(mddev_t *mddev)
5176 {
5177         raid5_conf_t *conf = mddev_to_conf(mddev);
5178
5179         if (mddev->delta_disks == 0 &&
5180             mddev->new_layout == mddev->layout &&
5181             mddev->new_chunk == mddev->chunk_size)
5182                 return -EINVAL; /* nothing to do */
5183         if (mddev->bitmap)
5184                 /* Cannot grow a bitmap yet */
5185                 return -EBUSY;
5186         if (mddev->degraded > conf->max_degraded)
5187                 return -EINVAL;
5188         if (mddev->delta_disks < 0) {
5189                 /* We might be able to shrink, but the devices must
5190                  * be made bigger first.
5191                  * For raid6, 4 is the minimum size.
5192                  * Otherwise 2 is the minimum
5193                  */
5194                 int min = 2;
5195                 if (mddev->level == 6)
5196                         min = 4;
5197                 if (mddev->raid_disks + mddev->delta_disks < min)
5198                         return -EINVAL;
5199         }
5200
5201         /* Can only proceed if there are plenty of stripe_heads.
5202          * We need a minimum of one full stripe,, and for sensible progress
5203          * it is best to have about 4 times that.
5204          * If we require 4 times, then the default 256 4K stripe_heads will
5205          * allow for chunk sizes up to 256K, which is probably OK.
5206          * If the chunk size is greater, user-space should request more
5207          * stripe_heads first.
5208          */
5209         if ((mddev->chunk_size / STRIPE_SIZE) * 4 > conf->max_nr_stripes ||
5210             (mddev->new_chunk / STRIPE_SIZE) * 4 > conf->max_nr_stripes) {
5211                 printk(KERN_WARNING "raid5: reshape: not enough stripes.  Needed %lu\n",
5212                        (max(mddev->chunk_size, mddev->new_chunk)
5213                         / STRIPE_SIZE)*4);
5214                 return -ENOSPC;
5215         }
5216
5217         return resize_stripes(conf, conf->raid_disks + mddev->delta_disks);
5218 }
5219
5220 static int raid5_start_reshape(mddev_t *mddev)
5221 {
5222         raid5_conf_t *conf = mddev_to_conf(mddev);
5223         mdk_rdev_t *rdev;
5224         int spares = 0;
5225         int added_devices = 0;
5226         unsigned long flags;
5227
5228         if (test_bit(MD_RECOVERY_RUNNING, &mddev->recovery))
5229                 return -EBUSY;
5230
5231         list_for_each_entry(rdev, &mddev->disks, same_set)
5232                 if (rdev->raid_disk < 0 &&
5233                     !test_bit(Faulty, &rdev->flags))
5234                         spares++;
5235
5236         if (spares - mddev->degraded < mddev->delta_disks - conf->max_degraded)
5237                 /* Not enough devices even to make a degraded array
5238                  * of that size
5239                  */
5240                 return -EINVAL;
5241
5242         /* Refuse to reduce size of the array.  Any reductions in
5243          * array size must be through explicit setting of array_size
5244          * attribute.
5245          */
5246         if (raid5_size(mddev, 0, conf->raid_disks + mddev->delta_disks)
5247             < mddev->array_sectors) {
5248                 printk(KERN_ERR "md: %s: array size must be reduced "
5249                        "before number of disks\n", mdname(mddev));
5250                 return -EINVAL;
5251         }
5252
5253         atomic_set(&conf->reshape_stripes, 0);
5254         spin_lock_irq(&conf->device_lock);
5255         conf->previous_raid_disks = conf->raid_disks;
5256         conf->raid_disks += mddev->delta_disks;
5257         conf->prev_chunk = conf->chunk_size;
5258         conf->chunk_size = mddev->new_chunk;
5259         conf->prev_algo = conf->algorithm;
5260         conf->algorithm = mddev->new_layout;
5261         if (mddev->delta_disks < 0)
5262                 conf->reshape_progress = raid5_size(mddev, 0, 0);
5263         else
5264                 conf->reshape_progress = 0;
5265         conf->reshape_safe = conf->reshape_progress;
5266         conf->generation++;
5267         spin_unlock_irq(&conf->device_lock);
5268
5269         /* Add some new drives, as many as will fit.
5270          * We know there are enough to make the newly sized array work.
5271          */
5272         list_for_each_entry(rdev, &mddev->disks, same_set)
5273                 if (rdev->raid_disk < 0 &&
5274                     !test_bit(Faulty, &rdev->flags)) {
5275                         if (raid5_add_disk(mddev, rdev) == 0) {
5276                                 char nm[20];
5277                                 set_bit(In_sync, &rdev->flags);
5278                                 added_devices++;
5279                                 rdev->recovery_offset = 0;
5280                                 sprintf(nm, "rd%d", rdev->raid_disk);
5281                                 if (sysfs_create_link(&mddev->kobj,
5282                                                       &rdev->kobj, nm))
5283                                         printk(KERN_WARNING
5284                                                "raid5: failed to create "
5285                                                " link %s for %s\n",
5286                                                nm, mdname(mddev));
5287                         } else
5288                                 break;
5289                 }
5290
5291         if (mddev->delta_disks > 0) {
5292                 spin_lock_irqsave(&conf->device_lock, flags);
5293                 mddev->degraded = (conf->raid_disks - conf->previous_raid_disks)
5294                         - added_devices;
5295                 spin_unlock_irqrestore(&conf->device_lock, flags);
5296         }
5297         mddev->raid_disks = conf->raid_disks;
5298         mddev->reshape_position = 0;
5299         set_bit(MD_CHANGE_DEVS, &mddev->flags);
5300
5301         clear_bit(MD_RECOVERY_SYNC, &mddev->recovery);
5302         clear_bit(MD_RECOVERY_CHECK, &mddev->recovery);
5303         set_bit(MD_RECOVERY_RESHAPE, &mddev->recovery);
5304         set_bit(MD_RECOVERY_RUNNING, &mddev->recovery);
5305         mddev->sync_thread = md_register_thread(md_do_sync, mddev,
5306                                                 "%s_reshape");
5307         if (!mddev->sync_thread) {
5308                 mddev->recovery = 0;
5309                 spin_lock_irq(&conf->device_lock);
5310                 mddev->raid_disks = conf->raid_disks = conf->previous_raid_disks;
5311                 conf->reshape_progress = MaxSector;
5312                 spin_unlock_irq(&conf->device_lock);
5313                 return -EAGAIN;
5314         }
5315         conf->reshape_checkpoint = jiffies;
5316         md_wakeup_thread(mddev->sync_thread);
5317         md_new_event(mddev);
5318         return 0;
5319 }
5320
5321 /* This is called from the reshape thread and should make any
5322  * changes needed in 'conf'
5323  */
5324 static void end_reshape(raid5_conf_t *conf)
5325 {
5326
5327         if (!test_bit(MD_RECOVERY_INTR, &conf->mddev->recovery)) {
5328
5329                 spin_lock_irq(&conf->device_lock);
5330                 conf->previous_raid_disks = conf->raid_disks;
5331                 conf->reshape_progress = MaxSector;
5332                 spin_unlock_irq(&conf->device_lock);
5333                 wake_up(&conf->wait_for_overlap);
5334
5335                 /* read-ahead size must cover two whole stripes, which is
5336                  * 2 * (datadisks) * chunksize where 'n' is the number of raid devices
5337                  */
5338                 {
5339                         int data_disks = conf->raid_disks - conf->max_degraded;
5340                         int stripe = data_disks * (conf->chunk_size
5341                                                    / PAGE_SIZE);
5342                         if (conf->mddev->queue->backing_dev_info.ra_pages < 2 * stripe)
5343                                 conf->mddev->queue->backing_dev_info.ra_pages = 2 * stripe;
5344                 }
5345         }
5346 }
5347
5348 /* This is called from the raid5d thread with mddev_lock held.
5349  * It makes config changes to the device.
5350  */
5351 static void raid5_finish_reshape(mddev_t *mddev)
5352 {
5353         struct block_device *bdev;
5354         raid5_conf_t *conf = mddev_to_conf(mddev);
5355
5356         if (!test_bit(MD_RECOVERY_INTR, &mddev->recovery)) {
5357
5358                 if (mddev->delta_disks > 0) {
5359                         md_set_array_sectors(mddev, raid5_size(mddev, 0, 0));
5360                         set_capacity(mddev->gendisk, mddev->array_sectors);
5361                         mddev->changed = 1;
5362
5363                         bdev = bdget_disk(mddev->gendisk, 0);
5364                         if (bdev) {
5365                                 mutex_lock(&bdev->bd_inode->i_mutex);
5366                                 i_size_write(bdev->bd_inode,
5367                                              (loff_t)mddev->array_sectors << 9);
5368                                 mutex_unlock(&bdev->bd_inode->i_mutex);
5369                                 bdput(bdev);
5370                         }
5371                 } else {
5372                         int d;
5373                         mddev->degraded = conf->raid_disks;
5374                         for (d = 0; d < conf->raid_disks ; d++)
5375                                 if (conf->disks[d].rdev &&
5376                                     test_bit(In_sync,
5377                                              &conf->disks[d].rdev->flags))
5378                                         mddev->degraded--;
5379                         for (d = conf->raid_disks ;
5380                              d < conf->raid_disks - mddev->delta_disks;
5381                              d++)
5382                                 raid5_remove_disk(mddev, d);
5383                 }
5384                 mddev->layout = conf->algorithm;
5385                 mddev->chunk_size = conf->chunk_size;
5386                 mddev->reshape_position = MaxSector;
5387                 mddev->delta_disks = 0;
5388         }
5389 }
5390
5391 static void raid5_quiesce(mddev_t *mddev, int state)
5392 {
5393         raid5_conf_t *conf = mddev_to_conf(mddev);
5394
5395         switch(state) {
5396         case 2: /* resume for a suspend */
5397                 wake_up(&conf->wait_for_overlap);
5398                 break;
5399
5400         case 1: /* stop all writes */
5401                 spin_lock_irq(&conf->device_lock);
5402                 conf->quiesce = 1;
5403                 wait_event_lock_irq(conf->wait_for_stripe,
5404                                     atomic_read(&conf->active_stripes) == 0 &&
5405                                     atomic_read(&conf->active_aligned_reads) == 0,
5406                                     conf->device_lock, /* nothing */);
5407                 spin_unlock_irq(&conf->device_lock);
5408                 break;
5409
5410         case 0: /* re-enable writes */
5411                 spin_lock_irq(&conf->device_lock);
5412                 conf->quiesce = 0;
5413                 wake_up(&conf->wait_for_stripe);
5414                 wake_up(&conf->wait_for_overlap);
5415                 spin_unlock_irq(&conf->device_lock);
5416                 break;
5417         }
5418 }
5419
5420
5421 static void *raid5_takeover_raid1(mddev_t *mddev)
5422 {
5423         int chunksect;
5424
5425         if (mddev->raid_disks != 2 ||
5426             mddev->degraded > 1)
5427                 return ERR_PTR(-EINVAL);
5428
5429         /* Should check if there are write-behind devices? */
5430
5431         chunksect = 64*2; /* 64K by default */
5432
5433         /* The array must be an exact multiple of chunksize */
5434         while (chunksect && (mddev->array_sectors & (chunksect-1)))
5435                 chunksect >>= 1;
5436
5437         if ((chunksect<<9) < STRIPE_SIZE)
5438                 /* array size does not allow a suitable chunk size */
5439                 return ERR_PTR(-EINVAL);
5440
5441         mddev->new_level = 5;
5442         mddev->new_layout = ALGORITHM_LEFT_SYMMETRIC;
5443         mddev->new_chunk = chunksect << 9;
5444
5445         return setup_conf(mddev);
5446 }
5447
5448 static void *raid5_takeover_raid6(mddev_t *mddev)
5449 {
5450         int new_layout;
5451
5452         switch (mddev->layout) {
5453         case ALGORITHM_LEFT_ASYMMETRIC_6:
5454                 new_layout = ALGORITHM_LEFT_ASYMMETRIC;
5455                 break;
5456         case ALGORITHM_RIGHT_ASYMMETRIC_6:
5457                 new_layout = ALGORITHM_RIGHT_ASYMMETRIC;
5458                 break;
5459         case ALGORITHM_LEFT_SYMMETRIC_6:
5460                 new_layout = ALGORITHM_LEFT_SYMMETRIC;
5461                 break;
5462         case ALGORITHM_RIGHT_SYMMETRIC_6:
5463                 new_layout = ALGORITHM_RIGHT_SYMMETRIC;
5464                 break;
5465         case ALGORITHM_PARITY_0_6:
5466                 new_layout = ALGORITHM_PARITY_0;
5467                 break;
5468         case ALGORITHM_PARITY_N:
5469                 new_layout = ALGORITHM_PARITY_N;
5470                 break;
5471         default:
5472                 return ERR_PTR(-EINVAL);
5473         }
5474         mddev->new_level = 5;
5475         mddev->new_layout = new_layout;
5476         mddev->delta_disks = -1;
5477         mddev->raid_disks -= 1;
5478         return setup_conf(mddev);
5479 }
5480
5481
5482 static int raid5_reconfig(mddev_t *mddev, int new_layout, int new_chunk)
5483 {
5484         /* For a 2-drive array, the layout and chunk size can be changed
5485          * immediately as not restriping is needed.
5486          * For larger arrays we record the new value - after validation
5487          * to be used by a reshape pass.
5488          */
5489         raid5_conf_t *conf = mddev_to_conf(mddev);
5490
5491         if (new_layout >= 0 && !algorithm_valid_raid5(new_layout))
5492                 return -EINVAL;
5493         if (new_chunk > 0) {
5494                 if (new_chunk & (new_chunk-1))
5495                         /* not a power of 2 */
5496                         return -EINVAL;
5497                 if (new_chunk < PAGE_SIZE)
5498                         return -EINVAL;
5499                 if (mddev->array_sectors & ((new_chunk>>9)-1))
5500                         /* not factor of array size */
5501                         return -EINVAL;
5502         }
5503
5504         /* They look valid */
5505
5506         if (mddev->raid_disks == 2) {
5507
5508                 if (new_layout >= 0) {
5509                         conf->algorithm = new_layout;
5510                         mddev->layout = mddev->new_layout = new_layout;
5511                 }
5512                 if (new_chunk > 0) {
5513                         conf->chunk_size = new_chunk;
5514                         mddev->chunk_size = mddev->new_chunk = new_chunk;
5515                 }
5516                 set_bit(MD_CHANGE_DEVS, &mddev->flags);
5517                 md_wakeup_thread(mddev->thread);
5518         } else {
5519                 if (new_layout >= 0)
5520                         mddev->new_layout = new_layout;
5521                 if (new_chunk > 0)
5522                         mddev->new_chunk = new_chunk;
5523         }
5524         return 0;
5525 }
5526
5527 static int raid6_reconfig(mddev_t *mddev, int new_layout, int new_chunk)
5528 {
5529         if (new_layout >= 0 && !algorithm_valid_raid6(new_layout))
5530                 return -EINVAL;
5531         if (new_chunk > 0) {
5532                 if (new_chunk & (new_chunk-1))
5533                         /* not a power of 2 */
5534                         return -EINVAL;
5535                 if (new_chunk < PAGE_SIZE)
5536                         return -EINVAL;
5537                 if (mddev->array_sectors & ((new_chunk>>9)-1))
5538                         /* not factor of array size */
5539                         return -EINVAL;
5540         }
5541
5542         /* They look valid */
5543
5544         if (new_layout >= 0)
5545                 mddev->new_layout = new_layout;
5546         if (new_chunk > 0)
5547                 mddev->new_chunk = new_chunk;
5548
5549         return 0;
5550 }
5551
5552 static void *raid5_takeover(mddev_t *mddev)
5553 {
5554         /* raid5 can take over:
5555          *  raid0 - if all devices are the same - make it a raid4 layout
5556          *  raid1 - if there are two drives.  We need to know the chunk size
5557          *  raid4 - trivial - just use a raid4 layout.
5558          *  raid6 - Providing it is a *_6 layout
5559          *
5560          * For now, just do raid1
5561          */
5562
5563         if (mddev->level == 1)
5564                 return raid5_takeover_raid1(mddev);
5565         if (mddev->level == 4) {
5566                 mddev->new_layout = ALGORITHM_PARITY_N;
5567                 mddev->new_level = 5;
5568                 return setup_conf(mddev);
5569         }
5570         if (mddev->level == 6)
5571                 return raid5_takeover_raid6(mddev);
5572
5573         return ERR_PTR(-EINVAL);
5574 }
5575
5576
5577 static struct mdk_personality raid5_personality;
5578
5579 static void *raid6_takeover(mddev_t *mddev)
5580 {
5581         /* Currently can only take over a raid5.  We map the
5582          * personality to an equivalent raid6 personality
5583          * with the Q block at the end.
5584          */
5585         int new_layout;
5586
5587         if (mddev->pers != &raid5_personality)
5588                 return ERR_PTR(-EINVAL);
5589         if (mddev->degraded > 1)
5590                 return ERR_PTR(-EINVAL);
5591         if (mddev->raid_disks > 253)
5592                 return ERR_PTR(-EINVAL);
5593         if (mddev->raid_disks < 3)
5594                 return ERR_PTR(-EINVAL);
5595
5596         switch (mddev->layout) {
5597         case ALGORITHM_LEFT_ASYMMETRIC:
5598                 new_layout = ALGORITHM_LEFT_ASYMMETRIC_6;
5599                 break;
5600         case ALGORITHM_RIGHT_ASYMMETRIC:
5601                 new_layout = ALGORITHM_RIGHT_ASYMMETRIC_6;
5602                 break;
5603         case ALGORITHM_LEFT_SYMMETRIC:
5604                 new_layout = ALGORITHM_LEFT_SYMMETRIC_6;
5605                 break;
5606         case ALGORITHM_RIGHT_SYMMETRIC:
5607                 new_layout = ALGORITHM_RIGHT_SYMMETRIC_6;
5608                 break;
5609         case ALGORITHM_PARITY_0:
5610                 new_layout = ALGORITHM_PARITY_0_6;
5611                 break;
5612         case ALGORITHM_PARITY_N:
5613                 new_layout = ALGORITHM_PARITY_N;
5614                 break;
5615         default:
5616                 return ERR_PTR(-EINVAL);
5617         }
5618         mddev->new_level = 6;
5619         mddev->new_layout = new_layout;
5620         mddev->delta_disks = 1;
5621         mddev->raid_disks += 1;
5622         return setup_conf(mddev);
5623 }
5624
5625
5626 static struct mdk_personality raid6_personality =
5627 {
5628         .name           = "raid6",
5629         .level          = 6,
5630         .owner          = THIS_MODULE,
5631         .make_request   = make_request,
5632         .run            = run,
5633         .stop           = stop,
5634         .status         = status,
5635         .error_handler  = error,
5636         .hot_add_disk   = raid5_add_disk,
5637         .hot_remove_disk= raid5_remove_disk,
5638         .spare_active   = raid5_spare_active,
5639         .sync_request   = sync_request,
5640         .resize         = raid5_resize,
5641         .size           = raid5_size,
5642         .check_reshape  = raid5_check_reshape,
5643         .start_reshape  = raid5_start_reshape,
5644         .finish_reshape = raid5_finish_reshape,
5645         .quiesce        = raid5_quiesce,
5646         .takeover       = raid6_takeover,
5647         .reconfig       = raid6_reconfig,
5648 };
5649 static struct mdk_personality raid5_personality =
5650 {
5651         .name           = "raid5",
5652         .level          = 5,
5653         .owner          = THIS_MODULE,
5654         .make_request   = make_request,
5655         .run            = run,
5656         .stop           = stop,
5657         .status         = status,
5658         .error_handler  = error,
5659         .hot_add_disk   = raid5_add_disk,
5660         .hot_remove_disk= raid5_remove_disk,
5661         .spare_active   = raid5_spare_active,
5662         .sync_request   = sync_request,
5663         .resize         = raid5_resize,
5664         .size           = raid5_size,
5665         .check_reshape  = raid5_check_reshape,
5666         .start_reshape  = raid5_start_reshape,
5667         .finish_reshape = raid5_finish_reshape,
5668         .quiesce        = raid5_quiesce,
5669         .takeover       = raid5_takeover,
5670         .reconfig       = raid5_reconfig,
5671 };
5672
5673 static struct mdk_personality raid4_personality =
5674 {
5675         .name           = "raid4",
5676         .level          = 4,
5677         .owner          = THIS_MODULE,
5678         .make_request   = make_request,
5679         .run            = run,
5680         .stop           = stop,
5681         .status         = status,
5682         .error_handler  = error,
5683         .hot_add_disk   = raid5_add_disk,
5684         .hot_remove_disk= raid5_remove_disk,
5685         .spare_active   = raid5_spare_active,
5686         .sync_request   = sync_request,
5687         .resize         = raid5_resize,
5688         .size           = raid5_size,
5689         .check_reshape  = raid5_check_reshape,
5690         .start_reshape  = raid5_start_reshape,
5691         .finish_reshape = raid5_finish_reshape,
5692         .quiesce        = raid5_quiesce,
5693 };
5694
5695 static int __init raid5_init(void)
5696 {
5697         register_md_personality(&raid6_personality);
5698         register_md_personality(&raid5_personality);
5699         register_md_personality(&raid4_personality);
5700         return 0;
5701 }
5702
5703 static void raid5_exit(void)
5704 {
5705         unregister_md_personality(&raid6_personality);
5706         unregister_md_personality(&raid5_personality);
5707         unregister_md_personality(&raid4_personality);
5708 }
5709
5710 module_init(raid5_init);
5711 module_exit(raid5_exit);
5712 MODULE_LICENSE("GPL");
5713 MODULE_ALIAS("md-personality-4"); /* RAID5 */
5714 MODULE_ALIAS("md-raid5");
5715 MODULE_ALIAS("md-raid4");
5716 MODULE_ALIAS("md-level-5");
5717 MODULE_ALIAS("md-level-4");
5718 MODULE_ALIAS("md-personality-8"); /* RAID6 */
5719 MODULE_ALIAS("md-raid6");
5720 MODULE_ALIAS("md-level-6");
5721
5722 /* This used to be two separate modules, they were: */
5723 MODULE_ALIAS("raid5");
5724 MODULE_ALIAS("raid6");