servlet实现防盗链及文件下载

package servlet;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.URLEncoder;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

@WebServlet("/Test1")
public class Test1 extends HttpServlet {
private static final long serialVersionUID = 1L;

protected void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
String refer = request.getHeader("referer");
String site = "http://" + request.getServerName();
if (refer != null && refer.startsWith(site)) {
String Directory = request.  
               getServletContext().getRealPath("/WEB-INF/");  
File file=new File(Directory+"\\file","测试.docx");
String filename=URLEncoder.encode(file.getName(),"utf-8");
response.setContentType("application/octet-stream");
response.setHeader("Content-Disposition","attachment;filename="+filename);
response.setHeader("Content-Length", file.length()+"");
BufferedInputStream bis=new BufferedInputStream(new FileInputStream(file));
BufferedOutputStream bos=new BufferedOutputStream(response.getOutputStream());
byte[] buffer = new byte[1024];
int i = -1;
while ((i = bis.read(buffer)) != -1) {
       bos.write(buffer, 0, i);
}
       bos.flush();
       bos.close();
       bis.close();       
}
else {
request.getRequestDispatcher("/download.html").forward(request, response);
}

}

protected void doPost(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
doGet(request, response);
}

}


download.html:

<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<title>download</title>
</head>
<body>
<a href="/chapter5/Test1">download here</a>
</body>
</html>

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值