碼上快樂

相關內容    簡體    繁體

HttpServletRequestWrapper的使用

本文轉載自   查看原文   2016-09-24 19:11   3520 

老大給了一個很實際的需求：有段程序，使用Http的方式與合作商交互，而且是明文傳輸數據。我方的代碼已經打包放在服務器上運行了很長時間，這時合作商突然要求修改數據傳輸的方式，要求加密后再傳輸，而我方的原有的代碼不能改變，以防止引發其它問題。 
問：如何在不修改我方現有的代碼的前提下，滿足合作商的要求？ 

可能大家都想到了，只要加上一個過濾器Filter不就可以了嗎？事實就是這樣的，采用Filter+HttpServletRequestWrapper就可以解決這個問題。 
首先：在filter中攔截到加密后的請求，將參數解密，然后組裝成一個新的明文請求串。 
然后：重寫HttpServletRequestWrapper中的getInputStream()方法，讓其返回過濾器解析后的明文串即可。 

具體代碼解釋如下。 

首先我寫了兩個一摸一樣的servlet，一個用來直接接收合作商的明文請求並打印；一個用來接收Filter處理后的合作商的請求並打印（Filter中將合作商加密后的參數解密再傳給這個Servlet）。

Java代碼   收藏代碼
  1. @WebServlet("/SiServlet")  
  2. public class SiServlet extends HttpServlet {  
  3.     private static final long serialVersionUID = 1L;  
  4.   
  5.     /** 
  6.      * @see HttpServlet#HttpServlet() 
  7.      */  
  8.     public SiServlet() {  
  9.         super();  
  10.     }  
  11.   
  12.     /** 
  13.      * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse 
  14.      *      response) 
  15.      */  
  16.     protected void doGet(HttpServletRequest request,  
  17.             HttpServletResponse response) throws ServletException, IOException {  
  18.         this.doPost(request, response);  
  19.     }  
  20.   
  21.     /** 
  22.      * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse 
  23.      *      response) 
  24.      *  
  25.      */  
  26.     protected void doPost(HttpServletRequest request,  
  27.             HttpServletResponse response) throws ServletException, IOException {  
  28.         String bizBindMsg = IOUtils.toString(request.getInputStream(), "UTF-8");  
  29.         bizBindMsg = URLDecoder.decode(bizBindMsg.toString(), "UTF-8");  
  30.         System.out.println("SiServlet接收到請求為: " + bizBindMsg);  
  31.   
  32.         response.getWriter().write("==========success=========");  
  33.     }  
  34. }  
Java代碼   收藏代碼
  1. @WebServlet("/SiServletNormal")  
  2. public class SiServletNormal extends HttpServlet {  
  3.     private static final long serialVersionUID = 1L;  
  4.   
  5.     /** 
  6.      * @see HttpServlet#HttpServlet() 
  7.      */  
  8.     public SiServletNormal() {  
  9.         super();  
  10.     }  
  11.   
  12.     /** 
  13.      * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse 
  14.      *      response) 
  15.      */  
  16.     protected void doGet(HttpServletRequest request,  
  17.             HttpServletResponse response) throws ServletException, IOException {  
  18.         this.doPost(request, response);  
  19.     }  
  20.   
  21.     /** 
  22.      * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse 
  23.      *      response) 
  24.      *  
  25.      */  
  26.     protected void doPost(HttpServletRequest request,  
  27.             HttpServletResponse response) throws ServletException, IOException {  
  28.         String bizBindMsg = IOUtils.toString(request.getInputStream(), "UTF-8");  
  29.         bizBindMsg = URLDecoder.decode(bizBindMsg.toString(), "UTF-8");  
  30.         System.out.println("SiServletNormal接收到請求為: " + bizBindMsg);  
  31.   
  32.         response.getWriter()  
  33.                 .write("==========SiServletNormal Success=========");  
  34.     }  
  35. }  



然后我使用HttpClient模擬了一下合作商發送明文和密文請求的過程，加密使用Base64簡單模擬一下。 

Java代碼   收藏代碼
  1. public class AdcClient {  
  2.     private HttpPost httpPost = null;  
  3.     private HttpClient client = null;  
  4.     private List<NameValuePair> pairs = null;  
  5.   
  6.     public AdcClient() {  
  7.         httpPost = new HttpPost("http://localhost:8080/filtertest/SiServlet");  
  8.         client = new DefaultHttpClient();  
  9.     }  
  10.   
  11.     /** 
  12.      * 發送明文消息 
  13.      *  
  14.      */  
  15.     public void sendMsg() {  
  16.   
  17.         try {  
  18.             httpPost = new HttpPost(  
  19.                     "http://localhost:8080/filtertest/SiServletNormal");  
  20.   
  21.             pairs = new ArrayList<NameValuePair>();  
  22.             pairs.add(new BasicNameValuePair(("param1"), "obama沒加密"));  
  23.             pairs.add(new BasicNameValuePair(("param2"), "男沒加密"));  
  24.             pairs.add(new BasicNameValuePair(("param3"), "漢沒加密"));  
  25.             pairs.add(new BasicNameValuePair(("param4"), "山東沒加密"));  
  26.   
  27.             httpPost.setEntity(new UrlEncodedFormEntity(pairs, "UTF-8"));  
  28.             // httpPost.setHeader("Cookie", "TOKEN=1234567890");  
  29.             HttpResponse response = client.execute(httpPost);  
  30.   
  31.             HttpEntity entity = response.getEntity();  
  32.             BufferedReader br = new BufferedReader(new InputStreamReader(  
  33.                     entity.getContent()));  
  34.             String line = null;  
  35.             StringBuffer result = new StringBuffer();  
  36.             while ((line = br.readLine()) != null) {  
  37.                 result.append(line);  
  38.                 line = br.readLine();  
  39.             }  
  40.   
  41.             System.out.println("來自SiServletNormal的響應為:" + result.toString());  
  42.         } catch (UnsupportedEncodingException e) {  
  43.             e.printStackTrace();  
  44.         } catch (ClientProtocolException e) {  
  45.             e.printStackTrace();  
  46.         } catch (IOException e) {  
  47.             e.printStackTrace();  
  48.         }  
  49.     }  
  50.   
  51.     /** 
  52.      * 發送加密后的消息 
  53.      */  
  54.     public void sendEncryptMsg() {  
  55.         try {  
  56.             pairs = new ArrayList<NameValuePair>();  
  57.             pairs.add(new BasicNameValuePair(("param1"), Base64EnDecrypt  
  58.                     .base64Encode("obama")));  
  59.             pairs.add(new BasicNameValuePair(("param2"), Base64EnDecrypt  
  60.                     .base64Encode("男")));  
  61.             pairs.add(new BasicNameValuePair(("param3"), Base64EnDecrypt  
  62.                     .base64Encode("漢")));  
  63.             pairs.add(new BasicNameValuePair(("param4"), Base64EnDecrypt  
  64.                     .base64Encode("山東")));  
  65.   
  66.             HttpEntity reqEntity = new UrlEncodedFormEntity(pairs, "UTF-8");  
  67.             httpPost.setEntity(reqEntity);  
  68.             // httpPost.setHeader("Cookie", "TOKEN=1234567890");  
  69.             HttpResponse response = client.execute(httpPost);  
  70.   
  71.             /** 
  72.              * 獲取響應信息 
  73.              */  
  74.             HttpEntity entity = response.getEntity();  
  75.             BufferedReader br = new BufferedReader(new InputStreamReader(  
  76.                     entity.getContent()));  
  77.             String line = null;  
  78.             StringBuffer result = new StringBuffer();  
  79.             while ((line = br.readLine()) != null) {  
  80.                 result.append(line);  
  81.                 line = br.readLine();  
  82.             }  
  83.   
  84.             System.out.println("來自SiServlet的響應為:" + result.toString());  
  85.         } catch (UnsupportedEncodingException e) {  
  86.             e.printStackTrace();  
  87.         } catch (ClientProtocolException e) {  
  88.             e.printStackTrace();  
  89.         } catch (IOException e) {  
  90.             e.printStackTrace();  
  91.         }  
  92.     }  
  93.   
  94.     /** 
  95.      * @param args 
  96.      * @throws UnsupportedEncodingException 
  97.      */  
  98.     public static void main(String[] args) throws UnsupportedEncodingException {  
  99.         new AdcClient().sendMsg();  
  100.   
  101.         new AdcClient().sendEncryptMsg();  
  102.     }  
  103. }  


重點是下面的這個HttpServletRequestWrapper，我重寫了它的getInputStream()方法，這個方法返回包含明文的ServletInputStream

Java代碼   收藏代碼
  1. public class MyRequestWrapper extends HttpServletRequestWrapper {  
  2.     private HttpServletRequest request;  
  3.   
  4.     public MyRequestWrapper(HttpServletRequest request) {  
  5.         super(request);  
  6.         this.request = request;  
  7.     }  
  8.   
  9.     /** 
  10.      * 先解密，獲取明文；然后將明文轉化為字節數組；然后再去讀取字節數組中的內容 
  11.      */  
  12.     @Override  
  13.     public ServletInputStream getInputStream() {  
  14.         String bizBindMsg = null;  
  15.         ServletInputStream stream = null;  
  16.   
  17.         try {  
  18.             stream = request.getInputStream();  
  19.             bizBindMsg = IOUtils.toString(stream, "UTF-8");  
  20.         } catch (IOException e) {  
  21.             e.printStackTrace();  
  22.         }  
  23.         try {  
  24.             bizBindMsg = URLDecoder.decode(bizBindMsg.toString(), "UTF-8");  
  25.         } catch (UnsupportedEncodingException e) {  
  26.             e.printStackTrace();  
  27.         }  
  28.         System.out.println("MyRequestWrapper接收到的請求為: " + bizBindMsg);  
  29.   
  30.         /** 
  31.          * 獲取加密的值進行解密 
  32.          */  
  33.         final StringBuffer reqStr = new StringBuffer();  
  34.         reqStr.append("param1=").append(  
  35.                 Base64EnDecrypt.base64Decode(bizBindMsg.substring(  
  36.                         bizBindMsg.indexOf("param1=") + 7,  
  37.                         bizBindMsg.indexOf("param2="))));  
  38.         reqStr.append("&");  
  39.         reqStr.append("param2=").append(  
  40.                 Base64EnDecrypt.base64Decode(bizBindMsg.substring(  
  41.                         bizBindMsg.indexOf("param2=") + 7,  
  42.                         bizBindMsg.indexOf("param3="))));  
  43.         reqStr.append("&");  
  44.         reqStr.append("param3=").append(  
  45.                 Base64EnDecrypt.base64Decode(bizBindMsg.substring(  
  46.                         bizBindMsg.indexOf("param3=") + 7,  
  47.                         bizBindMsg.indexOf("param4="))));  
  48.         reqStr.append("&");  
  49.         reqStr.append("param4=").append(  
  50.                 Base64EnDecrypt.base64Decode(bizBindMsg.substring(bizBindMsg  
  51.                         .indexOf("param4=") + 7)));  
  52.   
  53.         System.out.println("********MyRequestWrapper接收到的解密后的請求為*********");  
  54.         System.out.println(reqStr.toString());  
  55.   
  56.         /** 
  57.          * 將解密后的明文串放到buffer數組中 
  58.          */  
  59.         byte[] buffer = null;  
  60.         try {  
  61.             buffer = reqStr.toString().getBytes("UTF-8");  
  62.         } catch (UnsupportedEncodingException e) {  
  63.             e.printStackTrace();  
  64.         }  
  65.         final ByteArrayInputStream bais = new ByteArrayInputStream(buffer);  
  66.   
  67.         ServletInputStream newStream = new ServletInputStream() {  
  68.   
  69.             @Override  
  70.             public int read() throws IOException {  
  71.                 return bais.read();  
  72.             }  
  73.         };  
  74.         return newStream;  
  75.     }  
  76. }  


最后是簡單的Filter，在這里將加密后的ServletRequest重新包裝，交給SiServlet進行處理

Java代碼   收藏代碼
  1. public class EncryptFilter implements Filter {  
  2.   
  3.     @Override  
  4.     public void destroy() {  
  5.     }  
  6.   
  7.     @Override  
  8.     public void doFilter(ServletRequest request, ServletResponse response,  
  9.             FilterChain chain) throws IOException, ServletException {  
  10.   
  11.         chain.doFilter(new MyRequestWrapper((HttpServletRequest) request),  
  12.                 response);  
  13.     }  
  14.   
  15.     @Override  
  16.     public void init(FilterConfig arg0) throws ServletException {  
  17.   
  18.     }  
  19.   
  20. }  


我的web.xml中是這樣配置的 

Java代碼   收藏代碼
  1. <filter>  
  2.     <filter-name>encryptFilter</filter-name>  
  3.     <filter-class>com.test.filter.EncryptFilter</filter-class>  
  4.   </filter>  
  5.   <filter-mapping>  
  6.     <filter-name>encryptFilter</filter-name>  
  7.     <url-pattern>/SiServlet</url-pattern>  
  8.   </filter-mapping>  


確保過濾器entyptFilter只攔截到SiServlet的請求即可。 

運行AdcClient，可以看到下面的結果 


這里的重點是MyRequestWrapper中重寫的getInputStream()方法。大家可以看看API中關於HttpServletRequest的用法http://tomcat.apache.org/tomcat-5.5-doc/servletapi/index.html。

 

http://lichuanbao.iteye.com/blog/1102138

 


免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。



猜您在找 HttpServletRequestWrapper類的使用 使用HttpServletRequestWrapper在filter修改request參數 使用HttpServletRequestWrapper重寫Request請求參數 使用HttpServletRequestWrapper重寫Request請求參數 HttpServletRequestWrapper使用技巧（自定義session和緩存InputStream） 使用HttpServletRequestWrapper修改請求參數 和 使用HttpServletResponseWrapper截獲響應數據 JavaWeb：HttpServletRequestWrapper HttpServletRequestWrapper 用法（轉） HttpServletRequestWrapper 類&過濾指定文字 【轉】HttpServletRequestWrapper 實現xss注入
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM