Annotation of embedaddon/lighttpd/src/base64.c, revision 1.1.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>