Support SHA3 with ARMASM.

wolfSSL · Oct 30, 2024 · 3eff52a · 3eff52a
1 parent 2536eb2
commit 3eff52a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/arch.mk b/arch.mk
@@ -80,8 +80,9 @@ ifeq ($(ARCH),AARCH64)
   endif
   ifeq ($(NO_ASM),0)
     ARCH_FLAGS=-mstrict-align
-    CFLAGS+=$(ARCH_FLAGS) -DWOLFSSL_ARMASM -DWC_HASH_DATA_ALIGNMENT=8
+    CFLAGS+=$(ARCH_FLAGS) -DWOLFSSL_ARMASM -DWOLFSSL_ARMASM_INLINE -DWC_HASH_DATA_ALIGNMENT=8
     WOLFCRYPT_OBJS += lib/wolfssl/wolfcrypt/src/port/arm/armv8-sha256.o \
+                      lib/wolfssl/wolfcrypt/src/port/arm/armv8-sha3-asm_c.o \
                       lib/wolfssl/wolfcrypt/src/port/arm/armv8-aes.o
   endif
 endif

diff --git a/include/user_settings.h b/include/user_settings.h
@@ -42,15 +42,20 @@
 /* Stdlib Types */
 #define CTYPE_USER /* don't let wolfCrypt types.h include ctype.h */
 
-#ifndef WOLFSSL_ARMASM
-#ifndef toupper
-extern int toupper(int c);
-#endif
-#ifndef tolower
-extern int tolower(int c);
-#endif
-#define XTOUPPER(c)     toupper((c))
-#define XTOLOWER(c)     tolower((c))
+#ifdef WOLFSSL_ARMASM
+    /* armv8 uses stdint types like uint64_t */
+    #ifdef __aarch64__
+        #include <stdint.h>
+    #endif
+#else
+    #ifndef toupper
+        extern int toupper(int c);
+    #endif
+    #ifndef tolower
+        extern int tolower(int c);
+    #endif
+    #define XTOUPPER(c)     toupper((c))
+    #define XTOLOWER(c)     tolower((c))
 #endif
 
 #ifdef USE_FAST_MATH