зеркало из https://github.com/mozilla/gecko-dev.git
194 строки
5.2 KiB
C++
194 строки
5.2 KiB
C++
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*-
|
||
|
*
|
||
|
* The contents of this file are subject to the Netscape Public
|
||
|
* License Version 1.1 (the "License"); you may not use this file
|
||
|
* except in compliance with the License. You may obtain a copy of
|
||
|
* the License at http://www.mozilla.org/NPL/
|
||
|
*
|
||
|
* Software distributed under the License is distributed on an "AS
|
||
|
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
|
||
|
* implied. See the License for the specific language governing
|
||
|
* rights and limitations under the License.
|
||
|
*
|
||
|
* The Original Code is mozilla.org code.
|
||
|
*
|
||
|
* The Initial Developer of the Original Code is Netscape
|
||
|
* Communications Corporation. Portions created by Netscape are
|
||
|
* Copyright (C) 2001 Netscape Communications Corporation. All
|
||
|
* Rights Reserved.
|
||
|
*
|
||
|
* Contributor(s):
|
||
|
* Darin Fisher <darin@netscape.com> (original author)
|
||
|
*/
|
||
|
|
||
|
#include "nsIInputStreamTee.h"
|
||
|
#include "nsIInputStream.h"
|
||
|
#include "nsIOutputStream.h"
|
||
|
#include "nsCOMPtr.h"
|
||
|
|
||
|
class nsInputStreamTee : public nsIInputStreamTee
|
||
|
{
|
||
|
public:
|
||
|
NS_DECL_ISUPPORTS
|
||
|
NS_DECL_NSIINPUTSTREAM
|
||
|
NS_DECL_NSIINPUTSTREAMTEE
|
||
|
|
||
|
nsInputStreamTee();
|
||
|
virtual ~nsInputStreamTee();
|
||
|
|
||
|
private:
|
||
|
nsresult TeeSegment(const char *buf, PRUint32 count);
|
||
|
|
||
|
static NS_METHOD WriteSegmentFun(nsIInputStream *, void *, const char *,
|
||
|
PRUint32, PRUint32, PRUint32 *);
|
||
|
|
||
|
private:
|
||
|
nsCOMPtr<nsIInputStream> mSource;
|
||
|
nsCOMPtr<nsIOutputStream> mSink;
|
||
|
nsWriteSegmentFun mWriter; // for implementing ReadSegments
|
||
|
void *mClosure; // for implementing ReadSegments
|
||
|
};
|
||
|
|
||
|
nsInputStreamTee::nsInputStreamTee()
|
||
|
{
|
||
|
NS_INIT_ISUPPORTS();
|
||
|
}
|
||
|
|
||
|
nsInputStreamTee::~nsInputStreamTee()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
nsresult
|
||
|
nsInputStreamTee::TeeSegment(const char *buf, PRUint32 count)
|
||
|
{
|
||
|
nsresult rv = NS_OK;
|
||
|
PRUint32 bytesWritten = 0;
|
||
|
while (count) {
|
||
|
rv = mSink->Write(buf + bytesWritten, count, &bytesWritten);
|
||
|
if (NS_FAILED(rv)) break;
|
||
|
NS_ASSERTION(bytesWritten <= count, "wrote too much");
|
||
|
count -= bytesWritten;
|
||
|
}
|
||
|
return rv;
|
||
|
}
|
||
|
|
||
|
NS_METHOD
|
||
|
nsInputStreamTee::WriteSegmentFun(nsIInputStream *in, void *closure, const char *fromSegment,
|
||
|
PRUint32 offset, PRUint32 count, PRUint32 *writeCount)
|
||
|
{
|
||
|
nsInputStreamTee *tee = NS_REINTERPRET_CAST(nsInputStreamTee *, closure);
|
||
|
|
||
|
nsresult rv = tee->mWriter(in, tee->mClosure, fromSegment, offset, count, writeCount);
|
||
|
if (NS_FAILED(rv) || (writeCount == 0))
|
||
|
return rv;
|
||
|
|
||
|
return tee->TeeSegment(fromSegment, *writeCount);
|
||
|
}
|
||
|
|
||
|
NS_IMPL_ISUPPORTS2(nsInputStreamTee,
|
||
|
nsIInputStreamTee,
|
||
|
nsIInputStream)
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsInputStreamTee::Close()
|
||
|
{
|
||
|
NS_ENSURE_TRUE(mSource, NS_ERROR_NOT_INITIALIZED);
|
||
|
nsresult rv = mSource->Close();
|
||
|
mSource = 0;
|
||
|
mSink = 0;
|
||
|
return rv;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsInputStreamTee::Available(PRUint32 *avail)
|
||
|
{
|
||
|
NS_ENSURE_TRUE(mSource, NS_ERROR_NOT_INITIALIZED);
|
||
|
return mSource->Available(avail);
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsInputStreamTee::Read(char *buf, PRUint32 count, PRUint32 *bytesRead)
|
||
|
{
|
||
|
NS_ENSURE_TRUE(mSource, NS_ERROR_NOT_INITIALIZED);
|
||
|
NS_ENSURE_TRUE(mSink, NS_ERROR_NOT_INITIALIZED);
|
||
|
|
||
|
nsresult rv = mSource->Read(buf, count, bytesRead);
|
||
|
if (NS_FAILED(rv) || (*bytesRead == 0))
|
||
|
return rv;
|
||
|
|
||
|
return TeeSegment(buf, *bytesRead);
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsInputStreamTee::ReadSegments(nsWriteSegmentFun writer,
|
||
|
void *closure,
|
||
|
PRUint32 count,
|
||
|
PRUint32 *bytesRead)
|
||
|
{
|
||
|
NS_ENSURE_TRUE(mSource, NS_ERROR_NOT_INITIALIZED);
|
||
|
NS_ENSURE_TRUE(mSink, NS_ERROR_NOT_INITIALIZED);
|
||
|
|
||
|
mWriter = writer;
|
||
|
mClosure = closure;
|
||
|
|
||
|
return mSource->ReadSegments(WriteSegmentFun, this, count, bytesRead);
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsInputStreamTee::GetNonBlocking(PRBool *result)
|
||
|
{
|
||
|
NS_ENSURE_TRUE(mSource, NS_ERROR_NOT_INITIALIZED);
|
||
|
return mSource->GetNonBlocking(result);
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsInputStreamTee::GetObserver(nsIInputStreamObserver **obs)
|
||
|
{
|
||
|
NS_ENSURE_TRUE(mSource, NS_ERROR_NOT_INITIALIZED);
|
||
|
return mSource->GetObserver(obs);
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsInputStreamTee::SetObserver(nsIInputStreamObserver *obs)
|
||
|
{
|
||
|
NS_ENSURE_TRUE(mSource, NS_ERROR_NOT_INITIALIZED);
|
||
|
return mSource->SetObserver(obs);
|
||
|
}
|
||
|
|
||
|
nsresult
|
||
|
nsInputStreamTee::Init(nsIInputStream *source, nsIOutputStream *sink)
|
||
|
{
|
||
|
NS_ENSURE_TRUE(source, NS_ERROR_NOT_INITIALIZED);
|
||
|
NS_ENSURE_TRUE(sink, NS_ERROR_NOT_INITIALIZED);
|
||
|
|
||
|
PRBool nonBlocking = PR_FALSE;
|
||
|
nsresult rv = sink->GetNonBlocking(&nonBlocking);
|
||
|
if (NS_FAILED(rv))
|
||
|
return rv;
|
||
|
if (nonBlocking) {
|
||
|
NS_NOTREACHED("Can only tee data to a blocking output stream");
|
||
|
return NS_ERROR_NOT_IMPLEMENTED;
|
||
|
}
|
||
|
mSource = source;
|
||
|
mSink = sink;
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
// factory method
|
||
|
|
||
|
NS_COM nsresult
|
||
|
NS_NewInputStreamTee(nsIInputStream **result,
|
||
|
nsIInputStream *source,
|
||
|
nsIOutputStream *sink)
|
||
|
{
|
||
|
nsCOMPtr<nsIInputStreamTee> tee;
|
||
|
NS_NEWXPCOM(tee, nsInputStreamTee);
|
||
|
if (!tee)
|
||
|
return NS_ERROR_OUT_OF_MEMORY;
|
||
|
|
||
|
nsresult rv = tee->Init(source, sink);
|
||
|
if (NS_SUCCEEDED(rv))
|
||
|
NS_ADDREF(*result = tee);
|
||
|
return rv;
|
||
|
}
|