001: /*
002: * Copyright (c) 1998-2008 Caucho Technology -- all rights reserved
003: *
004: * This file is part of Resin(R) Open Source
005: *
006: * Each copy or derived work must preserve the copyright notice and this
007: * notice unmodified.
008: *
009: * Resin Open Source is free software; you can redistribute it and/or modify
010: * it under the terms of the GNU General Public License as published by
011: * the Free Software Foundation; either version 2 of the License, or
012: * (at your option) any later version.
013: *
014: * Resin Open Source is distributed in the hope that it will be useful,
015: * but WITHOUT ANY WARRANTY; without even the implied warranty of
016: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
017: * of NON-INFRINGEMENT. See the GNU General Public License for more
018: * details.
019: *
020: * You should have received a copy of the GNU General Public License
021: * along with Resin Open Source; if not, write to the
022: *
023: * Free Software Foundation, Inc.
024: * 59 Temple Place, Suite 330
025: * Boston, MA 02111-1307 USA
026: *
027: * @author Sam
028: */
029:
030: package com.caucho.server.rewrite;
031:
032: import com.caucho.server.connection.*;
033: import com.caucho.filters.*;
034: import com.caucho.server.dispatch.*;
035:
036: import javax.servlet.FilterChain;
037: import javax.servlet.ServletException;
038: import javax.servlet.ServletRequest;
039: import javax.servlet.ServletResponse;
040: import javax.servlet.http.HttpServletRequest;
041: import javax.servlet.http.HttpServletRequestWrapper;
042: import java.io.IOException;
043:
044: public class SetFilterChain extends ContinueMapFilterChain {
045: private final Boolean _requestSecure;
046: private final String _requestCharacterEncoding;
047: private final String _responseContentType;
048: private String _responseCharacterEncoding;
049:
050: public SetFilterChain(String uri, String queryString,
051: FilterChain accept,
052: FilterChainMapper nextFilterChainMapper,
053: String requestCharacterEncoding, Boolean requestSecure,
054: String responseCharacterEncoding, String responseContentType) {
055: super (uri, queryString, accept, nextFilterChainMapper);
056:
057: _requestCharacterEncoding = requestCharacterEncoding;
058: _requestSecure = requestSecure;
059: _responseCharacterEncoding = responseCharacterEncoding;
060: _responseContentType = responseContentType;
061: }
062:
063: @Override
064: public void doFilter(ServletRequest request,
065: ServletResponse response) throws ServletException,
066: IOException {
067: if (_requestCharacterEncoding != null)
068: request.setCharacterEncoding(_requestCharacterEncoding);
069:
070: CauchoRequest oldRequest = null;
071: AbstractHttpResponse cauchoResponse = null;
072: if (_requestSecure != null) {
073: CauchoRequest cauchoRequest = new SecureServletRequestWrapper(
074: (HttpServletRequest) request);
075:
076: if (response instanceof AbstractHttpResponse
077: && cauchoRequest.getWebApp() != null) {
078: cauchoResponse = (AbstractHttpResponse) response;
079:
080: oldRequest = cauchoResponse.getRequest();
081: cauchoResponse.setRequest(cauchoRequest);
082: }
083:
084: request = cauchoRequest;
085:
086: request.setCharacterEncoding(_requestCharacterEncoding);
087: }
088:
089: if (_responseCharacterEncoding != null)
090: response.setCharacterEncoding(_responseCharacterEncoding);
091:
092: if (_responseContentType != null)
093: response.setContentType(_responseContentType);
094:
095: try {
096: super .doFilter(request, response);
097: } finally {
098: if (cauchoResponse != null && request != oldRequest)
099: cauchoResponse.setRequest(oldRequest);
100: }
101: }
102:
103: private class SecureServletRequestWrapper extends RequestAdapter {
104: public SecureServletRequestWrapper(HttpServletRequest request) {
105: setRequest(request);
106:
107: if (request instanceof CauchoRequest)
108: setWebApp(((CauchoRequest) request).getWebApp());
109: }
110:
111: public boolean isSecure() {
112: return _requestSecure;
113: }
114:
115: /**
116: * Returns the request's scheme.
117: */
118: public String getScheme() {
119: return isSecure() ? "https" : "http";
120: }
121: }
122: }
|