org.easymock.asm.optimizer.JarOptimizer Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of easymock Show documentation
Show all versions of easymock Show documentation
EasyMock provides an easy way to create Mock Objects for interfaces and classes generating them on the fly
/***
* 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.easymock.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.easymock.asm.ClassReader;
import org.easymock.asm.ClassVisitor;
import org.easymock.asm.FieldVisitor;
import org.easymock.asm.MethodVisitor;
import org.easymock.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");
}
}
}
}