diff --git a/mozilla/xpcom/io/nsFileStream.cpp b/mozilla/xpcom/io/nsFileStream.cpp index 9798e07cd35..52617319424 100644 --- a/mozilla/xpcom/io/nsFileStream.cpp +++ b/mozilla/xpcom/io/nsFileStream.cpp @@ -308,6 +308,64 @@ void nsOutputFileStream::flush() mFileOutputStream->Flush(); } +//---------------------------------------------------------------------------------------- +void nsOutputFileStream::abort() +//---------------------------------------------------------------------------------------- +{ + mResult = NS_FILE_FAILURE; + close(); +} + +//======================================================================================== +// nsSaveViaTempStream +//======================================================================================== + +//---------------------------------------------------------------------------------------- +nsSaveViaTempStream::nsSaveViaTempStream(const nsFileSpec& inFileToSave) +//---------------------------------------------------------------------------------------- + : mFileToSave(inFileToSave) + , mTempFileSpec(new nsFileSpec(inFileToSave)) +{ + // Make sure the temp file's unique (in particular, different from the target file) + mTempFileSpec->MakeUnique(); + open( + *mTempFileSpec, + (PR_WRONLY | PR_CREATE_FILE | PR_TRUNCATE), + 0700 /*(octal)*/); +} // nsSaveViaTempStream::nsSaveViaTempStream + +//---------------------------------------------------------------------------------------- +void nsSaveViaTempStream::close() +//---------------------------------------------------------------------------------------- +{ + if (!mTempFileSpec) + return; + nsresult currentResult = mResult; + Inherited::close(); + mResult = currentResult; + if (error()) + { + mTempFileSpec->Delete(PR_FALSE); + } + else + { + nsFileSpec thirdSpec(mFileToSave); + thirdSpec.MakeUnique(); + nsSimpleCharString originalName(mFileToSave.GetLeafName()); + mFileToSave.Rename(nsSimpleCharString(thirdSpec.GetLeafName())); + if (NS_SUCCEEDED(mTempFileSpec->Rename(originalName)) && mTempFileSpec->Valid()) + mFileToSave.Delete(PR_FALSE); + } + delete mTempFileSpec; +} // nsSaveViaTempStream::~nsSaveViaTempStream + +//---------------------------------------------------------------------------------------- +nsSaveViaTempStream::~nsSaveViaTempStream() +//---------------------------------------------------------------------------------------- +{ + delete mTempFileSpec; +} // nsSaveViaTempStream::~nsSaveViaTempStream + //======================================================================================== // Manipulators //========================================================================================