transmission/libtransmission/utils-test.c

504 lines
12 KiB
C
Raw Normal View History

#include <math.h> /* sqrt () */
#include <string.h> /* strlen () */
#include "transmission.h"
#include "bitfield.h"
#include "ConvertUTF.h" /* tr_utf8_validate*/
#include "platform.h"
2008-09-06 02:09:53 +00:00
#include "crypto.h"
#include "utils.h"
#include "web.h"
/* #define VERBOSE */
#undef VERBOSE
#define NUM_LOOPS 1
#define SPEED_TEST 0
#if SPEED_TEST
#define VERBOSE
#undef NUM_LOOPS
#define NUM_LOOPS 200
#endif
#include "libtransmission-test.h"
static int
test_base64 (void)
{
char *in, *out;
int len;
/* base64 */
out = tr_base64_encode ("YOYO!", -1, &len);
check_streq ("WU9ZTyE=", out);
check_int_eq (8, len);
in = tr_base64_decode (out, -1, &len);
check_streq ("YOYO!", in);
check_int_eq (5, len);
tr_free (in);
tr_free (out);
out = tr_base64_encode (NULL, 0, &len);
check (out == NULL);
check_int_eq (0, len);
return 0;
}
static int
test_bitfield_count_range (void)
{
int i;
int n;
int begin;
int end;
int count1;
int count2;
const int bitCount = 100 + tr_cryptoWeakRandInt (1000);
tr_bitfield bf;
/* generate a random bitfield */
tr_bitfieldConstruct (&bf, bitCount);
for (i=0, n=tr_cryptoWeakRandInt (bitCount); i<n; ++i)
tr_bitfieldAdd (&bf, tr_cryptoWeakRandInt (bitCount));
begin = tr_cryptoWeakRandInt (bitCount);
do {
end = tr_cryptoWeakRandInt (bitCount);
} while (end == begin);
if (end < begin) {
const int tmp = begin;
begin = end;
end = tmp;
}
count1 = 0;
for (i=begin; i<end; ++i)
if (tr_bitfieldHas (&bf, i))
++count1;
count2 = tr_bitfieldCountRange (&bf, begin, end);
check (count1 == count2);
tr_bitfieldDestruct (&bf);
return 0;
}
static int
test_bitfields (void)
{
unsigned int i;
unsigned int bitcount = 500;
tr_bitfield field;
tr_bitfieldConstruct (&field, bitcount);
/* test tr_bitfieldAdd */
for (i = 0; i < bitcount; ++i)
if (! (i % 7))
tr_bitfieldAdd (&field, i);
for (i = 0; i < bitcount; ++i)
check (tr_bitfieldHas (&field, i) == (! (i % 7)));
/* test tr_bitfieldAddRange */
tr_bitfieldAddRange (&field, 0, bitcount);
for (i = 0; i < bitcount; ++i)
check (tr_bitfieldHas (&field, i));
/* test tr_bitfieldRemRange in the middle of a boundary */
tr_bitfieldRemRange (&field, 4, 21);
for (i = 0; i < 64; ++i)
check (tr_bitfieldHas (&field, i) == ((i < 4) || (i >= 21)));
/* test tr_bitfieldRemRange on the boundaries */
tr_bitfieldAddRange (&field, 0, 64);
tr_bitfieldRemRange (&field, 8, 24);
for (i = 0; i < 64; ++i)
check (tr_bitfieldHas (&field, i) == ((i < 8) || (i >= 24)));
/* test tr_bitfieldRemRange when begin & end is on the same word */
tr_bitfieldAddRange (&field, 0, 64);
tr_bitfieldRemRange (&field, 4, 5);
for (i = 0; i < 64; ++i)
check (tr_bitfieldHas (&field, i) == ((i < 4) || (i >= 5)));
/* test tr_bitfieldAddRange */
tr_bitfieldRemRange (&field, 0, 64);
tr_bitfieldAddRange (&field, 4, 21);
for (i = 0; i < 64; ++i)
check (tr_bitfieldHas (&field, i) == ((4 <= i) && (i < 21)));
/* test tr_bitfieldAddRange on the boundaries */
tr_bitfieldRemRange (&field, 0, 64);
tr_bitfieldAddRange (&field, 8, 24);
for (i = 0; i < 64; ++i)
check (tr_bitfieldHas (&field, i) == ((8 <= i) && (i < 24)));
/* test tr_bitfieldAddRange when begin & end is on the same word */
tr_bitfieldRemRange (&field, 0, 64);
tr_bitfieldAddRange (&field, 4, 5);
for (i = 0; i < 64; ++i)
check (tr_bitfieldHas (&field, i) == ((4 <= i) && (i < 5)));
tr_bitfieldDestruct (&field);
return 0;
}
static int
test_strip_positional_args (void)
{
const char * in;
const char * out;
const char * expected;
in = "Hello %1$s foo %2$.*f";
expected = "Hello %s foo %.*f";
out = tr_strip_positional_args (in);
check_streq (expected, out);
in = "Hello %1$'d foo %2$'f";
expected = "Hello %d foo %f";
out = tr_strip_positional_args (in);
check_streq (expected, out);
return 0;
}
static int
test_strstrip (void)
{
char *in, *out;
/* strstrip */
in = tr_strdup (" test ");
out = tr_strstrip (in);
check (in == out);
check_streq ("test", out);
tr_free (in);
/* strstrip */
in = tr_strdup (" test test ");
out = tr_strstrip (in);
check (in == out);
check_streq ("test test", out);
tr_free (in);
/* strstrip */
in = tr_strdup ("test");
out = tr_strstrip (in);
check (in == out);
check_streq ("test", out);
tr_free (in);
return 0;
}
static int
test_buildpath (void)
{
char * out;
out = tr_buildPath ("foo", "bar", NULL);
check_streq ("foo" TR_PATH_DELIMITER_STR "bar", out);
tr_free (out);
out = tr_buildPath ("", "foo", "bar", NULL);
check_streq (TR_PATH_DELIMITER_STR "foo" TR_PATH_DELIMITER_STR "bar", out);
tr_free (out);
return 0;
}
static int
test_utf8 (void)
{
const char * in;
char * out;
in = "hello world";
out = tr_utf8clean (in, -1);
check_streq (in, out);
tr_free (out);
in = "hello world";
out = tr_utf8clean (in, 5);
check_streq ("hello", out);
tr_free (out);
/* this version is not utf-8 */
in = "<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD>";
out = tr_utf8clean (in, 17);
check (out != NULL);
check ((strlen (out) == 17) || (strlen (out) == 32));
check (tr_utf8_validate (out, -1, NULL));
tr_free (out);
/* same string, but utf-8 clean */
in = "Òðóäíî áûòü Áîãîì";
out = tr_utf8clean (in, -1);
check (out != NULL);
check (tr_utf8_validate (out, -1, NULL));
check_streq (in, out);
tr_free (out);
return 0;
}
static int
test_numbers (void)
{
int i;
int count;
int * numbers;
numbers = tr_parseNumberRange ("1-10,13,16-19", -1, &count);
check_int_eq (15, count);
check_int_eq (1, numbers[0]);
check_int_eq (6, numbers[5]);
check_int_eq (10, numbers[9]);
check_int_eq (13, numbers[10]);
check_int_eq (16, numbers[11]);
check_int_eq (19, numbers[14]);
tr_free (numbers);
numbers = tr_parseNumberRange ("1-5,3-7,2-6", -1, &count);
check (count == 7);
check (numbers != NULL);
for (i=0; i<count; ++i)
check_int_eq (i+1, numbers[i]);
tr_free (numbers);
numbers = tr_parseNumberRange ("1-Hello", -1, &count);
check_int_eq (0, count);
check (numbers == NULL);
numbers = tr_parseNumberRange ("1-", -1, &count);
check_int_eq (0, count);
check (numbers == NULL);
numbers = tr_parseNumberRange ("Hello", -1, &count);
check_int_eq (0, count);
check (numbers == NULL);
return 0;
}
static int
compareInts (const void * va, const void * vb)
{
const int a = * (const int *)va;
const int b = * (const int*)vb;
return a - b;
}
static int
test_lowerbound (void)
{
int i;
const int A[] = { 1, 2, 3, 3, 3, 5, 8 };
const int expected_pos[] = { 0, 1, 2, 5, 5, 6, 6, 6, 7, 7 };
const int expected_exact[] = { true, true, true, false, true, false, false, true, false, false };
const int N = sizeof (A) / sizeof (A[0]);
for (i=1; i<=10; ++i)
{
bool exact;
const int pos = tr_lowerBound (&i, A, N, sizeof (int), compareInts, &exact);
#if 0
fprintf (stderr, "searching for %d. ", i);
fprintf (stderr, "result: index = %d, ", pos);
if (pos != N)
fprintf (stderr, "A[%d] == %d\n", pos, A[pos]);
else
fprintf (stderr, "which is off the end.\n");
#endif
check_int_eq (expected_pos[i-1], pos);
check_int_eq (expected_exact[i-1], exact);
}
return 0;
}
static int
test_memmem (void)
{
char const haystack[12] = "abcabcabcabc";
char const needle[3] = "cab";
check (tr_memmem (haystack, sizeof haystack, haystack, sizeof haystack) == haystack);
check (tr_memmem (haystack, sizeof haystack, needle, sizeof needle) == haystack + 2);
check (tr_memmem (needle, sizeof needle, haystack, sizeof haystack) == NULL);
return 0;
}
static int
test_hex (void)
{
char hex1[41];
char hex2[41];
uint8_t sha1[20];
/*uint8_t sha2[20];*/
memcpy (hex1, "fb5ef5507427b17e04b69cef31fa3379b456735a", 41);
tr_hex_to_sha1 (sha1, hex1);
tr_sha1_to_hex (hex2, sha1);
check_streq (hex1, hex2);
return 0;
}
2009-11-24 02:16:31 +00:00
static int
test_array (void)
2009-11-24 02:16:31 +00:00
{
int i;
int array[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
int n = sizeof (array) / sizeof (array[0]);
2009-11-24 02:16:31 +00:00
tr_removeElementFromArray (array, 5u, sizeof (int), n--);
for (i=0; i<n; ++i)
check_int_eq ((i<5 ? i : i+1), array[i]);
2009-11-24 02:16:31 +00:00
tr_removeElementFromArray (array, 0u, sizeof (int), n--);
for (i=0; i<n; ++i)
check_int_eq ((i<4 ? i+1 : i+2), array[i]);
2009-11-24 02:16:31 +00:00
tr_removeElementFromArray (array, n-1, sizeof (int), n); n--;
for (i=0; i<n; ++i)
check_int_eq ((i<4 ? i+1 : i+2), array[i]);
2009-11-24 02:16:31 +00:00
return 0;
}
static int
test_url (void)
{
int port;
char * scheme;
char * host;
char * path;
char * str;
const char * url;
url = "http://1";
check (!tr_urlParse (url, -1, &scheme, &host, &port, &path));
check_streq ("http", scheme);
check_streq ("1", host);
check_streq ("/", path);
check_int_eq (80, port);
tr_free (scheme);
tr_free (path);
tr_free (host);
url = "http://www.some-tracker.org/some/path";
check (!tr_urlParse (url, -1, &scheme, &host, &port, &path));
check_streq ("http", scheme);
check_streq ("www.some-tracker.org", host);
check_streq ("/some/path", path);
check_int_eq (80, port);
tr_free (scheme);
tr_free (path);
tr_free (host);
url = "http://www.some-tracker.org:80/some/path";
check (!tr_urlParse (url, -1, &scheme, &host, &port, &path));
check_streq ("http", scheme);
check_streq ("www.some-tracker.org", host);
check_streq ("/some/path", path);
check_int_eq (80, port);
tr_free (scheme);
tr_free (path);
tr_free (host);
url = "http%3A%2F%2Fwww.example.com%2F~user%2F%3Ftest%3D1%26test1%3D2";
str = tr_http_unescape (url, strlen (url));
check_streq ("http://www.example.com/~user/?test=1&test1=2", str);
tr_free (str);
return 0;
}
static int
test_truncd (void)
{
2009-07-14 20:35:48 +00:00
char buf[32];
const double nan = sqrt (-1);
tr_snprintf (buf, sizeof (buf), "%.2f%%", 99.999);
check_streq ("100.00%", buf);
tr_snprintf (buf, sizeof (buf), "%.2f%%", tr_truncd (99.999, 2));
check_streq ("99.99%", buf);
2009-07-14 20:35:48 +00:00
tr_snprintf (buf, sizeof (buf), "%.4f", tr_truncd (403650.656250, 4));
check_streq ("403650.6562", buf);
tr_snprintf (buf, sizeof (buf), "%.2f", tr_truncd (2.15, 2));
check_streq ("2.15", buf);
tr_snprintf (buf, sizeof (buf), "%.2f", tr_truncd (2.05, 2));
check_streq ("2.05", buf);
tr_snprintf (buf, sizeof (buf), "%.2f", tr_truncd (3.3333, 2));
check_streq ("3.33", buf);
tr_snprintf (buf, sizeof (buf), "%.0f", tr_truncd (3.3333, 0));
check_streq ("3", buf);
tr_snprintf (buf, sizeof (buf), "%.2f", tr_truncd (nan, 2));
check (strstr (buf, "nan") != NULL);
return 0;
}
static int
test_cryptoRand (void)
{
int i;
/* test that tr_cryptoRandInt () stays in-bounds */
for (i = 0; i < 100000; ++i)
{
const int val = tr_cryptoRandInt (100);
check (val >= 0);
check (val < 100);
}
return 0;
}
struct blah
{
uint8_t hash[SHA_DIGEST_LENGTH]; /* pieces hash */
int8_t priority; /* TR_PRI_HIGH, _NORMAL, or _LOW */
int8_t dnd; /* "do not download" flag */
time_t timeChecked; /* the last time we tested this piece */
};
int
main (void)
{
const testFunc tests[] = {
test_base64, test_hex, test_lowerbound, test_strip_positional_args,
test_strstrip, test_buildpath, test_utf8, test_numbers, test_memmem,
test_array, test_url, test_truncd, test_cryptoRand,
};
int ret;
int l;
if ((ret = runTests (tests, NUM_TESTS (tests))))
return ret;
/* simple bitfield tests */
for (l = 0; l < NUM_LOOPS; ++l)
if ((ret = test_bitfields ()))
return ret;
/* bitfield count range */
for (l=0; l<10000; ++l)
if ((ret = test_bitfield_count_range ()))
return ret;
return 0;
}