org.deegree.securityproxy.request.HttpServletRequestBodyWrapper.java Source code

Introduction

Here is the source code for org.deegree.securityproxy.request.HttpServletRequestBodyWrapper.java

Source

//$HeadURL$
/*----------------------------------------------------------------------------
 This file is part of deegree, http://deegree.org/
 Copyright (C) 2001-2014 by:
 - Department of Geography, University of Bonn -
 and
 - lat/lon GmbH -
    
 This library is free software; you can redistribute it and/or modify it under
 the terms of the GNU Lesser General Public License as published by the Free
 Software Foundation; either version 2.1 of the License, or (at your option)
 any later version.
 This library is distributed in the hope that it will be useful, but WITHOUT
 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 details.
 You should have received a copy of the GNU Lesser General Public License
 along with this library; if not, write to the Free Software Foundation, Inc.,
 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
    
 Contact information:
    
 lat/lon GmbH
 Aennchenstr. 19, 53177 Bonn
 Germany
 http://lat-lon.de/
    
 Department of Geography, University of Bonn
 Prof. Dr. Klaus Greve
 Postfach 1147, 53001 Bonn
 Germany
 http://www.geographie.uni-bonn.de/deegree/
    
 e-mail: info@deegree.org
 ----------------------------------------------------------------------------*/
package org.deegree.securityproxy.request;

import static java.util.Collections.enumeration;
import static java.util.Collections.singleton;
import static org.apache.commons.io.IOUtils.toByteArray;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Enumeration;

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;

/**
 * Wraps a {@link HttpServletRequest} and allows modifications of the body.
 * 
 * @author <a href="mailto:goltz@lat-lon.de">Lyn Goltz</a>
 * @author last edited by: $Author: lyn $
 * 
 * @version $Revision: $, $Date: $
 */
public class HttpServletRequestBodyWrapper extends HttpServletRequestWrapper {

    private InputStream wrappedInputStream;

    /**
     * 
     * @param request
     *            wrapped request, never <code>null</code>
     * @throws IOException
     *             if the input stream of the wrapped request could not be read
     */
    public HttpServletRequestBodyWrapper(HttpServletRequest request) throws IOException {
        super(request);
        this.wrappedInputStream = request.getInputStream();
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        return new ServletInputStream() {
            @Override
            public int read() throws IOException {
                return wrappedInputStream.read();
            }
        };
    }

    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(wrappedInputStream));
    }

    @Override
    public int getContentLength() {
        try {
            synchronized (wrappedInputStream) {
                byte[] byteArray = toByteArray(wrappedInputStream);
                renewInputStream(new ByteArrayInputStream(byteArray));
                return byteArray.length;
            }
        } catch (IOException e) {
            return -1;
        }
    }

    @Override
    public String getHeader(String name) {
        if ("Content-Length".equalsIgnoreCase(name))
            return getContentLengthAsString();
        return super.getHeader(name);
    }

    @Override
    public Enumeration<?> getHeaders(String name) {
        if ("Content-Length".equalsIgnoreCase(name))
            return enumeration(singleton(getContentLengthAsString()));
        return super.getHeaders(name);
    }

    /**
     * Renews the wrapped input stream
     * 
     * @param newInputStream
     *            to wrap, never <code>null</code>
     */
    public void renewInputStream(InputStream newInputStream) {
        this.wrappedInputStream = newInputStream;
    }

    private String getContentLengthAsString() {
        return Integer.toString(getContentLength());
    }

}