Check protected method/field target in visibility index (#157)

1 files changed, 19 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/MethodNodeWithAction.java b/src/main/java/cuchaz/enigma/analysis/MethodNodeWithAction.java
new file mode 100644
index 0000000..8117103
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/analysis/MethodNodeWithAction.java
@@ -0,0 +1,19 @@
+package cuchaz.enigma.analysis;
+
+import org.objectweb.asm.tree.MethodNode;
+
+import java.util.function.Consumer;
+
+public class MethodNodeWithAction extends MethodNode {
+    private final Consumer<MethodNode> action;
+
+    public MethodNodeWithAction(int api, int access, String name, String descriptor, String signature, String[] exceptions, Consumer<MethodNode> action) {
+        super(api, access, name, descriptor, signature, exceptions);
+        this.action = action;
+    }
+
+    @Override
+    public void visitEnd() {
+        action.accept(this);
+    }
+}