Annotation of parser3/src/main/pa_base64.C, revision 1.7
1.1 moko 1: /** @file
2: Parser: base64 functions impl.
3:
4: Copyright (c) 2001-2017 Art. Lebedev Studio (http://www.artlebedev.com)
5: Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
6: */
7:
8: #include "pa_base64.h"
9: #include "pa_common.h"
10:
1.7 ! moko 11: volatile const char * IDENT_PA_BASE64_C="$Id: pa_base64.C,v 1.6 2019/11/20 20:31:52 moko Exp $" IDENT_PA_BASE64_H;
1.1 moko 12:
13: /*
14: * BASE64 part
15: * Authors: Michael Zucchi <notzed@ximian.com>
16: * Jeffrey Stedfast <fejj@ximian.com>
17: *
18: * Copyright 2000-2004 Ximian, Inc. (www.ximian.com)
19: *
20: * This program is free software; you can redistribute it and/or modify
21: * it under the terms of the GNU General Public License as published by
22: * the Free Software Foundation; either version 2 of the License, or
23: * (at your option) any later version.
24: *
25: * This program is distributed in the hope that it will be useful,
26: * but WITHOUT ANY WARRANTY; without even the implied warranty of
27: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
28: * GNU General Public License for more details.
29: *
30: * You should have received a copy of the GNU General Public License
31: * along with this program; if not, write to the Free Software
32: * Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA.
33: *
34: */
35:
1.2 moko 36: static const char *base64_alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
1.3 moko 37: static const char *base64_alphabet_url_safe = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_";
38:
1.6 moko 39: Base64Options::Base64Options(bool awrap): strict(false), wrap(awrap), pad(true), abc(base64_alphabet) {}
1.3 moko 40:
41: void Base64Options::set_url_safe_abc() {
1.4 moko 42: abc = base64_alphabet_url_safe;
1.3 moko 43: }
1.1 moko 44:
45: /**
46: * g_mime_utils_base64_encode_step:
47: * @in: input stream
48: * @inlen: length of the input
49: * @out: output string
50: * @state: holds the number of bits that are stored in @save
51: * @save: leftover bits that have not yet been encoded
52: *
53: * Base64 encodes a chunk of data. Performs an 'encode step', only
54: * encodes blocks of 3 characters to the output at a time, saves
55: * left-over state in state and save (initialise to 0 on first
56: * invocation).
57: *
58: * Returns the number of bytes encoded.
59: **/
60:
61: #define BASE64_GROUPS_IN_LINE 19
62:
1.2 moko 63: static size_t g_mime_utils_base64_encode_step (const unsigned char *in, size_t inlen, unsigned char *out, int *state, int *save) {
1.1 moko 64: register const unsigned char *inptr;
65: register unsigned char *outptr;
66:
67: if (inlen <= 0)
68: return 0;
69:
70: inptr = in;
71: outptr = out;
72:
73: if (inlen + ((unsigned char *)save)[0] > 2) {
74: const unsigned char *inend = in + inlen - 2;
75: register int c1 = 0, c2 = 0, c3 = 0;
76: register int already;
77:
78: already = *state;
79:
80: switch (((char *)save)[0]) {
81: case 1: c1 = ((unsigned char *)save)[1]; goto skip1;
82: case 2: c1 = ((unsigned char *)save)[1];
83: c2 = ((unsigned char *)save)[2]; goto skip2;
84: }
85:
86: /* yes, we jump into the loop, no i'm not going to change it, its beautiful! */
87: while (inptr < inend) {
88: c1 = *inptr++;
89: skip1:
90: c2 = *inptr++;
91: skip2:
92: c3 = *inptr++;
93: *outptr++ = base64_alphabet [c1 >> 2];
94: *outptr++ = base64_alphabet [(c2 >> 4) | ((c1 & 0x3) << 4)];
95: *outptr++ = base64_alphabet [((c2 & 0x0f) << 2) | (c3 >> 6)];
96: *outptr++ = base64_alphabet [c3 & 0x3f];
97: /* this is a bit ugly ... */
98: if ((++already) >= BASE64_GROUPS_IN_LINE) {
99: *outptr++ = '\n';
100: already = 0;
101: }
102: }
103:
104: ((unsigned char *)save)[0] = 0;
105: inlen = 2 - (inptr - inend);
106: *state = already;
107: }
108:
109: //d(printf ("state = %d, inlen = %d\n", (int)((char *)save)[0], inlen));
110:
111: if (inlen > 0) {
112: register char *saveout;
113:
114: /* points to the slot for the next char to save */
115: saveout = & (((char *)save)[1]) + ((char *)save)[0];
116:
117: /* inlen can only be 0 1 or 2 */
118: switch (inlen) {
119: case 2: *saveout++ = *inptr++;
120: case 1: *saveout++ = *inptr++;
121: }
122: *(char *)save = *(char *)save+(char)inlen;
123: }
124:
125: /*d(printf ("mode = %d\nc1 = %c\nc2 = %c\n",
126: (int)((char *)save)[0],
127: (int)((char *)save)[1],
128: (int)((char *)save)[2]));*/
129:
130: return (outptr - out);
131: }
132:
133: /**
134: * g_mime_utils_base64_encode_close:
135: * @in: input stream
136: * @inlen: length of the input
137: * @out: output string
138: * @state: holds the number of bits that are stored in @save
139: * @save: leftover bits that have not yet been encoded
140: *
141: * Base64 encodes the input stream to the output stream. Call this
142: * when finished encoding data with g_mime_utils_base64_encode_step to
143: * flush off the last little bit.
144: *
145: * Returns the number of bytes encoded.
146: **/
1.2 moko 147: static size_t g_mime_utils_base64_encode_close (const unsigned char *in, size_t inlen, unsigned char *out, int *state, int *save) {
1.1 moko 148: unsigned char *outptr = out;
149: int c1, c2;
150:
151: if (inlen > 0)
152: outptr += g_mime_utils_base64_encode_step (in, inlen, outptr, state, save);
153:
154: c1 = ((unsigned char *)save)[1];
155: c2 = ((unsigned char *)save)[2];
156:
157: switch (((unsigned char *)save)[0]) {
158: case 2:
159: outptr[2] = base64_alphabet [(c2 & 0x0f) << 2];
160: goto skip;
161: case 1:
162: outptr[2] = '=';
163: skip:
164: outptr[0] = base64_alphabet [c1 >> 2];
165: outptr[1] = base64_alphabet [c2 >> 4 | ((c1 & 0x3) << 4)];
166: outptr[3] = '=';
167: outptr += 4;
168: break;
169: }
170:
171: *outptr++ = 0;
172:
173: *save = 0;
174: *state = 0;
175:
176: return (outptr - out);
177: }
178:
179: static unsigned char gmime_base64_rank[256] = {
180: 255,255,255,255,255,255,255,255,255,254,254,255,255,254,255,255,
181: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
182: 254,255,255,255,255,255,255,255,255,255,255, 62,255,255,255, 63,
1.6 moko 183: 52, 53, 54, 55, 56, 57, 58, 59, 60, 61,255,255,255,253,255,255,
1.1 moko 184: 255, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
185: 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,255,255,255,255,255,
186: 255, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
187: 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51,255,255,255,255,255,
188: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
189: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
190: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
191: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
192: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
193: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
194: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
195: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
196: };
197:
1.5 moko 198: static unsigned char gmime_base64_rank_url_safe[256] = {
199: 255,255,255,255,255,255,255,255,255,254,254,255,255,254,255,255,
200: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
201: 254,255,255,255,255,255,255,255,255,255,255,255,255, 62,255,255,
1.6 moko 202: 52, 53, 54, 55, 56, 57, 58, 59, 60, 61,255,255,255,253,255,255,
1.5 moko 203: 255, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
204: 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,255,255,255,255, 63,
205: 255, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
206: 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51,255,255,255,255,255,
207: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
208: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
209: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
210: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
211: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
212: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
213: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
214: 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,
215: };
216:
217: size_t g_mime_utils_base64_decode(const unsigned char *in, size_t inlen, unsigned char *out, Base64Options options) {
218: const unsigned char *inptr = in;
219: unsigned char *outptr = out;
220: const unsigned char *inend = in + inlen;
221:
222: int saved = 0;
223: int state = 0;
1.1 moko 224:
1.5 moko 225: unsigned char *abc_rank = options.abc == base64_alphabet ? gmime_base64_rank : gmime_base64_rank_url_safe;
226:
1.1 moko 227: /* convert 4 base64 bytes to 3 normal bytes */
228: while (inptr < inend) {
1.5 moko 229: unsigned char c = abc_rank[*inptr++];
1.1 moko 230: switch(c) {
1.6 moko 231: case 255: // non-base64 and non-whitespace chars. not allowed in strict mode
1.5 moko 232: if(options.strict)
1.4 moko 233: throw Exception(BASE64_FORMAT, 0, "Invalid base64 char on position %d is detected", inptr - in - 1);
1.6 moko 234: case 254: // whitespace chars 0x09, 0x0A, 0x0D, 0x20 are allowed in any mode
235: break;
236: case 253: // =
237: if(state < 2) {
238: if(options.strict)
239: throw Exception(BASE64_FORMAT, 0, "Unexpected '=' on position %d is detected", inptr - in - 1);
240: break;
241: }
242: if(state == 2) { // double '='
243: if(inptr == inend) {
244: if(options.strict)
245: throw Exception(BASE64_FORMAT, 0, "Unexpected end of chars");
246: break;
247: }
248: if(*inptr != '=') {
249: if(options.strict)
250: throw Exception(BASE64_FORMAT, 0, "Unexpected '=' on position %d is detected", inptr - in - 1);
251: break;
252: }
253: inptr++;
254: *outptr++ = (unsigned char)(saved >> 4);
255: } else { // single '='
256: *outptr++ = (unsigned char)(saved >> 10);
257: *outptr++ = (unsigned char)(saved >> 2);
258: }
259: state = 0;
1.1 moko 260: break;
261: default:
262: saved = (saved << 6) | c;
1.5 moko 263: state++;
264: if (state == 4) {
1.1 moko 265: *outptr++ = (unsigned char)(saved >> 16);
266: *outptr++ = (unsigned char)(saved >> 8);
267: *outptr++ = (unsigned char)(saved);
1.5 moko 268: state = 0;
1.1 moko 269: }
270: }
271: }
1.5 moko 272:
1.6 moko 273: if(state > 0) {
274: if(state > 1) {
275: if(options.pad && options.strict)
276: throw Exception(BASE64_FORMAT, 0, "Unexpected end of chars");
277: if(state == 2) {
278: *outptr++ = (unsigned char)(saved >> 4);
279: } else {
280: *outptr++ = (unsigned char)(saved >> 10);
281: *outptr++ = (unsigned char)(saved >> 2);
282: }
283: } else {
284: if(options.strict)
285: throw Exception(BASE64_FORMAT, 0, "Unexpected end of chars");
1.1 moko 286: }
287: }
1.5 moko 288:
1.7 ! moko 289: *outptr='\0'; // for text files
1.1 moko 290: return (outptr - out);
291: }
292:
1.4 moko 293: size_t pa_base64_size(size_t in_size, bool wrap){
294: size_t new_size = ((in_size / 3 + 1) * 4) + 1 /*zero terminator*/;
295: if (wrap) new_size += new_size / (BASE64_GROUPS_IN_LINE * 4) /*new lines*/;
296: return new_size;
297: }
1.1 moko 298:
1.3 moko 299: char* pa_base64_encode(const char *in, size_t in_size, Base64Options options) {
1.4 moko 300: size_t new_size = pa_base64_size(in_size, options.wrap);
1.1 moko 301: char* result = new(PointerFreeGC) char[new_size];
1.4 moko 302: int state = 0;
303: int save = 0;
304: size_t filled = g_mime_utils_base64_encode_close ((const unsigned char*)in, in_size, (unsigned char*)result, &state, &save);
1.1 moko 305:
306: //throw Exception(PARSER_RUNTIME, 0, "%d %d %d", in_size, new_size, filled);
307: assert(filled <= new_size);
308:
309: return result;
310: }
311:
312: struct File_base64_action_info {
313: unsigned char** base64;
314: };
315:
316: static void file_base64_file_action(struct stat& finfo, int f, const String& file_spec, void *context) {
1.4 moko 317: if(finfo.st_size) {
318: File_base64_action_info& info = *static_cast<File_base64_action_info *>(context);
1.5 moko 319: *info.base64 = new(PointerFreeGC) unsigned char[pa_base64_size(check_file_size(finfo.st_size, file_spec), true)];
1.1 moko 320: unsigned char* base64 = *info.base64;
1.4 moko 321: int state = 0;
322: int save = 0;
1.1 moko 323: int nCount;
324: do {
325: unsigned char buffer[FILE_BUFFER_SIZE];
326: nCount = file_block_read(f, buffer, sizeof(buffer));
327: if( nCount ){
1.4 moko 328: size_t filled = g_mime_utils_base64_encode_step ((const unsigned char*)buffer, nCount, base64, &state, &save);
329: base64 += filled;
1.1 moko 330: }
331: } while(nCount > 0);
332: g_mime_utils_base64_encode_close (0, 0, base64, &state, &save);
333: }
334: }
335:
1.3 moko 336: char* pa_base64_encode(const String& file_spec, Base64Options options){
1.4 moko 337: unsigned char* base64 = 0;
338: File_base64_action_info info = { &base64 };
1.1 moko 339:
340: file_read_action_under_lock(file_spec, "pa_base64_encode", file_base64_file_action, &info);
341:
342: return (char*)base64;
343: }
344:
1.7 ! moko 345: size_t pa_base64_decode(const char *in, size_t in_size, char*& result, Base64Options options) {
1.1 moko 346: // every 4 base64 bytes are converted into 3 normal bytes
1.4 moko 347: size_t new_size = (in_size + 3) / 4 * 3;
348: result = new(PointerFreeGC) char[new_size + 1 /*terminator*/];
1.1 moko 349:
1.7 ! moko 350: return g_mime_utils_base64_decode ((const unsigned char*)in, in_size, (unsigned char*)result, options);
1.1 moko 351: }
E-mail:
![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to pa_base64.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / parser3 / src / main