From 01e95de886413c252ee1a419ed0a7c41435847e6 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Sun, 18 Oct 2020 03:20:06 +0300 Subject: Process: add get_exe_path() --- src/process.cpp | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) (limited to 'src/process.cpp') diff --git a/src/process.cpp b/src/process.cpp index d3ce3a4..7e3a9e7 100644 --- a/src/process.cpp +++ b/src/process.cpp @@ -122,6 +122,32 @@ HMODULE Process::get_exe_module() { return module; } +std::string Process::get_exe_path() { + BOOST_STATIC_CONSTEXPR std::size_t init_buffer_size = MAX_PATH; + static_assert(init_buffer_size > 0, "init_buffer_size must be positive"); + + std::vector buffer; + buffer.resize(init_buffer_size); + + while (true) { + SetLastError(ERROR_SUCCESS); + + const auto nch = ::GetModuleFileNameW(NULL, buffer.data(), buffer.size()); + + if (nch == 0) { + throw error::windows(GetLastError(), "GetModuleFileNameW"); + } + + if (nch == buffer.size() && GetLastError() == ERROR_INSUFFICIENT_BUFFER) { + buffer.resize(2 * buffer.size()); + continue; + } + + buffer.resize(nch); + return narrow(buffer); + } +} + std::string Process::get_resource_string(unsigned int id) { wchar_t* s = nullptr; -- cgit v1.2.3