Netty4 WebSocket服務(wù)器：正確處理瀏覽器401響應(yīng)

使用Netty4開發(fā)WebSocket服務(wù)器時，常常需要驗(yàn)證客戶端Token。驗(yàn)證失敗，服務(wù)器應(yīng)返回401狀態(tài)碼并關(guān)閉連接。然而，瀏覽器有時無法正確接收此響應(yīng)。本文將詳細(xì)說明如何解決此問題。

問題：使用var socket = new WebSocket("ws://127.0.0.1:18080/ws?token=xxxx");連接服務(wù)器，服務(wù)器驗(yàn)證Token。失敗時，服務(wù)器返回401并關(guān)閉連接，但瀏覽器未收到401響應(yīng)。服務(wù)器代碼片段如下：

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協(xié)議。因此，401響應(yīng)必須在握手階段返回。

解決方案：在處理WebSocket握手請求的代碼中驗(yàn)證Token。驗(yàn)證失敗，直接返回401響應(yīng)，不執(zhí)行WebSocket連接建立邏輯。

改進(jìn)后的代碼示例：

@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
    if (msg instanceof FullHttpRequest) {
        FullHttpRequest request = (FullHttpRequest) msg;
        String token = extractTokenFromRequest(request); //提取Token的輔助函數(shù)

        if (!validateToken(token)) {
            httpResponse401(ctx, request);
            return;
        }

        // Token驗(yàn)證通過，繼續(xù)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) {
    // 這里實(shí)現(xiàn)token驗(yàn)證邏輯
    return token != null && token.equals("validToken"); // 示例，替換為實(shí)際驗(yàn)證邏輯
}

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

通過在握手階段進(jìn)行Token驗(yàn)證并返回401響應(yīng)，瀏覽器就能正確識別連接關(guān)閉的原因，從而實(shí)現(xiàn)更健壯的WebSocket服務(wù)器。 extractTokenFromRequest 函數(shù)增強(qiáng)了Token提取的魯棒性。 請將示例中的token驗(yàn)證替換為您的實(shí)際驗(yàn)證邏輯。

以上是在使用Netty4開發(fā)WebSocket服務(wù)器時，如何確保瀏覽器正確識別401響應(yīng)？的詳細(xì)內(nèi)容。更多信息請關(guān)注PHP中文網(wǎng)其他相關(guān)文章！