* Fixed mask application

gmu-swe · Nov 29, 2023 · 145b3bf · 145b3bf
1 parent 91696df
commit 145b3bf
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/Phosphor/src/main/java/edu/columbia/cs/psl/phosphor/mask/ReflectionHidingMV.java b/Phosphor/src/main/java/edu/columbia/cs/psl/phosphor/mask/ReflectionHidingMV.java
@@ -74,20 +74,20 @@ public void visitMethodInsn(int opcode, String owner, String name, String desc,
                 maskUnsafe(opcode, owner, name, desc, isInterface, descWithoutStackFrame);
                 return;
             default:
+                MaskInfo mask = MaskRegistry.getMask(owner, name, desc);
+                if (mask != null) {
+                    if (!name.equals("defineClass") || patchAnonymousClasses) {
+                        mask.accept(mv);
+                        return;
+                    }
+                }
                 super.visitMethodInsn(opcode, owner, name, desc, isInterface);
                 fixReturn(owner, name, name, descWithoutStackFrame);
         }
     }
 
     private void maskUnsafe(
             int opcode, String owner, String name, String desc, boolean isInterface, String descWithoutStackFrame) {
-        MaskInfo mask = MaskRegistry.getMask(owner, name, desc);
-        if (mask != null) {
-            if (!name.equals("defineClass") || patchAnonymousClasses) {
-                mask.accept(mv);
-                return;
-            }
-        }
         if (patchAnonymousClasses
                 && name.equals("defineAnonymousClass")
                 && descWithoutStackFrame.equals("(Ljava/lang/Class;[B[Ljava/lang/Object;)Ljava/lang/Class;")) {