This commit is contained in:
Dereku 2023-03-10 09:55:53 +07:00
parent 50f6d37416
commit d3a21672fd

View file

@ -1,24 +1,19 @@
package net.rubeta.legacyping; package net.rubeta.legacyping;
import org.bukkit.Bukkit;
import org.bukkit.Server; import org.bukkit.Server;
import org.bukkit.craftbukkit.CraftServer; import org.bukkit.craftbukkit.CraftServer;
import java.io.*; import java.io.*;
import java.net.InetAddress; import java.net.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Arrays;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean; import java.util.concurrent.atomic.AtomicBoolean;
public class ServerThread extends Thread { public class ServerThread extends Thread {
// Queries start private static final String MODERN_RESPONSE =
private static final byte[] RUBETA_MODDED_CLIENT = new byte[]{0, 0, 0, 1, (byte) 254}; "{\"version\": {\"name\": \"b1.7.3\",\"protocol\": -17}," +
private static final byte[] LEGACY_PING_B1_8_TO_R1_3 = new byte[]{(byte) 254}; "\"players\": {\"max\": %max%,\"online\": %now%,\"sample\": []}," +
private static final byte[] PING_R1_4_TO_R1_5 = new byte[]{(byte) 254, 1}; "\"description\": {\"text\": \"%motd%\"},\"enforcesSecureChat\": false}";
// Queries end
private final ServerSocket serverSocket; private final ServerSocket serverSocket;
private final AtomicBoolean working = new AtomicBoolean(false); private final AtomicBoolean working = new AtomicBoolean(false);
private final Server minecraftServer; private final Server minecraftServer;
@ -28,6 +23,7 @@ public class ServerThread extends Thread {
final InetAddress address = InetAddress.getByName(host); final InetAddress address = InetAddress.getByName(host);
this.serverSocket = new ServerSocket(port, 10, address); this.serverSocket = new ServerSocket(port, 10, address);
this.serverSocket.setSoTimeout((int) TimeUnit.SECONDS.toMillis(10L)); this.serverSocket.setSoTimeout((int) TimeUnit.SECONDS.toMillis(10L));
this.serverSocket.setPerformancePreferences(0, 2, 1);
} }
@Override @Override
@ -35,23 +31,23 @@ public class ServerThread extends Thread {
this.working.set(true); this.working.set(true);
while (this.working.get() || !this.serverSocket.isClosed()) { while (this.working.get() || !this.serverSocket.isClosed()) {
try { try {
final Socket newConnection = this.serverSocket.accept(); final Socket socket = this.serverSocket.accept();
final InputStream inputStream = newConnection.getInputStream(); socket.setTrafficClass(SocketOptions.TCP_NODELAY & SocketOptions.IP_TOS);
final ByteArrayOutputStream result = new ByteArrayOutputStream(); final InputStream inputStream = socket.getInputStream();
byte[] buffer = new byte[16]; final OutputStream outputStream = socket.getOutputStream();
for (int length; (length = inputStream.read(buffer)) != -1; ) { final int firstByte = inputStream.read();
result.write(buffer, 0, length); switch (firstByte) {
case 0:
this.responseForRuBetaModdedClient(outputStream);
break;
case 254:
this.responseForLegacyPing(outputStream);
break;
default:
this.responseForModernPing(outputStream);
} }
final byte[] query = result.toByteArray(); outputStream.flush();
final OutputStream outputStream = newConnection.getOutputStream(); socket.close();
if (Arrays.equals(RUBETA_MODDED_CLIENT, query)) {
this.responseForRuBetaModdedClient(outputStream);
} else if (Arrays.equals(LEGACY_PING_B1_8_TO_R1_3, query)) {
this.responseForLegacyPing(outputStream);
} else if (Arrays.equals(PING_R1_4_TO_R1_5, query)) {
this.responseForR14Ping(outputStream);
}
newConnection.close();
} catch (IOException ignore) { } catch (IOException ignore) {
} }
} }
@ -82,12 +78,13 @@ public class ServerThread extends Thread {
clientOutput.write(result.toByteArray()); clientOutput.write(result.toByteArray());
} }
@SuppressWarnings("UnnecessaryUnicodeEscape")
private void responseForLegacyPing(final OutputStream outputStream) throws IOException { private void responseForLegacyPing(final OutputStream outputStream) throws IOException {
final DataOutputStream clientOutput = new DataOutputStream(outputStream); final DataOutputStream clientOutput = new DataOutputStream(outputStream);
clientOutput.write(255); // Packet 0xFF (PacketDisconnect) clientOutput.write(255);
final String response = ((CraftServer) this.minecraftServer).getServer().motd final String response = ((CraftServer) this.minecraftServer).getServer().motd
+ '\u0000' + this.minecraftServer.getOnlinePlayers().length + '\u00a7' + this.minecraftServer.getOnlinePlayers().length
+ '\u0000' + this.minecraftServer.getMaxPlayers(); + '\u00a7' + this.minecraftServer.getMaxPlayers();
clientOutput.writeShort(response.length()); clientOutput.writeShort(response.length());
for (char c : response.toCharArray()) { for (char c : response.toCharArray()) {
clientOutput.writeChar(c); clientOutput.writeChar(c);
@ -95,16 +92,27 @@ public class ServerThread extends Thread {
// Packet data end // Packet data end
} }
private void responseForR14Ping(final OutputStream outputStream) throws IOException { private void responseForModernPing(final OutputStream outputStream) throws IOException {
final String responseJson = MODERN_RESPONSE.replace("%max%", "" + this.minecraftServer.getMaxPlayers())
.replace("%now%", "" + this.minecraftServer.getOnlinePlayers().length)
.replace("%motd%", ((CraftServer) this.minecraftServer).getServer().motd);
byte[] bytes = responseJson.getBytes();
final DataOutputStream clientOutput = new DataOutputStream(outputStream); final DataOutputStream clientOutput = new DataOutputStream(outputStream);
clientOutput.write(255); // Packet 0xFF (PacketDisconnect) this.writeVarInt(clientOutput, bytes.length + 3);
final String response = '§' + 1 this.writeVarInt(clientOutput, 0);
+ '\u0000' + "b1.7.3m" + '\u0000' + 17 // Protocol version this.writeVarInt(clientOutput, bytes.length);
+ '\u0000' + ((CraftServer) this.minecraftServer).getServer().motd clientOutput.write(bytes);
+ '\u0000' + this.minecraftServer.getOnlinePlayers().length
+ '\u0000' + this.minecraftServer.getMaxPlayers();
clientOutput.writeShort(response.length());
clientOutput.writeUTF(response);
// Packet data end // Packet data end
} }
private void writeVarInt(DataOutputStream out, int paramInt) throws IOException {
while (true) {
if ((paramInt & 0xFFFFFF80) == 0) {
out.writeByte(paramInt);
return;
}
out.writeByte(paramInt & 0x7F | 0x80);
paramInt >>>= 7;
}
}
} }