3
0
Mirror von https://github.com/ViaVersion/ViaVersion.git synchronisiert 2024-11-03 14:50:30 +01:00

Make several tweaks to Protocol to improve inlining and performance on hot paths

Dieser Commit ist enthalten in:
Andrew Steinborn 2020-08-17 19:37:55 -04:00
Ursprung be6e086aac
Commit fc7a7396ef
2 geänderte Dateien mit 67 neuen und 61 gelöschten Zeilen

Datei anzeigen

@ -411,34 +411,38 @@ public abstract class Protocol<C1 extends ClientboundPacketType, C2 extends Clie
int oldId = packetWrapper.getId(); int oldId = packetWrapper.getId();
int newId = direction == Direction.OUTGOING ? protocolPacket.getNewID() : protocolPacket.getOldID(); int newId = direction == Direction.OUTGOING ? protocolPacket.getNewID() : protocolPacket.getOldID();
packetWrapper.setId(newId); packetWrapper.setId(newId);
if (protocolPacket.getRemapper() == null) {
return; PacketRemapper remapper = protocolPacket.getRemapper();
if (remapper != null) {
try {
remapper.remap(packetWrapper);
} catch (InformativeException e) { // Catch InformativeExceptions, pass through CancelExceptions
throwRemapError(direction, state, oldId, newId, e);
return;
}
if (packetWrapper.isCancelled()) {
throw CancelException.generate();
}
} }
}
// Remap private void throwRemapError(Direction direction, State state, int oldId, int newId, InformativeException e) throws InformativeException {
try { // Don't print errors during handshake
protocolPacket.getRemapper().remap(packetWrapper); if (state == State.HANDSHAKE) {
} catch (InformativeException e) { // Catch InformativeExceptions, pass through CancelExceptions
// Don't print errors during handshake
if (state == State.HANDSHAKE) {
throw e;
}
Class<? extends PacketType> packetTypeClass = state == State.PLAY ? (direction == Direction.OUTGOING ? oldClientboundPacketEnum : newServerboundPacketEnum) : null;
if (packetTypeClass != null) {
PacketType[] enumConstants = packetTypeClass.getEnumConstants();
PacketType packetType = oldId < enumConstants.length && oldId >= 0 ? enumConstants[oldId] : null;
Via.getPlatform().getLogger().warning("ERROR IN " + getClass().getSimpleName() + " IN REMAP OF " + packetType + " (" + toNiceHex(oldId) + ")");
} else {
Via.getPlatform().getLogger().warning("ERROR IN " + getClass().getSimpleName()
+ " IN REMAP OF " + toNiceHex(oldId) + "->" + toNiceHex(newId));
}
throw e; throw e;
} }
if (packetWrapper.isCancelled()) { Class<? extends PacketType> packetTypeClass = state == State.PLAY ? (direction == Direction.OUTGOING ? oldClientboundPacketEnum : newServerboundPacketEnum) : null;
throw CancelException.generate(); if (packetTypeClass != null) {
PacketType[] enumConstants = packetTypeClass.getEnumConstants();
PacketType packetType = oldId < enumConstants.length && oldId >= 0 ? enumConstants[oldId] : null;
Via.getPlatform().getLogger().warning("ERROR IN " + getClass().getSimpleName() + " IN REMAP OF " + packetType + " (" + toNiceHex(oldId) + ")");
} else {
Via.getPlatform().getLogger().warning("ERROR IN " + getClass().getSimpleName()
+ " IN REMAP OF " + toNiceHex(oldId) + "->" + toNiceHex(newId));
} }
throw e;
} }
private String toNiceHex(int id) { private String toNiceHex(int id) {

Datei anzeigen

@ -1,5 +1,6 @@
package us.myles.ViaVersion.api.protocol; package us.myles.ViaVersion.api.protocol;
import com.google.common.collect.Lists;
import us.myles.ViaVersion.api.PacketWrapper; import us.myles.ViaVersion.api.PacketWrapper;
import us.myles.ViaVersion.api.Via; import us.myles.ViaVersion.api.Via;
import us.myles.ViaVersion.api.data.UserConnection; import us.myles.ViaVersion.api.data.UserConnection;
@ -9,13 +10,13 @@ import us.myles.ViaVersion.packets.State;
import us.myles.ViaVersion.protocols.base.ProtocolInfo; import us.myles.ViaVersion.protocols.base.ProtocolInfo;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Collections;
import java.util.List; import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList; import java.util.concurrent.CopyOnWriteArrayList;
import java.util.logging.Level; import java.util.logging.Level;
public class ProtocolPipeline extends SimpleProtocol { public class ProtocolPipeline extends SimpleProtocol {
private List<Protocol> protocolList; private List<Protocol> incoming;
private List<Protocol> outgoing;
private UserConnection userConnection; private UserConnection userConnection;
public ProtocolPipeline(UserConnection userConnection) { public ProtocolPipeline(UserConnection userConnection) {
@ -25,9 +26,10 @@ public class ProtocolPipeline extends SimpleProtocol {
@Override @Override
protected void registerPackets() { protected void registerPackets() {
protocolList = new CopyOnWriteArrayList<>(); incoming = new CopyOnWriteArrayList<>();
outgoing = Lists.reverse(incoming);
// This is a pipeline so we register basic pipes // This is a pipeline so we register basic pipes
protocolList.add(ProtocolRegistry.BASE_PROTOCOL); incoming.add(ProtocolRegistry.BASE_PROTOCOL);
} }
@Override @Override
@ -40,7 +42,7 @@ public class ProtocolPipeline extends SimpleProtocol {
userConnection.setProtocolInfo(protocolInfo); userConnection.setProtocolInfo(protocolInfo);
/* Init through all our pipes */ /* Init through all our pipes */
for (Protocol protocol : protocolList) { for (Protocol protocol : incoming) {
protocol.init(userConnection); protocol.init(userConnection);
} }
} }
@ -52,59 +54,59 @@ public class ProtocolPipeline extends SimpleProtocol {
* @param protocol The protocol to add to the end * @param protocol The protocol to add to the end
*/ */
public void add(Protocol protocol) { public void add(Protocol protocol) {
if (protocolList != null) { if (incoming != null) {
protocolList.add(protocol); incoming.add(protocol);
protocol.init(userConnection); protocol.init(userConnection);
// Move base Protocols to the end, so the login packets can be modified by other protocols // Move base Protocols to the end, so the login packets can be modified by other protocols
List<Protocol> toMove = new ArrayList<>(); List<Protocol> toMove = new ArrayList<>();
for (Protocol p : protocolList) { for (Protocol p : incoming) {
if (ProtocolRegistry.isBaseProtocol(p)) { if (ProtocolRegistry.isBaseProtocol(p)) {
toMove.add(p); toMove.add(p);
} }
} }
protocolList.removeAll(toMove); incoming.removeAll(toMove);
protocolList.addAll(toMove); incoming.addAll(toMove);
} else { } else {
throw new NullPointerException("Tried to add protocol to early"); throw new NullPointerException("Tried to add protocol too early");
} }
} }
@Override @Override
public void transform(Direction direction, State state, PacketWrapper packetWrapper) throws Exception { public void transform(Direction direction, State state, PacketWrapper packetWrapper) throws Exception {
int originalID = packetWrapper.getId(); int originalID = packetWrapper.getId();
List<Protocol> protocols = new ArrayList<>(protocolList); List<Protocol> protocols = direction == Direction.OUTGOING ? outgoing : incoming;
// Other way if outgoing
if (direction == Direction.OUTGOING)
Collections.reverse(protocols);
// Apply protocols // Apply protocols
packetWrapper.apply(direction, state, 0, protocols); packetWrapper.apply(direction, state, 0, protocols);
super.transform(direction, state, packetWrapper); super.transform(direction, state, packetWrapper);
if (Via.getManager().isDebug()) { if (Via.getManager().isDebug()) {
// Debug packet logPacket(direction, state, packetWrapper, originalID);
int serverProtocol = userConnection.getProtocolInfo().getServerProtocolVersion();
int clientProtocol = userConnection.getProtocolInfo().getProtocolVersion();
ViaPlatform platform = Via.getPlatform();
String actualUsername = packetWrapper.user().getProtocolInfo().getUsername();
String username = actualUsername != null ? actualUsername + " " : "";
platform.getLogger().log(Level.INFO, "{0}{1} {2}: {3} (0x{4}) -> {5} (0x{6}) [{7}] {8}",
new Object[]{
username,
direction,
state,
originalID,
Integer.toHexString(originalID),
packetWrapper.getId(),
Integer.toHexString(packetWrapper.getId()),
Integer.toString(clientProtocol),
packetWrapper
});
} }
} }
private void logPacket(Direction direction, State state, PacketWrapper packetWrapper, int originalID) {
// Debug packet
int clientProtocol = userConnection.getProtocolInfo().getProtocolVersion();
ViaPlatform platform = Via.getPlatform();
String actualUsername = packetWrapper.user().getProtocolInfo().getUsername();
String username = actualUsername != null ? actualUsername + " " : "";
platform.getLogger().log(Level.INFO, "{0}{1} {2}: {3} (0x{4}) -> {5} (0x{6}) [{7}] {8}",
new Object[]{
username,
direction,
state,
originalID,
Integer.toHexString(originalID),
packetWrapper.getId(),
Integer.toHexString(packetWrapper.getId()),
Integer.toString(clientProtocol),
packetWrapper
});
}
/** /**
* Check if the pipeline contains a protocol * Check if the pipeline contains a protocol
* *
@ -112,14 +114,14 @@ public class ProtocolPipeline extends SimpleProtocol {
* @return True if the protocol class is in the pipeline * @return True if the protocol class is in the pipeline
*/ */
public boolean contains(Class<? extends Protocol> pipeClass) { public boolean contains(Class<? extends Protocol> pipeClass) {
for (Protocol protocol : protocolList) { for (Protocol protocol : incoming) {
if (protocol.getClass().equals(pipeClass)) return true; if (protocol.getClass().equals(pipeClass)) return true;
} }
return false; return false;
} }
public <P extends Protocol> P getProtocol(Class<P> pipeClass) { public <P extends Protocol> P getProtocol(Class<P> pipeClass) {
for (Protocol protocol : protocolList) { for (Protocol protocol : incoming) {
if (protocol.getClass() == pipeClass) return (P) protocol; if (protocol.getClass() == pipeClass) return (P) protocol;
} }
return null; return null;
@ -134,7 +136,7 @@ public class ProtocolPipeline extends SimpleProtocol {
* @throws Exception If it failed to convert / packet cancelld. * @throws Exception If it failed to convert / packet cancelld.
*/ */
public boolean filter(Object o, List list) throws Exception { public boolean filter(Object o, List list) throws Exception {
for (Protocol protocol : protocolList) { for (Protocol protocol : incoming) {
if (protocol.isFiltered(o.getClass())) { if (protocol.isFiltered(o.getClass())) {
protocol.filterPacket(userConnection, o, list); protocol.filterPacket(userConnection, o, list);
return true; return true;
@ -145,7 +147,7 @@ public class ProtocolPipeline extends SimpleProtocol {
} }
public List<Protocol> pipes() { public List<Protocol> pipes() {
return protocolList; return incoming;
} }
/** /**