/*
 * 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 <assert.h>
#include <stdio.h>
#include <string.h> /* 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));
}