Add utility to map variable insns to their respective accessed/writte…

…n variable types
Col-E · Jan 13, 2025 · c7c6026 · c7c6026
1 parent 2431fc3
commit c7c6026
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/recaf-core/src/main/java/software/coley/recaf/util/AsmInsnUtil.java b/recaf-core/src/main/java/software/coley/recaf/util/AsmInsnUtil.java
@@ -7,6 +7,7 @@
 import org.objectweb.asm.tree.InsnNode;
 import org.objectweb.asm.tree.IntInsnNode;
 import org.objectweb.asm.tree.LdcInsnNode;
+import org.objectweb.asm.tree.VarInsnNode;
 
 /**
  * ASM instruction utilities.
@@ -31,6 +32,23 @@ public static AbstractInsnNode getDefaultValue(@Nonnull Type type) {
 		};
 	}
 
+	/**
+	 * @param varInsn
+	 * 		Variable instruction.
+	 *
+	 * @return Type that encompasses the variable being accessed/written to,
+	 */
+	@Nonnull
+	public static Type getTypeForVarInsn(@Nonnull VarInsnNode varInsn) {
+		return switch (varInsn.getOpcode()) {
+			case Opcodes.ISTORE, Opcodes.ILOAD -> Type.INT_TYPE;
+			case Opcodes.LSTORE, Opcodes.LLOAD -> Type.LONG_TYPE;
+			case Opcodes.FSTORE, Opcodes.FLOAD -> Type.FLOAT_TYPE;
+			case Opcodes.DSTORE, Opcodes.DLOAD -> Type.DOUBLE_TYPE;
+			default -> Types.OBJECT_TYPE;
+		};
+	}
+
 	/**
 	 * Create an instruction to hold a given {@code int} value.
 	 *