/* * This file Copyright (C) 2013-2014 Mnemosyne LLC * * It may be used under the GNU GPL versions 2 or 3 * or any future license endorsed by Mnemosyne LLC. * */ #include #include #include /* strlen() */ #include "transmission.h" #include "blocklist.h" #include "file.h" #include "net.h" #include "session.h" /* tr_sessionIsAddressBlocked() */ #include "utils.h" #include "libtransmission-test.h" static char const* contents1 = "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* contents2 = "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"; static void create_text_file(char const* path, char const* contents) { tr_sys_file_t fd; char* dir; dir = tr_sys_path_dirname(path, NULL); tr_sys_dir_create(dir, TR_SYS_DIR_CREATE_PARENTS, 0700, NULL); tr_free(dir); fd = tr_sys_file_open(path, TR_SYS_FILE_WRITE | TR_SYS_FILE_CREATE | TR_SYS_FILE_TRUNCATE, 0600, NULL); tr_sys_file_write(fd, contents, strlen(contents), NULL, NULL); tr_sys_file_close(fd, NULL); libttest_sync(); } static bool address_is_blocked(tr_session* session, char const* address_str) { struct tr_address addr; tr_address_from_string(&addr, address_str); return tr_sessionIsAddressBlocked(session, &addr); } static int test_parsing(void) { char* path; tr_session* session; /* init the session */ session = libttest_session_init(NULL); check(!tr_blocklistExists(session)); check_int_eq(0, tr_blocklistGetRuleCount(session)); /* init the blocklist */ path = tr_buildPath(tr_sessionGetConfigDir(session), "blocklists", "level1", NULL); create_text_file(path, contents1); tr_free(path); tr_sessionReloadBlocklists(session); check(tr_blocklistExists(session)); check_int_eq(4, tr_blocklistGetRuleCount(session)); /* enable the blocklist */ check(!tr_blocklistIsEnabled(session)); tr_blocklistSetEnabled(session, true); check(tr_blocklistIsEnabled(session)); /* test blocked addresses */ check(!address_is_blocked(session, "216.16.1.143")); check(address_is_blocked(session, "216.16.1.144")); check(address_is_blocked(session, "216.16.1.145")); check(address_is_blocked(session, "216.16.1.146")); check(address_is_blocked(session, "216.16.1.147")); check(address_is_blocked(session, "216.16.1.148")); check(address_is_blocked(session, "216.16.1.149")); check(address_is_blocked(session, "216.16.1.150")); check(address_is_blocked(session, "216.16.1.151")); check(!address_is_blocked(session, "216.16.1.152")); check(!address_is_blocked(session, "216.16.1.153")); check(!address_is_blocked(session, "217.0.0.1")); check(!address_is_blocked(session, "255.0.0.1")); /* cleanup */ libttest_session_close(session); return 0; } /*** **** ***/ static int test_updating(void) { char* path; tr_session* session; /* init the session */ session = libttest_session_init(NULL); path = tr_buildPath(tr_sessionGetConfigDir(session), "blocklists", "level1", NULL); /* no blocklist to start with... */ check_int_eq(0, tr_blocklistGetRuleCount(session)); /* test that updated source files will get loaded */ create_text_file(path, contents1); tr_sessionReloadBlocklists(session); check_int_eq(4, tr_blocklistGetRuleCount(session)); /* test that updated source files will get loaded */ create_text_file(path, contents2); tr_sessionReloadBlocklists(session); check_int_eq(5, tr_blocklistGetRuleCount(session)); /* test that updated source files will get loaded */ create_text_file(path, contents1); tr_sessionReloadBlocklists(session); check_int_eq(4, tr_blocklistGetRuleCount(session)); /* ensure that new files, if bad, get skipped */ create_text_file(path, "# nothing useful\n"); tr_sessionReloadBlocklists(session); check_int_eq(4, tr_blocklistGetRuleCount(session)); /* cleanup */ libttest_session_close(session); tr_free(path); return 0; } /*** **** ***/ int main(void) { testFunc const tests[] = { test_parsing, test_updating }; return runTests(tests, NUM_TESTS(tests)); }