/* -*- Mode: C++; tab-width: 4; 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) 1998 Netscape Communications Corporation. All * Rights Reserved. * * Contributor(s): */ #include "nsIStringStream.h" #include "nsIFileStream.h" #include "prerror.h" #include "nsFileSpec.h" #include "plstr.h" //======================================================================================== class BasicStringImpl : public nsIOutputStream , public nsIInputStream , public nsIRandomAccessStore //======================================================================================== { public: BasicStringImpl() : mOffset(0) , mLastResult(NS_OK) , mEOF(PR_FALSE) { NS_INIT_REFCNT(); } virtual ~BasicStringImpl() { } NS_IMETHOD Seek(PRSeekWhence whence, PRInt32 offset); NS_IMETHOD Tell(PRIntn* outWhere) { *outWhere = mOffset; return NS_OK; } NS_IMETHOD GetAtEOF(PRBool* outAtEOF) { *outAtEOF = mEOF; return NS_OK; } NS_IMETHOD SetAtEOF(PRBool inAtEOF) { mEOF = inAtEOF; return NS_OK; } NS_IMETHOD Available(PRUint32 *aLength) { NS_PRECONDITION(aLength != nsnull, "null ptr"); if (!aLength) return NS_ERROR_NULL_POINTER; *aLength = length(); return NS_OK; } NS_IMETHOD Read(char* aBuf, PRUint32 aCount, PRUint32 *aReadCount) { NS_PRECONDITION(aBuf != nsnull, "null ptr"); if (!aBuf) return NS_ERROR_NULL_POINTER; NS_PRECONDITION(aReadCount != nsnull, "null ptr"); if (!aReadCount) return NS_ERROR_NULL_POINTER; if (NS_FAILED(mLastResult)) return mLastResult; PRInt32 bytesRead = read(aBuf, aCount); if (NS_FAILED(mLastResult)) { *aReadCount = 0; return mLastResult; } *aReadCount = bytesRead; if (bytesRead < (PRInt32)aCount) SetAtEOF(PR_TRUE); return NS_OK; } // nsIOutputStream interface NS_IMETHOD Write(const char* aBuf, PRUint32 aCount, PRUint32 *aWriteCount) { NS_PRECONDITION(aBuf != nsnull, "null ptr"); NS_PRECONDITION(aWriteCount != nsnull, "null ptr"); if (NS_FAILED(mLastResult)) return mLastResult; PRInt32 bytesWrit = write(aBuf, aCount); if (NS_FAILED(mLastResult)) { *aWriteCount = 0; return mLastResult; } *aWriteCount = bytesWrit; return NS_OK; } public: // nsISupports interface NS_DECL_ISUPPORTS NS_IMETHOD Close() { return NS_OK; } // nsIInputStream interface NS_IMETHOD Flush() { return NS_OK; } public: nsresult get_result() const { return mLastResult; } protected: virtual PRInt32 length() const = 0; virtual PRInt32 read(char* buf, PRUint32 count) = 0; virtual PRInt32 write(const char*, PRUint32) { NS_ASSERTION(PR_FALSE, "Write to a const string"); mLastResult = NS_FILE_RESULT(PR_ILLEGAL_ACCESS_ERROR); return -1; } protected: PRUint32 mOffset; nsresult mLastResult; PRBool mEOF; }; // class BasicStringImpl //======================================================================================== class ConstCharImpl : public BasicStringImpl //======================================================================================== { public: ConstCharImpl(const char* inString, PRInt32 inLength = -1) : mConstString(inString), mLength(inLength == -1 ? (inString ? strlen(inString) : 0) : inLength) {} protected: virtual PRInt32 length() const { return mLength - mOffset; } virtual PRInt32 read(char* buf, PRUint32 aCount) { PRInt32 maxCount = mLength - mOffset; if ((PRInt32)aCount > maxCount) aCount = maxCount; memcpy(buf, mConstString + mOffset, aCount); mOffset += aCount; return aCount; } protected: const char* mConstString; size_t mLength; }; // class ConstCharImpl //======================================================================================== class CharImpl : public ConstCharImpl //======================================================================================== { enum { kAllocQuantum = 256 }; public: CharImpl(char** inString, PRInt32 inLength = -1) : ConstCharImpl(*inString, inLength) , mString(*inString) , mAllocLength(mLength + 1) , mOriginalLength(mLength) { if (!mString) { mAllocLength = kAllocQuantum; mString = new char[mAllocLength]; if (!mString) { mLastResult = NS_ERROR_OUT_OF_MEMORY; return; } mConstString = mString; *mString = '\0'; } } ~CharImpl() { if (mString) { delete [] mString; } } virtual PRInt32 write(const char* buf, PRUint32 aCount) { if (!buf) return 0; PRInt32 maxCount = mAllocLength - 1 - mOffset; if ((PRInt32)aCount > maxCount) { mAllocLength = aCount + 1 + mOffset + kAllocQuantum; char* newString = new char[mAllocLength]; if (!newString) { mLastResult = NS_ERROR_OUT_OF_MEMORY; return 0; } memcpy(newString, mString, mLength); delete [] mString; mString = newString; mConstString = newString; } memcpy(mString + mOffset, buf, aCount); mOffset += aCount; if (mOffset > mOriginalLength) mString[mOffset] = 0; return aCount; } protected: char*& mString; size_t mAllocLength; size_t mOriginalLength; }; // class CharImpl //======================================================================================== class ConstStringImpl : public ConstCharImpl //======================================================================================== { public: ConstStringImpl(const nsString& inString) : ConstCharImpl(inString.ToNewCString(), inString.Length()) { } ~ConstStringImpl() { Recycle((char*)mConstString); } protected: protected: }; // class ConstStringImpl //======================================================================================== class StringImpl : public ConstStringImpl // This is wrong, since it really converts to 1-char strings. //======================================================================================== { public: StringImpl(nsString& inString) : ConstStringImpl(inString) , mString(inString) { } protected: virtual PRInt32 write(const char* buf, PRUint32 count) { if (!buf) return 0; // Clone our string as chars char* cstring = mString.ToNewCString(); // Make a CharImpl and do the write CharImpl chars(&cstring, mString.Length()); chars.Seek(PR_SEEK_SET, mOffset); // Get the bytecount and result from the CharImpl PRInt32 result = chars.write(buf,count); mLastResult = chars.get_result(); // Set our string to match the new chars chars.Seek(PR_SEEK_SET, 0); PRUint32 newLength; chars.Available(&newLength); mString.Assign(cstring, newLength); // Set our const string also... delete [] (char*)mConstString; mConstString = cstring; return result; } protected: nsString& mString; }; // class StringImpl NS_IMPL_RELEASE(BasicStringImpl) NS_IMPL_ADDREF(BasicStringImpl) NS_IMPL_QUERY_HEAD(BasicStringImpl) NS_IMPL_QUERY_BODY(nsIRandomAccessStore) NS_IMPL_QUERY_BODY(nsIOutputStream) NS_IMPL_QUERY_BODY(nsIInputStream) if ( aIID.Equals(NS_GET_IID(nsIBaseStream)) ) foundInterface = NS_STATIC_CAST(nsIBaseStream*, NS_STATIC_CAST(nsIOutputStream*, this)); else NS_IMPL_QUERY_TAIL(nsIOutputStream) //---------------------------------------------------------------------------------------- NS_IMETHODIMP BasicStringImpl::Seek(PRSeekWhence whence, PRInt32 offset) //---------------------------------------------------------------------------------------- { mLastResult = NS_OK; // reset on a seek. mEOF = PR_FALSE; // reset on a seek. PRInt32 fileSize = length(); PRInt32 newPosition=-1; switch (whence) { case PR_SEEK_CUR: newPosition = mOffset + offset; break; case PR_SEEK_SET: newPosition = offset; break; case PR_SEEK_END: newPosition = fileSize + offset; break; } if (newPosition < 0) { newPosition = 0; mLastResult = NS_FILE_RESULT(PR_FILE_SEEK_ERROR); } if (newPosition >= fileSize) { newPosition = fileSize; mEOF = PR_TRUE; } mOffset = newPosition; return NS_OK; } // StringImpl::Seek //---------------------------------------------------------------------------------------- extern "C" NS_COM nsresult NS_NewStringInputStream( nsISupports** aStreamResult, const nsString& aStringToRead) // Factory method to get an nsInputStream from a string. Result will implement all the // file stream interfaces in nsIFileStream.h //---------------------------------------------------------------------------------------- { NS_PRECONDITION(aStreamResult != nsnull, "null ptr"); if (! aStreamResult) return NS_ERROR_NULL_POINTER; ConstStringImpl* stream = new ConstStringImpl(aStringToRead); if (! stream) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(stream); *aStreamResult = (nsISupports*)(void*)stream; return NS_OK; } //---------------------------------------------------------------------------------------- extern "C" NS_COM nsresult NS_NewStringOutputStream( nsISupports** aStreamResult, nsString& aStringToChange) // Factory method to get an nsOutputStream from a string. Result will implement all the // file stream interfaces in nsIFileStream.h //---------------------------------------------------------------------------------------- { NS_PRECONDITION(aStreamResult != nsnull, "null ptr"); if (! aStreamResult) return NS_ERROR_NULL_POINTER; StringImpl* stream = new StringImpl(aStringToChange); if (! stream) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(stream); *aStreamResult = (nsISupports*)(void*)stream; return NS_OK; } //---------------------------------------------------------------------------------------- extern "C" NS_COM nsresult NS_NewCharInputStream( nsISupports** aStreamResult, const char* aStringToRead) // Factory method to get an nsInputStream from a string. Result will implement all the // file stream interfaces in nsIFileStream.h //---------------------------------------------------------------------------------------- { NS_PRECONDITION(aStreamResult != nsnull, "null ptr"); if (! aStreamResult) return NS_ERROR_NULL_POINTER; ConstCharImpl* stream = new ConstCharImpl(aStringToRead); if (! stream) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(stream); *aStreamResult = (nsISupports*)(void*)stream; return NS_OK; } //---------------------------------------------------------------------------------------- extern "C" NS_COM nsresult NS_NewCharOutputStream( nsISupports** aStreamResult, char** aStringToChange) // Factory method to get an nsOutputStream to a string. Result will implement all the // file stream interfaces in nsIFileStream.h //---------------------------------------------------------------------------------------- { NS_PRECONDITION(aStreamResult != nsnull, "null ptr"); NS_PRECONDITION(aStringToChange != nsnull, "null ptr"); if (!aStreamResult || !aStringToChange) return NS_ERROR_NULL_POINTER; CharImpl* stream = new CharImpl(aStringToChange); if (! stream) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(stream); *aStreamResult = (nsISupports*)(void*)stream; return NS_OK; } //---------------------------------------------------------------------------------------- extern "C" NS_COM nsresult NS_NewStringIOStream( nsISupports** aStreamResult, nsString& aStringToChange) // Factory method to get an nsOutputStream to a string. Result will implement all the // file stream interfaces in nsIFileStream.h //---------------------------------------------------------------------------------------- { return NS_NewStringOutputStream(aStreamResult, aStringToChange); } //---------------------------------------------------------------------------------------- extern "C" NS_COM nsresult NS_NewCharIOStream( nsISupports** aStreamResult, char** aStringToChange) // Factory method to get an nsOutputStream to a string. Result will implement all the // file stream interfaces in nsIFileStream.h //---------------------------------------------------------------------------------------- { return NS_NewCharOutputStream(aStreamResult, aStringToChange); }