update

Author: guFalcon

pom.xml

@@ -10,7 +10,7 @@
 
 	<modelVersion>4.0.0</modelVersion>
 	<artifactId>websocket-server</artifactId>
-	<version>1.0.4</version>
+	<version>1.0.5</version>
 	<name>WebsocketServer</name>
 	<packaging>jar</packaging>
 

src/main/java/info/unterrainer/websocketserver/WebsocketServer.java

@@ -57,6 +57,7 @@ public WebsocketServer wsOauth(String path, WsOauthHandlerBase handler) {
 		wss.ws(path, ws -> {
 			ws.onConnect(handler::onConnect);
 			ws.onMessage(handler::onMessage);
+			ws.onBinaryMessage(handler::onBinaryMessage);
 			ws.onClose(handler::onClose);
 			ws.onError(handler::onError);
 		});

src/main/java/info/unterrainer/websocketserver/WsHandlerBase.java

@@ -1,5 +1,6 @@
 package info.unterrainer.websocketserver;
 
+import io.javalin.websocket.WsBinaryMessageContext;
 import io.javalin.websocket.WsCloseContext;
 import io.javalin.websocket.WsConnectContext;
 import io.javalin.websocket.WsErrorContext;
@@ -11,6 +12,8 @@ public abstract class WsHandlerBase {
 
 	public abstract void onMessage(WsMessageContext ctx) throws Exception;
 
+	public abstract void onBinaryMessage(WsBinaryMessageContext ctx) throws Exception;
+
 	public abstract void onClose(WsCloseContext ctx) throws Exception;
 
 	public abstract void onError(WsErrorContext ctx) throws Exception;

src/main/java/info/unterrainer/websocketserver/WsOauthHandlerBase.java

@@ -8,6 +8,7 @@
 import org.eclipse.jetty.websocket.api.Session;
 
 import info.unterrainer.oauthtokenmanager.OauthTokenManager;
+import io.javalin.websocket.WsBinaryMessageContext;
 import io.javalin.websocket.WsCloseContext;
 import io.javalin.websocket.WsConnectContext;
 import io.javalin.websocket.WsErrorContext;
@@ -98,6 +99,17 @@ public void onMessage(WsMessageContext ctx) throws Exception {
 		}
 	}
 
+	@Override
+	public void onBinaryMessage(WsBinaryMessageContext ctx) throws Exception {
+		log.debug("Received binary message from [{}]: [{}] bytes", ctx.session.getRemoteAddress(), ctx.data().length);
+		if (isQuarantined(ctx.session)) {
+			log.warn("Invalid Message from quarantined client [{}]. Disconnecting.", ctx.session.getRemoteAddress());
+			removeClient(ctx.session);
+			ctx.session.close();
+			return;
+		}
+	}
+
 	@Override
 	public void onClose(WsCloseContext ctx) throws Exception {
 		log.debug("Disconnected client: [{}]", ctx.session.getRemoteAddress());