transmission/libtransmission/blocklist-test.c

87 lines
2.9 KiB
C

#include <stdio.h>
#include "transmission.h"
#include "blocklist.h"
#include "net.h"
#include "utils.h"
#include "libtransmission-test.h"
#ifndef WIN32
#define TEMPDIR_PREFIX "/tmp/"
#else
#define TEMPDIR_PREFIX
#endif
#define TEMPFILE_TXT TEMPDIR_PREFIX "transmission-blocklist-test.txt"
#define TEMPFILE_BIN TEMPDIR_PREFIX "transmission-blocklist-test.bin"
static void
createTestBlocklist (const char * tmpfile)
{
const char * lines[] = { "Austin Law Firm:216.16.1.144-216.16.1.151",
"Sargent Controls and Aerospace:216.19.18.0-216.19.18.255",
"Corel Corporation:216.21.157.192-216.21.157.223",
"Fox Speed Channel:216.79.131.192-216.79.131.223" };
FILE * out;
int i;
const int lineCount = sizeof (lines) / sizeof (lines[0]);
/* create the ascii file to feed to libtransmission */
out = fopen (tmpfile, "w+");
for (i = 0; i < lineCount; ++i)
fprintf (out, "%s\n", lines[i]);
fclose (out);
}
static int
testBlockList (void)
{
const char * tmpfile_txt = TEMPFILE_TXT;
const char * tmpfile_bin = TEMPFILE_BIN;
struct tr_address addr;
tr_blocklist * b;
remove (tmpfile_txt);
remove (tmpfile_bin);
b = _tr_blocklistNew (tmpfile_bin, true);
createTestBlocklist (tmpfile_txt);
_tr_blocklistSetContent (b, tmpfile_txt);
/* now run some tests */
check (tr_address_from_string (&addr, "216.16.1.143"));
check (!_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.144"));
check (_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.145"));
check (_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.146"));
check (_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.147"));
check (_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.148"));
check (_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.149"));
check (_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.150"));
check (_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.151"));
check (_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.152"));
check (!_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "216.16.1.153"));
check (!_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "217.0.0.1"));
check (!_tr_blocklistHasAddress (b, &addr));
check (tr_address_from_string (&addr, "255.0.0.1"));
/* cleanup */
_tr_blocklistFree (b);
remove (tmpfile_txt);
remove (tmpfile_bin);
return 0;
}
MAIN_SINGLE_TEST (testBlockList)