windows 获取当前模块所在文件夹路径

1.getmodulepath.h

#ifndef GETMODULEPATH
#define GETMODULEPATH

#include <wtypes.h>
#include <string>

/**
* @class getmodulepath
* @brief 这个类用来获取当前模块的绝对路径
*/
class GetModulePath
{
public:
static HMODULE GetModuleHandleFromAddress(void* address);

static bool IsModuleHandleValid(HMODULE module_handle);

static HMODULE GetCurrentModuleHandle();

static std::wstring GetModulePathName(HMODULE module_handle);

static bool IsFilePathSeparator(const wchar_t separator);

static bool FilePathApartDirectory(const std::wstring &filepath_in, std::wstring &directory_out);

static std::wstring GetModuleDirectory(HMODULE module_handle);

static std::wstring GetCurrentModuleDirectory();
};
#endif // GETMODULEPATH



2.getmodulepath.cpp

#include "getmodulepath.h"

const wchar_t kEndChar = L'\0';
const wchar_t kFilePathSeparators[] = L"\\/";

HMODULE GetModulePath::GetModuleHandleFromAddress(void *address)
{
MEMORY_BASIC_INFORMATION mbi;
DWORD result = ::VirtualQuery(address, &mbi, sizeof(mbi));
if (result != sizeof(mbi))
{
return static_cast<HMODULE>(0);
}
return static_cast<HMODULE>(mbi.AllocationBase);
}

bool GetModulePath::IsModuleHandleValid(HMODULE module_handle)
{
if (!module_handle)
{
return false;
}

return module_handle == GetModuleHandleFromAddress(module_handle);
}

HMODULE GetModulePath::GetCurrentModuleHandle()
{
return GetModuleHandleFromAddress((void*)GetCurrentModuleHandle);
}

std::wstring GetModulePath::GetModulePathName(HMODULE module_handle)
{
std::wstring mod_path;
if (IsModuleHandleValid(module_handle))
{
mod_path.resize(MAX_PATH);
mod_path.resize(::GetModuleFileNameW(module_handle, &mod_path[0], mod_path.size()));
}

return mod_path;
}

bool GetModulePath::IsFilePathSeparator(const wchar_t separator)
{
if (separator == kEndChar)
{
return false;
}

size_t len = sizeof(kFilePathSeparators) / sizeof(wchar_t);
for (size_t i = 0; i < len; i++)
{
if (separator == kFilePathSeparators[i])
{
return true;
}
}

return false;
}

bool GetModulePath::FilePathApartDirectory(const std::wstring &filepath_in, std::wstring &directory_out)
{
size_t index = filepath_in.size() - 1;
if (index <= 0 || filepath_in.size() == 0)
{
return false;
}

for (; index != 0; index--)
{
if (IsFilePathSeparator(filepath_in[index]))
{
if (index == filepath_in.size() - 1)
{
directory_out = filepath_in;
}
else
{
directory_out = filepath_in.substr(0, index + 1);
}
return true;
}
}
return false;
}

std::wstring GetModulePath::GetModuleDirectory(HMODULE module_handle)
{
std::wstring module_directory;
if (IsModuleHandleValid(module_handle))
{
if (FilePathApartDirectory(GetModulePathName(module_handle), module_directory))
{
return module_directory;
}
}

return L"";
}

std::wstring GetModulePath::GetCurrentModuleDirectory()
{
return GetModuleDirectory(GetCurrentModuleHandle());
}



3.使用示例

std::wstring dir = GetModulePath::GetCurrentModuleDirectory();