Netty4 WebSocket服务器如何优雅处理未授权连接 (401状态码)
在使用Netty构建WebSocket服务器时，安全验证至关重要。本文探讨如何正确处理token验证失败的情况，并确保浏览器能够识别服务器返回的401未授权状态码，从而提升用户体验和调试效率。
问题：在WebSocket握手阶段，如果token验证失败，服务器返回401状态码并关闭连接，但浏览器无法感知具体的错误原因。
原始代码存在的问题在于，返回的401响应体为空，且缺少必要的HTTP头部信息，导致浏览器无法正确解析该响应。
改进后的解决方案：在返回401响应时，需要包含一个描述性的错误信息以及完整的HTTP头部信息。
改进后的服务器代码：
private void httpResponse401(ChannelHandlerContext ctx, FullHttpRequest request) {
FullHttpResponse response = new DefaultFullHttpResponse(
request.protocolVersion(),
HttpResponseStatus.UNAUTHORIZED,
Unpooled.copiedBuffer("Unauthorized: Invalid token", CharsetUtil.UTF_8)
);
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);
}
关键改进：
-
添加错误信息:
Unpooled.copiedBuffer("Unauthorized: Invalid token", CharsetUtil.UTF_8)向响应体中添加了清晰的错误消息，方便浏览器和开发者理解错误原因。 -
设置CONTENT_TYPE头部:
response.headers().set(HttpHeaderNames.CONTENT_TYPE, "text/plain; charset=UTF-8")指定响应内容类型为纯文本，并指定字符集为UTF-8，确保浏览器能够正确解码错误信息。 -
设置CONTENT_LENGTH头部:
response.headers().set(HttpHeaderNames.CONTENT_LENGTH, response.content().readableBytes())准确告知浏览器响应体长度，避免解析错误。
通过这些改进，浏览器在连接失败时将收到包含401状态码和错误信息的HTTP响应，从而能够在开发者工具中显示具体的错误信息，方便开发者进行调试和排查问题，提升用户体验和开发效率。 这确保了WebSocket握手阶段的错误处理更加健壮和可靠。
