From d49788b7d3c0c7eae3b0598c83d668b7cc6b6395 Mon Sep 17 00:00:00 2001
From: Jason Dillaman <dillaman@redhat.com>
Date: Wed, 18 Mar 2015 11:07:35 -0400
Subject: [PATCH] librbd: add preliminary methods for rebuilding object map

Signed-off-by: Jason Dillaman <dillaman@redhat.com>
---
 src/librbd/internal.cc | 63 ++++++++++++++++++++++++++++++++++++++++++
 src/librbd/internal.h  |  4 +++
 2 files changed, 67 insertions(+)

diff --git a/src/librbd/internal.cc b/src/librbd/internal.cc
index 9ef8c972c6d6d..900c06ffcdde5 100644
--- a/src/librbd/internal.cc
+++ b/src/librbd/internal.cc
@@ -2703,6 +2703,69 @@ reprotect_and_return_err:
     return 0;
   }
 
+  int rebuild_object_map(ImageCtx *ictx, ProgressContext &prog_ctx) {
+    CephContext *cct = ictx->cct;
+    ldout(cct, 10) << "rebuild_object_map" << dendl;
+
+    int r = ictx_check(ictx);
+    if (r < 0) {
+      return r;
+    }
+
+    {
+      RWLock::RLocker snap_locker(ictx->snap_lock);
+      if (ictx->read_only || ictx->snap_id != CEPH_NOSNAP) {
+        return -EROFS;
+      }
+    }
+
+    uint64_t request_id = ictx->async_request_seq.inc();
+    do {
+      C_SaferCond ctx;
+      {
+        RWLock::RLocker l(ictx->owner_lock);
+        while (ictx->image_watcher->is_lock_supported()) {
+          r = prepare_image_update(ictx);
+          if (r < 0) {
+            return -EROFS;
+          } else if (ictx->image_watcher->is_lock_owner()) {
+	    break;
+	  }
+
+          r = ictx->image_watcher->notify_rebuild_object_map(request_id,
+                                                             prog_ctx);
+          if (r != -ETIMEDOUT && r != -ERESTART) {
+            return r;
+          }
+          ldout(ictx->cct, 5) << "rebuild object map timed out notifying lock "
+                              << "owner" << dendl;
+        }
+
+        r = async_rebuild_object_map(ictx, &ctx, prog_ctx);
+        if (r < 0) {
+	  return r;
+        }
+      }
+
+      r = ctx.wait();
+      if (r == -ERESTART) {
+	ldout(ictx->cct, 5) << "rebuild object map interrupted: restarting"
+                            << dendl;
+      }
+    } while (r == -ERESTART);
+
+    // TODO rebuild snapshots
+
+    ldout(cct, 10) << "rebuild object map finished" << dendl;
+    return r;
+  }
+
+  int async_rebuild_object_map(ImageCtx *ictx, Context *ctx,
+                               ProgressContext &prog_ctx) {
+    // TODO rebuild image HEAD
+    return 0;
+  }
+
   int list_lockers(ImageCtx *ictx,
 		   std::list<locker_t> *lockers,
 		   bool *exclusive,
diff --git a/src/librbd/internal.h b/src/librbd/internal.h
index aa472a98af3ec..33107313f1d37 100644
--- a/src/librbd/internal.h
+++ b/src/librbd/internal.h
@@ -137,6 +137,8 @@ namespace librbd {
 		   const char *buf);
   int flatten(ImageCtx *ictx, ProgressContext &prog_ctx);
 
+  int rebuild_object_map(ImageCtx *ictx, ProgressContext &prog_ctx);
+
   /* cooperative locking */
   int list_lockers(ImageCtx *ictx,
 		   std::list<locker_t> *locks,
@@ -196,6 +198,8 @@ namespace librbd {
 		   ProgressContext &prog_ctx);
   void async_resize_helper(ImageCtx *ictx, Context *ctx, uint64_t new_size,
                            ProgressContext& prog_ctx);
+  int async_rebuild_object_map(ImageCtx *ictx, Context *ctx,
+                               ProgressContext &prog_ctx);
 
   int aio_write(ImageCtx *ictx, uint64_t off, size_t len, const char *buf,
 		AioCompletion *c, int op_flags);
-- 
2.39.5