| /* |
| * Copyright 1999,2004 The Apache Software Foundation. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| |
| package org.apache.catalina.connector; |
| |
| import java.io.IOException; |
| import java.io.PrintWriter; |
| import java.security.AccessController; |
| import java.security.PrivilegedAction; |
| import java.security.PrivilegedActionException; |
| import java.security.PrivilegedExceptionAction; |
| import java.util.Locale; |
| |
| import javax.servlet.ServletOutputStream; |
| import javax.servlet.http.Cookie; |
| import javax.servlet.http.HttpServletResponse; |
| |
| |
| /** |
| * Facade class that wraps a Coyote response object. |
| * All methods are delegated to the wrapped response. |
| * |
| * @author Remy Maucherat |
| * @author Jean-Francois Arcand |
| * @version $Revision$ $Date$ |
| */ |
| |
| |
| public class ResponseFacade |
| implements HttpServletResponse { |
| |
| |
| // ----------------------------------------------------------- DoPrivileged |
| |
| private final class SetContentTypePrivilegedAction implements PrivilegedAction{ |
| private String contentType; |
| public SetContentTypePrivilegedAction(String contentType){ |
| this.contentType = contentType; |
| } |
| |
| public Object run() { |
| response.setContentType(contentType); |
| return null; |
| } |
| } |
| |
| private final class DateHeaderPrivilegedAction implements PrivilegedAction { |
| private String name; |
| private long value; |
| private boolean add; |
| |
| DateHeaderPrivilegedAction(String name, long value, boolean add) { |
| this.name = name; |
| this.value = value; |
| this.add = add; |
| } |
| |
| public Object run() { |
| if(add) { |
| response.addDateHeader(name, value); |
| } else { |
| response.setDateHeader(name, value); |
| } |
| return null; |
| } |
| } |
| |
| // ----------------------------------------------------------- Constructors |
| |
| |
| /** |
| * Construct a wrapper for the specified response. |
| * |
| * @param response The response to be wrapped |
| */ |
| public ResponseFacade(Response response) { |
| |
| this.response = response; |
| |
| } |
| |
| |
| // ----------------------------------------------------- Instance Variables |
| |
| |
| /** |
| * The wrapped response. |
| */ |
| protected Response response = null; |
| |
| |
| // --------------------------------------------------------- Public Methods |
| |
| |
| /** |
| * Clear facade. |
| */ |
| public void clear() { |
| response = null; |
| } |
| |
| |
| public void finish() { |
| |
| response.setSuspended(true); |
| |
| } |
| |
| |
| public boolean isFinished() { |
| |
| return response.isSuspended(); |
| |
| } |
| |
| |
| // ------------------------------------------------ ServletResponse Methods |
| |
| |
| public String getCharacterEncoding() { |
| return response.getCharacterEncoding(); |
| } |
| |
| |
| public ServletOutputStream getOutputStream() |
| throws IOException { |
| |
| // if (isFinished()) |
| // throw new IllegalStateException |
| // (/*sm.getString("responseFacade.finished")*/); |
| |
| ServletOutputStream sos = response.getOutputStream(); |
| if (isFinished()) |
| response.setSuspended(true); |
| return (sos); |
| |
| } |
| |
| |
| public PrintWriter getWriter() |
| throws IOException { |
| |
| // if (isFinished()) |
| // throw new IllegalStateException |
| // (/*sm.getString("responseFacade.finished")*/); |
| |
| PrintWriter writer = response.getWriter(); |
| if (isFinished()) |
| response.setSuspended(true); |
| return (writer); |
| |
| } |
| |
| |
| public void setContentLength(int len) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.setContentLength(len); |
| |
| } |
| |
| |
| public void setContentType(String type) { |
| |
| if (isCommitted()) |
| return; |
| |
| if (System.getSecurityManager() != null){ |
| AccessController.doPrivileged(new SetContentTypePrivilegedAction(type)); |
| } else { |
| response.setContentType(type); |
| } |
| } |
| |
| |
| public void setBufferSize(int size) { |
| |
| if (isCommitted()) |
| throw new IllegalStateException |
| (/*sm.getString("responseBase.reset.ise")*/); |
| |
| response.setBufferSize(size); |
| |
| } |
| |
| |
| public int getBufferSize() { |
| return response.getBufferSize(); |
| } |
| |
| |
| public void flushBuffer() |
| throws IOException { |
| |
| if (isFinished()) |
| // throw new IllegalStateException |
| // (/*sm.getString("responseFacade.finished")*/); |
| return; |
| |
| if (System.getSecurityManager() != null){ |
| try{ |
| AccessController.doPrivileged(new PrivilegedExceptionAction(){ |
| |
| public Object run() throws IOException{ |
| response.setAppCommitted(true); |
| |
| response.flushBuffer(); |
| return null; |
| } |
| }); |
| } catch(PrivilegedActionException e){ |
| Exception ex = e.getException(); |
| if (ex instanceof IOException){ |
| throw (IOException)ex; |
| } |
| } |
| } else { |
| response.setAppCommitted(true); |
| |
| response.flushBuffer(); |
| } |
| |
| } |
| |
| |
| public void resetBuffer() { |
| |
| if (isCommitted()) |
| throw new IllegalStateException |
| (/*sm.getString("responseBase.reset.ise")*/); |
| |
| response.resetBuffer(); |
| |
| } |
| |
| |
| public boolean isCommitted() { |
| return (response.isAppCommitted()); |
| } |
| |
| |
| public void reset() { |
| |
| if (isCommitted()) |
| throw new IllegalStateException |
| (/*sm.getString("responseBase.reset.ise")*/); |
| |
| response.reset(); |
| |
| } |
| |
| |
| public void setLocale(Locale loc) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.setLocale(loc); |
| } |
| |
| |
| public Locale getLocale() { |
| return response.getLocale(); |
| } |
| |
| |
| public void addCookie(Cookie cookie) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.addCookie(cookie); |
| |
| } |
| |
| |
| public boolean containsHeader(String name) { |
| return response.containsHeader(name); |
| } |
| |
| |
| public String encodeURL(String url) { |
| return response.encodeURL(url); |
| } |
| |
| |
| public String encodeRedirectURL(String url) { |
| return response.encodeRedirectURL(url); |
| } |
| |
| |
| public String encodeUrl(String url) { |
| return response.encodeURL(url); |
| } |
| |
| |
| public String encodeRedirectUrl(String url) { |
| return response.encodeRedirectURL(url); |
| } |
| |
| |
| public void sendError(int sc, String msg) |
| throws IOException { |
| |
| if (isCommitted()) |
| throw new IllegalStateException |
| (/*sm.getString("responseBase.reset.ise")*/); |
| |
| response.setAppCommitted(true); |
| |
| response.sendError(sc, msg); |
| |
| } |
| |
| |
| public void sendError(int sc) |
| throws IOException { |
| |
| if (isCommitted()) |
| throw new IllegalStateException |
| (/*sm.getString("responseBase.reset.ise")*/); |
| |
| response.setAppCommitted(true); |
| |
| response.sendError(sc); |
| |
| } |
| |
| |
| public void sendRedirect(String location) |
| throws IOException { |
| |
| if (isCommitted()) |
| throw new IllegalStateException |
| (/*sm.getString("responseBase.reset.ise")*/); |
| |
| response.setAppCommitted(true); |
| |
| response.sendRedirect(location); |
| |
| } |
| |
| |
| public void setDateHeader(String name, long date) { |
| |
| if (isCommitted()) |
| return; |
| |
| if(System.getSecurityManager() != null) { |
| AccessController.doPrivileged(new DateHeaderPrivilegedAction |
| (name, date, false)); |
| } else { |
| response.setDateHeader(name, date); |
| } |
| |
| } |
| |
| |
| public void addDateHeader(String name, long date) { |
| |
| if (isCommitted()) |
| return; |
| |
| if(System.getSecurityManager() != null) { |
| AccessController.doPrivileged(new DateHeaderPrivilegedAction |
| (name, date, true)); |
| } else { |
| response.addDateHeader(name, date); |
| } |
| |
| } |
| |
| |
| public void setHeader(String name, String value) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.setHeader(name, value); |
| |
| } |
| |
| |
| public void addHeader(String name, String value) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.addHeader(name, value); |
| |
| } |
| |
| |
| public void setIntHeader(String name, int value) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.setIntHeader(name, value); |
| |
| } |
| |
| |
| public void addIntHeader(String name, int value) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.addIntHeader(name, value); |
| |
| } |
| |
| |
| public void setStatus(int sc) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.setStatus(sc); |
| |
| } |
| |
| |
| public void setStatus(int sc, String sm) { |
| |
| if (isCommitted()) |
| return; |
| |
| response.setStatus(sc, sm); |
| |
| } |
| |
| |
| public String getContentType() { |
| return response.getContentType(); |
| } |
| |
| |
| public void setCharacterEncoding(String arg0) { |
| response.setCharacterEncoding(arg0); |
| } |
| |
| |
| } |