Fixed module_stomping

b1nhack · b1nhack · commit 5f82eebdb069 · 2023-09-16T00:15:58.000+08:00
diff --git a/module_stomping/src/main.rs b/module_stomping/src/main.rs
@@ -87,8 +87,7 @@ fn main() {
             addr_of_mut!(needed),
         );
         let count = (needed as usize) / size_of::<HMODULE>();
-        for i in 0..count {
-            let module = modules[i];
+        for module in modules.into_iter().take(count) {
             let mut name: [u8; 128] = zeroed();
             GetModuleBaseNameA(
                 handle,
@@ -123,7 +122,7 @@ fn main() {
                 );
             };
 
-            CloseHandle(handle);
         }
+        CloseHandle(handle);
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -87,8 +87,7 @@ fn main() {`
`87`	`87`	`addr_of_mut!(needed),`
`88`	`88`	`);`
`89`	`89`	`let count = (needed as usize) / size_of::<HMODULE>();`
`90`		`- for i in 0..count {`
`91`		`- let module = modules[i];`
	`90`	`+ for module in modules.into_iter().take(count) {`
`92`	`91`	`let mut name: [u8; 128] = zeroed();`
`93`	`92`	`GetModuleBaseNameA(`
`94`	`93`	`handle,`
`@@ -123,7 +122,7 @@ fn main() {`
`123`	`122`	`);`
`124`	`123`	`};`
`125`	`124`
`126`		`- CloseHandle(handle);`
`127`	`125`	`}`
	`126`	`+ CloseHandle(handle);`
`128`	`127`	`}`
`129`	`128`	`}`