about summary refs log tree commit diff stats
path: root/Source/ChallengeRandomizer.cpp
diff options
context:
space:
mode:
authorjbzdarkid <jbzdarkid@gmail.com>2018-12-09 11:48:17 -0800
committerjbzdarkid <jbzdarkid@gmail.com>2018-12-09 11:48:17 -0800
commit4ad724eaae7be3780b5fb891b391bbf4ef26f410 (patch)
treed4daf1c5a8036098ee7dd4a52690802dbacaf3c3 /Source/ChallengeRandomizer.cpp
parentf485927337c329a4a691500a1b6c18c763863bb4 (diff)
downloadwitness-tutorializer-4ad724eaae7be3780b5fb891b391bbf4ef26f410.tar.gz
witness-tutorializer-4ad724eaae7be3780b5fb891b391bbf4ef26f410.tar.bz2
witness-tutorializer-4ad724eaae7be3780b5fb891b391bbf4ef26f410.zip
Improve internals of sigscanning, fix a small overwrite bug
Diffstat (limited to 'Source/ChallengeRandomizer.cpp')
-rw-r--r--Source/ChallengeRandomizer.cpp265
1 files changed, 95 insertions, 170 deletions
diff --git a/Source/ChallengeRandomizer.cpp b/Source/ChallengeRandomizer.cpp index 44886b7..fcd4e4a 100644 --- a/Source/ChallengeRandomizer.cpp +++ b/Source/ChallengeRandomizer.cpp
@@ -1,21 +1,6 @@
1#include "ChallengeRandomizer.h" 1#include "ChallengeRandomizer.h"
2#include <iostream> 2#include <iostream>
3 3
4int find(const std::vector<byte> &data, const std::vector<byte>& search, size_t startIndex = 0) {
5 for (size_t i=startIndex; i<data.size() - search.size(); i++) {
6 bool match = true;
7 for (size_t j=0; j<search.size(); j++) {
8 if (data[i+j] == search[j]) {
9 continue;
10 }
11 match = false;
12 break;
13 }
14 if (match) return static_cast<int>(i);
15 }
16 return -1;
17}
18
19// Reads the (relative!) address of the RNG, then shifts it to point at RNG2 4// Reads the (relative!) address of the RNG, then shifts it to point at RNG2
20void ChallengeRandomizer::AdjustRng(int offset) { 5void ChallengeRandomizer::AdjustRng(int offset) {
21 int currentRng = _memory->ReadData<int>({offset}, 0x1)[0]; 6 int currentRng = _memory->ReadData<int>({offset}, 0x1)[0];
@@ -33,164 +18,104 @@ ChallengeRandomizer::ChallengeRandomizer(const std::shared_ptr<Memory>& memory,
33 if (!alreadyInjected) _memory->WriteData<int>({GLOBALS + 0x30}, {RNG_ADDR + 4}); 18 if (!alreadyInjected) _memory->WriteData<int>({GLOBALS + 0x30}, {RNG_ADDR + 4});
34 _memory->WriteData<int>({GLOBALS + 0x30, 0}, {seed}); 19 _memory->WriteData<int>({GLOBALS + 0x30, 0}, {seed});
35 20
36 int do_success_side_effects = -1; 21 // do_success_side_effects
37 int reveal_exit_hall = -1; 22 _memory->AddSigScan({0xFF, 0xC8, 0x99, 0x2B, 0xC2, 0xD1, 0xF8, 0x8B, 0xD0}, [&](int index) {
38 int begin_endgame_1 = -1; 23 if (GLOBALS == 0x5B28C0) { // Version differences
39 24 index += 0x3E;
40 _memory->SigScan([&](int offset, const std::vector<byte>& data) { 25 } else if (GLOBALS == 0x62A080) {
41 // This injection ensures that the seed is set every time the challenge is started. 26 index += 0x42;
42 // We always do this sigscan since it affects the seed. 27 }
43 if (do_success_side_effects == -1) { 28 _memory->WriteData<byte>({index}, {
44 int index = find(data, {0xFF, 0xC8, 0x99, 0x2B, 0xC2, 0xD1, 0xF8, 0x8B, 0xD0}); 29 0x8B, 0x0D, 0x00, 0x00, 0x00, 0x00, // mov ecx, [0x00000000] ;This is going to be the address of the custom RNG
45 if (index != -1) { 30 0x67, 0xC7, 0x01, 0x00, 0x00, 0x00, 0x00, // mov dword ptr ds:[ecx], 0x00000000 ;This is going to be the seed value
46 do_success_side_effects = offset + index + 0x3E; 31 0x48, 0x83, 0xF8, 0x02, // cmp rax, 0x2 ;This is the short solve on the record player (which turns it off)
47 if (GLOBALS == 0x62A080) do_success_side_effects += 4; // There's an extra 4 opcodes in the new version 32 0x90, 0x90, 0x90 // nop nop nop
48 _memory->WriteData<byte>({do_success_side_effects}, { 33 });
49 0x8B, 0x0D, 0x00, 0x00, 0x00, 0x00, // mov ecx, [0x00000000] ;This is going to be the address of the custom RNG 34 int target = (GLOBALS + 0x30) - (index + 0x6); // +6 is for the length of the line
50 0x67, 0xC7, 0x01, 0x00, 0x00, 0x00, 0x00, // mov dword ptr ds:[ecx], 0x00000000 ;This is going to be the seed value 35 _memory->WriteData<int>({index + 0x2}, {target});
51 0x48, 0x83, 0xF8, 0x02, // cmp rax, 0x2 ;This is the short solve on the record player (which turns it off) 36 _memory->WriteData<int>({index + 0x9}, {seed});
52 0x90, 0x90, 0x90 // nop nop nop
53 });
54 int target = (GLOBALS + 0x30) - (do_success_side_effects + 0x6); // +6 is for the length of the line
55 _memory->WriteData<int>({do_success_side_effects + 0x2}, {target});
56 _memory->WriteData<int>({do_success_side_effects + 0x9}, {seed});
57 }
58 }
59
60 // BLEH.
61 if (reveal_exit_hall == -1) {
62 int index = find(data, {0x45, 0x8B, 0xF7, 0x48, 0x8B, 0x4D});
63 if (index != -1) {
64 reveal_exit_hall = offset + index;
65 _memory->WriteData<byte>({reveal_exit_hall + 0x15}, {0xEB});
66 }
67 }
68 if (begin_endgame_1 == -1) {
69 int index = find(data, {0x83, 0x7C, 0x01, 0xD0, 0x04});
70 if (index != -1) {
71 begin_endgame_1 = offset + index;
72 if (GLOBALS == 0x5B28C0) { // Version differences :/
73 begin_endgame_1 += 0x75;
74 } else if (GLOBALS == 0x62A080) {
75 begin_endgame_1 += 0x86;
76 }
77 _memory->WriteData<byte>({begin_endgame_1}, {0xEB});
78 }
79 }
80 }); 37 });
81 38
82 if (!alreadyInjected) HandleSigScans(); 39 // reveal_exit_hall
83} 40 _memory->AddSigScan({0x45, 0x8B, 0xF7, 0x48, 0x8B, 0x4D}, [&](int index){
41 _memory->WriteData<byte>({index + 0x15}, {0xEB});
42 });
84 43
85void ChallengeRandomizer::HandleSigScans() { 44 // begin_endgame_1
86 static int shuffle_integers = -1; 45 _memory->AddSigScan({0x83, 0x7C, 0x01, 0xD0, 0x04}, [&](int index){
87 static int shuffle_int = -1; 46 if (GLOBALS == 0x5B28C0) { // Version differences
88 static int cut_random_edges = -1; 47 index += 0x75;
89 static int get_empty_decoration_slot = -1; 48 } else if (GLOBALS == 0x62A080) {
90 static int get_empty_dot_spot = -1; 49 index += 0x86;
91 static int add_exactly_this_many_bisection_dots = -1; 50 }
92 static int make_a_shaper = -1; 51 _memory->WriteData<byte>({index}, {0xEB});
93 static int init_pattern_data_lotus = -1; 52 });
94 static int reroll_lotus_eater_stuff = -1;
95 static int do_lotus_minutes = -1;
96 static int do_lotus_tenths = -1;
97 static int do_lotus_eighths = -1;
98 53
99 _memory->SigScan([&](int offset, const std::vector<byte>& data) { 54 if (!alreadyInjected) {
100 if (shuffle_integers == -1) { 55 // shuffle_integers
101 int index = find(data, {0x48, 0x89, 0x5C, 0x24, 0x10, 0x56, 0x48, 0x83, 0xEC, 0x20, 0x48, 0x63, 0xDA, 0x48, 0x8B, 0xF1, 0x83, 0xFB, 0x01}); 56 _memory->AddSigScan({0x48, 0x89, 0x5C, 0x24, 0x10, 0x56, 0x48, 0x83, 0xEC, 0x20, 0x48, 0x63, 0xDA, 0x48, 0x8B, 0xF1, 0x83, 0xFB, 0x01}, [&](int index) {
102 if (index != -1) { 57 AdjustRng(index + 0x23);
103 shuffle_integers = offset + index; 58 });
104 AdjustRng(shuffle_integers + 0x23);
105 }
106 }
107 // shuffle<int> 59 // shuffle<int>
108 if (shuffle_int == -1) { 60 _memory->AddSigScan({0x33, 0xF6, 0x48, 0x8B, 0xD9, 0x39, 0x31, 0x7E, 0x51}, [&](int index) {
109 int index = find(data, {0x33, 0xF6, 0x48, 0x8B, 0xD9, 0x39, 0x31, 0x7E, 0x51}); 61 AdjustRng(index - 0x4);
110 if (index != -1) { 62 });
111 shuffle_int = offset + index - 0x16; 63 // cut_random_edges
112 AdjustRng(shuffle_int + 0x12); 64 _memory->AddSigScan({0x89, 0x44, 0x24, 0x3C, 0x33, 0xC0, 0x85, 0xC0, 0x75, 0xFA}, [&](int index) {
113 } 65 AdjustRng(index + 0x3B);
114 } 66 });
115 if (cut_random_edges == -1) { 67 // get_empty_decoration_slot
116 int index = find(data, {0x89, 0x44, 0x24, 0x3C, 0x33, 0xC0, 0x85, 0xC0, 0x75, 0xFA}); 68 _memory->AddSigScan({0x42, 0x83, 0x3C, 0x80, 0x00, 0x75, 0xDF}, [&](int index) {
117 if (index != -1) { 69 AdjustRng(index - 0x17);
118 cut_random_edges = offset + index - 0x22; 70 });
119 AdjustRng(cut_random_edges + 0x5D); 71 // get_empty_dot_spot
120 } 72 _memory->AddSigScan({0xF7, 0xF3, 0x85, 0xD2, 0x74, 0xEC}, [&](int index) {
121 } 73 AdjustRng(index - 0xB);
122 if (get_empty_decoration_slot == -1) { 74 });
123 int index = find(data, {0x42, 0x83, 0x3C, 0x80, 0x00, 0x75, 0xDF}); 75 // add_exactly_this_many_bisection_dots
124 if (index != -1) { 76 _memory->AddSigScan({0x48, 0x8B, 0xB4, 0x24, 0xB8, 0x00, 0x00, 0x00, 0x48, 0x8B, 0xBC, 0x24, 0xB0, 0x00, 0x00, 0x00}, [&](int index) {
125 get_empty_decoration_slot = offset + index - 0x2D; 77 AdjustRng(index - 0x4);
126 AdjustRng(get_empty_decoration_slot + 0x16); 78 });
127 } 79 // make_a_shaper
128 } 80 _memory->AddSigScan({0xF7, 0xE3, 0xD1, 0xEA, 0x8D, 0x0C, 0x52}, [&](int index) {
129 if (get_empty_dot_spot == -1) { 81 AdjustRng(index - 0x10);
130 int index = find(data, {0xF7, 0xF3, 0x85, 0xD2, 0x74, 0xEC}); 82 AdjustRng(index + 0x1C);
131 if (index != -1) { 83 AdjustRng(index + 0x49);
132 get_empty_dot_spot = offset + index - 0x2E; 84 });
133 AdjustRng(get_empty_dot_spot + 0x23); 85 // Entity_Machine_Panel::init_pattern_data_lotus
134 } 86 _memory->AddSigScan({0x40, 0x55, 0x56, 0x48, 0x8D, 0x6C, 0x24, 0xB1}, [&](int index) {
135 } 87 AdjustRng(index + 0x433);
136 if (add_exactly_this_many_bisection_dots == -1) { 88 AdjustRng(index + 0x45B);
137 int index = find(data, {0x48, 0x8B, 0xB4, 0x24, 0xB8, 0x00, 0x00, 0x00, 0x48, 0x8B, 0xBC, 0x24, 0xB0, 0x00, 0x00, 0x00}); 89 AdjustRng(index + 0x5A7);
138 if (index != -1) { 90 AdjustRng(index + 0x5D6);
139 add_exactly_this_many_bisection_dots = offset + index - 0x20; 91 AdjustRng(index + 0x6F6);
140 AdjustRng(add_exactly_this_many_bisection_dots + 0x1C); 92 AdjustRng(index + 0xD17);
141 } 93 AdjustRng(index + 0xFDA);
142 } 94 });
143 if (make_a_shaper == -1) { 95 // Entity_Record_Player::reroll_lotus_eater_stuff
144 int index = find(data, {0xF7, 0xE3, 0xD1, 0xEA, 0x8D, 0x0C, 0x52}); 96 _memory->AddSigScan({0xB8, 0xAB, 0xAA, 0xAA, 0xAA, 0x41, 0xC1, 0xE8}, [&](int index) {
145 if (index != -1) { 97 AdjustRng(index - 0x52);
146 make_a_shaper = offset + index - 0x19; 98 AdjustRng(index - 0xB);
147 AdjustRng(make_a_shaper + 0x9); 99 });
148 AdjustRng(make_a_shaper + 0x35); 100
149 AdjustRng(make_a_shaper + 0x62);
150 }
151 }
152 if (/*Entity_Machine_Panel::*/init_pattern_data_lotus == -1) {
153 int index = find(data, {0x40, 0x55, 0x56, 0x48, 0x8D, 0x6C, 0x24, 0xB1});
154 if (index != -1) {
155 init_pattern_data_lotus = offset + index;
156 AdjustRng(init_pattern_data_lotus + 0x433);
157 AdjustRng(init_pattern_data_lotus + 0x45B);
158 AdjustRng(init_pattern_data_lotus + 0x5A7);
159 AdjustRng(init_pattern_data_lotus + 0x5D6);
160 AdjustRng(init_pattern_data_lotus + 0x6F6);
161 AdjustRng(init_pattern_data_lotus + 0xD17);
162 AdjustRng(init_pattern_data_lotus + 0xFDA);
163 }
164 }
165 if (/*Entity_Record_Player::*/reroll_lotus_eater_stuff == -1) {
166 int index = find(data, {0xB8, 0xAB, 0xAA, 0xAA, 0xAA, 0x41, 0xC1, 0xE8});
167 if (index != -1) {
168 reroll_lotus_eater_stuff = offset + index - 0x37;
169 AdjustRng(reroll_lotus_eater_stuff + 0x24);
170 AdjustRng(reroll_lotus_eater_stuff + 0x6B);
171 }
172 }
173 // These disable the random locations on timer panels, which would otherwise increment the RNG. 101 // These disable the random locations on timer panels, which would otherwise increment the RNG.
174 if (do_lotus_minutes == -1) { 102 // I'm writing 31 C0 (xor eax, eax), then 3 NOPs, which pretends the RNG returns 0.
175 int index = find(data, {0x0F, 0xBE, 0x6C, 0x08, 0xFF, 0x45}); 103 // do_lotus_minutes
176 if (index != -1) { 104 _memory->AddSigScan({0x0F, 0xBE, 0x6C, 0x08, 0xFF, 0x45}, [&](int index) {
177 do_lotus_minutes = offset + index - 0x2B; 105 _memory->WriteData<byte>({index + 0x410}, {0x31, 0xC0, 0x90, 0x90, 0x90});
178 _memory->WriteData<byte>({do_lotus_minutes + 0x43B}, {0x31, 0xC0, 0x90, 0x90, 0x90}); // xor eax, eax ;RNG returns 0 106 });
179 } 107 // do_lotus_tenths
180 } 108 _memory->AddSigScan({0x00, 0x04, 0x00, 0x00, 0x41, 0x8D, 0x50, 0x09}, [&](int index) {
181 if (do_lotus_tenths == -1) { 109 _memory->WriteData<byte>({index + 0xA2}, {0x31, 0xC0, 0x90, 0x90, 0x90});
182 int index = find(data, {0x00, 0x04, 0x00, 0x00, 0x41, 0x8D, 0x50, 0x09}); 110 });
183 if (index != -1) { 111 // do_lotus_eighths
184 do_lotus_tenths = offset + index - 0x61; 112 _memory->AddSigScan({0x75, 0xF5, 0x0F, 0xBE, 0x44, 0x08, 0xFF}, [&](int index) {
185 _memory->WriteData<byte>({do_lotus_tenths + 0x103}, {0x31, 0xC0, 0x90, 0x90, 0x90}); // xor eax, eax ;RNG returns 0 113 _memory->WriteData<byte>({index + 0x1AE}, {0x31, 0xC0, 0x90, 0x90, 0x90});
186 } 114 });
187 } 115 }
188 if (do_lotus_eighths == -1) { 116
189 int index = find(data, {0x75, 0xF5, 0x0F, 0xBE, 0x44, 0x08, 0xFF}); 117 int failed = _memory->ExecuteSigScans();
190 if (index != -1) { 118 if (failed != 0) {
191 do_lotus_eighths = offset + index - 0x39; 119 std::cout << "Failed " << failed << " sigscans";
192 _memory->WriteData<byte>({do_lotus_eighths + 0x1E7}, {0x31, 0xC0, 0x90, 0x90, 0x90}); // xor eax, eax ;RNG returns 0 120 }
193 } 121}
194 }
195 });
196} \ No newline at end of file