Zengtudor/ccls
1
0
Fork 0
mirror of https://github.com/MaskRay/ccls.git synced 2025-02-21 16:09:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Convert GetFilesInFolderHelper to use BFS (#268)

Browse Source 
This function has a huge stack frame (> 8k, thanks to tinydir), and is
called recursively. I have seen this function causing some stack
overflows. So convert it to BFS to avoid that.
This commit is contained in:
yshui 2018-01-08 17:09:19 +00:00 committed by Fangrui Song
parent 0ef5aec3fc
commit b644b54614
1 changed files with 39 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/utils.cc
View File

@ -140,8 +140,11 @@ static void GetFilesInFolderHelper(
bool recursive,
std::string output_prefix,
const std::function<void(const std::string&)>& handler) {
std::queue<std::pair<std::string, std::string>> q;
q.push(make_pair(folder, output_prefix));
while(!q.empty()) {
tinydir_dir dir;
if (tinydir_open(&dir, folder.c_str()) == -1) {
if (tinydir_open(&dir, q.front().first.c_str()) == -1) {
LOG_S(WARNING) << "Unable to open directory " << folder;
goto bail;
}
@ -166,11 +169,10 @@ static void GetFilesInFolderHelper(
if (recursive) {
std::string child_dir = output_prefix + file.name + "/";
if (!IsSymLink(child_dir))
GetFilesInFolderHelper(file.path, true /*recursive*/, child_dir,
handler);
q.push(make_pair(file.path, child_dir));
}
} else {
handler(output_prefix + file.name);
handler(q.front().second + file.name);
}
}
@ -181,8 +183,10 @@ static void GetFilesInFolderHelper(
}
}
bail:
bail:
tinydir_close(&dir);
q.pop();
}
}
std::vector<std::string> GetFilesInFolder(std::string folder,