зеркало из https://github.com/mozilla/gecko-dev.git
142 строки
4.4 KiB
C++
142 строки
4.4 KiB
C++
/* -*- 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/. */
|
|
|
|
#include "WebBrowserPersistSerializeChild.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include "nsThreadUtils.h"
|
|
#include "ipc/IPCMessageUtils.h"
|
|
|
|
namespace mozilla {
|
|
|
|
NS_IMPL_ISUPPORTS(WebBrowserPersistSerializeChild,
|
|
nsIWebBrowserPersistWriteCompletion,
|
|
nsIWebBrowserPersistURIMap,
|
|
nsIOutputStream)
|
|
|
|
WebBrowserPersistSerializeChild::WebBrowserPersistSerializeChild(const WebBrowserPersistURIMap& aMap)
|
|
: mMap(aMap)
|
|
{
|
|
}
|
|
|
|
WebBrowserPersistSerializeChild::~WebBrowserPersistSerializeChild() = default;
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::OnFinish(nsIWebBrowserPersistDocument* aDocument,
|
|
nsIOutputStream* aStream,
|
|
const nsACString& aContentType,
|
|
nsresult aStatus)
|
|
{
|
|
MOZ_ASSERT(aStream == this);
|
|
nsCString contentType(aContentType);
|
|
Send__delete__(this, contentType, aStatus);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::GetNumMappedURIs(uint32_t* aNum)
|
|
{
|
|
*aNum = static_cast<uint32_t>(mMap.mapURIs().Length());
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::GetURIMapping(uint32_t aIndex,
|
|
nsACString& aMapFrom,
|
|
nsACString& aMapTo)
|
|
{
|
|
if (aIndex >= mMap.mapURIs().Length()) {
|
|
return NS_ERROR_INVALID_ARG;
|
|
}
|
|
aMapFrom = mMap.mapURIs()[aIndex].mapFrom();
|
|
aMapTo = mMap.mapURIs()[aIndex].mapTo();
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::GetTargetBaseURI(nsACString& aURI)
|
|
{
|
|
aURI = mMap.targetBaseURI();
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::Close()
|
|
{
|
|
NS_WARNING("WebBrowserPersistSerializeChild::Close()");
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::Flush()
|
|
{
|
|
NS_WARNING("WebBrowserPersistSerializeChild::Flush()");
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::Write(const char* aBuf, uint32_t aCount,
|
|
uint32_t* aWritten)
|
|
{
|
|
// Normally an nsIOutputStream would have to be thread-safe, but
|
|
// nsDocumentEncoder currently doesn't call this off the main
|
|
// thread (which also means it's difficult to test the
|
|
// thread-safety code this class doesn't yet have).
|
|
//
|
|
// This is *not* an NS_ERROR_NOT_IMPLEMENTED, because at this
|
|
// point we've probably already misused the non-thread-safe
|
|
// refcounting.
|
|
MOZ_RELEASE_ASSERT(NS_IsMainThread(), "Fix this class to be thread-safe.");
|
|
|
|
// Work around bug 1181433 by sending multiple messages if
|
|
// necessary to write the entire aCount bytes, even though
|
|
// nsIOutputStream.idl says we're allowed to do a short write.
|
|
const char* buf = aBuf;
|
|
uint32_t count = aCount;
|
|
*aWritten = 0;
|
|
while (count > 0) {
|
|
uint32_t toWrite = std::min(IPC::MAX_MESSAGE_SIZE, count);
|
|
nsTArray<uint8_t> arrayBuf;
|
|
// It would be nice if this extra copy could be avoided.
|
|
arrayBuf.AppendElements(buf, toWrite);
|
|
SendWriteData(std::move(arrayBuf));
|
|
*aWritten += toWrite;
|
|
buf += toWrite;
|
|
count -= toWrite;
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::WriteFrom(nsIInputStream* aFrom,
|
|
uint32_t aCount,
|
|
uint32_t* aWritten)
|
|
{
|
|
NS_WARNING("WebBrowserPersistSerializeChild::WriteFrom()");
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::WriteSegments(nsReadSegmentFun aFun,
|
|
void* aCtx,
|
|
uint32_t aCount,
|
|
uint32_t* aWritten)
|
|
{
|
|
NS_WARNING("WebBrowserPersistSerializeChild::WriteSegments()");
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
WebBrowserPersistSerializeChild::IsNonBlocking(bool* aNonBlocking)
|
|
{
|
|
// Writes will never fail with NS_BASE_STREAM_WOULD_BLOCK, so:
|
|
*aNonBlocking = false;
|
|
return NS_OK;
|
|
}
|
|
|
|
} // namespace mozilla
|