mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-24 13:30:18 +00:00
50 lines
1.6 KiB
C
50 lines
1.6 KiB
C
#ifdef PK_BUILD_MODULE_LZ4
|
|
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include "pocketpy/pocketpy.h"
|
|
#include "lz4/lib/lz4.h"
|
|
|
|
static bool lz4_compress(int argc, py_Ref argv) {
|
|
PY_CHECK_ARGC(1);
|
|
PY_CHECK_ARG_TYPE(0, tp_bytes);
|
|
int src_size;
|
|
const void* src = py_tobytes(argv, &src_size);
|
|
int dst_capacity = LZ4_compressBound(src_size);
|
|
char* p = (char*)py_newbytes(py_retval(), sizeof(int) + dst_capacity);
|
|
memcpy(p, &src_size, sizeof(int));
|
|
char* dst = p + sizeof(int);
|
|
int dst_size = LZ4_compress_default(src, dst, src_size, dst_capacity);
|
|
if(dst_size <= 0) return ValueError("LZ4 compression failed");
|
|
py_bytes_resize(py_retval(), sizeof(int) + dst_size);
|
|
return true;
|
|
}
|
|
|
|
static bool lz4_decompress(int argc, py_Ref argv) {
|
|
PY_CHECK_ARGC(1);
|
|
PY_CHECK_ARG_TYPE(0, tp_bytes);
|
|
int total_size;
|
|
const int* p = (int*)py_tobytes(argv, &total_size);
|
|
const char* src = (const char*)(p + 1);
|
|
if(total_size < sizeof(int)) return ValueError("invalid LZ4 data");
|
|
int uncompressed_size = *p;
|
|
if(uncompressed_size < 0) return ValueError("invalid LZ4 data");
|
|
char* dst = (char*)py_newbytes(py_retval(), uncompressed_size);
|
|
int dst_size = LZ4_decompress_safe(src, dst, total_size - sizeof(int), uncompressed_size);
|
|
if(dst_size < 0) return ValueError("LZ4 decompression failed");
|
|
assert(dst_size == uncompressed_size);
|
|
return true;
|
|
}
|
|
|
|
void pk__add_module_lz4() {
|
|
py_Ref mod = py_newmodule("lz4");
|
|
py_bindfunc(mod, "compress", lz4_compress);
|
|
py_bindfunc(mod, "decompress", lz4_decompress);
|
|
}
|
|
|
|
#else
|
|
|
|
void pk__add_module_lz4() {}
|
|
|
|
#endif
|