From 2b2249e873c4adfd2dd6e8f1f2489ccd9f6aa021 Mon Sep 17 00:00:00 2001 From: gegy1000 Date: Sat, 19 May 2018 17:02:46 +0200 Subject: Initial port to ASM --- .../java/cuchaz/enigma/analysis/ParsedJar.java | 72 ++++++++++++++++++++++ 1 file changed, 72 insertions(+) create mode 100644 src/main/java/cuchaz/enigma/analysis/ParsedJar.java (limited to 'src/main/java/cuchaz/enigma/analysis/ParsedJar.java') diff --git a/src/main/java/cuchaz/enigma/analysis/ParsedJar.java b/src/main/java/cuchaz/enigma/analysis/ParsedJar.java new file mode 100644 index 0000000..78ef722 --- /dev/null +++ b/src/main/java/cuchaz/enigma/analysis/ParsedJar.java @@ -0,0 +1,72 @@ +/******************************************************************************* + * Copyright (c) 2015 Jeff Martin. + * All rights reserved. This program and the accompanying materials + * are made available under the terms of the GNU Lesser General Public + * License v3.0 which accompanies this distribution, and is available at + * http://www.gnu.org/licenses/lgpl.html + *

+ * Contributors: + * Jeff Martin - initial API and implementation + ******************************************************************************/ + +package cuchaz.enigma.analysis; + +import cuchaz.enigma.mapping.ClassEntry; +import org.objectweb.asm.ClassReader; +import org.objectweb.asm.tree.ClassNode; + +import java.io.IOException; +import java.io.InputStream; +import java.util.*; +import java.util.function.Consumer; +import java.util.jar.JarEntry; +import java.util.jar.JarFile; + +public class ParsedJar { + private final Map nodes = new LinkedHashMap<>(); + + public ParsedJar(JarFile jar) throws IOException { + try { + // get the jar entries that correspond to classes + Enumeration entries = jar.entries(); + while (entries.hasMoreElements()) { + JarEntry entry = entries.nextElement(); + // is this a class file? + if (entry.getName().endsWith(".class")) { + try (InputStream input = jar.getInputStream(entry)) { + // read the ClassNode from the jar + ClassReader reader = new ClassReader(input); + ClassNode node = new ClassNode(); + reader.accept(node, 0); + String path = entry.getName().substring(0, entry.getName().length() - ".class".length()); + nodes.put(path, node); + } + } + } + } finally { + jar.close(); + } + } + + public void visit(Consumer visitor) { + for (ClassNode node : nodes.values()) { + visitor.accept(node); + } + } + + public int getClassCount() { + return nodes.size(); + } + + public List getClassEntries() { + List entries = new ArrayList<>(nodes.size()); + for (ClassNode node : nodes.values()) { + entries.add(new ClassEntry(node.name)); + } + return entries; + } + + public ClassNode getClassNode(String name) { + return nodes.get(name); + } +} -- cgit v1.2.3 From 406b9a89318473571d27de60b8aa1b51f84af245 Mon Sep 17 00:00:00 2001 From: gegy1000 Date: Sat, 19 May 2018 17:06:26 +0200 Subject: Package updates --- src/main/java/cuchaz/enigma/analysis/ParsedJar.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/main/java/cuchaz/enigma/analysis/ParsedJar.java') diff --git a/src/main/java/cuchaz/enigma/analysis/ParsedJar.java b/src/main/java/cuchaz/enigma/analysis/ParsedJar.java index 78ef722..fae58f8 100644 --- a/src/main/java/cuchaz/enigma/analysis/ParsedJar.java +++ b/src/main/java/cuchaz/enigma/analysis/ParsedJar.java @@ -11,7 +11,7 @@ package cuchaz.enigma.analysis; -import cuchaz.enigma.mapping.ClassEntry; +import cuchaz.enigma.mapping.entry.ClassEntry; import org.objectweb.asm.ClassReader; import org.objectweb.asm.tree.ClassNode; -- cgit v1.2.3 From 0809cffbade803caa34ad99f5409997f50d57b8b Mon Sep 17 00:00:00 2001 From: Thiakil Date: Tue, 10 Jul 2018 21:18:54 +0800 Subject: buffer the ParsedJar class entry inputstream, allow use with a jarinputstream --- .../java/cuchaz/enigma/analysis/ParsedJar.java | 25 +++++++++++++++++++++- 1 file changed, 24 insertions(+), 1 deletion(-) (limited to 'src/main/java/cuchaz/enigma/analysis/ParsedJar.java') diff --git a/src/main/java/cuchaz/enigma/analysis/ParsedJar.java b/src/main/java/cuchaz/enigma/analysis/ParsedJar.java index fae58f8..55f2141 100644 --- a/src/main/java/cuchaz/enigma/analysis/ParsedJar.java +++ b/src/main/java/cuchaz/enigma/analysis/ParsedJar.java @@ -15,12 +15,14 @@ import cuchaz.enigma.mapping.entry.ClassEntry; import org.objectweb.asm.ClassReader; import org.objectweb.asm.tree.ClassNode; +import java.io.BufferedInputStream; import java.io.IOException; import java.io.InputStream; import java.util.*; import java.util.function.Consumer; import java.util.jar.JarEntry; import java.util.jar.JarFile; +import java.util.jar.JarInputStream; public class ParsedJar { private final Map nodes = new LinkedHashMap<>(); @@ -33,7 +35,7 @@ public class ParsedJar { JarEntry entry = entries.nextElement(); // is this a class file? if (entry.getName().endsWith(".class")) { - try (InputStream input = jar.getInputStream(entry)) { + try (InputStream input = new BufferedInputStream(jar.getInputStream(entry))) { // read the ClassNode from the jar ClassReader reader = new ClassReader(input); ClassNode node = new ClassNode(); @@ -48,6 +50,27 @@ public class ParsedJar { } } + public ParsedJar(JarInputStream jar) throws IOException { + try { + // get the jar entries that correspond to classes + JarEntry entry; + while ((entry = jar.getNextJarEntry()) != null) { + // is this a class file? + if (entry.getName().endsWith(".class")) { + // read the ClassNode from the jar + ClassReader reader = new ClassReader(jar); + ClassNode node = new ClassNode(); + reader.accept(node, 0); + String path = entry.getName().substring(0, entry.getName().length() - ".class".length()); + nodes.put(path, node); + jar.closeEntry(); + } + } + } finally { + jar.close(); + } + } + public void visit(Consumer visitor) { for (ClassNode node : nodes.values()) { visitor.accept(node); -- cgit v1.2.3