commit 56d90038fdbb655e5d65238b1a628d1b9ad1ef99
author Aditya Deshpande <aditya.deshpande@arm.com> Fri Feb 03 16:15:30 2023 +0000
committer Aditya Deshpande <aditya.deshpande@arm.com> Sun Feb 05 15:47:45 2023 +0000
tree 9bd2cb6f1d15b7ab282ab0d33a9c823f17471962
parent e87ed421be0997b046d8eb754fffc35057f6b00f

Fix more argc checks

Signed-off-by: Aditya Deshpande <aditya.deshpande@arm.com>

programs/test/query_compile_time_config.c

diff --git a/programs/test/query_compile_time_config.c b/programs/test/query_compile_time_config.c
index d846031..ff470b2 100644
--- a/programs/test/query_compile_time_config.c
+++ b/programs/test/query_compile_time_config.c
@@ -38,7 +38,7 @@
 
 int main(int argc, char *argv[])
 {
-    if (argc != 2) {
+    if (argc < 2 || strcmp(argv[1], "-h") == 0) {
         mbedtls_printf(USAGE, argv[0]);
         return MBEDTLS_EXIT_FAILURE;
     }