From 06783ae22bb372e83611c718f6b15f1fde66bccb Mon Sep 17 00:00:00 2001
From: Manuel Thalmann <m@nuth.ch>
Date: Tue, 12 Dec 2023 18:58:20 +0100
Subject: [PATCH] Notify about invalid number of arguments

---
 aes-32bit/aes.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/aes-32bit/aes.cpp b/aes-32bit/aes.cpp
index 4445b93..e86abc9 100644
--- a/aes-32bit/aes.cpp
+++ b/aes-32bit/aes.cpp
@@ -1,5 +1,9 @@
+#include <cstdio>
+#include <cstdlib>
+#include <iostream>
 #include <stdio.h>
 #include <stdint.h>
+#include <system_error>
 
 /* AES-128 simple implementation template and testing */
 
@@ -184,6 +188,11 @@ void aes(uint8_t *in, uint8_t *out, uint8_t *skey)
 //****************************
 int main(int argc, char* argv[])
 {
+    if (argc > 1) {
+        std::cerr << "Invalid number of arguments\n";
+        exit(EXIT_FAILURE);
+    }
+
     for (int i = 0; i < 1000000; i++) {
         uint8_t key[16] = { 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff };
         uint8_t in[16] =  { 0xab, 0xcd, 0xef, 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0x01, 0x23, 0x45, 0x67, 0x89};