Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 17b5a443 authored by Akilesh Kailash's avatar Akilesh Kailash
Browse files

libsnapshot:snapuserd: Fix cow_snapuserd_test. 



This will test the IO operation
between dm-snapshot/dm-user/snapuserd daemon. This test was broken
after cow format and daemon related changes.

Test: cow_snapuserd_test

Bug: 168311203
Signed-off-by: default avatarAkilesh Kailash <akailash@google.com>
Change-Id: I3e5f435332957a2549fd029c84efe13efb98b909
parent f7c47f81

fs_mgr/libsnapshot/cow_snapuserd_test.cpp

+10 −42
Original line number Diff line number Diff line
@@ -41,6 +41,10 @@ using namespace std::chrono_literals; 
class SnapuserdTest : public ::testing::Test {

  protected:

    void SetUp() override {

        // TODO: Daemon started through first stage

        // init does not have permission to read files

        // from /data/nativetest.

        system("setenforce 0");

        cow_system_ = std::make_unique<TemporaryFile>();

        ASSERT_GE(cow_system_->fd, 0) << strerror(errno);
@@ -53,10 +57,6 @@ class SnapuserdTest : public ::testing::Test { 
        cow_product_1_ = std::make_unique<TemporaryFile>();

        ASSERT_GE(cow_product_1_->fd, 0) << strerror(errno);



        // Create temp files in the PWD as selinux

        // allows kernel domin to read from that directory only

        // on userdebug/eng builds. Creating files under /data/local/tmp

        // will have selinux denials.

        std::string path = android::base::GetExecutableDirectory();



        system_a_ = std::make_unique<TemporaryFile>(path);
@@ -65,10 +65,11 @@ class SnapuserdTest : public ::testing::Test { 
        product_a_ = std::make_unique<TemporaryFile>(path);

        ASSERT_GE(product_a_->fd, 0) << strerror(errno);



        size_ = 1_MiB;

        size_ = 100_MiB;

    }



    void TearDown() override {

        system("setenforce 1");

        cow_system_ = nullptr;

        cow_product_ = nullptr;
@@ -116,10 +117,10 @@ class SnapuserdTest : public ::testing::Test { 
    void SwitchSnapshotDevices();



    std::string GetSystemControlPath() {

        return std::string("/dev/dm-user-") + system_device_ctrl_name_;

        return std::string("/dev/dm-user/") + system_device_ctrl_name_;

    }

    std::string GetProductControlPath() {

        return std::string("/dev/dm-user-") + product_device_ctrl_name_;

        return std::string("/dev/dm-user/") + product_device_ctrl_name_;

    }



    void TestIO(unique_fd& snapshot_fd, std::unique_ptr<uint8_t[]>& buffer);
@@ -151,12 +152,12 @@ void SnapuserdTest::Init() { 
        offset += 1_MiB;

    }



    for (size_t j = 0; j < (8_MiB / 1_MiB); j++) {

    for (size_t j = 0; j < (800_MiB / 1_MiB); j++) {

        ASSERT_EQ(ReadFullyAtOffset(rnd_fd, (char*)random_buffer.get(), 1_MiB, 0), true);

        ASSERT_EQ(android::base::WriteFully(system_a_->fd, random_buffer.get(), 1_MiB), true);

    }



    for (size_t j = 0; j < (8_MiB / 1_MiB); j++) {

    for (size_t j = 0; j < (800_MiB / 1_MiB); j++) {

        ASSERT_EQ(ReadFullyAtOffset(rnd_fd, (char*)random_buffer.get(), 1_MiB, 0), true);

        ASSERT_EQ(android::base::WriteFully(product_a_->fd, random_buffer.get(), 1_MiB), true);

    }
@@ -451,42 +452,9 @@ TEST_F(SnapuserdTest, ReadWrite) { 


    snapshot_fd.reset(-1);



    // Sequence of operations for transition

    CreateCowDevice(cow_system_1_);

    CreateCowDevice(cow_product_1_);



    // Create dm-user which creates new control devices

    CreateSystemDmUser(cow_system_1_);

    CreateProductDmUser(cow_product_1_);



    // Send the path information to second stage daemon through vector

    std::vector<std::vector<std::string>> vec{

            {cow_system_1_->path, system_a_loop_->device(), GetSystemControlPath()},

            {cow_product_1_->path, product_a_loop_->device(), GetProductControlPath()}};



    // TODO: This is not switching snapshot device but creates a new table;

    // Second stage daemon will be ready to serve the IO request. From now

    // onwards, we can go ahead and shutdown the first stage daemon

    SwitchSnapshotDevices();



    DeleteDmUser(cow_system_, "system-snapshot");

    DeleteDmUser(cow_product_, "product-snapshot");



    // Test the IO again with the second stage daemon

    snapshot_fd.reset(open("/dev/block/mapper/system-snapshot-1", O_RDONLY));

    ASSERT_TRUE(snapshot_fd > 0);

    TestIO(snapshot_fd, system_buffer_);



    snapshot_fd.reset(open("/dev/block/mapper/product-snapshot-1", O_RDONLY));

    ASSERT_TRUE(snapshot_fd > 0);

    TestIO(snapshot_fd, product_buffer_);



    snapshot_fd.reset(-1);



    DeleteDmUser(cow_system_1_, "system-snapshot-1");

    DeleteDmUser(cow_product_1_, "product-snapshot-1");



    // Stop the second stage daemon

    ASSERT_TRUE(client_->StopSnapuserd());

}