/**
* JBoss, Home of Professional Open Source
* Copyright 2012, Red Hat Middleware LLC, and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jboss.arquillian.warp.impl.shared;
import java.io.ByteArrayInputStream;
import java.io.Externalizable;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectInputStream;
import java.io.ObjectOutput;
import java.io.ObjectStreamClass;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import org.jboss.arquillian.warp.Inspection;
import org.jboss.arquillian.warp.impl.client.transformation.MigratedInspection;
import org.jboss.arquillian.warp.impl.client.transformation.TransformedInspection;
public class RequestPayload implements Externalizable {
private static final long serialVersionUID = -5537112559937896153L;
public static final long FAILURE_SERIAL_ID = -1L;
private List<Inspection> inspections;
private long serialId;
private static final DynamicClassLoader cl = new DynamicClassLoader(Thread.currentThread().getContextClassLoader());
private static final ConcurrentHashMap<String, Class<?>> loadedClasses = new ConcurrentHashMap<String, Class<?>>();
public RequestPayload() {
}
public RequestPayload(Inspection... inspections) {
this(Arrays.asList(inspections));
}
public RequestPayload(List<Inspection> inspections) {
this.inspections = inspections;
this.serialId = UUID.randomUUID().getMostSignificantBits();
}
public List<Inspection> getInspections() {
return inspections;
}
public long getSerialId() {
return serialId;
}
@Override
public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
serialId = in.readLong();
int size = in.read();
inspections = new ArrayList<Inspection>(size);
for (int i = 0; i < size; i++) {
boolean anonymous = in.readBoolean();
if (anonymous) {
String className = (String) in.readObject();
byte[] classFile = (byte[]) in.readObject();
byte[] obj = (byte[]) in.readObject();
final Inspection inspection = (Inspection) loadObject(className, classFile, obj);
inspections.add(inspection);
} else {
inspections.add((Inspection) in.readObject());
}
}
}
@SuppressWarnings("unchecked")
private <T> T loadObject(String className, byte[] classFile, final byte[] obj) throws IOException, ClassNotFoundException {
final Class<?> clazz = loadClass(className, classFile);
ObjectInputStream input = new ObjectInputStream(new ByteArrayInputStream(obj)) {
protected Class<?> resolveClass(ObjectStreamClass desc) throws IOException, ClassNotFoundException {
if (desc.getName().equals(clazz.getName())) {
return clazz;
}
return super.resolveClass(desc);
}
};
return (T) input.readObject();
}
private Class<?> loadClass(String className, byte[] classFile) {
if (loadedClasses.containsKey(className)) {
return loadedClasses.get(className);
} else {
Class<?> clazz = cl.load(classFile);
loadedClasses.put(className, clazz);
return clazz;
}
}
@Override
public void writeExternal(ObjectOutput out) throws IOException {
out.writeLong(serialId);
out.write(inspections.size());
for (Inspection inspection : inspections) {
if (shouldBeTransformed(inspection)) {
try {
out.writeBoolean(true); // flag 'anonymous'
TransformedInspection transformed = new TransformedInspection(inspection);
MigratedInspection migrated = new MigratedInspection(transformed);
out.writeObject(transformed.getOriginalClass().getName());
out.writeObject(migrated.toBytecode());
out.writeObject(migrated.toSerializedForm());
} catch (Exception e) {
throw new RuntimeException("Could not transform and replicate class " + inspections.getClass() + ":\n" + e.getMessage(), e);
}
} else {
out.writeBoolean(false); // flag 'not anonymous'
out.writeObject(inspection);
}
}
}
private boolean shouldBeTransformed(Inspection inspection) {
// return inspection.getClass().isAnonymousClass() || (inspection.getClass().isMemberClass() && !Modifier.isStatic(inspection.getClass().getModifiers()));
return inspection.getClass().isAnonymousClass() || (inspection.getClass().isMemberClass());
}
public static class DynamicClassLoader extends ClassLoader {
public DynamicClassLoader(ClassLoader parent) {
super(parent);
}
public Class<?> load(byte[] classFile) {
return defineClass(null, classFile, 0, classFile.length);
}
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + (int) (serialId ^ (serialId >>> 32));
return result;
}
@Override
public boolean equals(Object obj) {
if (this == obj)
return true;
if (obj == null)
return false;
if (getClass() != obj.getClass())
return false;
RequestPayload other = (RequestPayload) obj;
if (serialId != other.serialId)
return false;
return true;
}
}