1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
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);
|
