Extern/Beef
1
0
Fork 0
mirror of https://github.com/beefytech/Beef.git synced 2025-06-08 11:38:21 +02:00
Code Activity

Hotswap extension module fix

Browse source
This commit is contained in:
Brian Fiete 2024-12-29 13:03:31 -08:00
parent 0bdfea7231
commit b23c1db0d9
1 changed files with 16 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

19
IDEHelper/DbgModule.cpp
View file

@ -5529,6 +5529,7 @@ void DbgModule::HotReplaceMethods(DbgType* newType, DbgType* primaryType)
auto _ReplaceMethod = [&](DbgSubprogram* newMethod)
{
bool didReplace = false;
if (!newMethod->mBlock.IsEmpty())
{
BfLogDbg("Hot added new method %p %s Address:%p\n", newMethod, newMethod->mName, newMethod->mBlock.mLowPC);
@ -5543,9 +5544,11 @@ void DbgModule::HotReplaceMethods(DbgType* newType, DbgType* primaryType)
{
auto oldMethod = itr.GetValue();
_HotJump(oldMethod, newMethod);
didReplace = true;
}
}
}
return didReplace;
};
if (newType == primaryType)
@ -5558,11 +5561,15 @@ void DbgModule::HotReplaceMethods(DbgType* newType, DbgType* primaryType)
if ((newMethod->mBlock.mLowPC >= mImageBase) && (newMethod->mBlock.mHighPC <= mImageBase + mImageSize))
{
// Our object file contains this function
_ReplaceMethod(newMethod);
auto nextMethod = newMethod->mNext;
newType->mMethodList.Remove(newMethod);
primaryType->mHotReplacedMethodList.PushFront(newMethod);
newMethod = nextMethod;
if (_ReplaceMethod(newMethod))
{
auto nextMethod = newMethod->mNext;
newType->mMethodList.Remove(newMethod);
primaryType->mHotReplacedMethodList.PushFront(newMethod);
newMethod = nextMethod;
}
else
newMethod = newMethod->mNext;
}
else
{
@ -6582,7 +6589,9 @@ bool DbgModule::ReadCOFF(DataStream* stream, DbgModuleKind moduleKind)
auto primaryType = dbgType->GetPrimaryType();
if ((primaryType->mHotNewType == NULL) && (HasHotReplacedMethods(primaryType)) && (dbgType == primaryType))
{
HotReplaceMethods(dbgType, primaryType);
}
if (primaryType != dbgType)
{