📄 bufdataf.cpp
字号:
/* ***** BEGIN LICENSE BLOCK *****
* Version: RCSL 1.0/RPSL 1.0
*
* Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved.
*
* The contents of this file, and the files included with this file, are
* subject to the current version of the RealNetworks Public Source License
* Version 1.0 (the "RPSL") available at
* http://www.helixcommunity.org/content/rpsl unless you have licensed
* the file under the RealNetworks Community Source License Version 1.0
* (the "RCSL") available at http://www.helixcommunity.org/content/rcsl,
* in which case the RCSL will apply. You may also obtain the license terms
* directly from RealNetworks. You may not use this file except in
* compliance with the RPSL or, if you have a valid RCSL with RealNetworks
* applicable to this file, the RCSL. Please see the applicable RPSL or
* RCSL for the rights, obligations and limitations governing use of the
* contents of the file.
*
* This file is part of the Helix DNA Technology. RealNetworks is the
* developer of the Original Code and owns the copyrights in the portions
* it created.
*
* This file, and the files included with this file, is distributed and made
* available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
* EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES,
* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
*
* Technology Compatibility Kit Test Suite(s) Location:
* http://www.helixcommunity.org/content/tck
*
* Contributor(s):
*
* ***** END LICENSE BLOCK ***** */
#include "hxtypes.h"
#include "hlxclib/stdio.h"
#if defined(_UNIX)
#include <unistd.h>
#endif // _UNIX
#include "hlxclib/windows.h"
#include "hlxclib/sys/stat.h"
#include "hlxclib/sys/types.h"
#include "hlxclib/fcntl.h"
#include "hlxclib/errno.h"
#include "hlxosstr.h"
#include "hxcom.h"
#include "hxresult.h"
#include "ihxpckts.h"
#include "hxbuffer.h"
#include "debug.h"
#include "hxdataf.h"
#include "bufdataf.h"
/////////////////////////////////////////////////////////////////////////
// Method:
// BufferedDataFile::QueryInterface
// Purpose:
// Implement this to export the interfaces supported by your
// object.
//
STDMETHODIMP
BufferedDataFile::QueryInterface(REFIID riid, void** ppvObj)
{
if (IsEqualIID(riid, IID_IHXDataFile))
{
AddRef();
*ppvObj = (IHXDataFile*)this;
return HXR_OK;
}
*ppvObj = NULL;
return HXR_NOINTERFACE;
}
/////////////////////////////////////////////////////////////////////////
// Method:
// BufferedDataFile::AddRef
// Purpose:
// Everyone usually implements this the same... feel free to use
// this implementation.
//
STDMETHODIMP_(ULONG32)
BufferedDataFile::AddRef()
{
DPRINTF(0x5d000000, ("UBDF::AddRef() = %ld\n", m_lRefCount+1));
return InterlockedIncrement(&m_lRefCount);
}
/////////////////////////////////////////////////////////////////////////
// Method:
// BufferedDataFile::Release
// Purpose:
// Everyone usually implements this the same... feel free to use
// this implementation.
//
STDMETHODIMP_(ULONG32)
BufferedDataFile::Release()
{
DPRINTF(0x5d000000, ("UBDF::Release() = %ld\n", m_lRefCount-1));
if (InterlockedDecrement(&m_lRefCount) > 0)
{
return m_lRefCount;
}
delete this;
return 0;
}
BufferedDataFile::BufferedDataFile()
: m_lRefCount(0)
, m_ulLastError(HXR_OK)
, m_pFilename(new CHXBuffer)
, m_pFile(0)
{
m_pFilename->AddRef();
DPRINTF(0x5d000000, ("BufferedDataFile::BufferedDataFile()\n"));
}
// ~CHXFile should close the file if it is open
BufferedDataFile::~BufferedDataFile()
{
// close the file if it is open
if (m_pFile)
{
fclose(m_pFile);
m_pFile = 0;
}
HX_RELEASE(m_pFilename);
DPRINTF(0x5d000000, ("BufferedDataFile::~BufferedDataFile()\n"));
}
/*
* IHXDataFile methods
*/
/* Bind DataFile Object with FileName */
STDMETHODIMP_(void)
BufferedDataFile::Bind(const char* pFilename)
{
m_pFilename->Set((BYTE *)pFilename, strlen(pFilename)+1);
DPRINTF(0x5d000000, ("BufferedDataFile::Bind(%s)\n",
(const char *)m_pFilename->GetBuffer()));
}
/* Creates a datafile using the specified mode
* uOpenMode --- File Open mode - HX_FILEFLAG_READ/HX_FILEFLAG_WRITE/HX_FILEFLAG_BINARY
*/
STDMETHODIMP
BufferedDataFile::Open(UINT16 uOpenMode)
{
return _Create(uOpenMode, FALSE);
}
STDMETHODIMP
BufferedDataFile::Create(UINT16 uOpenMode)
{
return _Create(uOpenMode, TRUE);
}
/* Open will open a file with the specified mode
*/
HX_RESULT
BufferedDataFile::_Create(UINT16 uOpenMode, BOOL bCreateFile)
{
HX_RESULT retVal = HXR_OK;
m_ulLastError = HXR_OK;
DPRINTF(0x5d000000, ("BufferedDataFile::Open()\n"));
char modeflags[4]; /* Flawfinder: ignore */
// close previous file if necessary
if (m_pFile)
{
fclose(m_pFile);
m_pFile = NULL;
}
modeflags[0] = 'r';
modeflags[1] = 0;
modeflags[2] = 0;
#ifdef HELIX_CONFIG_ALLFILES_BINARY
uOpenMode |= HX_FILEFLAG_BINARY;
#endif // HELIX_CONFIG_ALLFILES_BINARY
if (((uOpenMode & HX_FILEFLAG_NOTRUNC) == 0) && bCreateFile)
{
// Overwrite any existing file if in the way
modeflags[0] = 'w';
if (uOpenMode & HX_FILEFLAG_READ)
{
modeflags[1] = '+';
}
}
else
{
modeflags[0] = 'r';
if (uOpenMode & HX_FILEFLAG_WRITE)
{
modeflags[1] = '+';
}
}
if (uOpenMode & HX_FILEFLAG_BINARY)
{
strcat(modeflags, "b"); /* Flawfinder: ignore */
}
// open file
if ((m_pFilename == NULL) ||
(m_pFilename->GetSize() == 0))
{
retVal = HXR_FAIL;
}
if (retVal == HXR_OK)
{
m_pFile = fopen((const char *) m_pFilename->GetBuffer(), modeflags);
if (m_pFile)
{
if (bCreateFile && (uOpenMode & HX_FILEFLAG_NOTRUNC))
{
fclose(m_pFile);
m_pFile = NULL;
retVal = HXR_FAIL;
m_ulLastError = (UINT32)HXR_FAIL;
}
}
else
{
retVal = HXR_DOC_MISSING;
m_ulLastError = (UINT32)HXR_DOC_MISSING;
}
}
if (retVal == HXR_DOC_MISSING)
{
if (bCreateFile && (uOpenMode & HX_FILEFLAG_NOTRUNC))
{
// This a recovery on NO_TRUNC file creation
// Open failed so attempt to create file
modeflags[0] = 'w';
modeflags[1] = 0;
if (uOpenMode && HX_FILEFLAG_READ)
{
modeflags[1] = '+';
modeflags[2] = 0;
}
if (uOpenMode & HX_FILEFLAG_BINARY)
{
strcat(modeflags, "b"); /* Flawfinder: ignore */
}
m_pFile = fopen((const char *) m_pFilename->GetBuffer(), modeflags);
if (m_pFile)
{
retVal = m_ulLastError = HXR_OK;
}
else
{
retVal = HXR_FAIL;
m_ulLastError = (UINT32)retVal;
}
}
}
DPRINTF(0x5d000000, ("BDF::Open() -- %s\n", modeflags));
return retVal;
}
/* Close closes a file
* If the reference count on the IHXDataFile object is greater than 1,
* then the underlying file cannot be safely closed, so Close() becomes
* a noop in that case. Close it only when the object is destroyed.
* This would be safe, but could lead to a file descriptor leak.
*/
STDMETHODIMP
BufferedDataFile::Close()
{
if (m_pFile)
{
fclose(m_pFile);
m_pFile = 0;
}
return HXR_OK;
}
/* Name returns the currently bound file name in FileName.
* and returns TRUE, if the a name has been bound. Otherwise
* FALSE is returned.
*/
STDMETHODIMP_(BOOL)
BufferedDataFile::Name(REF(IHXBuffer*) pFileName)
{
BOOL bRetVal = FALSE;
if (m_pFilename && (m_pFilename->GetSize() != 0))
{
pFileName = m_pFilename;
m_pFilename->AddRef();
bRetVal = TRUE;
}
return bRetVal;
}
/*
* IsOpen returns TRUE if file is open. Otherwise FALSE.
*/
inline BOOL
BufferedDataFile::IsOpen()
{
return (m_pFile ? TRUE : FALSE);
}
/* Seek moves the current file position to the offset from the
* fromWhere specifier returns current position of file or -1 on
* error.
*/
STDMETHODIMP
BufferedDataFile::Seek(ULONG32 offset, UINT16 fromWhere)
{
if (m_pFile)
{
m_ulLastError = HXR_OK;
if (fseek(m_pFile, offset, fromWhere) < 0)
{
m_ulLastError = errno;
return HXR_INVALID_FILE;
}
return HXR_OK;
}
return HXR_INVALID_FILE;
}
/* Tell returns the current file position in the file */
STDMETHODIMP_(ULONG32)
BufferedDataFile::Tell()
{
INT32 offset = -1;
m_ulLastError = (UINT32)HXR_FAIL;
if (m_pFile)
{
// so we do this instead....
if ((offset = ftell(m_pFile)) != -1)
{
m_ulLastError = HXR_OK;
}
}
return (ULONG32) offset;
}
/* Read reads up to count bytes of data into buf.
* returns the number of bytes read, EOF, or -1 if the read failed
*/
STDMETHODIMP_(ULONG32)
BufferedDataFile::Read(REF(IHXBuffer*) pOutBuf, ULONG32 count)
{
int ncnt = 0; // number of bytes read
IHXBuffer* pBuf = NULL;
if (m_pFile)
{
if (pOutBuf)
{
pBuf = pOutBuf;
}
else
{
m_ulLastError = (UINT32)HXR_OUTOFMEMORY;
pBuf = new CHXBuffer();
}
if (pBuf)
{
pBuf->AddRef();
m_ulLastError = pBuf->SetSize(count);
}
if (m_ulLastError == HXR_OK)
{
Tell();
}
if (m_ulLastError == HXR_OK)
{
if ((ncnt = fread((void *)pBuf->GetBuffer(), sizeof(char),
count, m_pFile)) < (int)count)
{
m_ulLastError = ferror(m_pFile) ? HXR_FAIL : HXR_OK;
}
if (ncnt < (int)count)
{
pBuf->SetSize(ncnt);
}
if (pBuf != pOutBuf)
{
pOutBuf = pBuf;
pBuf->AddRef();
}
}
if (pBuf)
{
pBuf->Release();
}
}
return (ULONG32)ncnt;
}
/* Write writes up to count bytes of data from buf.
* returns the number of bytes written, or -1 if the write failed
*/
STDMETHODIMP_(ULONG32)
BufferedDataFile::Write(REF(IHXBuffer *) pBuf)
{
int ncnt = -1; // number of bytes written
if (m_pFile)
{
m_ulLastError = HXR_OK;
if ((ncnt = fwrite(pBuf->GetBuffer(), sizeof(char),
pBuf->GetSize(), m_pFile)) < (int)(pBuf->GetSize()))
{
m_ulLastError = (UINT32)HXR_FAIL;
}
}
return (ULONG32)ncnt;
}
/* Flush out the data in case of Buffered I/O
*/
STDMETHODIMP
BufferedDataFile::Flush()
{
if (m_pFile)
{
m_ulLastError = HXR_OK;
if (fflush(m_pFile) == 0)
{
return HXR_OK;
}
m_ulLastError = (UINT32)HXR_FAIL;
return HXR_FAIL;
}
return HXR_INVALID_FILE;
}
/*
* Return info about the data file such as permissions, time of creation
* size in bytes, etc.
*/
STDMETHODIMP
BufferedDataFile::Stat(struct stat* statbuf)
{
#ifndef WIN32_PLATFORM_PSPC
if (m_pFile)
{
if (!fstat(fileno(m_pFile), statbuf))
return HXR_OK;
} else
#endif
if (m_pFilename->GetSize())
{
if (!stat((const char *)m_pFilename->GetBuffer(), statbuf))
return HXR_OK;
}
return HXR_FAIL;
}
/* Return the file descriptor */
inline INT16
BufferedDataFile::GetFd()
{
return m_pFile ? (INT16) fileno(m_pFile) : -1;
}
/* GetLastError returns the platform specific file error */
STDMETHODIMP
BufferedDataFile::GetLastError()
{
return m_ulLastError;
}
/* GetLastError returns the platform specific file error in
* string form.
*/
STDMETHODIMP_(void)
BufferedDataFile::GetLastError(REF(IHXBuffer*) err)
{
}
STDMETHODIMP
BufferedDataFile::Delete()
{
Close();
if ((!m_pFilename) || (m_pFilename->GetSize() == 0))
{
return HXR_UNEXPECTED;
}
#if defined(_WINDOWS)
if (DeleteFile(OS_STRING(m_pFilename->GetBuffer())))
{
return HXR_OK;
}
#elif defined(_UNIX)
if (unlink(OS_STRING(m_pFilename->GetBuffer())) == 0)
{
return HXR_OK;
}
#endif // _UNIX
return HXR_FAIL;
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -