1 files changed, 129 insertions, 0 deletions
diff --git a/src/librbd/Operations.h b/src/librbd/Operations.h
new file mode 100644
index 00000000..253c9f92
--- /dev/null
+++ b/src/librbd/Operations.h
@@ -0,0 +1,129 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#ifndef CEPH_LIBRBD_OPERATIONS_H
+#define CEPH_LIBRBD_OPERATIONS_H
+
+#include "cls/rbd/cls_rbd_types.h"
+#include "include/int_types.h"
+#include "librbd/exclusive_lock/Policy.h"
+#include "librbd/operation/ObjectMapIterate.h"
+#include <atomic>
+#include <string>
+#include <boost/function.hpp>
+
+class Context;
+
+namespace librbd {
+
+class ImageCtx;
+class ProgressContext;
+
+template <typename ImageCtxT = ImageCtx>
+class Operations {
+public:
+  Operations(ImageCtxT &image_ctx);
+
+  int flatten(ProgressContext &prog_ctx);
+  void execute_flatten(ProgressContext &prog_ctx, Context *on_finish);
+
+  int rebuild_object_map(ProgressContext &prog_ctx);
+  void execute_rebuild_object_map(ProgressContext &prog_ctx,
+                                  Context *on_finish);
+
+  int check_object_map(ProgressContext &prog_ctx);
+  void check_object_map(ProgressContext &prog_ctx, Context *on_finish);
+
+  void object_map_iterate(ProgressContext &prog_ctx,
+			  operation::ObjectIterateWork<ImageCtxT> handle_mismatch,
+			  Context* on_finish);
+
+  int rename(const char *dstname);
+  void execute_rename(const std::string &dest_name, Context *on_finish);
+
+  int resize(uint64_t size, bool allow_shrink, ProgressContext& prog_ctx);
+  void execute_resize(uint64_t size, bool allow_shrink, ProgressContext &prog_ctx,
+                      Context *on_finish, uint64_t journal_op_tid);
+
+  int snap_create(const cls::rbd::SnapshotNamespace &snap_namespace,
+		  const std::string& snap_name);
+  void snap_create(const cls::rbd::SnapshotNamespace &snap_namespace,
+		   const std::string& snap_name, Context *on_finish);
+  void execute_snap_create(const cls::rbd::SnapshotNamespace &snap_namespace,
+			   const std::string &snap_name,
+			   Context *on_finish,
+                           uint64_t journal_op_tid, bool skip_object_map);
+
+  int snap_rollback(const cls::rbd::SnapshotNamespace& snap_namespace,
+		    const std::string& snap_name,
+		    ProgressContext& prog_ctx);
+  void execute_snap_rollback(const cls::rbd::SnapshotNamespace& snap_namespace,
+			     const std::string &snap_name,
+                             ProgressContext& prog_ctx, Context *on_finish);
+
+  int snap_remove(const cls::rbd::SnapshotNamespace& snap_namespace,
+		  const std::string& snap_name);
+  void snap_remove(const cls::rbd::SnapshotNamespace& snap_namespace,
+		   const std::string& snap_name,
+		   Context *on_finish);
+  void execute_snap_remove(const cls::rbd::SnapshotNamespace& snap_namespace,
+			   const std::string &snap_name,
+			   Context *on_finish);
+
+  int snap_rename(const char *srcname, const char *dstname);
+  void execute_snap_rename(const uint64_t src_snap_id,
+                           const std::string &dest_snap_name,
+                           Context *on_finish);
+
+  int snap_protect(const cls::rbd::SnapshotNamespace& snap_namespace,
+		   const std::string& snap_name);
+  void execute_snap_protect(const cls::rbd::SnapshotNamespace& snap_namespace,
+			    const std::string &snap_name,
+			    Context *on_finish);
+
+  int snap_unprotect(const cls::rbd::SnapshotNamespace& snap_namespace,
+		     const std::string& snap_name);
+  void execute_snap_unprotect(const cls::rbd::SnapshotNamespace& snap_namespace,
+			      const std::string &snap_name,
+			      Context *on_finish);
+
+  int snap_set_limit(uint64_t limit);
+  void execute_snap_set_limit(uint64_t limit, Context *on_finish);
+
+  int update_features(uint64_t features, bool enabled);
+  void execute_update_features(uint64_t features, bool enabled,
+                               Context *on_finish, uint64_t journal_op_tid);
+
+  int metadata_set(const std::string &key, const std::string &value);
+  void execute_metadata_set(const std::string &key, const std::string &value,
+                            Context *on_finish);
+
+  int metadata_remove(const std::string &key);
+  void execute_metadata_remove(const std::string &key, Context *on_finish);
+
+  int migrate(ProgressContext &prog_ctx);
+  void execute_migrate(ProgressContext &prog_ctx, Context *on_finish);
+
+  int sparsify(size_t sparse_size, ProgressContext &prog_ctx);
+  void execute_sparsify(size_t sparse_size, ProgressContext &prog_ctx,
+                        Context *on_finish);
+
+  int prepare_image_update(exclusive_lock::OperationRequestType request_type,
+                           bool request_lock);
+
+private:
+  ImageCtxT &m_image_ctx;
+  std::atomic<int> m_async_request_seq;
+
+  int invoke_async_request(const std::string& name,
+                           exclusive_lock::OperationRequestType request_type,
+                           bool permit_snapshot,
+                           const boost::function<void(Context*)>& local,
+                           const boost::function<void(Context*)>& remote);
+};
+
+} // namespace librbd
+
+extern template class librbd::Operations<librbd::ImageCtx>;
+
+#endif // CEPH_LIBRBD_OPERATIONS_H