+++ /dev/null
-// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
-// vim: ts=8 sw=2 smarttab
-
-#include "ObjectCopyRequest.h"
-#include "librados/snap_set_diff.h"
-#include "librbd/ExclusiveLock.h"
-#include "librbd/ObjectMap.h"
-#include "librbd/Utils.h"
-#include "common/errno.h"
-
-#define dout_context g_ceph_context
-#define dout_subsys ceph_subsys_rbd_mirror
-#undef dout_prefix
-#define dout_prefix *_dout << "rbd::mirror::image_sync::ObjectCopyRequest: " \
- << this << " " << __func__
-
-namespace librados {
-
-bool operator==(const clone_info_t& rhs, const clone_info_t& lhs) {
- return (rhs.cloneid == lhs.cloneid &&
- rhs.snaps == lhs.snaps &&
- rhs.overlap == lhs.overlap &&
- rhs.size == lhs.size);
-}
-
-bool operator==(const snap_set_t& rhs, const snap_set_t& lhs) {
- return (rhs.clones == lhs.clones &&
- rhs.seq == lhs.seq);
-}
-
-} // namespace librados
-
-namespace rbd {
-namespace mirror {
-namespace image_sync {
-
-using librbd::util::create_context_callback;
-using librbd::util::create_rados_callback;
-
-template <typename I>
-ObjectCopyRequest<I>::ObjectCopyRequest(I *local_image_ctx, I *remote_image_ctx,
- const SnapMap *snap_map,
- uint64_t object_number,
- Context *on_finish)
- : m_local_image_ctx(local_image_ctx), m_remote_image_ctx(remote_image_ctx),
- m_snap_map(snap_map), m_object_number(object_number),
- m_on_finish(on_finish) {
- assert(!snap_map->empty());
-
- m_local_io_ctx.dup(m_local_image_ctx->data_ctx);
- m_local_oid = m_local_image_ctx->get_object_name(object_number);
-
- m_remote_io_ctx.dup(m_remote_image_ctx->data_ctx);
- m_remote_oid = m_remote_image_ctx->get_object_name(object_number);
-
- dout(20) << ": "
- << "remote_oid=" << m_remote_oid << ", "
- << "local_oid=" << m_local_oid << dendl;
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::send() {
- send_list_snaps();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::send_list_snaps() {
- dout(20) << dendl;
-
- librados::AioCompletion *rados_completion = create_rados_callback<
- ObjectCopyRequest<I>, &ObjectCopyRequest<I>::handle_list_snaps>(this);
-
- librados::ObjectReadOperation op;
- m_snap_set = {};
- m_snap_ret = 0;
- op.list_snaps(&m_snap_set, &m_snap_ret);
-
- m_remote_io_ctx.snap_set_read(CEPH_SNAPDIR);
- int r = m_remote_io_ctx.aio_operate(m_remote_oid, rados_completion, &op,
- nullptr);
- assert(r == 0);
- rados_completion->release();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::handle_list_snaps(int r) {
- if (r == 0 && m_snap_ret < 0) {
- r = m_snap_ret;
- }
-
- dout(20) << ": r=" << r << dendl;
-
- if (r == -ENOENT) {
- finish(0);
- return;
- }
-
- if (r < 0) {
- derr << ": failed to list snaps: " << cpp_strerror(r) << dendl;
- finish(r);
- return;
- }
-
- if (m_retry_missing_read) {
- if (m_snap_set == m_retry_snap_set) {
- derr << ": read encountered missing object using up-to-date snap set"
- << dendl;
- finish(-ENOENT);
- return;
- }
-
- dout(20) << ": retrying using updated snap set" << dendl;
- m_retry_missing_read = false;
- m_retry_snap_set = {};
- }
-
- compute_diffs();
- send_read_object();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::send_read_object() {
- if (m_snap_sync_ops.empty()) {
- // no more snapshot diffs to read from remote
- finish(0);
- return;
- }
-
- // build the read request
- auto &sync_ops = m_snap_sync_ops.begin()->second;
- assert(!sync_ops.empty());
-
- bool read_required = false;
- librados::ObjectReadOperation op;
- for (auto &sync_op : sync_ops) {
- switch (sync_op.type) {
- case SYNC_OP_TYPE_WRITE:
- if (!read_required) {
- // map the sync op start snap id back to the necessary read snap id
- librados::snap_t remote_snap_seq =
- m_snap_sync_ops.begin()->first.second;
- m_remote_io_ctx.snap_set_read(remote_snap_seq);
-
- dout(20) << ": remote_snap_seq=" << remote_snap_seq << dendl;
- read_required = true;
- }
- dout(20) << ": read op: " << sync_op.offset << "~" << sync_op.length
- << dendl;
- op.sparse_read(sync_op.offset, sync_op.length, &sync_op.extent_map,
- &sync_op.out_bl, nullptr);
- op.set_op_flags2(LIBRADOS_OP_FLAG_FADVISE_SEQUENTIAL |
- LIBRADOS_OP_FLAG_FADVISE_NOCACHE);
- break;
- default:
- break;
- }
- }
-
- if (!read_required) {
- // nothing written to this object for this snapshot (must be trunc/remove)
- send_write_object();
- return;
- }
-
- librados::AioCompletion *comp = create_rados_callback<
- ObjectCopyRequest<I>, &ObjectCopyRequest<I>::handle_read_object>(this);
- int r = m_remote_io_ctx.aio_operate(m_remote_oid, comp, &op, nullptr);
- assert(r == 0);
- comp->release();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::handle_read_object(int r) {
- dout(20) << ": r=" << r << dendl;
-
- if (r == -ENOENT) {
- m_retry_snap_set = m_snap_set;
- m_retry_missing_read = true;
-
- dout(5) << ": object missing potentially due to removed snapshot" << dendl;
- send_list_snaps();
- return;
- }
-
- if (r < 0) {
- derr << ": failed to read from remote object: " << cpp_strerror(r)
- << dendl;
- finish(r);
- return;
- }
-
- send_write_object();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::send_write_object() {
- // retrieve the local snap context for the op
- SnapIds local_snap_ids;
- librados::snap_t local_snap_seq = 0;
- librados::snap_t remote_snap_seq = m_snap_sync_ops.begin()->first.first;
- if (remote_snap_seq != 0) {
- auto snap_map_it = m_snap_map->find(remote_snap_seq);
- assert(snap_map_it != m_snap_map->end());
-
- // write snapshot context should be before actual snapshot
- if (snap_map_it != m_snap_map->begin()) {
- --snap_map_it;
- assert(!snap_map_it->second.empty());
- local_snap_seq = snap_map_it->second.front();
- local_snap_ids = snap_map_it->second;
- }
- }
-
- Context *finish_op_ctx;
- {
- RWLock::RLocker owner_locker(m_local_image_ctx->owner_lock);
- finish_op_ctx = start_local_op(m_local_image_ctx->owner_lock);
- }
- if (finish_op_ctx == nullptr) {
- derr << ": lost exclusive lock" << dendl;
- finish(-EROFS);
- return;
- }
-
- dout(20) << ": "
- << "local_snap_seq=" << local_snap_seq << ", "
- << "local_snaps=" << local_snap_ids << dendl;
-
- auto &sync_ops = m_snap_sync_ops.begin()->second;
- assert(!sync_ops.empty());
- uint64_t object_offset;
- uint64_t buffer_offset;
- librados::ObjectWriteOperation op;
- for (auto &sync_op : sync_ops) {
- switch (sync_op.type) {
- case SYNC_OP_TYPE_WRITE:
- object_offset = sync_op.offset;
- buffer_offset = 0;
- for (auto it : sync_op.extent_map) {
- if (object_offset < it.first) {
- dout(20) << ": zero op: " << object_offset << "~"
- << it.first - object_offset << dendl;
- op.zero(object_offset, it.first - object_offset);
- }
- dout(20) << ": write op: " << it.first << "~" << it.second << dendl;
- bufferlist tmpbl;
- tmpbl.substr_of(sync_op.out_bl, buffer_offset, it.second);
- op.write(it.first, tmpbl);
- op.set_op_flags2(LIBRADOS_OP_FLAG_FADVISE_SEQUENTIAL |
- LIBRADOS_OP_FLAG_FADVISE_NOCACHE);
- buffer_offset += it.second;
- object_offset = it.first + it.second;
- }
- if (object_offset < sync_op.offset + sync_op.length) {
- uint64_t sync_op_end = sync_op.offset + sync_op.length;
- assert(sync_op_end <= m_snap_object_sizes[remote_snap_seq]);
- if (sync_op_end == m_snap_object_sizes[remote_snap_seq]) {
- dout(20) << ": trunc op: " << object_offset << dendl;
- op.truncate(object_offset);
- m_snap_object_sizes[remote_snap_seq] = object_offset;
- } else {
- dout(20) << ": zero op: " << object_offset << "~"
- << sync_op_end - object_offset << dendl;
- op.zero(object_offset, sync_op_end - object_offset);
- }
- }
- break;
- case SYNC_OP_TYPE_TRUNC:
- if (sync_op.offset > m_snap_object_sizes[remote_snap_seq]) {
- // skip (must have been updated in WRITE op case issuing trunc op)
- break;
- }
- dout(20) << ": trunc op: " << sync_op.offset << dendl;
- op.truncate(sync_op.offset);
- break;
- case SYNC_OP_TYPE_REMOVE:
- dout(20) << ": remove op" << dendl;
- op.remove();
- break;
- default:
- assert(false);
- }
- }
-
- auto ctx = new FunctionContext([this, finish_op_ctx](int r) {
- handle_write_object(r);
- finish_op_ctx->complete(0);
- });
- librados::AioCompletion *comp = create_rados_callback(ctx);
- int r = m_local_io_ctx.aio_operate(m_local_oid, comp, &op, local_snap_seq,
- local_snap_ids);
- assert(r == 0);
- comp->release();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::handle_write_object(int r) {
- dout(20) << ": r=" << r << dendl;
-
- if (r == -ENOENT) {
- r = 0;
- }
- if (r < 0) {
- derr << ": failed to write to local object: " << cpp_strerror(r)
- << dendl;
- finish(r);
- return;
- }
-
- m_snap_sync_ops.erase(m_snap_sync_ops.begin());
- if (!m_snap_sync_ops.empty()) {
- send_read_object();
- return;
- }
-
- send_update_object_map();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::send_update_object_map() {
- m_local_image_ctx->owner_lock.get_read();
- m_local_image_ctx->snap_lock.get_read();
- if (!m_local_image_ctx->test_features(RBD_FEATURE_OBJECT_MAP,
- m_local_image_ctx->snap_lock) ||
- m_snap_object_states.empty()) {
- m_local_image_ctx->snap_lock.put_read();
- m_local_image_ctx->owner_lock.put_read();
- finish(0);
- return;
- } else if (m_local_image_ctx->object_map == nullptr) {
- // possible that exclusive lock was lost in background
- derr << ": object map is not initialized" << dendl;
-
- m_local_image_ctx->snap_lock.put_read();
- m_local_image_ctx->owner_lock.put_read();
- finish(-EINVAL);
- return;
- }
-
- assert(m_local_image_ctx->object_map != nullptr);
-
- auto snap_object_state = *m_snap_object_states.begin();
- m_snap_object_states.erase(m_snap_object_states.begin());
-
- dout(20) << ": "
- << "local_snap_id=" << snap_object_state.first << ", "
- << "object_state=" << static_cast<uint32_t>(snap_object_state.second)
- << dendl;
-
- auto finish_op_ctx = start_local_op(m_local_image_ctx->owner_lock);
- if (finish_op_ctx == nullptr) {
- derr << ": lost exclusive lock" << dendl;
- m_local_image_ctx->snap_lock.put_read();
- m_local_image_ctx->owner_lock.put_read();
- finish(-EROFS);
- return;
- }
-
- auto ctx = new FunctionContext([this, finish_op_ctx](int r) {
- handle_update_object_map(r);
- finish_op_ctx->complete(0);
- });
-
- RWLock::WLocker object_map_locker(m_local_image_ctx->object_map_lock);
- bool sent = m_local_image_ctx->object_map->template aio_update<
- Context, &Context::complete>(
- snap_object_state.first, m_object_number, snap_object_state.second, {},
- {}, ctx);
- assert(sent);
- m_local_image_ctx->snap_lock.put_read();
- m_local_image_ctx->owner_lock.put_read();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::handle_update_object_map(int r) {
- dout(20) << ": r=" << r << dendl;
-
- assert(r == 0);
- if (!m_snap_object_states.empty()) {
- send_update_object_map();
- return;
- }
- finish(0);
-}
-
-template <typename I>
-Context *ObjectCopyRequest<I>::start_local_op(RWLock &owner_lock) {
- assert(m_local_image_ctx->owner_lock.is_locked());
- if (m_local_image_ctx->exclusive_lock == nullptr) {
- return nullptr;
- }
- return m_local_image_ctx->exclusive_lock->start_op();
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::compute_diffs() {
- CephContext *cct = m_local_image_ctx->cct;
-
- m_snap_sync_ops = {};
- m_snap_object_states = {};
- m_snap_object_sizes = {};
-
- librados::snap_t remote_sync_pont_snap_id = m_snap_map->rbegin()->first;
- uint64_t prev_end_size = 0;
- bool prev_exists = false;
- librados::snap_t start_remote_snap_id = 0;
- for (auto &pair : *m_snap_map) {
- assert(!pair.second.empty());
- librados::snap_t end_remote_snap_id = pair.first;
- librados::snap_t end_local_snap_id = pair.second.front();
-
- interval_set<uint64_t> diff;
- uint64_t end_size;
- bool exists;
- librados::snap_t clone_end_snap_id;
- calc_snap_set_diff(cct, m_snap_set, start_remote_snap_id,
- end_remote_snap_id, &diff, &end_size, &exists,
- &clone_end_snap_id);
-
- dout(20) << ": "
- << "start_remote_snap=" << start_remote_snap_id << ", "
- << "end_remote_snap_id=" << end_remote_snap_id << ", "
- << "clone_end_snap_id=" << clone_end_snap_id << ", "
- << "end_local_snap_id=" << end_local_snap_id << ", "
- << "diff=" << diff << ", "
- << "end_size=" << end_size << ", "
- << "exists=" << exists << dendl;
- if (exists) {
- // clip diff to size of object (in case it was truncated)
- if (end_size < prev_end_size) {
- interval_set<uint64_t> trunc;
- trunc.insert(end_size, prev_end_size);
- trunc.intersection_of(diff);
- diff.subtract(trunc);
- dout(20) << ": clearing truncate diff: " << trunc << dendl;
- }
-
- // prepare the object map state
- {
- RWLock::RLocker snap_locker(m_local_image_ctx->snap_lock);
- uint8_t object_state = OBJECT_EXISTS;
- if (m_local_image_ctx->test_features(RBD_FEATURE_FAST_DIFF,
- m_local_image_ctx->snap_lock) &&
- prev_exists && diff.empty() && end_size == prev_end_size) {
- object_state = OBJECT_EXISTS_CLEAN;
- }
- m_snap_object_states[end_local_snap_id] = object_state;
- }
-
- // reads should be issued against the newest (existing) snapshot within
- // the associated snapshot object clone. writes should be issued
- // against the oldest snapshot in the snap_map.
- assert(clone_end_snap_id >= end_remote_snap_id);
- if (clone_end_snap_id > remote_sync_pont_snap_id) {
- // do not read past the sync point snapshot
- clone_end_snap_id = remote_sync_pont_snap_id;
- }
-
- // object write/zero, or truncate
- // NOTE: a single snapshot clone might represent multiple snapshots, but
- // the write/zero and truncate ops will only be associated with the first
- // snapshot encountered within the clone since the diff will be empty for
- // subsequent snapshots and the size will remain constant for a clone.
- for (auto it = diff.begin(); it != diff.end(); ++it) {
- dout(20) << ": read/write op: " << it.get_start() << "~"
- << it.get_len() << dendl;
- m_snap_sync_ops[{end_remote_snap_id, clone_end_snap_id}].emplace_back(
- SYNC_OP_TYPE_WRITE, it.get_start(), it.get_len());
- }
- if (end_size < prev_end_size) {
- dout(20) << ": trunc op: " << end_size << dendl;
- m_snap_sync_ops[{end_remote_snap_id, clone_end_snap_id}].emplace_back(
- SYNC_OP_TYPE_TRUNC, end_size, 0U);
- }
- m_snap_object_sizes[end_remote_snap_id] = end_size;
- } else {
- if (prev_exists) {
- // object remove
- dout(20) << ": remove op" << dendl;
- m_snap_sync_ops[{end_remote_snap_id, end_remote_snap_id}].emplace_back(
- SYNC_OP_TYPE_REMOVE, 0U, 0U);
- }
- }
-
- prev_end_size = end_size;
- prev_exists = exists;
- start_remote_snap_id = end_remote_snap_id;
- }
-}
-
-template <typename I>
-void ObjectCopyRequest<I>::finish(int r) {
- dout(20) << ": r=" << r << dendl;
-
- // ensure IoCtxs are closed prior to proceeding
- auto on_finish = m_on_finish;
- delete this;
-
- on_finish->complete(r);
-}
-
-} // namespace image_sync
-} // namespace mirror
-} // namespace rbd
-
-template class rbd::mirror::image_sync::ObjectCopyRequest<librbd::ImageCtx>;