Various improvements, including execute ability

checkm8_payloads/src/aes.c

@@ -0,0 +1,57 @@
+#include "brfunc_aes.h"
+#include "brfunc_timing.h"
+#include "brfunc_sep.h"
+
+__attribute__ ((section (".payload_text")))
+int aes_hw_crypto_command(unsigned int cmd,
+                          void *src,
+                          void *dst,
+                          int len,
+                          unsigned int opts,
+                          void *key,
+                          void *iv)
+{
+    int seeded;
+    long start = 0, timeout = 0;
+    CLOCK_GATE(0x3C, 1);
+
+    seeded = DPA_SEEDED();
+    if(!seeded)
+    {
+        SEP_CREATE_SEND_DPA_MESSAGE();
+        start = SYSTEM_TIME();
+
+        while(!seeded && !timeout)
+        {
+            seeded = DPA_SEEDED();
+            timeout = TIME_HAS_ELAPSED(start, 1000);
+        }
+    }
+
+    if(timeout) return -1;
+
+    unsigned int key_command = CREATE_KEY_COMMAND(0, 0, 0, 0, 1, 0, 0, 0);
+    *rAES_INT_STATUS = 0x20;
+    *rAES_CONTROL = 1;
+
+    PUSH_COMMAND_KEY(key_command, key);
+    PUSH_COMMAND_IV(0, 0, 0, iv);
+    PUSH_COMMAND_DATA(0, 0, src, dst, len);
+    PUSH_COMMAND_FLAG(0, 1, 1);
+    WAIT_FOR_COMMAND_FLAG();
+
+    *rAES_CONTROL = 2;
+    CLOCK_GATE(0x3C, 0);
+    return 0;
+}
+
+int _start(unsigned int cmd,
+           void *src,
+           void *dst,
+           int len,
+           unsigned int opts,
+           void *key,
+           void *iv)
+{
+    return aes_hw_crypto_command(cmd, src, dst, len, opts, key, iv);
+}

checkm8_payloads/src/sysreg.c

@@ -0,0 +1,14 @@
+struct sysregs
+{
+    long pt_base;
+    long evt_base;
+};
+
+struct sysregs _start()
+{
+    struct sysregs res;
+    __asm__("mrs %0, ttbr1_el1" : "=r" (res.pt_base));
+    __asm__("mrs %0, vbar_el1"  : "=r" (res.evt_base));
+
+    return res;
+}