AuroraSupport/AuroraRuntime
2
1
Fork 0
Code Issues 27 Pull Requests Releases Wiki Activity

[+] Insert anon/other pages into process map of linux processes

Browse Source
This commit is contained in:
Reece Wilson 2022-04-21 13:29:47 +01:00
parent 237d2d070c
commit a66fb2c510
3 changed files with 32 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
Source/Process/ProcessMap.Linux.cpp
View File

@ -36,14 +36,6 @@ namespace Aurora::Process
size_t size,
void *data)
{
#if 0
auto index = file.find_last_of(".so");
if (index != AuString::npos)
{
file = file.substr(0, index);
}
#endif
AuString fileName = info->dlpi_name[0] ? info->dlpi_name : AuString{};
if (fileName.empty())
{
@ -81,6 +73,7 @@ namespace Aurora::Process
return;
}
Sections sections;
AuParse::SplitNewlines(map, [&](const AuString &line)
{
char *endPtr;
@ -92,6 +85,8 @@ namespace Aurora::Process
if (errno == ERANGE) return;
if (*endPtr != ' ') return;
auto perms = endPtr + 1;
auto A = line.find_first_of(':');
if (A == AuString::npos) return;
@ -131,10 +126,26 @@ namespace Aurora::Process
}
}
// TODO:
//printf("%llx %llx %s\n", base, end, name.c_str());
Section sect {};
sect.origVa = 0;
sect.baseVa = base;
sect.size = end - base;
sect.fsOff = 0;
sect.pt.readable = perms[0] == 'r';
sect.pt.writable = perms[1] == 'w';
sect.pt.NX = perms[2] != 'x';
sect.name = AuMove(name);
AuTryInsert(sections, sect);
}, false);
BorrowOtherSectionArray([&](AuList<Section> &out)
{
out.clear();
out.insert(out.end(), sections.begin(), sections.end());
});
}
static void PassThreeGatherMissingProgNames()
@ -205,9 +216,7 @@ namespace Aurora::Process
AuUInt fileOffset = section.sh_offset;
AuUInt programOffset = section.sh_addr;
Section sect {};
sect.origVa = section.sh_addr;
sect.baseVa = 0;
@ -238,7 +247,10 @@ namespace Aurora::Process
sect.name = sectionName;
AuTryInsert(modSections, sect);
if (sect.fsOff || sect.size)
{
AuTryInsert(modSections, sect);
}
}
auto pub = AuMakeShared<PublicModule>();

8
Source/Process/ProcessMap.cpp
View File

@ -131,6 +131,11 @@ namespace Aurora::Process
for (const auto &section : mod->sections)
{
if (!section.baseVa)
{
continue;
}
for (AuUInt i = section.baseVa; i < section.baseVa + section.size; i += (kMinPageAlignment * kPageBufferPad))
{
auto itr = gModulePtrMap.find(ToLowestPageAlignment(i));
@ -155,7 +160,6 @@ namespace Aurora::Process
static AuOptional<Section> FindInCache(AuUInt pointer)
{
auto curPtr = ToLowestPageAlignment(pointer);
auto temp = GetSectionCache(curPtr);
if (temp.has_value()) return temp;
@ -209,7 +213,7 @@ namespace Aurora::Process
#endif
}
void BorrowOtherSectionArray(const AuConsumer<AuList<Section>> &callback)
void BorrowOtherSectionArray(const AuConsumer<AuList<Section>&> &callback)
{
AU_LOCK_GUARD(gMutexUnique);
callback(gOtherSections);

2
Source/Process/ProcessMap.hpp
View File

@ -15,7 +15,7 @@ namespace Aurora::Process
AuUInt modBase;
};
void BorrowOtherSectionArray(const AuConsumer<AuList<Section>> &callback);
void BorrowOtherSectionArray(const AuConsumer<AuList<Section>&> &callback);
PublicModule GetFromModuleCache(AuUInt handle);
void InsertModuleCache(const ModuleBasePair &pair, const AuSPtr<PublicModule> &mod);