Fix: Fix memleak when exiting method by exception

code/Common/Importer.cpp

@@ -482,24 +482,19 @@ bool Importer::ValidateFlags(unsigned int pFlags) const {
 }
 
 // ------------------------------------------------------------------------------------------------
-const aiScene* Importer::ReadFileFromMemory( const void* pBuffer,
+const aiScene* Importer::ReadFileFromMemory(const void* pBuffer, size_t pLength, unsigned int pFlags, const char* pHint ) {
-    size_t pLength,
-    unsigned int pFlags,
-    const char* pHint /*= ""*/) {
     ai_assert(nullptr != pimpl);
 
-    ASSIMP_BEGIN_EXCEPTION_REGION();
+    IOSystem* io = pimpl->mIOHandler;
-    if (!pHint) {
+    try {
+        if (pHint == nullptr) {
             pHint = "";
         }
-
         if (!pBuffer || !pLength || strlen(pHint) > MaxLenHint ) {
             pimpl->mErrorString = "Invalid parameters passed to ReadFileFromMemory()";
             return nullptr;
         }
-
         // prevent deletion of the previous IOHandler
-    IOSystem* io = pimpl->mIOHandler;
         pimpl->mIOHandler = nullptr;
 
         SetIOHandler(new MemoryIOSystem((const uint8_t*)pBuffer,pLength,io));
@@ -511,8 +506,19 @@ const aiScene* Importer::ReadFileFromMemory( const void* pBuffer,
 
         ReadFile(fbuff,pFlags);
         SetIOHandler(io);
+    } catch(const DeadlyImportError &e) {
+        pimpl->mErrorString = e.what();
+        pimpl->mException = std::current_exception();
+        SetIOHandler(io);
+        return ExceptionSwallower<const aiScene*>()();                                                                                                    \
+    } catch(...) {
+        pimpl->mErrorString = "Unknown exception";
+        pimpl->mException = std::current_exception();
+        SetIOHandler(io);
+        return ExceptionSwallower<const aiScene*>()();                                                                                                    \
+
+    }
 
-    ASSIMP_END_EXCEPTION_REGION_WITH_ERROR_STRING(const aiScene*, pimpl->mErrorString, pimpl->mException);
     return pimpl->mScene;
 }
 

fuzz/assimp_fuzzer.cc

@@ -3,7 +3,7 @@
 Open Asset Import Library (assimp)
 ---------------------------------------------------------------------------
 
-Copyright (c) 2006-2020, assimp team
+Copyright (c) 2006-2023, assimp team
 
 All rights reserved.
 
@@ -57,3 +57,4 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t dataSize) {
 
     return 0;
 }
+

include/assimp/MemoryIOWrapper.h

@@ -4,7 +4,6 @@ Open Asset Import Library (assimp)
 
 Copyright (c) 2006-2022, assimp team
 
-
 All rights reserved.
 
 Redistribution and use of this software in source and binary forms,
@@ -153,11 +152,7 @@ private:
 class MemoryIOSystem : public IOSystem {
 public:
     /** Constructor. */
-    MemoryIOSystem(const uint8_t* buff, size_t len, IOSystem* io)
+    MemoryIOSystem(const uint8_t* buff, size_t len, IOSystem* io) : buffer(buff), length(len), existing_io(io) {
-    : buffer(buff)
-    , length(len)
-    , existing_io(io)
-    , created_streams() {
         // empty
     }
 
@@ -167,6 +162,7 @@ public:
     // -------------------------------------------------------------------
     /** Tests for the existence of a file at the given path. */
     bool Exists(const char* pFile) const override {
+        printf("Exists\n");
         if (0 == strncmp( pFile, AI_MEMORYIO_MAGIC_FILENAME, AI_MEMORYIO_MAGIC_FILENAME_LENGTH ) ) {
             return true;
         }
@@ -187,7 +183,7 @@ public:
             created_streams.emplace_back(new MemoryIOStream(buffer, length));
             return created_streams.back();
         }
-        return existing_io ? existing_io->Open(pFile, pMode) : NULL;
+        return existing_io ? existing_io->Open(pFile, pMode) : nullptr;
     }
 
     // -------------------------------------------------------------------
@@ -246,4 +242,4 @@ private:
 
 } // end namespace Assimp
 
-#endif
+#endif // AI_MEMORYIOSTREAM_H_INC