1 files changed, 143 insertions, 0 deletions
diff --git a/src/librbd/mirror/DisableRequest.h b/src/librbd/mirror/DisableRequest.h
new file mode 100644
index 000000000..f45d1a14c
--- /dev/null
+++ b/src/librbd/mirror/DisableRequest.h
@@ -0,0 +1,143 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#ifndef CEPH_LIBRBD_MIRROR_DISABLE_REQUEST_H
+#define CEPH_LIBRBD_MIRROR_DISABLE_REQUEST_H
+
+#include "include/buffer.h"
+#include "common/ceph_mutex.h"
+#include "cls/journal/cls_journal_types.h"
+#include "cls/rbd/cls_rbd_types.h"
+#include "librbd/mirror/Types.h"
+
+#include <map>
+#include <string>
+
+class Context;
+
+namespace librbd {
+
+class ImageCtx;
+
+namespace mirror {
+
+template <typename ImageCtxT = ImageCtx>
+class DisableRequest {
+public:
+  static DisableRequest *create(ImageCtxT *image_ctx, bool force,
+                                bool remove, Context *on_finish) {
+    return new DisableRequest(image_ctx, force, remove, on_finish);
+  }
+
+  DisableRequest(ImageCtxT *image_ctx, bool force, bool remove,
+                 Context *on_finish);
+
+  void send();
+
+private:
+  /**
+   * @verbatim
+   *
+   * <start>
+   *    |
+   *    v
+   * GET_MIRROR_INFO  * * * * * * * * * * * * * * * * * * * * * * *
+   *    |                                                         *
+   *    v                                                         *
+   * IMAGE_STATE_UPDATE * * * * * * * * * * * * * * * * * * * * * *
+   *    |                                                         *
+   *    v                                                         *
+   * PROMOTE_IMAGE (skip if primary)                              *
+   *    |                                                         *
+   *    v                                                         *
+   * REFRESH_IMAGE (skip if necessary)                            *
+   *    |                                                         *
+   *    v                                                         *
+   * GET_CLIENTS <----------------------------------------\ * * * *
+   *    |     | (unregister clients)                      |       *  (on error)
+   *    |     |/----------------------------\             |       *
+   *    |     |                             |             |       *
+   *    |     |   /-----------\ (repeat     | (repeat     | (repeat
+   *    |     |   |           |  as needed) |  as needed) |  as needed)
+   *    |     v   v           |             |             |       *
+   *    |  REMOVE_SYNC_SNAP --/ * * * * * * | * * * * * * | * * * *
+   *    |     |                             |             |       *
+   *    |     v                             |             |       *
+   *    |  UNREGISTER_CLIENT ---------------/-------------/ * * * *
+   *    |                                                         *
+   *    | (no more clients                                        *
+   *    |  to unregister)                                         *
+   *    v                                                         *
+   * REMOVE_MIRROR_IMAGE  * * * * * * * * * * * * * * * * * * * * *
+   *    |         (skip if no remove)                             *
+   *    v                                                         *
+   * <finish> < * * * * * * * * * * * * * * * * * * * * * * * * * *
+   *
+   * @endverbatim
+   */
+
+  ImageCtxT *m_image_ctx;
+  bool m_force;
+  bool m_remove;
+  Context *m_on_finish;
+
+  bool m_is_primary = false;
+  cls::rbd::MirrorImage m_mirror_image;
+  PromotionState m_promotion_state = PROMOTION_STATE_NON_PRIMARY;
+  std::string m_primary_mirror_uuid;
+  std::set<cls::journal::Client> m_clients;
+  std::map<std::string, int> m_ret;
+  std::map<std::string, int> m_current_ops;
+  int m_error_result = 0;
+  mutable ceph::mutex m_lock =
+    ceph::make_mutex("mirror::DisableRequest::m_lock");
+
+  void send_get_mirror_info();
+  Context *handle_get_mirror_info(int *result);
+
+  void send_image_state_update();
+  Context *handle_image_state_update(int *result);
+
+  void send_notify_mirroring_watcher();
+  Context *handle_notify_mirroring_watcher(int *result);
+
+  void send_promote_image();
+  Context *handle_promote_image(int *result);
+
+  void send_refresh_image();
+  Context* handle_refresh_image(int* result);
+
+  void clean_mirror_state();
+
+  void send_get_clients();
+  Context *handle_get_clients(int *result);
+
+  void remove_mirror_snapshots();
+
+  void send_remove_snap(const std::string &client_id,
+                        const cls::rbd::SnapshotNamespace &snap_namespace,
+			const std::string &snap_name);
+  Context *handle_remove_snap(int *result, const std::string &client_id);
+
+  void send_unregister_client(const std::string &client_id);
+  Context *handle_unregister_client(int *result, const std::string &client_id);
+
+  void send_remove_mirror_image();
+  Context *handle_remove_mirror_image(int *result);
+
+  void send_notify_mirroring_watcher_removed();
+  Context *handle_notify_mirroring_watcher_removed(int *result);
+
+  Context *create_context_callback(
+    Context*(DisableRequest<ImageCtxT>::*handle)(
+      int*, const std::string &client_id),
+    const std::string &client_id);
+
+};
+
+} // namespace mirror
+} // namespace librbd
+
+extern template class librbd::mirror::DisableRequest<librbd::ImageCtx>;
+
+#endif // CEPH_LIBRBD_MIRROR_DISABLE_REQUEST_H