const swift::FrontendOptions& options) {

for (const auto& input : options.InputsAndOutputs.getAllInputs()) {

if (const auto& module = input.getPrimarySpecificPaths().SupplementaryOutputs.ModuleOutputPath;

!module.empty()) {

if (auto target = codeql::createTargetTrapDomain(state, codeql::resolvePath(module),

codeql::TrapType::module)) {

target->emitComment(

"trap file deliberately empty\n"

" * this swiftmodule was created during the build, so its entities must have\n"

" * been extracted directly from source files\n");

}

}

}

swift::FrontendOptions& options) {

auto& inOuts = options.InputsAndOutputs;

std::vector<swift::InputFile> inputs;

inOuts.forEachInput([&](const swift::InputFile& input) {

swift::PrimarySpecificPaths psp{};

if (std::filesystem::path output = input.getPrimarySpecificPaths().OutputFilename;

!output.empty()) {

if (output.extension() == ".swiftmodule") {

psp.OutputFilename = codeql::redirect(output);

} else {

psp.OutputFilename = "/dev/null";

}

}

if (std::filesystem::path module =

input.getPrimarySpecificPaths().SupplementaryOutputs.ModuleOutputPath;

!module.empty()) {

psp.SupplementaryOutputs.ModuleOutputPath = codeql::redirect(module);

state.originalOutputModules.push_back(module);

}

auto inputCopy = input;

inputCopy.setPrimarySpecificPaths(std::move(psp));

inputs.push_back(std::move(inputCopy));

return false;

});

inOuts.clearInputs();

for (const auto& i : inputs) {

inOuts.addInput(i);

}

public:

explicit Observer(const codeql::SwiftExtractorConfiguration& config) : state{config} {}

void parsedArgs(swift::CompilerInvocation& invocation) override {

auto& options = invocation.getFrontendOptions();

options.KeepASTContext = true;

lockOutputSwiftModuleTraps(state, options);

processFrontendOptions(state, options);

}

void configuredCompiler(swift::CompilerInstance& instance) override {

// remove default consumers to avoid double messaging

instance.getDiags().takeConsumers();

instance.addDiagnosticConsumer(&diagConsumer);

}

void performedCompilation(swift::CompilerInstance& compiler) override {

codeql::extractSwiftFiles(state, compiler);

codeql::extractSwiftInvocation(state, compiler, invocationTrap);

codeql::extractExtractLazyDeclarations(state, compiler);

}

void markSuccessfullyExtractedFiles() {

codeql::SwiftLocationExtractor locExtractor{invocationTrap};

for (const auto& src : state.sourceFiles) {

auto fileLabel = locExtractor.emitFile(src);

invocationTrap.emit(codeql::FileIsSuccessfullyExtractedTrap{fileLabel});

}

}

private:

codeql::SwiftExtractorState state;

codeql::TrapDomain invocationTrap{invocationTrapDomain(state)};

codeql::SwiftDiagnosticsConsumer diagConsumer{invocationTrap};

if (const char* var = getenv(envvar)) {

return var;

}

return def;

auto runUnderFilter = getenv("CODEQL_EXTRACTOR_SWIFT_RUN_UNDER_FILTER");

if (runUnderFilter == nullptr) {

return true;

}

std::string call = argv[0];

for (auto i = 1; i < argc; ++i) {

call += ' ';

call += argv[i];

}

std::regex filter{runUnderFilter, std::regex_constants::basic | std::regex_constants::nosubs};

return std::regex_search(call, filter);

if (argc == 0) return;

auto runUnder = getenv("CODEQL_EXTRACTOR_SWIFT_RUN_UNDER");

if (runUnder == nullptr || !checkRunUnderFilter(argc, argv)) {

return;

}

std::vector<char*> args;

// split RUN_UNDER value by spaces to get args vector

for (auto word = std::strtok(runUnder, " "); word != nullptr; word = std::strtok(nullptr, " ")) {

args.push_back(word);

}

// append process args, including extractor executable path

args.insert(args.end(), argv, argv + argc);

args.push_back(nullptr);

// avoid looping on this function

unsetenv("CODEQL_EXTRACTOR_SWIFT_RUN_UNDER");

execvp(args[0], args.data());

auto timestamp = std::chrono::system_clock::now().time_since_epoch().count();

auto filename = absl::StrCat(timestamp, "-", getpid());

auto target = std::filesystem::path("invocations") / std::filesystem::path(filename);

auto maybeDomain = codeql::createTargetTrapDomain(state, target, codeql::TrapType::invocation);

CODEQL_ASSERT(maybeDomain, "Cannot create invocation trap file for {}", target);

return std::move(maybeDomain.value());

codeql::SwiftExtractorConfiguration configuration{};

configuration.trapDir = getenv_or("CODEQL_EXTRACTOR_SWIFT_TRAP_DIR", "extractor-out/trap/swift");

configuration.sourceArchiveDir =

getenv_or("CODEQL_EXTRACTOR_SWIFT_SOURCE_ARCHIVE_DIR", "extractor-out/src");

configuration.scratchDir =

getenv_or("CODEQL_EXTRACTOR_SWIFT_SCRATCH_DIR", "extractor-out/working");

configuration.frontendOptions.assign(argv + 1, argv + argc);

return configuration;

if (argc < 2) {

return ""s;

}

return absl::StrJoin(argv + 1, argv + argc, " ");

std::string ret;

for (auto env = envp; *env; ++env) {

ret += *env;

ret += '\n';

}

return ret;

checkWhetherToRunUnderTool(argc, argv);

if (argc == 1) {

// TODO: print usage

return 1;

}

// Required by Swift/LLVM

PROGRAM_START(argc, argv);

INITIALIZE_LLVM();

initializeSwiftModules();

const auto configuration = configure(argc, argv);

LOG_INFO("calling extractor with arguments \"{}\"", argDump(argc, argv));

LOG_DEBUG("environment:\n{}\n", envDump(envp));

auto openInterception = codeql::setupFileInterception(configuration);

Observer observer(configuration);

int frontend_rc = swift::performFrontend(configuration.frontendOptions, "swift-extractor",

(void*)main, &observer);

if (frontend_rc == 0) {

observer.markSuccessfullyExtractedFiles();

}

codeql::Log::flush();

return frontend_rc;