nold/Atmosphere
1
0
Fork 0
mirror of https://github.com/Atmosphere-NX/Atmosphere synced 2025-01-03 11:11:14 +00:00
Code Issues Projects Releases Packages Wiki Activity

dmnt-cheat: Add new, better math instruction to vm

Browse source
This commit is contained in:
Michael Scire 2019-02-27 19:30:29 -08:00
parent 68af2c1c2a
commit a3fc2c95b8
2 changed files with 106 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

82
stratosphere/dmnt/source/dmnt_cheat_vm.cpp
View file

@ -37,6 +37,22 @@ void DmntCheatVm::SkipConditionalBlock() {
} }
} }
u64 DmntCheatVm::GetVmInt(VmInt value, u32 bit_width) {
switch (bit_width) {
case 1:
return value.bit8;
case 2:
return value.bit16;
case 4:
return value.bit32;
case 8:
return value.bit64;
default:
/* Invalid bit width -> return 0. */
return 0;
}
}
void DmntCheatVm::Execute(const CheatProcessMetadata *metadata) { void DmntCheatVm::Execute(const CheatProcessMetadata *metadata) {
CheatVmOpcode cur_opcode; CheatVmOpcode cur_opcode;
u64 kDown = 0; u64 kDown = 0;
@ -111,6 +127,9 @@ void DmntCheatVm::Execute(const CheatProcessMetadata *metadata) {
case RegisterArithmeticType_RightShift: case RegisterArithmeticType_RightShift:
this->registers[cur_opcode.perform_math_static.reg_index] >>= (u64)cur_opcode.perform_math_static.value; this->registers[cur_opcode.perform_math_static.reg_index] >>= (u64)cur_opcode.perform_math_static.value;
break; break;
default:
/* Do not handle extensions here. */
break;
} }
/* Apply bit width. */ /* Apply bit width. */
switch (cur_opcode.perform_math_static.bit_width) { switch (cur_opcode.perform_math_static.bit_width) {
@ -136,6 +155,69 @@ void DmntCheatVm::Execute(const CheatProcessMetadata *metadata) {
this->SkipConditionalBlock(); this->SkipConditionalBlock();
} }
break; break;
case CheatVmOpcodeType_PerformArithmeticRegister:
{
const u64 operand_1_value = this->registers[cur_opcode.perform_math_reg.src_reg_1_index];
const u64 operand_2_value = cur_opcode.perform_math_reg.has_immediate ?
GetVmInt(cur_opcode.perform_math_reg.value, cur_opcode.perform_math_reg.bit_width) :
this->registers[cur_opcode.perform_math_reg.src_reg_2_index];
u64 res_val = 0;
/* Do requested math. */
switch (cur_opcode.perform_math_reg.math_type) {
case RegisterArithmeticType_Addition:
res_val = operand_1_value + operand_2_value;
break;
case RegisterArithmeticType_Subtraction:
res_val = operand_1_value - operand_2_value;
break;
case RegisterArithmeticType_Multiplication:
res_val = operand_1_value * operand_2_value;
break;
case RegisterArithmeticType_LeftShift:
res_val = operand_1_value << operand_2_value;
break;
case RegisterArithmeticType_RightShift:
res_val = operand_1_value >> operand_2_value;
break;
case RegisterArithmeticType_LogicalAnd:
res_val = operand_1_value & operand_2_value;
break;
case RegisterArithmeticType_LogicalOr:
res_val = operand_1_value | operand_2_value;
break;
case RegisterArithmeticType_LogicalNot:
res_val = ~operand_1_value;
break;
case RegisterArithmeticType_LogicalXor:
res_val = operand_1_value ^ operand_2_value;
break;
case RegisterArithmeticType_None:
res_val = operand_1_value;
break;
}
/* Apply bit width. */
switch (cur_opcode.perform_math_reg.bit_width) {
case 1:
res_val = static_cast<u8>(res_val);
break;
case 2:
res_val = static_cast<u16>(res_val);
break;
case 4:
res_val = static_cast<u32>(res_val);
break;
case 8:
res_val = static_cast<u64>(res_val);
break;
}
/* Save to register. */
this->registers[cur_opcode.perform_math_reg.dst_reg_index] = res_val;
}
break;
} }
} }
} }

24
stratosphere/dmnt/source/dmnt_cheat_vm.hpp
View file

@ -30,6 +30,9 @@ enum CheatVmOpcodeType : u32 {
CheatVmOpcodeType_StoreToRegisterAddress = 6, CheatVmOpcodeType_StoreToRegisterAddress = 6,
CheatVmOpcodeType_PerformArithmeticStatic = 7, CheatVmOpcodeType_PerformArithmeticStatic = 7,
CheatVmOpcodeType_BeginKeypressConditionalBlock = 8, CheatVmOpcodeType_BeginKeypressConditionalBlock = 8,
/* These are not implemented by Gateway's VM. */
CheatVmOpcodeType_PerformArithmeticRegister = 9,
}; };
enum MemoryAccessType : u32 { enum MemoryAccessType : u32 {
@ -52,6 +55,14 @@ enum RegisterArithmeticType : u32 {
RegisterArithmeticType_Multiplication = 2, RegisterArithmeticType_Multiplication = 2,
RegisterArithmeticType_LeftShift = 3, RegisterArithmeticType_LeftShift = 3,
RegisterArithmeticType_RightShift = 4, RegisterArithmeticType_RightShift = 4,
/* These are not supported by Gateway's VM. */
RegisterArithmeticType_LogicalAnd = 5,
RegisterArithmeticType_LogicalOr = 6,
RegisterArithmeticType_LogicalNot = 7,
RegisterArithmeticType_LogicalXor = 8,
RegisterArithmeticType_None = 9,
}; };
union VmInt { union VmInt {
@ -118,6 +129,16 @@ struct BeginKeypressConditionalOpcode {
u32 key_mask; u32 key_mask;
}; };
struct PerformArithmeticRegisterOpcode {
u32 bit_width;
RegisterArithmeticType math_type;
u32 dst_reg_index;
u32 src_reg_1_index;
u32 src_reg_2_index;
bool has_immediate;
VmInt value;
};
struct CheatVmOpcode { struct CheatVmOpcode {
CheatVmOpcodeType opcode; CheatVmOpcodeType opcode;
union { union {
@ -130,6 +151,7 @@ struct CheatVmOpcode {
StoreToRegisterAddressOpcode str_regaddr; StoreToRegisterAddressOpcode str_regaddr;
PerformArithmeticStaticOpcode perform_math_static; PerformArithmeticStaticOpcode perform_math_static;
BeginKeypressConditionalOpcode begin_keypress_cond; BeginKeypressConditionalOpcode begin_keypress_cond;
PerformArithmeticRegisterOpcode perform_math_reg;
}; };
}; };
@ -146,6 +168,8 @@ class DmntCheatVm {
private: private:
bool DecodeNextOpcode(CheatVmOpcode *out); bool DecodeNextOpcode(CheatVmOpcode *out);
void SkipConditionalBlock(); void SkipConditionalBlock();
static u64 GetVmInt(VmInt value, u32 bit_width);
public: public:
DmntCheatVm() { } DmntCheatVm() { }