/*
- * dm-snapshot.c
+ * dm-exception-store.c
  *
  * Copyright (C) 2001-2002 Sistina Software (UK) Limited.
+ * Copyright (C) 2006 Red Hat GmbH
  *
  * This file is released under the GPL.
  */
        atomic_t pending_count;
        uint32_t callback_count;
        struct commit_callback *callbacks;
+       struct dm_io_client *io_client;
 };
 
 static inline unsigned int sectors_to_pages(unsigned int sectors)
  */
 static int chunk_io(struct pstore *ps, uint32_t chunk, int rw)
 {
-       struct io_region where;
-       unsigned long bits;
-
-       where.bdev = ps->snap->cow->bdev;
-       where.sector = ps->snap->chunk_size * chunk;
-       where.count = ps->snap->chunk_size;
-
-       return dm_io_sync_vm(1, &where, rw, ps->area, &bits);
+       struct io_region where = {
+               .bdev = ps->snap->cow->bdev,
+               .sector = ps->snap->chunk_size * chunk,
+               .count = ps->snap->chunk_size,
+       };
+       struct dm_io_request io_req = {
+               .bi_rw = rw,
+               .mem.type = DM_IO_VMA,
+               .mem.ptr.vma = ps->area,
+               .client = ps->io_client,
+               .notify.fn = NULL,
+       };
+
+       return dm_io(&io_req, 1, &where, NULL);
 }
 
 /*
                chunk_size_supplied = 0;
        }
 
-       r = dm_io_get(sectors_to_pages(ps->snap->chunk_size));
-       if (r)
-               return r;
+       ps->io_client = dm_io_client_create(sectors_to_pages(ps->snap->
+                                                            chunk_size));
+       if (IS_ERR(ps->io_client))
+               return PTR_ERR(ps->io_client);
 
        r = alloc_area(ps);
        if (r)
-               goto bad1;
+               return r;
 
        r = chunk_io(ps, 0, READ);
        if (r)
-               goto bad2;
+               goto bad;
 
        dh = (struct disk_header *) ps->area;
 
        if (le32_to_cpu(dh->magic) != SNAP_MAGIC) {
                DMWARN("Invalid or corrupt snapshot");
                r = -ENXIO;
-               goto bad2;
+               goto bad;
        }
 
        *new_snapshot = 0;
               (unsigned long long)ps->snap->chunk_size);
 
        /* We had a bogus chunk_size. Fix stuff up. */
-       dm_io_put(sectors_to_pages(ps->snap->chunk_size));
        free_area(ps);
 
        ps->snap->chunk_size = chunk_size;
        ps->snap->chunk_mask = chunk_size - 1;
        ps->snap->chunk_shift = ffs(chunk_size) - 1;
 
-       r = dm_io_get(sectors_to_pages(chunk_size));
+       r = dm_io_client_resize(sectors_to_pages(ps->snap->chunk_size),
+                               ps->io_client);
        if (r)
                return r;
 
        r = alloc_area(ps);
-       if (r)
-               goto bad1;
-
-       return 0;
+       return r;
 
-bad2:
+bad:
        free_area(ps);
-bad1:
-       dm_io_put(sectors_to_pages(ps->snap->chunk_size));
        return r;
 }
 
 {
        struct pstore *ps = get_info(store);
 
-       dm_io_put(sectors_to_pages(ps->snap->chunk_size));
+       dm_io_client_destroy(ps->io_client);
        vfree(ps->callbacks);
        free_area(ps);
        kfree(ps);