Annotation of libaitwww/src/base64.c, revision 1.2
1.2 ! misho 1: /*************************************************************************
! 2: * (C) 2012 AITNET ltd - Sofia/Bulgaria - <misho@aitnet.org>
! 3: * by Michael Pounov <misho@elwix.org>
! 4: *
! 5: * $Author: misho $
! 6: * $Id: base64.c,v 1.1.2.1 2012/09/17 13:05:47 misho Exp $
! 7: *
! 8: **************************************************************************
! 9: The ELWIX and AITNET software is distributed under the following
! 10: terms:
! 11:
! 12: All of the documentation and software included in the ELWIX and AITNET
! 13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
! 14:
! 15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
! 16: by Michael Pounov <misho@elwix.org>. All rights reserved.
! 17:
! 18: Redistribution and use in source and binary forms, with or without
! 19: modification, are permitted provided that the following conditions
! 20: are met:
! 21: 1. Redistributions of source code must retain the above copyright
! 22: notice, this list of conditions and the following disclaimer.
! 23: 2. Redistributions in binary form must reproduce the above copyright
! 24: notice, this list of conditions and the following disclaimer in the
! 25: documentation and/or other materials provided with the distribution.
! 26: 3. All advertising materials mentioning features or use of this software
! 27: must display the following acknowledgement:
! 28: This product includes software developed by Michael Pounov <misho@elwix.org>
! 29: ELWIX - Embedded LightWeight unIX and its contributors.
! 30: 4. Neither the name of AITNET nor the names of its contributors
! 31: may be used to endorse or promote products derived from this software
! 32: without specific prior written permission.
! 33:
! 34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
! 35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
! 36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
! 37: ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
! 38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
! 39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
! 40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
! 41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
! 42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
! 43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
! 44: SUCH DAMAGE.
! 45: */
! 46: #include "global.h"
! 47:
! 48:
! 49: static const unsigned char base64_table[65] =
! 50: "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
! 51:
! 52:
! 53: /*
! 54: * www_b64encode() - Base64 encode function
! 55: *
! 56: * @src = source data
! 57: * return: NULL error or !=NULL encoded variable, after use call io_freeVar()
! 58: */
! 59: ait_val_t *
! 60: www_b64encode(ait_val_t * __restrict src)
! 61: {
! 62: ait_val_t v, *ret = NULL;
! 63: size_t olen;
! 64: const u_char *in, *end;
! 65: u_char *pos, *out;
! 66: int len = 0;
! 67:
! 68: if (!src || AIT_ISEMPTY(src))
! 69: return NULL;
! 70: olen = AIT_LEN(src) * 4 / 3 + 4; /* 3-byte blocks to 4-byte */
! 71: olen += olen / 72; /* LF */
! 72: olen++; /* \0 */
! 73: if (olen < AIT_LEN(src)) {
! 74: www_SetErr(EINVAL, "Integer overflow");
! 75: return NULL;
! 76: }
! 77:
! 78: AIT_SET_STRSIZ(&v, olen);
! 79: pos = out = (u_char*) AIT_GET_STR(&v);
! 80: in = (u_char*) AIT_GET_STR(src);
! 81: end = in + AIT_LEN(src);
! 82:
! 83: while (end - in >= 3) {
! 84: *pos++ = base64_table[in[0] >> 2];
! 85: *pos++ = base64_table[((in[0] & 0x03) << 4) | (in[1] >> 4)];
! 86: *pos++ = base64_table[((in[1] & 0x0f) << 2) | (in[2] >> 6)];
! 87: *pos++ = base64_table[in[2] & 0x3f];
! 88: in += 3;
! 89: len += 4;
! 90: if (len >= 72) {
! 91: *pos++ = '\n';
! 92: len = 0;
! 93: }
! 94: }
! 95:
! 96: if (end - in) {
! 97: *pos++ = base64_table[in[0] >> 2];
! 98: if (end - in == 1) {
! 99: *pos++ = base64_table[(in[0] & 0x03) << 4];
! 100: *pos++ = '=';
! 101: } else {
! 102: *pos++ = base64_table[((in[0] & 0x03) << 4) | (in[1] >> 4)];
! 103: *pos++ = base64_table[(in[1] & 0x0f) << 2];
! 104: }
! 105: *pos++ = '=';
! 106: len += 4;
! 107: }
! 108:
! 109: if (len)
! 110: *pos++ = '\n';
! 111:
! 112: *pos = 0;
! 113:
! 114: if (!(ret = io_allocVar())) {
! 115: www_SetErr(io_GetErrno(), "%s", io_GetError());
! 116: return NULL;
! 117: } else {
! 118: AIT_INIT_VAL2(ret, string);
! 119: AIT_SET_STRLCPY(ret, AIT_GET_STR(&v), pos - out);
! 120: AIT_FREE_VAL(&v);
! 121: }
! 122: return ret;
! 123: }
! 124:
! 125: /*
! 126: * www_b64decode() - Base64 decode function
! 127: *
! 128: * @src = source encoded data
! 129: * return: NULL error or !=NULL decoded variable, after use call io_freeVar()
! 130: */
! 131: ait_val_t *
! 132: www_b64decode(ait_val_t * __restrict src)
! 133: {
! 134: ait_val_t v, *ret = NULL;
! 135: u_char dtable[256], *s, *pos, *out, in[4], blk[4];
! 136: register size_t i, olen, cx = 0;
! 137:
! 138: if (!src || AIT_ISEMPTY(src))
! 139: return NULL;
! 140: else
! 141: s = (u_char*) AIT_GET_STR(src);
! 142:
! 143: memset(dtable, 0x80, sizeof dtable);
! 144: for (i = 0; i < sizeof base64_table - 1; i++)
! 145: dtable[base64_table[i]] = (u_char) i;
! 146: dtable['='] = 0;
! 147:
! 148: for (i = 0; i < AIT_LEN(src); i++)
! 149: if (dtable[s[i]] != 0x80)
! 150: cx++;
! 151: if (!cx || cx % 4) {
! 152: www_SetErr(EINVAL, "Invalid argument");
! 153: return NULL;
! 154: }
! 155:
! 156: olen = cx / 4 * 3;
! 157: AIT_SET_STRSIZ(&v, olen);
! 158: pos = out = (u_char*) AIT_GET_STR(&v);
! 159:
! 160: for (cx = i = 0; i < AIT_LEN(src); i++) {
! 161: if (dtable[s[i]] == 0x80)
! 162: continue;
! 163: in[cx] = s[i];
! 164: blk[cx++] = dtable[s[i]];
! 165: if (cx == 4) {
! 166: *pos++ = (blk[0] << 2) | (blk[1] >> 4);
! 167: *pos++ = (blk[1] << 4) | (blk[2] >> 2);
! 168: *pos++ = (blk[2] << 6) | blk[3];
! 169: cx = 0;
! 170: }
! 171: }
! 172:
! 173: if (pos > out) {
! 174: if (in[2] == '=')
! 175: pos -= 2;
! 176: else if (in[3] == '=')
! 177: pos--;
! 178: }
! 179:
! 180: if (!(ret = io_allocVar())) {
! 181: www_SetErr(io_GetErrno(), "%s", io_GetError());
! 182: return NULL;
! 183: } else {
! 184: AIT_INIT_VAL2(ret, string);
! 185: AIT_SET_STRLCPY(ret, AIT_GET_STR(&v), pos - out);
! 186: AIT_FREE_VAL(&v);
! 187: }
! 188: return ret;
! 189: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>