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/hash/generic_sum.c

diff --git a/programs/hash/generic_sum.c b/programs/hash/generic_sum.c
index edb40b6..66eaee0 100644
--- a/programs/hash/generic_sum.c
+++ b/programs/hash/generic_sum.c
@@ -173,7 +173,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 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;
     }