[ccan] [PATCH] base64: implements rfc4648, the base64 encoding
Peter Barker
pbarker at barker.dropbear.id.au
Thu Feb 12 10:46:14 AEDT 2015
On Thu, 22 Jan 2015, Peter Barker wrote:
> Encode buffers into base64 according to rfc4648.
> Decode base64-encoded buffers according to the same standard.
>
> Signed-off-by: <pb-ccan at barker.dropbear.id.au>
Does anybody have any opinions on this base64 implementation?
Thanks,
Peter
> ---
> Makefile-ccan | 1 +
> ccan/base64/GPL-2 | 1 +
> ccan/base64/LICENSE | 1 +
> ccan/base64/_info | 39 +++++
> ccan/base64/base64.c | 273 ++++++++++++++++++++++++++++++++++
> ccan/base64/base64.h | 200 +++++++++++++++++++++++++
> ccan/base64/test/moretap.h | 96 ++++++++++++
> ccan/base64/test/run.c | 351 ++++++++++++++++++++++++++++++++++++++++++++
> 8 files changed, 962 insertions(+)
> create mode 120000 ccan/base64/GPL-2
> create mode 120000 ccan/base64/LICENSE
> create mode 100644 ccan/base64/_info
> create mode 100644 ccan/base64/base64.c
> create mode 100644 ccan/base64/base64.h
> create mode 100644 ccan/base64/test/moretap.h
> create mode 100644 ccan/base64/test/run.c
>
> diff --git a/Makefile-ccan b/Makefile-ccan
> index acfbe19..8321ab0 100644
> --- a/Makefile-ccan
> +++ b/Makefile-ccan
> @@ -35,6 +35,7 @@ MODS_WITH_SRC := antithread \
> asprintf \
> autodata \
> avl \
> + base64 \
> bdelta \
> block_pool \
> breakpoint \
> diff --git a/ccan/base64/GPL-2 b/ccan/base64/GPL-2
> new file mode 120000
> index 0000000..9961ca9
> --- /dev/null
> +++ b/ccan/base64/GPL-2
> @@ -0,0 +1 @@
> +../../licenses/GPL-2
> \ No newline at end of file
> diff --git a/ccan/base64/LICENSE b/ccan/base64/LICENSE
> new file mode 120000
> index 0000000..9961ca9
> --- /dev/null
> +++ b/ccan/base64/LICENSE
> @@ -0,0 +1 @@
> +../../licenses/GPL-2
> \ No newline at end of file
> diff --git a/ccan/base64/_info b/ccan/base64/_info
> new file mode 100644
> index 0000000..d69a7cb
> --- /dev/null
> +++ b/ccan/base64/_info
> @@ -0,0 +1,39 @@
> +#include "config.h"
> +
> +/**
> + * base64 - base64 encoding and decoding (rfc4648).
> + *
> + * base64 encoding is used to encode data in a 7-bit clean manner.
> + * Commonly used for escaping data before encapsulation or transfer
> + *
> + * Example:
> + * #include <stdio.h>
> + * #include <string.h>
> + * #include <ccan/base64/base64.h>
> + *
> + * int main(int argc, char *argv[])
> + * {
> + * char *base64_encoded_string;
> + * int i;
> + *
> + * // print the base64-encoded form of the program arguments
> + * for(i=1;i<argc;i++) {
> + * size_t space_required = base64_encoded_length(strlen(argv[i]));
> + * base64_encoded_string = malloc(space_required);
> + * base64_encode(base64_encoded_string,sizeof(space_required),argv[i],strlen(argv[i]));
> + * printf("%s\n",base64_encoded_string);
> + * free(base64_encoded_string);
> + * }
> + *
> + * return 0;
> + * }
> + *
> + * License: GPL (v2 or any later version)
> + */
> +int main(int argc, char *argv[])
> +{
> + if (argc != 2)
> + return 1;
> +
> + return 1;
> +}
> diff --git a/ccan/base64/base64.c b/ccan/base64/base64.c
> new file mode 100644
> index 0000000..59d7c31
> --- /dev/null
> +++ b/ccan/base64/base64.c
> @@ -0,0 +1,273 @@
> +/* Licensed under LGPLv2+ - see LICENSE file for details */
> +#include "base64.h"
> +
> +#include <errno.h>
> +#include <string.h>
> +#include <assert.h>
> +
> +/**
> + * sixbit_to_b64 - maps a 6-bit value to the base64 alphabet
> + * @param alphabet A base 64 alphabet (see base64_init_alphabet)
> + * @param sixbit Six-bit value to map
> + * @return a base 64 character
> + */
> +static char sixbit_to_b64(const base64_alphabet_t *alphabet, const char sixbit)
> +{
> + assert(sixbit >= 0);
> + assert(sixbit <= 63);
> +
> + return alphabet->encode_map[(unsigned char)sixbit];
> +}
> +
> +/**
> + * sixbit_from_b64 - maps a base64-alphabet character to its 6-bit value
> + * @param alphabet A base 64 alphabet (see base64_init_alphabet)
> + * @param sixbit Six-bit value to map
> + * @return a six-bit value
> + */
> +static signed char sixbit_from_b64(const base64_alphabet_t *alphabet,
> + const unsigned char b64letter)
> +{
> + signed char ret;
> +
> + ret = alphabet->decode_map[(unsigned char)b64letter];
> + if (ret == '~') {
> + errno = ERANGE;
> + return -1;
> + }
> +
> + return ret;
> +}
> +
> +int base64_char_in_alphabet(const base64_alphabet_t *alphabet,
> + const char b64char) {
> + return (alphabet->decode_map[(unsigned char)b64char] != '~');
> +}
> +
> +void base64_init_alphabet(base64_alphabet_t *dest, const char src[64]) {
> + unsigned char i;
> +
> + memcpy(dest->encode_map,src,64);
> + memset(dest->decode_map,'~',256);
> + for (i=0; i<64; i++) {
> + dest->decode_map[(unsigned char)src[i]] = i;
> + }
> +}
> +
> +size_t base64_encoded_length(size_t srclen)
> +{
> + return ((srclen + 2) / 3) * 4;
> +}
> +
> +void base64_encode_triplet_using_alphabet(const base64_alphabet_t *alphabet,
> + char dest[4], const char src[3])
> +{
> + char a = src[0];
> + char b = src[1];
> + char c = src[2];
> +
> + dest[0] = sixbit_to_b64(alphabet, (a & 0xfc) >> 2);
> + dest[1] = sixbit_to_b64(alphabet, ((a & 0x3) << 4) | ((b & 0xf0) >> 4));
> + dest[2] = sixbit_to_b64(alphabet, ((c & 0xc0) >> 6) | ((b & 0xf) << 2));
> + dest[3] = sixbit_to_b64(alphabet, c & 0x3f);
> +}
> +
> +void base64_encode_tail_using_alphabet(const base64_alphabet_t *alphabet,
> + char dest[4],
> + const char *src, const size_t srclen)
> +{
> + char longsrc[3];
> +
> + memcpy(longsrc, src, srclen);
> + memset(longsrc+srclen, '\0', 3-srclen);
> + base64_encode_triplet_using_alphabet(alphabet, dest, longsrc);
> + memset(dest+1+srclen, '=', 3-srclen);
> +}
> +
> +size_t base64_encode_using_alphabet(const base64_alphabet_t *alphabet,
> + char *dest, const size_t destlen,
> + const char *src, const size_t srclen)
> +{
> + size_t src_offset = 0;
> + size_t dest_offset = 0;
> +
> + if (destlen < base64_encoded_length(srclen)) {
> + errno = EOVERFLOW;
> + return -1;
> + }
> +
> + while (srclen - src_offset >= 3) {
> + base64_encode_triplet_using_alphabet(alphabet, &dest[dest_offset], &src[src_offset]);
> + src_offset += 3;
> + dest_offset += 4;
> + }
> +
> + if (srclen - src_offset) {
> + base64_encode_tail_using_alphabet(alphabet, &dest[dest_offset], &src[src_offset], srclen-src_offset);
> + dest_offset += 4;
> + }
> +
> + memset(&dest[dest_offset], '\0', destlen-dest_offset);
> +
> + return dest_offset;
> +}
> +
> +size_t base64_decoded_length(size_t srclen)
> +{
> + return ((srclen+3)/4*3);
> +}
> +
> +int base64_decode_quartet_using_alphabet(const base64_alphabet_t *alphabet,
> + char dest[3], const char src[4])
> +{
> + signed char a;
> + signed char b;
> + signed char c;
> + signed char d;
> +
> + a = sixbit_from_b64(alphabet, src[0]);
> + b = sixbit_from_b64(alphabet, src[1]);
> + c = sixbit_from_b64(alphabet, src[2]);
> + d = sixbit_from_b64(alphabet, src[3]);
> +
> + if ((a == -1) || (b == -1) || (c == -1) || (d == -1)) {
> + return -1;
> + }
> +
> + dest[0] = (a << 2) | (b >> 4);
> + dest[1] = ((b & 0xf) << 4) | (c >> 2);
> + dest[2] = ((c & 0x3) << 6) | d;
> +
> + return 0;
> +}
> +
> +
> +int base64_decode_tail_using_alphabet(const base64_alphabet_t *alphabet, char dest[3],
> + const char * src, const size_t srclen)
> +{
> + char longsrc[4];
> + int quartet_result;
> + size_t insize = srclen;
> +
> + if (insize == 0) {
> + return 0;
> + }
> + while (src[insize-1] == '=') { /* throw away padding symbols */
> + insize--;
> + }
> + if (insize == 1) {
> + /* the input is malformed.... */
> + errno = EINVAL;
> + return -1;
> + }
> + memcpy(longsrc, src, insize);
> + memset(longsrc+insize, 'A', 4-insize);
> + quartet_result = base64_decode_quartet_using_alphabet(alphabet, dest, longsrc);
> + if (quartet_result == -1) {
> + return -1;
> + }
> +
> + return insize - 1;
> +}
> +
> +size_t base64_decode_using_alphabet(const base64_alphabet_t *alphabet,
> + char *dest, const size_t destlen,
> + const char *src, const size_t srclen)
> +{
> + size_t dest_offset = 0;
> + size_t i;
> + size_t more;
> +
> + if (destlen < base64_decoded_length(srclen)) {
> + errno = EOVERFLOW;
> + return -1;
> + }
> +
> + for(i=0; srclen - i > 4; i+=4) {
> + if (base64_decode_quartet_using_alphabet(alphabet, &dest[dest_offset], &src[i]) == -1) {
> + return -1;
> + }
> + dest_offset += 3;
> + }
> +
> + more = base64_decode_tail_using_alphabet(alphabet, &dest[dest_offset], &src[i], srclen - i);
> + if (more == -1) {
> + return -1;
> + }
> + dest_offset += more;
> +
> + memset(&dest[dest_offset], '\0', destlen-dest_offset);
> +
> + return dest_offset;
> +}
> +
> +
> +/* the rfc4648 functions: */
> +#define base64_decode_map_rfc4648 \
> + "~~~~~" /* 0 */ \
> + "~~~~~" /* 5 */ \
> + "~~~~~" /* 10 */ \
> + "~~~~~" /* 15 */ \
> + "~~~~~" /* 20 */ \
> + "~~~~~" /* 25 */ \
> + "~~~~~" /* 30 */ \
> + "~~~~~" /* 35 */ \
> + "~~~\x3e~" /* 40 */ \
> + "~~\x3f\x34\x35" /* 45 */ \
> + "\x36\x37\x38\x39\x3a" /* 50 */ \
> + "\x3b\x3c\x3d~~" /* 55 */ \
> + "~~~~~" /* 60 */ \
> + "\x00\x01\x02\x03\x04" /* 65 A */ \
> + "\x05\x06\x07\x08\x09" /* 70 */ \
> + "\x0a\x0b\x0c\x0d\x0e" /* 75 */ \
> + "\x0f\x10\x11\x12\x13" /* 80 */ \
> + "\x14\x15\x16\x17\x18" /* 85 */ \
> + "\x19~~~~" /* 90 */ \
> + "~~\x1a\x1b\x1c" /* 95 */ \
> + "\x1d\x1e\x1f\x20\x21" /* 100 */ \
> + "\x22\x23\x24\x25\x26" /* 105 */ \
> + "\x27\x28\x29\x2a\x2b" /* 110 */ \
> + "\x2c\x2d\x2e\x2f\x30" /* 115 */ \
> + "\x31\x32\x33~~" /* 120 */ \
> + "~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" /* 125 */ \
> + "~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" /* 175 */ \
> + "~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" /* 225 */
> +
> +static const base64_alphabet_t alphabet_rfc4648_pregen = {
> + "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/",
> + base64_decode_map_rfc4648
> +};
> +
> +#undef base64_decode_map_rfc4648
> +
> +int base64_decode_quartet(char dest[3], const char src[4])
> +{
> + return base64_decode_quartet_using_alphabet(&alphabet_rfc4648_pregen,
> + dest, src);
> +}
> +size_t base64_decode_tail(char dest[3], const char *src, const size_t srclen)
> +{
> + return base64_decode_tail_using_alphabet(&alphabet_rfc4648_pregen,
> + dest, src, srclen);
> +}
> +size_t base64_decode(char *dest, const size_t destlen, const char *src, const size_t srclen) {
> + return base64_decode_using_alphabet(&alphabet_rfc4648_pregen,
> + dest, destlen, src, srclen);
> +}
> +
> +
> +
> +void base64_encode_triplet(char dest[4], const char src[3]) {
> + base64_encode_triplet_using_alphabet(&alphabet_rfc4648_pregen,
> + dest, src);
> +}
> +void base64_encode_tail(char dest[4], const char *src, const size_t srclen) {
> + base64_encode_tail_using_alphabet(&alphabet_rfc4648_pregen,
> + dest, src, srclen);
> +}
> +size_t base64_encode(char *dest, const size_t destlen, const char *src, const size_t srclen) {
> + return base64_encode_using_alphabet(&alphabet_rfc4648_pregen,
> + dest, destlen, src, srclen);
> +}
> +
> +/* end rfc4648 functions */
> diff --git a/ccan/base64/base64.h b/ccan/base64/base64.h
> new file mode 100644
> index 0000000..929a9d4
> --- /dev/null
> +++ b/ccan/base64/base64.h
> @@ -0,0 +1,200 @@
> +/* Licensed under LGPLv2+ - see LICENSE file for details */
> +#ifndef CCAN_BASE64_H
> +#define CCAN_BASE64_H
> +
> +#include <stdio.h> /* For size_t */
> +
> +
> +/**
> + * base64_alphabet_rfc4648 - the base64 alphabet as defined in rfc4648
> + */
> +static const char base64_alphabet_rfc4648[] =
> + "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
> +
> +/**
> + * base64_alphabet_t - structure to hold maps for encode/decode
> + */
> +typedef struct {
> + char encode_map[64];
> + char decode_map[256];
> +} base64_alphabet_t;
> +
> +/**
> + * base64_encoded_length - Calculate encode buffer length
> + * @param srclen the size of the data to be encoded
> + * @note add 1 to this to get null-termination
> + * @return Buffer length required for encode
> + */
> +size_t base64_encoded_length(size_t srclen);
> +
> +/**
> + * base64_encode - Encode a buffer into base64 according to rfc4648
> + * @param dest Buffer to encode into
> + * @param destlen Length of the destination buffer
> + * @param src Buffer to encode
> + * @param srclen Length of the data to encode
> + * @return Number of encoded bytes set in dest. -1 on error (and errno set)
> + * @note dest will be nul-padded to destlen
> + *
> + * This function encodes src according to http://tools.ietf.org/html/rfc4648
> + *
> + * Example:
> + * size_t encoded_length;
> + * char dest[100];
> + * const char *src = "This string gets encoded";
> + * encoded_length = base64_encode(dest, sizeof(dest), src, strlen(src));
> + * printf("Returned data of length %zd @%p\n", encoded_length, &dest);
> + */
> +size_t base64_encode(char *dest, const size_t destlen, const char *src, const size_t srclen);
> +
> +/**
> + * base64_decoded_length - Calculate decode buffer length
> + * @param srclen Length of the data to be decoded
> + * @note This does not return the size of the decoded data! see base64_decode
> + * @return Minimum buffer length for safe decode
> + */
> +size_t base64_decoded_length(size_t srclen);
> +
> +/**
> + * base64_decode - decode An rfc4648 base64-encoded string
> + * @param dest Buffer to decode into
> + * @param destlen Length of the destination buffer
> + * @param src Buffer to decode
> + * @param srclen Length of the data to decode
> + * @return Number of decoded bytes set in dest. -1 on error (and errno set)
> + * @note dest will be nul-padded to destlen
> + *
> + * This function decodes the buffer according to
> + * http://tools.ietf.org/html/rfc4648
> + *
> + * Example:
> + * size_t decoded_length;
> + * char ret[100];
> + * const char *src = "Zm9vYmFyYmF6";
> + * decoded_length = base64_decode(ret, sizeof(ret), src, strlen(src));
> + * printf("Returned data of length %zd @%p\n", decoded_length, &ret);
> + */
> +size_t base64_decode(char *dest, const size_t destlen,
> + const char *src, const size_t srclen);
> +
> +
> +/**
> + * base64_encode_triplet - encode 3 bytes into base64
> + * @param dest Buffer containing at least 4 bytes
> + * @param src Buffer containing 3 bytes
> + */
> +void base64_encode_triplet(char dest[4], const char src[3]);
> +
> +/**
> + * base64_encode_tail - encode the final bytes of a source
> + * @param dest Buffer containing at least 4 bytes
> + * @param src Buffer containing at least srclen bytes
> + * @param srclen Number of bytes to encode in src
> + */
> +void base64_encode_tail(char dest[4], const char *src, const size_t srclen);
> +
> +
> +/**
> + * base64_decode_quartet - decode the first 4 characters in src into dest
> + * @param dest Buffer containing at least 3 bytes
> + * @param src Buffer containing at least 4 characters
> + * @return Number of decoded bytes set in dest. -1 on error (and errno set)
> + */
> +int base64_decode_quartet(char dest[3], const char src[4]);
> +
> +/**
> + * @brief decode the final bytes of a base64 string from src into dest
> + * @param dest Buffer containing 3 bytes
> + * @param src Buffer containing 4 bytes - padded with '=' as required
> + * @param srclen Number of bytes to decode in src
> + * @return Number of decoded bytes set in dest. -1 on error (and errno set)
> + */
> +size_t base64_decode_tail(char dest[3], const char src[4], const size_t srclen);
> +
> +
> +/**
> + * base64_encode_triplet_using_alphabet - encode 3 bytes into base64 using a specific alphabet
> + * @param alphabet Alphabet to use for encoding
> + * @param dest Buffer containing at least 3 bytes
> + * @param src Buffer containing at least 4 characters
> + */
> +void base64_encode_triplet_using_alphabet(const base64_alphabet_t *alphabet,
> + char dest[4], const char src[3]);
> +
> +/**
> + * base64_encode_tail_using_alphabet - encode the final bytes of a source using a specific alphabet
> + * @param alphabet Alphabet to use for encoding
> + * @param dest Buffer containing at least 4 bytes
> + * @param src Buffer containing at least srclen bytes
> + * @param srclen Number of bytes to encode in src
> + */
> +void base64_encode_tail_using_alphabet(const base64_alphabet_t *alphabet,
> + char dest[4],
> + const char *src, const size_t srclen);
> +
> +/**
> + * base64_encode_using_alphabet - encode a buffer into base64 using a specific alphabet
> + * @param alphabet Alphabet to use for encoding
> + * @param dest Buffer to encode into
> + * @param destlen Length of dest
> + * @param src Buffer to encode
> + * @param srclen Length of the data to encode
> + * @return Number of encoded bytes set in dest. -1 on error (and errno set)
> + * @note dest will be nul-padded to destlen
> + */
> +size_t base64_encode_using_alphabet(const base64_alphabet_t *alphabet,
> + char *dest, const size_t destlen,
> + const char *src, const size_t srclen);
> +
> +/**
> + * base64_init_alphabet - populate a base64_alphabet_t based on a supplied map
> + * @param dest A base64 alphabet
> + * @param src Map to populate the alphabet from (e.g. base64_alphabet_rfc4648)
> + */
> +void base64_init_alphabet(base64_alphabet_t *dest, const char src[64]);
> +
> +/*
> + * base64_char_in_alphabet - returns true if character can be part of an encoded string
> + * @param alphabet A base 64 alphabet (see base64_init_alphabet)
> + * @param b64char Character to check
> + */
> +int base64_char_in_alphabet(const base64_alphabet_t *alphabet,
> + const char b64char);
> +
> +/**
> + * base64_decode_using_alphabet - decode a base64-encoded string using a specific alphabet
> + * @param alphabet A base 64 alphabet (see base64_init_alphabet)
> + * @param dest Buffer to decode into
> + * @param destlen length of dest
> + * @param src the buffer to decode
> + * @param srclen the length of the data to decode
> + * @return Number of decoded bytes set in dest. -1 on error (and errno set)
> + * @note dest will be nul-padded to destlen
> + */
> +size_t base64_decode_using_alphabet(const base64_alphabet_t *alphabet,
> + char *dest, const size_t destlen,
> + const char *src, const size_t srclen);
> +
> +/**
> + * base64_decode_quartet_using_alphabet - decode 4 bytes from base64 using a specific alphabet
> + * @param alphabet A base 64 alphabet (see base64_init_alphabet)
> + * @param dest Buffer containing at least 3 bytes
> + * @param src Buffer containing at least 4 characters
> + * @return Number of decoded bytes set in dest. -1 on error (and errno set)
> + */
> +int base64_decode_quartet_using_alphabet(const base64_alphabet_t *alphabet,
> + char dest[3], const char src[4]);
> +
> +/**
> + * base64_decode_tail_using_alphabet - decode the final bytes of a base64 string using a specific alphabet
> + * @param alphabet A base 64 alphabet (see base64_init_alphabet)
> + * @param dest Buffer of at least 3 bytes
> + * @param src Buffer containing 4 bytes - padded with '=' as required
> + * @param srclen Number of bytes to decode in src
> + * @return Number of decoded bytes set in dest. -1 on error (and errno set)
> + */
> +int base64_decode_tail_using_alphabet(const base64_alphabet_t *alphabet,
> + char *dest,
> + const char *src, const size_t srclen);
> +
> +#endif /* CCAN_BASE64_H */
> diff --git a/ccan/base64/test/moretap.h b/ccan/base64/test/moretap.h
> new file mode 100644
> index 0000000..114445c
> --- /dev/null
> +++ b/ccan/base64/test/moretap.h
> @@ -0,0 +1,96 @@
> +#ifndef _BASE64_MORETAP_H
> +#define _BASE64_MORETAP_H
> +
> +#include <ccan/str/str.h>
> +
> +/**
> + * is_str - OK if strings are equal
> + * @e1: expression for the variable string
> + * @e2: expression for the expected string
> + *
> + * If the strings are equal, the test passes.
> + *
> + * Example:
> + * is_str(give_me_a_fred(),"fred");
> + */
> +static void _is_str(char *got,const char *expected, const char *got_string, const char *expected_string, const char *func, const char *file, int line) {
> + if (streq(expected,got)) {
> + _gen_result(1, func, file, line,"%s eq %s",
> + got_string,expected_string);
> + } else {
> + _gen_result(0, func, file, line,"%s eq %s",
> + got_string,expected_string);
> + diag("Expected: %s",expected);
> + diag(" Got: %s",got);
> + }
> +}
> +# define is_str(got,expected) _is_str(got,expected,#got,#expected,__func__, __FILE__, __LINE__)
> +
> +
> +/**
> + * is_int - OK if arguments are equal when cast to integers
> + * @e1: expression for the number
> + * @e2: expression for the expected number
> + *
> + * If the numbers are equal, the test passes.
> + *
> + * Example:
> + * is_int(give_me_17(),17);
> + */
> +# define is_int(e1,e2 ...) \
> + (((int)e1)==((int)e2) ? \
> + _gen_result(1, __func__, __FILE__, __LINE__,"%s == %s",#e1,#e2) : \
> + (_gen_result(0, __func__, __FILE__, __LINE__,"%s == %s",#e1,#e2)) || (diag("Expected: %d",e2),diag(" Got: %d",e1),0)) /* diag is void; note commas. */
> +
> +
> +
> +/**
> + * is_mem - OK if arguments are identical up to length @e3
> + * @e1: expression for the buffer
> + * @e2: expression for the expected buffer
> + * @e2: length to compare in buffers
> + *
> + * If the buffers are equal up to @e2, the test passes.
> + *
> + * Example:
> + * is_mem(give_me_foo(),"foo",3);
> + */
> +static void _is_mem(const char *got, const char *expected, const size_t len,
> + const char *got_string, const char *expected_string, const char *len_string,
> + const char *func, const char *file, int line) {
> + size_t offset = 0;
> +
> + for (offset=0; offset<len; offset++) {
> + if (got[offset] != expected[offset]) {
> + _gen_result(0, func, file, line,"%s eq %s",got_string,expected_string);
> + /* diag("Expected: %s",e2); */
> + /* diag(" Got: %s",e1); */
> + diag("Buffers differ at offset %zd (got=0x%02x expected=0x%02x)",
> + offset,got[offset],expected[offset]);
> + return;
> + }
> + }
> +
> + _gen_result(1, __func__, __FILE__, __LINE__,"%s eq %s",
> + expected_string,got_string);
> +}
> +# define is_mem(got,expected,len) \
> + _is_mem(got,expected,len,#got,#expected,#len,__func__, __FILE__, __LINE__)
> +
> +/**
> + * is_size_t - OK if arguments are equal when cast to size_t
> + * @e1: expression for the number
> + * @e2: expression for the expected number
> + *
> + * If the numbers are equal, the test passes.
> + *
> + * Example:
> + * is_size_t(give_me_17(),17);
> + */
> +# define is_size_t(e1,e2 ...) \
> + ((size_t)(e1)==((size_t)e2) ? \
> + _gen_result(1, __func__, __FILE__, __LINE__,"%s == %s",#e1,#e2) : \
> + (_gen_result(0, __func__, __FILE__, __LINE__, \
> + "%s == %s",#e1,#e2)) || (diag("Expected: %zd",(size_t)e2),diag(" Got: %zd",(size_t)e1),0)) /* diag is void; note commas. */
> +
> +#endif
> diff --git a/ccan/base64/test/run.c b/ccan/base64/test/run.c
> new file mode 100644
> index 0000000..2159409
> --- /dev/null
> +++ b/ccan/base64/test/run.c
> @@ -0,0 +1,351 @@
> +/* Start of run.c test */
> +#include "config.h"
> +
> +#include <stdlib.h>
> +#include <stdio.h>
> +#include <string.h>
> +
> +#include <ccan/base64/base64.h>
> +#include <ccan/tap/tap.h>
> +
> +#include <ccan/base64/base64.c>
> +#include "moretap.h"
> +
> +static void * xmalloc(size_t size);
> +
> +/* not defined in terms of test_encode_using_alphabet so we cross
> + appropriate paths in library */
> +#define test_encode(src,srclen,expected) \
> + do { \
> + size_t destlen; \
> + char * dest; \
> + destlen = base64_encoded_length(srclen); \
> + destlen++; /* null termination */ \
> + dest = xmalloc(destlen); \
> + ok1(base64_encode(dest,destlen,src,srclen) != -1); \
> + is_str(dest,expected); \
> + free(dest); \
> + } while (0)
> +
> +#define test_encode_using_alphabet(alphastring,src,srclen,expected) \
> + do { \
> + size_t destlen; \
> + char * dest; \
> + base64_alphabet_t alphabet; \
> + base64_init_alphabet(&alphabet,alphastring); \
> + destlen = base64_encoded_length(srclen); \
> + destlen++; /* null termination */ \
> + dest = xmalloc(destlen); \
> + ok1(base64_encode_using_alphabet(&alphabet,dest,destlen,src,srclen) != -1); \
> + is_str(dest,expected); \
> + free(dest); \
> + } while (0)
> +
> +/* not defined in terms of test_decode_using_alphabet so we cross
> + appropriate paths in library */
> +#define test_decode(src,srclen,expected,expectedlen) \
> + do { \
> + size_t destlen; \
> + size_t bytes_used; \
> + char * dest; \
> + destlen = base64_decoded_length(srclen); \
> + dest = xmalloc(destlen); \
> + ok1((bytes_used = base64_decode(dest,destlen,src,srclen)) != -1); \
> + is_size_t(bytes_used,expectedlen); \
> + is_mem(dest,expected,bytes_used); \
> + free(dest); \
> + } while (0)
> +
> +#define test_decode_using_alphabet(alphastring,src,srclen,expected,expectedlen) \
> + do { \
> + size_t destlen; \
> + size_t bytes_used; \
> + char * dest; \
> + base64_alphabet_t alphabet; \
> + \
> + base64_init_alphabet(&alphabet,alphastring); \
> + destlen = base64_decoded_length(srclen); \
> + dest = xmalloc(destlen); \
> + ok1((bytes_used = base64_decode_using_alphabet(&alphabet,dest,destlen,src,srclen)) != -1); \
> + is_size_t(bytes_used,expectedlen); \
> + is_mem(dest,expected,bytes_used); \
> + free(dest); \
> + } while (0)
> +
> +#define check_bad_range_decode(stuff_to_test,stufflen) \
> +do { \
> + char dest[10]; \
> + errno = 0; \
> + is_size_t(base64_decode(dest,sizeof(dest),stuff_to_test,(size_t)stufflen), \
> + (size_t)-1); \
> + is_int(errno,ERANGE); \
> +} while (0)
> +
> +int
> +main(int argc, char *argv[])
> +{
> + plan_tests(131);
> +
> + is_size_t(base64_encoded_length(0),(size_t)0);
> + is_size_t(base64_encoded_length(1),(size_t)4);
> + is_size_t(base64_encoded_length(2),(size_t)4);
> + is_size_t(base64_encoded_length(3),(size_t)4);
> + is_size_t(base64_encoded_length(512),(size_t)684);
> +
> + /* straight from page 11 of http://tools.ietf.org/html/rfc4648 */
> + test_encode("",0,"");
> + test_encode("f",1,"Zg==");
> + test_encode("fo",2,"Zm8=");
> +
> + test_encode("foo",3,"Zm9v");
> + test_encode("foob",4,"Zm9vYg==");
> + test_encode("fooba",5,"Zm9vYmE=");
> + test_encode("foobar",6,"Zm9vYmFy");
> +
> + /* a few more */
> + test_encode("foobarb",7,"Zm9vYmFyYg==");
> + test_encode("foobarba",8,"Zm9vYmFyYmE=");
> + test_encode("foobarbaz",9,"Zm9vYmFyYmF6");
> +
> + test_encode("foobart",7,"Zm9vYmFydA==");
> +
> + test_encode("abcdefghijklmnopqrstuvwxyz",26,"YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXo=");
> + test_encode("\x05\x05\x01\x00\x07",5,"BQUBAAc=");
> +
> + test_encode("FOO",3,"Rk9P");
> + test_encode("Z",1,"Wg==");
> +
> + /* decode testing */
> +
> + test_decode("",0,"",0);
> + test_decode("Zg==",4,"f",1);
> + test_decode("Zm8=",4,"fo",2);
> + test_decode("Zm9v",4,"foo",3);
> + test_decode("Zm9vYg==",8,"foob",4);
> + test_decode("Zm9vYmE=",8,"fooba",5);
> + test_decode("Zm9vYmFy",8,"foobar",6);
> + test_decode("Zm9vYmFyYg==",12,"foobarb",7);
> + test_decode("Zm9vYmFyYmE=",12,"foobarba",8);
> + test_decode("Zm9vYmFyYmF6",12,"foobarbaz",9);
> +
> + test_decode("Rk9P",4,"FOO",3);
> +
> + test_decode("Wg==",4,"Z",1);
> + test_decode("AA==",4,"\0",1);
> + test_decode("AAA=",4,"\0\0",2);
> +
> + {
> + const char *binary = "\x01\x00\x03";
> + const size_t binarylen = 3;
> +
> + char * decoded;
> + char * encoded;
> + size_t encoded_len;
> + size_t decoded_len;
> + size_t decoded_space_required;
> +
> + size_t encoded_space_required = base64_encoded_length(binarylen);
> + encoded_space_required++; /* null termination */
> + encoded = xmalloc(encoded_space_required);
> + encoded_len = base64_encode(encoded,encoded_space_required,binary,binarylen);
> + is_mem(encoded,"AQAD",encoded_len);
> +
> + decoded_space_required = base64_decoded_length(encoded_len);
> + decoded = xmalloc(decoded_space_required);
> + decoded_len = base64_decode(decoded,decoded_space_required,encoded,encoded_len);
> + is_size_t(decoded_len,binarylen);
> + is_mem(binary,decoded,decoded_len);
> + }
> +
> + /* some expected encode failures: */
> + {
> + size_t destlen = 1;
> + char dest[destlen];
> + errno = 0;
> + is_size_t(base64_encode(dest,destlen,"A",1),(size_t)-1);
> + is_int(errno,EOVERFLOW);
> + }
> +
> + /* some expected decode failures: */
> + {
> + base64_alphabet_t alphabet;
> + const char * src = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
> + base64_init_alphabet(&alphabet,src);
> +
> + is_int(sixbit_from_b64(&alphabet,'\xfe'),(signed char)-1);
> + is_int(errno,ERANGE);
> + }
> + {
> + size_t destlen = 10;
> + char dest[destlen];
> + errno = 0;
> + is_size_t(base64_decode(dest,destlen,"A",1),(size_t)-1);
> + is_int(errno,EINVAL);
> + }
> + {
> + size_t destlen = 1;
> + char dest[destlen];
> + errno = 0;
> + is_size_t(base64_decode(dest,destlen,"A",1),(size_t)-1);
> + is_int(errno,EOVERFLOW);
> + }
> + {
> + /* (char)1 is not a valid base64 character: */
> + check_bad_range_decode("A\x01",2);
> + /* (char)255 is not a valid base64 character: (char is signed on most platforms, so this is actually < 0 */
> + check_bad_range_decode("\xff""A",2);
> + check_bad_range_decode("A\xff",2);
> + check_bad_range_decode("AA\xff",3);
> + check_bad_range_decode("A\xff""A",3);
> + check_bad_range_decode("\xff""AA",3);
> + check_bad_range_decode("AAA\xff",4);
> + check_bad_range_decode("\xff\x41\x41\x41\x41",5);
> + check_bad_range_decode("A\xff\x41\x41\x41\x41",6);
> + check_bad_range_decode("AA\xff\x41\x41\x41\x41",7);
> + check_bad_range_decode("AAA\xff\x41\x41\x41\x41",8);
> + }
> + /* trigger some failures in the sixbit-to-b64 encoder: */
> + /* this function now aborts rather than returning -1/setting errno */
> + /* { */
> + /* is_int(sixbit_to_b64(base64_alphabet_rfc4648,'\x70'),(char)-1); */
> + /* is_int(sixbit_to_b64(base64_alphabet_rfc4648,'\xff'),(char)-1); */
> + /* } */
> + /* following tests all of the mapping from b64 chars to 6-bit values: */
> + test_decode("//+FwHRSRIsFU2IhAEGD+AMPhOA=",28,"\xff\xff\x85\xc0\x74\x52\x44\x8b\x05\x53\x62\x21\x00\x41\x83\xf8\x03\x0f\x84\xe0",20);
> + test_encode("\xff\xff\x85\xc0\x74\x52\x44\x8b\x05\x53\x62\x21\x00\x41\x83\xf8\x03\x0f\x84\xe0",20,"//+FwHRSRIsFU2IhAEGD+AMPhOA=");
> +
> +
> + /* check the null-padding stuff */
> + {
> + size_t destlen = 8;
> + char dest[destlen];
> + memset(dest,'\1',sizeof(dest));
> + is_size_t(base64_encode(dest,destlen,"A",1),(size_t)4);
> + is_mem(&dest[4],"\0\0\0\0",4);
> + }
> + {
> + size_t destlen = 3;
> + char dest[destlen];
> + memset(dest,'\1',sizeof(dest));
> + is_size_t(base64_decode(dest,destlen,"Wg==",4), 1);
> + is_mem(&dest[1],"\0",2);
> + }
> +
> + /* test encoding using different alphabets */
> + {
> + char alphabet_fs_safe[64];
> + memcpy(alphabet_fs_safe,base64_alphabet_rfc4648,sizeof(alphabet_fs_safe));
> + alphabet_fs_safe[62] = '-';
> + alphabet_fs_safe[63] = '_';
> + test_encode_using_alphabet(alphabet_fs_safe,"\xff\xff\x85\xc0\x74\x52\x44\x8b\x05\x53\x62\x21\x00\x41\x83\xf8\x03\x0f\x84\xe0",20,"__-FwHRSRIsFU2IhAEGD-AMPhOA=");
> + }
> +
> + /* test decoding using different alphabets */
> + {
> + char alphabet_fs_safe[64];
> + #define src "__-FwHRSRIsFU2IhAEGD-AMPhOA="
> + #define expected "\xff\xff\x85\xc0\x74\x52\x44\x8b\x05\x53\x62\x21\x00\x41\x83\xf8\x03\x0f\x84\xe0"
> +
> + memcpy(alphabet_fs_safe,base64_alphabet_rfc4648,sizeof(alphabet_fs_safe));
> + alphabet_fs_safe[62] = '-';
> + alphabet_fs_safe[63] = '_';
> +
> + test_decode_using_alphabet(alphabet_fs_safe,src,strlen(src),expected,20);
> + #undef src
> + #undef expected
> + }
> +
> + /* explicitly test the non-alphabet encode_triplet and
> + encode_tail functions */
> + {
> + size_t destlen = 4;
> + char dest[destlen];
> + const char *src = "AB\04";
> + memset(dest,'\1',sizeof(dest));
> + base64_encode_triplet(dest,src);
> + is_mem(dest,"QUIE",sizeof(dest));
> + }
> + {
> + size_t destlen = 4;
> + char dest[destlen];
> + const char *src = "A";
> + memset(dest,'\1',sizeof(dest));
> + base64_encode_tail(dest,src,strlen(src));
> + is_mem(dest,"QQ==",sizeof(dest));
> + }
> +
> + /* test the alphabet inversion */
> + {
> + base64_alphabet_t dest;
> + const char expected_inverse[] =
> + "~~~~~" /* 0 */
> + "~~~~~" /* 5 */
> + "~~~~~" /* 10 */
> + "~~~~~" /* 15 */
> + "~~~~~" /* 20 */
> + "~~~~~" /* 25 */
> + "~~~~~" /* 30 */
> + "~~~~~" /* 35 */
> + "~~~\x3e~" /* 40 */
> + "~~\x3f" /* 45 - */
> + "\x34\x35" /* 48 0 */
> + "\x36\x37\x38\x39\x3a" /* 50 */
> + "\x3b\x3c\x3d~~" /* 55 */
> + "~~~~~" /* 60 */
> + "\x00\x01\x02\x03\x04" /* 65 A */
> + "\x05\x06\x07\x08\x09" /* 70 */
> + "\x0a\x0b\x0c\x0d\x0e" /* 75 */
> + "\x0f\x10\x11\x12\x13" /* 80 */
> + "\x14\x15\x16\x17\x18" /* 85 */
> + "\x19~~~~" /* 90 */
> + "~~" /* 95 _ */
> + "\x1a\x1b\x1c" /* 97 a */
> + "\x1d\x1e\x1f\x20\x21" /* 100 */
> + "\x22\x23\x24\x25\x26" /* 105 */
> + "\x27\x28\x29\x2a\x2b" /* 110 */
> + "\x2c\x2d\x2e\x2f\x30" /* 115 */
> + "\x31\x32\x33~~" /* 120 */
> + "~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" /* 125 */
> + "~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" /* 175 */
> + "~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~" /* 225 */
> + ;
> + const char * src = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
> + base64_init_alphabet(&dest, src);
> + is_mem(dest.decode_map, expected_inverse, 256);
> + ok1(base64_char_in_alphabet(&dest,'A'));
> + ok1(!base64_char_in_alphabet(&dest,'\n'));
> + }
> +
> + /* explicitly test the non-alpha decode_tail and decode_quartet */
> + {
> + char dest[4];
> + const char *src = "QQ==";
> + const char * expected = "A";
> + memset(dest, '%', sizeof(dest));
> + base64_decode_tail(dest,src,4);
> + is_mem(dest, expected, 1);
> + }
> + {
> + char dest[4];
> + const char *src = "Zm9v";
> + const char * expected = "foo";
> + memset(dest, '%', sizeof(dest));
> + base64_decode_quartet(dest,src);
> + is_mem(dest, expected, 1);
> + }
> +
> + exit(exit_status());
> +}
> +
> +static void * xmalloc(size_t size)
> +{
> + char * ret;
> + ret = malloc(size);
> + if (ret == NULL) {
> + perror("malloc");
> + abort();
> + }
> + return ret;
> +}
> +
> +/* End of run.c test */
> --
> 1.7.10.4
>
Peter Barker | Programmer,Sysadmin,Geek.
pbarker at barker.dropbear.id.au | You need a bigger hammer.
:: It's a hack! Expect underscores! - Nigel Williams
More information about the ccan
mailing list