michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "nsInputStreamChannel.h"
michael@0: 
michael@0: //-----------------------------------------------------------------------------
michael@0: // nsInputStreamChannel
michael@0: 
michael@0: nsresult
michael@0: nsInputStreamChannel::OpenContentStream(bool async, nsIInputStream **result,
michael@0:                                         nsIChannel** channel)
michael@0: {
michael@0:   NS_ENSURE_TRUE(mContentStream, NS_ERROR_NOT_INITIALIZED);
michael@0: 
michael@0:   // If content length is unknown, then we must guess.  In this case, we assume
michael@0:   // the stream can tell us.  If the stream is a pipe, then this will not work.
michael@0: 
michael@0:   if (mContentLength < 0) {
michael@0:     uint64_t avail;
michael@0:     nsresult rv = mContentStream->Available(&avail);
michael@0:     if (rv == NS_BASE_STREAM_CLOSED) {
michael@0:       // This just means there's nothing in the stream
michael@0:       avail = 0;
michael@0:     } else if (NS_FAILED(rv)) {
michael@0:       return rv;
michael@0:     }
michael@0:     mContentLength = avail;
michael@0:   }
michael@0: 
michael@0:   EnableSynthesizedProgressEvents(true);
michael@0:   
michael@0:   NS_ADDREF(*result = mContentStream);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: //-----------------------------------------------------------------------------
michael@0: // nsInputStreamChannel::nsISupports
michael@0: 
michael@0: NS_IMPL_ISUPPORTS_INHERITED(nsInputStreamChannel,
michael@0:                             nsBaseChannel,
michael@0:                             nsIInputStreamChannel)
michael@0: 
michael@0: //-----------------------------------------------------------------------------
michael@0: // nsInputStreamChannel::nsIInputStreamChannel
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInputStreamChannel::SetURI(nsIURI *uri)
michael@0: {
michael@0:   NS_ENSURE_TRUE(!URI(), NS_ERROR_ALREADY_INITIALIZED);
michael@0:   nsBaseChannel::SetURI(uri);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInputStreamChannel::GetContentStream(nsIInputStream **stream)
michael@0: {
michael@0:   NS_IF_ADDREF(*stream = mContentStream);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInputStreamChannel::SetContentStream(nsIInputStream *stream)
michael@0: {
michael@0:   NS_ENSURE_TRUE(!mContentStream, NS_ERROR_ALREADY_INITIALIZED);
michael@0:   mContentStream = stream;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInputStreamChannel::GetSrcdocData(nsAString& aSrcdocData)
michael@0: {
michael@0:   aSrcdocData = mSrcdocData;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInputStreamChannel::SetSrcdocData(const nsAString& aSrcdocData)
michael@0: {
michael@0:   mSrcdocData = aSrcdocData;
michael@0:   mIsSrcdocChannel = true;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInputStreamChannel::GetIsSrcdocChannel(bool *aIsSrcdocChannel)
michael@0: {
michael@0:   *aIsSrcdocChannel = mIsSrcdocChannel;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInputStreamChannel::GetBaseURI(nsIURI** aBaseURI)
michael@0: {
michael@0:   *aBaseURI = mBaseURI;
michael@0:   NS_IF_ADDREF(*aBaseURI);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInputStreamChannel::SetBaseURI(nsIURI* aBaseURI)
michael@0: {
michael@0:   mBaseURI = aBaseURI;
michael@0:   return NS_OK;
michael@0: }