From a429447754baae5aeb008e2ed55e6c9f72a23141 Mon Sep 17 00:00:00 2001
From: Sen <sen@sen.gay>
Date: Wed, 4 Jun 2025 16:14:16 +0200
Subject: [PATCH] add early whitelist kick

---
 proxy/src/main/java/proxy/Proxy.java                | 5 +++++
 proxy/src/main/java/proxy/network/LoginHandler.java | 4 ++++
 2 files changed, 9 insertions(+)

diff --git a/proxy/src/main/java/proxy/Proxy.java b/proxy/src/main/java/proxy/Proxy.java
index ac2ce40..c401aad 100755
--- a/proxy/src/main/java/proxy/Proxy.java
+++ b/proxy/src/main/java/proxy/Proxy.java
@@ -98,6 +98,7 @@ public class Proxy {
 	private boolean epoll = false;
 	private boolean register = true;
 	private boolean checkCase = true;
+	private boolean kickOnConnect = true;
 	private String forwardHost = "127.0.0.1";
 	private int forwardPort = 25565;
 	private String proxyHost = "";
@@ -362,6 +363,10 @@ public class Proxy {
 	public boolean isCheckingCase() {
 		return this.checkCase;
 	}
+	
+	public boolean isKickingOnConnect() {
+		return this.kickOnConnect;
+	}
 
 	public int getMinimumPasswordLength() {
 		return this.minPassLength;
diff --git a/proxy/src/main/java/proxy/network/LoginHandler.java b/proxy/src/main/java/proxy/network/LoginHandler.java
index 42960ed..1bee675 100755
--- a/proxy/src/main/java/proxy/network/LoginHandler.java
+++ b/proxy/src/main/java/proxy/network/LoginHandler.java
@@ -54,6 +54,10 @@ public class LoginHandler implements Handler {
 			this.disconnect("That username is already taken");
 			return;
 		}
+		else if(this.proxy.isKickingOnConnect() && !this.proxy.canRegister() && this.proxy.getUser(this.username) == null) {
+			this.disconnect("You are not whitelisted on this server");
+			return;
+		}
 		this.state = LoginState.DONE;
 
 		if(this.proxy.getCompression() >= 0) {