ExtendedObjectInputStream.java

package uk.ac.imperial.lsds.seep.util;

import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectStreamClass;


public class ExtendedObjectInputStream extends ObjectInputStream {

	private RuntimeClassLoader rcl = null;
	
	public ExtendedObjectInputStream(InputStream in, RuntimeClassLoader rcl) throws IOException {
		super(in);
		this.rcl = rcl;
	}
	
	public ObjectStreamClass readClassDescriptor(){
		try {
			return super.readClassDescriptor();
		} 
		catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} 
		catch (ClassNotFoundException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		return null;
	}
	
	public Class<?> resolveClass(ObjectStreamClass osc){
		///\fixme{Check if the class is already loaded in the system before loading it}
		Class hack = null;
		hack = rcl.loadClass(osc.getName());
		return hack;
	}
	
	public Object resolveObject(Object mock){
		this.enableResolveObject(true);
		try {
			return super.resolveObject(this.readObject());
		} 
		catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (ClassNotFoundException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		return mock;
	}
}