mirror of
https://github.com/github/codeql.git
synced 2025-12-17 17:23:36 +01:00
521e6235b5
This replaces usages of `llvm::fs` and string manipulation with `std::filesystem`, also replacing `std::string` with `std::filesystem::path` where it made sense. Moreover MD5 hashing used in macOS file remapping was replaced by SHA256 hashing using a small header-only SHA256 C++ library with an MIT license, https://github.com/okdshin/PicoSHA2. File contents hashing was relocated to the newly created `file` library for later planned reuse.
86 lines
2.1 KiB
C++
86 lines
2.1 KiB
C++
#include "swift/extractor/remapping/SwiftOpenInterception.h"
|
|
|
|
#include <fishhook.h>
|
|
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <filesystem>
|
|
|
|
#include "swift/extractor/infra/file/FileHash.h"
|
|
|
|
namespace fs = std::filesystem;
|
|
|
|
namespace codeql {
|
|
|
|
static fs::path scratchDir;
|
|
static bool interceptionEnabled = false;
|
|
|
|
static int (*original_open)(const char*, int, ...) = nullptr;
|
|
|
|
static std::string originalHashFile(const fs::path& filename) {
|
|
int fd = original_open(filename.c_str(), O_RDONLY);
|
|
if (fd == -1) {
|
|
return {};
|
|
}
|
|
|
|
return hashFile(fd);
|
|
}
|
|
|
|
static int codeql_open(const char* path, int oflag, ...) {
|
|
va_list ap = {0};
|
|
mode_t mode = 0;
|
|
if ((oflag & O_CREAT) != 0) {
|
|
// mode only applies to O_CREAT
|
|
va_start(ap, oflag);
|
|
mode = va_arg(ap, int);
|
|
va_end(ap);
|
|
}
|
|
|
|
fs::path newPath(path);
|
|
|
|
if (interceptionEnabled && fs::exists(newPath)) {
|
|
// TODO: check file magic instead
|
|
if (newPath.extension() == ".swiftmodule") {
|
|
auto hash = originalHashFile(newPath);
|
|
auto hashed = scratchDir / hash;
|
|
if (!hash.empty() && fs::exists(hashed)) {
|
|
newPath = hashed;
|
|
}
|
|
}
|
|
}
|
|
|
|
return original_open(newPath.c_str(), oflag, mode);
|
|
}
|
|
|
|
void finalizeRemapping(
|
|
const std::unordered_map<std::filesystem::path, std::filesystem::path>& mapping) {
|
|
for (auto& [original, patched] : mapping) {
|
|
// TODO: Check file magic instead
|
|
if (original.extension() != ".swiftmodule") {
|
|
continue;
|
|
}
|
|
auto hash = originalHashFile(original);
|
|
auto hashed = scratchDir / hash;
|
|
if (!hash.empty() && fs::exists(hashed)) {
|
|
std::error_code ec;
|
|
fs::create_symlink(/* target */ patched, /* symlink */ hashed, ec);
|
|
if (ec) {
|
|
std::cerr << "Cannot remap file '" << patched << "' -> '" << hashed << "': " << ec.message()
|
|
<< "\n";
|
|
}
|
|
}
|
|
}
|
|
interceptionEnabled = false;
|
|
}
|
|
|
|
void initRemapping(const std::filesystem::path& dir) {
|
|
scratchDir = dir;
|
|
|
|
struct rebinding binding[] = {
|
|
{"open", reinterpret_cast<void*>(codeql_open), reinterpret_cast<void**>(&original_open)}};
|
|
rebind_symbols(binding, 1);
|
|
interceptionEnabled = true;
|
|
}
|
|
|
|
} // namespace codeql
|