From 9d9cfa9ef9d92374c6f0fb59c16c3634fa0511f5 Mon Sep 17 00:00:00 2001 From: Bimba Shrestha Date: Thu, 30 Jan 2020 17:54:01 -0800 Subject: [PATCH] Adding assert and using ZSTD_STRATEGY_MAX as ZSTD_NB_STRATEGIES --- programs/zstdcli.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/programs/zstdcli.c b/programs/zstdcli.c index 40d1e621c..fed0a179f 100644 --- a/programs/zstdcli.c +++ b/programs/zstdcli.c @@ -544,7 +544,7 @@ static int init_cLevel(void) { return ZSTDCLI_CLEVEL_DEFAULT; } -#define ZSTD_NB_STRATEGIES 9 +#define ZSTD_NB_STRATEGIES ZSTD_STRATEGY_MAX static const char* ZSTD_strategyMap[ZSTD_NB_STRATEGIES + 1] = { "", "ZSTD_fast", "ZSTD_dfast", "ZSTD_greedy", "ZSTD_lazy", "ZSTD_lazy2", "ZSTD_btlazy2", "ZSTD_btopt", "ZSTD_btultra", "ZSTD_btultra2"}; @@ -1240,6 +1240,7 @@ int main(int const argCount, const char* argv[]) DISPLAY(" - searchLog : %u\n", cParams.searchLog); DISPLAY(" - minMatch : %u\n", cParams.minMatch); DISPLAY(" - targetLength : %u\n", cParams.targetLength); + assert(cParams.strategy < ZSTD_NB_STRATEGIES + 1); DISPLAY(" - strategy : %s\n", ZSTD_strategyMap[(int)cParams.strategy]); } }