Commit 1daf5d10 authored by Skylot's avatar Skylot

core: fix condition processing (resolve #25)

parent 9d2c0e4a
...@@ -175,6 +175,8 @@ public class IfMakerHelper { ...@@ -175,6 +175,8 @@ public class IfMakerHelper {
if (isInversionNeeded(currentIf, nextIf)) { if (isInversionNeeded(currentIf, nextIf)) {
nextIf = IfInfo.invert(nextIf); nextIf = IfInfo.invert(nextIf);
} }
} else {
return currentIf;
} }
} }
......
...@@ -45,7 +45,7 @@ public class IfRegionVisitor extends AbstractVisitor implements IRegionVisitor, ...@@ -45,7 +45,7 @@ public class IfRegionVisitor extends AbstractVisitor implements IRegionVisitor,
@Override @Override
public boolean visitRegion(MethodNode mth, IRegion region) { public boolean visitRegion(MethodNode mth, IRegion region) {
if (region instanceof IfRegion) { if (region instanceof IfRegion) {
return removeRedundantElseBlock((IfRegion) region); return removeRedundantElseBlock(mth, (IfRegion) region);
} }
return false; return false;
} }
...@@ -136,28 +136,38 @@ public class IfRegionVisitor extends AbstractVisitor implements IRegionVisitor, ...@@ -136,28 +136,38 @@ public class IfRegionVisitor extends AbstractVisitor implements IRegionVisitor,
} }
} }
private static boolean removeRedundantElseBlock(IfRegion ifRegion) { private static boolean removeRedundantElseBlock(MethodNode mth, IfRegion ifRegion) {
if (ifRegion.getElseRegion() != null if (ifRegion.getElseRegion() == null
&& !ifRegion.contains(AFlag.ELSE_IF_CHAIN) || ifRegion.contains(AFlag.ELSE_IF_CHAIN)
&& !ifRegion.getElseRegion().contains(AFlag.ELSE_IF_CHAIN) || ifRegion.getElseRegion().contains(AFlag.ELSE_IF_CHAIN)) {
&& hasBranchTerminator(ifRegion) return false;
&& insnsCount(ifRegion.getThenRegion()) < 2) { }
IRegion parent = ifRegion.getParent(); if (!hasBranchTerminator(ifRegion.getThenRegion())) {
Region newRegion = new Region(parent); return false;
if (parent.replaceSubBlock(ifRegion, newRegion)) { }
newRegion.add(ifRegion); // code style check:
newRegion.add(ifRegion.getElseRegion()); // will remove 'return;' from 'then' and 'else' with one instruction
ifRegion.setElseRegion(null); // see #jadx.tests.integration.conditions.TestConditions9
return true; if (mth.getReturnType() == ArgType.VOID
} && insnsCount(ifRegion.getThenRegion()) == 2
&& insnsCount(ifRegion.getElseRegion()) == 2) {
return false;
}
IRegion parent = ifRegion.getParent();
Region newRegion = new Region(parent);
if (parent.replaceSubBlock(ifRegion, newRegion)) {
newRegion.add(ifRegion);
newRegion.add(ifRegion.getElseRegion());
ifRegion.setElseRegion(null);
return true;
} }
return false; return false;
} }
private static boolean hasBranchTerminator(IfRegion ifRegion) { private static boolean hasBranchTerminator(IContainer region) {
// TODO: check for exception throw // TODO: check for exception throw
return RegionUtils.hasExitBlock(ifRegion.getThenRegion()) return RegionUtils.hasExitBlock(region)
|| RegionUtils.hasBreakInsn(ifRegion.getThenRegion()); || RegionUtils.hasBreakInsn(region);
} }
private static void invertIfRegion(IfRegion ifRegion) { private static void invertIfRegion(IfRegion ifRegion) {
......
package jadx.tests.integration.conditions;
import jadx.core.dex.nodes.ClassNode;
import jadx.tests.api.IntegrationTest;
import org.junit.Test;
import static jadx.tests.api.utils.JadxMatchers.containsOne;
import static org.hamcrest.CoreMatchers.containsString;
import static org.hamcrest.CoreMatchers.not;
import static org.junit.Assert.assertThat;
public class TestNestedIf2 extends IntegrationTest {
public static class TestCls {
static int executedCount = 0;
static boolean finished = false;
static int repeatCount = 2;
static boolean test(float delta, Object object) {
if (executedCount != repeatCount && isRun(delta, object)) {
if (finished) {
return true;
}
if (repeatCount == -1) {
++executedCount;
action();
return false;
}
++executedCount;
if (executedCount >= repeatCount) {
return true;
}
action();
}
return false;
}
public static void action() {
}
public static boolean isRun(float delta, Object object) {
return delta == 0;
}
}
@Test
public void test() {
ClassNode cls = getClassNode(TestCls.class);
String code = cls.getCode().toString();
assertThat(code, containsOne("if (executedCount != repeatCount && isRun(delta, object)) {"));
assertThat(code, containsOne("if (finished) {"));
assertThat(code, not(containsString("else")));
}
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment