gecko-dev/dom/file/uri/BlobURLChannel.cpp
Andrea Marchesini f6768a8ff6 Bug 1228139 - Remove nsIURIWithPrincipal - part 3 - main part, r=bz
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.
2018-07-24 22:15:57 +02:00

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;
}