CodeChickenCore/codechicken/lib/tool/StripClassLoader.java

package codechicken.lib.tool;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.net.URLClassLoader;

public class StripClassLoader extends URLClassLoader
{
    public StripClassLoader() {
        super(new URL[0], StripClassLoader.class.getClassLoader());
    }

    @Override
    public Class<?> loadClass(String name) throws ClassNotFoundException {
        if(!name.startsWith("codechicken.lib"))
            return super.loadClass(name);

        try {
            String resName = name.replace('.', '/')+".class";
            InputStream res = getResourceAsStream(resName);
            if(res == null)
                throw new ClassNotFoundException("Could not find resource: "+resName);
            byte[] bytes = readFully(res);
            bytes = transform(bytes);
            return defineClass(name, bytes, 0, bytes.length);

        } catch(IOException e) {
            throw new ClassNotFoundException(name, e);
        }
    }

    public static byte[] readFully(InputStream is) throws IOException {
        ByteArrayOutputStream buffer = new ByteArrayOutputStream();

        int read;
        byte[] data = new byte[16384];
        while ((read = is.read(data, 0, data.length)) > 0)
            buffer.write(data, 0, read);

        return buffer.toByteArray();
    }

    private byte[] transform(byte[] bytes) {

        bytes = MCStripTransformer.transform(bytes);
        return bytes;
    }
}