076-scan_actor.cpp 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345
  1. #include "catch.hpp"
  2. #include "test-utils.h"
  3. #include "access.h"
  4. #include "test_supervisor.h"
  5. #include "model/cluster.h"
  6. #include "model/diff/aggregate.h"
  7. #include "model/diff/modify/create_folder.h"
  8. #include "model/diff/modify/share_folder.h"
  9. #include "model/diff/modify/clone_file.h"
  10. #include "hasher/hasher_proxy_actor.h"
  11. #include "hasher/hasher_actor.h"
  12. #include "fs/scan_actor.h"
  13. #include "net/names.h"
  14. #include "utils/error_code.h"
  15. using namespace syncspirit;
  16. using namespace syncspirit::test;
  17. using namespace syncspirit::model;
  18. using namespace syncspirit::net;
  19. using namespace syncspirit::hasher;
  20. struct fixture_t {
  21. using target_ptr_t = r::intrusive_ptr_t<fs::scan_actor_t>;
  22. using error_msg_t = model::message::io_error_t;
  23. using error_msg_ptr_t = r::intrusive_ptr_t<error_msg_t>;
  24. using errors_container_t = std::vector<error_msg_ptr_t>;
  25. fixture_t() noexcept: root_path{ bfs::unique_path() }, path_quard{root_path} {
  26. utils::set_default("trace");
  27. bfs::create_directory(root_path);
  28. }
  29. void run() noexcept {
  30. auto my_id = device_id_t::from_string("KHQNO2S-5QSILRK-YX4JZZ4-7L77APM-QNVGZJT-EKU7IFI-PNEPBMY-4MXFMQD").value();
  31. my_device = device_t::create(my_id, "my-device").value();
  32. auto peer_id = device_id_t::from_string("VUV42CZ-IQD5A37-RPEBPM4-VVQK6E4-6WSKC7B-PVJQHHD-4PZD44V-ENC6WAZ").value();
  33. peer_device = device_t::create(peer_id, "peer-device").value();
  34. cluster = new cluster_t(my_device, 1);
  35. cluster->get_devices().put(my_device);
  36. cluster->get_devices().put(peer_device);
  37. r::system_context_t ctx;
  38. sup = ctx.create_supervisor<supervisor_t>().timeout(timeout).create_registry().finish();
  39. sup->cluster = cluster;
  40. db::Folder db_folder;
  41. db_folder.set_id("1234-5678");
  42. db_folder.set_label("my-f1");
  43. db_folder.set_path(root_path.string());
  44. auto diffs = diff::aggregate_t::diffs_t{};
  45. diffs.push_back(new diff::modify::create_folder_t(db_folder));
  46. diffs.push_back(new diff::modify::share_folder_t(peer_id.get_sha256(), db_folder.id(), 123ul));
  47. auto diff = diff::cluster_diff_ptr_t(new diff::aggregate_t(std::move(diffs)));
  48. REQUIRE(diff->apply(*cluster));
  49. folder = cluster->get_folders().by_id(db_folder.id());
  50. folder_info = folder->get_folder_infos().by_device(my_device);
  51. files = &folder_info->get_file_infos();
  52. folder_info_peer = folder->get_folder_infos().by_device(peer_device);
  53. files_peer = &folder_info_peer->get_file_infos();
  54. sup->configure_callback = [&](r::plugin::plugin_base_t &plugin){
  55. plugin.template with_casted<r::plugin::starter_plugin_t>([&](auto &p) {
  56. p.subscribe_actor(r::lambda<error_msg_t>(
  57. [&](error_msg_t &msg) { errors.push_back(&msg); }));
  58. });};
  59. sup->start();
  60. sup->do_process();
  61. CHECK(static_cast<r::actor_base_t*>(sup.get())->access<to::state>() == r::state_t::OPERATIONAL);
  62. sup->create_actor<hasher_actor_t>().index(1).timeout(timeout).finish();
  63. auto proxy_addr = sup->create_actor<hasher::hasher_proxy_actor_t>()
  64. .timeout(timeout)
  65. .hasher_threads(1)
  66. .name(net::names::hasher_proxy)
  67. .finish()
  68. ->get_address();
  69. sup->do_process();
  70. auto fs_config = config::fs_config_t {
  71. 0,
  72. 3600
  73. };
  74. target = sup->create_actor<fs::scan_actor_t>()
  75. .timeout(timeout)
  76. .cluster(cluster)
  77. .hasher_proxy(proxy_addr)
  78. .fs_config(fs_config)
  79. .requested_hashes_limit(2ul)
  80. .finish();
  81. main();
  82. sup->do_process();
  83. sup->shutdown();
  84. sup->do_process();
  85. CHECK(static_cast<r::actor_base_t*>(sup.get())->access<to::state>() == r::state_t::SHUT_DOWN);
  86. }
  87. virtual void main() noexcept {
  88. }
  89. r::pt::time_duration timeout = r::pt::millisec{10};
  90. r::intrusive_ptr_t<supervisor_t> sup;
  91. cluster_ptr_t cluster;
  92. device_ptr_t my_device;
  93. bfs::path root_path;
  94. path_guard_t path_quard;
  95. target_ptr_t target;
  96. model::folder_ptr_t folder;
  97. model::folder_info_ptr_t folder_info;
  98. model::folder_info_ptr_t folder_info_peer;
  99. model::file_infos_map_t* files;
  100. model::file_infos_map_t* files_peer;
  101. errors_container_t errors;
  102. model::device_ptr_t peer_device;
  103. };
  104. void test_meta_changes() {
  105. struct F : fixture_t {
  106. void main() noexcept override {
  107. sys::error_code ec;
  108. SECTION("no files"){
  109. sup->do_process();
  110. CHECK(folder_info->get_file_infos().size() == 0);
  111. }
  112. SECTION("just 1 dir"){
  113. CHECK(bfs::create_directories(root_path / "abc"));
  114. sup->do_process();
  115. CHECK(folder_info->get_file_infos().size() == 0);
  116. }
  117. SECTION("just 1 subdir, which cannot be read"){
  118. CHECK(bfs::create_directories(root_path / "abc" / "def", ec));
  119. bfs::permissions(root_path / "abc", bfs::perms::no_perms);
  120. sup->do_process();
  121. CHECK(folder_info->get_file_infos().size() == 0);
  122. bfs::permissions(root_path / "abc", bfs::perms::all_all);
  123. REQUIRE(errors.size() == 1);
  124. auto& errs = errors.at(0)->payload.errors;
  125. REQUIRE(errs.size() == 1);
  126. REQUIRE(errs.at(0).path == (root_path / "abc"));
  127. REQUIRE(errs.at(0).ec);
  128. }
  129. proto::FileInfo pr_fi;
  130. std::int64_t modified = 1641828421;
  131. pr_fi.set_name("q.txt");
  132. pr_fi.set_modified_s(modified);
  133. pr_fi.set_block_size(5ul);
  134. pr_fi.set_size(5ul);
  135. auto version = pr_fi.mutable_version();
  136. auto counter = version->add_counters();
  137. counter->set_id(1);
  138. counter->set_value(peer_device->as_uint());
  139. auto bi = proto::BlockInfo();
  140. bi.set_size(5);
  141. bi.set_weak_hash(12);
  142. bi.set_hash(utils::sha256_digest("12345").value());
  143. bi.set_offset(0);
  144. auto b = block_info_t::create(bi).value();
  145. SECTION("a file does not physically exists"){
  146. auto file_peer = file_info_t::create(cluster->next_uuid(), pr_fi, folder_info_peer).value();
  147. file_peer->assign_block(b, 0);
  148. files_peer->put(file_peer);
  149. auto diff = diff::cluster_diff_ptr_t(new diff::modify::clone_file_t(*file_peer));
  150. REQUIRE(diff->apply(*cluster));
  151. auto file = files->by_name(pr_fi.name());
  152. sup->do_process();
  153. CHECK(files->size() == 1);
  154. CHECK(!file->is_locally_available());
  155. }
  156. SECTION("complete file exists") {
  157. auto file_peer = file_info_t::create(cluster->next_uuid(), pr_fi, folder_info_peer).value();
  158. file_peer->assign_block(b, 0);
  159. files_peer->put(file_peer);
  160. auto diff = diff::cluster_diff_ptr_t(new diff::modify::clone_file_t(*file_peer));
  161. REQUIRE(diff->apply(*cluster));
  162. auto file = files->by_name(pr_fi.name());
  163. file->assign_block(b, 0);
  164. file->set_source(nullptr);
  165. auto path = file->get_path();
  166. SECTION("meta is not changed") {
  167. write_file(path, "12345");
  168. bfs::last_write_time(path, modified);
  169. sup->do_process();
  170. CHECK(files->size() == 1);
  171. CHECK(file->is_locally_available());
  172. }
  173. SECTION("meta is changed (modification)") {
  174. write_file(path, "12345");
  175. sup->do_process();
  176. CHECK(files->size() == 1);
  177. CHECK(!file->is_locally_available());
  178. }
  179. SECTION("meta is changed (size)") {
  180. write_file(path, "123456");
  181. bfs::last_write_time(path, modified);
  182. sup->do_process();
  183. CHECK(files->size() == 1);
  184. CHECK(!file->is_locally_available());
  185. }
  186. }
  187. SECTION("incomplete file exists") {
  188. pr_fi.set_size(10ul);
  189. pr_fi.set_block_size(5ul);
  190. auto bi_2 = proto::BlockInfo();
  191. bi_2.set_size(5);
  192. bi_2.set_weak_hash(12);
  193. bi_2.set_hash(utils::sha256_digest("67890").value());
  194. bi_2.set_offset(5);
  195. auto b2 = block_info_t::create(bi_2).value();
  196. auto file_peer = file_info_t::create(cluster->next_uuid(), pr_fi, folder_info_peer).value();
  197. file_peer->assign_block(b, 0);
  198. file_peer->assign_block(b2, 1);
  199. files_peer->put(file_peer);
  200. auto diff = diff::cluster_diff_ptr_t(new diff::modify::clone_file_t(*file_peer));
  201. REQUIRE(diff->apply(*cluster));
  202. auto file = files->by_name(pr_fi.name());
  203. auto path = file->get_path().string() + ".syncspirit-tmp";
  204. auto content = "12345\0\0\0\0\0";
  205. write_file(path, std::string(content, 10));
  206. SECTION("outdated -> just remove") {
  207. bfs::last_write_time(path, modified - 24 * 3600);
  208. sup->do_process();
  209. CHECK(!file->is_locally_available());
  210. CHECK(!bfs::exists(path));
  211. }
  212. SECTION("just 1st block is valid, tmp is kept") {
  213. sup->do_process();
  214. CHECK(!file->is_locally_available());
  215. CHECK(!file->is_locally_available(0));
  216. CHECK(!file->is_locally_available(1));
  217. CHECK(!file_peer->is_locally_available());
  218. CHECK(file_peer->is_locally_available(0));
  219. CHECK(!file_peer->is_locally_available(1));
  220. CHECK(bfs::exists(path));
  221. }
  222. SECTION("corrupted content") {
  223. SECTION("1st block") {
  224. write_file(path, "2234567890");
  225. }
  226. SECTION("2nd block") {
  227. write_file(path, "1234567899");
  228. }
  229. SECTION("missing source file") {
  230. file->set_source(nullptr);
  231. }
  232. sup->do_process();
  233. CHECK(!file->is_locally_available(0));
  234. CHECK(!file->is_locally_available(1));
  235. CHECK(!file_peer->is_locally_available(0));
  236. CHECK(!file_peer->is_locally_available(1));
  237. CHECK(!bfs::exists(path));
  238. }
  239. SECTION("error on reading -> remove") {
  240. bfs::permissions(path, bfs::perms::no_perms);
  241. sup->do_process();
  242. CHECK(!file->is_locally_available());
  243. CHECK(!bfs::exists(path));
  244. REQUIRE(errors.size() == 1);
  245. auto& errs = errors.at(0)->payload.errors;
  246. REQUIRE(errs.size() == 1);
  247. CHECK(errs.at(0).path == path);
  248. CHECK(errs.at(0).ec);
  249. }
  250. }
  251. SECTION("local (previous) file exists") {
  252. pr_fi.set_size(10ul);
  253. pr_fi.set_block_size(5ul);
  254. auto bi_2 = proto::BlockInfo();
  255. bi_2.set_size(5);
  256. bi_2.set_weak_hash(12);
  257. bi_2.set_hash(utils::sha256_digest("67890").value());
  258. bi_2.set_offset(5);
  259. auto b2 = block_info_t::create(bi_2).value();
  260. pr_fi.set_size(5ul);
  261. auto file_my = file_info_t::create(cluster->next_uuid(), pr_fi, folder_info).value();
  262. file_my->assign_block(b, 0);
  263. files->put(file_my);
  264. pr_fi.set_size(10ul);
  265. counter->set_id(2);
  266. auto file_peer = file_info_t::create(cluster->next_uuid(), pr_fi, folder_info_peer).value();
  267. file_peer->assign_block(b, 0);
  268. file_peer->assign_block(b2, 1);
  269. files_peer->put(file_peer);
  270. auto diff = diff::cluster_diff_ptr_t(new diff::modify::clone_file_t(*file_peer));
  271. REQUIRE(diff->apply(*cluster));
  272. auto file = files->by_name(pr_fi.name());
  273. auto path_my = file->get_path().string();
  274. auto path_peer = file->get_path().string() + ".syncspirit-tmp";
  275. write_file(path_my, "12345");
  276. bfs::last_write_time(path_my, modified);
  277. auto content = "12345\0\0\0\0\0";
  278. write_file(path_peer, std::string(content, 10));
  279. sup->do_process();
  280. CHECK(file_my->is_locally_available());
  281. CHECK(file_my->get_source() == file_peer);
  282. CHECK(!file_peer->is_locally_available());
  283. CHECK(file_peer->is_locally_available(0));
  284. CHECK(!file_peer->is_locally_available(1));
  285. }
  286. }
  287. };
  288. F().run();
  289. }
  290. REGISTER_TEST_CASE(test_meta_changes, "test_meta_changes", "[fs]");