我要评论
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验证替换为您的实际验证逻辑。
以上就是在使用netty4开发websocket服务器时，如何确保浏览器正确识别401响应？的详细内容，更多请关注代码网其它相关文章！
发表评论