source: trunk/libtransmission/utils-test.c @ 10931

Last change on this file since 10931 was 10931, checked in by charles, 12 years ago

(trunk) #3045 "make libtransmission's API byte-oriented instead of KiB-oriented." -- implemented. This is a largish commit and will break the mac build for a little while.

File size: 11.2 KB
Line 
1#include <stdio.h> /* fprintf */
2#include <string.h> /* strcmp */
3
4#include "transmission.h"
5#include "bitfield.h"
6#include "ConvertUTF.h" /* tr_utf8_validate*/
7#include "platform.h"
8#include "crypto.h"
9#include "utils.h"
10#include "web.h"
11
12/* #define VERBOSE */
13#undef VERBOSE
14#define NUM_LOOPS 1
15#define SPEED_TEST 0
16
17#if SPEED_TEST
18 #define VERBOSE
19 #undef NUM_LOOPS
20 #define NUM_LOOPS 200
21#endif
22
23static int test = 0;
24
25#ifdef VERBOSE
26  #define check( A ) \
27    { \
28        ++test; \
29        if( A ){ \
30            fprintf( stderr, "PASS test #%d (%s, %d)\n", test, __FILE__, __LINE__ ); \
31        } else { \
32            fprintf( stderr, "FAIL test #%d (%s, %d)\n", test, __FILE__, __LINE__ ); \
33            return test; \
34        } \
35    }
36#else
37  #define check( A ) \
38    { \
39        ++test; \
40        if( !( A ) ){ \
41            fprintf( stderr, "FAIL test #%d (%s, %d)\n", test, __FILE__, __LINE__ ); \
42            return test; \
43        } \
44    }
45#endif
46
47static int
48test_bitfields( void )
49{
50    unsigned int  i;
51    unsigned int  bitcount = 5000000;
52    tr_bitfield * field = tr_bitfieldNew( bitcount );
53
54    /* test tr_bitfieldAdd */
55    for( i = 0; i < bitcount; ++i )
56        if( !( i % 7 ) )
57            tr_bitfieldAdd( field, i );
58    for( i = 0; i < bitcount; ++i )
59        check( tr_bitfieldHas( field, i ) == ( !( i % 7 ) ) );
60
61    /* test tr_bitfieldAddRange */
62    tr_bitfieldAddRange( field, 0, bitcount );
63    for( i = 0; i < bitcount; ++i )
64        check( tr_bitfieldHas( field, i ) );
65
66    /* test tr_bitfieldRemRange in the middle of a boundary */
67    tr_bitfieldRemRange( field, 4, 21 );
68    for( i = 0; i < 64; ++i )
69        check( tr_bitfieldHas( field, i ) == ( ( i < 4 ) || ( i >= 21 ) ) );
70
71    /* test tr_bitfieldRemRange on the boundaries */
72    tr_bitfieldAddRange( field, 0, 64 );
73    tr_bitfieldRemRange( field, 8, 24 );
74    for( i = 0; i < 64; ++i )
75        check( tr_bitfieldHas( field, i ) == ( ( i < 8 ) || ( i >= 24 ) ) );
76
77    /* test tr_bitfieldRemRange when begin & end is on the same word */
78    tr_bitfieldAddRange( field, 0, 64 );
79    tr_bitfieldRemRange( field, 4, 5 );
80    for( i = 0; i < 64; ++i )
81        check( tr_bitfieldHas( field, i ) == ( ( i < 4 ) || ( i >= 5 ) ) );
82
83    /* test tr_bitfieldAddRange */
84    tr_bitfieldRemRange( field, 0, 64 );
85    tr_bitfieldAddRange( field, 4, 21 );
86    for( i = 0; i < 64; ++i )
87        check( tr_bitfieldHas( field, i ) == ( ( 4 <= i ) && ( i < 21 ) ) );
88
89    /* test tr_bitfieldAddRange on the boundaries */
90    tr_bitfieldRemRange( field, 0, 64 );
91    tr_bitfieldAddRange( field, 8, 24 );
92    for( i = 0; i < 64; ++i )
93        check( tr_bitfieldHas( field, i ) == ( ( 8 <= i ) && ( i < 24 ) ) );
94
95    /* test tr_bitfieldAddRange when begin & end is on the same word */
96    tr_bitfieldRemRange( field, 0, 64 );
97    tr_bitfieldAddRange( field, 4, 5 );
98    for( i = 0; i < 64; ++i )
99        check( tr_bitfieldHas( field, i ) == ( ( 4 <= i ) && ( i < 5 ) ) );
100
101    tr_bitfieldFree( field );
102    return 0;
103}
104
105static int
106test_strip_positional_args( void )
107{
108    const char * in;
109    const char * out;
110    const char * expected;
111
112    in = "Hello %1$s foo %2$.*f";
113    expected = "Hello %s foo %.*f";
114    out = tr_strip_positional_args( in );
115    check( out != NULL )
116    check( !strcmp( out, expected ) )
117
118    in = "Hello %1$'d foo %2$'f";
119    expected = "Hello %d foo %f";
120    out = tr_strip_positional_args( in );
121    check( out != NULL )
122    check( !strcmp( out, expected ) )
123
124    return 0;
125}
126
127static int
128test_strstrip( void )
129{
130    char *in, *out;
131
132    /* strstrip */
133    in = tr_strdup( "   test    " );
134    out = tr_strstrip( in );
135    check( in == out );
136    check( !strcmp( in, "test" ) );
137    tr_free( in );
138
139    /* strstrip */
140    in = tr_strdup( " test test " );
141    out = tr_strstrip( in );
142    check( in == out );
143    check( !strcmp( in, "test test" ) );
144    tr_free( in );
145
146    /* strstrip */
147    in = tr_strdup( "test" );
148    out = tr_strstrip( in );
149    check( in == out );
150    check( !strcmp( in, "test" ) );
151    tr_free( in );
152
153    return 0;
154}
155
156static int
157test_buildpath( void )
158{
159    char * out;
160
161    out = tr_buildPath( "foo", "bar", NULL );
162    check( !strcmp( out, "foo" TR_PATH_DELIMITER_STR "bar" ) );
163    tr_free( out );
164
165    out = tr_buildPath( "", "foo", "bar", NULL );
166    check( !strcmp( out, TR_PATH_DELIMITER_STR "foo" TR_PATH_DELIMITER_STR "bar" ) );
167    tr_free( out );
168
169    return 0;
170}
171
172static int
173test_utf8( void )
174{
175    const char * in;
176    char * out;
177    tr_bool err;
178
179    in = "hello world";
180    out = tr_utf8clean( in, -1, &err );
181    check( err == FALSE )
182    check( out != NULL )
183    check( !strcmp( out, in ) )
184    tr_free( out );
185
186    in = "hello world";
187    out = tr_utf8clean( in, 5, &err );
188    check( err == FALSE )
189    check( out != NULL )
190    check( !strcmp( out, "hello" ) )
191    tr_free( out );
192
193    /* this version is not utf-8 */
194    in = "Òðóäíî áûòü Áîãîì";
195    out = tr_utf8clean( in, 17, &err );
196    check( out != NULL )
197    check( err != 0 )
198    check( strlen( out ) == 17 )
199    check( tr_utf8_validate( out, -1, NULL ) )
200    tr_free( out );
201
202    /* same string, but utf-8 clean */
203    in = "ÒðóÀíî áûòÌ Áîãîì";
204    out = tr_utf8clean( in, -1, &err );
205    check( out != NULL )
206    check( !err );
207    check( tr_utf8_validate( out, -1, NULL ) )
208    check ( !strcmp( in, out ) )
209    tr_free( out );
210
211    return 0;
212}
213
214static int
215test_numbers( void )
216{
217    int i;
218    int count;
219    int * numbers;
220
221    numbers = tr_parseNumberRange( "1-10,13,16-19", -1, &count );
222    check( count == 15 );
223    check( numbers != NULL );
224    check( numbers[0] == 1 );
225    check( numbers[5] == 6 );
226    check( numbers[9] == 10 );
227    check( numbers[10] == 13 );
228    check( numbers[11] == 16 );
229    check( numbers[14] == 19 );
230    tr_free( numbers );
231
232    numbers = tr_parseNumberRange( "1-5,3-7,2-6", -1, &count );
233    check( count == 7 );
234    check( numbers != NULL );
235    for( i=0; i<count; ++i )
236        check( numbers[i] == i+1 );
237    tr_free( numbers );
238
239    numbers = tr_parseNumberRange( "1-Hello", -1, &count );
240    check( count == 0 );
241    check( numbers == NULL );
242
243    numbers = tr_parseNumberRange( "1-", -1, &count );
244    check( count == 0 );
245    check( numbers == NULL );
246
247    numbers = tr_parseNumberRange( "Hello", -1, &count );
248    check( count == 0 );
249    check( numbers == NULL );
250
251    return 0;
252}
253
254static int
255compareInts( const void * va, const void * vb )
256{
257    const int a = *(const int *)va;
258    const int b = *(const int*)vb;
259    return a - b;
260}
261
262static int
263test_lowerbound( void )
264{
265    int i;
266    const int A[] = { 1, 2, 3, 3, 3, 5, 8 };
267    const int expected_pos[] = { 0, 1, 2, 5, 5, 6, 6, 6, 7, 7 };
268    const int expected_exact[] = { TRUE, TRUE, TRUE, FALSE, TRUE, FALSE, FALSE, TRUE, FALSE, FALSE };
269    const int N = sizeof(A) / sizeof(A[0]);
270
271    for( i=1; i<=10; ++i )
272    {
273        tr_bool exact;
274        const int pos = tr_lowerBound( &i, A, N, sizeof(int), compareInts, &exact );
275
276#if 0
277        fprintf( stderr, "searching for %d.  ", i );
278        fprintf( stderr, "result: index = %d, ", pos );
279        if( pos != N )
280            fprintf( stderr, "A[%d] == %d\n", pos, A[pos] );
281        else
282            fprintf( stderr, "which is off the end.\n" );
283#endif
284        check( pos == expected_pos[i-1] )
285        check( exact == expected_exact[i-1] )
286    }
287
288    return 0;
289}
290
291static int
292test_memmem( void )
293{
294    char const haystack[12] = "abcabcabcabc";
295    char const needle[3] = "cab";
296
297    check( tr_memmem( haystack, sizeof haystack, haystack, sizeof haystack) == haystack )
298    check( tr_memmem( haystack, sizeof haystack, needle, sizeof needle) == haystack + 2 )
299    check( tr_memmem( needle, sizeof needle, haystack, sizeof haystack) == NULL )
300    check( tr_memmem( haystack, sizeof haystack, "", 0) == haystack )
301    check( tr_memmem( haystack, sizeof haystack, NULL, 0) == haystack )
302    check( tr_memmem( haystack, 0, "", 0) == haystack )
303
304    return 0;
305}
306
307static int
308test_hex( void )
309{
310    char hex1[41];
311    char hex2[41];
312    uint8_t sha1[20];
313    /*uint8_t sha2[20];*/
314
315    memcpy( hex1, "fb5ef5507427b17e04b69cef31fa3379b456735a", 41 );
316    tr_hex_to_sha1( sha1, hex1 );
317    tr_sha1_to_hex( hex2, sha1 );
318    check( !strcmp( hex1, hex2 ) )
319
320    return 0;
321}
322
323static int
324test_array( void )
325{
326    int i;
327    int array[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
328    int n = sizeof( array ) / sizeof( array[0] );
329
330    tr_removeElementFromArray( array, 5u, sizeof( int ), n-- );
331    for( i=0; i<n; ++i )
332        check( array[i] == ( i<5 ? i : i+1 ) );
333
334    tr_removeElementFromArray( array, 0u, sizeof( int ), n-- );
335    for( i=0; i<n; ++i )
336        check( array[i] == ( i<4 ? i+1 : i+2 ) );
337
338    tr_removeElementFromArray( array, n-1, sizeof( int ), n ); n--;
339    for( i=0; i<n; ++i )
340        check( array[i] == ( i<4 ? i+1 : i+2 ) );
341
342    return 0;
343}
344
345static int
346test_url( void )
347{
348    int port;
349    char * scheme;
350    char * host;
351    char * path;
352    char * str;
353    const char * url;
354
355    url = "http://www.some-tracker.org/some/path";
356    check( !tr_urlParse( url, -1, &scheme, &host, &port, &path ) )
357    check( !strcmp( scheme, "http" ) )
358    check( !strcmp( host, "www.some-tracker.org" ) )
359    check( !strcmp( path, "/some/path" ) )
360    check( port == 80 )
361    tr_free( scheme );
362    tr_free( path );
363    tr_free( host );
364
365    url = "http://www.some-tracker.org:80/some/path";
366    check( !tr_urlParse( url, -1, &scheme, &host, &port, &path ) )
367    check( !strcmp( scheme, "http" ) )
368    check( !strcmp( host, "www.some-tracker.org" ) )
369    check( !strcmp( path, "/some/path" ) )
370    check( port == 80 )
371    tr_free( scheme );
372    tr_free( path );
373    tr_free( host );
374
375    url = "http%3A%2F%2Fwww.example.com%2F~user%2F%3Ftest%3D1%26test1%3D2";
376    str = tr_http_unescape( url, strlen( url ) );
377    check( !strcmp( str, "http://www.example.com/~user/?test=1&test1=2" ) )
378    tr_free( str );
379
380    return 0;
381}
382
383int
384main( void )
385{
386    char buf[32];
387    char *in, *out;
388    int   len;
389    int   i;
390    int   l;
391
392    /* tr_truncd */
393    tr_snprintf( buf, sizeof( buf ), "%.2f%%", 99.999 );
394    check( !strcmp( buf, "100.00%" ) );
395    tr_snprintf( buf, sizeof( buf ), "%.2f%%", tr_truncd( 99.999, 2 ) );
396    check( !strcmp( buf, "99.99%" ) );
397
398    /* base64 */
399    out = tr_base64_encode( "YOYO!", -1, &len );
400    check( out );
401    check( !strcmp( out, "WU9ZTyE=\n" ) );
402    check( len == 9 );
403    in = tr_base64_decode( out, -1, &len );
404    check( in );
405    check( !strcmp( in, "YOYO!" ) );
406    check( len == 5 );
407    tr_free( in );
408    tr_free( out );
409    out = tr_base64_encode( NULL, 0, &len );
410    check( out == NULL );
411    check( len == 0 );
412
413    if( ( i = test_hex( ) ) )
414        return i;
415    if( ( i = test_lowerbound( ) ) )
416        return i;
417    if( ( i = test_strip_positional_args( ) ) )
418        return i;
419    if( ( i = test_strstrip( ) ) )
420        return i;
421    if( ( i = test_buildpath( ) ) )
422        return i;
423    if( ( i = test_utf8( ) ) )
424        return i;
425    if( ( i = test_numbers( ) ) )
426        return i;
427    if( ( i = test_memmem( ) ) )
428        return i;
429    if( ( i = test_array( ) ) )
430        return i;
431    if( ( i = test_url( ) ) )
432        return i;
433
434    /* test that tr_cryptoRandInt() stays in-bounds */
435    for( i = 0; i < 100000; ++i )
436    {
437        const int val = tr_cryptoRandInt( 100 );
438        check( val >= 0 );
439        check( val < 100 );
440    }
441
442    /* simple bitfield tests */
443    for( l = 0; l < NUM_LOOPS; ++l )
444        if( ( i = test_bitfields( ) ) )
445            return i;
446
447    return 0;
448}
449
Note: See TracBrowser for help on using the repository browser.