mirror of
https://github.com/curl/curl.git
synced 2024-11-27 05:50:21 +08:00
fd00b382b2
Previous interfaces for these libcurl internal functions did not allow to tell apart a legitimate zero size result from an error condition. These functions now return a CURLcode indicating function success or otherwise specific error. Output size is returned using a pointer argument. All usage of these two functions, and others closely related, has been adapted to the new interfaces. Relative error and OOM handling adapted or added where missing. Unit test 1302 also adapted.
125 lines
4.1 KiB
C
125 lines
4.1 KiB
C
/***************************************************************************
|
|
* _ _ ____ _
|
|
* Project ___| | | | _ \| |
|
|
* / __| | | | |_) | |
|
|
* | (__| |_| | _ <| |___
|
|
* \___|\___/|_| \_\_____|
|
|
*
|
|
* Copyright (C) 1998 - 2011, Daniel Stenberg, <daniel@haxx.se>, et al.
|
|
*
|
|
* This software is licensed as described in the file COPYING, which
|
|
* you should have received as part of this distribution. The terms
|
|
* are also available at http://curl.haxx.se/docs/copyright.html.
|
|
*
|
|
* You may opt to use, copy, modify, merge, publish, distribute and/or sell
|
|
* copies of the Software, and permit persons to whom the Software is
|
|
* furnished to do so, under the terms of the COPYING file.
|
|
*
|
|
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
|
|
* KIND, either express or implied.
|
|
*
|
|
***************************************************************************/
|
|
#include "curlcheck.h"
|
|
|
|
#include "urldata.h"
|
|
#include "url.h" /* for Curl_safefree */
|
|
#include "curl_base64.h"
|
|
#include "memdebug.h" /* LAST include file */
|
|
|
|
static struct SessionHandle *data;
|
|
|
|
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);
|
|
}
|
|
|
|
UNITTEST_START
|
|
|
|
char *output;
|
|
unsigned char *decoded;
|
|
size_t size = 0;
|
|
unsigned char anychar = 'x';
|
|
CURLcode rc;
|
|
|
|
rc = Curl_base64_encode(data, "i", 1, &output, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 4, "size should be 4");
|
|
verify_memory( output, "aQ==", 4);
|
|
Curl_safefree(output);
|
|
|
|
rc = Curl_base64_encode(data, "ii", 2, &output, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 4, "size should be 4");
|
|
verify_memory( output, "aWk=", 4);
|
|
Curl_safefree(output);
|
|
|
|
rc = Curl_base64_encode(data, "iii", 3, &output, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 4, "size should be 4");
|
|
verify_memory( output, "aWlp", 4);
|
|
Curl_safefree(output);
|
|
|
|
rc = Curl_base64_encode(data, "iiii", 4, &output, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 8, "size should be 8");
|
|
verify_memory( output, "aWlpaQ==", 8);
|
|
Curl_safefree(output);
|
|
|
|
/* 0 length makes it do strlen() */
|
|
rc = Curl_base64_encode(data, "iiii", 0, &output, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 8, "size should be 8");
|
|
verify_memory( output, "aWlpaQ==", 8);
|
|
Curl_safefree(output);
|
|
|
|
rc = Curl_base64_decode("aWlpaQ==", &decoded, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 4, "size should be 4");
|
|
verify_memory(decoded, "iiii", 4);
|
|
Curl_safefree(decoded);
|
|
|
|
rc = Curl_base64_decode("aWlp", &decoded, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 3, "size should be 3");
|
|
verify_memory(decoded, "iii", 3);
|
|
Curl_safefree(decoded);
|
|
|
|
rc = Curl_base64_decode("aWk=", &decoded, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 2, "size should be 2");
|
|
verify_memory(decoded, "ii", 2);
|
|
Curl_safefree(decoded);
|
|
|
|
rc = Curl_base64_decode("aQ==", &decoded, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 1, "size should be 1");
|
|
verify_memory(decoded, "i", 2);
|
|
Curl_safefree(decoded);
|
|
|
|
/* this is an illegal input */
|
|
size = 1; /* not zero */
|
|
decoded = &anychar; /* not NULL */
|
|
rc = Curl_base64_decode("aQ", &decoded, &size);
|
|
/* return code indiferent, but output shall be as follows */
|
|
fail_unless(size == 0, "size should be 0");
|
|
fail_if(decoded, "returned pointer should be NULL");
|
|
|
|
/* this is garbage input that libcurl decodes as far as possible */
|
|
size = 0;
|
|
decoded = NULL;
|
|
rc = Curl_base64_decode("a\x1f==", &decoded, &size);
|
|
fail_unless(rc == CURLE_OK, "return code should be CURLE_OK");
|
|
fail_unless(size == 1, "size should be 1");
|
|
fail_if(!decoded, "returned pointer should not be NULL");
|
|
Curl_safefree(decoded);
|
|
|
|
UNITTEST_STOP
|