web后端--一个小小的下载和防盗链功能

下载功能代码:

//1获取要下载的文件名
		String filename = req.getParameter("filename");
		//如果没有文件名
		if(filename==null||"".equals(filename=filename.trim())){
			//设置响应头,防止响应乱码
			resp.setContentType("text/html;charset=UTF-8");
			resp.getWriter().write("请指定要下载的文件名");
			return;
		}
		//下载逻辑
		//2.获取要下载的MIME类型
		int lastIndexOf = filename.lastIndexOf(".");
		String mimeType=null;
		String substring =null;
		if(lastIndexOf<=0){
			// 2.1 没有后缀,MIME 类型设置为二进制文件类型
			mimeType = req.getServletContext().getMimeType(".bin");
		}else{
			// 2.2 拿到要下载的文件的后缀
			substring = filename.substring(lastIndexOf);//获取.后面的名称,比如:是soso.apk,则获取到.apk
			mimeType = req.getServletContext().getMimeType(substring);
			if(mimeType==null||"".equals(mimeType=mimeType.trim())){
				// 如果当前后缀没有查到相应的MIME 类型,同样的设置为二进制文件类型
				mimeType = req.getServletContext().getMimeType(".bin");
			}
		}
		//3.获取文件输入流
		InputStream in = req.getServletContext().getResourceAsStream("/download/"+filename);
		if(in==null){
			 // 如果没有找到要下载的文件
			resp.setContentType("text/html;charset=UTF-8");
            resp.getWriter().write("文件[" + filename + "]不存在,请检查文件名再试");
            return;
		}
		
		// 4. 设置响应头
        // 4.1 设置响应正文的MIME 类型
		resp.setContentType(mimeType);
        // 4.2 设置响应正文的长度,byte
        resp.setContentLength(in.available());
        // 4.3 设置下载文件的描述信息(待下载文件的文件名)
        resp.addHeader("Content-Disposition", "attachment;filename=" + filename);
        
		// 5.输出响应正文,也就是文件
		OutputStream out = resp.getOutputStream();
		byte[] buffer=new byte[1024];
		int length=0;
		while((length=in.read(buffer))!=-1){
			out.write(buffer, 0, length);
		}
		in.close();

防盗链功能:

//防盗链
	// 根据Referer 请求头,确定当前请求是否来自本网站
        // 如果不是,响应 404, 并狠狠地骂回去
		String referer = req.getHeader("Referer");
		if(referer!=null&&!"".equals(referer=referer.trim())){
			//获取请求的来路应用,http://loca
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
防盗链(Referer)是指在一个网站上,通过直接访问其他网站的内容,使得其他网站的服务器消耗流量和带宽的行为。为了防止盗链,我们可以在服务器端对HTTP请求的Referer进行检查,如果检查到请求的Referer不是本站,则返回403 Forbidden错误,从而达到防盗链的目的。 下面是一个使用Java编写的防盗链的示例代码: ```java import java.io.IOException; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; public class AntiLeechServlet extends HttpServlet { private static final long serialVersionUID = 1L; protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { String referer = request.getHeader("referer"); if (referer == null || !referer.startsWith("http://www.example.com")) { response.sendError(HttpServletResponse.SC_FORBIDDEN); return; } // 正常处理请求 // ... } protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { doGet(request, response); } } ``` 以上代码中,我们通过获取HTTP请求的Referer头部信息,判断请求的Referer是否是本站(例如本站域名为www.example.com),如果不是,则返回403 Forbidden错误。如果是本站,则继续正常处理请求。你可以根据自己的实际情况进行修改和优化。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值