From ebb9c7ae04f905f86f6beddba06a9b0c3e4e0ac8 Mon Sep 17 00:00:00 2001
From: Dan Fandrich <dan@coneharvesters.com>
Date: Wed, 5 Jan 2011 23:53:24 -0800
Subject: Made unit_setup() return an error code to abort the test early

This makes it possible to skip the call to unit_stop() in such
cases.  Also use Curl_safefree() in unit test 1302 so it will
pass the memory torture test.
---
 tests/unit/unit1302.c | 29 ++++++++++++++++++-----------
 1 file changed, 18 insertions(+), 11 deletions(-)

(limited to 'tests/unit/unit1302.c')

diff --git a/tests/unit/unit1302.c b/tests/unit/unit1302.c
index d300a14ee..1a6bf4e54 100644
--- a/tests/unit/unit1302.c
+++ b/tests/unit/unit1302.c
@@ -3,16 +3,21 @@
 #include "setup.h"
 
 #include "urldata.h"
+#include "url.h" /* for Curl_safefree */
 #include "curl_base64.h"
 #include "curlcheck.h"
 #include "memdebug.h" /* LAST include file */
 
 static struct SessionHandle *data;
 
-static void unit_setup( void )
+static CURLcode unit_setup( void )
 {
   data = curl_easy_init();
+  if (!data)
+    return CURLE_OUT_OF_MEMORY;
+  return CURLE_OK;
 }
+
 static void unit_stop( void )
 {
   curl_easy_cleanup(data);
@@ -27,56 +32,58 @@ size_t rc;
 rc = Curl_base64_encode(data, "i", 1, &output);
 fail_unless( rc == 4 , "return code should be 4" );
 verify_memory( output, "aQ==", 4);
-free(output);
+Curl_safefree(output);
 
 rc = Curl_base64_encode(data, "ii", 2, &output);
 fail_unless( rc == 4 , "return code should be 4" );
 verify_memory( output, "aWk=", 4);
-free(output);
+Curl_safefree(output);
 
 rc = Curl_base64_encode(data, "iii", 3, &output);
 fail_unless( rc == 4 , "return code should be 4" );
 verify_memory( output, "aWlp", 4);
-free(output);
+Curl_safefree(output);
 
 rc = Curl_base64_encode(data, "iiii", 4, &output);
 fail_unless( rc == 8 , "return code should be 8" );
 verify_memory( output, "aWlpaQ==", 8);
-free(output);
+Curl_safefree(output);
 
 /* 0 length makes it do strlen() */
 rc = Curl_base64_encode(data, "iiii", 0, &output);
 fail_unless( rc == 8 , "return code should be 8" );
 verify_memory( output, "aWlpaQ==", 8);
-free(output);
+Curl_safefree(output);
 
 rc = Curl_base64_decode("aWlpaQ==", &decoded);
 fail_unless(rc == 4, "return code should be 4");
 verify_memory(decoded, "iiii", 4);
-free(decoded);
+Curl_safefree(decoded);
 
 rc = Curl_base64_decode("aWlp", &decoded);
 fail_unless(rc == 3, "return code should be 3");
 verify_memory(decoded, "iii", 3);
-free(decoded);
+Curl_safefree(decoded);
 
 rc = Curl_base64_decode("aWk=", &decoded);
 fail_unless(rc == 2, "return code should be 2");
 verify_memory(decoded, "ii", 2);
-free(decoded);
+Curl_safefree(decoded);
 
 rc = Curl_base64_decode("aQ==", &decoded);
 fail_unless(rc == 1, "return code should be 1");
 verify_memory(decoded, "i", 2);
-free(decoded);
+Curl_safefree(decoded);
 
 /* this is an illegal input */
+decoded = NULL;
 rc = Curl_base64_decode("aQ", &decoded);
 fail_unless(rc == 0, "return code should be 0");
+fail_if(decoded, "returned pointer should be NULL");
 
 /* this is garbage input that libcurl decodes as far as possible */
 rc = Curl_base64_decode("a\x1f==", &decoded);
 fail_unless(rc == 1, "return code should be 1");
-free(decoded);
+Curl_safefree(decoded);
 
 UNITTEST_STOP
-- 
cgit v1.2.3