commit 9b45f6bb68f23d95841a90a226e9e36cb635def2
author Aditya Deshpande <aditya.deshpande@arm.com> Fri Feb 03 16:15:30 2023 +0000
committer Aditya Deshpande <aditya.deshpande@arm.com> Fri Feb 03 16:15:30 2023 +0000
tree 45a8f4f8365ae0c3ac4c78b53250788bd92ce20b
parent 3b18a29c13da779d7bb898feced00f7b150f69e5

Fix more argc checks

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

programs/hash/generic_sum.c

diff --git a/programs/hash/generic_sum.c b/programs/hash/generic_sum.c
index e2e49e3..995694a 100644
--- a/programs/hash/generic_sum.c
+++ b/programs/hash/generic_sum.c
@@ -169,7 +169,7 @@
 
     mbedtls_md_init(&md_ctx);
 
-    if (argc == 1) {
+    if (argc < 2) {
         const int *list;
 
         mbedtls_printf("print mode:  generic_sum <mbedtls_md> <file> <file> ...\n");

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 8f763fa..df0fe4a 100644
--- a/programs/test/query_compile_time_config.c
+++ b/programs/test/query_compile_time_config.c
@@ -40,7 +40,7 @@
 {
     int i;
 
-    if (argc == 1 || strcmp(argv[1], "-h") == 0) {
+    if (argc < 2 || strcmp(argv[1], "-h") == 0) {
         mbedtls_printf(USAGE, argv[0]);
         return MBEDTLS_EXIT_FAILURE;
     }