src/jp/sourceforge/rabbitBTS/interceptors/CSRFInterceptor.java

   1 /*
   2    Copyright 2009 senju@users.sourceforge.jp
   3
   4    Licensed under the Apache License, Version 2.0 (the "License");
   5    you may not use this file except in compliance with the License.
   6    You may obtain a copy of the License at
   7
   8        http://www.apache.org/licenses/LICENSE-2.0
   9
  10    Unless required by applicable law or agreed to in writing, software
  11    distributed under the License is distributed on an "AS IS" BASIS,
  12    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13    See the License for the specific language governing permissions and
  14    limitations under the License.
  15  */
  16
  17 package jp.sourceforge.rabbitBTS.interceptors;
  18
  19 import javax.servlet.http.HttpServletRequest;
  20 import javax.servlet.http.HttpServletResponse;
  21
  22 import jp.sourceforge.rabbitBTS.Sht;
  23 import jp.sourceforge.rabbitBTS.controllers.IController;
  24
  25 import org.apache.commons.lang.RandomStringUtils;
  26 import org.apache.commons.lang.StringUtils;
  27 import org.springframework.web.servlet.ModelAndView;
  28 import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
  29
  30 /**
  31  * CSRF対策用インターセプター
  32  */
  33 public class CSRFInterceptor extends HandlerInterceptorAdapter {
  34
  35         /**
  36          * POSTの場合チェック処理
  37          */
  38         @Override
  39         public boolean preHandle(HttpServletRequest request,
  40                         HttpServletResponse response, Object handler) throws Exception {
  41                 if (request.getMethod().equals("POST")
  42                                 && handler instanceof IController) {
  43                         final String sentToken = request.getParameter("secureToken");
  44                         final String sessionToken = (String) request.getSession()
  45                                         .getAttribute("secureToken");
  46                         final IController c = (IController) handler;
  47
  48                         if (StringUtils.equals(sentToken, sessionToken)) {
  49                                 c.setCsrfSafe(true);
  50                         } else {
  51                                 c.setCsrfSafe(false);
  52                                 Sht.log(this).warning("CSRF detected.");
  53                         }
  54
  55                 }
  56                 return true;
  57         }
  58
  59         /**
  60          * セッションとMAVにトークンを格納しておく
  61          */
  62         @Override
  63         public void postHandle(HttpServletRequest request,
  64                         HttpServletResponse response, Object handler,
  65                         ModelAndView modelAndView) throws Exception {
  66                 final String token = RandomStringUtils.randomAlphanumeric(128);
  67
  68                 request.getSession().setAttribute("secureToken", token);
  69                 modelAndView.addObject("secureToken", token);
  70
  71                 if (request.getMethod().equals("POST")
  72                                 && handler instanceof IController) {
  73                         // きちんとCSRFチェックが行われているかチェックする
  74                         final IController c = (IController) handler;
  75                         Sht.log(this).severe("CSRFチェックを行っていないPOST");
  76                         assert c.isCsrfChecked();
  77                 }
  78         }
  79
  80 }