added handler for 'ReturnType'

added handler for the parameter 'ReturnType'  inConvertFileOffsetToVA(),  ConvertVAtoFileOffset() and ConvertVAtoFileOffsetEx()
This commit is contained in:
cw2k 2020-11-03 22:53:57 +01:00 committed by GitHub
parent 951852bab1
commit 9f16d36504
Signed by: GitHub
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 26 additions and 9 deletions

View File

@ -567,21 +567,35 @@ public:
mappedFiles.erase(found);
return true;
}
ULONG_PTR ConvertFileOffsetToVA(ULONG_PTR FileMapVA, ULONG_PTR AddressToConvert, bool ReturnType)
////
///
// ConvertFileOffsetToVA - converts FileOffset to a virtual addresses
// checks if PE file is valid and if memory is accessible
// Returns: converted VA
// or NULL if conversion has failed.
//
ULONG_PTR ConvertFileOffsetToVA(
ULONG_PTR FileMapVA, // [in] Pointer to the mapped file content. It's either StaticFileLoad function or Windows API for file mapping.
ULONG_PTR AddressToConvert, // [in] physical address/file offset
bool ReturnType // [in] if true: add the FileMapVA to the return value
)
{
auto found = mappedFiles.find(FileMapVA);
if(found == mappedFiles.end())
__debugbreak(); //return 0;
if(!found->second.pe->IsValidPe())
__debugbreak(); //return 0;
auto offset = found->second.pe->ConvertOffsetToRva( uint32( AddressToConvert ) );
// convert: FileOffset -> VA
auto offset = found->second.pe->ConvertOffsetToRva(
uint32( AddressToConvert )
);
if (offset == INVALID_VALUE)
return 0;
return 0
else
return ReturnType ? FileMapVA + offset : offset;
}
////
@ -594,11 +608,11 @@ public:
ULONG_PTR ConvertVAtoFileOffset(
ULONG_PTR FileMapVA, // [in] Pointer to the mapped file content. It's either StaticFileLoad function or Windows API for file mapping.
ULONG_PTR AddressToConvert, // [in] Virtual address to convert to a physical address.
bool ReturnType // [in] Add the FileMapVA return value?
bool ReturnType // [in] if true: add the FileMapVA to the return value
)
{
return ConvertVAtoFileOffsetEx(
FileMapVA ,0 ,0 ,
FileMapVA ,0 ,0 ,
AddressToConvert, false, ReturnType );
}
@ -615,7 +629,7 @@ public:
ULONG_PTR ImageBase, // [in] ImageBase of the mapped file
ULONG_PTR AddressToConvert, // [in] Virtual address to convert to a physical address.
bool AddressIsRVA, // [in] true => AddressToConvert is relative virtual address
bool ReturnType // [in] Add the FileMapVA return value?
bool ReturnType // [in] if true: add the FileMapVA to the return value
)
{
auto found = mappedFiles.find(FileMapVA);
@ -630,12 +644,15 @@ public:
AddressToConvert :
AddressToConvert - ImageBase;
// convert: VA -> FileOffset
auto offset = found->second.pe->ConvertRvaToOffset(
uint32( RVA_ToConvert )
);
if (offset == INVALID_VALUE)
return 0;
return 0
else
return ReturnType ? FileMapVA + offset : offset;
}
template<typename T>