FreeRDP
Loading...
Searching...
No Matches
base64.c
1
20#include <freerdp/config.h>
21
22#include <winpr/crt.h>
23
24#include <freerdp/crypto/crypto.h>
25
26static const char enc_base64[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
27static const char enc_base64url[] =
28 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_";
29
30static const signed char dec_base64url[] = {
31 -1, /* 0 000 00 00000000 NUL &#00; Null character */
32 -1, /* 1 001 01 00000001 SOH &#01; Start of Heading */
33 -1, /* 2 002 02 00000010 STX &#02; Start of Text */
34 -1, /* 3 003 03 00000011 ETX &#03; End of Text */
35 -1, /* 4 004 04 00000100 EOT &#04; End of Transmission */
36 -1, /* 5 005 05 00000101 ENQ &#05; Enquiry */
37 -1, /* 6 006 06 00000110 ACK &#06; Acknowledge */
38 -1, /* 7 007 07 00000111 BEL &#07; Bell, Alert */
39 -1, /* 8 010 08 00001000 BS &#08; Backspace */
40 -1, /* 9 011 09 00001001 HT &#09; Horizontal Tab */
41 -1, /* 10 012 0A 00001010 LF &#10; Line Feed */
42 -1, /* 11 013 0B 00001011 VT &#11; Vertical Tabulation */
43 -1, /* 12 014 0C 00001100 FF &#12; Form Feed */
44 -1, /* 13 015 0D 00001101 CR &#13; Carriage Return */
45 -1, /* 14 016 0E 00001110 SO &#14; Shift Out */
46 -1, /* 15 017 0F 00001111 SI &#15; Shift In */
47 -1, /* 16 020 10 00010000 DLE &#16; Data Link Escape */
48 -1, /* 17 021 11 00010001 DC1 &#17; Device Control One (XON) */
49 -1, /* 18 022 12 00010010 DC2 &#18; Device Control Two */
50 -1, /* 19 023 13 00010011 DC3 &#19; Device Control Three (XOFF) */
51 -1, /* 20 024 14 00010100 DC4 &#20; Device Control Four */
52 -1, /* 21 025 15 00010101 NAK &#21; Negative Acknowledge */
53 -1, /* 22 026 16 00010110 SYN &#22; Synchronous Idle */
54 -1, /* 23 027 17 00010111 ETB &#23; End of Transmission Block */
55 -1, /* 24 030 18 00011000 CAN &#24; Cancel */
56 -1, /* 25 031 19 00011001 EM &#25; End of medium */
57 -1, /* 26 032 1A 00011010 SUB &#26; Substitute */
58 -1, /* 27 033 1B 00011011 ESC &#27; Escape */
59 -1, /* 28 034 1C 00011100 FS &#28; File Separator */
60 -1, /* 29 035 1D 00011101 GS &#29; Group Separator */
61 -1, /* 30 036 1E 00011110 RS &#30; Record Separator */
62 -1, /* 31 037 1F 00011111 US &#31; Unit Separator */
63 -1, /* 32 040 20 00100000 SP &#32; Space */
64 -1, /* 33 041 21 00100001 ! &#33; &excl; Exclamation mark */
65 -1, /* 34 042 22 00100010 " &#34; &quot; Double quotes (or speech marks) */
66 -1, /* 35 043 23 00100011 # &#35; &num; Number sign */
67 -1, /* 36 044 24 00100100 $ &#36; &dollar; Dollar */
68 -1, /* 37 045 25 00100101 % &#37; &percnt; Per cent sign */
69 -1, /* 38 046 26 00100110 & &#38; &amp; Ampersand */
70 -1, /* 39 047 27 00100111 ' &#39; &apos; Single quote */
71 -1, /* 40 050 28 00101000 ( &#40; &lparen; Open parenthesis (or open
72 * bracket)
73 */
74 -1, /* 41 051 29 00101001 ) &#41; &rparen; Close parenthesis (or close
75 * bracket)
76 */
77 -1, /* 42 052 2A 00101010 * &#42; &ast; Asterisk */
78 -1, /* 43 053 2B 00101011 + &#43; &plus; Plus */
79 -1, /* 44 054 2C 00101100 , &#44; &comma; Comma */
80 62, /* 45 055 2D 00101101 - &#45; Hyphen-minus */
81 -1, /* 46 056 2E 00101110 . &#46; &period; Period, dot or full stop */
82 -1, /* 47 057 2F 00101111 / &#47; &sol; Slash or divide */
83 52, /* 48 060 30 00110000 0 &#48; Zero */
84 53, /* 49 061 31 00110001 1 &#49; One */
85 54, /* 50 062 32 00110010 2 &#50; Two */
86 55, /* 51 063 33 00110011 3 &#51; Three */
87 56, /* 52 064 34 00110100 4 &#52; Four */
88 57, /* 53 065 35 00110101 5 &#53; Five */
89 58, /* 54 066 36 00110110 6 &#54; Six */
90 59, /* 55 067 37 00110111 7 &#55; Seven */
91 60, /* 56 070 38 00111000 8 &#56; Eight */
92 61, /* 57 071 39 00111001 9 &#57; Nine */
93 -1, /* 58 072 3A 00111010 : &#58; &colon; Colon */
94 -1, /* 59 073 3B 00111011 ; &#59; &semi; Semicolon */
95 -1, /* 60 074 3C 00111100 < &#60; &lt; Less than (or open angled bracket)
96 */
97 -1, /* 61 075 3D 00111101 = &#61; &equals; Equals */
98 -1, /* 62 076 3E 00111110 > &#62; &gt; Greater than (or close angled
99 * bracket)
100 */
101 -1, /* 63 077 3F 00111111 ? &#63; &quest; Question mark */
102 -1, /* 64 100 40 01000000 @ &#64; &commat; At sign */
103 0, /* 65 101 41 01000001 A &#65; Uppercase A */
104 1, /* 66 102 42 01000010 B &#66; Uppercase B */
105 2, /* 67 103 43 01000011 C &#67; Uppercase C */
106 3, /* 68 104 44 01000100 D &#68; Uppercase D */
107 4, /* 69 105 45 01000101 E &#69; Uppercase E */
108 5, /* 70 106 46 01000110 F &#70; Uppercase F */
109 6, /* 71 107 47 01000111 G &#71; Uppercase G */
110 7, /* 72 110 48 01001000 H &#72; Uppercase H */
111 8, /* 73 111 49 01001001 I &#73; Uppercase I */
112 9, /* 74 112 4A 01001010 J &#74; Uppercase J */
113 10, /* 75 113 4B 01001011 K &#75; Uppercase K */
114 11, /* 76 114 4C 01001100 L &#76; Uppercase L */
115 12, /* 77 115 4D 01001101 M &#77; Uppercase M */
116 13, /* 78 116 4E 01001110 N &#78; Uppercase N */
117 14, /* 79 117 4F 01001111 O &#79; Uppercase O */
118 15, /* 80 120 50 01010000 P &#80; Uppercase P */
119 16, /* 81 121 51 01010001 Q &#81; Uppercase Q */
120 17, /* 82 122 52 01010010 R &#82; Uppercase R */
121 18, /* 83 123 53 01010011 S &#83; Uppercase S */
122 19, /* 84 124 54 01010100 T &#84; Uppercase T */
123 20, /* 85 125 55 01010101 U &#85; Uppercase U */
124 21, /* 86 126 56 01010110 V &#86; Uppercase V */
125 22, /* 87 127 57 01010111 W &#87; Uppercase W */
126 23, /* 88 130 58 01011000 X &#88; Uppercase X */
127 24, /* 89 131 59 01011001 Y &#89; Uppercase Y */
128 25, /* 90 132 5A 01011010 Z &#90; Uppercase Z */
129 -1, /* 91 133 5B 01011011 [ &#91; &lsqb; Opening bracket */
130 -1, /* 92 134 5C 01011100 \ &#92; &bsol; Backslash */
131 -1, /* 93 135 5D 01011101 ] &#93; &rsqb; Closing bracket */
132 -1, /* 94 136 5E 01011110 ^ &#94; &Hat; Caret - circumflex */
133 63, /* 95 137 5F 01011111 _ &#95; &lowbar; Underscore */
134 -1, /* 96 140 60 01100000 ` &#96; &grave; Grave accent */
135 26, /* 97 141 61 01100001 a &#97; Lowercase a */
136 27, /* 98 142 62 01100010 b &#98; Lowercase b */
137 28, /* 99 143 63 01100011 c &#99; Lowercase c */
138 29, /* 100 144 64 01100100 d &#100; Lowercase d */
139 30, /* 101 145 65 01100101 e &#101; Lowercase e */
140 31, /* 102 146 66 01100110 f &#102; Lowercase f */
141 32, /* 103 147 67 01100111 g &#103; Lowercase g */
142 33, /* 104 150 68 01101000 h &#104; Lowercase h */
143 34, /* 105 151 69 01101001 i &#105; Lowercase i */
144 35, /* 106 152 6A 01101010 j &#106; Lowercase j */
145 36, /* 107 153 6B 01101011 k &#107; Lowercase k */
146 37, /* 108 154 6C 01101100 l &#108; Lowercase l */
147 38, /* 109 155 6D 01101101 m &#109; Lowercase m */
148 39, /* 110 156 6E 01101110 n &#110; Lowercase n */
149 40, /* 111 157 6F 01101111 o &#111; Lowercase o */
150 41, /* 112 160 70 01110000 p &#112; Lowercase p */
151 42, /* 113 161 71 01110001 q &#113; Lowercase q */
152 43, /* 114 162 72 01110010 r &#114; Lowercase r */
153 44, /* 115 163 73 01110011 s &#115; Lowercase s */
154 45, /* 116 164 74 01110100 t &#116; Lowercase t */
155 46, /* 117 165 75 01110101 u &#117; Lowercase u */
156 47, /* 118 166 76 01110110 v &#118; Lowercase v */
157 48, /* 119 167 77 01110111 w &#119; Lowercase w */
158 49, /* 120 170 78 01111000 x &#120; Lowercase x */
159 50, /* 121 171 79 01111001 y &#121; Lowercase y */
160 51, /* 122 172 7A 01111010 z &#122; Lowercase z */
161 -1, /* 123 173 7B 01111011 { &#123; &lcub; Opening brace */
162 -1, /* 124 174 7C 01111100 | &#124; &verbar; Vertical bar */
163 -1, /* 125 175 7D 01111101 } &#125; &rcub; Closing brace */
164 -1, /* 126 176 7E 01111110 ~ &#126; &tilde; Equivalency sign - tilde */
165 -1, /* 127 177 7F 01111111 DEL &#127; Delete */
166};
167static const signed char dec_base64[] = {
168 -1, /* 0 000 00 00000000 NUL &#00; Null character */
169 -1, /* 1 001 01 00000001 SOH &#01; Start of Heading */
170 -1, /* 2 002 02 00000010 STX &#02; Start of Text */
171 -1, /* 3 003 03 00000011 ETX &#03; End of Text */
172 -1, /* 4 004 04 00000100 EOT &#04; End of Transmission */
173 -1, /* 5 005 05 00000101 ENQ &#05; Enquiry */
174 -1, /* 6 006 06 00000110 ACK &#06; Acknowledge */
175 -1, /* 7 007 07 00000111 BEL &#07; Bell, Alert */
176 -1, /* 8 010 08 00001000 BS &#08; Backspace */
177 -1, /* 9 011 09 00001001 HT &#09; Horizontal Tab */
178 -1, /* 10 012 0A 00001010 LF &#10; Line Feed */
179 -1, /* 11 013 0B 00001011 VT &#11; Vertical Tabulation */
180 -1, /* 12 014 0C 00001100 FF &#12; Form Feed */
181 -1, /* 13 015 0D 00001101 CR &#13; Carriage Return */
182 -1, /* 14 016 0E 00001110 SO &#14; Shift Out */
183 -1, /* 15 017 0F 00001111 SI &#15; Shift In */
184 -1, /* 16 020 10 00010000 DLE &#16; Data Link Escape */
185 -1, /* 17 021 11 00010001 DC1 &#17; Device Control One (XON) */
186 -1, /* 18 022 12 00010010 DC2 &#18; Device Control Two */
187 -1, /* 19 023 13 00010011 DC3 &#19; Device Control Three (XOFF) */
188 -1, /* 20 024 14 00010100 DC4 &#20; Device Control Four */
189 -1, /* 21 025 15 00010101 NAK &#21; Negative Acknowledge */
190 -1, /* 22 026 16 00010110 SYN &#22; Synchronous Idle */
191 -1, /* 23 027 17 00010111 ETB &#23; End of Transmission Block */
192 -1, /* 24 030 18 00011000 CAN &#24; Cancel */
193 -1, /* 25 031 19 00011001 EM &#25; End of medium */
194 -1, /* 26 032 1A 00011010 SUB &#26; Substitute */
195 -1, /* 27 033 1B 00011011 ESC &#27; Escape */
196 -1, /* 28 034 1C 00011100 FS &#28; File Separator */
197 -1, /* 29 035 1D 00011101 GS &#29; Group Separator */
198 -1, /* 30 036 1E 00011110 RS &#30; Record Separator */
199 -1, /* 31 037 1F 00011111 US &#31; Unit Separator */
200 -1, /* 32 040 20 00100000 SP &#32; Space */
201 -1, /* 33 041 21 00100001 ! &#33; &excl; Exclamation mark */
202 -1, /* 34 042 22 00100010 " &#34; &quot; Double quotes (or speech marks) */
203 -1, /* 35 043 23 00100011 # &#35; &num; Number sign */
204 -1, /* 36 044 24 00100100 $ &#36; &dollar; Dollar */
205 -1, /* 37 045 25 00100101 % &#37; &percnt; Per cent sign */
206 -1, /* 38 046 26 00100110 & &#38; &amp; Ampersand */
207 -1, /* 39 047 27 00100111 ' &#39; &apos; Single quote */
208 -1, /* 40 050 28 00101000 ( &#40; &lparen; Open parenthesis (or open
209 * bracket)
210 */
211 -1, /* 41 051 29 00101001 ) &#41; &rparen; Close parenthesis (or close
212 * bracket)
213 */
214 -1, /* 42 052 2A 00101010 * &#42; &ast; Asterisk */
215 62, /* 43 053 2B 00101011 + &#43; &plus; Plus */
216 -1, /* 44 054 2C 00101100 , &#44; &comma; Comma */
217 -1, /* 45 055 2D 00101101 - &#45; Hyphen-minus */
218 -1, /* 46 056 2E 00101110 . &#46; &period; Period, dot or full stop */
219 63, /* 47 057 2F 00101111 / &#47; &sol; Slash or divide */
220 52, /* 48 060 30 00110000 0 &#48; Zero */
221 53, /* 49 061 31 00110001 1 &#49; One */
222 54, /* 50 062 32 00110010 2 &#50; Two */
223 55, /* 51 063 33 00110011 3 &#51; Three */
224 56, /* 52 064 34 00110100 4 &#52; Four */
225 57, /* 53 065 35 00110101 5 &#53; Five */
226 58, /* 54 066 36 00110110 6 &#54; Six */
227 59, /* 55 067 37 00110111 7 &#55; Seven */
228 60, /* 56 070 38 00111000 8 &#56; Eight */
229 61, /* 57 071 39 00111001 9 &#57; Nine */
230 -1, /* 58 072 3A 00111010 : &#58; &colon; Colon */
231 -1, /* 59 073 3B 00111011 ; &#59; &semi; Semicolon */
232 -1, /* 60 074 3C 00111100 < &#60; &lt; Less than (or open angled bracket)
233 */
234 -1, /* 61 075 3D 00111101 = &#61; &equals; Equals */
235 -1, /* 62 076 3E 00111110 > &#62; &gt; Greater than (or close angled
236 * bracket)
237 */
238 -1, /* 63 077 3F 00111111 ? &#63; &quest; Question mark */
239 -1, /* 64 100 40 01000000 @ &#64; &commat; At sign */
240 0, /* 65 101 41 01000001 A &#65; Uppercase A */
241 1, /* 66 102 42 01000010 B &#66; Uppercase B */
242 2, /* 67 103 43 01000011 C &#67; Uppercase C */
243 3, /* 68 104 44 01000100 D &#68; Uppercase D */
244 4, /* 69 105 45 01000101 E &#69; Uppercase E */
245 5, /* 70 106 46 01000110 F &#70; Uppercase F */
246 6, /* 71 107 47 01000111 G &#71; Uppercase G */
247 7, /* 72 110 48 01001000 H &#72; Uppercase H */
248 8, /* 73 111 49 01001001 I &#73; Uppercase I */
249 9, /* 74 112 4A 01001010 J &#74; Uppercase J */
250 10, /* 75 113 4B 01001011 K &#75; Uppercase K */
251 11, /* 76 114 4C 01001100 L &#76; Uppercase L */
252 12, /* 77 115 4D 01001101 M &#77; Uppercase M */
253 13, /* 78 116 4E 01001110 N &#78; Uppercase N */
254 14, /* 79 117 4F 01001111 O &#79; Uppercase O */
255 15, /* 80 120 50 01010000 P &#80; Uppercase P */
256 16, /* 81 121 51 01010001 Q &#81; Uppercase Q */
257 17, /* 82 122 52 01010010 R &#82; Uppercase R */
258 18, /* 83 123 53 01010011 S &#83; Uppercase S */
259 19, /* 84 124 54 01010100 T &#84; Uppercase T */
260 20, /* 85 125 55 01010101 U &#85; Uppercase U */
261 21, /* 86 126 56 01010110 V &#86; Uppercase V */
262 22, /* 87 127 57 01010111 W &#87; Uppercase W */
263 23, /* 88 130 58 01011000 X &#88; Uppercase X */
264 24, /* 89 131 59 01011001 Y &#89; Uppercase Y */
265 25, /* 90 132 5A 01011010 Z &#90; Uppercase Z */
266 -1, /* 91 133 5B 01011011 [ &#91; &lsqb; Opening bracket */
267 -1, /* 92 134 5C 01011100 \ &#92; &bsol; Backslash */
268 -1, /* 93 135 5D 01011101 ] &#93; &rsqb; Closing bracket */
269 -1, /* 94 136 5E 01011110 ^ &#94; &Hat; Caret - circumflex */
270 -1, /* 95 137 5F 01011111 _ &#95; &lowbar; Underscore */
271 -1, /* 96 140 60 01100000 ` &#96; &grave; Grave accent */
272 26, /* 97 141 61 01100001 a &#97; Lowercase a */
273 27, /* 98 142 62 01100010 b &#98; Lowercase b */
274 28, /* 99 143 63 01100011 c &#99; Lowercase c */
275 29, /* 100 144 64 01100100 d &#100; Lowercase d */
276 30, /* 101 145 65 01100101 e &#101; Lowercase e */
277 31, /* 102 146 66 01100110 f &#102; Lowercase f */
278 32, /* 103 147 67 01100111 g &#103; Lowercase g */
279 33, /* 104 150 68 01101000 h &#104; Lowercase h */
280 34, /* 105 151 69 01101001 i &#105; Lowercase i */
281 35, /* 106 152 6A 01101010 j &#106; Lowercase j */
282 36, /* 107 153 6B 01101011 k &#107; Lowercase k */
283 37, /* 108 154 6C 01101100 l &#108; Lowercase l */
284 38, /* 109 155 6D 01101101 m &#109; Lowercase m */
285 39, /* 110 156 6E 01101110 n &#110; Lowercase n */
286 40, /* 111 157 6F 01101111 o &#111; Lowercase o */
287 41, /* 112 160 70 01110000 p &#112; Lowercase p */
288 42, /* 113 161 71 01110001 q &#113; Lowercase q */
289 43, /* 114 162 72 01110010 r &#114; Lowercase r */
290 44, /* 115 163 73 01110011 s &#115; Lowercase s */
291 45, /* 116 164 74 01110100 t &#116; Lowercase t */
292 46, /* 117 165 75 01110101 u &#117; Lowercase u */
293 47, /* 118 166 76 01110110 v &#118; Lowercase v */
294 48, /* 119 167 77 01110111 w &#119; Lowercase w */
295 49, /* 120 170 78 01111000 x &#120; Lowercase x */
296 50, /* 121 171 79 01111001 y &#121; Lowercase y */
297 51, /* 122 172 7A 01111010 z &#122; Lowercase z */
298 -1, /* 123 173 7B 01111011 { &#123; &lcub; Opening brace */
299 -1, /* 124 174 7C 01111100 | &#124; &verbar; Vertical bar */
300 -1, /* 125 175 7D 01111101 } &#125; &rcub; Closing brace */
301 -1, /* 126 176 7E 01111110 ~ &#126; &tilde; Equivalency sign - tilde */
302 -1, /* 127 177 7F 01111111 DEL &#127; Delete */
303};
304
305WINPR_ATTR_MALLOC(free, 1)
306static inline char* base64_encode_ex(const char* WINPR_RESTRICT alphabet,
307 WINPR_ATTR_UNUSED size_t alphabetCount,
308 const BYTE* WINPR_RESTRICT data, size_t length, BOOL pad,
309 BOOL crLf, size_t lineSize, size_t* plen)
310{
311 /* ensure the output buffer size calculation does not overflow */
312 if (length > SIZE_MAX / 4 - 3)
313 return nullptr;
314
315 size_t outLen = (length + 3) * 4 / 3;
316 size_t extra = 0;
317 if (crLf)
318 {
319 size_t nCrLf = (outLen + lineSize - 1) / lineSize;
320 extra = nCrLf * 2ull;
321 }
322 size_t outCounter = 0;
323
324 const BYTE* q = data;
325 const size_t allocsize = outLen + extra + 1ull;
326 char* p = calloc(allocsize, sizeof(char));
327 if (!p)
328 return nullptr;
329
330 char* ret = p;
331
332 /* b1, b2, b3 are input bytes
333 *
334 * 0 1 2
335 * 012345678901234567890123
336 * | b1 | b2 | b3 |
337 *
338 * [ c1 ] [ c3 ]
339 * [ c2 ] [ c4 ]
340 *
341 * c1, c2, c3, c4 are output chars in base64
342 */
343
344 /* first treat complete blocks */
345 const size_t blocks = length - (length % 3);
346 for (size_t i = 0; i < blocks; i += 3, q += 3)
347 {
348 const unsigned c = ((unsigned)q[0] << 16) + ((unsigned)q[1] << 8) + q[2];
349 const unsigned idx0 = (c & 0x00FC0000) >> 18;
350 const unsigned idx1 = (c & 0x0003F000) >> 12;
351 const unsigned idx2 = (c & 0x00000FC0) >> 6;
352 const unsigned idx3 = c & 0x0000003F;
353
354 WINPR_ASSERT(idx0 < alphabetCount);
355 WINPR_ASSERT(idx1 < alphabetCount);
356 WINPR_ASSERT(idx2 < alphabetCount);
357 WINPR_ASSERT(idx3 < alphabetCount);
358
359 *p++ = alphabet[idx0];
360 *p++ = alphabet[idx1];
361 *p++ = alphabet[idx2];
362 *p++ = alphabet[idx3];
363
364 outCounter += 4;
365 if (crLf && (outCounter % lineSize == 0))
366 {
367 *p++ = '\r';
368 *p++ = '\n';
369 }
370 }
371
372 /* then remainder */
373 switch (length % 3)
374 {
375 case 0:
376 break;
377 case 1:
378 {
379 const unsigned c = ((unsigned)q[0] << 16);
380 const unsigned idx0 = (c & 0x00FC0000) >> 18;
381 const unsigned idx1 = (c & 0x0003F000) >> 12;
382
383 WINPR_ASSERT(idx0 < alphabetCount);
384 WINPR_ASSERT(idx1 < alphabetCount);
385
386 *p++ = alphabet[idx0];
387 *p++ = alphabet[idx1];
388 if (pad)
389 {
390 *p++ = '=';
391 *p++ = '=';
392 }
393 }
394 break;
395 case 2:
396 {
397 const unsigned c = ((unsigned)q[0] << 16) + ((unsigned)q[1] << 8);
398 const unsigned idx0 = (c & 0x00FC0000) >> 18;
399 const unsigned idx1 = (c & 0x0003F000) >> 12;
400 const unsigned idx2 = (c & 0x00000FC0) >> 6;
401
402 WINPR_ASSERT(idx0 < alphabetCount);
403 WINPR_ASSERT(idx1 < alphabetCount);
404 WINPR_ASSERT(idx2 < alphabetCount);
405
406 *p++ = alphabet[idx0];
407 *p++ = alphabet[idx1];
408 *p++ = alphabet[idx2];
409 if (pad)
410 *p++ = '=';
411 }
412 break;
413 default:
414 break;
415 }
416
417 if (crLf && length % 3)
418 {
419 *p++ = '\r';
420 *p++ = '\n';
421 }
422 *p = 0;
423
424 if (plen)
425 *plen = strnlen(ret, allocsize);
426 return ret;
427}
428
429WINPR_ATTR_MALLOC(free, 1)
430static inline char* base64_encode(const char* WINPR_RESTRICT alphabet, size_t alphabetCount,
431 const BYTE* WINPR_RESTRICT data, size_t length, BOOL pad,
432 size_t* plen)
433{
434 return base64_encode_ex(alphabet, alphabetCount, data, length, pad, FALSE, 64, plen);
435}
436
437WINPR_ATTR_NODISCARD
438static inline int base64_decode_char(const signed char* WINPR_RESTRICT alphabet,
439 size_t alphabetCount, char c)
440{
441 /* ensure char is signed for this check */
442 const int ic = (int)c;
443 if ((ic <= 0) || ((size_t)ic >= alphabetCount))
444 return -1;
445
446 return alphabet[(size_t)c];
447}
448
449WINPR_ATTR_MALLOC(free, 1)
450static inline void* base64_decode(const signed char* WINPR_RESTRICT alphabet, size_t alphabetCount,
451 const char* WINPR_RESTRICT s, size_t length,
452 size_t* WINPR_RESTRICT data_len, BOOL pad)
453{
454 int n[4] = WINPR_C_ARRAY_INIT;
455 BYTE* data = nullptr;
456 const size_t remainder = length % 4;
457
458 if ((pad && remainder > 0) || (remainder == 1))
459 return nullptr;
460
461 if (!pad && remainder)
462 length += 4 - remainder;
463
464 BYTE* q = data = (BYTE*)malloc(length / 4 * 3 + 1);
465 if (!q)
466 return nullptr;
467
468 /* first treat complete blocks */
469 const size_t nBlocks = (length / 4);
470 size_t outputLen = 0;
471
472 if (nBlocks < 1)
473 {
474 free(data);
475 return nullptr;
476 }
477
478 for (size_t i = 0; i < nBlocks - 1; i++, q += 3)
479 {
480 n[0] = base64_decode_char(alphabet, alphabetCount, *s++);
481 n[1] = base64_decode_char(alphabet, alphabetCount, *s++);
482 n[2] = base64_decode_char(alphabet, alphabetCount, *s++);
483 n[3] = base64_decode_char(alphabet, alphabetCount, *s++);
484
485 if ((n[0] == -1) || (n[1] == -1) || (n[2] == -1) || (n[3] == -1))
486 goto out_free;
487
488 q[0] = (BYTE)((n[0] << 2) + (n[1] >> 4));
489 q[1] = (BYTE)(((n[1] & 15) << 4) + (n[2] >> 2));
490 q[2] = (BYTE)(((n[2] & 3) << 6) + n[3]);
491 outputLen += 3;
492 }
493
494 /* treat last block */
495 n[0] = base64_decode_char(alphabet, alphabetCount, *s++);
496 n[1] = base64_decode_char(alphabet, alphabetCount, *s++);
497 if ((n[0] == -1) || (n[1] == -1))
498 goto out_free;
499
500 n[2] = remainder == 2 ? -1 : base64_decode_char(alphabet, alphabetCount, *s++);
501 n[3] = remainder >= 2 ? -1 : base64_decode_char(alphabet, alphabetCount, *s++);
502
503 q[0] = (BYTE)((n[0] << 2) + (n[1] >> 4));
504 if (n[2] == -1)
505 {
506 /* XX== */
507 outputLen += 1;
508 if (n[3] != -1)
509 goto out_free;
510
511 q[1] = (BYTE)((n[1] & 15) << 4);
512 }
513 else if (n[3] == -1)
514 {
515 /* yyy= */
516 outputLen += 2;
517 q[1] = (BYTE)(((n[1] & 15) << 4) + (n[2] >> 2));
518 q[2] = (BYTE)((n[2] & 3) << 6);
519 }
520 else
521 {
522 /* XXXX */
523 outputLen += 3;
524 q[0] = (BYTE)((n[0] << 2) + (n[1] >> 4));
525 q[1] = (BYTE)(((n[1] & 15) << 4) + (n[2] >> 2));
526 q[2] = (BYTE)(((n[2] & 3) << 6) + n[3]);
527 }
528
529 if (data_len)
530 *data_len = outputLen;
531 data[outputLen] = '\0';
532
533 return data;
534out_free:
535 free(data);
536 return nullptr;
537}
538
539char* crypto_base64_encode_ex(const BYTE* WINPR_RESTRICT data, size_t length, BOOL withCrLf)
540{
541 return crypto_base64_encode_ex_len(data, length, withCrLf, nullptr);
542}
543
544FREERDP_API char* crypto_base64_encode_ex_len(const void* WINPR_RESTRICT data, size_t length,
545 BOOL withCrLf, size_t* plen)
546{
547 return base64_encode_ex(enc_base64, ARRAYSIZE(enc_base64), data, length, TRUE, withCrLf, 64,
548 plen);
549}
550
551char* crypto_base64_encode(const BYTE* WINPR_RESTRICT data, size_t length)
552{
553 return crypto_base64_encode_len(data, length, nullptr);
554}
555
556char* crypto_base64_encode_len(const void* WINPR_RESTRICT data, size_t length, size_t* plen)
557{
558 return base64_encode(enc_base64, ARRAYSIZE(enc_base64), data, length, TRUE, plen);
559}
560
561void crypto_base64_decode(const char* WINPR_RESTRICT enc_data, size_t length,
562 BYTE** WINPR_RESTRICT dec_data, size_t* WINPR_RESTRICT res_length)
563{
564 *dec_data =
565 base64_decode(dec_base64, ARRAYSIZE(dec_base64), enc_data, length, res_length, TRUE);
566}
567
568char* crypto_base64url_encode(const BYTE* WINPR_RESTRICT data, size_t length)
569{
570 return crypto_base64url_encode_len(data, length, nullptr);
571}
572
573char* crypto_base64url_encode_len(const void* WINPR_RESTRICT data, size_t length, size_t* plen)
574{
575 return base64_encode(enc_base64url, ARRAYSIZE(enc_base64url), data, length, FALSE, plen);
576}
577
578void crypto_base64url_decode(const char* WINPR_RESTRICT enc_data, size_t length,
579 BYTE** WINPR_RESTRICT dec_data, size_t* WINPR_RESTRICT res_length)
580{
581 *dec_data =
582 base64_decode(dec_base64url, ARRAYSIZE(dec_base64url), enc_data, length, res_length, FALSE);
583}