1 files changed, 141 insertions, 0 deletions
diff --git a/src/tools/rbd_mirror/image_replayer/PrepareRemoteImageRequest.h b/src/tools/rbd_mirror/image_replayer/PrepareRemoteImageRequest.h
new file mode 100644
index 00000000..100a066b
--- /dev/null
+++ b/src/tools/rbd_mirror/image_replayer/PrepareRemoteImageRequest.h
@@ -0,0 +1,141 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#ifndef RBD_MIRROR_IMAGE_REPLAYER_PREPARE_REMOTE_IMAGE_REQUEST_H
+#define RBD_MIRROR_IMAGE_REPLAYER_PREPARE_REMOTE_IMAGE_REQUEST_H
+
+#include "include/buffer_fwd.h"
+#include "include/rados/librados_fwd.hpp"
+#include "cls/journal/cls_journal_types.h"
+#include "journal/Settings.h"
+#include "librbd/journal/TypeTraits.h"
+#include <string>
+
+namespace journal { class Journaler; }
+namespace journal { class Settings; }
+namespace librbd { struct ImageCtx; }
+namespace librbd { namespace journal { struct MirrorPeerClientMeta; } }
+
+struct Context;
+struct ContextWQ;
+
+namespace rbd {
+namespace mirror {
+
+template <typename> struct Threads;
+
+namespace image_replayer {
+
+template <typename ImageCtxT = librbd::ImageCtx>
+class PrepareRemoteImageRequest {
+public:
+  typedef librbd::journal::TypeTraits<ImageCtxT> TypeTraits;
+  typedef typename TypeTraits::Journaler Journaler;
+  typedef librbd::journal::MirrorPeerClientMeta MirrorPeerClientMeta;
+
+  static PrepareRemoteImageRequest *create(Threads<ImageCtxT> *threads,
+                                           librados::IoCtx &remote_io_ctx,
+                                           const std::string &global_image_id,
+                                           const std::string &local_mirror_uuid,
+                                           const std::string &local_image_id,
+                                           const journal::Settings &settings,
+                                           std::string *remote_mirror_uuid,
+                                           std::string *remote_image_id,
+                                           Journaler **remote_journaler,
+                                           cls::journal::ClientState *client_state,
+                                           MirrorPeerClientMeta *client_meta,
+                                           Context *on_finish) {
+    return new PrepareRemoteImageRequest(threads, remote_io_ctx,
+                                         global_image_id, local_mirror_uuid,
+                                         local_image_id, settings,
+                                         remote_mirror_uuid, remote_image_id,
+                                         remote_journaler, client_state,
+                                         client_meta, on_finish);
+  }
+
+  PrepareRemoteImageRequest(Threads<ImageCtxT> *threads,
+                           librados::IoCtx &remote_io_ctx,
+                           const std::string &global_image_id,
+                           const std::string &local_mirror_uuid,
+                           const std::string &local_image_id,
+                           const journal::Settings &journal_settings,
+                           std::string *remote_mirror_uuid,
+                           std::string *remote_image_id,
+                           Journaler **remote_journaler,
+                           cls::journal::ClientState *client_state,
+                           MirrorPeerClientMeta *client_meta,
+                           Context *on_finish)
+    : m_threads(threads), m_remote_io_ctx(remote_io_ctx),
+      m_global_image_id(global_image_id),
+      m_local_mirror_uuid(local_mirror_uuid), m_local_image_id(local_image_id),
+      m_journal_settings(journal_settings),
+      m_remote_mirror_uuid(remote_mirror_uuid),
+      m_remote_image_id(remote_image_id),
+      m_remote_journaler(remote_journaler), m_client_state(client_state),
+      m_client_meta(client_meta), m_on_finish(on_finish) {
+  }
+
+  void send();
+
+private:
+  /**
+   * @verbatim
+   *
+   * <start>
+   *    |
+   *    v
+   * GET_REMOTE_MIRROR_UUID
+   *    |
+   *    v
+   * GET_REMOTE_IMAGE_ID
+   *    |
+   *    v
+   * GET_CLIENT
+   *    |
+   *    v (skip if not needed)
+   * REGISTER_CLIENT
+   *    |
+   *    v
+   * <finish>
+
+   * @endverbatim
+   */
+
+  Threads<ImageCtxT> *m_threads;
+  librados::IoCtx &m_remote_io_ctx;
+  std::string m_global_image_id;
+  std::string m_local_mirror_uuid;
+  std::string m_local_image_id;
+  journal::Settings m_journal_settings;
+  std::string *m_remote_mirror_uuid;
+  std::string *m_remote_image_id;
+  Journaler **m_remote_journaler;
+  cls::journal::ClientState *m_client_state;
+  MirrorPeerClientMeta *m_client_meta;
+  Context *m_on_finish;
+
+  bufferlist m_out_bl;
+  cls::journal::Client m_client;
+
+  void get_remote_mirror_uuid();
+  void handle_get_remote_mirror_uuid(int r);
+
+  void get_remote_image_id();
+  void handle_get_remote_image_id(int r);
+
+  void get_client();
+  void handle_get_client(int r);
+
+  void register_client();
+  void handle_register_client(int r);
+
+  void finish(int r);
+};
+
+} // namespace image_replayer
+} // namespace mirror
+} // namespace rbd
+
+extern template class rbd::mirror::image_replayer::PrepareRemoteImageRequest<librbd::ImageCtx>;
+
+#endif // RBD_MIRROR_IMAGE_REPLAYER_PREPARE_REMOTE_IMAGE_REQUEST_H