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:
Ursprung
be6e086aac
Commit
fc7a7396ef
@ -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) {
|
||||||
|
@ -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;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
Laden…
In neuem Issue referenzieren
Einen Benutzer sperren