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};