| 530 | } |
| 531 | |
| 532 | DWORD GetModulePathName(HMODULE hModule, CEStr& lsPathName) |
| 533 | { |
| 534 | DWORD cchMax = MAX_PATH; |
| 535 | wchar_t* pszData = lsPathName.GetBuffer(cchMax); |
| 536 | if (!pszData) |
| 537 | return 0; // Memory allocation error |
| 538 | |
| 539 | DWORD nRc = GetModuleFileName(hModule, pszData, cchMax); |
| 540 | if (!nRc) |
| 541 | return 0; |
| 542 | |
| 543 | const DWORD errCode = GetLastError(); |
| 544 | |
| 545 | // Not enough space? |
| 546 | if (nRc == cchMax || errCode == ERROR_INSUFFICIENT_BUFFER) |
| 547 | { |
| 548 | cchMax = MAX_WIDE_PATH_LENGTH; |
| 549 | pszData = lsPathName.GetBuffer(cchMax); |
| 550 | if (!pszData) |
| 551 | return 0; // Memory allocation error |
| 552 | nRc = GetModuleFileName(hModule, pszData, cchMax); |
| 553 | if (!nRc || (nRc == cchMax)) |
| 554 | return 0; |
| 555 | } |
| 556 | |
| 557 | return nRc; |
| 558 | } |
| 559 | |
| 560 | #ifndef __GNUC__ |
| 561 | extern "C" IMAGE_DOS_HEADER __ImageBase; |