From a8c30b898c2e4dec02c75a25e72c211d9dba473d Mon Sep 17 00:00:00 2001 From: Curtis McEnroe Date: Mon, 3 Dec 2018 23:05:01 -0500 Subject: Implement base64 encoding Will be used for SASL authentication. --- Makefile | 1 + chat.h | 3 +++ pls.c | 60 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 64 insertions(+) diff --git a/Makefile b/Makefile index e09be8b..cd96001 100644 --- a/Makefile +++ b/Makefile @@ -30,6 +30,7 @@ OBJS += ui.o OBJS += url.o TESTS += format.t +TESTS += pls.t TESTS += term.t all: tags $(BINS) test diff --git a/chat.h b/chat.h index b6411d7..6c1cf0f 100644 --- a/chat.h +++ b/chat.h @@ -30,6 +30,8 @@ #define err(...) do { uiHide(); err(__VA_ARGS__); } while (0) #define errx(...) do { uiHide(); errx(__VA_ARGS__); } while (0) +typedef unsigned char byte; + struct { char *host; char *port; @@ -193,6 +195,7 @@ char *awcstombs(const wchar_t *src); char *awcsntombs(const wchar_t *src, size_t nwc); int vaswprintf(wchar_t **ret, const wchar_t *format, va_list ap); int aswprintf(wchar_t **ret, const wchar_t *format, ...); +char *base64(const byte *src, size_t len); // HACK: clang won't check wchar_t *format strings. #ifdef NDEBUG diff --git a/pls.c b/pls.c index d91fc97..4e032c8 100644 --- a/pls.c +++ b/pls.c @@ -131,3 +131,63 @@ int aswprintf(wchar_t **ret, const wchar_t *format, ...) { va_end(ap); return n; } + +static const char Base64[64] = + "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; + +char *base64(const byte *src, size_t len) { + char *dst = malloc(1 + 4 * (len + 2) / 3); + if (!dst) return NULL; + size_t i = 0; + while (len > 2) { + dst[i++] = Base64[0x3F & (src[0] >> 2)]; + dst[i++] = Base64[0x3F & (src[0] << 4 | src[1] >> 4)]; + dst[i++] = Base64[0x3F & (src[1] << 2 | src[2] >> 6)]; + dst[i++] = Base64[0x3F & src[2]]; + src += 3; + len -= 3; + } + if (len) { + dst[i++] = Base64[0x3F & (src[0] >> 2)]; + if (len > 1) { + dst[i++] = Base64[0x3F & (src[0] << 4 | src[1] >> 4)]; + dst[i++] = Base64[0x3F & (src[1] << 2)]; + } else { + dst[i++] = Base64[0x3F & (src[0] << 4)]; + dst[i++] = '='; + } + dst[i++] = '='; + } + dst[i] = '\0'; + return dst; +} + +#ifdef TEST +#include +#include + +int main() { + char *cat = base64((byte *)"cat", 3); + char *ca = base64((byte *)"ca", 2); + char *c = base64((byte *)"c", 1); + assert(cat); + assert(ca); + assert(c); + assert(!strcmp("Y2F0", cat)); + assert(!strcmp("Y2E=", ca)); + assert(!strcmp("Yw==", c)); + free(cat); + free(ca); + free(c); + + char *fzf = base64((byte []) { 0xFF, 0x00, 0xFF }, 3); + char *zfz = base64((byte []) { 0x00, 0xFF, 0x00 }, 3); + assert(fzf); + assert(zfz); + assert(!strcmp("/wD/", fzf)); + assert(!strcmp("AP8A", zfz)); + free(fzf); + free(zfz); +} + +#endif -- cgit 1.4.1 0400'>2014-06-29Add nrformats settingJune McEnroe 2014-06-28Add multiple cursorsJune McEnroe 2014-06-28Add indent guidesJune McEnroe 2014-06-10Add delimitMateJune McEnroe 2014-05-28Use ~ for home directory in gitconfigJune McEnroe 2014-05-28Add .DS_store to global gitignoreJune McEnroe 2014-05-15Add OS X specific shell configurationsJune McEnroe 2014-04-19Heavily comment .vimrcJune McEnroe 2014-04-19Autocorrect git commandsJune McEnroe 2014-04-19Fix gitprompt for git 1.9.1June McEnroe