aarch64 JIT: support soft AES

SChernykh · Sep 27, 2023 · a99d803 · a99d803
1 parent fcf9f12
commit a99d803
Show file tree

Hide file tree

Showing 3 changed files with 527 additions and 4 deletions.
diff --git a/src/jit_compiler_a64.cpp b/src/jit_compiler_a64.cpp
@@ -173,8 +173,15 @@ void JitCompilerA64::generateProgram(Program& program, ProgramConfiguration& con
 	if (flags & RANDOMX_FLAG_V2) {
 		codePos = ((uint8_t*)randomx_program_aarch64_v2_FE_mix) - ((uint8_t*)randomx_program_aarch64);
 
-		// Disable the jump to RandomX v1 FE mix code by writing "movi v28.4s, 0" instruction
-		emit32(0x4F00041C, code, codePos);
+		if (flags & RANDOMX_FLAG_HARD_AES) {
+			// Disable the jump to RandomX v1 FE mix code by writing "movi v28.4s, 0" instruction
+			emit32(0x4F00041C, code, codePos);
+		}
+		else {
+			// Jump to RandomX v2 FE mix soft AES code by writing "b randomx_program_aarch64_v2_FE_mix_soft_aes" instruction
+			const uint32_t offset = (uint8_t*)randomx_program_aarch64_v2_FE_mix_soft_aes - (uint8_t*)randomx_program_aarch64_v2_FE_mix;
+			emit32(ARMV8A::B | (offset / 4), code, codePos);
+		}
 	}
 
 #ifdef __GNUC__
@@ -228,8 +235,15 @@ void JitCompilerA64::generateProgramLight(Program& program, ProgramConfiguration
 	if (flags & RANDOMX_FLAG_V2) {
 		codePos = ((uint8_t*)randomx_program_aarch64_v2_FE_mix) - ((uint8_t*)randomx_program_aarch64);
 
-		// Disable the jump to RandomX v1 FE mix code by writing "movi v28.4s, 0" instruction
-		emit32(0x4F00041C, code, codePos);
+		if (flags & RANDOMX_FLAG_HARD_AES) {
+			// Disable the jump to RandomX v1 FE mix code by writing "movi v28.4s, 0" instruction
+			emit32(0x4F00041C, code, codePos);
+		}
+		else {
+			// Jump to RandomX v2 FE mix soft AES code by writing "b randomx_program_aarch64_v2_FE_mix_soft_aes" instruction
+			const uint32_t offset = (uint8_t*)randomx_program_aarch64_v2_FE_mix_soft_aes - (uint8_t*)randomx_program_aarch64_v2_FE_mix;
+			emit32(ARMV8A::B | (offset / 4), code, codePos);
+		}
 	}
 
 	// Apply dataset offset