1 files changed, 124 insertions, 0 deletions
diff --git a/src/base64.c b/src/base64.c
new file mode 100644
index 0000000..0fc1caa
--- /dev/null
+++ b/src/base64.c
@@ -0,0 +1,124 @@
+#include <stdlib.h>
+#include <string.h>
+#include <stddef.h>
+
+// Stole from Jouni Malinen <j@w1.fi>
+unsigned char *base64_encode(const unsigned char *src, size_t len, size_t *out_len) {
+	const unsigned char base64_table[] =
+		"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+
+	unsigned char *out, *pos;
+	const unsigned char *end, *in;
+	size_t olen;
+
+	olen = len * 4 / 3 + 4; /* 3-byte blocks to 4-byte */
+	olen += olen / 72; /* line feeds */
+	olen++; /* nul termination */
+	if (olen < len) {
+	    return NULL; /* integer overflow */
+	}
+	out = malloc(olen);
+	if (out == NULL) {
+		return NULL;
+	}
+
+	end = src + len;
+	in = src;
+	pos = out;
+
+	while (end - in >= 3) {
+		*pos++ = base64_table[in[0] >> 2];
+		*pos++ = base64_table[((in[0] & 0x03) << 4) | (in[1] >> 4)];
+		*pos++ = base64_table[((in[1] & 0x0f) << 2) | (in[2] >> 6)];
+		*pos++ = base64_table[in[2] & 0x3f];
+		in += 3;
+	}
+
+	if (end - in) {
+		*pos++ = base64_table[in[0] >> 2];
+		if (end - in == 1) {
+			*pos++ = base64_table[(in[0] & 0x03) << 4];
+			*pos++ = '=';
+		} else {
+			*pos++ = base64_table[((in[0] & 0x03) << 4) |
+								  (in[1] >> 4)];
+			*pos++ = base64_table[(in[1] & 0x0f) << 2];
+		}
+		*pos++ = '=';
+	}
+
+	*pos = '\0';
+	if (out_len) {
+		*out_len = pos - out;
+	}
+
+	return out;
+}
+
+// Stole from Jouni Malinen <j@w1.fi>
+unsigned char *base64_decode(const unsigned char *src, size_t len, size_t *out_len) {
+	const unsigned char base64_table[] =
+		"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+
+	unsigned char dtable[256], *out, *pos, block[4], tmp;
+	size_t i, count, olen;
+	int pad = 0;
+
+	memset(dtable, 0x80, 256);
+	for (i = 0; i < sizeof(base64_table) - 1; i++) {
+		dtable[base64_table[i]] = (unsigned char) i;
+	}
+	dtable['='] = 0;
+
+	count = 0;
+	for (i = 0; i < len; i++) {
+		if (dtable[src[i]] != 0x80) {
+			count++;
+		}
+	}
+
+	if (count == 0 || count % 4) {
+		return NULL;
+	}
+
+	olen = count / 4 * 3;
+	pos = out = malloc(olen);
+	if (out == NULL) {
+		return NULL;
+	}
+
+	count = 0;
+	for (i = 0; i < len; i++) {
+		tmp = dtable[src[i]];
+		if (tmp == 0x80) {
+			continue;
+		}
+
+		if (src[i] == '=') {
+			pad++;
+		}
+		block[count] = tmp;
+		count++;
+		if (count == 4) {
+			*pos++ = (block[0] << 2) | (block[1] >> 4);
+			*pos++ = (block[1] << 4) | (block[2] >> 2);
+			*pos++ = (block[2] << 6) | block[3];
+			count = 0;
+			if (pad) {
+				if (pad == 1) {
+					pos--;
+				} else if (pad == 2) {
+					pos -= 2;
+				} else {
+					/* Invalid padding */
+					free(out);
+					return NULL;
+				}
+				break;
+			}
+		}
+	}
+
+	*out_len = pos - out;
+	return out;
+}