forked from Tydus/tcptunnel
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathb64.c
104 lines (84 loc) · 1.79 KB
/
b64.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#include <stdio.h>
#include <string.h>
#include <stdint.h>
static const char encode[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
"abcdefghijklmnopqrstuvwxyz0123456789+/";
static const char decode[] = "|$$$}rstuvwxyz{$$$$$$$>?@ABCDEFGHIJKLMNOPQRSTUVW"
"$$$$$$XYZ[\\]^_`abcdefghijklmnopq";
int b64_encode(const uint8_t *in, int in_len, char *out, int out_size)
{
uint8_t triple[3];
int i;
int len;
int line = 0;
int done = 0;
while(in_len){
len = 0;
for(i = 0; i < 3; i++){
if(in_len){
triple[i] = *in++;
len++;
in_len--;
}else
triple[i] = 0;
}
if(!len)
continue;
if(done + 4 >= out_size)
return -1;
*out++ = encode[triple[0] >> 2];
*out++ = encode[((triple[0] & 0x03) << 4) |
((triple[1] & 0xf0) >> 4)];
*out++ = (len > 1 ? encode[((triple[1] & 0x0f) << 2) |
((triple[2] & 0xc0) >> 6)] : '=');
*out++ = (len > 2 ? encode[triple[2] & 0x3f] : '=');
done += 4;
line += 4;
}
if(done + 1 >= out_size)
return -1;
*out++ = '\0';
return done;
}
int b64_decode(const char *in, uint8_t *out, int out_size)
{
int len;
int i;
int done = 0;
uint8_t v;
uint8_t quad[4];
while(*in){
len = 0;
for(i = 0; i < 4 && *in; i++){
v = 0;
while(*in && !v){
v = *in++;
v = (v < 43 || v > 122) ? 0 : decode[v - 43];
if(v)
v = (v == '$') ? 0 : v - 61;
if(*in){
len++;
if(v)
quad[i] = v - 1;
}else
quad[i] = 0;
}
}
if(!len)
continue;
if(out_size < (done + len - 1))
/* out buffer is too small */
return -1;
if(len >= 2)
*out++ = quad[0] << 2 | quad[1] >> 4;
if(len >= 3)
*out++ = quad[1] << 4 | quad[2] >> 2;
if(len >= 4)
*out++ = ((quad[2] << 6) & 0xc0) | quad[3];
done += len - 1;
}
if(done + 1 >= out_size)
return -1;
*out++ = '\0';
return done;
}