From: Nick Mathewson <nickm@torproject.org>
Date: Mon, 17 Sep 2012 14:03:11 +0000 (-0400)
Subject: Merge branch '6044_nm_squashed'
X-Git-Tag: tor-0.2.4.3-alpha~26
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=c3f526ed64012f64589b926082ff5e542ec7d18d;p=thirdparty%2Ftor.git

Merge branch '6044_nm_squashed'
---

c3f526ed64012f64589b926082ff5e542ec7d18d
diff --cc src/test/test_util.c
index 2c65903ddb,1734108635..f615ead7d2
--- a/src/test/test_util.c
+++ b/src/test/test_util.c
@@@ -18,19 -18,74 +18,87 @@@
  #include <tchar.h>
  #endif
  
 +/* XXXX this is a minimal wrapper to make the unit tests compile with the
 + * changed tor_timegm interface. */
 +static time_t
 +tor_timegm_wrapper(const struct tm *tm)
 +{
 +  time_t t;
 +  if (tor_timegm(tm, &t) < 0)
 +    return -1;
 +  return t;
 +}
 +
 +#define tor_timegm tor_timegm_wrapper
 +
+ static void
+ test_util_read_until_eof_impl(const char *fname, size_t file_len,
+                               size_t read_limit)
+ {
+   char *fifo_name = NULL;
+   char *test_str = NULL;
+   char *str = NULL;
+   size_t sz = 9999999;
+   int fd = -1;
+   int r;
+ 
+   fifo_name = tor_strdup(get_fname(fname));
+   test_str = tor_malloc(file_len);
+   crypto_rand(test_str, file_len);
+ 
+   r = write_bytes_to_file(fifo_name, test_str, file_len, 1);
+   tt_int_op(r, ==, 0);
+ 
+   fd = open(fifo_name, O_RDONLY|O_BINARY);
+   tt_int_op(fd, >=, 0);
+   str = read_file_to_str_until_eof(fd, read_limit, &sz);
+   close(fd);
+   tt_assert(str != NULL);
+ 
+   if (read_limit < file_len)
+     tt_int_op(sz, ==, read_limit);
+   else
+     tt_int_op(sz, ==, file_len);
+ 
+   test_mem_op(test_str, ==, str, sz);
+   test_assert(str[sz] == '\0');
+ 
+  done:
+   unlink(fifo_name);
+   tor_free(fifo_name);
+   tor_free(test_str);
+   tor_free(str);
+ }
+ 
+ static void
+ test_util_read_file_eof_tiny_limit(void *arg)
+ {
+   (void)arg;
+   // purposely set limit shorter than what we wrote to the FIFO to
+   // test the maximum, and that it puts the NUL in the right spot
+ 
+   test_util_read_until_eof_impl("tor_test_fifo_tiny", 5, 4);
+ }
+ 
+ static void
+ test_util_read_file_eof_two_loops(void *arg)
+ {
+   (void)arg;
+   // write more than 1024 bytes to the FIFO to test two passes through
+   // the loop in the method; if the re-alloc size is changed this
+   // should be updated as well.
+ 
+   test_util_read_until_eof_impl("tor_test_fifo_2k", 2048, 10000);
+ }
+ 
+ static void
+ test_util_read_file_eof_zero_bytes(void *arg)
+ {
+   (void)arg;
+   // zero-byte fifo
+   test_util_read_until_eof_impl("tor_test_fifo_empty", 0, 10000);
+ }
+ 
  static void
  test_util_time(void)
  {