netty模拟文件列表http服务器

学习链接

Netty网络框架 :HTTP模拟文件列表服务器

HttpObjectDecoder&HttpObjectEncoder源码浅析

ChunkedWriteHandler源码浅析

HttpPostRequestEncoder使用示例

HTTP模拟文件列表服务器

FileServer

packagecom.zzhua.file;importio.netty.bootstrap.ServerBootstrap;importio.netty.channel.ChannelFuture;importio.netty.channel.ChannelInitializer;importio.netty.channel.ChannelPipeline;importio.netty.channel.nio.NioEventLoopGroup;importio.netty.channel.socket.SocketChannel;importio.netty.channel.socket.nio.NioServerSocketChannel;importio.netty.handler.codec.http.HttpObjectAggregator;importio.netty.handler.codec.http.HttpServerCodec;importio.netty.handler.logging.LoggingHandler;importio.netty.handler.stream.ChunkedWriteHandler;//通过 HTTP 请求 返回访问指定目录的文件列表publicclassFileServer{publicstaticvoidmain(String[]args){NioEventLoopGroupbossGroup=newNioEventLoopGroup();NioEventLoopGroupworkerGroup=newNioEventLoopGroup();//连接指引对象ServerBootstrapserverBootstrap=newServerBootstrap();//链式编程serverBootstrap.group(bossGroup,workerGroup).channel(NioServerSocketChannel.class).handler(newLoggingHandler()).childHandler(newChannelInitializer<SocketChannel>(){@OverrideprotectedvoidinitChannel(SocketChannelsocketChannel)throwsException{ChannelPipelinepipeline=socketChannel.pipeline();//设定编解码器 Netty提供的http 编解码器pipeline.addLast(newHttpServerCodec());//数据聚合器 因为是分段传输 处理Http消息的聚合器pipeline.addLast(newHttpObjectAggregator(512*1024));//支持大数据的文件传输 控制对内存的使用pipeline.addLast(newChunkedWriteHandler());// 添加自定义处理器pipeline.addLast(newFileServerHandler());}});// 启动服务器try{ChannelFuturefuture=serverBootstrap.bind(9090).sync();// 阻塞关闭future.channel().closeFuture().sync();}catch(Exceptione){e.printStackTrace();}finally{workerGroup.shutdownGracefully();bossGroup.shutdownGracefully();}}}

FileServerHandler

packagecom.zzhua.file;importio.netty.buffer.ByteBuf;importio.netty.buffer.Unpooled;importio.netty.channel.ChannelFutureListener;importio.netty.channel.ChannelHandlerContext;importio.netty.channel.SimpleChannelInboundHandler;importio.netty.handler.codec.http.*;importio.netty.util.CharsetUtil;importjava.io.File;importjava.io.UnsupportedEncodingException;importjava.net.URLDecoder;importjava.util.regex.Pattern;/** * @projectName: gchatsystem * @package: com.hyc.gchatsystem.file * @className: FileServerHandler * @author: 冷环渊 doomwatcher * @description: TODO * @date: 2022/4/13 17:36 * @version: 1.0 *///文件服务器处理器publicclassFileServerHandlerextendsSimpleChannelInboundHandler<FullHttpRequest>{privatestaticfinalPatternINSECURE_URI=Pattern.compile(".*[<>&\"].*");//文件名称匹配规则，必须是英文、数字下划线、中划线privatestaticfinalPatternALLOWED_FILE_NAME=Pattern.compile("[A-Za-z0-9][-_A-Za-z0-9\\.]*");//文件路徑privatefinalStringurl;publicFileServerHandler(){this.url="/";}@OverrideprotectedvoidchannelRead0(ChannelHandlerContextctx,FullHttpRequestrequest)throwsException{// 如果无法解码,返回400if(!request.decoderResult().isSuccess()){sendError(ctx,HttpResponseStatus.BAD_REQUEST);return;}//只支持GET方法,其他方法返回if(request.method()!=HttpMethod.GET){sendError(ctx,HttpResponseStatus.METHOD_NOT_ALLOWED);return;}//获取文件路径finalStringuri=request.uri();// 格式化URL，并且获取文件的磁盘路径finalStringpath=sanitizeUri(uri);System.out.println(path);if(path==null){sendError(ctx,HttpResponseStatus.FORBIDDEN);return;}Filefile=newFile(path);// 如果文件隐藏不可访问或者文件不存在if(file.isHidden()||!file.exists()){sendError(ctx,HttpResponseStatus.NOT_FOUND);return;}//如果是文件目录if(file.isDirectory()){//以/结尾时，就列出文件夹下的所有文件if(uri.endsWith("/")){sendListing(ctx,file);}else{//否则进行重定向，打开文件夹，继续深入sendRedirect(ctx,uri+'/');}return;}//既不是文件夹，也不是文件if(!file.isFile()){sendError(ctx,HttpResponseStatus.FORBIDDEN);}else{sendFile(ctx,request);}}privatevoidsendFile(ChannelHandlerContextctx,FullHttpRequestrequest){try{Stringuri=request.uri();Filefile=newFile(System.getProperty("user.dir")+File.separator+uri);if(!file.exists()){thrownewFileNotFoundException(file.getPath());}DefaultHttpResponseresponse=newDefaultHttpResponse(HttpVersion.HTTP_1_1,HttpResponseStatus.OK);response.headers().set(HttpHeaderNames.CONTENT_TYPE,"application/octet-stream");response.headers().set(HttpHeaderNames.CONTENT_DISPOSITION,"attachment; filename="+URLEncoder.encode(file.getName()));response.headers().set(HttpHeaderNames.CONTENT_LENGTH,file.length());ctx.writeAndFlush(response);HttpChunkedInputhttpChunkedInput=newHttpChunkedInput(newChunkedFile(file));ctx.writeAndFlush(httpChunkedInput);}catch(Exceptione){ctx.fireExceptionCaught(e);}}/** * 异常处理 */@OverridepublicvoidexceptionCaught(ChannelHandlerContextctx,Throwablecause)throwsException{cause.printStackTrace();if(ctx.channel().isActive()){sendError(ctx,HttpResponseStatus.INTERNAL_SERVER_ERROR);}}/** * 格式化uri，返回文件的磁盘路径 * @param uri * @return * */privateStringsanitizeUri(Stringuri){try{uri=URLDecoder.decode(uri,"UTF-8");}catch(UnsupportedEncodingExceptione){//用utf-8解码出现错误时，就换种解码方式try{uri=URLDecoder.decode(uri,"ISO-8859-1");}catch(UnsupportedEncodingExceptione1){thrownewError();}}if(!uri.startsWith(url)){returnnull;}if(!uri.startsWith("/")){returnnull;}uri=uri.replace('/',File.separatorChar);if(uri.contains(File.separator+'.')||uri.contains('.'+File.separator)||uri.startsWith(".")||uri.endsWith(".")||INSECURE_URI.matcher(uri).matches()){returnnull;}System.out.println(System.getProperty("user.dir"));//returnSystem.getProperty("user.dir")+File.separator+uri;}/** * 展示文件列表 * @param ctx * @param dir * */privatestaticvoidsendListing(ChannelHandlerContextctx,Filedir){FullHttpResponseresponse=newDefaultFullHttpResponse(HttpVersion.HTTP_1_1,HttpResponseStatus.OK);response.headers().set(HttpHeaderNames.CONTENT_TYPE,"text/html; charset=UTF-8");StringBuilderbuf=newStringBuilder();StringdirPath=dir.getPath();buf.append("<!DOCTYPE html>\r\n");buf.append("<html><head><title>");buf.append("使用netty做下载文件服务器");buf.append("</title></head><body>\r\n");buf.append("<h3>");buf.append("当前文件夹位置：");buf.append(dirPath);buf.append("</h3>\r\n");buf.append("<h4>");buf.append("文件列表");buf.append("</h4>\r\n");buf.append("<ul>");buf.append("<li>上一级链接：<a href=\"../\">..</a></li>\r\n");for(Filef:dir.listFiles()){//隐藏文件，不可读文件直接跳过if(f.isHidden()||!f.canRead()){continue;}Stringname=f.getName();//非英文、数字、下划线、中划线组成的文件，也跳过if(!ALLOWED_FILE_NAME.matcher(name).matches()){continue;}buf.append("<li>链接：<a href=\"");buf.append(name);buf.append("\">");buf.append(name);buf.append("</a></li>\r\n");}buf.append("</ul></body></html>\r\n");ByteBufbuffer=Unpooled.copiedBuffer(buf,CharsetUtil.UTF_8);response.content().writeBytes(buffer);buffer.release();ctx.writeAndFlush(response).addListener(ChannelFutureListener.CLOSE);}/** * 发送重定向 * @param ctx * @param newUri * */privatestaticvoidsendRedirect(ChannelHandlerContextctx,StringnewUri){FullHttpResponseresponse=newDefaultFullHttpResponse(HttpVersion.HTTP_1_1,HttpResponseStatus.FOUND);//设置访问的url路径为新的文件夹路径response.headers().set(HttpHeaderNames.LOCATION,newUri);ctx.writeAndFlush(response).addListener(ChannelFutureListener.CLOSE);}/** * 显示错误信息 * @param ctx * @param status * */privatestaticvoidsendError(ChannelHandlerContextctx,HttpResponseStatusstatus){FullHttpResponseresponse=newDefaultFullHttpResponse(HttpVersion.HTTP_1_1,status,Unpooled.copiedBuffer("Failure: "+status.toString()+"\r\n",CharsetUtil.UTF_8));response.headers().set(HttpHeaderNames.CONTENT_TYPE,"text/plain; charset=UTF-8");ctx.writeAndFlush(response).addListener(ChannelFutureListener.CLOSE);}}