LunaHook-mirror/LunaHook/enginecontrol.cpp

137 lines
4.0 KiB
C++
Raw Normal View History

2024-02-07 20:59:24 +08:00
WCHAR* processName, // cached
processPath[MAX_PATH]; // cached
WCHAR processName_lower[MAX_PATH];
uintptr_t processStartAddress, processStopAddress;
std::vector<ENGINE*> check_engines();
std::vector<ENGINE*> ignore_engines();
std::vector<ENGINE*> unsafe_check_atlast();
bool ENGINE::check_function(){
switch (check_by)
{
case CHECK_BY::ALL_TRUE:{
is_engine_certain=false;
return true;
}
case CHECK_BY::FILE:{
return (Util::CheckFile(std::get<check_by_single>(check_by_target))) ;
}
case CHECK_BY::FILE_ALL:{
auto _list=std::get<check_by_list>(check_by_target);
return std::all_of(_list.begin(),_list.end(),Util::CheckFile);
}
case CHECK_BY::FILE_ANY:{
auto _list=std::get<check_by_list>(check_by_target);
return std::any_of(_list.begin(),_list.end(),Util::CheckFile);
}
case CHECK_BY::RESOURCE_STR:{
return (Util::SearchResourceString(std::get<check_by_single>(check_by_target))) ;
}
case CHECK_BY::CUSTOM:{
return std::get<check_by_custom_function>(check_by_target)();
}
default:
return false;
}
}
bool safematch(ENGINE* m){
bool matched=false;
__try {
matched=m->check_function();
}
__except (EXCEPTION_EXECUTE_HANDLER) {
ConsoleOutput(Match_Error,m->getenginename());
//ConsoleOutput("match ERROR");
}
return matched;
}
bool safeattach(ENGINE* m){
bool attached=false;
__try {
attached=m->attach_function();
}
__except (EXCEPTION_EXECUTE_HANDLER) {
ConsoleOutput(Attach_Error,m->getenginename());
//ConsoleOutput("attach ERROR");
}
return attached;
}
bool checkengine(){
auto engines=check_engines();
auto engineatlast=unsafe_check_atlast();
auto engineignore=ignore_engines();
std::vector<const char*> infomations={
"match failed",
"attach failed",
"attach success"
};
auto allengines={engines,engineignore,engineatlast};
int total=[allengines](){int _=0;for(auto eng:allengines)_+=eng.size();return _;}();
int current=0;
for(auto eng:allengines){
for(auto m:eng) {
current+=1;
bool matched=safematch(m);
bool attached=matched&&safeattach(m);
//ConsoleOutput("Progress %d/%d, checked engine %s, %s",current,total,m->getenginename(),infomations[matched+attached]);
//ConsoleOutput("Progress %d/%d, %s",current,total,infomations[matched+attached]);
if(matched==false)continue;
ConsoleOutput(MatchedEngine,m->getenginename());
if(m->dontstop){
continue;
}
if(m->is_engine_certain){
ConsoleOutput(ConfirmStop,m->getenginename());
return attached;
}
if(attached){
ConsoleOutput(Attach_Stop,m->getenginename());
return true;
}
}
}
return false;
}
2024-05-06 23:30:27 +08:00
void HIJACK(){
2024-02-07 20:59:24 +08:00
static bool once = false;
if(once)return;
once=true;
GetModuleFileNameW(nullptr, processPath, MAX_PATH);
processName = wcsrchr(processPath, L'\\') + 1;
wcscpy_s(processName_lower, processName);
_wcslwr_s(processName_lower); // lower case
std::tie(processStartAddress,processStopAddress)=Util::QueryModuleLimits(GetModuleHandleW(nullptr),0,1+PAGE_NOACCESS);
spDefault.minAddress = processStartAddress;
spDefault.maxAddress = processStopAddress;
ConsoleOutput(ProcessRange, processStartAddress, processStopAddress);
if (processStartAddress + 0x40000 > processStopAddress) ConsoleOutput(WarningDummy);
bool result=false;
__try {
result = checkengine();
}
2024-02-10 12:43:51 +08:00
__except (EXCEPTION_EXECUTE_HANDLER) { ConsoleOutput(HIJACK_ERROR); }
2024-02-07 20:59:24 +08:00
if(result==false){
PcHooks::hookOtherPcFunctions();
}
}