View Javadoc

1   // ========================================================================
2   // Copyright (c) 2004-2009 Mort Bay Consulting Pty. Ltd.
3   // ------------------------------------------------------------------------
4   // All rights reserved. This program and the accompanying materials
5   // are made available under the terms of the Eclipse Public License v1.0
6   // and Apache License v2.0 which accompanies this distribution.
7   // The Eclipse Public License is available at 
8   // http://www.eclipse.org/legal/epl-v10.html
9   // The Apache License v2.0 is available at
10  // http://www.opensource.org/licenses/apache2.0.php
11  // You may elect to redistribute this code under either of these licenses. 
12  // ========================================================================
13  
14  package com.acme;
15  
16  import java.io.IOException;
17  import java.util.HashSet;
18  import java.util.Set;
19  
20  import javax.servlet.Filter;
21  import javax.servlet.FilterChain;
22  import javax.servlet.FilterConfig;
23  import javax.servlet.ServletContext;
24  import javax.servlet.ServletException;
25  import javax.servlet.ServletRequest;
26  import javax.servlet.ServletRequestWrapper;
27  import javax.servlet.ServletResponse;
28  import javax.servlet.http.HttpServletRequest;
29  import javax.servlet.http.HttpServletRequestWrapper;
30  import javax.servlet.http.HttpServletResponse;
31  
32  import org.eclipse.jetty.util.log.Log;
33  
34  /* ------------------------------------------------------------ */
35  /** TestFilter.
36   * 
37   * This filter checks for a none local request, and if the init parameter
38   * "remote" is not set to true, then all non local requests are forwarded
39   * to /remote.html
40   * 
41   */
42  public class TestFilter implements Filter
43  {
44      private boolean _remote;
45      private ServletContext _context;
46      private final Set<String> _allowed = new HashSet<String>();
47      
48      /* ------------------------------------------------------------ */
49      /* 
50       * @see javax.servlet.Filter#init(javax.servlet.FilterConfig)
51       */
52      public void init(FilterConfig filterConfig) throws ServletException
53      {
54          _context= filterConfig.getServletContext();
55          _remote=Boolean.parseBoolean(filterConfig.getInitParameter("remote"));
56          _allowed.add("/favicon.ico");
57          _allowed.add("/jetty_banner.gif");
58          
59          Log.debug("TestFilter#remote="+_remote);
60      }
61  
62      /* ------------------------------------------------------------ */
63      /* 
64       * @see javax.servlet.Filter#doFilter(javax.servlet.ServletRequest, javax.servlet.ServletResponse, javax.servlet.FilterChain)
65       */
66      public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
67              throws IOException, ServletException
68      {
69          String from = request.getRemoteHost();
70          String to = request.getServerName();
71          String path=((HttpServletRequest)request).getServletPath();
72  
73          if (!_remote && !_allowed.contains(path) && (
74              !from.equals("localhost") && !from.startsWith("127.") && from.indexOf(":1")<0 ||
75              !to.equals("localhost")&&!to.startsWith("127.0.0.") && to.indexOf(":1")<0))
76          {
77              if ("/".equals(path))
78                  _context.getRequestDispatcher("/remote.html").forward(request,response);
79              else
80                  ((HttpServletResponse)response).sendRedirect("/remote.html");
81              return;
82          }
83          
84          Integer old_value=null;
85          ServletRequest r = request;
86          while (r instanceof ServletRequestWrapper)
87              r=((ServletRequestWrapper)r).getRequest();
88          
89          try
90          {
91              old_value=(Integer)request.getAttribute("testFilter");
92              
93              Integer value=(old_value==null)?new Integer(1):new Integer(old_value.intValue()+1);
94                          
95              request.setAttribute("testFilter", value);
96              
97              String qString = ((HttpServletRequest)request).getQueryString();
98              if (qString != null && qString.indexOf("wrap")>=0)
99              {
100                 request=new HttpServletRequestWrapper((HttpServletRequest)request);
101             }
102             _context.setAttribute("request"+r.hashCode(),value);
103             
104             chain.doFilter(request, response);
105         }
106         finally
107         {
108             request.setAttribute("testFilter", old_value);
109             _context.setAttribute("request"+r.hashCode(),old_value);
110         }
111     }
112 
113     /* ------------------------------------------------------------ */
114     /* 
115      * @see javax.servlet.Filter#destroy()
116      */
117     public void destroy()
118     {
119     }
120 
121 }