zaaarf/lillero
2
1
Fork 0
mirror of https://github.com/zaaarf/lillero.git synced 2024-11-21 23:14:51 +01:00
Code Issues Projects Releases Packages Wiki Activity

fix: failure to restart matcher after failing to find pattern once

Browse source
This commit is contained in:
zaaarf 2023-03-13 02:47:09 +01:00
parent 592e38066f
commit 30f7660ded
No known key found for this signature in database
GPG key ID: 82240E075E31FA4C
2 changed files with 14 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/main/java/ftbsc/lll/tools/PatternMatcher.java
View file

@ -78,31 +78,25 @@ public class PatternMatcher {
*/ */
public InsnSequence find(AbstractInsnNode node) { public InsnSequence find(AbstractInsnNode node) {
if(node != null) { if(node != null) {
int match = 0; AbstractInsnNode first;
AbstractInsnNode first = null; AbstractInsnNode last;
AbstractInsnNode last = null;
for(AbstractInsnNode cur = node; cur != null; cur = reverse ? cur.getPrevious() : cur.getNext()) { for(AbstractInsnNode cur = node; cur != null; cur = reverse ? cur.getPrevious() : cur.getNext()) {
if(ignoreLabels && cur.getType() == AbstractInsnNode.LABEL) continue; if(ignoreLabels && cur.getType() == AbstractInsnNode.LABEL) continue;
if(ignoreFrames && cur.getType() == AbstractInsnNode.FRAME) continue; if(ignoreFrames && cur.getType() == AbstractInsnNode.FRAME) continue;
if(ignoreLineNumbers && cur.getType() == AbstractInsnNode.LINE) continue; if(ignoreLineNumbers && cur.getType() == AbstractInsnNode.LINE) continue;
if(predicates.get(match) != null) { if(predicates.size() == 0)
if(predicates.get(match).test(cur)) {
match++;
if (first == null)
first = cur;
} else { //reset
first = null;
match = 0;
}
}
//check if we found the last one
if(match == predicates.size()) {
if(match == 0)
return new InsnSequence(cur); //match whatever return new InsnSequence(cur); //match whatever
first = cur;
last = cur; last = cur;
for(int match = 0; match < predicates.size(); match++) {
if(last == null) break;
if(!predicates.get(match).test(last)) break;
if(match == predicates.size() - 1) {
if(reverse) return new InsnSequence(last, first); //we are matching backwards if(reverse) return new InsnSequence(last, first); //we are matching backwards
else return new InsnSequence(first, last); else return new InsnSequence(first, last);
} }
last = reverse ? last.getPrevious() : last.getNext();
}
} }
} }
throw new PatternNotFoundException("Failed to find pattern!"); throw new PatternNotFoundException("Failed to find pattern!");

2
src/main/java/ftbsc/lll/tools/StackTools.java
View file

@ -43,7 +43,7 @@ public class StackTools implements Opcodes {
public static InsnList instantiate(String name, String desc, InsnList args) { public static InsnList instantiate(String name, String desc, InsnList args) {
InsnSequence list = new InsnSequence(); InsnSequence list = new InsnSequence();
list.add(new TypeInsnNode(NEW, name), new InsnNode(DUP)); list.add(new TypeInsnNode(NEW, name), new InsnNode(DUP));
if (args != null) list.add(args); if(args != null) list.add(args);
list.add(new MethodInsnNode(INVOKESPECIAL, name, "<init>", desc, false)); list.add(new MethodInsnNode(INVOKESPECIAL, name, "<init>", desc, false));
return list; return list;
} }