// This file Copyright (C) 2013-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 // strlen() // #include // sync() #include "transmission.h" #include "blocklist.h" #include "file.h" #include "net.h" #include "peer-socket.h" #include "session.h" // tr_sessionIsAddressBlocked() #include "tr-strbuf.h" #include "utils.h" #include "test-fixtures.h" namespace libtransmission { namespace test { class BlocklistTest : public SessionTest { protected: static char const constexpr* const Contents1 = "10.5.6.7/8\n" "Austin Law Firm:216.16.1.144-216.16.1.151\n" "Sargent Controls and Aerospace:216.19.18.0-216.19.18.255\n" "Corel Corporation:216.21.157.192-216.21.157.223\n" "Fox Speed Channel:216.79.131.192-216.79.131.223\n"; static char const constexpr* const Contents2 = "10.5.6.7/8\n" "Austin Law Firm:216.16.1.144-216.16.1.151\n" "Sargent Controls and Aerospace:216.19.18.0-216.19.18.255\n" "Corel Corporation:216.21.157.192-216.21.157.223\n" "Fox Speed Channel:216.79.131.192-216.79.131.223\n" "Evilcorp:216.88.88.0-216.88.88.255\n"; #if 0 void createFileWithContents(char const* path, char const* contents) { auto const dir = tr_sys_path_dirname(path); tr_sys_dir_create(dir, TR_SYS_DIR_CREATE_PARENTS, 0700); auto const fd = tr_sys_file_open(path, TR_SYS_FILE_WRITE | TR_SYS_FILE_CREATE | TR_SYS_FILE_TRUNCATE, 0600); blockingFileWrite(fd, contents, strlen(contents)); tr_sys_file_close(fd); sync(); } #endif bool addressIsBlocked(char const* address_str) { struct tr_address addr = {}; return !tr_address_from_string(&addr, address_str) || tr_sessionIsAddressBlocked(session_, &addr); } }; TEST_F(BlocklistTest, parsing) { EXPECT_EQ(0U, tr_blocklistGetRuleCount(session_)); // init the blocklist auto const path = tr_pathbuf{ tr_sessionGetConfigDir(session_), "/blocklists/level1" }; createFileWithContents(path, Contents1); tr_sessionReloadBlocklists(session_); EXPECT_TRUE(tr_blocklistExists(session_)); EXPECT_EQ(size_t{ 5 }, tr_blocklistGetRuleCount(session_)); // enable the blocklist EXPECT_FALSE(tr_blocklistIsEnabled(session_)); tr_blocklistSetEnabled(session_, true); EXPECT_TRUE(tr_blocklistIsEnabled(session_)); // test blocked addresses EXPECT_FALSE(addressIsBlocked("0.0.0.1")); EXPECT_TRUE(addressIsBlocked("10.1.2.3")); EXPECT_FALSE(addressIsBlocked("216.16.1.143")); EXPECT_TRUE(addressIsBlocked("216.16.1.144")); EXPECT_TRUE(addressIsBlocked("216.16.1.145")); EXPECT_TRUE(addressIsBlocked("216.16.1.146")); EXPECT_TRUE(addressIsBlocked("216.16.1.147")); EXPECT_TRUE(addressIsBlocked("216.16.1.148")); EXPECT_TRUE(addressIsBlocked("216.16.1.149")); EXPECT_TRUE(addressIsBlocked("216.16.1.150")); EXPECT_TRUE(addressIsBlocked("216.16.1.151")); EXPECT_FALSE(addressIsBlocked("216.16.1.152")); EXPECT_FALSE(addressIsBlocked("216.16.1.153")); EXPECT_FALSE(addressIsBlocked("217.0.0.1")); EXPECT_FALSE(addressIsBlocked("255.0.0.1")); } /*** **** ***/ TEST_F(BlocklistTest, updating) { // init the session auto const path = tr_pathbuf{ tr_sessionGetConfigDir(session_), "/blocklists/level1" }; // no blocklist to start with... EXPECT_EQ(0U, tr_blocklistGetRuleCount(session_)); // test that updated source files will get loaded createFileWithContents(path, Contents1); tr_sessionReloadBlocklists(session_); EXPECT_EQ(5U, tr_blocklistGetRuleCount(session_)); // test that updated source files will get loaded createFileWithContents(path, Contents2); tr_sessionReloadBlocklists(session_); EXPECT_EQ(6U, tr_blocklistGetRuleCount(session_)); // test that updated source files will get loaded createFileWithContents(path, Contents1); tr_sessionReloadBlocklists(session_); EXPECT_EQ(5U, tr_blocklistGetRuleCount(session_)); // ensure that new files, if bad, get skipped createFileWithContents(path, "# nothing useful\n"); tr_sessionReloadBlocklists(session_); EXPECT_EQ(5U, tr_blocklistGetRuleCount(session_)); // cleanup } } // namespace test } // namespace libtransmission