Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 03:20:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix base64

Browse Source
This commit is contained in:
blueloveTH 2025-04-05 03:07:24 +08:00
parent 4bf59223b3
commit 95d59cd4c6
3 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/modules/base64.md
View File

@ -7,6 +7,6 @@ label: base64
Encode bytes-like object `b` using the standard Base64 alphabet. Encode bytes-like object `b` using the standard Base64 alphabet.
### `base64.b64decode(b: bytes) -> bytes` ### `base64.b64decode(b: str | bytes) -> bytes`
Decode Base64 encoded bytes-like object `b`. Decode Base64 encoded bytes-like object `b`.

12
src/modules/base64.c
View File

@ -178,9 +178,17 @@ static bool base64_b64encode(int argc, py_Ref argv) {
static bool base64_b64decode(int argc, py_Ref argv) { static bool base64_b64decode(int argc, py_Ref argv) {
PY_CHECK_ARGC(1); PY_CHECK_ARGC(1);
PY_CHECK_ARG_TYPE(0, tp_bytes);
int src_size; int src_size;
unsigned char* src_data = py_tobytes(argv, &src_size); void* src_data;
if(py_istype(argv, tp_str)) {
c11_sv sv = py_tosv(argv);
src_data = (void*)sv.data;
src_size = sv.size;
} else if(py_istype(argv, tp_bytes)) {
src_data = py_tobytes(argv, &src_size);
} else {
return TypeError("expect bytes or str, got %t", argv->type);
}
unsigned char* dst_data = py_newbytes(py_retval(), src_size); unsigned char* dst_data = py_newbytes(py_retval(), src_size);
int size = base64_decode((const char*)src_data, src_size, dst_data); int size = base64_decode((const char*)src_data, src_size, dst_data);
py_bytes_resize(py_retval(), size); py_bytes_resize(py_retval(), size);

4
tests/70_base64.py
View File

@ -17,4 +17,6 @@ res = 'Qm6RErAN/8qtbbLCq8aPGHEuRl5HjJ+/hua+4N9e2RTximh4+VuGMGwxAPnr4eS+P8zYZpkzT
assert encoded.decode() == res assert encoded.decode() == res
decoded = base64.b64decode(encoded) decoded = base64.b64decode(encoded)
assert decoded == data assert decoded == data
assert base64.b64decode('8J+llQ==') == '🥕'.encode()