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
|
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include "../crypto.h"
int char_index(int c){
if(c <= 25) return 65 + c;
if(c <= 51) return 71 + c;
if(c <= 61) return (c - 52)+48;
if(c == 62) return 43;
if(c == 63) return 47;
return 61;
}
int ichar_index(int c){
if(65 <= c && c <= 90) return c - 65;
if(97 <= c && c <= 122) return c - 97 + 26;
if(48 <= c && c <= 57) return c - 48 + 52;
if(c == 47) return 63;
return 0;
}
int de_base64(char* in, char* out){
int len = 0;
for(int i = 0; in[i]!='\0'; i++) len++;
//char out[len];
for(int i = 0; i < len; i+=4){
uint8_t u1 = i>len?0:in[i];
uint8_t u2 = i+1>len?0:in[i+1];
uint8_t u3 = i+2>len?0:in[i+2];
uint8_t u4 = i+3>len?0:in[i+3];
u1 = ichar_index(u1);
u2 = ichar_index(u2);
u3 = ichar_index(u3);
u4 = ichar_index(u4);
uint8_t left = u1 << 2 | u2 >> 4;
uint8_t middle = u2 << 4 | u3 >> 2;
uint8_t right = u3 << 6 | u4;
if(u4==0) sprintf(out,"%s%c%c",out,left,middle);
else if(u3==0) sprintf(out,"%s%c",out,left);
else sprintf(out,"%s%c%c%c",out,left,middle,right);
}
return 0;
}
int en_base64(char* in, char* out){
int len = 0;
for(int i = 0; in[i]!='\0'; i++) len++;
//char out[(len+1)*3];
for(int i = 0; i < len; i+=3){
uint8_t f = i>len?0:in[i];
uint8_t s = i+1>len?0:in[i+1];
uint8_t t = i+2>len?0:in[i+2];
uint8_t i1 = f>>2;
uint8_t i2 = (uint8_t)(f<<6)>>2 | (s>>4);
uint8_t i3 = (uint8_t)(s<<4)>>2 | (t>>6);
uint8_t i4 = t & 0x3f;
if(t==0)i4 = 64;
if(s==0)i3 = 64;
sprintf(out,"%s%c%c%c%c",out,char_index(i1),char_index(i2),
char_index(i3),char_index(i4));
}
return 0;
}
int l_base64encode(lua_State* L){
size_t len;
const char* _a = lua_tolstring(L, 1, &len);
char *a = calloc(len, sizeof * a);
memcpy(a, _a, len);
char* encode = calloc(len * 3,sizeof * encode);
en_base64(a, encode);
lua_pushstring(L, encode);
free(a);
free(encode);
return 1;
};
int l_base64decode(lua_State* L){
size_t len;
const char* _a = lua_tolstring(L, 1, &len);
char *a = calloc(len,sizeof *a);
memcpy(a, _a, len);
char* encode = calloc(len, sizeof *encode);
de_base64(a, encode);
lua_pushstring(L, encode);
free(a);
free(encode);
return 1;
};
|
