152 lines
3.1 KiB
C++
152 lines
3.1 KiB
C++
#include "mozilla/net/AltDataOutputStreamChild.h"
|
|
#include "mozilla/Unused.h"
|
|
#include "nsIInputStream.h"
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
NS_IMPL_ADDREF(AltDataOutputStreamChild)
|
|
|
|
NS_IMETHODIMP_(MozExternalRefCountType) AltDataOutputStreamChild::Release()
|
|
{
|
|
NS_PRECONDITION(0 != mRefCnt, "dup release");
|
|
MOZ_ASSERT(NS_IsMainThread(), "Main thread only");
|
|
--mRefCnt;
|
|
NS_LOG_RELEASE(this, mRefCnt, "AltDataOutputStreamChild");
|
|
|
|
if (mRefCnt == 1 && mIPCOpen) {
|
|
// Send_delete calls NeckoChild::PAltDataOutputStreamChild, which will release
|
|
// again to refcount == 0
|
|
PAltDataOutputStreamChild::Send__delete__(this);
|
|
return 0;
|
|
}
|
|
|
|
if (mRefCnt == 0) {
|
|
mRefCnt = 1; /* stabilize */
|
|
delete this;
|
|
return 0;
|
|
}
|
|
return mRefCnt;
|
|
}
|
|
|
|
NS_INTERFACE_MAP_BEGIN(AltDataOutputStreamChild)
|
|
NS_INTERFACE_MAP_ENTRY(nsIOutputStream)
|
|
NS_INTERFACE_MAP_ENTRY(nsISupports)
|
|
NS_INTERFACE_MAP_END
|
|
|
|
AltDataOutputStreamChild::AltDataOutputStreamChild()
|
|
: mIPCOpen(false)
|
|
, mError(NS_OK)
|
|
{
|
|
MOZ_ASSERT(NS_IsMainThread(), "Main thread only");
|
|
}
|
|
|
|
AltDataOutputStreamChild::~AltDataOutputStreamChild()
|
|
{
|
|
}
|
|
|
|
void
|
|
AltDataOutputStreamChild::AddIPDLReference()
|
|
{
|
|
MOZ_ASSERT(!mIPCOpen, "Attempt to retain more than one IPDL reference");
|
|
mIPCOpen = true;
|
|
AddRef();
|
|
}
|
|
|
|
void
|
|
AltDataOutputStreamChild::ReleaseIPDLReference()
|
|
{
|
|
MOZ_ASSERT(mIPCOpen, "Attempt to release nonexistent IPDL reference");
|
|
mIPCOpen = false;
|
|
Release();
|
|
}
|
|
|
|
bool
|
|
AltDataOutputStreamChild::WriteDataInChunks(const nsCString& data)
|
|
{
|
|
const uint32_t kChunkSize = 128*1024;
|
|
uint32_t next = std::min(data.Length(), kChunkSize);
|
|
for (uint32_t i = 0; i < data.Length();
|
|
i = next, next = std::min(data.Length(), next + kChunkSize)) {
|
|
nsCString chunk(Substring(data, i, kChunkSize));
|
|
if (mIPCOpen && !SendWriteData(chunk)) {
|
|
mIPCOpen = false;
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
AltDataOutputStreamChild::Close()
|
|
{
|
|
if (!mIPCOpen) {
|
|
return NS_ERROR_NOT_AVAILABLE;
|
|
}
|
|
if (NS_FAILED(mError)) {
|
|
return mError;
|
|
}
|
|
Unused << SendClose();
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
AltDataOutputStreamChild::Flush()
|
|
{
|
|
if (!mIPCOpen) {
|
|
return NS_ERROR_NOT_AVAILABLE;
|
|
}
|
|
if (NS_FAILED(mError)) {
|
|
return mError;
|
|
}
|
|
|
|
// This is a no-op
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
AltDataOutputStreamChild::Write(const char * aBuf, uint32_t aCount, uint32_t *_retval)
|
|
{
|
|
if (!mIPCOpen) {
|
|
return NS_ERROR_NOT_AVAILABLE;
|
|
}
|
|
if (NS_FAILED(mError)) {
|
|
return mError;
|
|
}
|
|
if (WriteDataInChunks(nsCString(aBuf, aCount))) {
|
|
*_retval = aCount;
|
|
return NS_OK;
|
|
}
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
AltDataOutputStreamChild::WriteFrom(nsIInputStream *aFromStream, uint32_t aCount, uint32_t *_retval)
|
|
{
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
AltDataOutputStreamChild::WriteSegments(nsReadSegmentFun aReader, void *aClosure, uint32_t aCount, uint32_t *_retval)
|
|
{
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
AltDataOutputStreamChild::IsNonBlocking(bool *_retval)
|
|
{
|
|
*_retval = false;
|
|
return NS_OK;
|
|
}
|
|
|
|
bool
|
|
AltDataOutputStreamChild::RecvError(const nsresult& err)
|
|
{
|
|
mError = err;
|
|
return true;
|
|
}
|
|
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|