Filter实现全站违法关键词屏蔽
思路:客户端请求服务器数据,经过Filter过滤(请求放行,响应拦截),服务器向客户端返回数据时,在Filter中修改掉返回数据中违法的部分。
修改服务器的响应需要自定义一个HttpServletResponse的包装类,继承自HttpServletResponseWrapper即可实现自己的HttpServletResponse的包装类。
大概是这样的:
//package com.ecshop.tools; import java.io.CharArrayWriter; import java.io.InputStream; import java.io.PrintWriter; import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletResponseWrapper; public class ShieldKeywordResponseWrapper extends HttpServletResponseWrapper{ private CharArrayWriter caw; private PrintWriter pw; public ShieldKeywordResponseWrapper(HttpServletResponse response) { super(response); // 这个是我们保存返回结果的地方 caw = new CharArrayWriter(); // 这个是包装PrintWriter的,让所有结果通过这个PrintWriter写入到caw中 pw = new PrintWriter(caw); } @Override public PrintWriter getWriter() { return pw; } public String getResult() { return caw.toString(); } }
因为违法关键词都包含在响应文本消息中,服务器向请求客户端响应文本消息必须获取HttpServletResponse对象中的PrintWriter文本输出流对象进行输出消息,所以我们的这个封装类覆盖HttpServletResponse的getWriter是很重要的一步,我这个封装类的意思是让服务器响应数据时使用我们修改过的PrintWriter文本输出流,输出的数据并不立刻响应给客户,而是把输出的数据都保留在了CharArrayWriter这个对象中,当控制层执行完毕我们就可以在Filter中进行获取CharArrayWriter对象中的数据进行修改了,修改完毕我们在返回给你客户,就是我们想要的效果了。(而默认HttpServletResponse的PrintWriter对象是由tomcat自己所扩展自PrintWriter的CoyoteWriter对象,CoyoteWriter对象所打印的数据都保留在OutputBuffer对象(也是tomcat自己所扩展自Writer的一个对象)中,我们不方便获取它的文本数据,所以我们修改掉它,当OutputBuffer执行flush就发送消息,或者不执行让servlet结束后自动发送消息)。
以上是我看了一点tomcat源码然后瞎推理,对不对无所谓。。
当我们HttpServletResponse的包装类弄完剩下的就好办了。
//我们看下过滤器中的doFilter函数部分代码就足够了 public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException { HttpServletResponse res= (HttpServletResponse) response; //我们刚刚做的包装类在这里 ShieldKeywordResponseWrapper skrw= new ShieldKeywordResponseWrapper(res); chain.doFilter(request, skrw); //在servlet处理完毕后我们开始做处理 String result = skrw.getResult(); //替换关键词为* Properties p = new Properties(); p.load(ShieldKeywords.class.getResourceAsStream("/keywords.properties")); Set<Entry<Object, Object>> se= p.entrySet(); for(Entry<Object, Object> e:se){ result = result.replace(String.valueOf(e.getKey()), String.valueOf(e.getValue())); } // 输出最终的结果 PrintWriter out = response.getWriter(); out.write(result); //以下两句无作用,CharArrayWriter中为空函数,tomcat自己提供的OutputBuffer执行这两句才有作用 out.flush(); out.close(); }
最后看下keywords.properties文件的内容:
违法词1=*
违法词2=*
建议web.xml中过滤器配置时只映射响应文本数据的请求url,因有的请求是要获取一个文件流,服务器HttpServletResponse响应时文本输出流和字节流是冲突的,正常情况下两者只可取一,不然会报错