From aece0f258adfbd72401707d39094a477106af07a Mon Sep 17 00:00:00 2001 From: Danielle Rozenblit Date: Tue, 13 Dec 2022 08:15:16 -0800 Subject: [PATCH] free memory in test case --- .../seekable_format/tests/seekable_tests.c | 22 +++++++++++-------- 1 file changed, 13 insertions(+), 9 deletions(-) diff --git a/contrib/seekable_format/tests/seekable_tests.c b/contrib/seekable_format/tests/seekable_tests.c index cf98e734f..1bb2d0e81 100644 --- a/contrib/seekable_format/tests/seekable_tests.c +++ b/contrib/seekable_format/tests/seekable_tests.c @@ -189,18 +189,18 @@ int main(int argc, const char** argv) printf("Test %u - check ZSTD magic in compressing empty string: ", testNb++); { // compressing empty string should return a zstd header - char *data_in = (char *) malloc(255 * sizeof(char)); - assert(data_in != NULL); - data_in = "\0"; - - char *data_out = (char *) malloc(255 * 255 * sizeof(char)); - assert(data_out != NULL); + size_t const capacity = 255; + char* inBuffer = malloc(capacity); + assert(inBuffer != NULL); + inBuffer[0] = '\0'; + void* const outBuffer = malloc(capacity); + assert(outBuffer != NULL); ZSTD_seekable_CStream *s = ZSTD_seekable_createCStream(); - ZSTD_seekable_initCStream(s, 1, 1, 1024 * 1024); + ZSTD_seekable_initCStream(s, 1, 1, 255); - ZSTD_inBuffer input = { data_in, 0, 0 }; - ZSTD_outBuffer output = { data_out, 255*255, 0 }; + ZSTD_inBuffer input = { .src=inBuffer, .pos=0, .size=0 }; + ZSTD_outBuffer output = { .dst=outBuffer, .pos=0, .size=capacity }; ZSTD_seekable_compressStream(s, &output, &input); ZSTD_seekable_endStream(s, &output); @@ -208,10 +208,14 @@ int main(int argc, const char** argv) if((((char*)output.dst)[0] != '\x28') | (((char*)output.dst)[1] != '\xb5') | (((char*)output.dst)[2] != '\x2f') | (((char*)output.dst)[3] != '\xfd')) { printf("%#02x %#02x %#02x %#02x\n", ((char*)output.dst)[0], ((char*)output.dst)[1] , ((char*)output.dst)[2] , ((char*)output.dst)[3] ); + free(inBuffer); + free(outBuffer); ZSTD_seekable_freeCStream(s); goto _test_error; } + free(inBuffer); + free(outBuffer); ZSTD_seekable_freeCStream(s); } printf("Success!\n"); -- 2.47.2