Reformatted code using IntelliJ
This commit is contained in:
@@ -21,7 +21,6 @@ public class CallTracker<A extends Annotation> extends GenericReflectClass<A> {
|
||||
|
||||
public CallTracker(CallInterceptor interceptor) {
|
||||
super();
|
||||
|
||||
atomicClass.set(this.persistentClass);
|
||||
}
|
||||
|
||||
@@ -31,8 +30,7 @@ public class CallTracker<A extends Annotation> extends GenericReflectClass<A> {
|
||||
new AgentBuilder.Default()
|
||||
.type(any()) // instrument all classes, you can restrict here
|
||||
.transform((builder, type, classLoader, module, protectionDomain) ->
|
||||
builder.visit(Advice.to(CallInterceptor.class).on(isMethod()))
|
||||
).installOn(inst);
|
||||
builder.visit(Advice.to(CallInterceptor.class).on(isMethod()))).installOn(inst);
|
||||
}
|
||||
|
||||
public abstract static class CallInterceptor {
|
||||
@@ -44,15 +42,12 @@ public class CallTracker<A extends Annotation> extends GenericReflectClass<A> {
|
||||
|
||||
@Advice.OnMethodEnter
|
||||
static void intercept(@Advice.Origin Method method) {
|
||||
|
||||
for (CallInterceptor interceptor : interceptors) {
|
||||
StackTraceElement[] stack = Thread.currentThread().getStackTrace();
|
||||
|
||||
if (stack.length <= 3)
|
||||
return;
|
||||
if (stack.length <= 3) return;
|
||||
|
||||
StackTraceElement caller = stack[3];
|
||||
|
||||
interceptor.onCall(method, caller);
|
||||
}
|
||||
|
||||
|
||||
@@ -38,45 +38,34 @@ public class ProtocolInfoProcessing extends AnnotationProcessor<ProtocolInfo> {
|
||||
@Override
|
||||
public void onCall(Method method, @Nullable StackTraceElement callerMethod) {
|
||||
ProtocolVersion.ProtocolSide side, callerSide;
|
||||
|
||||
Object o;
|
||||
|
||||
if ((o = methodGetByName(callerMethod.getMethodName())) != null)
|
||||
callerSide = ((Method) o).getAnnotation(ProtocolInfo.class).protocolSide();
|
||||
else if ((o = typeHasAnnotation(callerMethod.getClassName())) != null)
|
||||
callerSide = ((Class<?>) o).getAnnotation(ProtocolInfo.class).protocolSide();
|
||||
else
|
||||
return;
|
||||
else return;
|
||||
|
||||
|
||||
if (methodReferences.get().contains(method))
|
||||
side = method.getAnnotation(ProtocolInfo.class).protocolSide();
|
||||
|
||||
else if (typeReferences.get().contains(method.getDeclaringClass()))
|
||||
side = method.getDeclaringClass().getAnnotation(ProtocolInfo.class).protocolSide();
|
||||
else
|
||||
return;
|
||||
else return;
|
||||
|
||||
if (callerSide.equals(ProtocolVersion.ProtocolSide.CLIENT) &&
|
||||
!side.equals(callerSide))
|
||||
if (callerSide.equals(ProtocolVersion.ProtocolSide.CLIENT) && !side.equals(callerSide))
|
||||
throw new IncompatibleProtocolSideException(callerSide, side);
|
||||
|
||||
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private Method methodGetByName(String methodName) {
|
||||
for (Method method : this.annotatedMethods)
|
||||
if (method.getName().equals(methodName))
|
||||
return method;
|
||||
for (Method method : this.annotatedMethods) if (method.getName().equals(methodName)) return method;
|
||||
return null;
|
||||
}
|
||||
|
||||
private Class<?> typeHasAnnotation(String typeName) {
|
||||
for (Class<?> type : this.annotatedTypes)
|
||||
if (type.getName().equals(typeName))
|
||||
return type;
|
||||
for (Class<?> type : this.annotatedTypes) if (type.getName().equals(typeName)) return type;
|
||||
return null;
|
||||
}
|
||||
|
||||
|
||||
@@ -21,7 +21,6 @@ import org.openautonomousconnection.protocol.versions.v1_0_0.classic.objects.Cla
|
||||
import java.io.File;
|
||||
import java.io.FileNotFoundException;
|
||||
import java.io.IOException;
|
||||
import java.net.Proxy;
|
||||
import java.security.cert.CertificateException;
|
||||
|
||||
/**
|
||||
|
||||
@@ -135,10 +135,7 @@ public final class WebClient {
|
||||
// Start the receive thread
|
||||
this.receiveThread.start();
|
||||
}
|
||||
} /**
|
||||
* Thread for receiving data from the web server.
|
||||
*/
|
||||
private final Thread receiveThread = new Thread(this::receive);
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the NetworkClient used for the pipeline connection to the web server.
|
||||
@@ -147,7 +144,10 @@ public final class WebClient {
|
||||
*/
|
||||
public NetworkClient getClientPipelineConnection() {
|
||||
return clientToWebPipeline;
|
||||
}
|
||||
} /**
|
||||
* Thread for receiving data from the web server.
|
||||
*/
|
||||
private final Thread receiveThread = new Thread(this::receive);
|
||||
|
||||
/**
|
||||
* Gets the SSLSocket used for communication with the web server.
|
||||
|
||||
@@ -55,11 +55,7 @@ public final class ConnectedWebClient {
|
||||
* Indicates if the client version has been loaded.
|
||||
*/
|
||||
@Getter
|
||||
private boolean clientVersionLoaded = false; /**
|
||||
* Thread for receiving data from the client.
|
||||
*/
|
||||
private final Thread receiveThread = new Thread(this::receive);
|
||||
|
||||
private boolean clientVersionLoaded = false;
|
||||
/**
|
||||
* Constructs a ConnectedWebClient with the given connection handler.
|
||||
*
|
||||
@@ -67,7 +63,10 @@ public final class ConnectedWebClient {
|
||||
*/
|
||||
public ConnectedWebClient(ConnectionHandler pipelineConnection) {
|
||||
this.pipelineConnection = pipelineConnection;
|
||||
}
|
||||
} /**
|
||||
* Thread for receiving data from the client.
|
||||
*/
|
||||
private final Thread receiveThread = new Thread(this::receive);
|
||||
|
||||
/**
|
||||
* Sends an HTTP redirect response to the client.
|
||||
@@ -717,4 +716,5 @@ public final class ConnectedWebClient {
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user