Files
palemoon27/netwerk/base/MemoryDownloader.cpp
roytam1 50d80fcfc7 import changes from `dev' branch of rmottola/Arctic-Fox:
- Bug 1237091. Remove WAP telemetry probe. r=mcmanus (8538288c45)
- bug 718797 - allow heuristic cache of query string resources r=hurley (df7c0d050c)
- Bug 1238290 - fix bad necko deps on unified_sources r=valentin.gosu (fee60661f9)
- Bug 1237371: Asynchronously shutdown the predictor IO thread. r=hurley (2b971d9714)
- Bug 407537 - Dont normalize a nonexistant file r=biesi (e9d4b81b0f)
- Bug 1247733 part 1: Create a helper function for nsStandardURL's code to add/remove/replace a port in the URL string. r=valentin (81db1064a1)
- Bug 1247733 part 2: Give nsIStandardURL an API to set its default port, and use it when upgrading HTTP connections to HTTPS. r=valentin (de4ee7a9c3)
- Bug 1247733 part 3: Add mochitest to ensure that SVG <use> is rendered correctly in documents that have been upgraded using HSTS. r=valentin (a929b68f0c)
- Bug 1247733 part 4: Add xpcshell test for nsIStandardURL setDefaultPot() API. r=valentin (4599dc1a46)
- Bug 524232 - cache about: protocol handlers r=mayhemer (7ac918c396)
- Bug 1238010 - Turn off ClosingService. r=mcmanus (e46aa99310)
- Bug 1238017 - Remove ClosingService. r=mcmanus (c5d027507b)
- Bug 1238910 - Rework shutdown necko. r=mcmanus (c95508d202)
- Bug 1240269 - Do not open UDP socket during shutdown. r=mcmanus (e62a2008b5)
- Bug 1240481 - Limit PR_Close calls during shutdown. r=mcmanus (aa3d4bd35c)
- Bug 1242755 - Move nsHttpConnectionMgr->Shutdown back to nsHttpHandler. r=mcmanus (a630e3baf7)
- add webapps (1b90fe85a8)
- bug 1069556 - sync to Breakpad c53ed143108948eb7e2d7ee77dc8c0d92050ce7c. r=glandium, benwa (e0608cc38b)
- more work to add pdfjs (ef88fb0b25)
- var-let (df1123752d)
- Bug 1242254 - Enable initial set of eslint rules for PSM. r=dkeeler (f68806c68d)
2023-07-26 10:44:09 +08:00

93 lines
2.6 KiB
C++

/* -*- Mode: C++; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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/. */
#include "MemoryDownloader.h"
#include "mozilla/Assertions.h"
#include "nsIInputStream.h"
namespace mozilla {
namespace net {
NS_IMPL_ISUPPORTS(MemoryDownloader,
nsIStreamListener,
nsIRequestObserver)
MemoryDownloader::MemoryDownloader(IObserver* aObserver)
: mObserver(aObserver)
{
}
MemoryDownloader::~MemoryDownloader()
{
}
NS_IMETHODIMP
MemoryDownloader::OnStartRequest(nsIRequest* aRequest, nsISupports* aCtxt)
{
MOZ_ASSERT(!mData);
mData.reset(new FallibleTArray<uint8_t>());
mStatus = NS_OK;
return NS_OK;
}
NS_IMETHODIMP
MemoryDownloader::OnStopRequest(nsIRequest* aRequest,
nsISupports* aCtxt,
nsresult aStatus)
{
MOZ_ASSERT_IF(NS_FAILED(mStatus), NS_FAILED(aStatus));
MOZ_ASSERT(!mData == NS_FAILED(mStatus));
Data data;
data.swap(mData);
RefPtr<IObserver> observer;
observer.swap(mObserver);
observer->OnDownloadComplete(this, aRequest, aCtxt, aStatus,
mozilla::Move(data));
return NS_OK;
}
NS_METHOD
MemoryDownloader::ConsumeData(nsIInputStream* aIn,
void* aClosure,
const char* aFromRawSegment,
uint32_t aToOffset,
uint32_t aCount,
uint32_t* aWriteCount)
{
MemoryDownloader* self = static_cast<MemoryDownloader*>(aClosure);
if (!self->mData->AppendElements(aFromRawSegment, aCount, fallible)) {
// The error returned by ConsumeData isn't propagated to the
// return of ReadSegments, so it has to be passed as state.
self->mStatus = NS_ERROR_OUT_OF_MEMORY;
return NS_ERROR_OUT_OF_MEMORY;
}
*aWriteCount = aCount;
return NS_OK;
}
NS_IMETHODIMP
MemoryDownloader::OnDataAvailable(nsIRequest* aRequest,
nsISupports* aCtxt,
nsIInputStream* aInStr,
uint64_t aSourceOffset,
uint32_t aCount)
{
uint32_t n;
MOZ_ASSERT(mData);
nsresult rv = aInStr->ReadSegments(ConsumeData, this, aCount, &n);
if (NS_SUCCEEDED(mStatus) && NS_FAILED(rv)) {
mStatus = rv;
}
if (NS_WARN_IF(NS_FAILED(mStatus))) {
mData.reset(nullptr);
return mStatus;
}
return NS_OK;
}
} // namespace net
} // namespace mozilla