Cleanup RAM allocated by concatenated files.

jbenden · May 9, 2018 · d9d3c84 · d9d3c84
1 parent 86e7812
commit d9d3c84
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/include/u-lang/Basic/VirtualFileSystem.hpp b/include/u-lang/Basic/VirtualFileSystem.hpp
@@ -471,9 +471,13 @@ class ConcatenatedOverlayFileSystem : public FileSystem
   /// their addition.
   FileSystemList FSList;
 
+  std::vector<void*> Cleanup;
+
 public:
   explicit ConcatenatedOverlayFileSystem(IntrusiveRefCntPtr<FileSystem> Base);
 
+  ~ConcatenatedOverlayFileSystem();
+
   /// \brief Pushes a file system on top of the stack.
   void pushOverlay(IntrusiveRefCntPtr<FileSystem> FS);
 

diff --git a/lib/Basic/VirtualFileSystem.cpp b/lib/Basic/VirtualFileSystem.cpp
@@ -932,6 +932,14 @@ ConcatenatedOverlayFileSystem::ConcatenatedOverlayFileSystem(IntrusiveRefCntPtr<
   FSList.push_back(std::move(BaseFS));
 }
 
+ConcatenatedOverlayFileSystem::~ConcatenatedOverlayFileSystem()
+{
+  for (auto item : Cleanup)
+  {
+    delete reinterpret_cast<detail::InMemoryFile*>(item);
+  }
+}
+
 void
 ConcatenatedOverlayFileSystem::pushOverlay(IntrusiveRefCntPtr<FileSystem> FS)
 {
@@ -1018,6 +1026,8 @@ ConcatenatedOverlayFileSystem::openFileForRead(const llvm::Twine& Path)
 
   auto* F = new detail::InMemoryFile(std::move(Stat), std::move(memoryBuffer));
 
+  Cleanup.push_back(F);
+
   return std::unique_ptr<File>(new detail::InMemoryFileAdaptor(*F));
 }