Function main

IntelPresentMon/FlashInjector/Main.cpp:33–114

}

int main(int argc, char** argv)
{
    try {
        // Initial logging
        LOGI << "Injector process started" << std::endl;

        // Initialize arguments
        if (auto res = clio::Options::Init(argc, argv, true)) {
            return *res;
        }
        auto& opts = clio::Options::Get();

        stdfs::path injectorPath;
        {
            std::vector<char> buffer(MAX_PATH);
            auto size = GetModuleFileNameA(NULL, buffer.data(), static_cast<DWORD>(buffer.size()));
            if (size == 0) {
                LOGE << "Failed to get this executable path." << std::endl;;
            }
            injectorPath = std::string(buffer.begin(), buffer.begin() + size);
            injectorPath = injectorPath.parent_path();
        }

        // DLL to inject
        const stdfs::path libraryPath = injectorPath / std::format("FlashInjectorLibrary-{}.dll", PM_BUILD_PLATFORM);

        const bool weAre32Bit = PM_BUILD_PLATFORM == "Win32"s;

        if (!stdfs::exists(libraryPath)) {
            LOGE << "Cannot find library: " << libraryPath << std::endl;;
            exit(1);
        }

        LOGI << "Waiting for processes that match executable name..." << std::endl;

        std::mutex targetModuleNameMtx;
        std::string targetModuleName;
        // thread whose sole job is to read from stdin without blocking the main thread
        std::thread{ [&] {
            std::string line;
            while (true) {
                std::getline(std::cin, line);
                std::lock_guard lk{ targetModuleNameMtx };
                targetModuleName = str::ToLower(line);
            }
        } }.detach();

        // keep a set of processes already attached so we don't attempt multiple attachments per process
        std::unordered_set<DWORD> processesAttached;
        while (true) {
            // atomic load target name and skip if empty string
            const auto tgt = [&] { std::lock_guard lk{ targetModuleNameMtx }; return targetModuleName; }();
            if (!tgt.empty()) {
                for (auto&& [processId, processName] : LibraryInject::GetProcessNames()) {
                    const auto processNameLower = str::ToLower(processName);
                    if (processNameLower == tgt && !processesAttached.contains(processId)) {
                        auto hProcTarget = win::OpenProcess(processId, PROCESS_QUERY_LIMITED_INFORMATION);
                        if (win::ProcessIs32Bit(hProcTarget) == weAre32Bit) {