在使用Netty4开发WebSocket服务器时，如何确保浏览器正确识别401响应？_技术教程

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);
}