The Tor Browser: content/base/src/nsSyncLoadService.cpp@6474c204b198 (annotated)

content/base/src/nsSyncLoadService.cpp@6474c204b198 (annotated)

content/base/src/nsSyncLoadService.cpp

Wed, 31 Dec 2014 06:09:35 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Wed, 31 Dec 2014 06:09:35 +0100
changeset 0: 6474c204b198
permissions: -rw-r--r--

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
 /* This Source Code Form is subject to the terms of the Mozilla Public
  * License, v. 2.0. If a copy of the MPL was not distributed with this
  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 /*
  * A service that provides methods for synchronously loading a DOM in various ways.
  */
 #include "nsSyncLoadService.h"
 #include "nsCOMPtr.h"
 #include "nsIChannel.h"
 #include "nsIChannelEventSink.h"
 #include "nsIAsyncVerifyRedirectCallback.h"
 #include "nsIInterfaceRequestor.h"
 #include "nsString.h"
 #include "nsWeakReference.h"
 #include "nsIDocument.h"
 #include "nsIDOMDocument.h"
 #include "nsIPrincipal.h"
 #include "nsContentUtils.h" // for kLoadAsData
 #include "nsThreadUtils.h"
 #include "nsNetUtil.h"
 #include "nsAutoPtr.h"
 #include "nsStreamUtils.h"
 #include "nsCrossSiteListenerProxy.h"
 #include <algorithm>
 /**
  * This class manages loading a single XML document
  */
 class nsSyncLoader : public nsIStreamListener,
                      public nsIChannelEventSink,
                      public nsIInterfaceRequestor,
                      public nsSupportsWeakReference
 {
 public:
     nsSyncLoader() : mLoading(false) {}
     virtual ~nsSyncLoader();
     NS_DECL_ISUPPORTS
     nsresult LoadDocument(nsIChannel* aChannel, nsIPrincipal *aLoaderPrincipal,
                           bool aChannelIsSync, bool aForceToXML,
                           nsIDOMDocument** aResult);
     NS_FORWARD_NSISTREAMLISTENER(mListener->)
     NS_DECL_NSIREQUESTOBSERVER
     NS_DECL_NSICHANNELEVENTSINK
     NS_DECL_NSIINTERFACEREQUESTOR
 private:
     nsresult PushAsyncStream(nsIStreamListener* aListener);
     nsresult PushSyncStream(nsIStreamListener* aListener);
     nsCOMPtr<nsIChannel> mChannel;
     nsCOMPtr<nsIStreamListener> mListener;
     bool mLoading;
     nsresult mAsyncLoadStatus;
 };
 class nsForceXMLListener : public nsIStreamListener
 {
 public:
     nsForceXMLListener(nsIStreamListener* aListener);
     virtual ~nsForceXMLListener();
     NS_DECL_ISUPPORTS
     NS_FORWARD_NSISTREAMLISTENER(mListener->)
     NS_DECL_NSIREQUESTOBSERVER
 private:
     nsCOMPtr<nsIStreamListener> mListener;
 };
 nsForceXMLListener::nsForceXMLListener(nsIStreamListener* aListener)
     : mListener(aListener)
 {
 }
 nsForceXMLListener::~nsForceXMLListener()
 {
 }
 NS_IMPL_ISUPPORTS(nsForceXMLListener,
                   nsIStreamListener,
                   nsIRequestObserver)
 NS_IMETHODIMP
 nsForceXMLListener::OnStartRequest(nsIRequest *aRequest, nsISupports *aContext)
 {
     nsresult status;
     aRequest->GetStatus(&status);
     nsCOMPtr<nsIChannel> channel = do_QueryInterface(aRequest);
     if (channel && NS_SUCCEEDED(status)) {
       channel->SetContentType(NS_LITERAL_CSTRING("text/xml"));
     }
     return mListener->OnStartRequest(aRequest, aContext);
 }
 NS_IMETHODIMP
 nsForceXMLListener::OnStopRequest(nsIRequest *aRequest, nsISupports *aContext,
                                   nsresult aStatusCode)
 {
     return mListener->OnStopRequest(aRequest, aContext, aStatusCode);
 }
 nsSyncLoader::~nsSyncLoader()
 {
     if (mLoading && mChannel) {
         mChannel->Cancel(NS_BINDING_ABORTED);
     }
 }
 NS_IMPL_ISUPPORTS(nsSyncLoader,
                   nsIStreamListener,
                   nsIRequestObserver,
                   nsIChannelEventSink,
                   nsIInterfaceRequestor,
                   nsISupportsWeakReference)
 nsresult
 nsSyncLoader::LoadDocument(nsIChannel* aChannel,
                            nsIPrincipal *aLoaderPrincipal,
                            bool aChannelIsSync,
                            bool aForceToXML,
                            nsIDOMDocument **aResult)
 {
     NS_ENSURE_ARG_POINTER(aResult);
     *aResult = nullptr;
     nsresult rv = NS_OK;
     nsCOMPtr<nsIURI> loaderUri;
     if (aLoaderPrincipal) {
         aLoaderPrincipal->GetURI(getter_AddRefs(loaderUri));
     }
     mChannel = aChannel;
     nsCOMPtr<nsIHttpChannel> http = do_QueryInterface(mChannel);
     if (http) {
         http->SetRequestHeader(NS_LITERAL_CSTRING("Accept"),
                                NS_LITERAL_CSTRING("text/xml,application/xml,application/xhtml+xml,*/*;q=0.1"),
                                false);
         if (loaderUri) {
             http->SetReferrer(loaderUri);
         }
     }
     // Hook us up to listen to redirects and the like.
     // Do this before setting up the cross-site proxy since
     // that installs its own proxies.
     mChannel->SetNotificationCallbacks(this);
     // Get the loadgroup of the channel
     nsCOMPtr<nsILoadGroup> loadGroup;
     rv = aChannel->GetLoadGroup(getter_AddRefs(loadGroup));
     NS_ENSURE_SUCCESS(rv, rv);
     // Create document
     nsCOMPtr<nsIDocument> document;
     rv = NS_NewXMLDocument(getter_AddRefs(document));
     NS_ENSURE_SUCCESS(rv, rv);
     // Start the document load. Do this before we attach the load listener
     // since we reset the document which drops all observers.
     nsCOMPtr<nsIStreamListener> listener;
     rv = document->StartDocumentLoad(kLoadAsData, mChannel,
                                      loadGroup, nullptr,
                                      getter_AddRefs(listener),
                                      true);
     NS_ENSURE_SUCCESS(rv, rv);
     if (aForceToXML) {
         nsCOMPtr<nsIStreamListener> forceListener =
             new nsForceXMLListener(listener);
         listener.swap(forceListener);
     }
     if (aLoaderPrincipal) {
         nsRefPtr<nsCORSListenerProxy> corsListener =
           new nsCORSListenerProxy(listener, aLoaderPrincipal, false);
         rv = corsListener->Init(mChannel);
         NS_ENSURE_SUCCESS(rv, rv);
         listener = corsListener;
     }
     if (aChannelIsSync) {
         rv = PushSyncStream(listener);
     }
     else {
         rv = PushAsyncStream(listener);
     }
     http = do_QueryInterface(mChannel);
     if (NS_SUCCEEDED(rv) && http) {
         bool succeeded;
         if (NS_FAILED(http->GetRequestSucceeded(&succeeded)) || !succeeded) {
             rv = NS_ERROR_FAILURE;
         }
     }
     mChannel = nullptr;
     // check that the load succeeded
     NS_ENSURE_SUCCESS(rv, rv);
     NS_ENSURE_TRUE(document->GetRootElement(), NS_ERROR_FAILURE);
     return CallQueryInterface(document, aResult);
 }
 nsresult
 nsSyncLoader::PushAsyncStream(nsIStreamListener* aListener)
 {
     mListener = aListener;
     mAsyncLoadStatus = NS_OK;
     // Start reading from the channel
     nsresult rv = mChannel->AsyncOpen(this, nullptr);
     if (NS_SUCCEEDED(rv)) {
         // process events until we're finished.
         mLoading = true;
         nsIThread *thread = NS_GetCurrentThread();
         while (mLoading && NS_SUCCEEDED(rv)) {
             bool processedEvent;
             rv = thread->ProcessNextEvent(true, &processedEvent);
             if (NS_SUCCEEDED(rv) && !processedEvent)
                 rv = NS_ERROR_UNEXPECTED;
         }
     }
     mListener = nullptr;
     NS_ENSURE_SUCCESS(rv, rv);
     // Note that if AsyncOpen failed that's ok -- the only caller of
     // this method nulls out mChannel immediately after we return.
     return mAsyncLoadStatus;
 }
 nsresult
 nsSyncLoader::PushSyncStream(nsIStreamListener* aListener)
 {
     nsCOMPtr<nsIInputStream> in;
     nsresult rv = mChannel->Open(getter_AddRefs(in));
     NS_ENSURE_SUCCESS(rv, rv);
     mLoading = true;
     rv = nsSyncLoadService::PushSyncStreamToListener(in, aListener, mChannel);
     mLoading = false;
     return rv;
 }
 NS_IMETHODIMP
 nsSyncLoader::OnStartRequest(nsIRequest *aRequest, nsISupports *aContext)
 {
     return mListener->OnStartRequest(aRequest, aContext);
 }
 NS_IMETHODIMP
 nsSyncLoader::OnStopRequest(nsIRequest *aRequest, nsISupports *aContext,
                             nsresult aStatusCode)
 {
     if (NS_SUCCEEDED(mAsyncLoadStatus) && NS_FAILED(aStatusCode)) {
         mAsyncLoadStatus = aStatusCode;
     }
     nsresult rv = mListener->OnStopRequest(aRequest, aContext, aStatusCode);
     if (NS_SUCCEEDED(mAsyncLoadStatus) && NS_FAILED(rv)) {
         mAsyncLoadStatus = rv;
     }
     mLoading = false;
     return rv;
 }
 NS_IMETHODIMP
 nsSyncLoader::AsyncOnChannelRedirect(nsIChannel *aOldChannel,
                                      nsIChannel *aNewChannel,
                                      uint32_t aFlags,
                                      nsIAsyncVerifyRedirectCallback *callback)
 {
     NS_PRECONDITION(aNewChannel, "Redirecting to null channel?");
     mChannel = aNewChannel;
     callback->OnRedirectVerifyCallback(NS_OK);
     return NS_OK;
 }
 NS_IMETHODIMP
 nsSyncLoader::GetInterface(const nsIID & aIID,
                            void **aResult)
 {
     return QueryInterface(aIID, aResult);
 }
 /* static */
 nsresult
 nsSyncLoadService::LoadDocument(nsIURI *aURI, nsIPrincipal *aLoaderPrincipal,
                                 nsILoadGroup *aLoadGroup, bool aForceToXML,
                                 nsIDOMDocument** aResult)
 {
     nsCOMPtr<nsIChannel> channel;
     nsresult rv = NS_NewChannel(getter_AddRefs(channel), aURI, nullptr,
                                 aLoadGroup);
     NS_ENSURE_SUCCESS(rv, rv);
     if (!aForceToXML) {
         channel->SetContentType(NS_LITERAL_CSTRING("text/xml"));
     }
     bool isChrome = false, isResource = false;
     bool isSync = (NS_SUCCEEDED(aURI->SchemeIs("chrome", &isChrome)) &&
                      isChrome) ||
                     (NS_SUCCEEDED(aURI->SchemeIs("resource", &isResource)) &&
                      isResource);
     nsRefPtr<nsSyncLoader> loader = new nsSyncLoader();
     return loader->LoadDocument(channel, aLoaderPrincipal, isSync,
                                 aForceToXML, aResult);
 }
 /* static */
 nsresult
 nsSyncLoadService::PushSyncStreamToListener(nsIInputStream* aIn,
                                             nsIStreamListener* aListener,
                                             nsIChannel* aChannel)
 {
     // Set up buffering stream
     nsresult rv;
     nsCOMPtr<nsIInputStream> bufferedStream;
     if (!NS_InputStreamIsBuffered(aIn)) {
         int64_t chunkSize;
         rv = aChannel->GetContentLength(&chunkSize);
         if (NS_FAILED(rv) || chunkSize < 1) {
             chunkSize = 4096;
         }
         chunkSize = std::min(int64_t(UINT16_MAX), chunkSize);
         rv = NS_NewBufferedInputStream(getter_AddRefs(bufferedStream), aIn,
                                        chunkSize);
         NS_ENSURE_SUCCESS(rv, rv);
         aIn = bufferedStream;
     }
     // Load
     rv = aListener->OnStartRequest(aChannel, nullptr);
     if (NS_SUCCEEDED(rv)) {
         uint64_t sourceOffset = 0;
         while (1) {
             uint64_t readCount = 0;
             rv = aIn->Available(&readCount);
             if (NS_FAILED(rv) || !readCount) {
                 if (rv == NS_BASE_STREAM_CLOSED) {
                     // End of file, but not an error
                     rv = NS_OK;
                 }
                 break;
             }
             if (readCount > UINT32_MAX)
                 readCount = UINT32_MAX;
             rv = aListener->OnDataAvailable(aChannel, nullptr, aIn,
                                             (uint32_t)std::min(sourceOffset, (uint64_t)UINT32_MAX),
                                             (uint32_t)readCount);
             if (NS_FAILED(rv)) {
                 break;
             }
             sourceOffset += readCount;
         }
     }
     if (NS_FAILED(rv)) {
         aChannel->Cancel(rv);
     }
     aListener->OnStopRequest(aChannel, nullptr, rv);
     return rv;
 }

The Tor Browser / annotate

content/base/src/nsSyncLoadService.cpp@6474c204b198 (annotated)

content/base/src/nsSyncLoadService.cpp