src/ceph/src/tools/rbd/action/Clone.cc

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3
   4 #include "tools/rbd/ArgumentTypes.h"
   5 #include "tools/rbd/Shell.h"
   6 #include "tools/rbd/Utils.h"
   7 #include "common/errno.h"
   8 #include <iostream>
   9 #include <boost/program_options.hpp>
  10
  11 namespace rbd {
  12 namespace action {
  13 namespace clone {
  14
  15 namespace at = argument_types;
  16 namespace po = boost::program_options;
  17
  18 int do_clone(librbd::RBD &rbd, librados::IoCtx &p_ioctx,
  19              const char *p_name, const char *p_snapname,
  20              librados::IoCtx &c_ioctx, const char *c_name,
  21              librbd::ImageOptions& opts) {
  22   return rbd.clone3(p_ioctx, p_name, p_snapname, c_ioctx, c_name, opts);
  23 }
  24
  25 void get_arguments(po::options_description *positional,
  26                    po::options_description *options) {
  27   at::add_snap_spec_options(positional, options, at::ARGUMENT_MODIFIER_SOURCE);
  28   at::add_image_spec_options(positional, options, at::ARGUMENT_MODIFIER_DEST);
  29   at::add_create_image_options(options, false);
  30 }
  31
  32 int execute(const po::variables_map &vm) {
  33   size_t arg_index = 0;
  34   std::string pool_name;
  35   std::string image_name;
  36   std::string snap_name;
  37   int r = utils::get_pool_image_snapshot_names(
  38     vm, at::ARGUMENT_MODIFIER_SOURCE, &arg_index, &pool_name, &image_name,
  39     &snap_name, utils::SNAPSHOT_PRESENCE_REQUIRED, utils::SPEC_VALIDATION_NONE);
  40   if (r < 0) {
  41     return r;
  42   }
  43
  44   std::string dst_pool_name;
  45   std::string dst_image_name;
  46   std::string dst_snap_name;
  47   r = utils::get_pool_image_snapshot_names(
  48     vm, at::ARGUMENT_MODIFIER_DEST, &arg_index, &dst_pool_name, &dst_image_name,
  49     &dst_snap_name, utils::SNAPSHOT_PRESENCE_NONE, utils::SPEC_VALIDATION_FULL);
  50   if (r < 0) {
  51     return r;
  52   }
  53
  54   librbd::ImageOptions opts;
  55   r = utils::get_image_options(vm, false, &opts);
  56   if (r < 0) {
  57     return r;
  58   }
  59   opts.set(RBD_IMAGE_OPTION_FORMAT, static_cast<uint64_t>(2));
  60
  61   librados::Rados rados;
  62   librados::IoCtx io_ctx;
  63   r = utils::init(pool_name, &rados, &io_ctx);
  64   if (r < 0) {
  65     return r;
  66   }
  67
  68   librados::IoCtx dst_io_ctx;
  69   r = utils::init_io_ctx(rados, dst_pool_name, &dst_io_ctx);
  70   if (r < 0) {
  71     return r;
  72   }
  73
  74   librbd::RBD rbd;
  75   r = do_clone(rbd, io_ctx, image_name.c_str(), snap_name.c_str(), dst_io_ctx,
  76                dst_image_name.c_str(), opts);
  77   if (r < 0) {
  78     std::cerr << "rbd: clone error: " << cpp_strerror(r) << std::endl;
  79     return r;
  80   }
  81   return 0;
  82 }
  83
  84 Shell::Action action(
  85   {"clone"}, {}, "Clone a snapshot into a COW child image.",
  86   at::get_long_features_help(), &get_arguments, &execute);
  87
  88 } // namespace clone
  89 } // namespace action
  90 } // namespace rbd