mirror of
https://github.com/mozilla/gecko-dev.git
synced 2025-03-03 23:30:46 +00:00

nsIURIWithPrincipal is currently used to retrieve the nsIPrincipal from a BlobURL object. BlobURLProtocolHandler has a hashtable containing, for each blobURL, a BlobImpl and its nsIPrincipal. This patch introduces BlobURLProtocolHandler::GetBlobURLPrincipal() that retrieves the nsIPrincipal from this hashtable. This patch fixes also a bug in how the revocation of blobURLs is broadcasted to other processes. This should be done immediately because each process creates its own timer to revoke them after 5 seconds. An important change is related to NS_SecurityCompareURIs() where, if 1 (or both) of the 2 URIs to compare, is a revoked BlobURL, we will QI its URL to nsIStandardURL and fail out at that point.
94 lines
2.0 KiB
C++
94 lines
2.0 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* 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 "BlobURLChannel.h"
|
|
#include "mozilla/dom/BlobImpl.h"
|
|
|
|
using namespace mozilla::dom;
|
|
|
|
BlobURLChannel::BlobURLChannel(nsIURI* aURI,
|
|
nsILoadInfo* aLoadInfo)
|
|
: mInitialized(false)
|
|
{
|
|
SetURI(aURI);
|
|
SetOriginalURI(aURI);
|
|
SetLoadInfo(aLoadInfo);
|
|
|
|
// If we're sandboxed, make sure to clear any owner the channel
|
|
// might already have.
|
|
if (aLoadInfo && aLoadInfo->GetLoadingSandboxed()) {
|
|
SetOwner(nullptr);
|
|
}
|
|
}
|
|
|
|
BlobURLChannel::~BlobURLChannel() = default;
|
|
|
|
void
|
|
BlobURLChannel::InitFailed()
|
|
{
|
|
MOZ_ASSERT(!mInitialized);
|
|
MOZ_ASSERT(!mInputStream);
|
|
mInitialized = true;
|
|
}
|
|
|
|
void
|
|
BlobURLChannel::Initialize(BlobImpl* aBlobImpl)
|
|
{
|
|
MOZ_ASSERT(!mInitialized);
|
|
|
|
nsAutoString contentType;
|
|
aBlobImpl->GetType(contentType);
|
|
SetContentType(NS_ConvertUTF16toUTF8(contentType));
|
|
|
|
if (aBlobImpl->IsFile()) {
|
|
nsString filename;
|
|
aBlobImpl->GetName(filename);
|
|
SetContentDispositionFilename(filename);
|
|
}
|
|
|
|
ErrorResult rv;
|
|
uint64_t size = aBlobImpl->GetSize(rv);
|
|
if (NS_WARN_IF(rv.Failed())) {
|
|
InitFailed();
|
|
return;
|
|
}
|
|
|
|
SetContentLength(size);
|
|
|
|
aBlobImpl->CreateInputStream(getter_AddRefs(mInputStream), rv);
|
|
if (NS_WARN_IF(rv.Failed())) {
|
|
InitFailed();
|
|
return;
|
|
}
|
|
|
|
MOZ_ASSERT(mInputStream);
|
|
mInitialized = true;
|
|
}
|
|
|
|
nsresult
|
|
BlobURLChannel::OpenContentStream(bool aAsync, nsIInputStream** aResult,
|
|
nsIChannel** aChannel)
|
|
{
|
|
MOZ_ASSERT(mInitialized);
|
|
|
|
if (!mInputStream) {
|
|
return NS_ERROR_MALFORMED_URI;
|
|
}
|
|
|
|
EnableSynthesizedProgressEvents(true);
|
|
|
|
nsCOMPtr<nsIInputStream> stream = mInputStream;
|
|
stream.forget(aResult);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
BlobURLChannel::OnChannelDone()
|
|
{
|
|
mInputStream = nullptr;
|
|
}
|