fix use-after-free in the IModelsArchiveReader class

The implementation has also been moved into a cpp file.

1 files changed, 15 insertions, 0 deletions

diff --git a/library/cpp/archive/models_archive_reader.cpp b/library/cpp/archive/models_archive_reader.cpp
new file mode 100644
index 0000000000..abbb91d628
--- /dev/null
+++ b/library/cpp/archive/models_archive_reader.cpp
@@ -0,0 +1,15 @@
+#include "models_archive_reader.h"
+
+#include <util/generic/hash_set.h>
+
+THashSet<TStringBuf> IModelsArchiveReader::FilterByPrefix(TStringBuf prefix, TStringBuf suffix) const {
+    THashSet<TStringBuf> result;
+    const size_t count = Count();
+    for (size_t ind = 0; ind < count; ++ind) {
+        TString path = KeyByIndex(ind);
+        if (path.StartsWith(prefix) && path.EndsWith(suffix)) {
+            result.insert(std::move(path));
+        }
+    }
+    return result;
+}