org.testifyproject.asm.optimizer.JarOptimizer Maven / Gradle / Ivy
The newest version!
/***
* ASM: a very small and fast Java bytecode manipulation framework
* Copyright (c) 2000-2011 INRIA, France Telecom
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the copyright holders nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.testifyproject.asm.optimizer;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.LineNumberReader;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.zip.GZIPInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;
import org.testifyproject.asm.ClassReader;
import org.testifyproject.asm.ClassVisitor;
import org.testifyproject.asm.FieldVisitor;
import org.testifyproject.asm.MethodVisitor;
import org.testifyproject.asm.Opcodes;
/**
* A Jar file optimizer.
*
* @author Eric Bruneton
*/
public class JarOptimizer {
static final Set API = new HashSet();
static final Map HIERARCHY = new HashMap();
static boolean nodebug = false;
public static void main(final String[] args) throws IOException {
File f = new File(args[0]);
InputStream is = new GZIPInputStream(new FileInputStream(f));
BufferedReader lnr = new LineNumberReader(new InputStreamReader(is));
while (true) {
String line = lnr.readLine();
if (line != null) {
if (line.startsWith("class")) {
String c = line.substring(6, line.lastIndexOf(' '));
String sc = line.substring(line.lastIndexOf(' ') + 1);
HIERARCHY.put(c, sc);
} else {
API.add(line);
}
} else {
break;
}
}
int argIndex = 1;
if (args[argIndex].equals("-nodebug")) {
nodebug = true;
argIndex++;
}
optimize(new File(args[argIndex]));
}
static void optimize(final File f) throws IOException {
if (nodebug && f.getName().contains("debug")) {
return;
}
if (f.isDirectory()) {
File[] files = f.listFiles();
for (int i = 0; i < files.length; ++i) {
optimize(files[i]);
}
} else if (f.getName().endsWith(".jar")) {
File g = new File(f.getParentFile(), f.getName() + ".new");
ZipFile zf = new ZipFile(f);
ZipOutputStream out = new ZipOutputStream(new FileOutputStream(g));
Enumeration e = zf.entries();
byte[] buf = new byte[10000];
while (e.hasMoreElements()) {
ZipEntry ze = e.nextElement();
if (ze.isDirectory()) {
out.putNextEntry(ze);
continue;
}
out.putNextEntry(ze);
if (ze.getName().endsWith(".class")) {
ClassReader cr = new ClassReader(zf.getInputStream(ze));
// cr.accept(new ClassDump(), 0);
cr.accept(new ClassVerifier(), 0);
}
InputStream is = zf.getInputStream(ze);
int n;
do {
n = is.read(buf, 0, buf.length);
if (n != -1) {
out.write(buf, 0, n);
}
} while (n != -1);
out.closeEntry();
}
out.close();
zf.close();
if (!f.delete()) {
throw new IOException("Cannot delete file " + f);
}
if (!g.renameTo(f)) {
throw new IOException("Cannot rename file " + g);
}
}
}
static class ClassDump extends ClassVisitor {
String owner;
public ClassDump() {
super(Opcodes.ASM5);
}
@Override
public void visit(final int version, final int access,
final String name, final String signature,
final String superName, final String[] interfaces) {
owner = name;
if (owner.startsWith("java/")) {
System.out.println("class " + name + ' ' + superName);
}
}
@Override
public FieldVisitor visitField(final int access, final String name,
final String desc, final String signature, final Object value) {
if (owner.startsWith("java/")) {
System.out.println(owner + ' ' + name);
}
return null;
}
@Override
public MethodVisitor visitMethod(final int access, final String name,
final String desc, final String signature,
final String[] exceptions) {
if (owner.startsWith("java/")) {
System.out.println(owner + ' ' + name + desc);
}
return null;
}
}
static class ClassVerifier extends ClassVisitor {
String owner;
String method;
public ClassVerifier() {
super(Opcodes.ASM5);
}
@Override
public void visit(final int version, final int access,
final String name, final String signature,
final String superName, final String[] interfaces) {
owner = name;
}
@Override
public MethodVisitor visitMethod(final int access, final String name,
final String desc, final String signature,
final String[] exceptions) {
method = name + desc;
return new MethodVisitor(Opcodes.ASM5) {
@Override
public void visitFieldInsn(final int opcode,
final String owner, final String name, final String desc) {
check(owner, name);
}
@Override
public void visitMethodInsn(final int opcode,
final String owner, final String name,
final String desc, final boolean itf) {
check(owner, name + desc);
}
};
}
void check(String owner, String member) {
if (owner.startsWith("java/")) {
String o = owner;
while (o != null) {
if (API.contains(o + ' ' + member)) {
return;
}
o = HIERARCHY.get(o);
}
System.out.println("WARNING: " + owner + ' ' + member
+ " called in " + this.owner + ' ' + method
+ " is not defined in JDK 1.3 API");
}
}
}
}