mirror of
https://github.com/transmission/transmission
synced 2024-12-26 09:37:56 +00:00
503 lines
12 KiB
C
503 lines
12 KiB
C
#include <math.h> /* sqrt() */
|
||
#include <string.h> /* strlen() */
|
||
|
||
#include "transmission.h"
|
||
#include "bitfield.h"
|
||
#include "ConvertUTF.h" /* tr_utf8_validate*/
|
||
#include "platform.h"
|
||
#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;
|
||
}
|
||
|
||
static int
|
||
test_array( void )
|
||
{
|
||
int i;
|
||
int array[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
|
||
int n = sizeof( array ) / sizeof( array[0] );
|
||
|
||
tr_removeElementFromArray( array, 5u, sizeof( int ), n-- );
|
||
for( i=0; i<n; ++i )
|
||
check_int_eq( ( i<5 ? i : i+1 ), array[i] );
|
||
|
||
tr_removeElementFromArray( array, 0u, sizeof( int ), n-- );
|
||
for( i=0; i<n; ++i )
|
||
check_int_eq( ( i<4 ? i+1 : i+2 ), array[i] );
|
||
|
||
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] );
|
||
|
||
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 )
|
||
{
|
||
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);
|
||
|
||
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;
|
||
}
|
||
|