From 579288d663f7859c3bc4834f1509a8069a509a7c Mon Sep 17 00:00:00 2001
From: Volker Berlin <volker.berlin@googlemail.com>
Date: Sun, 2 Oct 2022 20:25:36 +0200
Subject: [PATCH] more Java 11 Unsafe replacements

---
 .../jwebassembly/module/UnsafeManager.java    | 30 +++++++++++++++++--
 .../nativecode/ReplacementForClass.java       |  8 +++++
 2 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/src/de/inetsoftware/jwebassembly/module/UnsafeManager.java b/src/de/inetsoftware/jwebassembly/module/UnsafeManager.java
index 8f93647..4a59ee1 100644
--- a/src/de/inetsoftware/jwebassembly/module/UnsafeManager.java
+++ b/src/de/inetsoftware/jwebassembly/module/UnsafeManager.java
@@ -163,8 +163,15 @@ class UnsafeManager {
             case "jdk/internal/misc/Unsafe.isBigEndian()Z":
                 patch_isBigEndian( instructions, idx, callInst );
                 break;
+            case "jdk/internal/misc/Unsafe.shouldBeInitialized(Ljava/lang/Class;)Z":
+                replaceWithConstNumber( instructions, idx, callInst, 2, 0 );
+                break;
             case "jdk/internal/misc/Unsafe.storeFence()V":
-                remove( instructions, idx, callInst );
+                remove( instructions, idx, callInst, 1 );
+                break;
+            case "jdk/internal/misc/Unsafe.ensureClassInitialized(Ljava/lang/Class;)V":
+            case "jdk/internal/misc/Unsafe.unpark(Ljava/lang/Object;)V":
+                remove( instructions, idx, callInst, 2 );
                 break;
             default:
                 throw new WasmException( "Unsupported Unsafe method: " + name.signatureName, -1 );
@@ -495,9 +502,28 @@ class UnsafeManager {
      * @param callInst
      *            the method call to Unsafe
      */
-    private void remove( List<WasmInstruction> instructions, int idx, final WasmCallInstruction callInst ) {
+    private void replaceWithConstNumber( List<WasmInstruction> instructions, int idx, final WasmCallInstruction callInst, int paramCount, int number ) {
         int from = StackInspector.findInstructionThatPushValue( instructions.subList( 0, idx ), 1, callInst.getCodePosition() ).idx;
 
+        nop( instructions, from, idx );
+
+        // on x86 use little endian
+        instructions.set( idx, new WasmConstNumberInstruction( number, callInst.getCodePosition(), callInst.getLineNumber() ) );
+    }
+
+    /**
+     * Patch an unsafe function that access a field
+     * 
+     * @param instructions
+     *            the instruction list
+     * @param idx
+     *            the index in the instructions
+     * @param callInst
+     *            the method call to Unsafe
+     */
+    private void remove( List<WasmInstruction> instructions, int idx, final WasmCallInstruction callInst, int paramCount ) {
+        int from = StackInspector.findInstructionThatPushValue( instructions.subList( 0, idx ), paramCount, callInst.getCodePosition() ).idx;
+
         nop( instructions, from, idx + 1 );
     }
 
diff --git a/src/de/inetsoftware/jwebassembly/module/nativecode/ReplacementForClass.java b/src/de/inetsoftware/jwebassembly/module/nativecode/ReplacementForClass.java
index 7dff105..6df3c9e 100644
--- a/src/de/inetsoftware/jwebassembly/module/nativecode/ReplacementForClass.java
+++ b/src/de/inetsoftware/jwebassembly/module/nativecode/ReplacementForClass.java
@@ -335,6 +335,14 @@ class ReplacementForClass<T> {
         return null;
     }
 
+    /**
+     * Get the declaring class
+     * @return the declaring class for this class
+     */
+    private Class<?> getDeclaringClass0() {
+        return null;
+    }
+
     /**
      * Replacement of the Java method getSimpleName()
      *