Annotation of embedaddon/lighttpd/src/base64.c, revision 1.1

1.1     ! misho       1: #include "first.h"
        !             2: 
        !             3: #include "base64.h"
        !             4: 
        !             5: /* reverse mapping:
        !             6:  * >= 0: base64 value
        !             7:  * -1: invalid character
        !             8:  * -2: skip character (whitespace/control)
        !             9:  * -3: padding
        !            10:  */
        !            11: 
        !            12: /* BASE64_STANDARD: "A-Z a-z 0-9 + /" maps to 0-63, pad with "=" */
        !            13: static const char base64_standard_table[66] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=";
        !            14: static const short base64_standard_reverse_table[128] = {
        !            15: /*      0   1   2   3   4   5   6   7   8   9   A   B   C   D   E   F */
        !            16:        -1, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, /* 0x00 - 0x0F */
        !            17:        -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, /* 0x10 - 0x1F */
        !            18:        -2, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, -1, -1, 63, /* 0x20 - 0x2F */
        !            19:        52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -3, -1, -1, /* 0x30 - 0x3F */
        !            20:        -1,  0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, /* 0x40 - 0x4F */
        !            21:        15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1, /* 0x50 - 0x5F */
        !            22:        -1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, /* 0x60 - 0x6F */
        !            23:        41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -1, -1, -1, -1, -1, /* 0x70 - 0x7F */
        !            24: };
        !            25: 
        !            26: /* BASE64_URL: "A-Z a-z 0-9 - _" maps to 0-63, pad with "." */
        !            27: static const char base64_url_table[66] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_.";
        !            28: static const short base64_url_reverse_table[128] = {
        !            29: /*      0   1   2   3   4   5   6   7   8   9   A   B   C   D   E   F */
        !            30:        -1, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, /* 0x00 - 0x0F */
        !            31:        -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, /* 0x10 - 0x1F */
        !            32:        -2, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -3, -1, /* 0x20 - 0x2F */
        !            33:        52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1, /* 0x30 - 0x3F */
        !            34:        -1,  0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, /* 0x40 - 0x4F */
        !            35:        15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, 63, /* 0x50 - 0x5F */
        !            36:        -1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, /* 0x60 - 0x6F */
        !            37:        41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -1, -1, -1, -1, -1, /* 0x70 - 0x7F */
        !            38: };
        !            39: 
        !            40: unsigned char* buffer_append_base64_decode(buffer *out, const char* in, size_t in_length, base64_charset charset) {
        !            41:        unsigned char *result;
        !            42:        size_t out_pos = 0; /* current output character (position) that is decoded. can contain partial result */
        !            43:        unsigned int group = 0; /* how many base64 digits in the current group were decoded already. each group has up to 4 digits */
        !            44:        size_t i;
        !            45:        const short* base64_reverse_table;
        !            46: 
        !            47:        switch (charset) {
        !            48:        case BASE64_STANDARD:
        !            49:                base64_reverse_table = base64_standard_reverse_table;
        !            50:                break;
        !            51:        case BASE64_URL:
        !            52:                base64_reverse_table = base64_url_reverse_table;
        !            53:                break;
        !            54:        default:
        !            55:                return NULL;
        !            56:        }
        !            57: 
        !            58:        result = (unsigned char *) buffer_string_prepare_append(out, 3*(in_length / 4) + 3);
        !            59: 
        !            60:        /* run through the whole string, converting as we go */
        !            61:        for (i = 0; i < in_length; i++) {
        !            62:                unsigned char c = (unsigned char) in[i];
        !            63:                short ch;
        !            64: 
        !            65:                if (c == '\0') break;
        !            66:                if (c >= 128) return NULL; /* only 7-bit characters allowed */
        !            67: 
        !            68:                ch = base64_reverse_table[c];
        !            69:                if (-3 == ch) {
        !            70:                        /* pad character; can only come after 2 base64 digits in a group */
        !            71:                        if (group < 2) return NULL;
        !            72:                        break;
        !            73:                } else if (-2 == ch) {
        !            74:                        continue; /* skip character */
        !            75:                } else if (ch < 0) {
        !            76:                        return NULL; /* invalid character, abort */
        !            77:                }
        !            78: 
        !            79:                switch(group) {
        !            80:                case 0:
        !            81:                        result[out_pos] = ch << 2;
        !            82:                        group = 1;
        !            83:                        break;
        !            84:                case 1:
        !            85:                        result[out_pos++] |= ch >> 4;
        !            86:                        result[out_pos] = (ch & 0x0f) << 4;
        !            87:                        group = 2;
        !            88:                        break;
        !            89:                case 2:
        !            90:                        result[out_pos++] |= ch >>2;
        !            91:                        result[out_pos] = (ch & 0x03) << 6;
        !            92:                        group = 3;
        !            93:                        break;
        !            94:                case 3:
        !            95:                        result[out_pos++] |= ch;
        !            96:                        group = 0;
        !            97:                        break;
        !            98:                }
        !            99:        }
        !           100: 
        !           101:        switch(group) {
        !           102:        case 0:
        !           103:                /* ended on boundary */
        !           104:                break;
        !           105:        case 1:
        !           106:                /* need at least 2 base64 digits per group */
        !           107:                return NULL;
        !           108:        case 2:
        !           109:                /* have 2 base64 digits in last group => one real octect, two zeroes padded */
        !           110:        case 3:
        !           111:                /* have 3 base64 digits in last group => two real octects, one zero padded */
        !           112: 
        !           113:                /* for both cases the current index already is on the first zero padded octet
        !           114:                 * - check it really is zero (overlapping bits) */
        !           115:                if (0 != result[out_pos]) return NULL;
        !           116:                break;
        !           117:        }
        !           118: 
        !           119:        buffer_commit(out, out_pos);
        !           120: 
        !           121:        return result;
        !           122: }
        !           123: 
        !           124: size_t li_to_base64_no_padding(char* out, size_t out_length, const unsigned char* in, size_t in_length, base64_charset charset) {
        !           125:        const size_t full_tuples = in_length / 3;
        !           126:        const size_t in_tuple_remainder = in_length % 3;
        !           127:        const size_t out_tuple_remainder = in_tuple_remainder ? 1 + in_tuple_remainder : 0;
        !           128:        const size_t require_space = 4 * full_tuples + out_tuple_remainder;
        !           129:        const char* base64_table;
        !           130:        size_t i;
        !           131:        size_t out_pos = 0;
        !           132: 
        !           133:        switch (charset) {
        !           134:        case BASE64_STANDARD:
        !           135:                base64_table = base64_standard_table;
        !           136:                break;
        !           137:        case BASE64_URL:
        !           138:                base64_table = base64_url_table;
        !           139:                break;
        !           140:        default:
        !           141:                force_assert(0 && "invalid charset");
        !           142:        }
        !           143: 
        !           144:        /* check overflows */
        !           145:        force_assert(full_tuples <= 2*full_tuples);
        !           146:        force_assert(full_tuples <= 4*full_tuples);
        !           147:        force_assert(4*full_tuples <= 4*full_tuples + out_tuple_remainder);
        !           148:        force_assert(require_space <= out_length);
        !           149: 
        !           150:        for (i = 2; i < in_length; i += 3) {
        !           151:                unsigned int v = (in[i-2] << 16) | (in[i-1] << 8) | in[i];
        !           152:                out[out_pos+3] = base64_table[v & 0x3f]; v >>= 6;
        !           153:                out[out_pos+2] = base64_table[v & 0x3f]; v >>= 6;
        !           154:                out[out_pos+1] = base64_table[v & 0x3f]; v >>= 6;
        !           155:                out[out_pos+0] = base64_table[v & 0x3f];
        !           156:                out_pos += 4;
        !           157:        }
        !           158:        switch (in_tuple_remainder) {
        !           159:        case 0:
        !           160:                break;
        !           161:        case 1:
        !           162:                {
        !           163:                        /* pretend in[i-1] = in[i] = 0, don't write last two (out_pos+3, out_pos+2) characters */
        !           164:                        unsigned int v = (in[i-2] << 4);
        !           165:                        out[out_pos+1] = base64_table[v & 0x3f]; v >>= 6;
        !           166:                        out[out_pos+0] = base64_table[v & 0x3f];
        !           167:                        out_pos += 2;
        !           168:                }
        !           169:                break;
        !           170:        case 2:
        !           171:                {
        !           172:                        /* pretend in[i] = 0, don't write last (out_pos+3) character */
        !           173:                        unsigned int v = (in[i-2] << 10) | (in[i-1] << 2);
        !           174:                        out[out_pos+2] = base64_table[v & 0x3f]; v >>= 6;
        !           175:                        out[out_pos+1] = base64_table[v & 0x3f]; v >>= 6;
        !           176:                        out[out_pos+0] = base64_table[v & 0x3f];
        !           177:                        out_pos += 3;
        !           178:                }
        !           179:                break;
        !           180:        }
        !           181:        force_assert(out_pos <= out_length);
        !           182:        return out_pos;
        !           183: }
        !           184: 
        !           185: size_t li_to_base64(char* out, size_t out_length, const unsigned char* in, size_t in_length, base64_charset charset) {
        !           186:        const size_t in_tuple_remainder = in_length % 3;
        !           187:        size_t padding_length = in_tuple_remainder ? 3 - in_tuple_remainder : 0;
        !           188: 
        !           189:        char padding;
        !           190:        size_t out_pos;
        !           191: 
        !           192:        switch (charset) {
        !           193:        case BASE64_STANDARD:
        !           194:                padding = base64_standard_table[64];
        !           195:                break;
        !           196:        case BASE64_URL:
        !           197:                padding = base64_url_table[64];
        !           198:                break;
        !           199:        default:
        !           200:                force_assert(0 && "invalid charset");
        !           201:        }
        !           202: 
        !           203:        force_assert(out_length >= padding_length);
        !           204:        out_pos = li_to_base64_no_padding(out, out_length - padding_length, in, in_length, charset);
        !           205: 
        !           206:        while (padding_length > 0) {
        !           207:                out[out_pos++] = padding;
        !           208:                padding_length--;
        !           209:        }
        !           210: 
        !           211:        force_assert(out_pos <= out_length);
        !           212: 
        !           213:        return out_pos;
        !           214: }
        !           215: 
        !           216: 
        !           217: char* buffer_append_base64_encode_no_padding(buffer *out, const unsigned char* in, size_t in_length, base64_charset charset) {
        !           218:        size_t reserve = 4*(in_length/3) + 4;
        !           219:        char* result = buffer_string_prepare_append(out, reserve);
        !           220:        size_t out_pos = li_to_base64_no_padding(result, reserve, in, in_length, charset);
        !           221: 
        !           222:        buffer_commit(out, out_pos);
        !           223: 
        !           224:        return result;
        !           225: }
        !           226: 
        !           227: char* buffer_append_base64_encode(buffer *out, const unsigned char* in, size_t in_length, base64_charset charset) {
        !           228:        size_t reserve = 4*(in_length/3) + 4;
        !           229:        char* result = buffer_string_prepare_append(out, reserve);
        !           230:        size_t out_pos = li_to_base64(result, reserve, in, in_length, charset);
        !           231: 
        !           232:        buffer_commit(out, out_pos);
        !           233: 
        !           234:        return result;
        !           235: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>