github-repos/testable-mock
1
0
Fork 0
mirror of https://github.com/alibaba/testable-mock.git synced 2025-03-23 16:20:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

handle bytecode jump opcode

Browse Source
This commit is contained in:
金戟 2021-02-24 17:18:10 +08:00
parent 190cf4cc3c
commit 8dfd0b4516
2 changed files with 54 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
testable-agent/src/main/java/com/alibaba/testable/agent/handler/SourceClassHandler.java
View File

@ -1,10 +1,12 @@
package com.alibaba.testable.agent.handler; package com.alibaba.testable.agent.handler;
import com.alibaba.testable.agent.model.MethodInfo; import com.alibaba.testable.agent.model.MethodInfo;
import com.alibaba.testable.agent.model.TravelStatus;
import com.alibaba.testable.agent.util.BytecodeUtil; import com.alibaba.testable.agent.util.BytecodeUtil;
import com.alibaba.testable.agent.util.ClassUtil; import com.alibaba.testable.agent.util.ClassUtil;
import com.alibaba.testable.agent.util.MethodUtil; import com.alibaba.testable.agent.util.MethodUtil;
import com.alibaba.testable.core.util.LogUtil; import com.alibaba.testable.core.util.LogUtil;
import org.objectweb.asm.Label;
import org.objectweb.asm.Opcodes; import org.objectweb.asm.Opcodes;
import org.objectweb.asm.tree.*; import org.objectweb.asm.tree.*;
@ -145,10 +147,36 @@ public class SourceClassHandler extends BaseClassHandler {
if (stackLevel < 0) { if (stackLevel < 0) {
return rangeEnd; return rangeEnd;
} }
Label labelToJump = null;
TravelStatus status = TravelStatus.Normal;
for (int i = rangeEnd - 1; i >= 0; i--) { for (int i = rangeEnd - 1; i >= 0; i--) {
stackLevel += getStackLevelChange(instructions[i]); switch (status) {
if (stackLevel < 0) { case Normal:
return i; if (instructions[i] instanceof FrameNode) {
status = TravelStatus.LookingForLabel;
} else {
stackLevel += getStackLevelChange(instructions[i]);
if (stackLevel < 0) {
return i;
}
}
break;
case LookingForLabel:
if (instructions[i] instanceof LabelNode) {
labelToJump = ((LabelNode)instructions[i]).getLabel();
status = TravelStatus.LookingForJump;
}
break;
case LookingForJump:
if (instructions[i] instanceof JumpInsnNode &&
((JumpInsnNode)instructions[i]).label.getLabel().equals(labelToJump)) {
stackLevel += getStackLevelChange(instructions[i]);
labelToJump = null;
status = TravelStatus.Normal;
}
break;
default:
break;
} }
} }
return -1; return -1;

23
testable-agent/src/main/java/com/alibaba/testable/agent/model/TravelStatus.java Normal file
View File

@ -0,0 +1,23 @@
package com.alibaba.testable.agent.model;
/**
* @author flin
*/
public enum TravelStatus {
/**
* traveling common bytecode
*/
Normal,
/**
* looking for label opcode
*/
LookingForLabel,
/**
* looking for jump opcode
*/
LookingForJump
}