chizui/Il2CppInspectorRedux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

IL2CPP: Dramatically speed up image scanning

Browse Source
This commit is contained in:
Katy Coe
2020-12-14 02:49:13 +01:00
parent 05ba1678c5
commit 14f4e034b0
1 changed files with 21 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
Il2CppInspector.Common/IL2CPP/ImageScan.cs
View File

@@ -1,44 +1,42 @@
using System; /*
using System.Collections; Copyright 2020 Katy Coe - http://www.djkaty.com - https://github.com/djkaty
All rights reserved.
*/
using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Diagnostics;
using System.Linq; using System.Linq;
using System.Text; using System.Text;
using NoisyCowStudios.Bin2Object;
namespace Il2CppInspector namespace Il2CppInspector
{ {
partial class Il2CppBinary partial class Il2CppBinary
{ {
// Find a sequence of bytes // Find a sequence of bytes
private int FindBytes(byte[] blob, byte[] signature, int step = 1, int startOffset = 0) { // Adapted from https://stackoverflow.com/a/332667
int sigMatch = 0; private int FindBytes(byte[] blob, byte[] signature, int requiredAlignment = 1, int startOffset = 0) {
int foundOffset = -1; var firstMatchByte = Array.IndexOf(blob, signature[0], startOffset);
var test = new byte[signature.Length];
for (int p = startOffset; p < blob.Length - signature.Length && foundOffset == -1; p += sigMatch == 0? step : 1) { while (firstMatchByte >= 0 && firstMatchByte <= blob.Length - signature.Length) {
sigMatch = blob[p] != signature[sigMatch] ? 0 : ++sigMatch; Buffer.BlockCopy(blob, firstMatchByte, test, 0, signature.Length);
if (test.SequenceEqual(signature) && firstMatchByte % requiredAlignment == 0)
return firstMatchByte;
if (sigMatch == 0) firstMatchByte = Array.IndexOf(blob, signature[0], firstMatchByte + 1);
p = p - p % step + step;
if (sigMatch == 0 && blob[p] == signature[0])
sigMatch++;
if (sigMatch == signature.Length)
foundOffset = p + 1 - signature.Length;
} }
return foundOffset; return -1;
} }
// Find all occurrences of a sequence of bytes // Find all occurrences of a sequence of bytes, using word alignment by default
private IEnumerable<uint> FindAllBytes(byte[] blob, byte[] signature, int step = 0) { private IEnumerable<uint> FindAllBytes(byte[] blob, byte[] signature, int alignment = 0) {
var ptrs = new List<uint>();
var offset = 0; var offset = 0;
while (offset != -1) { while (offset != -1) {
offset = FindBytes(blob, signature, step != 0 ? step : Image.Bits / 4, offset); offset = FindBytes(blob, signature, alignment != 0 ? alignment : Image.Bits / 8, offset);
if (offset != -1) { if (offset != -1) {
yield return (uint) offset; yield return (uint) offset;
offset += Image.Bits / 4; offset += Image.Bits / 8;
} }
} }
} }