mirror of
				https://github.com/hedge-dev/XenonRecomp.git
				synced 2025-11-04 06:47:09 +00:00 
			
		
		
		
	Allow midasm hook return behavior to be configured.
This commit is contained in:
		@@ -328,7 +328,14 @@ bool Recompiler::Recompile(
 | 
			
		||||
    auto midAsmHook = config.midAsmHooks.find(base);
 | 
			
		||||
    if (midAsmHook != config.midAsmHooks.end())
 | 
			
		||||
    {
 | 
			
		||||
        print("\t{}(", midAsmHook->second.name);
 | 
			
		||||
        bool returnsBool = midAsmHook->second.returnOnFalse || midAsmHook->second.returnOnTrue ||
 | 
			
		||||
            midAsmHook->second.jumpAddressOnFalse != NULL || midAsmHook->second.jumpAddressOnTrue != NULL;
 | 
			
		||||
 | 
			
		||||
        print("\t");
 | 
			
		||||
        if (returnsBool)
 | 
			
		||||
            print("if (");
 | 
			
		||||
 | 
			
		||||
        print("{}(", midAsmHook->second.name);
 | 
			
		||||
        for (auto& reg : midAsmHook->second.registers)
 | 
			
		||||
        {
 | 
			
		||||
            if (out.back() != '(')
 | 
			
		||||
@@ -367,7 +374,35 @@ bool Recompiler::Recompile(
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if (returnsBool)
 | 
			
		||||
        {
 | 
			
		||||
            println(")) {{");
 | 
			
		||||
 | 
			
		||||
            if (midAsmHook->second.returnOnTrue)
 | 
			
		||||
                println("\t\treturn;");
 | 
			
		||||
            else if (midAsmHook->second.jumpAddressOnTrue != NULL)
 | 
			
		||||
                println("\t\tgoto loc_{:X};", midAsmHook->second.jumpAddressOnTrue);
 | 
			
		||||
 | 
			
		||||
            println("\t}}");
 | 
			
		||||
 | 
			
		||||
            println("\telse {{");
 | 
			
		||||
 | 
			
		||||
            if (midAsmHook->second.returnOnFalse)
 | 
			
		||||
                println("\t\treturn;");
 | 
			
		||||
            else if (midAsmHook->second.jumpAddressOnFalse != NULL)
 | 
			
		||||
                println("\t\tgoto loc_{:X};", midAsmHook->second.jumpAddressOnFalse);
 | 
			
		||||
 | 
			
		||||
            println("\t}}");
 | 
			
		||||
        }
 | 
			
		||||
        else
 | 
			
		||||
        {
 | 
			
		||||
            println(");");
 | 
			
		||||
 | 
			
		||||
            if (midAsmHook->second.ret)
 | 
			
		||||
                println("\treturn;");
 | 
			
		||||
            else if (midAsmHook->second.jumpAddress != NULL)
 | 
			
		||||
                println("\tgoto loc_{:X};", midAsmHook->second.jumpAddress);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    int id = insn.opcode->id;
 | 
			
		||||
@@ -2141,7 +2176,17 @@ bool Recompiler::Recompile(const Function& fn)
 | 
			
		||||
        auto midAsmHook = config.midAsmHooks.find(addr);
 | 
			
		||||
        if (midAsmHook != config.midAsmHooks.end())
 | 
			
		||||
        {
 | 
			
		||||
            print("extern void {}(", midAsmHook->second.name);
 | 
			
		||||
            if (midAsmHook->second.returnOnFalse || midAsmHook->second.returnOnTrue ||
 | 
			
		||||
                midAsmHook->second.jumpAddressOnFalse != NULL || midAsmHook->second.jumpAddressOnTrue != NULL)
 | 
			
		||||
            {
 | 
			
		||||
                print("extern bool ");
 | 
			
		||||
            }
 | 
			
		||||
            else
 | 
			
		||||
            {
 | 
			
		||||
                print("extern void ");
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            print("{}(", midAsmHook->second.name);
 | 
			
		||||
            for (auto& reg : midAsmHook->second.registers)
 | 
			
		||||
            {
 | 
			
		||||
                if (out.back() != '(')
 | 
			
		||||
@@ -2178,6 +2223,13 @@ bool Recompiler::Recompile(const Function& fn)
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            println(");\n");
 | 
			
		||||
 | 
			
		||||
            if (midAsmHook->second.jumpAddress != NULL)
 | 
			
		||||
                labels.emplace(midAsmHook->second.jumpAddress);       
 | 
			
		||||
            if (midAsmHook->second.jumpAddressOnTrue != NULL)
 | 
			
		||||
                labels.emplace(midAsmHook->second.jumpAddressOnTrue);    
 | 
			
		||||
            if (midAsmHook->second.jumpAddressOnFalse != NULL)
 | 
			
		||||
                labels.emplace(midAsmHook->second.jumpAddressOnFalse);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -82,10 +82,34 @@ void RecompilerConfig::Load(const std::string_view& configFilePath)
 | 
			
		||||
 | 
			
		||||
            RecompilerMidAsmHook midAsmHook;
 | 
			
		||||
            midAsmHook.name = *table["name"].value<std::string>();
 | 
			
		||||
            for (auto& reg : *table["registers"].as_array())
 | 
			
		||||
            if (auto registerArray = table["registers"].as_array())
 | 
			
		||||
            {
 | 
			
		||||
                for (auto& reg : *registerArray)
 | 
			
		||||
                    midAsmHook.registers.push_back(*reg.value<std::string>());
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            midAsmHook.ret = table["return"].value_or(false);
 | 
			
		||||
            midAsmHook.returnOnTrue = table["return_on_true"].value_or(false);
 | 
			
		||||
            midAsmHook.returnOnFalse = table["return_on_false"].value_or(false);
 | 
			
		||||
 | 
			
		||||
            midAsmHook.jumpAddress = table["jump_address"].value_or(0u);
 | 
			
		||||
            midAsmHook.jumpAddressOnTrue = table["jump_address_on_true"].value_or(0u);
 | 
			
		||||
            midAsmHook.jumpAddressOnFalse = table["jump_address_on_false"].value_or(0u);
 | 
			
		||||
 | 
			
		||||
            if ((midAsmHook.ret && midAsmHook.jumpAddress != NULL) ||
 | 
			
		||||
                (midAsmHook.returnOnTrue && midAsmHook.jumpAddressOnTrue != NULL) ||
 | 
			
		||||
                (midAsmHook.returnOnFalse && midAsmHook.jumpAddressOnFalse != NULL))
 | 
			
		||||
            {
 | 
			
		||||
                std::println("{}: can't return and jump at the same time", midAsmHook.name);
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            if ((midAsmHook.ret || midAsmHook.jumpAddress != NULL) &&
 | 
			
		||||
                (midAsmHook.returnOnFalse != NULL || midAsmHook.returnOnTrue != NULL ||
 | 
			
		||||
                    midAsmHook.jumpAddressOnFalse != NULL || midAsmHook.jumpAddressOnTrue != NULL))
 | 
			
		||||
            {
 | 
			
		||||
                std::println("{}: can't mix direct and conditional return/jump", midAsmHook.name);
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            midAsmHooks.emplace(*table["address"].value<uint32_t>(), std::move(midAsmHook));
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 
 | 
			
		||||
@@ -10,6 +10,14 @@ struct RecompilerMidAsmHook
 | 
			
		||||
{
 | 
			
		||||
    std::string name;
 | 
			
		||||
    std::vector<std::string> registers;
 | 
			
		||||
 | 
			
		||||
    bool ret = false;
 | 
			
		||||
    bool returnOnTrue = false;
 | 
			
		||||
    bool returnOnFalse = false;
 | 
			
		||||
 | 
			
		||||
    uint32_t jumpAddress = 0;
 | 
			
		||||
    uint32_t jumpAddressOnTrue = 0;
 | 
			
		||||
    uint32_t jumpAddressOnFalse = 0;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
struct RecompilerConfig
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user