intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 123 insertions, 0 deletions

diff --git a/util/folder/fts_ut.cpp b/util/folder/fts_ut.cpp
new file mode 100644
index 0000000000..c5d59e35f4
--- /dev/null
+++ b/util/folder/fts_ut.cpp
@@ -0,0 +1,123 @@
+#include "fts.h"
+#include "dirut.h"
+#include "tempdir.h"
+
+#include <library/cpp/testing/unittest/registar.h>
+#include <library/cpp/threading/future/async.h>
+
+#include <util/system/file.h>
+#include <util/system/tempfile.h>
+#include <util/generic/string.h>
+
+class TFtsTest: public TTestBase {
+    UNIT_TEST_SUITE(TFtsTest);
+    UNIT_TEST(TestSimple);
+    UNIT_TEST(TestNoLeakChangingAccessToFolder);
+    UNIT_TEST_SUITE_END();
+
+public:
+    void TestSimple();
+    void TestNoLeakChangingAccessToFolder();
+};
+
+void MakeFile(const char* path) {
+    TFile(path, CreateAlways);
+}
+
+//There potentially could be problems in listing order on different platforms
+int FtsCmp(const FTSENT** ent1, const FTSENT** ent2) {
+    return strcmp((*ent1)->fts_accpath, (*ent2)->fts_accpath);
+}
+
+void CheckEnt(FTSENT* ent, const char* name, int type) {
+    UNIT_ASSERT(ent);
+    UNIT_ASSERT_STRINGS_EQUAL(ent->fts_path, name);
+    UNIT_ASSERT_EQUAL(ent->fts_info, type);
+}
+
+class TFileTree {
+public:
+    TFileTree(char* const* argv, int options, int (*compar)(const FTSENT**, const FTSENT**)) {
+        Fts_ = yfts_open(argv, options, compar);
+    }
+
+    ~TFileTree() {
+        yfts_close(Fts_);
+    }
+
+    FTS* operator()() {
+        return Fts_;
+    }
+
+private:
+    FTS* Fts_;
+};
+
+void TFtsTest::TestSimple() {
+    const char* dotPath[2] = {"." LOCSLASH_S, nullptr};
+    TFileTree currentDirTree((char* const*)dotPath, 0, FtsCmp);
+    UNIT_ASSERT(currentDirTree());
+    TTempDir tempDir = MakeTempName(yfts_read(currentDirTree())->fts_path);
+    MakeDirIfNotExist(tempDir().data());
+    MakeDirIfNotExist((tempDir() + LOCSLASH_S "dir1").data());
+    MakeFile((tempDir() + LOCSLASH_S "dir1" LOCSLASH_S "file1").data());
+    MakeFile((tempDir() + LOCSLASH_S "dir1" LOCSLASH_S "file2").data());
+    MakeDirIfNotExist((tempDir() + LOCSLASH_S "dir2").data());
+    MakeFile((tempDir() + LOCSLASH_S "dir2" LOCSLASH_S "file3").data());
+    MakeFile((tempDir() + LOCSLASH_S "dir2" LOCSLASH_S "file4").data());
+
+    const char* path[2] = {tempDir().data(), nullptr};
+    TFileTree fileTree((char* const*)path, 0, FtsCmp);
+    UNIT_ASSERT(fileTree());
+    CheckEnt(yfts_read(fileTree()), tempDir().data(), FTS_D);
+    CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir1").data(), FTS_D);
+    CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir1" LOCSLASH_S "file1").data(), FTS_F);
+    CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir1" LOCSLASH_S "file2").data(), FTS_F);
+    CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir1").data(), FTS_DP);
+    CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir2").data(), FTS_D);
+    CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir2" LOCSLASH_S "file3").data(), FTS_F);
+    CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir2" LOCSLASH_S "file4").data(), FTS_F);
+    CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir2").data(), FTS_DP);
+    CheckEnt(yfts_read(fileTree()), (tempDir()).data(), FTS_DP);
+    UNIT_ASSERT_EQUAL(yfts_read(fileTree()), nullptr);
+}
+
+class TTempDirWithLostAccess: public TTempDir {
+public:
+    ~TTempDirWithLostAccess() {
+        chmod(Name().data(), 0777);
+    }
+};
+
+// https://st.yandex-team.ru/YQ-318
+// Test that detects memory leak in case of error in chdir in fts_build function.
+void TFtsTest::TestNoLeakChangingAccessToFolder() {
+    TTempDirWithLostAccess tempDir;
+    TString tmpPath = tempDir();
+    if (tmpPath.EndsWith(LOCSLASH_S)) {
+        tmpPath.resize(tmpPath.size() - 1);
+    }
+    MakeDirIfNotExist((tmpPath + LOCSLASH_S + "subdir").data());
+
+    const char* path[2] = {tmpPath.data(), nullptr};
+    TFileTree fileTree((char* const*)path, FTS_SEEDOT, FtsCmp);
+    UNIT_ASSERT(fileTree());
+
+    CheckEnt(yfts_read(fileTree()), tmpPath.data(), FTS_D);
+#ifndef _win32_
+    CheckEnt(yfts_read(fileTree()), (tmpPath + LOCSLASH_S ".").data(), FTS_DOT);
+#endif // _win32_
+    CheckEnt(yfts_read(fileTree()), (tmpPath + LOCSLASH_S "..").data(), FTS_DOT);
+    CheckEnt(yfts_read(fileTree()), (tmpPath + LOCSLASH_S "subdir").data(), FTS_D);
+    auto pool = CreateThreadPool(2);
+    auto chmodFuture = NThreading::Async([name = tmpPath] {
+        UNIT_ASSERT_C(!chmod(name.data(), 0), "Errno: " << errno);
+    }, *pool);
+    auto childrenFuture = NThreading::Async([&] {
+        yfts_children(fileTree(), 0);
+    }, *pool);
+    childrenFuture.Wait();
+    chmodFuture.Wait();
+}
+
+UNIT_TEST_SUITE_REGISTRATION(TFtsTest);