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.
This commit is contained in:
Dan Fandrich 2011-01-05 23:53:24 -08:00
parent 53014175e8
commit ebb9c7ae04
5 changed files with 32 additions and 17 deletions

@ -46,12 +46,13 @@ extern int unitfail;
int test(char *unused) \ int test(char *unused) \
{ \ { \
(void)unused; \ (void)unused; \
unit_setup(); \ if (unit_setup()) { \
{ fail("unit_setup() failure"); \
} else {
#define UNITTEST_STOP \ #define UNITTEST_STOP \
unit_stop(); \
} \ } \
unit_stop(); \
return unitfail; \ return unitfail; \
} }

@ -14,9 +14,12 @@ static void test_curl_llist_dtor(void *key , void *value)
(void)value; (void)value;
} }
static void unit_setup( void ) static CURLcode unit_setup( void )
{ {
llist = Curl_llist_alloc( test_curl_llist_dtor ); llist = Curl_llist_alloc( test_curl_llist_dtor );
if (!llist)
return CURLE_OUT_OF_MEMORY;
return CURLE_OK;
} }
static void unit_stop( void ) static void unit_stop( void )

@ -5,7 +5,7 @@
#include "strequal.h" #include "strequal.h"
#include "curlcheck.h" #include "curlcheck.h"
static void unit_setup( void ) {} static CURLcode unit_setup( void ) {return CURLE_OK;}
static void unit_stop( void ) {} static void unit_stop( void ) {}
UNITTEST_START UNITTEST_START

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

@ -9,10 +9,14 @@
static struct SessionHandle *data; static struct SessionHandle *data;
static void unit_setup( void ) static CURLcode unit_setup( void )
{ {
data = curl_easy_init(); data = curl_easy_init();
if (!data)
return CURLE_OUT_OF_MEMORY;
return CURLE_OK;
} }
static void unit_stop( void ) static void unit_stop( void )
{ {
curl_easy_cleanup(data); curl_easy_cleanup(data);