cfileio: fix leaks by not closing files in the destructor
Numerous importers rely on the default C++ IOSystem implementation having the destructor close the file for them. The C IOSystem wrapper wasn't and instead assumed that the Close() method was going to be called. This brings the C IOSystem wrapper in line with the default C++ IOSystem by having the destructor close the file.pull/1037/head
parent
09ad67f469
commit
cbcaa107eb
|
@ -0,0 +1,134 @@
|
||||||
|
/*
|
||||||
|
---------------------------------------------------------------------------
|
||||||
|
Open Asset Import Library (assimp)
|
||||||
|
---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
Copyright (c) 2006-2016, assimp team
|
||||||
|
|
||||||
|
All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use of this software in source and binary forms,
|
||||||
|
with or without modification, are permitted provided that the following
|
||||||
|
conditions are met:
|
||||||
|
|
||||||
|
* Redistributions of source code must retain the above
|
||||||
|
copyright notice, this list of conditions and the
|
||||||
|
following disclaimer.
|
||||||
|
|
||||||
|
* Redistributions in binary form must reproduce the above
|
||||||
|
copyright notice, this list of conditions and the
|
||||||
|
following disclaimer in the documentation and/or other
|
||||||
|
materials provided with the distribution.
|
||||||
|
|
||||||
|
* Neither the name of the assimp team, nor the names of its
|
||||||
|
contributors may be used to endorse or promote products
|
||||||
|
derived from this software without specific prior
|
||||||
|
written permission of the assimp team.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||||
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||||
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||||
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||||
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||||
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||||
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||||
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||||
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||||
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
---------------------------------------------------------------------------
|
||||||
|
*/
|
||||||
|
|
||||||
|
/** @file aiFileIO -> IOSystem wrapper*/
|
||||||
|
|
||||||
|
#include "CInterfaceIOWrapper.h"
|
||||||
|
|
||||||
|
namespace Assimp {
|
||||||
|
|
||||||
|
CIOStreamWrapper::~CIOStreamWrapper(void)
|
||||||
|
{
|
||||||
|
/* Various places depend on this destructor to close the file */
|
||||||
|
if (mFile) {
|
||||||
|
mIO->mFileSystem->CloseProc(mIO->mFileSystem, mFile);
|
||||||
|
mFile = nullptr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
size_t CIOStreamWrapper::Read(void* pvBuffer,
|
||||||
|
size_t pSize,
|
||||||
|
size_t pCount
|
||||||
|
){
|
||||||
|
// need to typecast here as C has no void*
|
||||||
|
return mFile->ReadProc(mFile,(char*)pvBuffer,pSize,pCount);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
size_t CIOStreamWrapper::Write(const void* pvBuffer,
|
||||||
|
size_t pSize,
|
||||||
|
size_t pCount
|
||||||
|
){
|
||||||
|
// need to typecast here as C has no void*
|
||||||
|
return mFile->WriteProc(mFile,(const char*)pvBuffer,pSize,pCount);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
aiReturn CIOStreamWrapper::Seek(size_t pOffset,
|
||||||
|
aiOrigin pOrigin
|
||||||
|
){
|
||||||
|
return mFile->SeekProc(mFile,pOffset,pOrigin);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
size_t CIOStreamWrapper::Tell(void) const {
|
||||||
|
return mFile->TellProc(mFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
size_t CIOStreamWrapper::FileSize() const {
|
||||||
|
return mFile->FileSizeProc(mFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
void CIOStreamWrapper::Flush () {
|
||||||
|
return mFile->FlushProc(mFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
// Custom IOStream implementation for the C-API
|
||||||
|
bool CIOSystemWrapper::Exists( const char* pFile) const {
|
||||||
|
aiFile* p = mFileSystem->OpenProc(mFileSystem,pFile,"rb");
|
||||||
|
if (p){
|
||||||
|
mFileSystem->CloseProc(mFileSystem,p);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
char CIOSystemWrapper::getOsSeparator() const {
|
||||||
|
#ifndef _WIN32
|
||||||
|
return '/';
|
||||||
|
#else
|
||||||
|
return '\\';
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
IOStream* CIOSystemWrapper::Open(const char* pFile,const char* pMode) {
|
||||||
|
aiFile* p = mFileSystem->OpenProc(mFileSystem,pFile,pMode);
|
||||||
|
if (!p) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
return new CIOStreamWrapper(p, this);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ...................................................................
|
||||||
|
void CIOSystemWrapper::Close( IOStream* pFile) {
|
||||||
|
if (!pFile) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
delete pFile;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -50,106 +50,43 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
|
||||||
namespace Assimp {
|
namespace Assimp {
|
||||||
|
|
||||||
|
class CIOSystemWrapper;
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
// Custom IOStream implementation for the C-API
|
// Custom IOStream implementation for the C-API
|
||||||
class CIOStreamWrapper : public IOStream
|
class CIOStreamWrapper : public IOStream
|
||||||
{
|
{
|
||||||
friend class CIOSystemWrapper;
|
|
||||||
public:
|
public:
|
||||||
|
explicit CIOStreamWrapper(aiFile* pFile, CIOSystemWrapper* io)
|
||||||
explicit CIOStreamWrapper(aiFile* pFile)
|
: mFile(pFile),
|
||||||
: mFile(pFile)
|
mIO(io)
|
||||||
{}
|
{}
|
||||||
|
~CIOStreamWrapper(void);
|
||||||
|
|
||||||
// ...................................................................
|
size_t Read(void* pvBuffer, size_t pSize, size_t pCount);
|
||||||
size_t Read(void* pvBuffer,
|
size_t Write(const void* pvBuffer, size_t pSize, size_t pCount);
|
||||||
size_t pSize,
|
aiReturn Seek(size_t pOffset, aiOrigin pOrigin);
|
||||||
size_t pCount
|
size_t Tell(void) const;
|
||||||
){
|
size_t FileSize() const;
|
||||||
// need to typecast here as C has no void*
|
void Flush();
|
||||||
return mFile->ReadProc(mFile,(char*)pvBuffer,pSize,pCount);
|
|
||||||
}
|
|
||||||
|
|
||||||
// ...................................................................
|
|
||||||
size_t Write(const void* pvBuffer,
|
|
||||||
size_t pSize,
|
|
||||||
size_t pCount
|
|
||||||
){
|
|
||||||
// need to typecast here as C has no void*
|
|
||||||
return mFile->WriteProc(mFile,(const char*)pvBuffer,pSize,pCount);
|
|
||||||
}
|
|
||||||
|
|
||||||
// ...................................................................
|
|
||||||
aiReturn Seek(size_t pOffset,
|
|
||||||
aiOrigin pOrigin
|
|
||||||
){
|
|
||||||
return mFile->SeekProc(mFile,pOffset,pOrigin);
|
|
||||||
}
|
|
||||||
|
|
||||||
// ...................................................................
|
|
||||||
size_t Tell(void) const {
|
|
||||||
return mFile->TellProc(mFile);
|
|
||||||
}
|
|
||||||
|
|
||||||
// ...................................................................
|
|
||||||
size_t FileSize() const {
|
|
||||||
return mFile->FileSizeProc(mFile);
|
|
||||||
}
|
|
||||||
|
|
||||||
// ...................................................................
|
|
||||||
void Flush () {
|
|
||||||
return mFile->FlushProc(mFile);
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
aiFile* mFile;
|
aiFile* mFile;
|
||||||
|
CIOSystemWrapper* mIO;
|
||||||
};
|
};
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
|
||||||
// Custom IOStream implementation for the C-API
|
|
||||||
class CIOSystemWrapper : public IOSystem
|
class CIOSystemWrapper : public IOSystem
|
||||||
{
|
{
|
||||||
|
friend class CIOStreamWrapper;
|
||||||
public:
|
public:
|
||||||
explicit CIOSystemWrapper(aiFileIO* pFile)
|
explicit CIOSystemWrapper(aiFileIO* pFile)
|
||||||
: mFileSystem(pFile)
|
: mFileSystem(pFile)
|
||||||
{}
|
{}
|
||||||
|
|
||||||
// ...................................................................
|
bool Exists( const char* pFile) const;
|
||||||
bool Exists( const char* pFile) const {
|
char getOsSeparator() const;
|
||||||
aiFile* p = mFileSystem->OpenProc(mFileSystem,pFile,"rb");
|
IOStream* Open(const char* pFile,const char* pMode = "rb");
|
||||||
if (p){
|
void Close( IOStream* pFile);
|
||||||
mFileSystem->CloseProc(mFileSystem,p);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// ...................................................................
|
|
||||||
char getOsSeparator() const {
|
|
||||||
#ifndef _WIN32
|
|
||||||
return '/';
|
|
||||||
#else
|
|
||||||
return '\\';
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
// ...................................................................
|
|
||||||
IOStream* Open(const char* pFile,const char* pMode = "rb") {
|
|
||||||
aiFile* p = mFileSystem->OpenProc(mFileSystem,pFile,pMode);
|
|
||||||
if (!p) {
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
return new CIOStreamWrapper(p);
|
|
||||||
}
|
|
||||||
|
|
||||||
// ...................................................................
|
|
||||||
void Close( IOStream* pFile) {
|
|
||||||
if (!pFile) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
mFileSystem->CloseProc(mFileSystem,((CIOStreamWrapper*) pFile)->mFile);
|
|
||||||
delete pFile;
|
|
||||||
}
|
|
||||||
private:
|
private:
|
||||||
aiFileIO* mFileSystem;
|
aiFileIO* mFileSystem;
|
||||||
};
|
};
|
||||||
|
|
|
@ -128,6 +128,7 @@ SET( Common_SRCS
|
||||||
DefaultIOStream.h
|
DefaultIOStream.h
|
||||||
DefaultIOSystem.cpp
|
DefaultIOSystem.cpp
|
||||||
DefaultIOSystem.h
|
DefaultIOSystem.h
|
||||||
|
CInterfaceIOWrapper.cpp
|
||||||
CInterfaceIOWrapper.h
|
CInterfaceIOWrapper.h
|
||||||
Hash.h
|
Hash.h
|
||||||
Importer.cpp
|
Importer.cpp
|
||||||
|
|
Loading…
Reference in New Issue