averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[lldb-moduleimport-test] Validate input to prevent crashes.

Browse Source
This commit is contained in:
Davide Italiano
2018-04-17 11:22:15 -07:00
parent 36c4205e96
commit 5c99c35d24
2 changed files with 38 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tools/lldb-moduleimport-test/lldb-moduleimport-test.cpp
View File

@@ -199,6 +199,28 @@ int main(int argc, char **argv) {
DumpTypeFromMangled.removeArgument();
InputNames.removeArgument();
auto validateInputFile = [](std::string Filename) {
if (Filename.empty())
return true;
if (!llvm::sys::fs::exists(llvm::Twine(Filename))) {
llvm::errs() << Filename << " does not exists, exiting.\n";
return false;
}
if (!llvm::sys::fs::is_regular_file(llvm::Twine(Filename))) {
llvm::errs() << Filename << " is not a regular file, exiting.\n";
return false;
}
return true;
};
for (auto &InputFilename : InputNames)
if (!validateInputFile(InputFilename))
return 1;
if (!validateInputFile(DumpTypeFromMangled))
return 1;
if (!validateInputFile(DumpDeclFromMangled))
return 1;
// Fetch the serialized module bitstreams from the Mach-O files and
// register them with the module loader.
llvm::SmallVector<std::pair<char *, uint64_t>, 8> Modules;