1 files changed, 105 insertions, 0 deletions
diff --git a/src/librbd/api/Migration.h b/src/librbd/api/Migration.h
new file mode 100644
index 00000000..52a4517b
--- /dev/null
+++ b/src/librbd/api/Migration.h
@@ -0,0 +1,105 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#ifndef CEPH_LIBRBD_API_MIGRATION_H
+#define CEPH_LIBRBD_API_MIGRATION_H
+
+#include "include/int_types.h"
+#include "include/rados/librados_fwd.hpp"
+#include "include/rbd/librbd.hpp"
+#include "cls/rbd/cls_rbd_types.h"
+
+#include <vector>
+
+namespace librbd {
+
+class ImageCtx;
+
+namespace api {
+
+template <typename ImageCtxT = librbd::ImageCtx>
+class Migration {
+public:
+  static int prepare(librados::IoCtx& io_ctx, const std::string &image_name,
+                     librados::IoCtx& dest_io_ctx,
+                     const std::string &dest_image_name, ImageOptions& opts);
+  static int execute(librados::IoCtx& io_ctx, const std::string &image_name,
+                     ProgressContext &prog_ctx);
+  static int abort(librados::IoCtx& io_ctx, const std::string &image_name,
+                   ProgressContext &prog_ctx);
+  static int commit(librados::IoCtx& io_ctx, const std::string &image_name,
+                    ProgressContext &prog_ctx);
+  static int status(librados::IoCtx& io_ctx, const std::string &image_name,
+                    image_migration_status_t *status);
+
+private:
+  CephContext* m_cct;
+  ImageCtxT *m_src_image_ctx;
+  librados::IoCtx m_src_io_ctx;
+  librados::IoCtx &m_dst_io_ctx;
+  bool m_src_old_format;
+  std::string m_src_image_name;
+  std::string m_src_image_id;
+  std::string m_src_header_oid;
+  std::string m_dst_image_name;
+  std::string m_dst_image_id;
+  std::string m_dst_header_oid;
+  ImageOptions &m_image_options;
+  bool m_flatten;
+  bool m_mirroring;
+  ProgressContext *m_prog_ctx;
+
+  cls::rbd::MigrationSpec m_src_migration_spec;
+  cls::rbd::MigrationSpec m_dst_migration_spec;
+
+  Migration(ImageCtxT *image_ctx, librados::IoCtx& dest_io_ctx,
+            const std::string &dest_image_name, const std::string &dst_image_id,
+            ImageOptions& opts, bool flatten, bool mirroring,
+            cls::rbd::MigrationState state, const std::string &state_desc,
+            ProgressContext *prog_ctx);
+
+  int prepare();
+  int execute();
+  int abort();
+  int commit();
+  int status(image_migration_status_t *status);
+
+  int set_state(cls::rbd::MigrationState state, const std::string &description);
+
+  int list_src_snaps(std::vector<librbd::snap_info_t> *snaps);
+  int validate_src_snaps();
+  int disable_mirroring(ImageCtxT *image_ctx, bool *was_enabled);
+  int enable_mirroring(ImageCtxT *image_ctx, bool was_enabled);
+  int set_migration();
+  int unlink_src_image();
+  int relink_src_image();
+  int create_dst_image();
+  int remove_group(ImageCtxT *image_ctx, group_info_t *group_info);
+  int add_group(ImageCtxT *image_ctx, group_info_t &group_info);
+  int update_group(ImageCtxT *from_image_ctx, ImageCtxT *to_image_ctx);
+  int remove_migration(ImageCtxT *image_ctx);
+  int relink_children(ImageCtxT *from_image_ctx, ImageCtxT *to_image_ctx);
+  int remove_src_image();
+
+  int v1_set_migration();
+  int v2_set_migration();
+  int v1_unlink_src_image();
+  int v2_unlink_src_image();
+  int v1_relink_src_image();
+  int v2_relink_src_image();
+
+  int relink_child(ImageCtxT *from_image_ctx, ImageCtxT *to_image_ctx,
+                   const librbd::snap_info_t &src_snap,
+                   const librbd::linked_image_spec_t &child_image,
+                   bool migration_abort, bool reattach_child);
+
+  int revert_data(ImageCtxT* src_image_ctx, ImageCtxT* dst_image_ctx,
+                  ProgressContext *prog_ctx);
+};
+
+} // namespace api
+} // namespace librbd
+
+extern template class librbd::api::Migration<librbd::ImageCtx>;
+
+#endif // CEPH_LIBRBD_API_MIGRATION_H