tyler.durden
/
snappy-java
mirror of https://github.com/xerial/snappy-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add bytecode generator

This commit is contained in:
Taro L. Saito 2011-06-24 13:42:14 +09:00
parent d19a00ddb6
commit b531cb36dc
1 changed files with 91 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

135
src/test/java/org/xerial/snappy/SnappyLoaderTest.java
View File

@ -24,57 +24,104 @@
//--------------------------------------
package org.xerial.snappy;
import static org.junit.Assert.*;
import java.io.BufferedInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.URL;
import java.security.ProtectionDomain;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtField;
import javassist.CtNewMethod;
import org.codehaus.plexus.classworlds.ClassWorld;
import org.codehaus.plexus.classworlds.realm.ClassRealm;
import org.junit.Test;
import org.xerial.util.FileResource;
import org.xerial.util.log.Logger;
public class SnappyLoaderTest
{
private static Logger _logger = Logger.getLogger(SnappyLoaderTest.class);
// @Ignore
// @Test
// public void loadFromSytemClassLoader() throws Exception {
//
// ClassLoader parent = this.getClass().getClassLoader().getParent();
// ClassWorld cw = new ClassWorld();
// ClassRealm L1 = cw.newRealm("l1", parent);
// ClassRealm L2 = cw.newRealm("l2", parent);
//
// File nativeLib = SnappyLoader.findNativeLibrary();
// assertNotNull(nativeLib);
//
// ClassPool pool = ClassPool.getDefault();
// CtClass cl = pool.makeClass("org.xerial.snappy.SnappyNativeLoader");
// cl.addField(CtField.make("static boolean isLoaded = false;", cl));
// String m1 = FileResource.loadIntoString(SnappyLoaderTest.class, "load.code");
// String m2 = FileResource.loadIntoString(SnappyLoaderTest.class, "loadLibrary.code");
// cl.addMethod(CtNewMethod.make(m1, cl));
// cl.addMethod(CtNewMethod.make(m2, cl));
//
// ProtectionDomain systemPD = System.class.getProtectionDomain();
// byte[] bytecode = cl.toBytecode();
// FileOutputStream f = new FileOutputStream("src/main/resources/org/xerial/snappy/SnappyNativeLoader.bytecode");
// f.write(bytecode);
// f.close();
//
// //Class< ? > loaderClass = cl.toClass(parent, System.class.getProtectionDomain());
// //_logger.info(cl.getName());
// //Class< ? > loaderClass = cl.toClass();
//
// Class< ? > classLoader = Class.forName("java.lang.ClassLoader");
// java.lang.reflect.Method defineClass = classLoader.getDeclaredMethod("defineClass", new Class[] { String.class,
// byte[].class, int.class, int.class, ProtectionDomain.class });
//
// defineClass.setAccessible(true);
// defineClass.invoke(parent, cl.getName(), bytecode, 0, bytecode.length, System.class.getProtectionDomain());
//
// Class< ? > forName = parent.loadClass("org.xerial.snappy.SnappyNativeLoader");
// _logger.info(forName.toString());
//
// //Class< ? > snappyClass = L1.loadClass("org.xerial.snappy.Snappy"); // not found
// //_logger.info(snappyClass.getName());
//
// }
public static BufferedInputStream openByteStream(Class< ? > referenceClass, String resourceFileName)
throws IOException {
URL url = FileResource.find(referenceClass, resourceFileName);
if (url != null) {
return new BufferedInputStream(url.openStream());
}
else
return null;
}
public static <T> String loadIntoString(Class<T> referenceClass, String path) throws IOException {
BufferedInputStream in = openByteStream(referenceClass, path);
if (in == null)
throw new FileNotFoundException(
String.format("reference class:%s, path:%s", referenceClass.getName(), path));
ByteArrayOutputStream buf = new ByteArrayOutputStream();
try {
byte[] tmp = new byte[4028];
for (int readBytes = 0; (readBytes = in.read(tmp)) != -1;) {
buf.write(tmp, 0, readBytes);
}
buf.flush();
return buf.toString();
}
finally {
in.close();
}
}
@Test
public void loadFromSytemClassLoader() throws Exception {
ClassLoader parent = this.getClass().getClassLoader().getParent();
ClassWorld cw = new ClassWorld();
ClassRealm L1 = cw.newRealm("l1", parent);
ClassRealm L2 = cw.newRealm("l2", parent);
File nativeLib = SnappyLoader.findNativeLibrary();
assertNotNull(nativeLib);
ClassPool pool = ClassPool.getDefault();
CtClass cl = pool.makeClass("org.xerial.snappy.SnappyNativeLoader");
cl.addField(CtField.make("static boolean isLoaded = false;", cl));
String m1 = loadIntoString(SnappyLoaderTest.class, "load.code");
String m2 = loadIntoString(SnappyLoaderTest.class, "loadLibrary.code");
cl.addMethod(CtNewMethod.make(m1, cl));
cl.addMethod(CtNewMethod.make(m2, cl));
ProtectionDomain systemPD = System.class.getProtectionDomain();
byte[] bytecode = cl.toBytecode();
FileOutputStream f = new FileOutputStream("target/SnappyNativeLoader.bytecode");
f.write(bytecode);
f.close();
//Class< ? > loaderClass = cl.toClass(parent, System.class.getProtectionDomain());
//_logger.info(cl.getName());
//Class< ? > loaderClass = cl.toClass();
Class< ? > classLoader = Class.forName("java.lang.ClassLoader");
java.lang.reflect.Method defineClass = classLoader.getDeclaredMethod("defineClass", new Class[] { String.class,
byte[].class, int.class, int.class, ProtectionDomain.class });
defineClass.setAccessible(true);
defineClass.invoke(parent, cl.getName(), bytecode, 0, bytecode.length, System.class.getProtectionDomain());
Class< ? > forName = parent.loadClass("org.xerial.snappy.SnappyNativeLoader");
//Class< ? > snappyClass = L1.loadClass("org.xerial.snappy.Snappy"); // not found
//_logger.info(snappyClass.getName());
}
@Test
public void load() throws Exception {