mirror of
https://github.com/MariaDB/server.git
synced 2025-01-22 14:54:20 +01:00
102 lines
2.8 KiB
C
102 lines
2.8 KiB
C
// test sub block compression and decompression
|
|
|
|
#include <toku_portability.h>
|
|
#include "test.h"
|
|
#include <stdio.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
|
|
#include "sub_block.h"
|
|
|
|
int verbose;
|
|
|
|
static void
|
|
test_sub_block_compression(void *buf, int total_size, int my_max_sub_blocks, int n_cores) {
|
|
if (verbose)
|
|
printf("%s:%d %d %d\n", __FUNCTION__, __LINE__, total_size, my_max_sub_blocks);
|
|
|
|
int r;
|
|
|
|
int sub_block_size, n_sub_blocks;
|
|
r = choose_sub_block_size(total_size, my_max_sub_blocks, &sub_block_size, &n_sub_blocks);
|
|
assert(r == 0);
|
|
if (verbose)
|
|
printf("%s:%d %d %d\n", __FUNCTION__, __LINE__, sub_block_size, n_sub_blocks);
|
|
|
|
struct sub_block sub_blocks[n_sub_blocks];
|
|
set_all_sub_block_sizes(total_size, sub_block_size, n_sub_blocks, sub_blocks);
|
|
|
|
size_t cbuf_size_bound = get_sum_compressed_size_bound(n_sub_blocks, sub_blocks);
|
|
void *cbuf = toku_malloc(cbuf_size_bound);
|
|
assert(cbuf);
|
|
|
|
size_t cbuf_size = compress_all_sub_blocks(n_sub_blocks, sub_blocks, buf, cbuf, n_cores, NULL);
|
|
assert(cbuf_size <= cbuf_size_bound);
|
|
|
|
void *ubuf = toku_malloc(total_size);
|
|
assert(ubuf);
|
|
|
|
r = decompress_all_sub_blocks(n_sub_blocks, sub_blocks, cbuf, ubuf, n_cores, NULL);
|
|
assert(r == 0);
|
|
|
|
assert(memcmp(buf, ubuf, total_size) == 0);
|
|
|
|
toku_free(ubuf);
|
|
toku_free(cbuf);
|
|
}
|
|
|
|
static void
|
|
set_random(void *buf, int total_size) {
|
|
char *bp = (char *) buf;
|
|
for (int i = 0; i < total_size; i++)
|
|
bp[i] = random();
|
|
}
|
|
|
|
static void
|
|
run_test(int total_size, int n_cores) {
|
|
void *buf = toku_malloc(total_size);
|
|
assert(buf);
|
|
|
|
for (int my_max_sub_blocks = 1; my_max_sub_blocks <= max_sub_blocks; my_max_sub_blocks++) {
|
|
memset(buf, 0, total_size);
|
|
test_sub_block_compression(buf, total_size, my_max_sub_blocks, n_cores);
|
|
|
|
set_random(buf, total_size);
|
|
test_sub_block_compression(buf, total_size, my_max_sub_blocks, n_cores);
|
|
}
|
|
|
|
toku_free(buf);
|
|
}
|
|
int
|
|
test_main (int argc, const char *argv[]) {
|
|
int n_cores = 1;
|
|
int e = 1;
|
|
|
|
for (int i = 1; i < argc; i++) {
|
|
const char *arg = argv[i];
|
|
if (strcmp(arg, "-v") == 0 || strcmp(arg, "--verbose") == 0) {
|
|
verbose++;
|
|
continue;
|
|
}
|
|
if (strcmp(arg, "-n") == 0) {
|
|
if (i+1 < argc) {
|
|
n_cores = atoi(argv[++i]);
|
|
continue;
|
|
}
|
|
}
|
|
if (strcmp(arg, "-e") == 0) {
|
|
if (i+1 < argc) {
|
|
e = atoi(argv[++i]);
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
|
|
for (int total_size = 256*1024; total_size <= 4*1024*1024; total_size *= 2) {
|
|
for (int size = total_size - e; size <= total_size + e; size++) {
|
|
run_test(size, n_cores);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|