--- /dev/null
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#include "test/rbd_mirror/test_mock_fixture.h"
+#include "include/interval_set.h"
+#include "include/rbd/librbd.hpp"
+#include "include/rbd/object_map_types.h"
+#include "librbd/ImageCtx.h"
+#include "librbd/ImageState.h"
+#include "librbd/internal.h"
+#include "librbd/Operations.h"
+#include "librbd/io/ImageRequestWQ.h"
+#include "librbd/io/ReadResult.h"
+#include "test/librados_test_stub/MockTestMemIoCtxImpl.h"
+#include "test/librbd/mock/MockImageCtx.h"
+#include "tools/rbd_mirror/Threads.h"
+#include "tools/rbd_mirror/image_sync/ObjectCopyRequest.h"
+
+namespace librbd {
+namespace {
+
+struct MockTestImageCtx : public librbd::MockImageCtx {
+ MockTestImageCtx(librbd::ImageCtx &image_ctx)
+ : librbd::MockImageCtx(image_ctx) {
+ }
+};
+
+} // anonymous namespace
+} // namespace librbd
+
+// template definitions
+#include "tools/rbd_mirror/image_sync/ObjectCopyRequest.cc"
+template class rbd::mirror::image_sync::ObjectCopyRequest<librbd::MockTestImageCtx>;
+
+bool operator==(const SnapContext& rhs, const SnapContext& lhs) {
+ return (rhs.seq == lhs.seq && rhs.snaps == lhs.snaps);
+}
+
+namespace rbd {
+namespace mirror {
+namespace image_sync {
+
+using ::testing::_;
+using ::testing::DoAll;
+using ::testing::DoDefault;
+using ::testing::InSequence;
+using ::testing::Invoke;
+using ::testing::Return;
+using ::testing::ReturnNew;
+using ::testing::WithArg;
+
+namespace {
+
+void scribble(librbd::ImageCtx *image_ctx, int num_ops, size_t max_size,
+ interval_set<uint64_t> *what)
+{
+ uint64_t object_size = 1 << image_ctx->order;
+ for (int i=0; i<num_ops; i++) {
+ uint64_t off = rand() % (object_size - max_size + 1);
+ uint64_t len = 1 + rand() % max_size;
+
+ bufferlist bl;
+ bl.append(std::string(len, '1'));
+
+ int r = image_ctx->io_work_queue->write(off, len, std::move(bl), 0);
+ ASSERT_EQ(static_cast<int>(len), r);
+
+ interval_set<uint64_t> w;
+ w.insert(off, len);
+ what->union_of(w);
+ }
+ std::cout << " wrote " << *what << std::endl;
+}
+
+} // anonymous namespace
+
+class TestMockImageSyncObjectCopyRequest : public TestMockFixture {
+public:
+ typedef ObjectCopyRequest<librbd::MockTestImageCtx> MockObjectCopyRequest;
+
+ void SetUp() override {
+ TestMockFixture::SetUp();
+
+ librbd::RBD rbd;
+ ASSERT_EQ(0, create_image(rbd, m_remote_io_ctx, m_image_name, m_image_size));
+ ASSERT_EQ(0, open_image(m_remote_io_ctx, m_image_name, &m_remote_image_ctx));
+
+ ASSERT_EQ(0, create_image(rbd, m_local_io_ctx, m_image_name, m_image_size));
+ ASSERT_EQ(0, open_image(m_local_io_ctx, m_image_name, &m_local_image_ctx));
+ }
+
+ void expect_start_op(librbd::MockExclusiveLock &mock_exclusive_lock) {
+ EXPECT_CALL(mock_exclusive_lock, start_op()).WillOnce(
+ ReturnNew<FunctionContext>([](int) {}));
+ }
+
+ void expect_list_snaps(librbd::MockTestImageCtx &mock_image_ctx,
+ librados::MockTestMemIoCtxImpl &mock_io_ctx,
+ const librados::snap_set_t &snap_set) {
+ expect_set_snap_read(mock_io_ctx, CEPH_SNAPDIR);
+ EXPECT_CALL(mock_io_ctx,
+ list_snaps(mock_image_ctx.image_ctx->get_object_name(0), _))
+ .WillOnce(DoAll(WithArg<1>(Invoke([&snap_set](librados::snap_set_t *out_snap_set) {
+ *out_snap_set = snap_set;
+ })),
+ Return(0)));
+ }
+
+ void expect_list_snaps(librbd::MockTestImageCtx &mock_image_ctx,
+ librados::MockTestMemIoCtxImpl &mock_io_ctx, int r) {
+ expect_set_snap_read(mock_io_ctx, CEPH_SNAPDIR);
+ auto &expect = EXPECT_CALL(mock_io_ctx,
+ list_snaps(mock_image_ctx.image_ctx->get_object_name(0),
+ _));
+ if (r < 0) {
+ expect.WillOnce(Return(r));
+ } else {
+ expect.WillOnce(DoDefault());
+ }
+ }
+
+ void expect_get_object_name(librbd::MockTestImageCtx &mock_image_ctx) {
+ EXPECT_CALL(mock_image_ctx, get_object_name(0))
+ .WillOnce(Return(mock_image_ctx.image_ctx->get_object_name(0)));
+ }
+
+ MockObjectCopyRequest *create_request(librbd::MockTestImageCtx &mock_remote_image_ctx,
+ librbd::MockTestImageCtx &mock_local_image_ctx,
+ Context *on_finish) {
+ expect_get_object_name(mock_local_image_ctx);
+ expect_get_object_name(mock_remote_image_ctx);
+ return new MockObjectCopyRequest(&mock_local_image_ctx,
+ &mock_remote_image_ctx, &m_snap_map,
+ 0, on_finish);
+ }
+
+ void expect_set_snap_read(librados::MockTestMemIoCtxImpl &mock_io_ctx,
+ uint64_t snap_id) {
+ EXPECT_CALL(mock_io_ctx, set_snap_read(snap_id));
+ }
+
+ void expect_sparse_read(librados::MockTestMemIoCtxImpl &mock_io_ctx, uint64_t offset,
+ uint64_t length, int r) {
+
+ auto &expect = EXPECT_CALL(mock_io_ctx, sparse_read(_, offset, length, _, _));
+ if (r < 0) {
+ expect.WillOnce(Return(r));
+ } else {
+ expect.WillOnce(DoDefault());
+ }
+ }
+
+ void expect_sparse_read(librados::MockTestMemIoCtxImpl &mock_io_ctx,
+ const interval_set<uint64_t> &extents, int r) {
+ for (auto extent : extents) {
+ expect_sparse_read(mock_io_ctx, extent.first, extent.second, r);
+ if (r < 0) {
+ break;
+ }
+ }
+ }
+
+ void expect_write(librados::MockTestMemIoCtxImpl &mock_io_ctx,
+ uint64_t offset, uint64_t length,
+ const SnapContext &snapc, int r) {
+ auto &expect = EXPECT_CALL(mock_io_ctx, write(_, _, length, offset, snapc));
+ if (r < 0) {
+ expect.WillOnce(Return(r));
+ } else {
+ expect.WillOnce(DoDefault());
+ }
+ }
+
+ void expect_write(librados::MockTestMemIoCtxImpl &mock_io_ctx,
+ const interval_set<uint64_t> &extents,
+ const SnapContext &snapc, int r) {
+ for (auto extent : extents) {
+ expect_write(mock_io_ctx, extent.first, extent.second, snapc, r);
+ if (r < 0) {
+ break;
+ }
+ }
+ }
+
+ void expect_truncate(librados::MockTestMemIoCtxImpl &mock_io_ctx,
+ uint64_t offset, int r) {
+ auto &expect = EXPECT_CALL(mock_io_ctx, truncate(_, offset, _));
+ if (r < 0) {
+ expect.WillOnce(Return(r));
+ } else {
+ expect.WillOnce(DoDefault());
+ }
+ }
+
+ void expect_remove(librados::MockTestMemIoCtxImpl &mock_io_ctx, int r) {
+ auto &expect = EXPECT_CALL(mock_io_ctx, remove(_, _));
+ if (r < 0) {
+ expect.WillOnce(Return(r));
+ } else {
+ expect.WillOnce(DoDefault());
+ }
+ }
+
+ void expect_update_object_map(librbd::MockTestImageCtx &mock_image_ctx,
+ librbd::MockObjectMap &mock_object_map,
+ librados::snap_t snap_id, uint8_t state,
+ int r) {
+ if (mock_image_ctx.image_ctx->object_map != nullptr) {
+ auto &expect = EXPECT_CALL(mock_object_map, aio_update(snap_id, 0, 1, state, _, _, _));
+ if (r < 0) {
+ expect.WillOnce(DoAll(WithArg<6>(Invoke([this, r](Context *ctx) {
+ m_threads->work_queue->queue(ctx, r);
+ })),
+ Return(true)));
+ } else {
+ expect.WillOnce(DoAll(WithArg<6>(Invoke([&mock_image_ctx, snap_id, state, r](Context *ctx) {
+ assert(mock_image_ctx.image_ctx->snap_lock.is_locked());
+ assert(mock_image_ctx.image_ctx->object_map_lock.is_wlocked());
+ mock_image_ctx.image_ctx->object_map->aio_update<Context>(
+ snap_id, 0, 1, state, boost::none, {}, ctx);
+ })),
+ Return(true)));
+ }
+ }
+ }
+
+ using TestFixture::create_snap;
+ int create_snap(const char* snap_name) {
+ librados::snap_t remote_snap_id;
+ int r = create_snap(m_remote_image_ctx, snap_name, &remote_snap_id);
+ if (r < 0) {
+ return r;
+ }
+
+ librados::snap_t local_snap_id;
+ r = create_snap(m_local_image_ctx, snap_name, &local_snap_id);
+ if (r < 0) {
+ return r;
+ }
+
+ // collection of all existing snaps in local image
+ MockObjectCopyRequest::SnapIds local_snap_ids({local_snap_id});
+ if (!m_snap_map.empty()) {
+ local_snap_ids.insert(local_snap_ids.end(),
+ m_snap_map.rbegin()->second.begin(),
+ m_snap_map.rbegin()->second.end());
+ }
+ m_snap_map[remote_snap_id] = local_snap_ids;
+ m_remote_snap_ids.push_back(remote_snap_id);
+ m_local_snap_ids.push_back(local_snap_id);
+
+ return 0;
+ }
+
+ std::string get_snap_name(librbd::ImageCtx *image_ctx,
+ librados::snap_t snap_id) {
+ auto it = std::find_if(image_ctx->snap_ids.begin(),
+ image_ctx->snap_ids.end(),
+ [snap_id](const std::pair<std::pair<cls::rbd::SnapshotNamespace,
+ std::string>,
+ librados::snap_t> &pair) {
+ return (pair.second == snap_id);
+ });
+ if (it == image_ctx->snap_ids.end()) {
+ return "";
+ }
+ return it->first.second;
+ }
+
+ int compare_objects() {
+ MockObjectCopyRequest::SnapMap snap_map(m_snap_map);
+ if (snap_map.empty()) {
+ return -ENOENT;
+ }
+
+ int r;
+ uint64_t object_size = 1 << m_remote_image_ctx->order;
+ while (!snap_map.empty()) {
+ librados::snap_t remote_snap_id = snap_map.begin()->first;
+ librados::snap_t local_snap_id = *snap_map.begin()->second.begin();
+ snap_map.erase(snap_map.begin());
+
+ std::string snap_name = get_snap_name(m_remote_image_ctx, remote_snap_id);
+ if (snap_name.empty()) {
+ return -ENOENT;
+ }
+
+ std::cout << "comparing '" << snap_name << " (" << remote_snap_id
+ << " to " << local_snap_id << ")" << std::endl;
+
+ r = librbd::snap_set(m_remote_image_ctx,
+ cls::rbd::UserSnapshotNamespace(),
+ snap_name.c_str());
+ if (r < 0) {
+ return r;
+ }
+
+ r = librbd::snap_set(m_local_image_ctx,
+ cls::rbd::UserSnapshotNamespace(),
+ snap_name.c_str());
+ if (r < 0) {
+ return r;
+ }
+
+ bufferlist remote_bl;
+ remote_bl.append(std::string(object_size, '1'));
+ r = m_remote_image_ctx->io_work_queue->read(
+ 0, object_size, librbd::io::ReadResult{&remote_bl}, 0);
+ if (r < 0) {
+ return r;
+ }
+
+ bufferlist local_bl;
+ local_bl.append(std::string(object_size, '1'));
+ r = m_local_image_ctx->io_work_queue->read(
+ 0, object_size, librbd::io::ReadResult{&local_bl}, 0);
+ if (r < 0) {
+ return r;
+ }
+
+ if (!remote_bl.contents_equal(local_bl)) {
+ return -EBADMSG;
+ }
+ }
+
+ r = librbd::snap_set(m_remote_image_ctx,
+ cls::rbd::UserSnapshotNamespace(),
+ nullptr);
+ if (r < 0) {
+ return r;
+ }
+ r = librbd::snap_set(m_local_image_ctx,
+ cls::rbd::UserSnapshotNamespace(),
+ nullptr);
+ if (r < 0) {
+ return r;
+ }
+
+ return 0;
+ }
+
+ librbd::ImageCtx *m_remote_image_ctx;
+ librbd::ImageCtx *m_local_image_ctx;
+
+ MockObjectCopyRequest::SnapMap m_snap_map;
+ std::vector<librados::snap_t> m_remote_snap_ids;
+ std::vector<librados::snap_t> m_local_snap_ids;
+};
+
+TEST_F(TestMockImageSyncObjectCopyRequest, DNE) {
+ ASSERT_EQ(0, create_snap("sync"));
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, -ENOENT);
+
+ request->send();
+ ASSERT_EQ(0, ctx.wait());
+}
+
+TEST_F(TestMockImageSyncObjectCopyRequest, Write) {
+ // scribble some data
+ interval_set<uint64_t> one;
+ scribble(m_remote_image_ctx, 10, 102400, &one);
+
+ ASSERT_EQ(0, create_snap("sync"));
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+ librados::MockTestMemIoCtxImpl &mock_local_io_ctx(get_mock_io_ctx(
+ request->get_local_io_ctx()));
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[0]);
+ expect_sparse_read(mock_remote_io_ctx, 0, one.range_end(), 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_write(mock_local_io_ctx, 0, one.range_end(), {0, {}}, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[0], OBJECT_EXISTS, 0);
+
+ request->send();
+ ASSERT_EQ(0, ctx.wait());
+ ASSERT_EQ(0, compare_objects());
+}
+
+TEST_F(TestMockImageSyncObjectCopyRequest, ReadMissingStaleSnapSet) {
+ ASSERT_EQ(0, create_snap("one"));
+ ASSERT_EQ(0, create_snap("two"));
+
+ // scribble some data
+ interval_set<uint64_t> one;
+ scribble(m_remote_image_ctx, 10, 102400, &one);
+ ASSERT_EQ(0, create_snap("three"));
+
+ ASSERT_EQ(0, create_snap("sync"));
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+ librados::MockTestMemIoCtxImpl &mock_local_io_ctx(get_mock_io_ctx(
+ request->get_local_io_ctx()));
+
+ librados::clone_info_t dummy_clone_info;
+ dummy_clone_info.cloneid = librados::SNAP_HEAD;
+ dummy_clone_info.size = 123;
+
+ librados::snap_set_t dummy_snap_set1;
+ dummy_snap_set1.clones.push_back(dummy_clone_info);
+
+ dummy_clone_info.size = 234;
+ librados::snap_set_t dummy_snap_set2;
+ dummy_snap_set2.clones.push_back(dummy_clone_info);
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, dummy_snap_set1);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[3]);
+ expect_sparse_read(mock_remote_io_ctx, 0, 123, -ENOENT);
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, dummy_snap_set2);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[3]);
+ expect_sparse_read(mock_remote_io_ctx, 0, 234, -ENOENT);
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[3]);
+ expect_sparse_read(mock_remote_io_ctx, 0, one.range_end(), 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_write(mock_local_io_ctx, 0, one.range_end(),
+ {m_local_snap_ids[1], {m_local_snap_ids[1],
+ m_local_snap_ids[0]}},
+ 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[2], OBJECT_EXISTS, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[3], OBJECT_EXISTS_CLEAN, 0);
+
+ request->send();
+ ASSERT_EQ(0, ctx.wait());
+ ASSERT_EQ(0, compare_objects());
+}
+
+TEST_F(TestMockImageSyncObjectCopyRequest, ReadMissingUpToDateSnapMap) {
+ // scribble some data
+ interval_set<uint64_t> one;
+ scribble(m_remote_image_ctx, 10, 102400, &one);
+
+ ASSERT_EQ(0, create_snap("sync"));
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[0]);
+ expect_sparse_read(mock_remote_io_ctx, 0, one.range_end(), -ENOENT);
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+
+ request->send();
+ ASSERT_EQ(-ENOENT, ctx.wait());
+}
+
+TEST_F(TestMockImageSyncObjectCopyRequest, ReadError) {
+ // scribble some data
+ interval_set<uint64_t> one;
+ scribble(m_remote_image_ctx, 10, 102400, &one);
+
+ ASSERT_EQ(0, create_snap("sync"));
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[0]);
+ expect_sparse_read(mock_remote_io_ctx, 0, one.range_end(), -EINVAL);
+
+ request->send();
+ ASSERT_EQ(-EINVAL, ctx.wait());
+}
+
+TEST_F(TestMockImageSyncObjectCopyRequest, WriteError) {
+ // scribble some data
+ interval_set<uint64_t> one;
+ scribble(m_remote_image_ctx, 10, 102400, &one);
+
+ ASSERT_EQ(0, create_snap("sync"));
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+ librados::MockTestMemIoCtxImpl &mock_local_io_ctx(get_mock_io_ctx(
+ request->get_local_io_ctx()));
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[0]);
+ expect_sparse_read(mock_remote_io_ctx, 0, one.range_end(), 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_write(mock_local_io_ctx, 0, one.range_end(), {0, {}}, -EINVAL);
+
+ request->send();
+ ASSERT_EQ(-EINVAL, ctx.wait());
+}
+
+TEST_F(TestMockImageSyncObjectCopyRequest, WriteSnaps) {
+ // scribble some data
+ interval_set<uint64_t> one;
+ scribble(m_remote_image_ctx, 10, 102400, &one);
+ ASSERT_EQ(0, create_snap("one"));
+
+ interval_set<uint64_t> two;
+ scribble(m_remote_image_ctx, 10, 102400, &two);
+ ASSERT_EQ(0, create_snap("two"));
+
+ if (one.range_end() < two.range_end()) {
+ interval_set<uint64_t> resize_diff;
+ resize_diff.insert(one.range_end(), two.range_end() - one.range_end());
+ two.union_of(resize_diff);
+ }
+
+ ASSERT_EQ(0, create_snap("sync"));
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+ librados::MockTestMemIoCtxImpl &mock_local_io_ctx(get_mock_io_ctx(
+ request->get_local_io_ctx()));
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[0]);
+ expect_sparse_read(mock_remote_io_ctx, 0, one.range_end(), 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_write(mock_local_io_ctx, 0, one.range_end(), {0, {}}, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[2]);
+ expect_sparse_read(mock_remote_io_ctx, two, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_write(mock_local_io_ctx, two,
+ {m_local_snap_ids[0], {m_local_snap_ids[0]}}, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[0], OBJECT_EXISTS, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[1], OBJECT_EXISTS, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[2], OBJECT_EXISTS_CLEAN, 0);
+
+ request->send();
+ ASSERT_EQ(0, ctx.wait());
+ ASSERT_EQ(0, compare_objects());
+}
+
+TEST_F(TestMockImageSyncObjectCopyRequest, Trim) {
+ ASSERT_EQ(0, m_remote_image_ctx->operations->metadata_set(
+ "conf_rbd_skip_partial_discard", "false"));
+ // scribble some data
+ interval_set<uint64_t> one;
+ scribble(m_remote_image_ctx, 10, 102400, &one);
+ ASSERT_EQ(0, create_snap("one"));
+
+ // trim the object
+ uint64_t trim_offset = rand() % one.range_end();
+ ASSERT_LE(0, m_remote_image_ctx->io_work_queue->discard(
+ trim_offset, one.range_end() - trim_offset, m_remote_image_ctx->skip_partial_discard));
+ ASSERT_EQ(0, create_snap("sync"));
+
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+ librados::MockTestMemIoCtxImpl &mock_local_io_ctx(get_mock_io_ctx(
+ request->get_local_io_ctx()));
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[0]);
+ expect_sparse_read(mock_remote_io_ctx, 0, one.range_end(), 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_write(mock_local_io_ctx, 0, one.range_end(), {0, {}}, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_truncate(mock_local_io_ctx, trim_offset, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[0], OBJECT_EXISTS, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[1], OBJECT_EXISTS, 0);
+
+ request->send();
+ ASSERT_EQ(0, ctx.wait());
+ ASSERT_EQ(0, compare_objects());
+}
+
+TEST_F(TestMockImageSyncObjectCopyRequest, Remove) {
+ // scribble some data
+ interval_set<uint64_t> one;
+ scribble(m_remote_image_ctx, 10, 102400, &one);
+ ASSERT_EQ(0, create_snap("one"));
+ ASSERT_EQ(0, create_snap("two"));
+
+ // remove the object
+ uint64_t object_size = 1 << m_remote_image_ctx->order;
+ ASSERT_LE(0, m_remote_image_ctx->io_work_queue->discard(0, object_size, m_remote_image_ctx->skip_partial_discard));
+ ASSERT_EQ(0, create_snap("sync"));
+ librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
+ librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
+
+ librbd::MockExclusiveLock mock_exclusive_lock;
+ mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
+
+ librbd::MockObjectMap mock_object_map;
+ mock_local_image_ctx.object_map = &mock_object_map;
+
+ expect_test_features(mock_local_image_ctx);
+
+ C_SaferCond ctx;
+ MockObjectCopyRequest *request = create_request(mock_remote_image_ctx,
+ mock_local_image_ctx, &ctx);
+
+ librados::MockTestMemIoCtxImpl &mock_remote_io_ctx(get_mock_io_ctx(
+ request->get_remote_io_ctx()));
+ librados::MockTestMemIoCtxImpl &mock_local_io_ctx(get_mock_io_ctx(
+ request->get_local_io_ctx()));
+
+ InSequence seq;
+ expect_list_snaps(mock_remote_image_ctx, mock_remote_io_ctx, 0);
+ expect_set_snap_read(mock_remote_io_ctx, m_remote_snap_ids[1]);
+ expect_sparse_read(mock_remote_io_ctx, 0, one.range_end(), 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_write(mock_local_io_ctx, 0, one.range_end(), {0, {}}, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_remove(mock_local_io_ctx, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[0], OBJECT_EXISTS, 0);
+ expect_start_op(mock_exclusive_lock);
+ expect_update_object_map(mock_local_image_ctx, mock_object_map,
+ m_local_snap_ids[1], OBJECT_EXISTS_CLEAN, 0);
+
+ request->send();
+ ASSERT_EQ(0, ctx.wait());
+ ASSERT_EQ(0, compare_objects());
+}
+
+} // namespace image_sync
+} // namespace mirror
+} // namespace rbd
Code Review / stor4nfv.git / blobdiff