From 153a6a67a93af0f0bd00d76670e59e741c9a13c7 Mon Sep 17 00:00:00 2001 From: Jun Zhao Date: Thu, 9 May 2019 01:16:27 +0800 Subject: tools/crypto_bench: check malloc fail before using it Need to check malloc fail before using it, so adjust the location in the code. Reviewed-by: Nicolas George Signed-off-by: Jun Zhao --- tools/crypto_bench.c | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'tools') diff --git a/tools/crypto_bench.c b/tools/crypto_bench.c index aca8bbb1a8..ac9fcc465e 100644 --- a/tools/crypto_bench.c +++ b/tools/crypto_bench.c @@ -665,8 +665,8 @@ struct hash_impl implementations[] = { int main(int argc, char **argv) { - uint8_t *input = av_malloc(MAX_INPUT_SIZE * 2); - uint8_t *output = input + MAX_INPUT_SIZE; + uint8_t *input; + uint8_t *output; unsigned i, impl, size; int opt; @@ -702,12 +702,14 @@ int main(int argc, char **argv) exit(opt != 'h'); } } - + input = av_malloc(MAX_INPUT_SIZE * 2); if (!input) fatal_error("out of memory"); for (i = 0; i < MAX_INPUT_SIZE; i += 4) AV_WB32(input + i, i); + output = input + MAX_INPUT_SIZE; + size = MAX_INPUT_SIZE; for (impl = 0; impl < FF_ARRAY_ELEMS(implementations); impl++) run_implementation(input, output, &implementations[impl], size); -- cgit v1.2.3