116 lines
3.5 KiB
Java
116 lines
3.5 KiB
Java
/*
|
|
Copyright 2018-2023 Dmitry Isaenko
|
|
|
|
This file is part of NS-USBloader.
|
|
|
|
NS-USBloader is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
NS-USBloader is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with NS-USBloader. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
package nsusbloader.Utilities.patches.fs.finders;
|
|
|
|
import nsusbloader.Utilities.patches.AHeuristic;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.List;
|
|
import java.util.stream.Collectors;
|
|
|
|
public class HeuristicFsWizard {
|
|
private final List<AHeuristic> all;
|
|
private final List<AHeuristic> found;
|
|
private final List<AHeuristic> wantLessEntropy;
|
|
|
|
private final StringBuilder errorsAndNotes;
|
|
|
|
private int offset1 = -1;
|
|
private int offset2 = -1;
|
|
|
|
public HeuristicFsWizard(byte[] where) throws Exception{
|
|
this.errorsAndNotes = new StringBuilder();
|
|
|
|
this.all = Arrays.asList(
|
|
new HeuristicFs1(where),
|
|
new HeuristicFs2(where)
|
|
);
|
|
|
|
this.found = all.stream()
|
|
.filter(AHeuristic::isFound)
|
|
.collect(Collectors.toList());
|
|
|
|
if (found.isEmpty())
|
|
throw new Exception("Nothing found!");
|
|
|
|
this.wantLessEntropy = all.stream()
|
|
.filter(AHeuristic::wantLessEntropy)
|
|
.collect(Collectors.toList());
|
|
|
|
shareOffsetsWithEachOther();
|
|
|
|
assignOffset1();
|
|
assignOffset2();
|
|
}
|
|
|
|
private void shareOffsetsWithEachOther(){
|
|
for (AHeuristic es : wantLessEntropy) {
|
|
if (shareWithNext(es))
|
|
return;
|
|
}
|
|
}
|
|
private boolean shareWithNext(AHeuristic es){
|
|
try {
|
|
for (AHeuristic foundEs : found) {
|
|
if (es.setOffsetsNearby(foundEs.getOffset())) {
|
|
found.add(es);
|
|
wantLessEntropy.remove(es);
|
|
shareOffsetsWithEachOther();
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
catch (Exception e){ e.printStackTrace(); }
|
|
return false;
|
|
}
|
|
|
|
private void assignOffset1(){
|
|
try {
|
|
offset1 = all.get(0).getOffset();
|
|
}
|
|
catch (Exception e){ errorsAndNotes.append(e.getLocalizedMessage()).append("\n"); }
|
|
}
|
|
private void assignOffset2(){
|
|
try {
|
|
offset2 = all.get(1).getOffset();
|
|
}
|
|
catch (Exception e){ errorsAndNotes.append(e.getLocalizedMessage()).append("\n"); }
|
|
}
|
|
|
|
public String getErrorsAndNotes(){
|
|
return errorsAndNotes.toString();
|
|
}
|
|
|
|
public String getDebug(){
|
|
StringBuilder builder = new StringBuilder();
|
|
if (all.get(0).isFound()){
|
|
builder.append("\t\t-=== 1 ===-\n");
|
|
builder.append(all.get(0).getDetails());
|
|
}
|
|
if (all.get(1).isFound()){
|
|
builder.append("\t\t-=== 2 ===-\n");
|
|
builder.append(all.get(1).getDetails());
|
|
}
|
|
return builder.toString();
|
|
}
|
|
|
|
public int getOffset1() { return offset1; }
|
|
public int getOffset2() { return offset2; }
|
|
}
|