зеркало из https://github.com/mozilla/pjs.git
463 строки
17 KiB
C++
463 строки
17 KiB
C++
#include "nsIFileStream.h"
|
|
#include "nsFileSpec.h"
|
|
|
|
#include "prerror.h"
|
|
|
|
#ifdef XP_MAC
|
|
#include "pprio.h" // To get PR_ImportFile
|
|
#else
|
|
#include "prio.h"
|
|
#endif
|
|
|
|
#ifdef XP_MAC
|
|
#include <Errors.h>
|
|
#include <iostream>
|
|
#endif
|
|
|
|
//========================================================================================
|
|
class FileImpl
|
|
: public nsIOutputStream
|
|
, public nsIInputStream
|
|
, public nsIFileOutputStream
|
|
, public nsIFileInputStream
|
|
, public nsIFile
|
|
//========================================================================================
|
|
{
|
|
public:
|
|
FileImpl(PRFileDesc* inDesc)
|
|
: mFileDesc(inDesc)
|
|
, mFailed(PR_FALSE)
|
|
, mEOF(PR_FALSE)
|
|
, mLength(-1)
|
|
{
|
|
NS_INIT_REFCNT();
|
|
}
|
|
FileImpl(
|
|
const nsFileSpec& inFile,
|
|
int nsprMode,
|
|
PRIntn accessMode)
|
|
: mFileDesc(nsnull)
|
|
, mFailed(PR_FALSE)
|
|
, mEOF(PR_FALSE)
|
|
{
|
|
NS_INIT_REFCNT();
|
|
Open(inFile, nsprMode, accessMode);
|
|
}
|
|
virtual ~FileImpl()
|
|
{
|
|
Close();
|
|
}
|
|
|
|
// nsISupports interface
|
|
NS_DECL_ISUPPORTS
|
|
|
|
// nsIFile interface
|
|
NS_IMETHOD Open(
|
|
const nsFileSpec& inFile,
|
|
int nsprMode,
|
|
PRIntn accessMode);
|
|
NS_IMETHOD Close();
|
|
NS_IMETHOD Seek(PRSeekWhence whence, PRInt32 offset);
|
|
|
|
NS_IMETHOD GetIsOpen(PRBool* outOpen)
|
|
{
|
|
*outOpen = (mFileDesc != nsnull);
|
|
return NS_OK;
|
|
}
|
|
NS_IMETHOD Tell(PRIntn* outWhere);
|
|
|
|
// nsIInputStream interface
|
|
NS_IMETHOD GetLength(PRUint32 *aLength)
|
|
{
|
|
NS_PRECONDITION(aLength != nsnull, "null ptr");
|
|
if (!aLength)
|
|
return NS_ERROR_NULL_POINTER;
|
|
if (mLength < 0)
|
|
return NS_FILE_RESULT(NS_ERROR_UNEXPECTED);
|
|
*aLength = mLength;
|
|
return NS_OK;
|
|
}
|
|
NS_IMETHOD Read(char* aBuf,
|
|
PRUint32 aOffset,
|
|
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 (!mFileDesc)
|
|
return NS_FILE_RESULT(PR_BAD_DESCRIPTOR_ERROR);
|
|
if (mFailed)
|
|
return NS_ERROR_FAILURE;
|
|
if (aOffset)
|
|
PR_Seek(mFileDesc, aOffset, PR_SEEK_CUR);
|
|
PRInt32 bytesRead = PR_Read(mFileDesc, aBuf, aCount);
|
|
if (bytesRead < 0)
|
|
{
|
|
*aReadCount = 0;
|
|
mFailed = PR_TRUE;
|
|
return NS_FILE_RESULT(PR_GetError());
|
|
}
|
|
*aReadCount = bytesRead;
|
|
return NS_OK;
|
|
}
|
|
// nsIOutputStream interface
|
|
NS_IMETHOD Write(const char* aBuf,
|
|
PRUint32 aOffset,
|
|
PRUint32 aCount,
|
|
PRUint32 *aWriteCount)
|
|
{
|
|
NS_PRECONDITION(aBuf != nsnull, "null ptr");
|
|
NS_PRECONDITION(aWriteCount != nsnull, "null ptr");
|
|
|
|
|
|
#ifdef XP_MAC
|
|
// Calling PR_Write on stdout is sure suicide.
|
|
if (mFileDesc == PR_STDOUT || mFileDesc == PR_STDERR)
|
|
{
|
|
cout.write(aBuf, aCount);
|
|
*aWriteCount = aCount;
|
|
return NS_OK;
|
|
}
|
|
#endif
|
|
if (!mFileDesc)
|
|
return NS_FILE_RESULT(PR_BAD_DESCRIPTOR_ERROR);
|
|
if (mFailed)
|
|
return NS_ERROR_FAILURE;
|
|
if (aOffset)
|
|
PR_Seek(mFileDesc, aOffset, PR_SEEK_CUR);
|
|
PRInt32 bytesWrit = PR_Write(mFileDesc, aBuf, aCount);
|
|
if (bytesWrit != aCount)
|
|
{
|
|
mFailed = PR_TRUE;
|
|
*aWriteCount = 0;
|
|
return NS_FILE_RESULT(PR_GetError());
|
|
}
|
|
*aWriteCount = bytesWrit;
|
|
return NS_OK;
|
|
}
|
|
NS_IMETHOD Flush();
|
|
|
|
NS_IMETHOD GetAtEOF(PRBool* outAtEOF)
|
|
{
|
|
*outAtEOF = mEOF;
|
|
return NS_OK;
|
|
}
|
|
NS_IMETHOD SetAtEOF(PRBool inAtEOF)
|
|
{
|
|
mEOF = inAtEOF;
|
|
return NS_OK;
|
|
}
|
|
|
|
protected:
|
|
|
|
PRFileDesc* mFileDesc;
|
|
int mNSPRMode;
|
|
PRBool mFailed;
|
|
PRBool mEOF;
|
|
PRInt32 mLength;
|
|
}; // class FileImpl
|
|
|
|
#define SAY_I_IMPLEMENT(classname) \
|
|
if (aIID.Equals(classname::GetIID())) \
|
|
{ \
|
|
*aInstancePtr = (void*)((classname*)this); \
|
|
NS_ADDREF_THIS(); \
|
|
return NS_OK; \
|
|
}
|
|
|
|
NS_IMPL_RELEASE(FileImpl)
|
|
NS_IMPL_ADDREF(FileImpl)
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_IMETHODIMP FileImpl::QueryInterface(REFNSIID aIID, void** aInstancePtr)
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
if (!aInstancePtr)
|
|
return NS_ERROR_NULL_POINTER;
|
|
|
|
*aInstancePtr = nsnull;
|
|
|
|
SAY_I_IMPLEMENT(nsIFile)
|
|
SAY_I_IMPLEMENT(nsIOutputStream)
|
|
SAY_I_IMPLEMENT(nsIInputStream)
|
|
SAY_I_IMPLEMENT(nsIFileInputStream)
|
|
SAY_I_IMPLEMENT(nsIFileOutputStream)
|
|
// Note that we derive from two copies of nsIBaseStream (and hence
|
|
// of nsISupports), one through
|
|
// nsIOutputStream, the other through nsIInputStream. Resolve this
|
|
// by giving them a specific one
|
|
if (aIID.Equals(((nsIBaseStream*)(nsIOutputStream*)this)->GetIID()))
|
|
{
|
|
*aInstancePtr = (void*)((nsIBaseStream*)(nsIOutputStream*)this);
|
|
NS_ADDREF_THIS();
|
|
return NS_OK;
|
|
}
|
|
if (aIID.Equals(((nsISupports*)(nsIOutputStream*)this)->GetIID()))
|
|
{
|
|
*aInstancePtr = (void*)((nsISupports*)(nsIOutputStream*)this);
|
|
NS_ADDREF_THIS();
|
|
return NS_OK;
|
|
}
|
|
return NS_NOINTERFACE;
|
|
} // FileImpl::QueryInterface
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_IMETHODIMP FileImpl::Open(
|
|
const nsFileSpec& inFile,
|
|
int nsprMode,
|
|
PRIntn accessMode)
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
if (mFileDesc)
|
|
if ((nsprMode & mNSPRMode) == nsprMode)
|
|
return NS_OK;
|
|
else
|
|
return NS_FILE_RESULT(PR_ILLEGAL_ACCESS_ERROR);
|
|
|
|
const int nspr_modes[]={
|
|
PR_WRONLY | PR_CREATE_FILE,
|
|
PR_WRONLY | PR_CREATE_FILE | PR_APPEND,
|
|
PR_WRONLY | PR_CREATE_FILE | PR_TRUNCATE,
|
|
PR_RDONLY,
|
|
PR_RDONLY | PR_APPEND,
|
|
PR_RDWR | PR_CREATE_FILE,
|
|
PR_RDWR | PR_CREATE_FILE | PR_TRUNCATE,
|
|
// "wb",
|
|
// "ab",
|
|
// "wb",
|
|
// "rb",
|
|
// "r+b",
|
|
// "w+b",
|
|
0 };
|
|
const int* currentLegalMode = nspr_modes;
|
|
while (*currentLegalMode && nsprMode != *currentLegalMode)
|
|
++currentLegalMode;
|
|
if (!*currentLegalMode)
|
|
return NS_FILE_RESULT(PR_ILLEGAL_ACCESS_ERROR);
|
|
|
|
#ifdef XP_MAC
|
|
// Use the file spec to open the file, because one path can be common to
|
|
// several files on the Macintosh (you can have several volumes with the
|
|
// same name, see).
|
|
mFileDesc = 0;
|
|
if (inFile.Error() != noErr)
|
|
return NS_FILE_RESULT(inFile.Error());
|
|
OSErr err = noErr;
|
|
#if DEBUG
|
|
const OSType kCreator = 'CWIE';
|
|
#else
|
|
const OSType kCreator = 'MOSS';
|
|
#endif
|
|
// Resolve the alias to the original file.
|
|
nsFileSpec original = inFile;
|
|
PRBool ignoredResult;
|
|
original.ResolveAlias(ignoredResult);
|
|
const FSSpec& spec = original.operator const FSSpec&();
|
|
if (nsprMode & PR_CREATE_FILE)
|
|
err = FSpCreate(&spec, kCreator, 'TEXT', 0);
|
|
if (err == dupFNErr)
|
|
err = noErr;
|
|
if (err != noErr)
|
|
return NS_FILE_RESULT(err);
|
|
|
|
SInt8 perm;
|
|
if (nsprMode & PR_RDWR)
|
|
perm = fsRdWrPerm;
|
|
else if (nsprMode & PR_WRONLY)
|
|
perm = fsWrPerm;
|
|
else
|
|
perm = fsRdPerm;
|
|
|
|
short refnum;
|
|
err = FSpOpenDF(&spec, perm, &refnum);
|
|
|
|
if (err == noErr && (nsprMode & PR_TRUNCATE))
|
|
err = SetEOF(refnum, 0);
|
|
if (err == noErr && (nsprMode & PR_APPEND))
|
|
err = SetFPos(refnum, fsFromLEOF, 0);
|
|
if (err != noErr)
|
|
return NS_FILE_RESULT(err);
|
|
|
|
if ((mFileDesc = PR_ImportFile(refnum)) == 0)
|
|
return NS_FILE_RESULT(PR_GetError());
|
|
#else
|
|
// Platforms other than Macintosh...
|
|
// Another bug in NSPR: Mac PR_Open assumes a unix style path, but Win PR_Open assumes
|
|
// a windows path.
|
|
if ((mFileDesc = PR_Open((const char*)nsFileSpec(inFile), nsprMode, accessMode)) == 0)
|
|
return NS_FILE_RESULT(PR_GetError());
|
|
#endif
|
|
mNSPRMode = nsprMode;
|
|
mLength = PR_Available(mFileDesc);
|
|
return NS_OK;
|
|
} // FileImpl::Open
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_IMETHODIMP FileImpl::Seek(PRSeekWhence whence, PRInt32 offset)
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
if (mFileDesc==PR_STDIN || mFileDesc==PR_STDOUT || mFileDesc==PR_STDERR || !mFileDesc)
|
|
return NS_FILE_RESULT(PR_BAD_DESCRIPTOR_ERROR);
|
|
mFailed = PR_FALSE; // reset on a seek.
|
|
mEOF = PR_FALSE; // reset on a seek.
|
|
PRInt32 position = PR_Seek(mFileDesc, 0, PR_SEEK_CUR);
|
|
PRInt32 available = PR_Available(mFileDesc);
|
|
PRInt32 fileSize = position + available;
|
|
PRInt32 newPosition = 0;
|
|
switch (whence)
|
|
{
|
|
case PR_SEEK_CUR: newPosition = position + offset; break;
|
|
case PR_SEEK_SET: newPosition = offset; break;
|
|
case PR_SEEK_END: newPosition = fileSize + offset; break;
|
|
}
|
|
if (newPosition < 0)
|
|
{
|
|
newPosition = 0;
|
|
mFailed = PR_TRUE;
|
|
}
|
|
else if (newPosition >= fileSize)
|
|
{
|
|
newPosition = fileSize;
|
|
mEOF = PR_TRUE;
|
|
}
|
|
if (PR_Seek(mFileDesc, newPosition, PR_SEEK_SET) < 0)
|
|
mFailed = PR_TRUE;
|
|
return NS_OK;
|
|
} // FileImpl::Seek
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_IMETHODIMP FileImpl::Tell(PRIntn* outWhere)
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
if (mFileDesc==PR_STDIN || mFileDesc==PR_STDOUT || mFileDesc==PR_STDERR || !mFileDesc)
|
|
return NS_FILE_RESULT(PR_BAD_DESCRIPTOR_ERROR);
|
|
*outWhere = PR_Seek(mFileDesc, 0, PR_SEEK_CUR);
|
|
return NS_OK;
|
|
} // FileImpl::Tell
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_IMETHODIMP FileImpl::Close()
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
if (mFileDesc==PR_STDIN || mFileDesc==PR_STDOUT || mFileDesc==PR_STDERR || !mFileDesc)
|
|
return NS_OK;
|
|
if (PR_Close(mFileDesc) == PR_SUCCESS)
|
|
mFileDesc = 0;
|
|
else
|
|
return NS_FILE_RESULT(PR_GetError());
|
|
return NS_OK;
|
|
} // FileImpl::close
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_IMETHODIMP FileImpl::Flush()
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
#ifdef XP_MAC
|
|
if (mFileDesc == PR_STDOUT || mFileDesc == PR_STDERR)
|
|
{
|
|
cout.flush();
|
|
return NS_OK;
|
|
}
|
|
#endif
|
|
if (!mFileDesc)
|
|
return NS_FILE_RESULT(PR_BAD_DESCRIPTOR_ERROR);
|
|
PRBool itFailed = PR_Sync(mFileDesc) != PR_SUCCESS;
|
|
#ifdef XP_MAC
|
|
// On unix, it seems to fail always.
|
|
if (itFailed)
|
|
mFailed = PR_TRUE;
|
|
#endif
|
|
return NS_OK;
|
|
} // FileImpl::flush
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_BASE nsresult NS_NewTypicalInputFileStream(
|
|
nsISupports** aResult,
|
|
const nsFileSpec& inFile
|
|
/*Default nsprMode == PR_RDONLY*/
|
|
/*Default accessmode = 0700 (octal)*/)
|
|
// Factory method to get an nsInputStream from a file, using most common options
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
return NS_NewIOFileStream(aResult, inFile, PR_RDONLY, 0700);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_BASE nsresult NS_NewOutputConsoleStream(
|
|
nsISupports** aResult)
|
|
// Factory method to get an nsOutputStream to the console.
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
NS_PRECONDITION(aResult != nsnull, "null ptr");
|
|
if (! aResult)
|
|
return NS_ERROR_NULL_POINTER;
|
|
|
|
FileImpl* stream = new FileImpl(PR_STDOUT);
|
|
if (! stream)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
NS_ADDREF(stream);
|
|
*aResult = (nsISupports*)(void*)stream;
|
|
return NS_OK;
|
|
}
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_BASE nsresult NS_NewTypicalOutputFileStream(
|
|
nsISupports** aResult,
|
|
const nsFileSpec& inFile
|
|
/*default nsprMode= (PR_WRONLY | PR_CREATE_FILE | PR_TRUNCATE)*/
|
|
/*Default accessMode= 0700 (octal)*/)
|
|
// Factory method to get an nsOutputStream to a file - most common case.
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
return NS_NewIOFileStream(
|
|
aResult,
|
|
inFile,
|
|
(PR_WRONLY | PR_CREATE_FILE | PR_TRUNCATE),
|
|
0700);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_BASE nsresult NS_NewIOFileStream(
|
|
nsISupports** aResult,
|
|
const nsFileSpec& inFile,
|
|
PRInt32 nsprMode /*default = (PR_WRONLY | PR_CREATE_FILE | PR_TRUNCATE)*/,
|
|
PRInt32 accessMode /*Default = 0700 (octal)*/)
|
|
// Factory method to get an object that implements both nsIInputStream
|
|
// and nsIOutputStream, associated with a file.
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
NS_PRECONDITION(aResult != nsnull, "null ptr");
|
|
if (!aResult)
|
|
return NS_ERROR_NULL_POINTER;
|
|
|
|
FileImpl* stream = new FileImpl(inFile, nsprMode, accessMode);
|
|
if (! stream)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
NS_ADDREF(stream);
|
|
*aResult = (nsISupports*)(void*)stream;
|
|
return NS_OK;
|
|
}
|
|
|
|
//----------------------------------------------------------------------------------------
|
|
NS_BASE nsresult NS_NewTypicalIOFileStream(
|
|
nsISupports** aResult,
|
|
const nsFileSpec& inFile
|
|
/*default nsprMode= (PR_RDWR | PR_CREATE_FILE)*/
|
|
/*Default accessMode= 0700 (octal)*/)
|
|
// Factory method to get an object that implements both nsIInputStream
|
|
// and nsIOutputStream, associated with a single file.
|
|
//----------------------------------------------------------------------------------------
|
|
{
|
|
return NS_NewIOFileStream(
|
|
aResult,
|
|
inFile,
|
|
(PR_RDWR | PR_CREATE_FILE),
|
|
0700);
|
|
}
|