/**
    * Logback: the reliable, generic, fast and flexible logging framework.
    * Copyright (C) 1999-2011, QOS.ch. All rights reserved.
    *
    * This program and the accompanying materials are dual-licensed under
    * either the terms of the Eclipse Public License v1.0 as published by
    * the Eclipse Foundation
    *
    *   or (per the licensee's choosing)
   *
   * under the terms of the GNU Lesser General Public License version 2.1
   * as published by the Free Software Foundation.
   */
  package ch.qos.logback.access.servlet;
  
  import java.io.ByteArrayInputStream;
  import java.io.ByteArrayOutputStream;
  import java.io.IOException;
  import java.io.InputStream;
  
  import javax.servlet.ServletInputStream;
  import javax.servlet.http.HttpServletRequest;
  
  class TeeServletInputStream extends ServletInputStream {
  
    InputStream in;
    byte[] inputBuffer;
  
    TeeServletInputStream(HttpServletRequest request) {
      duplicateInputStream(request);
    }
  
    @Override
    public int read() throws IOException {
      return in.read();
    }
  
    private void duplicateInputStream(HttpServletRequest request) {
      ServletInputStream originalSIS = null;
      try {
        originalSIS = request.getInputStream();
        inputBuffer = consumeBufferAndReturnAsByteArray(originalSIS);
        this.in = new ByteArrayInputStream(inputBuffer);
      } catch (IOException e) {
        e.printStackTrace();
      } finally {
        closeStrean(originalSIS);
      }
    }
  
    byte[] consumeBufferAndReturnAsByteArray(InputStream is) throws IOException {
      int len = 1024;
      byte[] temp = new byte[len];
      int c = -1;
      ByteArrayOutputStream baos = new ByteArrayOutputStream();
      while ((c = is.read(temp, 0, len)) != -1) {
        baos.write(temp, 0, c);
      }
      return baos.toByteArray();
    }
  
  
    void closeStrean(ServletInputStream is) {
      if (is != null) {
        try {
          is.close();
        } catch (IOException e) {
        }
      }
    }
  
    byte[] getInputBuffer() {
      return inputBuffer;
    }
  }