74 lines
3.8 KiB
Java
74 lines
3.8 KiB
Java
package net.minecraft.network.protocol;
|
|
|
|
import com.mojang.logging.LogUtils;
|
|
import net.minecraft.CrashReport;
|
|
import net.minecraft.CrashReportCategory;
|
|
import net.minecraft.CrashReportDetail;
|
|
import net.minecraft.ReportedException;
|
|
import net.minecraft.network.PacketListener;
|
|
import net.minecraft.server.RunningOnDifferentThreadException;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.util.thread.BlockableEventLoop;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import org.slf4j.Logger;
|
|
|
|
public class PacketUtils {
|
|
private static final Logger LOGGER = LogUtils.getLogger();
|
|
|
|
/**
|
|
* Ensures that the given packet is handled on the main thread. If the current thread is not the main thread, this method
|
|
* throws {@link net.minecraft.server.RunningOnDifferentThreadException}, which is caught and ignored in the outer call ({@link net.minecraft.network.Connection#channelRead0(io.netty.channel.ChannelHandlerContext, net.minecraft.network.protocol.Packet)}). Additionally, it then re-schedules the packet to be handled on the main thread,
|
|
* which will then end up back here, but this time on the main thread.
|
|
*/
|
|
public static <T extends PacketListener> void ensureRunningOnSameThread(Packet<T> packet, T processor, ServerLevel level) throws RunningOnDifferentThreadException {
|
|
ensureRunningOnSameThread(packet, processor, level.getServer());
|
|
}
|
|
|
|
/**
|
|
* Ensures that the given packet is handled on the main thread. If the current thread is not the main thread, this method
|
|
* throws {@link net.minecraft.server.RunningOnDifferentThreadException}, which is caught and ignored in the outer call ({@link net.minecraft.network.Connection#channelRead0(io.netty.channel.ChannelHandlerContext, net.minecraft.network.protocol.Packet)}). Additionally, it then re-schedules the packet to be handled on the main thread,
|
|
* which will then end up back here, but this time on the main thread.
|
|
*/
|
|
public static <T extends PacketListener> void ensureRunningOnSameThread(Packet<T> packet, T processor, BlockableEventLoop<?> executor) throws RunningOnDifferentThreadException {
|
|
if (!executor.isSameThread()) {
|
|
executor.executeIfPossible(() -> {
|
|
if (processor.shouldHandleMessage(packet)) {
|
|
try {
|
|
packet.handle(processor);
|
|
} catch (Exception var4) {
|
|
if (var4 instanceof ReportedException reportedException && reportedException.getCause() instanceof OutOfMemoryError) {
|
|
throw makeReportedException(var4, packet, processor);
|
|
}
|
|
|
|
processor.onPacketError(packet, var4);
|
|
}
|
|
} else {
|
|
LOGGER.debug("Ignoring packet due to disconnection: {}", packet);
|
|
}
|
|
});
|
|
throw RunningOnDifferentThreadException.RUNNING_ON_DIFFERENT_THREAD;
|
|
}
|
|
}
|
|
|
|
public static <T extends PacketListener> ReportedException makeReportedException(Exception exception, Packet<T> packet, T packetListener) {
|
|
if (exception instanceof ReportedException reportedException) {
|
|
fillCrashReport(reportedException.getReport(), packetListener, packet);
|
|
return reportedException;
|
|
} else {
|
|
CrashReport crashReport = CrashReport.forThrowable(exception, "Main thread packet handler");
|
|
fillCrashReport(crashReport, packetListener, packet);
|
|
return new ReportedException(crashReport);
|
|
}
|
|
}
|
|
|
|
public static <T extends PacketListener> void fillCrashReport(CrashReport crashReport, T packetListener, @Nullable Packet<T> packet) {
|
|
if (packet != null) {
|
|
CrashReportCategory crashReportCategory = crashReport.addCategory("Incoming Packet");
|
|
crashReportCategory.setDetail("Type", (CrashReportDetail<String>)(() -> packet.type().toString()));
|
|
crashReportCategory.setDetail("Is Terminal", (CrashReportDetail<String>)(() -> Boolean.toString(packet.isTerminal())));
|
|
crashReportCategory.setDetail("Is Skippable", (CrashReportDetail<String>)(() -> Boolean.toString(packet.isSkippable())));
|
|
}
|
|
|
|
packetListener.fillCrashReport(crashReport);
|
|
}
|
|
}
|