drivers/md/dm-raid1.c

   1 /*
   2  * Copyright (C) 2003 Sistina Software Limited.
   3  *
   4  * This file is released under the GPL.
   5  */
   6
   7 #include "dm.h"
   8 #include "dm-bio-list.h"
   9 #include "dm-io.h"
  10 #include "dm-log.h"
  11 #include "kcopyd.h"
  12
  13 #include <linux/ctype.h>
  14 #include <linux/init.h>
  15 #include <linux/mempool.h>
  16 #include <linux/module.h>
  17 #include <linux/pagemap.h>
  18 #include <linux/slab.h>
  19 #include <linux/time.h>
  20 #include <linux/vmalloc.h>
  21 #include <linux/workqueue.h>
  22 #include <linux/log2.h>
  23 #include <linux/hardirq.h>
  24
  25 #define DM_MSG_PREFIX "raid1"
  26 #define DM_IO_PAGES 64
  27
  28 #define DM_RAID1_HANDLE_ERRORS 0x01
  29 #define errors_handled(p)       ((p)->features & DM_RAID1_HANDLE_ERRORS)
  30
  31 static DECLARE_WAIT_QUEUE_HEAD(_kmirrord_recovery_stopped);
  32
  33 /*-----------------------------------------------------------------
  34  * Region hash
  35  *
  36  * The mirror splits itself up into discrete regions.  Each
  37  * region can be in one of three states: clean, dirty,
  38  * nosync.  There is no need to put clean regions in the hash.
  39  *
  40  * In addition to being present in the hash table a region _may_
  41  * be present on one of three lists.
  42  *
  43  *   clean_regions: Regions on this list have no io pending to
  44  *   them, they are in sync, we are no longer interested in them,
  45  *   they are dull.  rh_update_states() will remove them from the
  46  *   hash table.
  47  *
  48  *   quiesced_regions: These regions have been spun down, ready
  49  *   for recovery.  rh_recovery_start() will remove regions from
  50  *   this list and hand them to kmirrord, which will schedule the
  51  *   recovery io with kcopyd.
  52  *
  53  *   recovered_regions: Regions that kcopyd has successfully
  54  *   recovered.  rh_update_states() will now schedule any delayed
  55  *   io, up the recovery_count, and remove the region from the
  56  *   hash.
  57  *
  58  * There are 2 locks:
  59  *   A rw spin lock 'hash_lock' protects just the hash table,
  60  *   this is never held in write mode from interrupt context,
  61  *   which I believe means that we only have to disable irqs when
  62  *   doing a write lock.
  63  *
  64  *   An ordinary spin lock 'region_lock' that protects the three
  65  *   lists in the region_hash, with the 'state', 'list' and
  66  *   'bhs_delayed' fields of the regions.  This is used from irq
  67  *   context, so all other uses will have to suspend local irqs.
  68  *---------------------------------------------------------------*/
  69 struct mirror_set;
  70 struct region_hash {
  71         struct mirror_set *ms;
  72         uint32_t region_size;
  73         unsigned region_shift;
  74
  75         /* holds persistent region state */
  76         struct dirty_log *log;
  77
  78         /* hash table */
  79         rwlock_t hash_lock;
  80         mempool_t *region_pool;
  81         unsigned int mask;
  82         unsigned int nr_buckets;
  83         struct list_head *buckets;
  84
  85         spinlock_t region_lock;
  86         atomic_t recovery_in_flight;
  87         struct semaphore recovery_count;
  88         struct list_head clean_regions;
  89         struct list_head quiesced_regions;
  90         struct list_head recovered_regions;
  91         struct list_head failed_recovered_regions;
  92 };
  93
  94 enum {
  95         RH_CLEAN,
  96         RH_DIRTY,
  97         RH_NOSYNC,
  98         RH_RECOVERING
  99 };
 100
 101 struct region {
 102         struct region_hash *rh; /* FIXME: can we get rid of this ? */
 103         region_t key;
 104         int state;
 105
 106         struct list_head hash_list;
 107         struct list_head list;
 108
 109         atomic_t pending;
 110         struct bio_list delayed_bios;
 111 };
 112
 113
 114 /*-----------------------------------------------------------------
 115  * Mirror set structures.
 116  *---------------------------------------------------------------*/
 117 enum dm_raid1_error {
 118         DM_RAID1_WRITE_ERROR,
 119         DM_RAID1_SYNC_ERROR,
 120         DM_RAID1_READ_ERROR
 121 };
 122
 123 struct mirror {
 124         struct mirror_set *ms;
 125         atomic_t error_count;
 126         uint32_t error_type;
 127         struct dm_dev *dev;
 128         sector_t offset;
 129 };
 130
 131 struct mirror_set {
 132         struct dm_target *ti;
 133         struct list_head list;
 134         struct region_hash rh;
 135         struct kcopyd_client *kcopyd_client;
 136         uint64_t features;
 137
 138         spinlock_t lock;        /* protects the lists */
 139         struct bio_list reads;
 140         struct bio_list writes;
 141         struct bio_list failures;
 142
 143         struct dm_io_client *io_client;
 144
 145         /* recovery */
 146         region_t nr_regions;
 147         int in_sync;
 148         int log_failure;
 149
 150         atomic_t default_mirror;        /* Default mirror */
 151
 152         struct workqueue_struct *kmirrord_wq;
 153         struct work_struct kmirrord_work;
 154         struct work_struct trigger_event;
 155
 156         unsigned int nr_mirrors;
 157         struct mirror mirror[0];
 158 };
 159
 160 /*
 161  * Conversion fns
 162  */
 163 static inline region_t bio_to_region(struct region_hash *rh, struct bio *bio)
 164 {
 165         return (bio->bi_sector - rh->ms->ti->begin) >> rh->region_shift;
 166 }
 167
 168 static inline sector_t region_to_sector(struct region_hash *rh, region_t region)
 169 {
 170         return region << rh->region_shift;
 171 }
 172
 173 static void wake(struct mirror_set *ms)
 174 {
 175         queue_work(ms->kmirrord_wq, &ms->kmirrord_work);
 176 }
 177
 178 /* FIXME move this */
 179 static void queue_bio(struct mirror_set *ms, struct bio *bio, int rw);
 180
 181 #define MIN_REGIONS 64
 182 #define MAX_RECOVERY 1
 183 static int rh_init(struct region_hash *rh, struct mirror_set *ms,
 184                    struct dirty_log *log, uint32_t region_size,
 185                    region_t nr_regions)
 186 {
 187         unsigned int nr_buckets, max_buckets;
 188         size_t i;
 189
 190         /*
 191          * Calculate a suitable number of buckets for our hash
 192          * table.
 193          */
 194         max_buckets = nr_regions >> 6;
 195         for (nr_buckets = 128u; nr_buckets < max_buckets; nr_buckets <<= 1)
 196                 ;
 197         nr_buckets >>= 1;
 198
 199         rh->ms = ms;
 200         rh->log = log;
 201         rh->region_size = region_size;
 202         rh->region_shift = ffs(region_size) - 1;
 203         rwlock_init(&rh->hash_lock);
 204         rh->mask = nr_buckets - 1;
 205         rh->nr_buckets = nr_buckets;
 206
 207         rh->buckets = vmalloc(nr_buckets * sizeof(*rh->buckets));
 208         if (!rh->buckets) {
 209                 DMERR("unable to allocate region hash memory");
 210                 return -ENOMEM;
 211         }
 212
 213         for (i = 0; i < nr_buckets; i++)
 214                 INIT_LIST_HEAD(rh->buckets + i);
 215
 216         spin_lock_init(&rh->region_lock);
 217         sema_init(&rh->recovery_count, 0);
 218         atomic_set(&rh->recovery_in_flight, 0);
 219         INIT_LIST_HEAD(&rh->clean_regions);
 220         INIT_LIST_HEAD(&rh->quiesced_regions);
 221         INIT_LIST_HEAD(&rh->recovered_regions);
 222         INIT_LIST_HEAD(&rh->failed_recovered_regions);
 223
 224         rh->region_pool = mempool_create_kmalloc_pool(MIN_REGIONS,
 225                                                       sizeof(struct region));
 226         if (!rh->region_pool) {
 227                 vfree(rh->buckets);
 228                 rh->buckets = NULL;
 229                 return -ENOMEM;
 230         }
 231
 232         return 0;
 233 }
 234
 235 static void rh_exit(struct region_hash *rh)
 236 {
 237         unsigned int h;
 238         struct region *reg, *nreg;
 239
 240         BUG_ON(!list_empty(&rh->quiesced_regions));
 241         for (h = 0; h < rh->nr_buckets; h++) {
 242                 list_for_each_entry_safe(reg, nreg, rh->buckets + h, hash_list) {
 243                         BUG_ON(atomic_read(&reg->pending));
 244                         mempool_free(reg, rh->region_pool);
 245                 }
 246         }
 247
 248         if (rh->log)
 249                 dm_destroy_dirty_log(rh->log);
 250         if (rh->region_pool)
 251                 mempool_destroy(rh->region_pool);
 252         vfree(rh->buckets);
 253 }
 254
 255 #define RH_HASH_MULT 2654435387U
 256
 257 static inline unsigned int rh_hash(struct region_hash *rh, region_t region)
 258 {
 259         return (unsigned int) ((region * RH_HASH_MULT) >> 12) & rh->mask;
 260 }
 261
 262 static struct region *__rh_lookup(struct region_hash *rh, region_t region)
 263 {
 264         struct region *reg;
 265
 266         list_for_each_entry (reg, rh->buckets + rh_hash(rh, region), hash_list)
 267                 if (reg->key == region)
 268                         return reg;
 269
 270         return NULL;
 271 }
 272
 273 static void __rh_insert(struct region_hash *rh, struct region *reg)
 274 {
 275         unsigned int h = rh_hash(rh, reg->key);
 276         list_add(&reg->hash_list, rh->buckets + h);
 277 }
 278
 279 static struct region *__rh_alloc(struct region_hash *rh, region_t region)
 280 {
 281         struct region *reg, *nreg;
 282
 283         read_unlock(&rh->hash_lock);
 284         nreg = mempool_alloc(rh->region_pool, GFP_ATOMIC);
 285         if (unlikely(!nreg))
 286                 nreg = kmalloc(sizeof(struct region), GFP_NOIO);
 287         nreg->state = rh->log->type->in_sync(rh->log, region, 1) ?
 288                 RH_CLEAN : RH_NOSYNC;
 289         nreg->rh = rh;
 290         nreg->key = region;
 291
 292         INIT_LIST_HEAD(&nreg->list);
 293
 294         atomic_set(&nreg->pending, 0);
 295         bio_list_init(&nreg->delayed_bios);
 296         write_lock_irq(&rh->hash_lock);
 297
 298         reg = __rh_lookup(rh, region);
 299         if (reg)
 300                 /* we lost the race */
 301                 mempool_free(nreg, rh->region_pool);
 302
 303         else {
 304                 __rh_insert(rh, nreg);
 305                 if (nreg->state == RH_CLEAN) {
 306                         spin_lock(&rh->region_lock);
 307                         list_add(&nreg->list, &rh->clean_regions);
 308                         spin_unlock(&rh->region_lock);
 309                 }
 310                 reg = nreg;
 311         }
 312         write_unlock_irq(&rh->hash_lock);
 313         read_lock(&rh->hash_lock);
 314
 315         return reg;
 316 }
 317
 318 static inline struct region *__rh_find(struct region_hash *rh, region_t region)
 319 {
 320         struct region *reg;
 321
 322         reg = __rh_lookup(rh, region);
 323         if (!reg)
 324                 reg = __rh_alloc(rh, region);
 325
 326         return reg;
 327 }
 328
 329 static int rh_state(struct region_hash *rh, region_t region, int may_block)
 330 {
 331         int r;
 332         struct region *reg;
 333
 334         read_lock(&rh->hash_lock);
 335         reg = __rh_lookup(rh, region);
 336         read_unlock(&rh->hash_lock);
 337
 338         if (reg)
 339                 return reg->state;
 340
 341         /*
 342          * The region wasn't in the hash, so we fall back to the
 343          * dirty log.
 344          */
 345         r = rh->log->type->in_sync(rh->log, region, may_block);
 346
 347         /*
 348          * Any error from the dirty log (eg. -EWOULDBLOCK) gets
 349          * taken as a RH_NOSYNC
 350          */
 351         return r == 1 ? RH_CLEAN : RH_NOSYNC;
 352 }
 353
 354 static inline int rh_in_sync(struct region_hash *rh,
 355                              region_t region, int may_block)
 356 {
 357         int state = rh_state(rh, region, may_block);
 358         return state == RH_CLEAN || state == RH_DIRTY;
 359 }
 360
 361 static void dispatch_bios(struct mirror_set *ms, struct bio_list *bio_list)
 362 {
 363         struct bio *bio;
 364
 365         while ((bio = bio_list_pop(bio_list))) {
 366                 queue_bio(ms, bio, WRITE);
 367         }
 368 }
 369
 370 static void complete_resync_work(struct region *reg, int success)
 371 {
 372         struct region_hash *rh = reg->rh;
 373
 374         rh->log->type->set_region_sync(rh->log, reg->key, success);
 375         dispatch_bios(rh->ms, &reg->delayed_bios);
 376         if (atomic_dec_and_test(&rh->recovery_in_flight))
 377                 wake_up_all(&_kmirrord_recovery_stopped);
 378         up(&rh->recovery_count);
 379 }
 380
 381 static void rh_update_states(struct region_hash *rh)
 382 {
 383         struct region *reg, *next;
 384
 385         LIST_HEAD(clean);
 386         LIST_HEAD(recovered);
 387         LIST_HEAD(failed_recovered);
 388
 389         /*
 390          * Quickly grab the lists.
 391          */
 392         write_lock_irq(&rh->hash_lock);
 393         spin_lock(&rh->region_lock);
 394         if (!list_empty(&rh->clean_regions)) {
 395                 list_splice(&rh->clean_regions, &clean);
 396                 INIT_LIST_HEAD(&rh->clean_regions);
 397
 398                 list_for_each_entry(reg, &clean, list)
 399                         list_del(&reg->hash_list);
 400         }
 401
 402         if (!list_empty(&rh->recovered_regions)) {
 403                 list_splice(&rh->recovered_regions, &recovered);
 404                 INIT_LIST_HEAD(&rh->recovered_regions);
 405
 406                 list_for_each_entry (reg, &recovered, list)
 407                         list_del(&reg->hash_list);
 408         }
 409
 410         if (!list_empty(&rh->failed_recovered_regions)) {
 411                 list_splice(&rh->failed_recovered_regions, &failed_recovered);
 412                 INIT_LIST_HEAD(&rh->failed_recovered_regions);
 413
 414                 list_for_each_entry(reg, &failed_recovered, list)
 415                         list_del(&reg->hash_list);
 416         }
 417
 418         spin_unlock(&rh->region_lock);
 419         write_unlock_irq(&rh->hash_lock);
 420
 421         /*
 422          * All the regions on the recovered and clean lists have
 423          * now been pulled out of the system, so no need to do
 424          * any more locking.
 425          */
 426         list_for_each_entry_safe (reg, next, &recovered, list) {
 427                 rh->log->type->clear_region(rh->log, reg->key);
 428                 complete_resync_work(reg, 1);
 429                 mempool_free(reg, rh->region_pool);
 430         }
 431
 432         list_for_each_entry_safe(reg, next, &failed_recovered, list) {
 433                 complete_resync_work(reg, errors_handled(rh->ms) ? 0 : 1);
 434                 mempool_free(reg, rh->region_pool);
 435         }
 436
 437         list_for_each_entry_safe(reg, next, &clean, list) {
 438                 rh->log->type->clear_region(rh->log, reg->key);
 439                 mempool_free(reg, rh->region_pool);
 440         }
 441
 442         rh->log->type->flush(rh->log);
 443 }
 444
 445 static void rh_inc(struct region_hash *rh, region_t region)
 446 {
 447         struct region *reg;
 448
 449         read_lock(&rh->hash_lock);
 450         reg = __rh_find(rh, region);
 451
 452         spin_lock_irq(&rh->region_lock);
 453         atomic_inc(&reg->pending);
 454
 455         if (reg->state == RH_CLEAN) {
 456                 reg->state = RH_DIRTY;
 457                 list_del_init(&reg->list);      /* take off the clean list */
 458                 spin_unlock_irq(&rh->region_lock);
 459
 460                 rh->log->type->mark_region(rh->log, reg->key);
 461         } else
 462                 spin_unlock_irq(&rh->region_lock);
 463
 464
 465         read_unlock(&rh->hash_lock);
 466 }
 467
 468 static void rh_inc_pending(struct region_hash *rh, struct bio_list *bios)
 469 {
 470         struct bio *bio;
 471
 472         for (bio = bios->head; bio; bio = bio->bi_next)
 473                 rh_inc(rh, bio_to_region(rh, bio));
 474 }
 475
 476 static void rh_dec(struct region_hash *rh, region_t region)
 477 {
 478         unsigned long flags;
 479         struct region *reg;
 480         int should_wake = 0;
 481
 482         read_lock(&rh->hash_lock);
 483         reg = __rh_lookup(rh, region);
 484         read_unlock(&rh->hash_lock);
 485
 486         spin_lock_irqsave(&rh->region_lock, flags);
 487         if (atomic_dec_and_test(&reg->pending)) {
 488                 /*
 489                  * There is no pending I/O for this region.
 490                  * We can move the region to corresponding list for next action.
 491                  * At this point, the region is not yet connected to any list.
 492                  *
 493                  * If the state is RH_NOSYNC, the region should be kept off
 494                  * from clean list.
 495                  * The hash entry for RH_NOSYNC will remain in memory
 496                  * until the region is recovered or the map is reloaded.
 497                  */
 498
 499                 /* do nothing for RH_NOSYNC */
 500                 if (reg->state == RH_RECOVERING) {
 501                         list_add_tail(&reg->list, &rh->quiesced_regions);
 502                 } else if (reg->state == RH_DIRTY) {
 503                         reg->state = RH_CLEAN;
 504                         list_add(&reg->list, &rh->clean_regions);
 505                 }
 506                 should_wake = 1;
 507         }
 508         spin_unlock_irqrestore(&rh->region_lock, flags);
 509
 510         if (should_wake)
 511                 wake(rh->ms);
 512 }
 513
 514 /*
 515  * Starts quiescing a region in preparation for recovery.
 516  */
 517 static int __rh_recovery_prepare(struct region_hash *rh)
 518 {
 519         int r;
 520         struct region *reg;
 521         region_t region;
 522
 523         /*
 524          * Ask the dirty log what's next.
 525          */
 526         r = rh->log->type->get_resync_work(rh->log, &region);
 527         if (r <= 0)
 528                 return r;
 529
 530         /*
 531          * Get this region, and start it quiescing by setting the
 532          * recovering flag.
 533          */
 534         read_lock(&rh->hash_lock);
 535         reg = __rh_find(rh, region);
 536         read_unlock(&rh->hash_lock);
 537
 538         spin_lock_irq(&rh->region_lock);
 539         reg->state = RH_RECOVERING;
 540
 541         /* Already quiesced ? */
 542         if (atomic_read(&reg->pending))
 543                 list_del_init(&reg->list);
 544         else
 545                 list_move(&reg->list, &rh->quiesced_regions);
 546
 547         spin_unlock_irq(&rh->region_lock);
 548
 549         return 1;
 550 }
 551
 552 static void rh_recovery_prepare(struct region_hash *rh)
 553 {
 554         /* Extra reference to avoid race with rh_stop_recovery */
 555         atomic_inc(&rh->recovery_in_flight);
 556
 557         while (!down_trylock(&rh->recovery_count)) {
 558                 atomic_inc(&rh->recovery_in_flight);
 559                 if (__rh_recovery_prepare(rh) <= 0) {
 560                         atomic_dec(&rh->recovery_in_flight);
 561                         up(&rh->recovery_count);
 562                         break;
 563                 }
 564         }
 565
 566         /* Drop the extra reference */
 567         if (atomic_dec_and_test(&rh->recovery_in_flight))
 568                 wake_up_all(&_kmirrord_recovery_stopped);
 569 }
 570
 571 /*
 572  * Returns any quiesced regions.
 573  */
 574 static struct region *rh_recovery_start(struct region_hash *rh)
 575 {
 576         struct region *reg = NULL;
 577
 578         spin_lock_irq(&rh->region_lock);
 579         if (!list_empty(&rh->quiesced_regions)) {
 580                 reg = list_entry(rh->quiesced_regions.next,
 581                                  struct region, list);
 582                 list_del_init(&reg->list);      /* remove from the quiesced list */
 583         }
 584         spin_unlock_irq(&rh->region_lock);
 585
 586         return reg;
 587 }
 588
 589 static void rh_recovery_end(struct region *reg, int success)
 590 {
 591         struct region_hash *rh = reg->rh;
 592
 593         spin_lock_irq(&rh->region_lock);
 594         if (success)
 595                 list_add(&reg->list, &reg->rh->recovered_regions);
 596         else {
 597                 reg->state = RH_NOSYNC;
 598                 list_add(&reg->list, &reg->rh->failed_recovered_regions);
 599         }
 600         spin_unlock_irq(&rh->region_lock);
 601
 602         wake(rh->ms);
 603 }
 604
 605 static int rh_flush(struct region_hash *rh)
 606 {
 607         return rh->log->type->flush(rh->log);
 608 }
 609
 610 static void rh_delay(struct region_hash *rh, struct bio *bio)
 611 {
 612         struct region *reg;
 613
 614         read_lock(&rh->hash_lock);
 615         reg = __rh_find(rh, bio_to_region(rh, bio));
 616         bio_list_add(&reg->delayed_bios, bio);
 617         read_unlock(&rh->hash_lock);
 618 }
 619
 620 static void rh_stop_recovery(struct region_hash *rh)
 621 {
 622         int i;
 623
 624         /* wait for any recovering regions */
 625         for (i = 0; i < MAX_RECOVERY; i++)
 626                 down(&rh->recovery_count);
 627 }
 628
 629 static void rh_start_recovery(struct region_hash *rh)
 630 {
 631         int i;
 632
 633         for (i = 0; i < MAX_RECOVERY; i++)
 634                 up(&rh->recovery_count);
 635
 636         wake(rh->ms);
 637 }
 638
 639 /*
 640  * Every mirror should look like this one.
 641  */
 642 #define DEFAULT_MIRROR 0
 643
 644 /*
 645  * This is yucky.  We squirrel the mirror_set struct away inside
 646  * bi_next for write buffers.  This is safe since the bh
 647  * doesn't get submitted to the lower levels of block layer.
 648  */
 649 static struct mirror_set *bio_get_ms(struct bio *bio)
 650 {
 651         return (struct mirror_set *) bio->bi_next;
 652 }
 653
 654 static void bio_set_ms(struct bio *bio, struct mirror_set *ms)
 655 {
 656         bio->bi_next = (struct bio *) ms;
 657 }
 658
 659 static struct mirror *get_default_mirror(struct mirror_set *ms)
 660 {
 661         return &ms->mirror[atomic_read(&ms->default_mirror)];
 662 }
 663
 664 static void set_default_mirror(struct mirror *m)
 665 {
 666         struct mirror_set *ms = m->ms;
 667         struct mirror *m0 = &(ms->mirror[0]);
 668
 669         atomic_set(&ms->default_mirror, m - m0);
 670 }
 671
 672 /* fail_mirror
 673  * @m: mirror device to fail
 674  * @error_type: one of the enum's, DM_RAID1_*_ERROR
 675  *
 676  * If errors are being handled, record the type of
 677  * error encountered for this device.  If this type
 678  * of error has already been recorded, we can return;
 679  * otherwise, we must signal userspace by triggering
 680  * an event.  Additionally, if the device is the
 681  * primary device, we must choose a new primary, but
 682  * only if the mirror is in-sync.
 683  *
 684  * This function must not block.
 685  */
 686 static void fail_mirror(struct mirror *m, enum dm_raid1_error error_type)
 687 {
 688         struct mirror_set *ms = m->ms;
 689         struct mirror *new;
 690
 691         if (!errors_handled(ms))
 692                 return;
 693
 694         /*
 695          * error_count is used for nothing more than a
 696          * simple way to tell if a device has encountered
 697          * errors.
 698          */
 699         atomic_inc(&m->error_count);
 700
 701         if (test_and_set_bit(error_type, &m->error_type))
 702                 return;
 703
 704         if (m != get_default_mirror(ms))
 705                 goto out;
 706
 707         if (!ms->in_sync) {
 708                 /*
 709                  * Better to issue requests to same failing device
 710                  * than to risk returning corrupt data.
 711                  */
 712                 DMERR("Primary mirror (%s) failed while out-of-sync: "
 713                       "Reads may fail.", m->dev->name);
 714                 goto out;
 715         }
 716
 717         for (new = ms->mirror; new < ms->mirror + ms->nr_mirrors; new++)
 718                 if (!atomic_read(&new->error_count)) {
 719                         set_default_mirror(new);
 720                         break;
 721                 }
 722
 723         if (unlikely(new == ms->mirror + ms->nr_mirrors))
 724                 DMWARN("All sides of mirror have failed.");
 725
 726 out:
 727         schedule_work(&ms->trigger_event);
 728 }
 729
 730 /*-----------------------------------------------------------------
 731  * Recovery.
 732  *
 733  * When a mirror is first activated we may find that some regions
 734  * are in the no-sync state.  We have to recover these by
 735  * recopying from the default mirror to all the others.
 736  *---------------------------------------------------------------*/
 737 static void recovery_complete(int read_err, unsigned int write_err,
 738                               void *context)
 739 {
 740         struct region *reg = (struct region *)context;
 741         struct mirror_set *ms = reg->rh->ms;
 742         int m, bit = 0;
 743
 744         if (read_err) {
 745                 /* Read error means the failure of default mirror. */
 746                 DMERR_LIMIT("Unable to read primary mirror during recovery");
 747                 fail_mirror(get_default_mirror(ms), DM_RAID1_SYNC_ERROR);
 748         }
 749
 750         if (write_err) {
 751                 DMERR_LIMIT("Write error during recovery (error = 0x%x)",
 752                             write_err);
 753                 /*
 754                  * Bits correspond to devices (excluding default mirror).
 755                  * The default mirror cannot change during recovery.
 756                  */
 757                 for (m = 0; m < ms->nr_mirrors; m++) {
 758                         if (&ms->mirror[m] == get_default_mirror(ms))
 759                                 continue;
 760                         if (test_bit(bit, &write_err))
 761                                 fail_mirror(ms->mirror + m,
 762                                             DM_RAID1_SYNC_ERROR);
 763                         bit++;
 764                 }
 765         }
 766
 767         rh_recovery_end(reg, !(read_err || write_err));
 768 }
 769
 770 static int recover(struct mirror_set *ms, struct region *reg)
 771 {
 772         int r;
 773         unsigned int i;
 774         struct io_region from, to[KCOPYD_MAX_REGIONS], *dest;
 775         struct mirror *m;
 776         unsigned long flags = 0;
 777
 778         /* fill in the source */
 779         m = get_default_mirror(ms);
 780         from.bdev = m->dev->bdev;
 781         from.sector = m->offset + region_to_sector(reg->rh, reg->key);
 782         if (reg->key == (ms->nr_regions - 1)) {
 783                 /*
 784                  * The final region may be smaller than
 785                  * region_size.
 786                  */
 787                 from.count = ms->ti->len & (reg->rh->region_size - 1);
 788                 if (!from.count)
 789                         from.count = reg->rh->region_size;
 790         } else
 791                 from.count = reg->rh->region_size;
 792
 793         /* fill in the destinations */
 794         for (i = 0, dest = to; i < ms->nr_mirrors; i++) {
 795                 if (&ms->mirror[i] == get_default_mirror(ms))
 796                         continue;
 797
 798                 m = ms->mirror + i;
 799                 dest->bdev = m->dev->bdev;
 800                 dest->sector = m->offset + region_to_sector(reg->rh, reg->key);
 801                 dest->count = from.count;
 802                 dest++;
 803         }
 804
 805         /* hand to kcopyd */
 806         set_bit(KCOPYD_IGNORE_ERROR, &flags);
 807         r = kcopyd_copy(ms->kcopyd_client, &from, ms->nr_mirrors - 1, to, flags,
 808                         recovery_complete, reg);
 809
 810         return r;
 811 }
 812
 813 static void do_recovery(struct mirror_set *ms)
 814 {
 815         int r;
 816         struct region *reg;
 817         struct dirty_log *log = ms->rh.log;
 818
 819         /*
 820          * Start quiescing some regions.
 821          */
 822         rh_recovery_prepare(&ms->rh);
 823
 824         /*
 825          * Copy any already quiesced regions.
 826          */
 827         while ((reg = rh_recovery_start(&ms->rh))) {
 828                 r = recover(ms, reg);
 829                 if (r)
 830                         rh_recovery_end(reg, 0);
 831         }
 832
 833         /*
 834          * Update the in sync flag.
 835          */
 836         if (!ms->in_sync &&
 837             (log->type->get_sync_count(log) == ms->nr_regions)) {
 838                 /* the sync is complete */
 839                 dm_table_event(ms->ti->table);
 840                 ms->in_sync = 1;
 841         }
 842 }
 843
 844 /*-----------------------------------------------------------------
 845  * Reads
 846  *---------------------------------------------------------------*/
 847 static struct mirror *choose_mirror(struct mirror_set *ms, sector_t sector)
 848 {
 849         /* FIXME: add read balancing */
 850         return get_default_mirror(ms);
 851 }
 852
 853 /*
 854  * remap a buffer to a particular mirror.
 855  */
 856 static void map_bio(struct mirror_set *ms, struct mirror *m, struct bio *bio)
 857 {
 858         bio->bi_bdev = m->dev->bdev;
 859         bio->bi_sector = m->offset + (bio->bi_sector - ms->ti->begin);
 860 }
 861
 862 static void do_reads(struct mirror_set *ms, struct bio_list *reads)
 863 {
 864         region_t region;
 865         struct bio *bio;
 866         struct mirror *m;
 867
 868         while ((bio = bio_list_pop(reads))) {
 869                 region = bio_to_region(&ms->rh, bio);
 870
 871                 /*
 872                  * We can only read balance if the region is in sync.
 873                  */
 874                 if (rh_in_sync(&ms->rh, region, 1))
 875                         m = choose_mirror(ms, bio->bi_sector);
 876                 else
 877                         m = get_default_mirror(ms);
 878
 879                 map_bio(ms, m, bio);
 880                 generic_make_request(bio);
 881         }
 882 }
 883
 884 /*-----------------------------------------------------------------
 885  * Writes.
 886  *
 887  * We do different things with the write io depending on the
 888  * state of the region that it's in:
 889  *
 890  * SYNC:        increment pending, use kcopyd to write to *all* mirrors
 891  * RECOVERING:  delay the io until recovery completes
 892  * NOSYNC:      increment pending, just write to the default mirror
 893  *---------------------------------------------------------------*/
 894
 895 /* __bio_mark_nosync
 896  * @ms
 897  * @bio
 898  * @done
 899  * @error
 900  *
 901  * The bio was written on some mirror(s) but failed on other mirror(s).
 902  * We can successfully endio the bio but should avoid the region being
 903  * marked clean by setting the state RH_NOSYNC.
 904  *
 905  * This function is _not_ safe in interrupt context!
 906  */
 907 static void __bio_mark_nosync(struct mirror_set *ms,
 908                               struct bio *bio, unsigned done, int error)
 909 {
 910         unsigned long flags;
 911         struct region_hash *rh = &ms->rh;
 912         struct dirty_log *log = ms->rh.log;
 913         struct region *reg;
 914         region_t region = bio_to_region(rh, bio);
 915         int recovering = 0;
 916
 917         /* We must inform the log that the sync count has changed. */
 918         log->type->set_region_sync(log, region, 0);
 919         ms->in_sync = 0;
 920
 921         read_lock(&rh->hash_lock);
 922         reg = __rh_find(rh, region);
 923         read_unlock(&rh->hash_lock);
 924
 925         /* region hash entry should exist because write was in-flight */
 926         BUG_ON(!reg);
 927         BUG_ON(!list_empty(&reg->list));
 928
 929         spin_lock_irqsave(&rh->region_lock, flags);
 930         /*
 931          * Possible cases:
 932          *   1) RH_DIRTY
 933          *   2) RH_NOSYNC: was dirty, other preceeding writes failed
 934          *   3) RH_RECOVERING: flushing pending writes
 935          * Either case, the region should have not been connected to list.
 936          */
 937         recovering = (reg->state == RH_RECOVERING);
 938         reg->state = RH_NOSYNC;
 939         BUG_ON(!list_empty(&reg->list));
 940         spin_unlock_irqrestore(&rh->region_lock, flags);
 941
 942         bio_endio(bio, error);
 943         if (recovering)
 944                 complete_resync_work(reg, 0);
 945 }
 946
 947 static void write_callback(unsigned long error, void *context)
 948 {
 949         unsigned i, ret = 0;
 950         struct bio *bio = (struct bio *) context;
 951         struct mirror_set *ms;
 952         int uptodate = 0;
 953         int should_wake = 0;
 954         unsigned long flags;
 955
 956         ms = bio_get_ms(bio);
 957         bio_set_ms(bio, NULL);
 958
 959         /*
 960          * NOTE: We don't decrement the pending count here,
 961          * instead it is done by the targets endio function.
 962          * This way we handle both writes to SYNC and NOSYNC
 963          * regions with the same code.
 964          */
 965         if (likely(!error))
 966                 goto out;
 967
 968         for (i = 0; i < ms->nr_mirrors; i++)
 969                 if (test_bit(i, &error))
 970                         fail_mirror(ms->mirror + i, DM_RAID1_WRITE_ERROR);
 971                 else
 972                         uptodate = 1;
 973
 974         if (unlikely(!uptodate)) {
 975                 DMERR("All replicated volumes dead, failing I/O");
 976                 /* None of the writes succeeded, fail the I/O. */
 977                 ret = -EIO;
 978         } else if (errors_handled(ms)) {
 979                 /*
 980                  * Need to raise event.  Since raising
 981                  * events can block, we need to do it in
 982                  * the main thread.
 983                  */
 984                 spin_lock_irqsave(&ms->lock, flags);
 985                 if (!ms->failures.head)
 986                         should_wake = 1;
 987                 bio_list_add(&ms->failures, bio);
 988                 spin_unlock_irqrestore(&ms->lock, flags);
 989                 if (should_wake)
 990                         wake(ms);
 991                 return;
 992         }
 993 out:
 994         bio_endio(bio, ret);
 995 }
 996
 997 static void do_write(struct mirror_set *ms, struct bio *bio)
 998 {
 999         unsigned int i;
1000         struct io_region io[KCOPYD_MAX_REGIONS+1];
1001         struct mirror *m;
1002         struct dm_io_request io_req = {
1003                 .bi_rw = WRITE,
1004                 .mem.type = DM_IO_BVEC,
1005                 .mem.ptr.bvec = bio->bi_io_vec + bio->bi_idx,
1006                 .notify.fn = write_callback,
1007                 .notify.context = bio,
1008                 .client = ms->io_client,
1009         };
1010
1011         for (i = 0; i < ms->nr_mirrors; i++) {
1012                 m = ms->mirror + i;
1013
1014                 io[i].bdev = m->dev->bdev;
1015                 io[i].sector = m->offset + (bio->bi_sector - ms->ti->begin);
1016                 io[i].count = bio->bi_size >> 9;
1017         }
1018
1019         bio_set_ms(bio, ms);
1020
1021         (void) dm_io(&io_req, ms->nr_mirrors, io, NULL);
1022 }
1023
1024 static void do_writes(struct mirror_set *ms, struct bio_list *writes)
1025 {
1026         int state;
1027         struct bio *bio;
1028         struct bio_list sync, nosync, recover, *this_list = NULL;
1029
1030         if (!writes->head)
1031                 return;
1032
1033         /*
1034          * Classify each write.
1035          */
1036         bio_list_init(&sync);
1037         bio_list_init(&nosync);
1038         bio_list_init(&recover);
1039
1040         while ((bio = bio_list_pop(writes))) {
1041                 state = rh_state(&ms->rh, bio_to_region(&ms->rh, bio), 1);
1042                 switch (state) {
1043                 case RH_CLEAN:
1044                 case RH_DIRTY:
1045                         this_list = &sync;
1046                         break;
1047
1048                 case RH_NOSYNC:
1049                         this_list = &nosync;
1050                         break;
1051
1052                 case RH_RECOVERING:
1053                         this_list = &recover;
1054                         break;
1055                 }
1056
1057                 bio_list_add(this_list, bio);
1058         }
1059
1060         /*
1061          * Increment the pending counts for any regions that will
1062          * be written to (writes to recover regions are going to
1063          * be delayed).
1064          */
1065         rh_inc_pending(&ms->rh, &sync);
1066         rh_inc_pending(&ms->rh, &nosync);
1067         ms->log_failure = rh_flush(&ms->rh) ? 1 : 0;
1068
1069         /*
1070          * Dispatch io.
1071          */
1072         if (unlikely(ms->log_failure))
1073                 while ((bio = bio_list_pop(&sync)))
1074                         bio_endio(bio, -EIO);
1075         else while ((bio = bio_list_pop(&sync)))
1076                 do_write(ms, bio);
1077
1078         while ((bio = bio_list_pop(&recover)))
1079                 rh_delay(&ms->rh, bio);
1080
1081         while ((bio = bio_list_pop(&nosync))) {
1082                 map_bio(ms, get_default_mirror(ms), bio);
1083                 generic_make_request(bio);
1084         }
1085 }
1086
1087 static void do_failures(struct mirror_set *ms, struct bio_list *failures)
1088 {
1089         struct bio *bio;
1090
1091         if (!failures->head)
1092                 return;
1093
1094         while ((bio = bio_list_pop(failures)))
1095                 __bio_mark_nosync(ms, bio, bio->bi_size, 0);
1096 }
1097
1098 static void trigger_event(struct work_struct *work)
1099 {
1100         struct mirror_set *ms =
1101                 container_of(work, struct mirror_set, trigger_event);
1102
1103         dm_table_event(ms->ti->table);
1104 }
1105
1106 /*-----------------------------------------------------------------
1107  * kmirrord
1108  *---------------------------------------------------------------*/
1109 static int _do_mirror(struct work_struct *work)
1110 {
1111         struct mirror_set *ms =container_of(work, struct mirror_set,
1112                                             kmirrord_work);
1113         struct bio_list reads, writes, failures;
1114         unsigned long flags;
1115
1116         spin_lock_irqsave(&ms->lock, flags);
1117         reads = ms->reads;
1118         writes = ms->writes;
1119         failures = ms->failures;
1120         bio_list_init(&ms->reads);
1121         bio_list_init(&ms->writes);
1122         bio_list_init(&ms->failures);
1123         spin_unlock_irqrestore(&ms->lock, flags);
1124
1125         rh_update_states(&ms->rh);
1126         do_recovery(ms);
1127         do_reads(ms, &reads);
1128         do_writes(ms, &writes);
1129         do_failures(ms, &failures);
1130
1131         return (ms->failures.head) ? 1 : 0;
1132 }
1133
1134 static void do_mirror(struct work_struct *work)
1135 {
1136         /*
1137          * If _do_mirror returns 1, we give it
1138          * another shot.  This helps for cases like
1139          * 'suspend' where we call flush_workqueue
1140          * and expect all work to be finished.  If
1141          * a failure happens during a suspend, we
1142          * couldn't issue a 'wake' because it would
1143          * not be honored.  Therefore, we return '1'
1144          * from _do_mirror, and retry here.
1145          */
1146         while (_do_mirror(work))
1147                 schedule();
1148 }
1149
1150
1151 /*-----------------------------------------------------------------
1152  * Target functions
1153  *---------------------------------------------------------------*/
1154 static struct mirror_set *alloc_context(unsigned int nr_mirrors,
1155                                         uint32_t region_size,
1156                                         struct dm_target *ti,
1157                                         struct dirty_log *dl)
1158 {
1159         size_t len;
1160         struct mirror_set *ms = NULL;
1161
1162         if (array_too_big(sizeof(*ms), sizeof(ms->mirror[0]), nr_mirrors))
1163                 return NULL;
1164
1165         len = sizeof(*ms) + (sizeof(ms->mirror[0]) * nr_mirrors);
1166
1167         ms = kzalloc(len, GFP_KERNEL);
1168         if (!ms) {
1169                 ti->error = "Cannot allocate mirror context";
1170                 return NULL;
1171         }
1172
1173         spin_lock_init(&ms->lock);
1174
1175         ms->ti = ti;
1176         ms->nr_mirrors = nr_mirrors;
1177         ms->nr_regions = dm_sector_div_up(ti->len, region_size);
1178         ms->in_sync = 0;
1179         atomic_set(&ms->default_mirror, DEFAULT_MIRROR);
1180
1181         ms->io_client = dm_io_client_create(DM_IO_PAGES);
1182         if (IS_ERR(ms->io_client)) {
1183                 ti->error = "Error creating dm_io client";
1184                 kfree(ms);
1185                 return NULL;
1186         }
1187
1188         if (rh_init(&ms->rh, ms, dl, region_size, ms->nr_regions)) {
1189                 ti->error = "Error creating dirty region hash";
1190                 dm_io_client_destroy(ms->io_client);
1191                 kfree(ms);
1192                 return NULL;
1193         }
1194
1195         return ms;
1196 }
1197
1198 static void free_context(struct mirror_set *ms, struct dm_target *ti,
1199                          unsigned int m)
1200 {
1201         while (m--)
1202                 dm_put_device(ti, ms->mirror[m].dev);
1203
1204         dm_io_client_destroy(ms->io_client);
1205         rh_exit(&ms->rh);
1206         kfree(ms);
1207 }
1208
1209 static inline int _check_region_size(struct dm_target *ti, uint32_t size)
1210 {
1211         return !(size % (PAGE_SIZE >> 9) || !is_power_of_2(size) ||
1212                  size > ti->len);
1213 }
1214
1215 static int get_mirror(struct mirror_set *ms, struct dm_target *ti,
1216                       unsigned int mirror, char **argv)
1217 {
1218         unsigned long long offset;
1219
1220         if (sscanf(argv[1], "%llu", &offset) != 1) {
1221                 ti->error = "Invalid offset";
1222                 return -EINVAL;
1223         }
1224
1225         if (dm_get_device(ti, argv[0], offset, ti->len,
1226                           dm_table_get_mode(ti->table),
1227                           &ms->mirror[mirror].dev)) {
1228                 ti->error = "Device lookup failure";
1229                 return -ENXIO;
1230         }
1231
1232         ms->mirror[mirror].ms = ms;
1233         atomic_set(&(ms->mirror[mirror].error_count), 0);
1234         ms->mirror[mirror].error_type = 0;
1235         ms->mirror[mirror].offset = offset;
1236
1237         return 0;
1238 }
1239
1240 /*
1241  * Create dirty log: log_type #log_params <log_params>
1242  */
1243 static struct dirty_log *create_dirty_log(struct dm_target *ti,
1244                                           unsigned int argc, char **argv,
1245                                           unsigned int *args_used)
1246 {
1247         unsigned int param_count;
1248         struct dirty_log *dl;
1249
1250         if (argc < 2) {
1251                 ti->error = "Insufficient mirror log arguments";
1252                 return NULL;
1253         }
1254
1255         if (sscanf(argv[1], "%u", &param_count) != 1) {
1256                 ti->error = "Invalid mirror log argument count";
1257                 return NULL;
1258         }
1259
1260         *args_used = 2 + param_count;
1261
1262         if (argc < *args_used) {
1263                 ti->error = "Insufficient mirror log arguments";
1264                 return NULL;
1265         }
1266
1267         dl = dm_create_dirty_log(argv[0], ti, param_count, argv + 2);
1268         if (!dl) {
1269                 ti->error = "Error creating mirror dirty log";
1270                 return NULL;
1271         }
1272
1273         if (!_check_region_size(ti, dl->type->get_region_size(dl))) {
1274                 ti->error = "Invalid region size";
1275                 dm_destroy_dirty_log(dl);
1276                 return NULL;
1277         }
1278
1279         return dl;
1280 }
1281
1282 static int parse_features(struct mirror_set *ms, unsigned argc, char **argv,
1283                           unsigned *args_used)
1284 {
1285         unsigned num_features;
1286         struct dm_target *ti = ms->ti;
1287
1288         *args_used = 0;
1289
1290         if (!argc)
1291                 return 0;
1292
1293         if (sscanf(argv[0], "%u", &num_features) != 1) {
1294                 ti->error = "Invalid number of features";
1295                 return -EINVAL;
1296         }
1297
1298         argc--;
1299         argv++;
1300         (*args_used)++;
1301
1302         if (num_features > argc) {
1303                 ti->error = "Not enough arguments to support feature count";
1304                 return -EINVAL;
1305         }
1306
1307         if (!strcmp("handle_errors", argv[0]))
1308                 ms->features |= DM_RAID1_HANDLE_ERRORS;
1309         else {
1310                 ti->error = "Unrecognised feature requested";
1311                 return -EINVAL;
1312         }
1313
1314         (*args_used)++;
1315
1316         return 0;
1317 }
1318
1319 /*
1320  * Construct a mirror mapping:
1321  *
1322  * log_type #log_params <log_params>
1323  * #mirrors [mirror_path offset]{2,}
1324  * [#features <features>]
1325  *
1326  * log_type is "core" or "disk"
1327  * #log_params is between 1 and 3
1328  *
1329  * If present, features must be "handle_errors".
1330  */
1331 static int mirror_ctr(struct dm_target *ti, unsigned int argc, char **argv)
1332 {
1333         int r;
1334         unsigned int nr_mirrors, m, args_used;
1335         struct mirror_set *ms;
1336         struct dirty_log *dl;
1337
1338         dl = create_dirty_log(ti, argc, argv, &args_used);
1339         if (!dl)
1340                 return -EINVAL;
1341
1342         argv += args_used;
1343         argc -= args_used;
1344
1345         if (!argc || sscanf(argv[0], "%u", &nr_mirrors) != 1 ||
1346             nr_mirrors < 2 || nr_mirrors > KCOPYD_MAX_REGIONS + 1) {
1347                 ti->error = "Invalid number of mirrors";
1348                 dm_destroy_dirty_log(dl);
1349                 return -EINVAL;
1350         }
1351
1352         argv++, argc--;
1353
1354         if (argc < nr_mirrors * 2) {
1355                 ti->error = "Too few mirror arguments";
1356                 dm_destroy_dirty_log(dl);
1357                 return -EINVAL;
1358         }
1359
1360         ms = alloc_context(nr_mirrors, dl->type->get_region_size(dl), ti, dl);
1361         if (!ms) {
1362                 dm_destroy_dirty_log(dl);
1363                 return -ENOMEM;
1364         }
1365
1366         /* Get the mirror parameter sets */
1367         for (m = 0; m < nr_mirrors; m++) {
1368                 r = get_mirror(ms, ti, m, argv);
1369                 if (r) {
1370                         free_context(ms, ti, m);
1371                         return r;
1372                 }
1373                 argv += 2;
1374                 argc -= 2;
1375         }
1376
1377         ti->private = ms;
1378         ti->split_io = ms->rh.region_size;
1379
1380         ms->kmirrord_wq = create_singlethread_workqueue("kmirrord");
1381         if (!ms->kmirrord_wq) {
1382                 DMERR("couldn't start kmirrord");
1383                 r = -ENOMEM;
1384                 goto err_free_context;
1385         }
1386         INIT_WORK(&ms->kmirrord_work, do_mirror);
1387         INIT_WORK(&ms->trigger_event, trigger_event);
1388
1389         r = parse_features(ms, argc, argv, &args_used);
1390         if (r)
1391                 goto err_destroy_wq;
1392
1393         argv += args_used;
1394         argc -= args_used;
1395
1396         /*
1397          * Any read-balancing addition depends on the
1398          * DM_RAID1_HANDLE_ERRORS flag being present.
1399          * This is because the decision to balance depends
1400          * on the sync state of a region.  If the above
1401          * flag is not present, we ignore errors; and
1402          * the sync state may be inaccurate.
1403          */
1404
1405         if (argc) {
1406                 ti->error = "Too many mirror arguments";
1407                 r = -EINVAL;
1408                 goto err_destroy_wq;
1409         }
1410
1411         r = kcopyd_client_create(DM_IO_PAGES, &ms->kcopyd_client);
1412         if (r)
1413                 goto err_destroy_wq;
1414
1415         wake(ms);
1416         return 0;
1417
1418 err_destroy_wq:
1419         destroy_workqueue(ms->kmirrord_wq);
1420 err_free_context:
1421         free_context(ms, ti, ms->nr_mirrors);
1422         return r;
1423 }
1424
1425 static void mirror_dtr(struct dm_target *ti)
1426 {
1427         struct mirror_set *ms = (struct mirror_set *) ti->private;
1428
1429         flush_workqueue(ms->kmirrord_wq);
1430         kcopyd_client_destroy(ms->kcopyd_client);
1431         destroy_workqueue(ms->kmirrord_wq);
1432         free_context(ms, ti, ms->nr_mirrors);
1433 }
1434
1435 static void queue_bio(struct mirror_set *ms, struct bio *bio, int rw)
1436 {
1437         unsigned long flags;
1438         int should_wake = 0;
1439         struct bio_list *bl;
1440
1441         bl = (rw == WRITE) ? &ms->writes : &ms->reads;
1442         spin_lock_irqsave(&ms->lock, flags);
1443         should_wake = !(bl->head);
1444         bio_list_add(bl, bio);
1445         spin_unlock_irqrestore(&ms->lock, flags);
1446
1447         if (should_wake)
1448                 wake(ms);
1449 }
1450
1451 /*
1452  * Mirror mapping function
1453  */
1454 static int mirror_map(struct dm_target *ti, struct bio *bio,
1455                       union map_info *map_context)
1456 {
1457         int r, rw = bio_rw(bio);
1458         struct mirror *m;
1459         struct mirror_set *ms = ti->private;
1460
1461         map_context->ll = bio_to_region(&ms->rh, bio);
1462
1463         if (rw == WRITE) {
1464                 queue_bio(ms, bio, rw);
1465                 return DM_MAPIO_SUBMITTED;
1466         }
1467
1468         r = ms->rh.log->type->in_sync(ms->rh.log,
1469                                       bio_to_region(&ms->rh, bio), 0);
1470         if (r < 0 && r != -EWOULDBLOCK)
1471                 return r;
1472
1473         if (r == -EWOULDBLOCK)  /* FIXME: ugly */
1474                 r = DM_MAPIO_SUBMITTED;
1475
1476         /*
1477          * We don't want to fast track a recovery just for a read
1478          * ahead.  So we just let it silently fail.
1479          * FIXME: get rid of this.
1480          */
1481         if (!r && rw == READA)
1482                 return -EIO;
1483
1484         if (!r) {
1485                 /* Pass this io over to the daemon */
1486                 queue_bio(ms, bio, rw);
1487                 return DM_MAPIO_SUBMITTED;
1488         }
1489
1490         m = choose_mirror(ms, bio->bi_sector);
1491         if (!m)
1492                 return -EIO;
1493
1494         map_bio(ms, m, bio);
1495         return DM_MAPIO_REMAPPED;
1496 }
1497
1498 static int mirror_end_io(struct dm_target *ti, struct bio *bio,
1499                          int error, union map_info *map_context)
1500 {
1501         int rw = bio_rw(bio);
1502         struct mirror_set *ms = (struct mirror_set *) ti->private;
1503         region_t region = map_context->ll;
1504
1505         /*
1506          * We need to dec pending if this was a write.
1507          */
1508         if (rw == WRITE)
1509                 rh_dec(&ms->rh, region);
1510
1511         return 0;
1512 }
1513
1514 static void mirror_postsuspend(struct dm_target *ti)
1515 {
1516         struct mirror_set *ms = (struct mirror_set *) ti->private;
1517         struct dirty_log *log = ms->rh.log;
1518
1519         rh_stop_recovery(&ms->rh);
1520
1521         /* Wait for all I/O we generated to complete */
1522         wait_event(_kmirrord_recovery_stopped,
1523                    !atomic_read(&ms->rh.recovery_in_flight));
1524
1525         if (log->type->postsuspend && log->type->postsuspend(log))
1526                 /* FIXME: need better error handling */
1527                 DMWARN("log suspend failed");
1528 }
1529
1530 static void mirror_resume(struct dm_target *ti)
1531 {
1532         struct mirror_set *ms = (struct mirror_set *) ti->private;
1533         struct dirty_log *log = ms->rh.log;
1534         if (log->type->resume && log->type->resume(log))
1535                 /* FIXME: need better error handling */
1536                 DMWARN("log resume failed");
1537         rh_start_recovery(&ms->rh);
1538 }
1539
1540 static int mirror_status(struct dm_target *ti, status_type_t type,
1541                          char *result, unsigned int maxlen)
1542 {
1543         unsigned int m, sz = 0;
1544         struct mirror_set *ms = (struct mirror_set *) ti->private;
1545
1546         switch (type) {
1547         case STATUSTYPE_INFO:
1548                 DMEMIT("%d ", ms->nr_mirrors);
1549                 for (m = 0; m < ms->nr_mirrors; m++)
1550                         DMEMIT("%s ", ms->mirror[m].dev->name);
1551
1552                 DMEMIT("%llu/%llu 0 ",
1553                         (unsigned long long)ms->rh.log->type->
1554                                 get_sync_count(ms->rh.log),
1555                         (unsigned long long)ms->nr_regions);
1556
1557                 sz += ms->rh.log->type->status(ms->rh.log, type, result+sz, maxlen-sz);
1558
1559                 break;
1560
1561         case STATUSTYPE_TABLE:
1562                 sz = ms->rh.log->type->status(ms->rh.log, type, result, maxlen);
1563
1564                 DMEMIT("%d", ms->nr_mirrors);
1565                 for (m = 0; m < ms->nr_mirrors; m++)
1566                         DMEMIT(" %s %llu", ms->mirror[m].dev->name,
1567                                 (unsigned long long)ms->mirror[m].offset);
1568
1569                 if (ms->features & DM_RAID1_HANDLE_ERRORS)
1570                         DMEMIT(" 1 handle_errors");
1571         }
1572
1573         return 0;
1574 }
1575
1576 static struct target_type mirror_target = {
1577         .name    = "mirror",
1578         .version = {1, 0, 3},
1579         .module  = THIS_MODULE,
1580         .ctr     = mirror_ctr,
1581         .dtr     = mirror_dtr,
1582         .map     = mirror_map,
1583         .end_io  = mirror_end_io,
1584         .postsuspend = mirror_postsuspend,
1585         .resume  = mirror_resume,
1586         .status  = mirror_status,
1587 };
1588
1589 static int __init dm_mirror_init(void)
1590 {
1591         int r;
1592
1593         r = dm_dirty_log_init();
1594         if (r)
1595                 return r;
1596
1597         r = dm_register_target(&mirror_target);
1598         if (r < 0) {
1599                 DMERR("Failed to register mirror target");
1600                 dm_dirty_log_exit();
1601         }
1602
1603         return r;
1604 }
1605
1606 static void __exit dm_mirror_exit(void)
1607 {
1608         int r;
1609
1610         r = dm_unregister_target(&mirror_target);
1611         if (r < 0)
1612                 DMERR("unregister failed %d", r);
1613
1614         dm_dirty_log_exit();
1615 }
1616
1617 /* Module hooks */
1618 module_init(dm_mirror_init);
1619 module_exit(dm_mirror_exit);
1620
1621 MODULE_DESCRIPTION(DM_NAME " mirror target");
1622 MODULE_AUTHOR("Joe Thornber");
1623 MODULE_LICENSE("GPL");