netwerk/base/src/nsStreamListenerTee.cpp

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is Mozilla.
  15  *
  16  * The Initial Developer of the Original Code is
  17  * Netscape Communications Corporation.
  18  * Portions created by the Initial Developer are Copyright (C) 2002
  19  * the Initial Developer. All Rights Reserved.
  20  *
  21  * Contributor(s):
  22  *   Darin Fisher <darin@netscape.com> (original author)
  23  *
  24  * Alternatively, the contents of this file may be used under the terms of
  25  * either the GNU General Public License Version 2 or later (the "GPL"), or
  26  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27  * in which case the provisions of the GPL or the LGPL are applicable instead
  28  * of those above. If you wish to allow use of your version of this file only
  29  * under the terms of either the GPL or the LGPL, and not to allow others to
  30  * use your version of this file under the terms of the MPL, indicate your
  31  * decision by deleting the provisions above and replace them with the notice
  32  * and other provisions required by the GPL or the LGPL. If you do not delete
  33  * the provisions above, a recipient may use your version of this file under
  34  * the terms of any one of the MPL, the GPL or the LGPL.
  35  *
  36  * ***** END LICENSE BLOCK ***** */
  37
  38 #include "nsStreamListenerTee.h"
  39
  40 NS_IMPL_ISUPPORTS3(nsStreamListenerTee,
  41                    nsIStreamListener,
  42                    nsIRequestObserver,
  43                    nsIStreamListenerTee)
  44
  45 NS_IMETHODIMP
  46 nsStreamListenerTee::OnStartRequest(nsIRequest *request,
  47                                     nsISupports *context)
  48 {
  49     NS_ENSURE_TRUE(mListener, NS_ERROR_NOT_INITIALIZED);
  50     return mListener->OnStartRequest(request, context);
  51 }
  52
  53 NS_IMETHODIMP
  54 nsStreamListenerTee::OnStopRequest(nsIRequest *request,
  55                                    nsISupports *context,
  56                                    nsresult status)
  57 {
  58     NS_ENSURE_TRUE(mListener, NS_ERROR_NOT_INITIALIZED);
  59     // it is critical that we close out the input stream tee
  60     if (mInputTee) {
  61         mInputTee->SetSink(nsnull);
  62         mInputTee = 0;
  63     }
  64     mSink = 0;
  65     return mListener->OnStopRequest(request, context, status);
  66 }
  67
  68 NS_IMETHODIMP
  69 nsStreamListenerTee::OnDataAvailable(nsIRequest *request,
  70                                      nsISupports *context,
  71                                      nsIInputStream *input,
  72                                      PRUint32 offset,
  73                                      PRUint32 count)
  74 {
  75     NS_ENSURE_TRUE(mListener, NS_ERROR_NOT_INITIALIZED);
  76     NS_ENSURE_TRUE(mSink, NS_ERROR_NOT_INITIALIZED);
  77
  78     nsCOMPtr<nsIInputStream> tee;
  79     nsresult rv;
  80
  81     if (!mInputTee) {
  82         rv = NS_NewInputStreamTee(getter_AddRefs(tee), input, mSink);
  83         if (NS_FAILED(rv)) return rv;
  84
  85         mInputTee = do_QueryInterface(tee, &rv);
  86         if (NS_FAILED(rv)) return rv;
  87     }
  88     else {
  89         // re-initialize the input tee since the input stream may have changed.
  90         rv = mInputTee->SetSource(input);
  91         if (NS_FAILED(rv)) return rv;
  92
  93         tee = do_QueryInterface(mInputTee, &rv);
  94         if (NS_FAILED(rv)) return rv;
  95     }
  96
  97     return mListener->OnDataAvailable(request, context, tee, offset, count);
  98 }
  99
 100 NS_IMETHODIMP
 101 nsStreamListenerTee::Init(nsIStreamListener *listener,
 102                           nsIOutputStream *sink)
 103 {
 104     mListener = listener;
 105     mSink = sink;
 106     return NS_OK;
 107 }