342 lines
17 KiB
C++
342 lines
17 KiB
C++
// This file Copyright (C) 2022 Mnemosyne LLC.
|
|
// It may be used under GPLv2 (SPDX: GPL-2.0), GPLv3 (SPDX: GPL-3.0),
|
|
// or any future license endorsed by Mnemosyne LLC.
|
|
// License text can be found in the licenses/ folder.
|
|
|
|
#include <array>
|
|
#include <cstdio>
|
|
#include <set>
|
|
#include <string_view>
|
|
#include <utility>
|
|
|
|
#include "transmission.h"
|
|
|
|
#include "file.h"
|
|
#include "torrent-files.h"
|
|
#include "tr-strbuf.h"
|
|
|
|
#include "test-fixtures.h"
|
|
|
|
using namespace std::literals;
|
|
using SubpathAndSize = std::pair<std::string_view, uint64_t>;
|
|
|
|
class RemoveTest : public libtransmission::test::SandboxedTest
|
|
{
|
|
protected:
|
|
static constexpr std::string_view Content = "Hello, World!"sv;
|
|
static constexpr std::string_view JunkBasename = ".DS_Store"sv;
|
|
static constexpr std::string_view NonJunkBasename = "passwords.txt"sv;
|
|
|
|
static void sysPathRemove(char const* filename)
|
|
{
|
|
tr_sys_path_remove(filename, nullptr);
|
|
}
|
|
|
|
static auto aliceFiles()
|
|
{
|
|
static constexpr std::array<SubpathAndSize, 106> AliceFiles = { {
|
|
{ "alice_in_wonderland_librivox/AliceInWonderland_librivox.m4b", 87525736ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland.jpg", 81464ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland.pdf", 185367ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_abbyy.gz", 24582ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_chocr.html.gz", 22527ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_djvu.txt", 2039ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_djvu.xml", 28144ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_hocr.html", 56942ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_hocr_pageindex.json.gz", 40ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_hocr_searchtext.txt.gz", 943ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_jp2.zip", 1499986ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_page_numbers.json", 136ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_scandata.xml", 538ULL },
|
|
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_thumb.jpg", 26987ULL },
|
|
{ "alice_in_wonderland_librivox/__ia_thumb.jpg", 16557ULL },
|
|
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox.json", 13740ULL },
|
|
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox.storj-store.trigger", 0ULL },
|
|
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox_128kb.m3u", 984ULL },
|
|
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox_64kb.m3u", 1044ULL },
|
|
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox_meta.sqlite", 20480ULL },
|
|
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox_meta.xml", 2805ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_01.mp3", 10249859ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_01.ogg", 7509828ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_01.png", 10779ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_01_64kb.mp3", 5124992ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_01_esshigh.json.gz", 1977ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_01_esslow.json.gz", 29258ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_01_spectrogram.png", 234022ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_02.mp3", 11772312ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_02.ogg", 5148365ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_02.png", 10962ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_02_64kb.mp3", 5886455ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_02_esshigh.json.gz", 1980ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_02_esslow.json.gz", 30287ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_02_spectrogram.png", 326161ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_03.mp3", 17024560ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_03.ogg", 12046177ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_03.png", 8725ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_03_64kb.mp3", 8512448ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_03_esshigh.json.gz", 1966ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_03_esslow.json.gz", 34110ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_03_spectrogram.png", 218264ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_04.mp3", 19087768ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_04.ogg", 9880920ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_04.png", 6055ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_04_64kb.mp3", 9544016ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_04_esshigh.json.gz", 1967ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_04_esslow.json.gz", 36154ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_04_spectrogram.png", 282145ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_05.mp3", 12946949ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_05.ogg", 7470734ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_05.png", 12061ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_05_64kb.mp3", 6473687ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_05_esshigh.json.gz", 1963ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_05_esslow.json.gz", 31048ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_05_spectrogram.png", 304150ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_06.mp3", 12413304ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_06.ogg", 7154040ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_06.png", 11383ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_06_64kb.mp3", 6206820ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_06_esshigh.json.gz", 1942ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_06_esslow.json.gz", 30295ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_06_spectrogram.png", 288601ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_07.mp3", 16742808ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_07.ogg", 10513847ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_07.png", 8180ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_07_64kb.mp3", 8371136ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_07_esshigh.json.gz", 1963ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_07_esslow.json.gz", 33992ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_07_spectrogram.png", 233725ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_08.mp3", 12784781ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_08.ogg", 9306961ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_08.png", 11470ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_08_64kb.mp3", 6392576ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_08_esshigh.json.gz", 1973ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_08_esslow.json.gz", 31964ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_08_spectrogram.png", 233626ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_09.mp3", 14528920ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_09.ogg", 8062952ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_09.png", 9439ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_09_64kb.mp3", 7264466ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_09_esshigh.json.gz", 1946ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_09_esslow.json.gz", 32295ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_09_spectrogram.png", 282898ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_10.mp3", 21894203ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_10.ogg", 15226220ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_10.png", 8796ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_10_64kb.mp3", 10947200ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_10_esshigh.json.gz", 1970ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_10_esslow.json.gz", 37062ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_10_spectrogram.png", 221277ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_11.mp3", 9919894ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_11.ogg", 7676067ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_11.png", 7140ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_11_64kb.mp3", 4959296ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_11_esshigh.json.gz", 1959ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_11_esslow.json.gz", 28694ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_11_spectrogram.png", 229471ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_12.mp3", 12359368ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_12.ogg", 8065179ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_12.png", 11074ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_12_64kb.mp3", 6179840ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_12_esshigh.json.gz", 1981ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_12_esslow.json.gz", 30975ULL },
|
|
{ "alice_in_wonderland_librivox/wonderland_ch_12_spectrogram.png", 235568ULL },
|
|
{ "alice_in_wonderland_librivox/history/files/alice_in_wonderland_librivox.storj-store.trigger.~1~", 0ULL },
|
|
} };
|
|
|
|
auto files = tr_torrent_files{};
|
|
|
|
for (auto const& file : AliceFiles)
|
|
{
|
|
auto const& [filename, size] = file;
|
|
files.add(filename, size);
|
|
}
|
|
|
|
return files;
|
|
}
|
|
|
|
static auto ubuntuFiles()
|
|
{
|
|
static auto constexpr Files = std::array<SubpathAndSize, 1>{ { { "ubuntu-20.04.4-desktop-amd64.iso"sv,
|
|
3379068928ULL } } };
|
|
|
|
auto files = tr_torrent_files{};
|
|
|
|
for (auto const& file : Files)
|
|
{
|
|
auto const& [filename, size] = file;
|
|
files.add(filename, size);
|
|
}
|
|
|
|
return files;
|
|
}
|
|
|
|
auto createFiles(tr_torrent_files const& files, char const* parent)
|
|
{
|
|
auto paths = std::set<std::string>{};
|
|
|
|
for (tr_file_index_t i = 0, n = files.fileCount(); i < n; ++i)
|
|
{
|
|
auto walk = tr_pathbuf{ parent, '/', files.path(i) };
|
|
createFileWithContents(walk, std::data(Content), std::size(Content));
|
|
paths.emplace(walk);
|
|
|
|
while (!tr_sys_path_is_same(parent, walk))
|
|
{
|
|
walk.popdir();
|
|
paths.emplace(walk);
|
|
}
|
|
}
|
|
|
|
return paths;
|
|
}
|
|
|
|
static auto getSubtreeContents(std::string_view parent_dir)
|
|
{
|
|
auto filenames = std::set<std::string>{};
|
|
|
|
auto file_func = [&filenames](char const* filename)
|
|
{
|
|
filenames.emplace(filename);
|
|
};
|
|
|
|
libtransmission::test::depthFirstWalk(tr_pathbuf{ parent_dir }, file_func);
|
|
|
|
return filenames;
|
|
}
|
|
};
|
|
|
|
TEST_F(RemoveTest, RemovesSingleFile)
|
|
{
|
|
auto const parent = sandboxDir();
|
|
auto expected_tree = std::set<std::string>{ parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
auto const files = ubuntuFiles();
|
|
expected_tree = createFiles(files, parent.c_str());
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
|
|
expected_tree = { parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
}
|
|
|
|
TEST_F(RemoveTest, RemovesSubtree)
|
|
{
|
|
auto const parent = sandboxDir();
|
|
auto expected_tree = std::set<std::string>{ parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
auto const files = aliceFiles();
|
|
expected_tree = createFiles(files, parent.c_str());
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
|
|
expected_tree = { parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
}
|
|
|
|
TEST_F(RemoveTest, RemovesLeftoverJunk)
|
|
{
|
|
auto const parent = sandboxDir();
|
|
auto expected_tree = std::set<std::string>{ parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
auto const files = aliceFiles();
|
|
expected_tree = createFiles(files, parent.c_str());
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
// add a junk file *inside of* the torrent's top directory.
|
|
auto const junk_file = tr_pathbuf{ parent, "/alice_in_wonderland_librivox/", JunkBasename };
|
|
createFileWithContents(junk_file, std::data(Content), std::size(Content));
|
|
expected_tree.emplace(junk_file);
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
|
|
expected_tree = { parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
}
|
|
|
|
TEST_F(RemoveTest, LeavesSiblingsAlone)
|
|
{
|
|
auto const parent = sandboxDir();
|
|
auto expected_tree = std::set<std::string>{ parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
auto const files = aliceFiles();
|
|
expected_tree = createFiles(files, parent.c_str());
|
|
EXPECT_GT(std::size(expected_tree), 100U);
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
// add a junk file *as a sibling of* the torrent's top directory.
|
|
auto const junk_file = tr_pathbuf{ parent, '/', JunkBasename };
|
|
createFileWithContents(junk_file, std::data(Content), std::size(Content));
|
|
expected_tree.emplace(junk_file);
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
// add a non-junk file *as a sibling of* the torrent's top directory.
|
|
auto const non_junk_file = tr_pathbuf{ parent, '/', NonJunkBasename };
|
|
createFileWithContents(non_junk_file, std::data(Content), std::size(Content));
|
|
expected_tree.emplace(non_junk_file);
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
|
|
expected_tree = { parent, junk_file.c_str(), non_junk_file.c_str() };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
}
|
|
|
|
TEST_F(RemoveTest, LeavesNonJunkAlone)
|
|
{
|
|
auto const parent = sandboxDir();
|
|
auto expected_tree = std::set<std::string>{ parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
auto const files = aliceFiles();
|
|
expected_tree = createFiles(files, parent.c_str());
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
// add a non-junk file.
|
|
auto const nonjunk_file = tr_pathbuf{ parent, "/alice_in_wonderland_librivox/", NonJunkBasename };
|
|
createFileWithContents(nonjunk_file, std::data(Content), std::size(Content));
|
|
expected_tree.emplace(nonjunk_file);
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
|
|
expected_tree = { parent, std::string{ tr_sys_path_dirname(nonjunk_file) }, nonjunk_file.c_str() };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
}
|
|
|
|
TEST_F(RemoveTest, PreservesDirectoryHierarchyIfPossible)
|
|
{
|
|
auto const parent = sandboxDir();
|
|
auto expected_tree = std::set<std::string>{ parent };
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
// add a recycle bin
|
|
auto const recycle_bin = tr_pathbuf{ parent, "/Trash"sv };
|
|
tr_sys_dir_create(recycle_bin, TR_SYS_DIR_CREATE_PARENTS, 0777);
|
|
expected_tree.emplace(recycle_bin);
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
auto const files = aliceFiles();
|
|
expected_tree = createFiles(files, parent.c_str());
|
|
expected_tree.emplace(recycle_bin);
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
|
|
auto const recycle_func = [&recycle_bin](char const* filename)
|
|
{
|
|
tr_sys_path_rename(filename, tr_pathbuf{ recycle_bin, '/', tr_sys_path_basename(filename) });
|
|
};
|
|
files.remove(parent, "tmpdir_prefix"sv, recycle_func);
|
|
|
|
// after remove, the subtree should be:
|
|
expected_tree = { parent, recycle_bin.c_str() };
|
|
for (tr_file_index_t i = 0, n = files.fileCount(); i < n; ++i)
|
|
{
|
|
expected_tree.emplace(tr_pathbuf{ recycle_bin, '/', files.path(i) });
|
|
}
|
|
expected_tree.emplace(tr_pathbuf{ recycle_bin, "/alice_in_wonderland_librivox"sv });
|
|
expected_tree.emplace(tr_pathbuf{ recycle_bin, "/alice_in_wonderland_librivox/history"sv });
|
|
expected_tree.emplace(tr_pathbuf{ recycle_bin, "/alice_in_wonderland_librivox/history/files"sv });
|
|
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
|
|
}
|