geforkt von Mirrors/Velocity
Add unit tests
Dieser Commit ist enthalten in:
Ursprung
9ce4294e6e
Commit
5bf936d1b9
@ -138,8 +138,15 @@ public class HandshakeSessionHandler implements MinecraftSessionHandler {
|
||||
}
|
||||
}
|
||||
|
||||
private String cleanVhost(String hostname) {
|
||||
// Clean out any anything after any zero byte
|
||||
/**
|
||||
* Cleans the specified virtual host hostname.
|
||||
*
|
||||
* @param hostname the host name to clean
|
||||
* @return the cleaned hostname
|
||||
*/
|
||||
static String cleanVhost(String hostname) {
|
||||
// Clean out any anything after any zero bytes (this includes BungeeCord forwarding and the
|
||||
// legacy Forge handshake indicator).
|
||||
String cleaned = hostname;
|
||||
int zeroIdx = cleaned.indexOf('\0');
|
||||
if (zeroIdx > -1) {
|
||||
|
@ -0,0 +1,31 @@
|
||||
package com.velocitypowered.proxy.connection.client;
|
||||
|
||||
import static com.velocitypowered.proxy.connection.client.HandshakeSessionHandler.cleanVhost;
|
||||
import static com.velocitypowered.proxy.connection.forge.legacy.LegacyForgeConstants.*;
|
||||
import static org.junit.jupiter.api.Assertions.*;
|
||||
|
||||
import com.velocitypowered.proxy.connection.forge.legacy.LegacyForgeConstants;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
class HandshakeSessionHandlerTest {
|
||||
|
||||
@Test
|
||||
void cleanVhostHandlesGoodHostname() {
|
||||
assertEquals("localhost", cleanVhost("localhost"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void cleanVhostHandlesTrailingOctet() {
|
||||
assertEquals("localhost", cleanVhost("localhost."));
|
||||
}
|
||||
|
||||
@Test
|
||||
void cleanVhostHandlesForge() {
|
||||
assertEquals("localhost", cleanVhost("localhost" + HANDSHAKE_HOSTNAME_TOKEN));
|
||||
}
|
||||
|
||||
@Test
|
||||
void cleanVhostHandlesOctetsAndForge() {
|
||||
assertEquals("localhost", cleanVhost("localhost." + HANDSHAKE_HOSTNAME_TOKEN));
|
||||
}
|
||||
}
|
Laden…
In neuem Issue referenzieren
Einen Benutzer sperren