aboutsummaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
m---------3rdparty/winapi/utf80
-rw-r--r--include/winapi/process.hpp2
-rw-r--r--src/path.cpp5
-rw-r--r--src/process.cpp26
-rw-r--r--test/unit_tests/process.cpp5
5 files changed, 37 insertions, 1 deletions
diff --git a/3rdparty/winapi/utf8 b/3rdparty/winapi/utf8
-Subproject 8fae97c1400493c23853496d0f59a3047bdd3da
+Subproject 25f39c16c2024bd7c9e03b4bc84625d0c8eed84
diff --git a/include/winapi/process.hpp b/include/winapi/process.hpp
index 469653c..5f841a0 100644
--- a/include/winapi/process.hpp
+++ b/include/winapi/process.hpp
@@ -56,6 +56,8 @@ public:
int get_exit_code() const;
+ static std::string get_exe_path();
+
static Resource get_resource(unsigned int id);
static std::string get_resource_string(unsigned int id);
diff --git a/src/path.cpp b/src/path.cpp
index 2c2deda..3df5918 100644
--- a/src/path.cpp
+++ b/src/path.cpp
@@ -16,8 +16,11 @@ namespace winapi {
namespace {
std::wstring do_canonicalize(const std::wstring& 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<wchar_t> buffer;
- buffer.resize(MAX_PATH);
+ buffer.resize(init_buffer_size);
while (true) {
const auto nch = ::GetFullPathNameW(path.c_str(), buffer.size(), buffer.data(), NULL);
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<wchar_t> 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;
diff --git a/test/unit_tests/process.cpp b/test/unit_tests/process.cpp
index 1c87b27..0a6cb27 100644
--- a/test/unit_tests/process.cpp
+++ b/test/unit_tests/process.cpp
@@ -21,6 +21,11 @@ using namespace winapi::process;
BOOST_AUTO_TEST_SUITE(process_tests)
+BOOST_AUTO_TEST_CASE(get_exe_path) {
+ const auto path = Process::get_exe_path();
+ BOOST_TEST_MESSAGE("Executable path: " << path);
+}
+
BOOST_FIXTURE_TEST_CASE(echo, WithEchoExe) {
const CommandLine cmd_line{get_echo_exe(), {"1", "2", "3"}};
const auto process = Process::create(cmd_line);