ceph/src/rbd_replay/actions.cc

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3 /*
   4  * Ceph - scalable distributed file system
   5  *
   6  * Copyright (C) 2014 Adam Crume <adamcrume@gmail.com>
   7  *
   8  * This is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License version 2.1, as published by the Free Software
  11  * Foundation.  See file COPYING.
  12  *
  13  */
  14
  15 #include "actions.hpp"
  16 #include <boost/foreach.hpp>
  17 #include <cstdlib>
  18 #include "PendingIO.hpp"
  19 #include "rbd_replay_debug.hpp"
  20
  21 #define dout_context g_ceph_context
  22
  23 using std::cerr;
  24 using namespace rbd_replay;
  25
  26 namespace {
  27
  28 std::string create_fake_data() {
  29   char data[1 << 20]; // 1 MB
  30   for (unsigned int i = 0; i < sizeof(data); i++) {
  31     data[i] = (char) i;
  32   }
  33   return std::string(data, sizeof(data));
  34 }
  35
  36 struct ConstructVisitor : public boost::static_visitor<Action::ptr> {
  37   inline Action::ptr operator()(const action::StartThreadAction &action) const {
  38     return Action::ptr(new StartThreadAction(action));
  39   }
  40
  41   inline Action::ptr operator()(const action::StopThreadAction &action) const{
  42     return Action::ptr(new StopThreadAction(action));
  43   }
  44
  45   inline Action::ptr operator()(const action::ReadAction &action) const {
  46     return Action::ptr(new ReadAction(action));
  47   }
  48
  49   inline Action::ptr operator()(const action::AioReadAction &action) const {
  50     return Action::ptr(new AioReadAction(action));
  51   }
  52
  53   inline Action::ptr operator()(const action::WriteAction &action) const {
  54     return Action::ptr(new WriteAction(action));
  55   }
  56
  57   inline Action::ptr operator()(const action::AioWriteAction &action) const {
  58     return Action::ptr(new AioWriteAction(action));
  59   }
  60
  61   inline Action::ptr operator()(const action::DiscardAction &action) const {
  62     return Action::ptr(new DiscardAction(action));
  63   }
  64
  65   inline Action::ptr operator()(const action::AioDiscardAction &action) const {
  66     return Action::ptr(new AioDiscardAction(action));
  67   }
  68
  69   inline Action::ptr operator()(const action::OpenImageAction &action) const {
  70     return Action::ptr(new OpenImageAction(action));
  71   }
  72
  73   inline Action::ptr operator()(const action::CloseImageAction &action) const {
  74     return Action::ptr(new CloseImageAction(action));
  75   }
  76
  77   inline Action::ptr operator()(const action::AioOpenImageAction &action) const {
  78     return Action::ptr(new AioOpenImageAction(action));
  79   }
  80
  81   inline Action::ptr operator()(const action::AioCloseImageAction &action) const {
  82     return Action::ptr(new AioCloseImageAction(action));
  83   }
  84
  85   inline Action::ptr operator()(const action::UnknownAction &action) const {
  86     return Action::ptr();
  87   }
  88 };
  89
  90 } // anonymous namespace
  91
  92 std::ostream& rbd_replay::operator<<(std::ostream& o, const Action& a) {
  93   return a.dump(o);
  94 }
  95
  96 Action::ptr Action::construct(const action::ActionEntry &action_entry) {
  97   return boost::apply_visitor(ConstructVisitor(), action_entry.action);
  98 }
  99
 100 void StartThreadAction::perform(ActionCtx &ctx) {
 101   cerr << "StartThreadAction should never actually be performed" << std::endl;
 102   exit(1);
 103 }
 104
 105 void StopThreadAction::perform(ActionCtx &ctx) {
 106   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 107   ctx.stop();
 108 }
 109
 110 void AioReadAction::perform(ActionCtx &worker) {
 111   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 112   librbd::Image *image = worker.get_image(m_action.imagectx_id);
 113   ceph_assert(image);
 114   PendingIO::ptr io(new PendingIO(pending_io_id(), worker));
 115   worker.add_pending(io);
 116   int r = image->aio_read(m_action.offset, m_action.length, io->bufferlist(), &io->completion());
 117   assertf(r >= 0, "id = %d, r = %d", id(), r);
 118 }
 119
 120 void ReadAction::perform(ActionCtx &worker) {
 121   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 122   librbd::Image *image = worker.get_image(m_action.imagectx_id);
 123   PendingIO::ptr io(new PendingIO(pending_io_id(), worker));
 124   worker.add_pending(io);
 125   ssize_t r = image->read(m_action.offset, m_action.length, io->bufferlist());
 126   assertf(r >= 0, "id = %d, r = %d", id(), r);
 127   worker.remove_pending(io);
 128 }
 129
 130 void AioWriteAction::perform(ActionCtx &worker) {
 131   static const std::string fake_data(create_fake_data());
 132   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 133   librbd::Image *image = worker.get_image(m_action.imagectx_id);
 134   PendingIO::ptr io(new PendingIO(pending_io_id(), worker));
 135   uint64_t remaining = m_action.length;
 136   while (remaining > 0) {
 137     uint64_t n = std::min(remaining, (uint64_t)fake_data.length());
 138     io->bufferlist().append(fake_data.data(), n);
 139     remaining -= n;
 140   }
 141   worker.add_pending(io);
 142   if (worker.readonly()) {
 143     worker.remove_pending(io);
 144   } else {
 145     int r = image->aio_write(m_action.offset, m_action.length, io->bufferlist(), &io->completion());
 146     assertf(r >= 0, "id = %d, r = %d", id(), r);
 147   }
 148 }
 149
 150 void WriteAction::perform(ActionCtx &worker) {
 151   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 152   librbd::Image *image = worker.get_image(m_action.imagectx_id);
 153   PendingIO::ptr io(new PendingIO(pending_io_id(), worker));
 154   worker.add_pending(io);
 155   io->bufferlist().append_zero(m_action.length);
 156   if (!worker.readonly()) {
 157     ssize_t r = image->write(m_action.offset, m_action.length, io->bufferlist());
 158     assertf(r >= 0, "id = %d, r = %d", id(), r);
 159   }
 160   worker.remove_pending(io);
 161 }
 162
 163 void AioDiscardAction::perform(ActionCtx &worker) {
 164   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 165   librbd::Image *image = worker.get_image(m_action.imagectx_id);
 166   PendingIO::ptr io(new PendingIO(pending_io_id(), worker));
 167   worker.add_pending(io);
 168   if (worker.readonly()) {
 169     worker.remove_pending(io);
 170   } else {
 171     int r = image->aio_discard(m_action.offset, m_action.length, &io->completion());
 172     assertf(r >= 0, "id = %d, r = %d", id(), r);
 173   }
 174 }
 175
 176 void DiscardAction::perform(ActionCtx &worker) {
 177   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 178   librbd::Image *image = worker.get_image(m_action.imagectx_id);
 179   PendingIO::ptr io(new PendingIO(pending_io_id(), worker));
 180   worker.add_pending(io);
 181   if (!worker.readonly()) {
 182     ssize_t r = image->discard(m_action.offset, m_action.length);
 183     assertf(r >= 0, "id = %d, r = %d", id(), r);
 184   }
 185   worker.remove_pending(io);
 186 }
 187
 188 void OpenImageAction::perform(ActionCtx &worker) {
 189   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 190   PendingIO::ptr io(new PendingIO(pending_io_id(), worker));
 191   worker.add_pending(io);
 192   librbd::Image *image = new librbd::Image();
 193   librbd::RBD *rbd = worker.rbd();
 194   rbd_loc name(worker.map_image_name(m_action.name, m_action.snap_name));
 195   int r;
 196   if (m_action.read_only || worker.readonly()) {
 197     r = rbd->open_read_only(*worker.ioctx(), *image, name.image.c_str(), name.snap.c_str());
 198   } else {
 199     r = rbd->open(*worker.ioctx(), *image, name.image.c_str(), name.snap.c_str());
 200   }
 201   if (r) {
 202     cerr << "Unable to open image '" << m_action.name
 203          << "' with snap '" << m_action.snap_name
 204          << "' (mapped to '" << name.str()
 205          << "') and readonly " << m_action.read_only
 206          << ": (" << -r << ") " << strerror(-r) << std::endl;
 207     exit(1);
 208   }
 209   worker.put_image(m_action.imagectx_id, image);
 210   worker.remove_pending(io);
 211 }
 212
 213 void CloseImageAction::perform(ActionCtx &worker) {
 214   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 215   worker.erase_image(m_action.imagectx_id);
 216   worker.set_action_complete(pending_io_id());
 217 }
 218
 219 void AioOpenImageAction::perform(ActionCtx &worker) {
 220   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 221   // TODO: Make it async
 222   PendingIO::ptr io(new PendingIO(pending_io_id(), worker));
 223   worker.add_pending(io);
 224   librbd::Image *image = new librbd::Image();
 225   librbd::RBD *rbd = worker.rbd();
 226   rbd_loc name(worker.map_image_name(m_action.name, m_action.snap_name));
 227   int r;
 228   if (m_action.read_only || worker.readonly()) {
 229     r = rbd->open_read_only(*worker.ioctx(), *image, name.image.c_str(), name.snap.c_str());
 230   } else {
 231     r = rbd->open(*worker.ioctx(), *image, name.image.c_str(), name.snap.c_str());
 232   }
 233   if (r) {
 234     cerr << "Unable to open image '" << m_action.name
 235          << "' with snap '" << m_action.snap_name
 236          << "' (mapped to '" << name.str()
 237          << "') and readonly " << m_action.read_only
 238          << ": (" << -r << ") " << strerror(-r) << std::endl;
 239     exit(1);
 240   }
 241   worker.put_image(m_action.imagectx_id, image);
 242   worker.remove_pending(io);
 243 }
 244
 245 void AioCloseImageAction::perform(ActionCtx &worker) {
 246   dout(ACTION_LEVEL) << "Performing " << *this << dendl;
 247   // TODO: Make it async
 248   worker.erase_image(m_action.imagectx_id);
 249   worker.set_action_complete(pending_io_id());
 250 }