mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-20 11:30:18 +00:00
port base64 module from v1.x
This commit is contained in:
blueloveTH
parent
0443cbcb16
commit
1e1f310022
12
docs/modules/base64.md
Normal file
12
docs/modules/base64.md
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
---
|
||||||
|
icon: package
|
||||||
|
label: base64
|
||||||
|
---
|
||||||
|
|
||||||
|
### `base64.b64encode(b: bytes) -> bytes`
|
||||||
|
|
||||||
|
Encode bytes-like object `b` using the standard Base64 alphabet.
|
||||||
|
|
||||||
|
### `base64.b64decode(b: bytes) -> bytes`
|
||||||
|
|
||||||
|
Decode Base64 encoded bytes-like object `b`.
|
||||||
@ -15,6 +15,7 @@ void pk__add_module_traceback();
|
|||||||
void pk__add_module_enum();
|
void pk__add_module_enum();
|
||||||
void pk__add_module_inspect();
|
void pk__add_module_inspect();
|
||||||
void pk__add_module_pickle();
|
void pk__add_module_pickle();
|
||||||
|
void pk__add_module_base64();
|
||||||
void pk__add_module_importlib();
|
void pk__add_module_importlib();
|
||||||
|
|
||||||
void pk__add_module_linalg();
|
void pk__add_module_linalg();
|
||||||
|
|||||||
@ -232,6 +232,7 @@ void VM__ctor(VM* self) {
|
|||||||
pk__add_module_enum();
|
pk__add_module_enum();
|
||||||
pk__add_module_inspect();
|
pk__add_module_inspect();
|
||||||
pk__add_module_pickle();
|
pk__add_module_pickle();
|
||||||
|
pk__add_module_base64();
|
||||||
pk__add_module_importlib();
|
pk__add_module_importlib();
|
||||||
|
|
||||||
pk__add_module_conio();
|
pk__add_module_conio();
|
||||||
|
|||||||
195
src/modules/base64.c
Normal file
195
src/modules/base64.c
Normal file
@ -0,0 +1,195 @@
|
|||||||
|
#include "pocketpy/interpreter/vm.h"
|
||||||
|
#include "pocketpy/pocketpy.h"
|
||||||
|
#include <limits.h>
|
||||||
|
|
||||||
|
// https://github.com/zhicheng/base64/blob/master/base64.c
|
||||||
|
|
||||||
|
const char BASE64_PAD = '=';
|
||||||
|
const char BASE64DE_FIRST = '+';
|
||||||
|
const char BASE64DE_LAST = 'z';
|
||||||
|
|
||||||
|
/* BASE 64 encode table */
|
||||||
|
const char base64en[] = {
|
||||||
|
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',
|
||||||
|
'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',
|
||||||
|
'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X',
|
||||||
|
'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f',
|
||||||
|
'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',
|
||||||
|
'o', 'p', 'q', 'r', 's', 't', 'u', 'v',
|
||||||
|
'w', 'x', 'y', 'z', '0', '1', '2', '3',
|
||||||
|
'4', '5', '6', '7', '8', '9', '+', '/',
|
||||||
|
};
|
||||||
|
|
||||||
|
/* ASCII order for BASE 64 decode, 255 in unused character */
|
||||||
|
const unsigned char base64de[] = {
|
||||||
|
/* nul, soh, stx, etx, eot, enq, ack, bel, */
|
||||||
|
255, 255, 255, 255, 255, 255, 255, 255,
|
||||||
|
|
||||||
|
/* bs, ht, nl, vt, np, cr, so, si, */
|
||||||
|
255, 255, 255, 255, 255, 255, 255, 255,
|
||||||
|
|
||||||
|
/* dle, dc1, dc2, dc3, dc4, nak, syn, etb, */
|
||||||
|
255, 255, 255, 255, 255, 255, 255, 255,
|
||||||
|
|
||||||
|
/* can, em, sub, esc, fs, gs, rs, us, */
|
||||||
|
255, 255, 255, 255, 255, 255, 255, 255,
|
||||||
|
|
||||||
|
/* sp, '!', '"', '#', '$', '%', '&', ''', */
|
||||||
|
255, 255, 255, 255, 255, 255, 255, 255,
|
||||||
|
|
||||||
|
/* '(', ')', '*', '+', ',', '-', '.', '/', */
|
||||||
|
255, 255, 255, 62, 255, 255, 255, 63,
|
||||||
|
|
||||||
|
/* '0', '1', '2', '3', '4', '5', '6', '7', */
|
||||||
|
52, 53, 54, 55, 56, 57, 58, 59,
|
||||||
|
|
||||||
|
/* '8', '9', ':', ';', '<', '=', '>', '?', */
|
||||||
|
60, 61, 255, 255, 255, 255, 255, 255,
|
||||||
|
|
||||||
|
/* '@', 'A', 'B', 'C', 'D', 'E', 'F', 'G', */
|
||||||
|
255, 0, 1, 2, 3, 4, 5, 6,
|
||||||
|
|
||||||
|
/* 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', */
|
||||||
|
7, 8, 9, 10, 11, 12, 13, 14,
|
||||||
|
|
||||||
|
/* 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', */
|
||||||
|
15, 16, 17, 18, 19, 20, 21, 22,
|
||||||
|
|
||||||
|
/* 'X', 'Y', 'Z', '[', '\', ']', '^', '_', */
|
||||||
|
23, 24, 25, 255, 255, 255, 255, 255,
|
||||||
|
|
||||||
|
/* '`', 'a', 'b', 'c', 'd', 'e', 'f', 'g', */
|
||||||
|
255, 26, 27, 28, 29, 30, 31, 32,
|
||||||
|
|
||||||
|
/* 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', */
|
||||||
|
33, 34, 35, 36, 37, 38, 39, 40,
|
||||||
|
|
||||||
|
/* 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', */
|
||||||
|
41, 42, 43, 44, 45, 46, 47, 48,
|
||||||
|
|
||||||
|
/* 'x', 'y', 'z', '{', '|', '}', '~', del, */
|
||||||
|
49, 50, 51, 255, 255, 255, 255, 255
|
||||||
|
};
|
||||||
|
|
||||||
|
static unsigned int
|
||||||
|
base64_encode(const unsigned char *in, unsigned int inlen, char *out)
|
||||||
|
{
|
||||||
|
int s;
|
||||||
|
unsigned int i;
|
||||||
|
unsigned int j;
|
||||||
|
unsigned char c;
|
||||||
|
unsigned char l;
|
||||||
|
|
||||||
|
s = 0;
|
||||||
|
l = 0;
|
||||||
|
for (i = j = 0; i < inlen; i++) {
|
||||||
|
c = in[i];
|
||||||
|
|
||||||
|
switch (s) {
|
||||||
|
case 0:
|
||||||
|
s = 1;
|
||||||
|
out[j++] = base64en[(c >> 2) & 0x3F];
|
||||||
|
break;
|
||||||
|
case 1:
|
||||||
|
s = 2;
|
||||||
|
out[j++] = base64en[((l & 0x3) << 4) | ((c >> 4) & 0xF)];
|
||||||
|
break;
|
||||||
|
case 2:
|
||||||
|
s = 0;
|
||||||
|
out[j++] = base64en[((l & 0xF) << 2) | ((c >> 6) & 0x3)];
|
||||||
|
out[j++] = base64en[c & 0x3F];
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
l = c;
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (s) {
|
||||||
|
case 1:
|
||||||
|
out[j++] = base64en[(l & 0x3) << 4];
|
||||||
|
out[j++] = BASE64_PAD;
|
||||||
|
out[j++] = BASE64_PAD;
|
||||||
|
break;
|
||||||
|
case 2:
|
||||||
|
out[j++] = base64en[(l & 0xF) << 2];
|
||||||
|
out[j++] = BASE64_PAD;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
out[j] = 0;
|
||||||
|
|
||||||
|
return j;
|
||||||
|
}
|
||||||
|
|
||||||
|
static unsigned int
|
||||||
|
base64_decode(const char *in, unsigned int inlen, unsigned char *out)
|
||||||
|
{
|
||||||
|
unsigned int i;
|
||||||
|
unsigned int j;
|
||||||
|
unsigned char c;
|
||||||
|
|
||||||
|
if (inlen & 0x3) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = j = 0; i < inlen; i++) {
|
||||||
|
if (in[i] == BASE64_PAD) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (in[i] < BASE64DE_FIRST || in[i] > BASE64DE_LAST) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
c = base64de[(unsigned char)in[i]];
|
||||||
|
if (c == 255) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (i & 0x3) {
|
||||||
|
case 0:
|
||||||
|
out[j] = (c << 2) & 0xFF;
|
||||||
|
break;
|
||||||
|
case 1:
|
||||||
|
out[j++] |= (c >> 4) & 0x3;
|
||||||
|
out[j] = (c & 0xF) << 4;
|
||||||
|
break;
|
||||||
|
case 2:
|
||||||
|
out[j++] |= (c >> 2) & 0xF;
|
||||||
|
out[j] = (c & 0x3) << 6;
|
||||||
|
break;
|
||||||
|
case 3:
|
||||||
|
out[j++] |= c;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return j;
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool base64_b64encode(int argc, py_Ref argv) {
|
||||||
|
PY_CHECK_ARGC(1);
|
||||||
|
PY_CHECK_ARG_TYPE(0, tp_bytes);
|
||||||
|
int src_size;
|
||||||
|
unsigned char* src_data = py_tobytes(argv, &src_size);
|
||||||
|
unsigned char* dst_data = py_newbytes(py_retval(), src_size * 2);
|
||||||
|
int size = base64_encode(src_data, src_size, (char*)dst_data);
|
||||||
|
py_bytes_resize(py_retval(), size);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool base64_b64decode(int argc, py_Ref argv) {
|
||||||
|
PY_CHECK_ARGC(1);
|
||||||
|
PY_CHECK_ARG_TYPE(0, tp_bytes);
|
||||||
|
int src_size;
|
||||||
|
unsigned char* src_data = py_tobytes(argv, &src_size);
|
||||||
|
unsigned char* dst_data = py_newbytes(py_retval(), src_size);
|
||||||
|
int size = base64_decode((const char*)src_data, src_size, dst_data);
|
||||||
|
py_bytes_resize(py_retval(), size);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void pk__add_module_base64(){
|
||||||
|
py_GlobalRef mod = py_newmodule("base64");
|
||||||
|
|
||||||
|
py_bindfunc(mod, "b64encode", base64_b64encode);
|
||||||
|
py_bindfunc(mod, "b64decode", base64_b64decode);
|
||||||
|
}
|
||||||
20
tests/70_base64.py
Normal file
20
tests/70_base64.py
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
a = '测试 123'
|
||||||
|
a = a.encode()
|
||||||
|
|
||||||
|
import base64
|
||||||
|
|
||||||
|
b = base64.b64encode(a)
|
||||||
|
c = base64.b64decode(b)
|
||||||
|
|
||||||
|
assert a == c
|
||||||
|
|
||||||
|
|
||||||
|
data = [66, 110, 145, 18, 176, 13, 255, 202, 173, 109, 178, 194, 171, 198, 143, 24, 113, 46, 70, 94, 71, 140, 159, 191, 134, 230, 190, 224, 223, 94, 217, 20, 241, 138, 104, 120, 249, 91, 134, 48, 108, 49, 0, 249, 235, 225, 228, 190, 63, 204, 216, 102, 153, 51, 79, 221, 234, 252, 231, 156, 74, 23, 131, 161, 172, 157, 26, 15, 88, 28, 21, 170, 86, 177, 177, 249, 111, 230, 35, 180, 61, 140, 33, 14, 74, 238, 253, 19, 177, 76, 249, 21, 35, 105, 24, 136, 187, 121, 71, 202, 239, 235, 71, 126, 60, 37, 83, 186, 102, 114, 95, 212, 81, 48, 102, 167, 208, 66, 250, 132, 199, 137, 141, 231, 126, 219, 125, 1, 86, 87, 132, 161, 55, 166, 192, 27, 95, 27, 237, 225, 32, 240, 234, 160, 247, 143, 241, 232, 195, 117, 83, 133, 69, 178, 239, 123, 144, 172, 34, 43, 56, 136, 184, 68, 65, 70, 61, 164, 109, 134, 142, 153, 125, 154, 62, 117, 166, 86, 234, 39, 73, 207, 67, 91, 88, 220, 43, 148, 201, 185, 128, 93, 151, 210, 167, 82, 87, 246, 171, 125, 210, 46, 60, 156, 4, 173, 219, 149, 24, 226, 63, 176, 92, 103, 126, 201, 254, 6, 186, 233, 165, 169, 237, 141, 252, 0, 195, 212, 222, 186, 103, 15, 137, 41, 251, 16, 163, 22, 177, 232, 205, 58, 50, 205, 89, 249, 38, 45, 98, 42, 155, 33, 225, 232, 16, 157, 91, 246, 207, 164, 150, 214, 76, 151, 179, 203, 67, 194, 213, 83, 2, 106, 109, 254, 15, 110, 168, 19, 114, 185, 174, 20, 106, 141, 116, 222, 205, 135, 222, 110, 90, 27, 61, 6, 118, 50, 155, 6, 224, 213, 109, 98, 252, 84, 166, 77, 124, 187, 187, 113, 173, 45, 17, 232, 208, 126, 248, 239, 18, 33, 205, 117, 44, 32, 223, 1, 221, 210, 41, 67, 28, 218, 218, 161, 209, 11, 93, 250, 96, 2, 43, 157, 217, 134, 183, 24, 105, 177, 74, 214, 18, 114, 191, 64, 195, 94, 194, 19, 115, 211, 103, 49, 218, 87, 8, 199, 50, 225, 174, 222, 75, 23, 159, 76, 56, 208, 224, 172, 48, 197, 126, 159, 191, 80, 216, 148, 30, 114, 231, 142, 100, 159, 67, 77, 190, 64, 182, 21, 108, 4, 232, 73, 145, 247, 196, 220, 197, 234, 55, 241, 212, 115, 115, 142, 172, 248, 132, 117, 115, 107, 176, 230, 130, 189, 160, 150, 63, 79, 253, 240, 113, 61, 222, 46, 102, 118, 100, 208, 170, 0, 60, 154, 102, 168, 241, 159, 146, 71, 55, 244, 123, 82, 49, 64, 231, 190, 49, 51, 16, 111, 153, 209, 208, 116, 19, 68, 139, 208, 105, 248, 80, 12, 237, 29, 63, 80, 127, 1, 118, 22, 39, 83, 25, 220, 75, 31, 152, 16, 94, 254, 141, 55, 6, 89, 45, 247, 229, 209, 239, 223, 226, 124, 50, 51, 219, 110, 100, 251, 122, 53, 166, 63, 43, 116, 190, 114, 169, 72, 18, 190, 55, 4, 249, 3, 200, 99, 0, 37, 94, 50, 58, 37, 56, 154, 18, 154, 127, 123, 187, 123, 110, 131, 14, 185, 76, 193, 11, 227, 36, 184, 88, 3, 222, 126, 32, 143, 125, 180, 104, 142, 84, 22, 53, 2, 38, 188, 187, 51, 163, 189, 25, 215, 94, 190, 196, 213, 155, 23, 84, 206, 237, 125, 76, 185, 12, 111, 201, 249, 101, 50, 217, 32, 3, 37, 49, 177, 4, 10, 123, 29, 126, 106, 108, 246, 89, 42, 182, 135, 11, 152, 122, 12, 23, 159, 212, 53, 44, 244, 48, 251, 130, 109, 191, 76, 148, 226, 83, 55, 225, 100, 196, 166, 171, 108, 91, 67, 226, 207, 143, 73, 81, 95, 69, 92, 141, 150, 108, 168, 235, 1, 33, 160, 158, 62, 149, 0, 200, 228, 176, 38, 112, 18, 253, 239, 107, 214, 17, 22, 112, 255, 117, 155, 248, 59, 113, 100, 145, 101, 245, 113, 230, 167, 58, 232, 195, 51, 76, 26, 7, 94, 201, 198, 96, 93, 8, 231, 60, 139, 37, 191, 37, 101, 155, 83, 246, 181, 109, 149, 241, 96, 168, 126, 232, 54, 230, 197, 179, 214, 148, 79, 13, 27, 195, 164, 146, 183, 129, 82, 82, 177, 2, 255, 8, 85, 214, 83, 244, 237, 143, 104, 107, 28, 215, 178, 46, 71, 175, 186, 77, 191, 93, 13, 204, 154, 234, 193, 231, 49, 27, 7, 66, 53, 170, 63, 3, 172, 177, 176, 255, 249, 116, 172, 165, 78, 64, 218, 147, 214, 206, 68, 42, 186, 119, 75, 28, 141, 187, 117, 21, 89, 69, 96, 79, 211, 1, 141]
|
||||||
|
data = bytes(data)
|
||||||
|
encoded = base64.b64encode(data)
|
||||||
|
|
||||||
|
res = 'Qm6RErAN/8qtbbLCq8aPGHEuRl5HjJ+/hua+4N9e2RTximh4+VuGMGwxAPnr4eS+P8zYZpkzT93q/OecSheDoaydGg9YHBWqVrGx+W/mI7Q9jCEOSu79E7FM+RUjaRiIu3lHyu/rR348JVO6ZnJf1FEwZqfQQvqEx4mN537bfQFWV4ShN6bAG18b7eEg8Oqg94/x6MN1U4VFsu97kKwiKziIuERBRj2kbYaOmX2aPnWmVuonSc9DW1jcK5TJuYBdl9KnUlf2q33SLjycBK3blRjiP7BcZ37J/ga66aWp7Y38AMPU3rpnD4kp+xCjFrHozToyzVn5Ji1iKpsh4egQnVv2z6SW1kyXs8tDwtVTAmpt/g9uqBNyua4Uao103s2H3m5aGz0GdjKbBuDVbWL8VKZNfLu7ca0tEejQfvjvEiHNdSwg3wHd0ilDHNraodELXfpgAiud2Ya3GGmxStYScr9Aw17CE3PTZzHaVwjHMuGu3ksXn0w40OCsMMV+n79Q2JQecueOZJ9DTb5AthVsBOhJkffE3MXqN/HUc3OOrPiEdXNrsOaCvaCWP0/98HE93i5mdmTQqgA8mmao8Z+SRzf0e1IxQOe+MTMQb5nR0HQTRIvQafhQDO0dP1B/AXYWJ1MZ3EsfmBBe/o03Blkt9+XR79/ifDIz225k+3o1pj8rdL5yqUgSvjcE+QPIYwAlXjI6JTiaEpp/e7t7boMOuUzBC+MkuFgD3n4gj320aI5UFjUCJry7M6O9GddevsTVmxdUzu19TLkMb8n5ZTLZIAMlMbEECnsdfmps9lkqtocLmHoMF5/UNSz0MPuCbb9MlOJTN+FkxKarbFtD4s+PSVFfRVyNlmyo6wEhoJ4+lQDI5LAmcBL972vWERZw/3Wb+DtxZJFl9XHmpzrowzNMGgdeycZgXQjnPIslvyVlm1P2tW2V8WCofug25sWz1pRPDRvDpJK3gVJSsQL/CFXWU/Ttj2hrHNeyLkevuk2/XQ3MmurB5zEbB0I1qj8DrLGw//l0rKVOQNqT1s5EKrp3SxyNu3UVWUVgT9MBjQ=='
|
||||||
|
assert encoded.decode() == res
|
||||||
|
|
||||||
|
decoded = base64.b64decode(encoded)
|
||||||
|
assert decoded == data
|
||||||
Loading…
x
Reference in New Issue
Block a user