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

Last change on this file since 13708 was 13708, checked in by jordan, 9 years ago

(trunk, libT) #5199 'tr_sessionGetNextQueuedTorrent() can be faster' -- copy peer-mgr.c's partial-sorting peer candidate code to a reusable function in utils.c, tr_quickfindFirstK()"

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