/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ /* ***** BEGIN LICENSE BLOCK ***** * Version: MPL 1.1/GPL 2.0/LGPL 2.1 * * The contents of this file are subject to the Mozilla 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/MPL/ * * 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 frightening to behold. * * The Initial Developer of the Original Code is * Jonas Sicking. * Portions created by the Initial Developer are Copyright (C) 2001 * the Initial Developer. All Rights Reserved. * * Contributor(s): * Jonas Sicking * * Alternatively, the contents of this file may be used under the terms of * either the GNU General Public License Version 2 or later (the "GPL"), or * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), * in which case the provisions of the GPL or the LGPL are applicable instead * of those above. If you wish to allow use of your version of this file only * under the terms of either the GPL or the LGPL, and not to allow others to * use your version of this file under the terms of the MPL, indicate your * decision by deleting the provisions above and replace them with the notice * and other provisions required by the GPL or the LGPL. If you do not delete * the provisions above, a recipient may use your version of this file under * the terms of any one of the MPL, the GPL or the LGPL. * * ***** END LICENSE BLOCK ***** */ /** * The MIME stream separates headers and a datastream. It also allows * automatic creation of the content-length header. */ #include "nsCOMPtr.h" #include "nsComponentManagerUtils.h" #include "nsIMultiplexInputStream.h" #include "nsIMIMEInputStream.h" #include "nsISeekableStream.h" #include "nsIStringStream.h" #include "nsString.h" class nsMIMEInputStream : public nsIMIMEInputStream, public nsISeekableStream { public: nsMIMEInputStream(); virtual ~nsMIMEInputStream(); NS_DECL_ISUPPORTS NS_DECL_NSIINPUTSTREAM NS_DECL_NSIMIMEINPUTSTREAM NS_DECL_NSISEEKABLESTREAM NS_METHOD Init(); private: void InitStreams(); struct ReadSegmentsState { nsIInputStream* mThisStream; nsWriteSegmentFun mWriter; void* mClosure; }; static NS_METHOD ReadSegCb(nsIInputStream* aIn, void* aClosure, const char* aFromRawSegment, PRUint32 aToOffset, PRUint32 aCount, PRUint32 *aWriteCount); nsCString mHeaders; nsCOMPtr mHeaderStream; nsCString mContentLength; nsCOMPtr mCLStream; nsCOMPtr mData; nsCOMPtr mStream; PRPackedBool mAddContentLength; PRPackedBool mStartedReading; }; NS_IMPL_THREADSAFE_ISUPPORTS3(nsMIMEInputStream, nsIMIMEInputStream, nsIInputStream, nsISeekableStream) nsMIMEInputStream::nsMIMEInputStream() : mAddContentLength(PR_FALSE), mStartedReading(PR_FALSE) { } nsMIMEInputStream::~nsMIMEInputStream() { } NS_METHOD nsMIMEInputStream::Init() { nsresult rv = NS_OK; mStream = do_CreateInstance("@mozilla.org/io/multiplex-input-stream;1", &rv); NS_ENSURE_SUCCESS(rv, rv); mHeaderStream = do_CreateInstance("@mozilla.org/io/string-input-stream;1", &rv); NS_ENSURE_SUCCESS(rv, rv); mCLStream = do_CreateInstance("@mozilla.org/io/string-input-stream;1", &rv); NS_ENSURE_SUCCESS(rv, rv); rv = mStream->AppendStream(mHeaderStream); NS_ENSURE_SUCCESS(rv, rv); rv = mStream->AppendStream(mCLStream); NS_ENSURE_SUCCESS(rv, rv); return NS_OK; } /* attribute boolean addContentLength; */ NS_IMETHODIMP nsMIMEInputStream::GetAddContentLength(PRBool *aAddContentLength) { *aAddContentLength = mAddContentLength; return NS_OK; } NS_IMETHODIMP nsMIMEInputStream::SetAddContentLength(PRBool aAddContentLength) { NS_ENSURE_FALSE(mStartedReading, NS_ERROR_FAILURE); mAddContentLength = aAddContentLength; return NS_OK; } /* void addHeader ([const] in string name, [const] in string value); */ NS_IMETHODIMP nsMIMEInputStream::AddHeader(const char *aName, const char *aValue) { NS_ENSURE_FALSE(mStartedReading, NS_ERROR_FAILURE); mHeaders.Append(aName); mHeaders.AppendLiteral(": "); mHeaders.Append(aValue); mHeaders.AppendLiteral("\r\n"); // Just in case someone somehow uses our stream, lets at least // let the stream have a valid pointer. The stream will be properly // initialized in nsMIMEInputStream::InitStreams mHeaderStream->ShareData(mHeaders.get(), 0); return NS_OK; } /* void setData (in nsIInputStream stream); */ NS_IMETHODIMP nsMIMEInputStream::SetData(nsIInputStream *aStream) { NS_ENSURE_FALSE(mStartedReading, NS_ERROR_FAILURE); // Remove the old stream if there is one if (mData) mStream->RemoveStream(2); mData = aStream; if (aStream) mStream->AppendStream(mData); return NS_OK; } // set up the internal streams void nsMIMEInputStream::InitStreams() { NS_ASSERTION(!mStartedReading, "Don't call initStreams twice without rewinding"); mStartedReading = PR_TRUE; // We'll use the content-length stream to add the final \r\n if (mAddContentLength) { PRUint32 cl = 0; if (mData) { mData->Available(&cl); } mContentLength.AssignLiteral("Content-Length: "); mContentLength.AppendInt((PRInt32)cl); mContentLength.AppendLiteral("\r\n\r\n"); } else { mContentLength.AssignLiteral("\r\n"); } mCLStream->ShareData(mContentLength.get(), -1); mHeaderStream->ShareData(mHeaders.get(), -1); } #define INITSTREAMS \ if (!mStartedReading) { \ InitStreams(); \ } // Reset mStartedReading when Seek-ing to start NS_IMETHODIMP nsMIMEInputStream::Seek(PRInt32 whence, PRInt64 offset) { nsresult rv; nsCOMPtr stream = do_QueryInterface(mStream); if (whence == NS_SEEK_SET && LL_EQ(offset, LL_Zero())) { rv = stream->Seek(whence, offset); if (NS_SUCCEEDED(rv)) mStartedReading = PR_FALSE; } else { INITSTREAMS; rv = stream->Seek(whence, offset); } return rv; } // Proxy ReadSegments since we need to be a good little nsIInputStream NS_IMETHODIMP nsMIMEInputStream::ReadSegments(nsWriteSegmentFun aWriter, void *aClosure, PRUint32 aCount, PRUint32 *_retval) { INITSTREAMS; ReadSegmentsState state; state.mThisStream = this; state.mWriter = aWriter; state.mClosure = aClosure; return mStream->ReadSegments(ReadSegCb, &state, aCount, _retval); } NS_METHOD nsMIMEInputStream::ReadSegCb(nsIInputStream* aIn, void* aClosure, const char* aFromRawSegment, PRUint32 aToOffset, PRUint32 aCount, PRUint32 *aWriteCount) { ReadSegmentsState* state = (ReadSegmentsState*)aClosure; return (state->mWriter)(state->mThisStream, state->mClosure, aFromRawSegment, aToOffset, aCount, aWriteCount); } /** * Forward everything else to the mStream after calling InitStreams() */ // nsIInputStream NS_IMETHODIMP nsMIMEInputStream::Close(void) { INITSTREAMS; return mStream->Close(); } NS_IMETHODIMP nsMIMEInputStream::Available(PRUint32 *_retval) { INITSTREAMS; return mStream->Available(_retval); } NS_IMETHODIMP nsMIMEInputStream::Read(char * buf, PRUint32 count, PRUint32 *_retval) { INITSTREAMS; return mStream->Read(buf, count, _retval); } NS_IMETHODIMP nsMIMEInputStream::IsNonBlocking(PRBool *aNonBlocking) { INITSTREAMS; return mStream->IsNonBlocking(aNonBlocking); } // nsISeekableStream NS_IMETHODIMP nsMIMEInputStream::Tell(PRInt64 *_retval) { INITSTREAMS; nsCOMPtr stream = do_QueryInterface(mStream); return stream->Tell(_retval); } NS_IMETHODIMP nsMIMEInputStream::SetEOF(void) { INITSTREAMS; nsCOMPtr stream = do_QueryInterface(mStream); return stream->SetEOF(); } /** * Factory method used by do_CreateInstance */ nsresult nsMIMEInputStreamConstructor(nsISupports *outer, REFNSIID iid, void **result) { *result = nsnull; if (outer) return NS_ERROR_NO_AGGREGATION; nsMIMEInputStream *inst = new nsMIMEInputStream(); if (!inst) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(inst); nsresult rv = inst->Init(); if (NS_FAILED(rv)) { NS_RELEASE(inst); return rv; } rv = inst->QueryInterface(iid, result); NS_RELEASE(inst); return rv; }