package com.caucho.hessian.io;
import java.io.IOException;
import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.HashMap;
import com.caucho.hessian.io.FieldDeserializer2Factory.NullFieldDeserializer;
public class JavaDeserializer extends AbstractMapDeserializer {
private Class<?> _type;
private HashMap<?,FieldDeserializer2> _fieldMap;
private Method _readResolve;
private Constructor<?> _constructor;
private Object []_constructorArgs;
public JavaDeserializer(Class<?> cl, FieldDeserializer2Factory fieldFactory)
{
_type = cl;
_fieldMap = getFieldMap(cl, fieldFactory);
_readResolve = getReadResolve(cl);
if (_readResolve != null) {
_readResolve.setAccessible(true);
}
_constructor = getConstructor(cl);
_constructorArgs = getConstructorArgs(_constructor);
}
protected Constructor<?> getConstructor(Class<?> cl)
{
Constructor<?> []constructors = cl.getDeclaredConstructors();
long bestCost = Long.MAX_VALUE;
Constructor<?> constructor = null;
for (int i = 0; i < constructors.length; i++) {
Class<?> []param = constructors[i].getParameterTypes();
long cost = 0;
for (int j = 0; j < param.length; j++) {
cost = 4 * cost;
if (Object.class.equals(param[j]))
cost += 1;
else if (String.class.equals(param[j]))
cost += 2;
else if (int.class.equals(param[j]))
cost += 3;
else if (long.class.equals(param[j]))
cost += 4;
else if (param[j].isPrimitive())
cost += 5;
else
cost += 6;
}
if (cost < 0 || cost > (1 << 48))
cost = 1 << 48;
cost += (long) param.length << 48;
if (cost < bestCost) {
constructor = constructors[i];
bestCost = cost;
}
}
if (constructor != null) {
constructor.setAccessible(true);
}
return constructor;
}
protected Object []getConstructorArgs(Constructor<?> constructor)
{
Object []constructorArgs = null;
if (constructor != null) {
Class<?> []params = constructor.getParameterTypes();
constructorArgs = new Object[params.length];
for (int i = 0; i < params.length; i++) {
constructorArgs[i] = getParamArg(params[i]);
}
}
return constructorArgs;
}
@Override
public Class<?> getType()
{
return _type;
}
@Override
public boolean isReadResolve()
{
return _readResolve != null;
}
public Object readMap(AbstractHessianInput in)
throws IOException
{
try {
Object obj = instantiate();
return readMap(in, obj);
} catch (IOException e) {
throw e;
} catch (RuntimeException e) {
throw e;
} catch (Exception e) {
throw new IOExceptionWrapper(_type.getName() + ":" + e.getMessage(), e);
}
}
@Override
public Object []createFields(int len)
{
return new FieldDeserializer2[len];
}
@Override
public Object createField(String name)
{
Object reader = _fieldMap.get(name);
if (reader == null)
reader = NullFieldDeserializer.DESER;
return reader;
}
@Override
public Object readObject(AbstractHessianInput in,
Object []fields)
throws IOException
{
try {
Object obj = instantiate();
return readObject(in, obj, (FieldDeserializer2 []) fields);
} catch (IOException e) {
throw e;
} catch (RuntimeException e) {
throw e;
} catch (Exception e) {
throw new IOExceptionWrapper(_type.getName() + ":" + e.getMessage(), e);
}
}
@Override
public Object readObject(AbstractHessianInput in,
String []fieldNames)
throws IOException
{
try {
Object obj = instantiate();
return readObject(in, obj, fieldNames);
} catch (IOException e) {
throw e;
} catch (RuntimeException e) {
throw e;
} catch (Exception e) {
throw new IOExceptionWrapper(_type.getName() + ":" + e.getMessage(), e);
}
}
protected Method getReadResolve(Class<?> cl)
{
for (; cl != null; cl = cl.getSuperclass()) {
Method []methods = cl.getDeclaredMethods();
for (int i = 0; i < methods.length; i++) {
Method method = methods[i];
if (method.getName().equals("readResolve")
&& method.getParameterTypes().length == 0)
return method;
}
}
return null;
}
public Object readMap(AbstractHessianInput in, Object obj)
throws IOException
{
try {
int ref = in.addRef(obj);
while (! in.isEnd()) {
Object key = in.readObject();
FieldDeserializer2 deser = _fieldMap.get(key);
if (deser != null)
deser.deserialize(in, obj);
else
in.readObject();
}
in.readMapEnd();
Object resolve = resolve(in, obj);
if (obj != resolve)
in.setRef(ref, resolve);
return resolve;
} catch (IOException e) {
throw e;
} catch (Exception e) {
throw new IOExceptionWrapper(e);
}
}
private Object readObject(AbstractHessianInput in,
Object obj,
FieldDeserializer2 []fields)
throws IOException
{
try {
int ref = in.addRef(obj);
for (FieldDeserializer2 reader : fields) {
reader.deserialize(in, obj);
}
Object resolve = resolve(in, obj);
if (obj != resolve)
in.setRef(ref, resolve);
return resolve;
} catch (IOException e) {
throw e;
} catch (Exception e) {
throw new IOExceptionWrapper(obj.getClass().getName() + ":" + e, e);
}
}
public Object readObject(AbstractHessianInput in,
Object obj,
String []fieldNames)
throws IOException
{
try {
int ref = in.addRef(obj);
for (String fieldName : fieldNames) {
FieldDeserializer2 reader = _fieldMap.get(fieldName);
if (reader != null)
reader.deserialize(in, obj);
else
in.readObject();
}
Object resolve = resolve(in, obj);
if (obj != resolve)
in.setRef(ref, resolve);
return resolve;
} catch (IOException e) {
throw e;
} catch (Exception e) {
throw new IOExceptionWrapper(obj.getClass().getName() + ":" + e, e);
}
}
protected Object resolve(AbstractHessianInput in, Object obj)
throws Exception
{
try {
if (_readResolve != null)
return _readResolve.invoke(obj, new Object[0]);
} catch (InvocationTargetException e) {
if (e.getCause() instanceof Exception)
throw (Exception) e.getCause();
else
throw e;
}
return obj;
}
protected Object instantiate()
throws Exception
{
try {
if (_constructor != null)
return _constructor.newInstance(_constructorArgs);
else
return _type.newInstance();
} catch (Exception e) {
throw new HessianProtocolException("'" + _type.getName() + "' could not be instantiated", e);
}
}
protected HashMap<String,FieldDeserializer2>
getFieldMap(Class<?> cl, FieldDeserializer2Factory fieldFactory)
{
HashMap<String,FieldDeserializer2> fieldMap
= new HashMap<String,FieldDeserializer2>();
for (; cl != null; cl = cl.getSuperclass()) {
Field []fields = cl.getDeclaredFields();
for (int i = 0; i < fields.length; i++) {
Field field = fields[i];
if (Modifier.isTransient(field.getModifiers())
|| Modifier.isStatic(field.getModifiers()))
continue;
else if (fieldMap.get(field.getName()) != null)
continue;
FieldDeserializer2 deser = fieldFactory.create(field);
fieldMap.put(field.getName(), deser);
}
}
return fieldMap;
}
protected static Object getParamArg(Class<?> cl)
{
if (! cl.isPrimitive())
return null;
else if (boolean.class.equals(cl))
return Boolean.FALSE;
else if (byte.class.equals(cl))
return new Byte((byte) 0);
else if (short.class.equals(cl))
return new Short((short) 0);
else if (char.class.equals(cl))
return new Character((char) 0);
else if (int.class.equals(cl))
return Integer.valueOf(0);
else if (long.class.equals(cl))
return Long.valueOf(0);
else if (float.class.equals(cl))
return Float.valueOf(0);
else if (double.class.equals(cl))
return Double.valueOf(0);
else
throw new UnsupportedOperationException();
}
static void logDeserializeError(Field field, Object obj, Object value,
Throwable e)
throws IOException
{
String fieldName = (field.getDeclaringClass().getName()
+ "." + field.getName());
if (e instanceof HessianFieldException)
throw (HessianFieldException) e;
else if (e instanceof IOException)
throw new HessianFieldException(fieldName + ": " + e.getMessage(), e);
if (value != null)
throw new HessianFieldException(fieldName + ": " + value.getClass().getName() + " (" + value + ")"
+ " cannot be assigned to '" + field.getType().getName() + "'", e);
else
throw new HessianFieldException(fieldName + ": " + field.getType().getName() + " cannot be assigned from null", e);
}
}