fork of https://github.com/rust-rocksdb/rust-rocksdb for nextgraph
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
239 lines
7.4 KiB
239 lines
7.4 KiB
// Basic test for LZ4_FREESTANDING
|
|
|
|
// $ gcc -ffreestanding -nostdlib freestanding.c && ./a.out || echo $?
|
|
|
|
// $ strace ./a.out
|
|
// execve("./a.out", ["./a.out"], 0x7fffaf5fa580 /* 22 vars */) = 0
|
|
// brk(NULL) = 0x56536f4fe000
|
|
// arch_prctl(0x3001 /* ARCH_??? */, 0x7fffc9e74950) = -1 EINVAL (Invalid argument)
|
|
// mmap(NULL, 8192, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0) = 0x7fd5c9c2b000
|
|
// access("/etc/ld.so.preload", R_OK) = -1 ENOENT (No such file or directory)
|
|
// arch_prctl(ARCH_SET_FS, 0x7fd5c9c2bc40) = 0
|
|
// set_tid_address(0x7fd5c9c2bf10) = 381
|
|
// set_robust_list(0x7fd5c9c2bf20, 24) = 0
|
|
// rseq(0x7fd5c9c2c5e0, 0x20, 0, 0x53053053) = 0
|
|
// mprotect(0x56536ea63000, 4096, PROT_READ) = 0
|
|
// exit(0) = ?
|
|
// +++ exited with 0 +++
|
|
|
|
// $ ltrace ./a.out
|
|
// +++ exited (status 0) +++
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
#if defined(__cplusplus)
|
|
# define EXTERN_C extern "C"
|
|
#else
|
|
# define EXTERN_C
|
|
#endif
|
|
|
|
|
|
#if !defined(__x86_64__) || !defined(__linux__)
|
|
EXTERN_C void _start(void) { }
|
|
int main(int argc, char** argv) { return 0; }
|
|
#else
|
|
|
|
static void MY_exit(int exitCode);
|
|
static void MY_abort(void);
|
|
EXTERN_C void *memmove(void *dst, const void *src, size_t n);
|
|
EXTERN_C void *memcpy(void * __restrict__ dst, const void * __restrict__ src, size_t n);
|
|
EXTERN_C void *memset(void *s, int c, size_t n);
|
|
EXTERN_C int memcmp(const void *s1, const void *s2, size_t n);
|
|
|
|
// LZ4/HC basic freestanding setup
|
|
#define LZ4_FREESTANDING 1
|
|
#define LZ4_memmove(dst, src, size) memmove((dst),(src),(size))
|
|
#define LZ4_memcpy(dst, src, size) memcpy((dst),(src),(size))
|
|
#define LZ4_memset(p,v,s) memset((p),(v),(s))
|
|
|
|
#include "../lib/lz4.c"
|
|
#include "../lib/lz4hc.c"
|
|
|
|
// Test for LZ4
|
|
static void test_lz4(const uint8_t* srcData, int srcSize) {
|
|
// Compress
|
|
static uint8_t compressBuffer[1024 * 1024];
|
|
const int compressedSize = LZ4_compress_default(
|
|
(const char*) srcData,
|
|
(char*) compressBuffer,
|
|
srcSize,
|
|
sizeof(compressBuffer)
|
|
);
|
|
if (compressedSize <= 0) {
|
|
MY_exit(__LINE__);
|
|
}
|
|
|
|
// Decompress
|
|
static uint8_t decompressBuffer[1024 * 1024];
|
|
const int decompressedSize = LZ4_decompress_safe(
|
|
(const char*) compressBuffer,
|
|
(char*) decompressBuffer,
|
|
compressedSize,
|
|
sizeof(decompressBuffer)
|
|
);
|
|
if (decompressedSize <= 0) {
|
|
MY_exit(__LINE__);
|
|
}
|
|
|
|
// Verify
|
|
if (decompressedSize != srcSize) {
|
|
MY_exit(__LINE__);
|
|
}
|
|
if (memcmp(srcData, decompressBuffer, srcSize) != 0) {
|
|
MY_exit(__LINE__);
|
|
}
|
|
}
|
|
|
|
|
|
// Test for LZ4HC
|
|
static void test_lz4hc(const uint8_t* srcData, int srcSize) {
|
|
// Compress
|
|
static uint8_t compressBuffer[1024 * 1024];
|
|
const int compressedSize = LZ4_compress_HC(
|
|
(const char*) srcData,
|
|
(char*) compressBuffer,
|
|
srcSize,
|
|
sizeof(compressBuffer),
|
|
LZ4HC_CLEVEL_DEFAULT
|
|
);
|
|
if (compressedSize <= 0) {
|
|
MY_exit(__LINE__);
|
|
}
|
|
|
|
// Decompress
|
|
static uint8_t decompressBuffer[1024 * 1024];
|
|
const int decompressedSize = LZ4_decompress_safe(
|
|
(const char*) compressBuffer,
|
|
(char*) decompressBuffer,
|
|
compressedSize,
|
|
sizeof(decompressBuffer)
|
|
);
|
|
if (decompressedSize <= 0) {
|
|
MY_exit(__LINE__);
|
|
}
|
|
|
|
// Verify
|
|
if (decompressedSize != srcSize) {
|
|
MY_exit(__LINE__);
|
|
}
|
|
if (memcmp(srcData, decompressBuffer, srcSize) != 0) {
|
|
MY_exit(__LINE__);
|
|
}
|
|
}
|
|
|
|
|
|
static void test(void) {
|
|
// First 256 bytes of lz4/README.md
|
|
static const uint8_t README_md[] = {
|
|
0x4c, 0x5a, 0x34, 0x20, 0x2d, 0x20, 0x45, 0x78, 0x74, 0x72, 0x65, 0x6d, 0x65, 0x6c, 0x79, 0x20,
|
|
0x66, 0x61, 0x73, 0x74, 0x20, 0x63, 0x6f, 0x6d, 0x70, 0x72, 0x65, 0x73, 0x73, 0x69, 0x6f, 0x6e,
|
|
0x0a, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d,
|
|
0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d, 0x3d,
|
|
0x3d, 0x0a, 0x0a, 0x4c, 0x5a, 0x34, 0x20, 0x69, 0x73, 0x20, 0x6c, 0x6f, 0x73, 0x73, 0x6c, 0x65,
|
|
0x73, 0x73, 0x20, 0x63, 0x6f, 0x6d, 0x70, 0x72, 0x65, 0x73, 0x73, 0x69, 0x6f, 0x6e, 0x20, 0x61,
|
|
0x6c, 0x67, 0x6f, 0x72, 0x69, 0x74, 0x68, 0x6d, 0x2c, 0x0a, 0x70, 0x72, 0x6f, 0x76, 0x69, 0x64,
|
|
0x69, 0x6e, 0x67, 0x20, 0x63, 0x6f, 0x6d, 0x70, 0x72, 0x65, 0x73, 0x73, 0x69, 0x6f, 0x6e, 0x20,
|
|
0x73, 0x70, 0x65, 0x65, 0x64, 0x20, 0x3e, 0x20, 0x35, 0x30, 0x30, 0x20, 0x4d, 0x42, 0x2f, 0x73,
|
|
0x20, 0x70, 0x65, 0x72, 0x20, 0x63, 0x6f, 0x72, 0x65, 0x2c, 0x0a, 0x73, 0x63, 0x61, 0x6c, 0x61,
|
|
0x62, 0x6c, 0x65, 0x20, 0x77, 0x69, 0x74, 0x68, 0x20, 0x6d, 0x75, 0x6c, 0x74, 0x69, 0x2d, 0x63,
|
|
0x6f, 0x72, 0x65, 0x73, 0x20, 0x43, 0x50, 0x55, 0x2e, 0x0a, 0x49, 0x74, 0x20, 0x66, 0x65, 0x61,
|
|
0x74, 0x75, 0x72, 0x65, 0x73, 0x20, 0x61, 0x6e, 0x20, 0x65, 0x78, 0x74, 0x72, 0x65, 0x6d, 0x65,
|
|
0x6c, 0x79, 0x20, 0x66, 0x61, 0x73, 0x74, 0x20, 0x64, 0x65, 0x63, 0x6f, 0x64, 0x65, 0x72, 0x2c,
|
|
0x0a, 0x77, 0x69, 0x74, 0x68, 0x20, 0x73, 0x70, 0x65, 0x65, 0x64, 0x20, 0x69, 0x6e, 0x20, 0x6d,
|
|
0x75, 0x6c, 0x74, 0x69, 0x70, 0x6c, 0x65, 0x20, 0x47, 0x42, 0x2f, 0x73, 0x20, 0x70, 0x65, 0x72,
|
|
};
|
|
|
|
static const uint8_t* srcData = README_md;
|
|
static const int srcSize = (int) sizeof(README_md);
|
|
test_lz4 (srcData, srcSize);
|
|
test_lz4hc(srcData, srcSize);
|
|
}
|
|
|
|
|
|
// low level syscall
|
|
#define SYS_exit (60)
|
|
|
|
static __inline long os_syscall1(long n, long a1) {
|
|
register long rax __asm__ ("rax") = n;
|
|
register long rdi __asm__ ("rdi") = a1;
|
|
__asm__ __volatile__ ("syscall" : "+r"(rax) : "r"(rdi) : "rcx", "r11", "memory");
|
|
return rax;
|
|
}
|
|
|
|
static void MY_exit(int exitCode) {
|
|
(void) os_syscall1(SYS_exit, exitCode);
|
|
__builtin_unreachable(); // suppress "warning: 'noreturn' function does return"
|
|
}
|
|
|
|
static void MY_abort(void) {
|
|
MY_exit(-1);
|
|
}
|
|
|
|
// https://refspecs.linuxbase.org/LSB_3.0.0/LSB-Core-generic/LSB-Core-generic/baselib---assert-fail-1.html
|
|
void __assert_fail(const char * assertion, const char * file, unsigned int line, const char * function) {
|
|
MY_abort();
|
|
}
|
|
|
|
|
|
// GCC requires memcpy, memmove, memset and memcmp.
|
|
// https://gcc.gnu.org/onlinedocs/gcc/Standards.html
|
|
// > GCC requires the freestanding environment provide memcpy, memmove, memset and memcmp.
|
|
EXTERN_C void *memmove(void *dst, const void *src, size_t n) {
|
|
uint8_t* d = (uint8_t*) dst;
|
|
const uint8_t* s = (const uint8_t*) src;
|
|
|
|
if (d > s) {
|
|
d += n;
|
|
s += n;
|
|
while (n--) {
|
|
*--d = *--s;
|
|
}
|
|
} else {
|
|
while (n--) {
|
|
*d++ = *s++;
|
|
}
|
|
}
|
|
return dst;
|
|
}
|
|
|
|
EXTERN_C void *memcpy(void * __restrict__ dst, const void * __restrict__ src, size_t n) {
|
|
return memmove(dst, src, n);
|
|
}
|
|
|
|
EXTERN_C void *memset(void *s, int c, size_t n) {
|
|
uint8_t* p = (uint8_t*) s;
|
|
while (n--) {
|
|
*p++ = (uint8_t) c;
|
|
}
|
|
return s;
|
|
}
|
|
|
|
EXTERN_C int memcmp(const void *s1, const void *s2, size_t n) {
|
|
const uint8_t* p1 = (const uint8_t*) s1;
|
|
const uint8_t* p2 = (const uint8_t*) s2;
|
|
while (n--) {
|
|
const uint8_t c1 = *p1++;
|
|
const uint8_t c2 = *p2++;
|
|
if (c1 < c2) {
|
|
return -1;
|
|
} else if (c1 > c2) {
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
//
|
|
EXTERN_C void _start(void) {
|
|
test();
|
|
MY_exit(0);
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
test();
|
|
MY_exit(0);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|