Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

client/windows: fix GetModuleFileNameA

This commit is contained in:
Oleksii Shevchuk 2020-07-02 10:55:26 +03:00
parent fa8c1f461f
commit d285229b5e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
client/sources/MyLoadLibrary.c
View File

@ -410,12 +410,12 @@ DWORD CALLBACK MyGetModuleFileNameA(HMODULE hModule, LPSTR lpStr, DWORD dwSize)
DWORD dwRet = MemoryModuleFileNameA(lib->module, lpStr, dwSize);
if (dwRet == 0xFFFFFFFF) {
size_t reqSize = strlen(lib->fileName);
if (reqSize < dwSize) {
if (reqSize > dwSize) {
SetLastError(ERROR_INSUFFICIENT_BUFFER);
dwRet = 0;
} else {
memcpy(lpStr, lib->fileName, reqSize);
if (dwSize+1 == reqSize) {
if (dwSize >= reqSize + 1) {
lpStr[reqSize] = '\0';
}
dwRet = reqSize;