Netty4 websocket服务器：正确处理浏览器401响应
使用Netty4开发WebSocket服务器时，常常需要验证客户端Token。验证失败，服务器应返回401状态码并关闭连接。然而，浏览器有时无法正确接收此响应。本文将详细说明如何解决此问题。
问题：使用var socket = new WebSocket(“ws://127.0.0.1:18080/ws?token=xxxx”);连接服务器，服务器验证Token。失败时，服务器返回401并关闭连接，但浏览器未收到401响应。服务器代码片段如下：
private void httpResponse401(ChannelHandlerContext ctx, FullHttpRequest request){ FullHttpResponse response = new DefaultFullHttpResponse(request.protocolVersion(), HttpResponseStatus.UNAUTHORIZED); response.headers().set(HttpHeaderNames.CONTENT_LENGTH, 0); ctx.writeAndFlush(response).addListener(ChannelFutureListener.CLOSE); ReferenceCountUtil.release(request); }
原因：问题在于WebSocket握手阶段。握手请求是HTTP请求，但握手成功后，通信不再是HTTP协议。因此，401响应必须在握手阶段返回。
解决方案：在处理WebSocket握手请求的代码中验证Token。验证失败，直接返回401响应，不执行WebSocket连接建立逻辑。
改进后的代码示例：
@Override public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception { if (msg instanceof FullHttpRequest) { FullHttpRequest request = (FullHttpRequest) msg; String token = extractTokenFromRequest(request); //提取Token的辅助函数 if (!validateToken(token)) { httpResponse401(ctx, request); return; } // Token验证通过，继续WebSocket握手 WebSocketServerHandshakerFactory wsFactory = new WebSocketServerHandshakerFactory( getWebSocketLocation(request), null, false); WebSocketServerHandshaker handshaker = wsFactory.newHandshaker(request); if (handshaker == null) { WebSocketServerHandshakerFactory.sendUnsupportedVersionResponse(ctx.channel()); } else { handshaker.handshake(ctx.channel(), request); } } else if (msg instanceof WebSocketFrame) { // 处理WebSocket帧 } } private String extractTokenFromRequest(FullHttpRequest request) { String uri = request.uri(); String[] parts = uri.split("?"); if (parts.length > 1) { String[] params = parts[1].split("&"); for (String param : params) { String[] keyValue = param.split("="); if (keyValue.length == 2 && keyValue[0].equals("token")) { return keyValue[1]; } } } return null; } private boolean validateToken(String token) { // 这里实现token验证逻辑 return token != null && token.equals("validToken"); // 示例，替换为实际验证逻辑 } private void httpResponse401(ChannelHandlerContext ctx, FullHttpRequest request) { FullHttpResponse response = new DefaultFullHttpResponse( HttpVersion.HTTP_1_1, HttpResponseStatus.UNAUTHORIZED); response.headers().set(HttpHeaderNames.CONTENT_TYPE, "text/plain; charset=UTF-8"); response.headers().set(HttpHeaderNames.CONTENT_LENGTH, response.content().readableBytes()); ctx.writeAndFlush(response).addListener(ChannelFutureListener.CLOSE); ReferenceCountUtil.release(request); }
通过在握手阶段进行Token验证并返回401响应，浏览器就能正确识别连接关闭的原因，从而实现更健壮的WebSocket服务器。 extractTokenFromRequest 函数增强了Token提取的鲁棒性。 请将示例中的token验证替换为您的实际验证逻辑。
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
