The base code has been taken from zstream project which was written by Steven Barth.
Signed-off-by: Luka Perkov <l...@openwrt.org> CC: Steven Barth <ste...@midlink.org> --- => changes in v2: Use new API: size_t b64decode(void **out, const char *in, size_t len); size_t b64encode(char **out, const void *in, size_t len); => changes in v3: Use new API: static inline int b64_decode_size(const void *in, size_t len); static inline int b64_encode_size(size_t len); size_t b64decode(void *out, const void *in, size_t len); size_t b64encode(void *out, const void *in, size_t len); In this set a few corner cases of invalid writes in b64decode were fixed. Furthermore, b64decode() works fine when in == out. => changes in v4: Use new API: static inline int b64_decode_size(size_t len) static inline int b64_encode_size(size_t len) size_t b64decode(void *out, const void *in, size_t len); size_t b64encode(void *out, const void *in, size_t len); Add terminating NULL byte and make changes reported by Jonas, John and Felix. --- CMakeLists.txt | 2 +- b64.c | 117 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ b64.h | 53 ++++++++++++++++++++++++++ 3 files changed, 171 insertions(+), 1 deletion(-) create mode 100644 b64.c create mode 100644 b64.h diff --git a/CMakeLists.txt b/CMakeLists.txt index 58381da..77f4842 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -14,7 +14,7 @@ IF(JSONC_FOUND) INCLUDE_DIRECTORIES(${JSONC_INCLUDE_DIRS}) ENDIF() -SET(SOURCES avl.c avl-cmp.c blob.c blobmsg.c uloop.c usock.c ustream.c ustream-fd.c vlist.c utils.c safe_list.c runqueue.c md5.c kvlist.c ulog.c) +SET(SOURCES avl.c avl-cmp.c blob.c blobmsg.c uloop.c usock.c ustream.c ustream-fd.c vlist.c utils.c safe_list.c runqueue.c md5.c kvlist.c ulog.c b64.c) ADD_LIBRARY(ubox SHARED ${SOURCES}) ADD_LIBRARY(ubox-static STATIC ${SOURCES}) diff --git a/b64.c b/b64.c new file mode 100644 index 0000000..b16b42d --- /dev/null +++ b/b64.c @@ -0,0 +1,117 @@ +/* + * Copyright (C) 2011 Steven Barth <ste...@midlink.org> + * Copyright (C) 2015 Luka Perkov <l...@openwrt.org> + * + * Permission to use, copy, modify, and/or distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + * + * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES + * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR + * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES + * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN + * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF + * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + */ + +#include <stdint.h> +#include <stdlib.h> + +#include "b64.h" + +static const uint8_t b64decode_tbl[] = { + 0x3e, 0xff, 0xff, 0xff, 0x3f, 0x34, 0x35, 0x36, + 0x37, 0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0xff, + 0xff, 0xff, 0x00, 0xff, 0xff, 0xff, 0x00, 0x01, + 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, + 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10, 0x11, + 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19, + 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x1a, 0x1b, + 0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23, + 0x24, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, + 0x2c, 0x2d, 0x2e, 0x2f, 0x30, 0x31, 0x32, 0x33 +}; + +size_t b64decode(void *out, const void *in, size_t len) +{ + uint8_t *o = (uint8_t *) out; + const uint8_t *data = (const uint8_t *) in; + size_t lenout, i, j; + uint32_t cv = 0; + + lenout = b64_decode_size(len); + if (!lenout) + return 0; + + o[--lenout] = '\0'; + + for (i = 0; i < len; i += 4) { + cv = 0; + for (j = 0; j < 4; j++) { + uint8_t c = data[i + j] - 43; + if (c > 79 || (c = b64decode_tbl[c]) == 0xff) + return 0; + + cv |= c; + if (j != 3) + cv <<= 6; + } + + o[2] = (uint8_t)(cv & 0xff); + o[1] = (uint8_t)((cv >> 8) & 0xff); + o[0] = (uint8_t)((cv >> 16) & 0xff); + o += 3; + } + + for (i = 1; i <= 2; i++) { + if (data[len - i] == '=') { + o[-i] = '\0'; + lenout--; + } else + break; + } + + return lenout; +} + +static const uint8_t b64encode_tbl[] = + "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; + +size_t b64encode(void *out, const void *in, size_t len) +{ + uint8_t *o = (uint8_t *) out; + const uint8_t *data = (const uint8_t *) in; + size_t lenout, pad, i; + uint32_t cv; + + lenout = b64_encode_size(len); + if (!lenout) + return 0; + + o[--lenout] = '\0'; + + for (i = 0; i < len; i += 3) { + cv = (data[i] << 16) | (data[i + 1] << 8) | data[i + 2]; + o[3] = b64encode_tbl[cv & 0x3f]; + o[2] = b64encode_tbl[(cv >> 6) & 0x3f]; + o[1] = b64encode_tbl[(cv >> 12) & 0x3f]; + o[0] = b64encode_tbl[(cv >> 18) & 0x3f]; + o += 4; + } + + pad = len % 3; + if (pad) { + cv = data[len - pad] << 16; + o[-1] = '='; + o[-2] = '='; + if (pad == 2) { + cv |= data[len - pad + 1] << 8; + o[-2] = b64encode_tbl[(cv >> 6) & 0x3f]; + } + o[-3] = b64encode_tbl[(cv >> 12) & 0x3f]; + o[-4] = b64encode_tbl[(cv >> 18) & 0x3f]; + } + + return lenout; +} diff --git a/b64.h b/b64.h new file mode 100644 index 0000000..0c41919 --- /dev/null +++ b/b64.h @@ -0,0 +1,53 @@ +/* + * Copyright (C) 2011 Steven Barth <ste...@midlink.org> + * Copyright (C) 2015 Luka Perkov <l...@openwrt.org> + * + * Permission to use, copy, modify, and/or distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + * + * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES + * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR + * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES + * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN + * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF + * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + */ + +#ifndef __LIBUBOX_B64_H +#define __LIBUBOX_B64_H + +#include <stddef.h> + +static inline int b64_decode_size(size_t len) +{ + int ret; + + if ((len == 0) || (len % 4)) + return 0; + + ret = (len / 4) * 3; + + return ret + 1; +} + +static inline int b64_encode_size(size_t len) +{ + int ret; + + if (len == 0) + return 0; + + ret = (len / 3) * 4; + + if (len % 3) + ret += 4; + + return ret + 1; +} + +size_t b64decode(void *out, const void *in, size_t len); +size_t b64encode(void *out, const void *in, size_t len); + +#endif -- 2.3.5 _______________________________________________ openwrt-devel mailing list openwrt-devel@lists.openwrt.org https://lists.openwrt.org/cgi-bin/mailman/listinfo/openwrt-devel