From ad9ce8f8502b01572826aab414fbe0177bb17640 Mon Sep 17 00:00:00 2001
From: Gregor Haas <gregorhaas1997@gmail.com>
Date: Fri, 10 Jan 2020 16:06:44 -0500
Subject: [PATCH] Loop and sleep

---
 c8_payloads/src/aes_sw.c | 7 ++++++-
 c8_remote/main.c         | 6 ++++--
 include/checkm8_config.h | 3 +--
 3 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/c8_payloads/src/aes_sw.c b/c8_payloads/src/aes_sw.c
index fd7b5e7..c0eb49f 100644
--- a/c8_payloads/src/aes_sw.c
+++ b/c8_payloads/src/aes_sw.c
@@ -1,4 +1,5 @@
 #include "util.h"
+#include "brfunc_timing.h"
 
 PAYLOAD_SECTION
 void sub_bytes(unsigned char block[16], unsigned char sbox[16][16])
@@ -145,6 +146,10 @@ unsigned int _start(unsigned char *msg, unsigned int msg_len, unsigned char *key
                     unsigned char sbox[16][16], unsigned char rc_lookup[11],
                     unsigned char mul2[256], unsigned char mul3[256])
 {
-    aes128_encrypt_ecb(msg, msg_len, key, sbox, rc_lookup, mul2, mul3);
+    while(1)
+    {
+        aes128_encrypt_ecb(msg, msg_len, key, sbox, rc_lookup, mul2, mul3);
+        task_sleep(1000);
+    }
     return 0xDEADBEEF;
 }
\ No newline at end of file
diff --git a/c8_remote/main.c b/c8_remote/main.c
index 97339fa..04b8fef 100644
--- a/c8_remote/main.c
+++ b/c8_remote/main.c
@@ -202,7 +202,9 @@ int main()
     write_aes_utils(dev);
 
     free_dev_cmd_resp(resp);
-    for(int i = 0; i < 65536; i++)
+
+    int i = 0;
+    while(1)
     {
         resp = execute_payload(dev, PAYLOAD_AES_SW, 0, 7,
                                0x180153000, 16, 0x180152000,
@@ -221,7 +223,7 @@ int main()
             printf("failed to read encrypted data from memory\n");
         }
 
-        printf("got ");
+        printf("%i) got ", i++);
         for(int j = 0; j < 16; j++)
         {
             printf("%02x", resp->data[j]);
diff --git a/include/checkm8_config.h b/include/checkm8_config.h
index 5132351..56033f0 100644
--- a/include/checkm8_config.h
+++ b/include/checkm8_config.h
@@ -1,8 +1,7 @@
 #ifndef CHECKM8_TOOL_CHECKM8_CONFIG_H
 #define CHECKM8_TOOL_CHECKM8_CONFIG_H
 
-//#define LIBUSB_LOGGING
-//#define CHECKM8_LOGGING
+#define CHECKM8_LOGGING
 
 #define WITH_ARDUINO
 #define ARDUINO_DEV "/dev/ttyACM0"