content/media/video/src/nsChannelToPipeListener.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent: */
   3 /* ***** BEGIN LICENSE BLOCK *****
   4  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version
   7  * 1.1 (the "License"); you may not use this file except in compliance with
   8  * the License. You may obtain a copy of the License at
   9  * http://www.mozilla.org/MPL/
  10  *
  11  * Software distributed under the License is distributed on an "AS IS" basis,
  12  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13  * for the specific language governing rights and limitations under the
  14  * License.
  15  *
  16  * The Original Code is Mozilla code.
  17  *
  18  * The Initial Developer of the Original Code is the Mozilla Corporation.
  19  * Portions created by the Initial Developer are Copyright (C) 2007
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *  Chris Double <chris.double@double.co.nz>
  24  *
  25  * Alternatively, the contents of this file may be used under the terms of
  26  * either the GNU General Public License Version 2 or later (the "GPL"), or
  27  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  28  * in which case the provisions of the GPL or the LGPL are applicable instead
  29  * of those above. If you wish to allow use of your version of this file only
  30  * under the terms of either the GPL or the LGPL, and not to allow others to
  31  * use your version of this file under the terms of the MPL, indicate your
  32  * decision by deleting the provisions above and replace them with the notice
  33  * and other provisions required by the GPL or the LGPL. If you do not delete
  34  * the provisions above, a recipient may use your version of this file under
  35  * the terms of any one of the MPL, the GPL or the LGPL.
  36  *
  37  * ***** END LICENSE BLOCK ***** */
  38 #include "nsAString.h"
  39 #include "nsNetUtil.h"
  40 #include "nsMediaDecoder.h"
  41 #include "nsIScriptSecurityManager.h"
  42 #include "nsChannelToPipeListener.h"
  43 #include "nsICachingChannel.h"
  44
  45 nsChannelToPipeListener::nsChannelToPipeListener(nsMediaDecoder* aDecoder) :
  46   mDecoder(aDecoder),
  47   mIntervalStart(0),
  48   mIntervalEnd(0),
  49   mTotalBytes(0)
  50 {
  51 }
  52
  53 nsresult nsChannelToPipeListener::Init()
  54 {
  55   nsresult rv = NS_NewPipe(getter_AddRefs(mInput),
  56                            getter_AddRefs(mOutput),
  57                            0,
  58                            PR_UINT32_MAX);
  59   NS_ENSURE_SUCCESS(rv, rv);
  60
  61   return NS_OK;
  62 }
  63
  64 void nsChannelToPipeListener::Stop()
  65 {
  66   mDecoder = nsnull;
  67   mInput = nsnull;
  68   mOutput = nsnull;
  69 }
  70
  71 void nsChannelToPipeListener::Cancel()
  72 {
  73   if (mOutput)
  74     mOutput->Close();
  75
  76   if (mInput)
  77     mInput->Close();
  78 }
  79
  80 double nsChannelToPipeListener::BytesPerSecond() const
  81 {
  82   return mOutput ? mTotalBytes / ((PR_IntervalToMilliseconds(mIntervalEnd-mIntervalStart)) / 1000.0) : NS_MEDIA_UNKNOWN_RATE;
  83 }
  84
  85 nsresult nsChannelToPipeListener::GetInputStream(nsIInputStream** aStream)
  86 {
  87   NS_IF_ADDREF(*aStream = mInput);
  88   return NS_OK;
  89 }
  90
  91 nsresult nsChannelToPipeListener::OnStartRequest(nsIRequest* aRequest, nsISupports* aContext)
  92 {
  93   mIntervalStart = PR_IntervalNow();
  94   mIntervalEnd = mIntervalStart;
  95   mTotalBytes = 0;
  96   mDecoder->UpdateBytesDownloaded(mTotalBytes);
  97   nsCOMPtr<nsIHttpChannel> hc = do_QueryInterface(aRequest);
  98   if (hc) {
  99     PRUint32 responseStatus = 0;
 100     hc->GetResponseStatus(&responseStatus);
 101     if (responseStatus == 200) {
 102       PRInt32 cl = 0;
 103       hc->GetContentLength(&cl);
 104       mDecoder->SetTotalBytes(cl);
 105     }
 106   }
 107
 108   nsCOMPtr<nsICachingChannel> cc = do_QueryInterface(aRequest);
 109   if (cc) {
 110     PRBool fromCache = PR_FALSE;
 111     nsresult rv = cc->IsFromCache(&fromCache);
 112
 113     if (NS_SUCCEEDED(rv) && !fromCache) {
 114       cc->SetCacheAsFile(PR_TRUE);
 115     }
 116   }
 117
 118   /* Get our principal */
 119   nsCOMPtr<nsIChannel> chan(do_QueryInterface(aRequest));
 120   if (chan) {
 121     nsCOMPtr<nsIScriptSecurityManager> secMan =
 122       do_GetService("@mozilla.org/scriptsecuritymanager;1");
 123     if (secMan) {
 124       nsresult rv = secMan->GetChannelPrincipal(chan,
 125                                                 getter_AddRefs(mPrincipal));
 126       if (NS_FAILED(rv)) {
 127         return rv;
 128       }
 129     }
 130   }
 131
 132   return NS_OK;
 133 }
 134
 135 nsresult nsChannelToPipeListener::OnStopRequest(nsIRequest* aRequest, nsISupports* aContext, nsresult aStatus)
 136 {
 137   mOutput = nsnull;
 138   if (mDecoder) {
 139     mDecoder->ResourceLoaded();
 140   }
 141   return NS_OK;
 142 }
 143
 144 nsresult nsChannelToPipeListener::OnDataAvailable(nsIRequest* aRequest,
 145                                                 nsISupports* aContext,
 146                                                 nsIInputStream* aStream,
 147                                                 PRUint32 aOffset,
 148                                                 PRUint32 aCount)
 149 {
 150   if (!mOutput)
 151     return NS_ERROR_FAILURE;
 152
 153   PRUint32 bytes = 0;
 154
 155   do {
 156     nsresult rv = mOutput->WriteFrom(aStream, aCount, &bytes);
 157     NS_ENSURE_SUCCESS(rv, rv);
 158
 159     aCount -= bytes;
 160     mTotalBytes += bytes;
 161     mDecoder->UpdateBytesDownloaded(mTotalBytes);
 162   } while (aCount) ;
 163
 164   nsresult rv = mOutput->Flush();
 165   NS_ENSURE_SUCCESS(rv, rv);
 166   mIntervalEnd = PR_IntervalNow();
 167   return NS_OK;
 168 }
 169
 170 nsIPrincipal*
 171 nsChannelToPipeListener::GetCurrentPrincipal()
 172 {
 173   return mPrincipal;
 174 }
 175
 176 NS_IMPL_THREADSAFE_ISUPPORTS2(nsChannelToPipeListener, nsIRequestObserver, nsIStreamListener)
 177