Fix more argc checks

Signed-off-by: Aditya Deshpande <[email protected]>
Mbed-TLS · Feb 3, 2023 · f9ea38f · f9ea38f
1 parent e87ed42
commit f9ea38f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/programs/hash/generic_sum.c b/programs/hash/generic_sum.c
@@ -173,7 +173,7 @@ int main(int argc, char *argv[])
 
     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");

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