1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
|
/* SPDX-License-Identifier: LGPL-2.1+ */
#include <fcntl.h>
#include "cachunk.h"
#include "def.h"
static void test_chunk_file(void) {
uint8_t buffer[BUFFER_SIZE*4];
_cleanup_(realloc_buffer_free) ReallocBuffer rb = {}, rb2 = {};
const char *d;
char *path;
_cleanup_(safe_closep) int fd = -1;
int r;
assert(var_tmp_dir(&d) >= 0);
path = strjoina(d, "/chunk-test.XXXXXX");
assert_se(dev_urandom(buffer, sizeof(buffer)) >= 0);
fd = mkostemp(path, O_RDWR|O_CLOEXEC);
assert_se(fd >= 0);
assert_se(unlink(path) >= 0);
r = ca_save_and_compress_fd(fd, CA_COMPRESSION_DEFAULT, buffer, sizeof(buffer));
assert_se(r >= 0);
assert_se(lseek(fd, 0, SEEK_SET) == 0);
r = ca_load_and_decompress_fd(fd, &rb);
safe_close(r >= 0);
assert_se(realloc_buffer_size(&rb) == sizeof(buffer));
assert_se(memcmp(realloc_buffer_data(&rb), buffer, sizeof(buffer)) == 0);
realloc_buffer_empty(&rb);
r = ca_compress(CA_COMPRESSION_DEFAULT, buffer, sizeof(buffer), &rb);
assert_se(r >= 0);
assert_se(lseek(fd, 0, SEEK_SET) == 0);
assert_se(ftruncate(fd, 0) == 0);
r = ca_save_and_decompress_fd(fd, realloc_buffer_data(&rb), realloc_buffer_size(&rb));
assert_se(r >= 0);
realloc_buffer_empty(&rb);
assert_se(lseek(fd, 0, SEEK_SET) == 0);
r = ca_load_and_compress_fd(fd, CA_COMPRESSION_DEFAULT, &rb);
assert_se(r >= 0);
r = ca_decompress(realloc_buffer_data(&rb), realloc_buffer_size(&rb), &rb2);
assert_se(r >= 0);
assert_se(realloc_buffer_size(&rb2) == sizeof(buffer));
assert_se(memcmp(realloc_buffer_data(&rb2), buffer, sizeof(buffer)) == 0);
}
int main(int argc, char *argv[]) {
test_chunk_file();
return 0;
}
|