[
{
"path": ".gitignore",
"content": "# Compiled class file\n*.class\n\n# Log file\n*.log\n\n# BlueJ files\n*.ctxt\n\n# Mobile Tools for Java (J2ME)\n.mtj.tmp/\n\n# Package Files #\n*.jar\n*.war\n*.nar\n*.ear\n*.zip\n*.tar.gz\n*.rar\n\n# virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml\nhs_err_pid*\n\n# IDEA\n\n.idea/\n*.iml\n*.kotlin_module\ntarget/\nlib/\nbuild/\ndist/\n.*\n*.data\n!.gitignore\n"
},
{
"path": "LICENSE",
"content": "MIT License\n\nCopyright (c) 2020 lsieun\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in all\ncopies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\nSOFTWARE.\n"
},
{
"path": "README.md",
"content": "# Learn Java ASM\n\n[](https://gitee.com/lsieun/learn-java-asm)\n[](https://github.com/lsieun/learn-java-asm)\n\n\n\n\n\n\n:maple_leaf: Java [ASM](https://asm.ow2.io/) is an open-source java library for manipulating bytecode.\n\n本项目旨在系统地介绍如何学习Java ASM的知识，主要涉及Core API、OPCODE和Tree API等内容。至于学习的预期目标就是，用一个形象的说法来讲，让字节码在你的手中“跳舞”：看看你的左手，一个完整的ClassFile拆解成不同粒度的字节码内容；看看你的右手，不同粒度的字节码内容又重新组织成一个ClassFile结构。\n\n```text\n _______ , _ _ \n (, /' /'/ /' `\\ ' ) _)\n /' /' / /' ._) // _/~/' \n /'____ . , ____ ,/' / (____ /'/_/~ /' \n _ /'/' )| / /' ) /`--,/ ) /' /~ /' \n /' ` /'/' /' | /' /' /' /' / /' /' /' \n(_____,/' (___,/(___|/(__(___,/(__ (,/' (_,(_____,/'(,/' (_, \n```\n\n---\n\n如果我们学会了Java ASM之后，可能还是需要一个具体的应用场景来进行使用，\n这个场景就是由 [Java Agent](https://docs.oracle.com/en/java/javase/25/docs/api/java.instrument/java/lang/instrument/Instrumentation.html) 开启的。\n\n那么，Java ASM和Java Agent这两者之间是什么关系呢？\nJava ASM是一个操作字节码的工具（tool），而Java Agent提供了修改字节码的机会（opportunity）。\n想像这样一个场景：\n有一个JVM正在运行，突然Java Agent在JVM上打开一扇大门，Java ASM通过大门冲进JVM里面，就要开始修改字节码了。\n\n```text\n.class --- Java ASM --- Java Agent --- JVM\n```\n\n再打个比方，Java ASM就是“一匹千里马”，而Java Agent就是“伯乐”。\n如果遇不到“伯乐”，可能“千里马”的才能就埋没了；正因为有了“伯乐”，“千里马”就有了施展才能的机会。\n\n```text\n世有伯乐，然后有千里马。\n千里马常有，而伯乐不常有。\n故虽有名马，祗辱于奴隶人之手，骈死于槽枥之间，不以千里称也。\n```\n\n---\n\n## 1. 如何使用\n\n### 1.1 代码下载\n\n[](https://gitee.com/lsieun/learn-java-asm)\n[](https://github.com/lsieun/learn-java-asm)\n\n从[Gitee](https://gitee.com/lsieun/learn-java-asm) 仓库下载代码，使用如下命令：\n\n```text\ngit clone https://gitee.com/lsieun/learn-java-asm\n```\n\n从[GitHub](https://github.com/lsieun/learn-java-asm) 仓库下载代码，使用如下命令：\n\n```text\ngit clone https://github.com/lsieun/learn-java-asm\n```\n\n### 1.2 开发环境\n\n[](./LICENSE)\n\n\n\n\n\n- [Git](https://git-scm.com/)\n- [Java 8](https://www.oracle.com/java/technologies/javase/javase8-archive-downloads.html)\n- [Apache Maven](https://maven.apache.org/)\n- [IntelliJ IDEA](https://www.jetbrains.com/idea/download/other.html) (Ultimate or Community Edition)\n\n在`learn-java-asm`项目当中，使用的ASM版本为`9.0`。如果想使用最新 版本，可以修改`pom.xml`文件中的`asm.version`属性：\n\n```text\n9.0\n```\n\n### 1.3. 运行代码\n\n在`learn-java-asm`项目当中，包含`main`方法的类主要位于`run`包（`src/main/java/run`）。\n\n## 2. 课程资料\n\n[](https://edu.51cto.com/lecturer/9210464.html)\n[](https://space.bilibili.com/1321054247)\n[](https://lsieun.github.io)\n\n- 《Java ASM系列一：Core API》\n - 文章地址： [lsieun.github.io](https://lsieun.github.io/java/asm/java-asm-season-01.html) | [51CTO](https://blog.51cto.com/lsieun/2924583)\n - 视频地址： [51CTO](https://edu.51cto.com/course/28517.html) | [Bilibili](https://space.bilibili.com/1321054247/channel/seriesdetail?sid=381716)\n- 《Java ASM系列二：OPCODE》\n - 文章地址：[lsieun.github.io](https://lsieun.github.io/java/asm/java-asm-season-02.html) | [51CTO](https://blog.51cto.com/lsieun/3273965)\n - 视频地址：[51CTO](https://edu.51cto.com/course/28870.html) | [Bilibili](https://space.bilibili.com/1321054247/channel/seriesdetail?sid=381716)\n- 《Java ASM系列三：Tree API》\n - 文章地址：[lsieun.github.io](https://lsieun.github.io/java/asm/java-asm-season-03.html) | [51CTO](https://blog.51cto.com/lsieun/4034588)\n - 视频地址：[51CTO](https://edu.51cto.com/course/29459.html) | [Bilibili](https://space.bilibili.com/1321054247/channel/seriesdetail?sid=381716)\n\n### 2.1. ASM的组成部分\n\n从组成结构上来说，Java ASM有Core API和Tree API两部分组成。\n\n```text\n ┌─── asm.jar\n │\n ┌─── Core API ─────────┼─── asm-util.jar\n │ │\n │ └─── asm-commons.jar\nJava ASM ───┤\n │\n │ ┌─── asm-tree.jar\n └─── Tree API ─────────┤\n └─── asm-analysis.jar\n```\n\n从依赖关系角度上说，Java ASM当中的各个`.jar`之间的依赖关系如下：\n\n```text\n┌────────────────────────────┬─────────────────────────────┐\n│ ┌───────┴────────┐ │\n│ util │ analysis │ commons │\n│ ┌──────┴────────────────┴──────┐ │\n│ │ tree │ │\n├─────────────┴──────────────────────────────┴─────────────┤\n│ core │\n└──────────────────────────────────────────────────────────┘\n```\n\n### 2.2. ASM能够做什么\n\n从应用的角度来说，Java ASM可以进行Class Generation、Class Transformation和Class Analysis三个类型的操作。\n\n```text\n ┌─── find potential bugs\n │\n ┌─── analysis ─────────┼─── detect unused code\n │ │\n │ └─── reverse engineer code\n │\nJava ASM ───┼─── generation\n │\n │ ┌─── optimize programs\n │ │\n └─── transformation ───┼─── obfuscate programs\n │\n └─── insert performance monitoring code\n```\n\n### 2.3. top, null和void\n\n在下表当中，top、null和void三者相对应的转换值：\n\n```text\n┌─────────────┬────────────────────────────┬────────────────────────────────┐\n│ .class │ ASM Type │ ASM Value in Frame │\n├─────────────┼────────────────────────────┼────────────────────────────────┤\n│ top │ null │ BasicValue.UNINITIALIZED_VALUE │\n├─────────────┼────────────────────────────┼────────────────────────────────┤\n│ aconst_null │ BasicInterpreter.NULL_TYPE │ BasicValue.REFERENCE_VALUE │\n├─────────────┼────────────────────────────┼────────────────────────────────┤\n│ void │ Type.VOID_TYPE │ null │\n└─────────────┴────────────────────────────┴────────────────────────────────┘\n```\n\n## 3. 注意事项\n\n### 3.1. 添加typo字典\n\n在编写代码的过程中，会遇到一些Typo提示，原因是`insn`等内容不是合法的单词。\n\n解决方法：借助于IntelliJ IDEA的[Spellchecking](https://www.jetbrains.com/help/idea/spellchecking.html) 的功能。\n\n操作步骤：\n\n- 第一步，在`Settings/Preferences`当中，找到`Editor | Natural Languages | Spelling`位置。\n- 第二步，在右侧的Custom dictionaries位置，添加**custom dictionary**，在`learn-java-asm`项目根目录下，有一个`accepted-words.dic`文件，添加该文件即可。\n\n配置完成之后，需要**重新启动IntelliJ IDEA**才能生效。\n\n### 3.2. 查看笔记\n\n在编写代码的过程中，为了方便理解代码，我添加了一些笔记，格式如下：\n\n```text\nNOTE: 希望这是一条有用的笔记\n```\n\n但是，在默认情况下，它并不会高亮显示，因此不容易被察觉到。\n\n解决方法：借助于IntelliJ IDEA的[TODO comments](https://www.jetbrains.com/help/idea/using-todo.html) 功能。\n\n操作步骤：\n\n- 第一步，在`Settings/Preferences`当中，找到`Editor | TODO`位置。\n- 第二步，在右侧的Patterns位置，添加以下内容：\n\n```text\n\\bnote\\b.*\n```\n\n配置完成之后，需要**重新启动IntelliJ IDEA**才能生效。\n\n### 3.3. 关闭调试信息\n\n在默认情况下，运行任何类，都会输出调试信息。在调试信息中，会带有`[DEBUG]`标识。\n\n如果想关闭调试信息，可以修改`lsieun.cst.Const`类的`DEBUG`字段值为`false`（默认值为`true`）：\n\n```java\npublic class Const {\n public static final boolean DEBUG = false;\n}\n```\n\n然后，执行`mvn clean compile`对类进行重新编译：\n\n```text\nmvn clean compile\n```\n\n等待编译完成之后，再次运行程序。\n\n## 4. 交流反馈\n\n- 如果您有好的想法，可以提issues\n- 如果您想贡献代码，可以进行fork\n- 如果您有其它问题，可以添加QQ群（参考联系方式）\n\n## 5. 联系方式\n\n[](https://lsieun.github.io/assets/images/contact/we-chat.jpg)\n[](https://lsieun.github.io/assets/images/contact/qq.png)\n[](https://lsieun.github.io/assets/images/contact/qq-group.jpg)\n[](https://jq.qq.com/?_wv=1027&k=yOBiOaJV)\n\n## 6. License\n\nThis project is licensed under the MIT License.\nSee the [LICENSE](./LICENSE) file for the full license text.\n"
},
{
"path": "accepted-words.dic",
"content": "bilibili\ngitee\ninsn\nlsieun\n"
},
{
"path": "pom.xml",
"content": "\n\n 4.0.0\n\n lsieun\n learn-java-asm\n 1.0-SNAPSHOT\n\n \n \n UTF-8\n UTF-8\n\n \n 1.8\n ${java.version}\n ${java.version}\n 9.9.1\n \n\n \n \n org.ow2.asm\n asm\n ${asm.version}\n \n \n org.ow2.asm\n asm-commons\n ${asm.version}\n \n \n org.ow2.asm\n asm-util\n ${asm.version}\n \n \n org.ow2.asm\n asm-tree\n ${asm.version}\n \n \n org.ow2.asm\n asm-analysis\n ${asm.version}\n \n \n\n \n \n \n \n org.apache.maven.plugins\n maven-compiler-plugin\n 3.13.0\n \n ${java.version}\n ${java.version}\n true\n \n -g\n -parameters\n \n \n \n\n \n \n org.apache.maven.plugins\n maven-source-plugin\n 3.3.1\n \n \n default-source-jar\n package\n \n jar\n \n \n \n \n \n \n\n"
},
{
"path": "src/main/java/lsieun/annotation/todo/ToDo.java",
"content": "package lsieun.annotation.todo;\n\nimport java.lang.annotation.Retention;\nimport java.lang.annotation.RetentionPolicy;\nimport java.lang.annotation.Target;\n\nimport static java.lang.annotation.ElementType.*;\n\n@Retention(RetentionPolicy.SOURCE)\n@Target({\n TYPE,\n FIELD,\n CONSTRUCTOR,\n METHOD,\n PARAMETER,\n LOCAL_VARIABLE,\n ANNOTATION_TYPE,\n PACKAGE,\n TYPE_PARAMETER,\n TYPE_USE,\n// MODULE,\n// RECORD_COMPONENT,\n})\npublic @interface ToDo {\n String[] value();\n}"
},
{
"path": "src/main/java/lsieun/asm/analysis/ControlFlowAnalyzer.java",
"content": "package lsieun.asm.analysis;\n\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.tree.*;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\n/**\n * 将 {@link org.objectweb.asm.tree.analysis.Analyzer}类的control flow analysis功能抽取出来.\n *\n * NOTE: 当前类抛弃了所有泛型相关的信息(V)，抛弃了Frame、抛弃了Interpreter、抛弃了Value。\n * 但是，它仍然能够实现control flow analysis的功能。\n *\n * @see org.objectweb.asm.tree.analysis.Analyzer\n */\npublic class ControlFlowAnalyzer implements Opcodes {\n\n private InsnList insnList;\n private int insnListSize;\n private List[] handlers;\n\n /**\n * 记录需要处理的instructions.\n *\n * NOTE: 这三个字段为一组，应该一起处理，最好是放到同一个方法里来处理。\n * 因此，我就加了三个新方法。\n * {@link #initInstructionsToProcess()}、{@link #addInstructionsToProcess(int)}和\n * {@link #removeInstructionsToProcess()}\n *\n * @see #initInstructionsToProcess()\n * @see #addInstructionsToProcess(int)\n * @see #removeInstructionsToProcess()\n */\n private boolean[] inInstructionsToProcess;\n private int[] instructionsToProcess;\n private int numInstructionsToProcess;\n\n public ControlFlowAnalyzer() {\n }\n\n public List getHandlers(final int insnIndex) {\n return handlers[insnIndex];\n }\n\n\n @SuppressWarnings(\"unchecked\")\n // NOTE: analyze方法的返回值类型变成了void类型。\n public void analyze(final String owner, final MethodNode method) throws AnalyzerException {\n if ((method.access & (ACC_ABSTRACT | ACC_NATIVE)) != 0) {\n return;\n }\n insnList = method.instructions;\n insnListSize = insnList.size();\n handlers = (List[]) new List[insnListSize];\n\n initInstructionsToProcess();\n\n // For each exception handler, and each instruction within its range, record in 'handlers' the\n // fact that execution can flow from this instruction to the exception handler.\n for (int i = 0; i < method.tryCatchBlocks.size(); ++i) {\n TryCatchBlockNode tryCatchBlock = method.tryCatchBlocks.get(i);\n int startIndex = insnList.indexOf(tryCatchBlock.start);\n int endIndex = insnList.indexOf(tryCatchBlock.end);\n for (int j = startIndex; j < endIndex; ++j) {\n List insnHandlers = handlers[j];\n if (insnHandlers == null) {\n insnHandlers = new ArrayList<>();\n handlers[j] = insnHandlers;\n }\n insnHandlers.add(tryCatchBlock);\n }\n }\n\n\n // Initializes the data structures for the control flow analysis.\n // NOTE: 调用addInstructionsToProcess方法，传入参数0，启动整个循环过程。\n addInstructionsToProcess(0);\n init(owner, method);\n\n // Control flow analysis.\n while (numInstructionsToProcess > 0) {\n // Get and remove one instruction from the list of instructions to process.\n int insnIndex = removeInstructionsToProcess();\n\n // Simulate the execution of this instruction.\n AbstractInsnNode insnNode = method.instructions.get(insnIndex);\n int insnOpcode = insnNode.getOpcode();\n int insnType = insnNode.getType();\n\n if (insnType == AbstractInsnNode.LABEL\n || insnType == AbstractInsnNode.LINE\n || insnType == AbstractInsnNode.FRAME) {\n newControlFlowEdge(insnIndex, insnIndex + 1);\n }\n else {\n if (insnNode instanceof JumpInsnNode) {\n JumpInsnNode jumpInsn = (JumpInsnNode) insnNode;\n if (insnOpcode != GOTO && insnOpcode != JSR) {\n newControlFlowEdge(insnIndex, insnIndex + 1);\n }\n int jumpInsnIndex = insnList.indexOf(jumpInsn.label);\n newControlFlowEdge(insnIndex, jumpInsnIndex);\n }\n else if (insnNode instanceof LookupSwitchInsnNode) {\n LookupSwitchInsnNode lookupSwitchInsn = (LookupSwitchInsnNode) insnNode;\n int targetInsnIndex = insnList.indexOf(lookupSwitchInsn.dflt);\n newControlFlowEdge(insnIndex, targetInsnIndex);\n for (int i = 0; i < lookupSwitchInsn.labels.size(); ++i) {\n LabelNode label = lookupSwitchInsn.labels.get(i);\n targetInsnIndex = insnList.indexOf(label);\n newControlFlowEdge(insnIndex, targetInsnIndex);\n }\n }\n else if (insnNode instanceof TableSwitchInsnNode) {\n TableSwitchInsnNode tableSwitchInsn = (TableSwitchInsnNode) insnNode;\n int targetInsnIndex = insnList.indexOf(tableSwitchInsn.dflt);\n newControlFlowEdge(insnIndex, targetInsnIndex);\n for (int i = 0; i < tableSwitchInsn.labels.size(); ++i) {\n LabelNode label = tableSwitchInsn.labels.get(i);\n targetInsnIndex = insnList.indexOf(label);\n newControlFlowEdge(insnIndex, targetInsnIndex);\n }\n }\n else if (insnOpcode != ATHROW && (insnOpcode < IRETURN || insnOpcode > RETURN)) {\n newControlFlowEdge(insnIndex, insnIndex + 1);\n }\n }\n\n List insnHandlers = handlers[insnIndex];\n if (insnHandlers != null) {\n for (TryCatchBlockNode tryCatchBlock : insnHandlers) {\n newControlFlowExceptionEdge(insnIndex, tryCatchBlock);\n }\n }\n\n }\n }\n\n\n protected void init(final String owner, final MethodNode method) throws AnalyzerException {\n // Nothing to do.\n }\n\n // NOTE: 这是一个新添加的方法\n private void initInstructionsToProcess() {\n inInstructionsToProcess = new boolean[insnListSize];\n instructionsToProcess = new int[insnListSize];\n numInstructionsToProcess = 0;\n }\n\n // NOTE: 这是一个新添加的方法\n private int removeInstructionsToProcess() {\n int insnIndex = this.instructionsToProcess[--numInstructionsToProcess];\n inInstructionsToProcess[insnIndex] = false;\n return insnIndex;\n }\n\n // NOTE: 这是一个新添加的方法\n private void addInstructionsToProcess(final int insnIndex) {\n if (!inInstructionsToProcess[insnIndex]) {\n inInstructionsToProcess[insnIndex] = true;\n instructionsToProcess[numInstructionsToProcess++] = insnIndex;\n }\n }\n\n protected void newControlFlowEdge(final int insnIndex, final int successorIndex) {\n // Nothing to do.\n addInstructionsToProcess(successorIndex);\n }\n\n protected void newControlFlowExceptionEdge(final int insnIndex, final TryCatchBlockNode tryCatchBlock) {\n newControlFlowExceptionEdge(insnIndex, insnList.indexOf(tryCatchBlock.handler));\n }\n\n protected void newControlFlowExceptionEdge(final int insnIndex, final int successorIndex) {\n // Nothing to do.\n addInstructionsToProcess(successorIndex);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/ControlFlowAnalyzer2.java",
"content": "package lsieun.asm.analysis;\n\nimport lsieun.asm.analysis.graph.InsnBlock;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\n\nimport java.util.List;\n\npublic class ControlFlowAnalyzer2 extends ControlFlowAnalyzer{\n private AbstractInsnNode[] nodeArray;\n public InsnBlock[] blocks;\n\n public ControlFlowAnalyzer2() {\n\n }\n\n @Override\n public void analyze(String owner, MethodNode method) throws AnalyzerException {\n nodeArray = method.instructions.toArray();\n int length = nodeArray.length;\n blocks = new InsnBlock[length];\n InsnText insnText = new InsnText();\n for (int i = 0; i < length; i++) {\n blocks[i] = getBlock(i);\n AbstractInsnNode node = nodeArray[i];\n List lines = insnText.toLines(node);\n blocks[i].addLines(lines);\n }\n\n super.analyze(owner, method);\n }\n\n @Override\n protected void newControlFlowEdge(int insnIndex, int successorIndex) {\n // 首先，处理自己的代码逻辑\n AbstractInsnNode insnNode = nodeArray[insnIndex];\n int insnOpcode = insnNode.getOpcode();\n int insnType = insnNode.getType();\n\n if (insnType == AbstractInsnNode.JUMP_INSN) {\n if ((insnIndex + 1) == successorIndex) {\n addNext(insnIndex, successorIndex);\n }\n else {\n addJump(insnIndex, successorIndex);\n }\n }\n else if (insnOpcode == LOOKUPSWITCH) {\n addJump(insnIndex, successorIndex);\n }\n else if (insnOpcode == TABLESWITCH) {\n addJump(insnIndex, successorIndex);\n }\n else if (insnOpcode == RET) {\n addJump(insnIndex, successorIndex);\n }\n else if (insnOpcode == ATHROW || (insnOpcode >= IRETURN && insnOpcode <= RETURN)) {\n assert false : \"should not be here\";\n removeNextAndJump(insnIndex);\n }\n else {\n addNext(insnIndex, successorIndex);\n }\n\n // 其次，调用父类的方法实现\n super.newControlFlowEdge(insnIndex, successorIndex);\n }\n\n private void addNext(int fromIndex, int toIndex) {\n InsnBlock currentBlock = getBlock(fromIndex);\n InsnBlock nextBlock = getBlock(toIndex);\n currentBlock.addNext(nextBlock);\n }\n\n private void addJump(int fromIndex, int toIndex) {\n InsnBlock currentBlock = getBlock(fromIndex);\n InsnBlock nextBlock = getBlock(toIndex);\n currentBlock.addJump(nextBlock);\n }\n\n private void removeNextAndJump(int insnIndex) {\n InsnBlock currentBlock = getBlock(insnIndex);\n currentBlock.nextBlockList.clear();\n currentBlock.jumpBlockList.clear();\n }\n\n private InsnBlock getBlock(int insnIndex) {\n InsnBlock block = blocks[insnIndex];\n if (block == null){\n block = new InsnBlock();\n blocks[insnIndex] = block;\n }\n return block;\n }\n\n public InsnBlock[] getBlocks() {\n return blocks;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/ControlFlowEdgeAnalyzer.java",
"content": "package lsieun.asm.analysis;\n\nimport lsieun.asm.analysis.graph.InsnBlock;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.*;\n\nimport java.util.List;\n\npublic class ControlFlowEdgeAnalyzer extends Analyzer {\n private AbstractInsnNode[] nodeArray;\n public InsnBlock[] blocks;\n\n public ControlFlowEdgeAnalyzer(Interpreter interpreter) {\n super(interpreter);\n }\n\n @Override\n public Frame[] analyze(String owner, MethodNode method) throws AnalyzerException {\n nodeArray = method.instructions.toArray();\n int length = nodeArray.length;\n blocks = new InsnBlock[length];\n InsnText insnText = new InsnText();\n for (int i = 0; i < length; i++) {\n blocks[i] = getBlock(i);\n AbstractInsnNode node = nodeArray[i];\n List lines = insnText.toLines(node);\n blocks[i].addLines(lines);\n }\n\n return super.analyze(owner, method);\n }\n\n @Override\n protected void newControlFlowEdge(int insnIndex, int successorIndex) {\n // 首先，处理自己的代码逻辑\n AbstractInsnNode insnNode = nodeArray[insnIndex];\n int insnOpcode = insnNode.getOpcode();\n int insnType = insnNode.getType();\n\n if (insnType == AbstractInsnNode.JUMP_INSN) {\n if ((insnIndex + 1) == successorIndex) {\n addNext(insnIndex, successorIndex);\n }\n else {\n addJump(insnIndex, successorIndex);\n }\n }\n else if (insnOpcode == LOOKUPSWITCH) {\n addJump(insnIndex, successorIndex);\n }\n else if (insnOpcode == TABLESWITCH) {\n addJump(insnIndex, successorIndex);\n }\n else if (insnOpcode == RET) {\n addJump(insnIndex, successorIndex);\n }\n else if (insnOpcode == ATHROW || (insnOpcode >= IRETURN && insnOpcode <= RETURN)) {\n assert false : \"should not be here\";\n removeNextAndJump(insnIndex);\n }\n else {\n addNext(insnIndex, successorIndex);\n }\n\n // 其次，调用父类的方法实现\n super.newControlFlowEdge(insnIndex, successorIndex);\n }\n\n private void addNext(int fromIndex, int toIndex) {\n InsnBlock currentBlock = getBlock(fromIndex);\n InsnBlock nextBlock = getBlock(toIndex);\n currentBlock.addNext(nextBlock);\n }\n\n private void addJump(int fromIndex, int toIndex) {\n InsnBlock currentBlock = getBlock(fromIndex);\n InsnBlock nextBlock = getBlock(toIndex);\n currentBlock.addJump(nextBlock);\n }\n\n private void removeNextAndJump(int insnIndex) {\n InsnBlock currentBlock = getBlock(insnIndex);\n currentBlock.nextBlockList.clear();\n currentBlock.jumpBlockList.clear();\n }\n\n private InsnBlock getBlock(int insnIndex) {\n InsnBlock block = blocks[insnIndex];\n if (block == null){\n block = new InsnBlock();\n blocks[insnIndex] = block;\n }\n return block;\n }\n\n public InsnBlock[] getBlocks() {\n return blocks;\n }\n\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/ControlFlowEdgeAnalyzer2.java",
"content": "package lsieun.asm.analysis;\n\nimport lsieun.asm.analysis.graph.InsnBlock;\nimport org.objectweb.asm.tree.analysis.Interpreter;\nimport org.objectweb.asm.tree.analysis.Value;\n\nimport java.util.ArrayList;\nimport java.util.HashSet;\nimport java.util.List;\nimport java.util.Set;\n\npublic class ControlFlowEdgeAnalyzer2 extends ControlFlowEdgeAnalyzer {\n\n public ControlFlowEdgeAnalyzer2(Interpreter interpreter) {\n super(interpreter);\n }\n\n @Override\n public InsnBlock[] getBlocks() {\n //（1）调用父类实现\n InsnBlock[] blocks = super.getBlocks();\n\n //（2）如果结果为空，则提前返回\n if (blocks == null || blocks.length < 1) {\n return blocks;\n }\n\n //（3）记录“分隔点”\n Set newBlockSet = new HashSet<>();\n int length = blocks.length;\n for (int i = 0; i < length; i++) {\n InsnBlock currentBlock = blocks[i];\n List nextBlockList = currentBlock.nextBlockList;\n List jumpBlockList = currentBlock.jumpBlockList;\n\n boolean hasNext = false;\n boolean hasJump = false;\n\n if (nextBlockList.size() > 0) {\n hasNext = true;\n }\n\n if (jumpBlockList.size() > 0) {\n hasJump = true;\n }\n\n if (!hasNext && (i + 1) < length) {\n newBlockSet.add(blocks[i + 1]);\n }\n\n if (hasJump) {\n newBlockSet.addAll(jumpBlockList);\n\n if (hasNext) {\n newBlockSet.add(blocks[i + 1]);\n }\n }\n }\n\n //（4）利用“分隔点”，合并成不同的分组\n List resultList = new ArrayList<>();\n for (int i = 0; i < length; i++) {\n InsnBlock currentBlock = blocks[i];\n\n if (i == 0) {\n resultList.add(currentBlock);\n }\n else if (newBlockSet.contains(currentBlock)) {\n resultList.add(currentBlock);\n }\n else {\n int size = resultList.size();\n InsnBlock lastBlock = resultList.get(size - 1);\n lastBlock.lines.addAll(currentBlock.lines);\n lastBlock.jumpBlockList.clear();\n lastBlock.jumpBlockList.addAll(currentBlock.jumpBlockList);\n lastBlock.nextBlockList.clear();\n lastBlock.nextBlockList.addAll(currentBlock.nextBlockList);\n }\n }\n\n return resultList.toArray(new InsnBlock[0]);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/ControlFlowGraphAnalyzer.java",
"content": "package lsieun.asm.analysis;\n\nimport lsieun.asm.analysis.graph.InsnBlock;\nimport org.objectweb.asm.tree.*;\n\nimport java.util.*;\n\nimport static org.objectweb.asm.Opcodes.*;\n\npublic class ControlFlowGraphAnalyzer {\n private final Set jumpLabelSet = new HashSet<>();\n private final Map> jumpLabelMap = new HashMap<>();\n private final Map labelInBlockMap = new HashMap<>();\n private final InsnText insnText = new InsnText();\n\n private InsnBlock currentBlock;\n private final List nodeList = new ArrayList<>();\n private final List blockList = new ArrayList<>();\n\n public ControlFlowGraphAnalyzer() {\n this.currentBlock = new InsnBlock();\n }\n\n public void analyze(MethodNode mn) {\n findJumpLabel(mn);\n instruction2Block(mn);\n linkBlocks();\n }\n\n\n private void findJumpLabel(MethodNode mn) {\n InsnList instructions = mn.instructions;\n\n for (AbstractInsnNode node : instructions) {\n if (node instanceof JumpInsnNode) {\n // 当前block与跳转目标的关系\n JumpInsnNode currentNode = (JumpInsnNode) node;\n jumpLabelSet.add(currentNode.label);\n }\n else if (node instanceof TableSwitchInsnNode) {\n // 当前block与跳转目标的关系\n TableSwitchInsnNode currentNode = (TableSwitchInsnNode) node;\n jumpLabelSet.add(currentNode.dflt);\n jumpLabelSet.addAll(currentNode.labels);\n }\n else if (node instanceof LookupSwitchInsnNode) {\n // 当前block与跳转目标的关系\n LookupSwitchInsnNode currentNode = (LookupSwitchInsnNode) node;\n jumpLabelSet.add(currentNode.dflt);\n jumpLabelSet.addAll(currentNode.labels);\n }\n }\n\n List tryCatchBlocks = mn.tryCatchBlocks;\n for (TryCatchBlockNode node : tryCatchBlocks) {\n jumpLabelSet.add(node.handler);\n }\n }\n\n private void instruction2Block(MethodNode mn) {\n InsnList instructions = mn.instructions;\n\n for (AbstractInsnNode node : instructions) {\n int opcode = node.getOpcode();\n\n if (node instanceof JumpInsnNode) {\n nodeList.add(node);\n\n // 当前block收集数据完成\n completeBlock();\n\n // 当前block与跳转目标的关系\n JumpInsnNode currentNode = (JumpInsnNode) node;\n addJumpFromBlockToLabel(currentNode.label);\n\n // 当前block与下一个block的关系\n InsnBlock nextBlock = new InsnBlock();\n if ((opcode >= IFEQ && opcode <= IF_ACMPNE) || (opcode >= IFNULL && opcode <= IFNONNULL)) {\n currentBlock.nextBlockList.add(nextBlock);\n }\n\n // 下一个block成为当前block\n currentBlock = nextBlock;\n }\n else if (node instanceof TableSwitchInsnNode) {\n nodeList.add(node);\n\n // 当前block收集数据完成\n completeBlock();\n\n // 当前block与跳转目标的关系\n TableSwitchInsnNode currentNode = (TableSwitchInsnNode) node;\n int min = currentNode.min;\n int max = currentNode.max;\n for (int i = min; i <= max; i++) {\n addJumpFromBlockToLabel(currentNode.labels.get(i - min));\n }\n addJumpFromBlockToLabel(currentNode.dflt);\n\n // 下一个block成为当前block\n InsnBlock nextBlock = new InsnBlock();\n currentBlock = nextBlock;\n }\n else if (node instanceof LookupSwitchInsnNode) {\n nodeList.add(node);\n\n // 当前block收集数据完成\n completeBlock();\n\n // 当前block与跳转目标的关系\n LookupSwitchInsnNode currentNode = (LookupSwitchInsnNode) node;\n List labels = currentNode.labels;\n for (LabelNode labelNode : labels) {\n addJumpFromBlockToLabel(labelNode);\n }\n addJumpFromBlockToLabel(currentNode.dflt);\n\n // 下一个block成为当前block\n InsnBlock nextBlock = new InsnBlock();\n currentBlock = nextBlock;\n }\n else if (node instanceof LabelNode) {\n LabelNode currentNode = (LabelNode) node;\n\n if (jumpLabelSet.contains(currentNode)) {\n if (nodeList.size() > 0) {\n // 当前block收集数据完成\n completeBlock();\n\n // 下一个block成为当前block\n InsnBlock nextBlock = new InsnBlock();\n currentBlock.nextBlockList.add(nextBlock);\n currentBlock = nextBlock;\n }\n\n }\n\n nodeList.add(node);\n labelInBlockMap.put(currentNode, currentBlock);\n }\n else if (node instanceof InsnNode) {\n nodeList.add(node);\n\n // 当前block收集数据完成\n if ((opcode >= IRETURN && opcode <= RETURN) || (opcode == ATHROW)) {\n completeBlock();\n\n InsnBlock nextBlock = new InsnBlock();\n currentBlock = nextBlock;\n }\n\n }\n else {\n nodeList.add(node);\n }\n }\n\n if (nodeList.size() > 0) {\n // 当前block收集数据完成\n completeBlock();\n }\n }\n\n private void linkBlocks() {\n for (Map.Entry> item : jumpLabelMap.entrySet()) {\n LabelNode key = item.getKey();\n Set set = item.getValue();\n\n InsnBlock targetBlock = labelInBlockMap.get(key);\n for (InsnBlock block : set) {\n block.jumpBlockList.add(targetBlock);\n }\n }\n }\n\n\n private void addJumpFromBlockToLabel(LabelNode labelNode) {\n Set list = jumpLabelMap.get(labelNode);\n if (list != null) {\n list.add(currentBlock);\n\n }\n else {\n list = new HashSet<>();\n list.add(currentBlock);\n jumpLabelMap.put(labelNode, list);\n }\n }\n\n private void completeBlock() {\n for (AbstractInsnNode node : nodeList) {\n List lines = insnText.toLines(node);\n currentBlock.addLines(lines);\n }\n\n nodeList.clear();\n blockList.add(currentBlock);\n }\n\n public InsnBlock[] getBlocks() {\n return blockList.toArray(new InsnBlock[0]);\n }\n\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/DataFlowAnalyzer.java",
"content": "package lsieun.asm.analysis;\n\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.*;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\nimport org.objectweb.asm.tree.analysis.Frame;\nimport org.objectweb.asm.tree.analysis.Interpreter;\nimport org.objectweb.asm.tree.analysis.Value;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\npublic class DataFlowAnalyzer implements Opcodes {\n\n private final Interpreter interpreter;\n\n // 数据输入：指令集和异常处理\n private InsnList insnList;\n private int insnListSize;\n private List[] handlers;\n\n // 中间状态：记录需要哪一个指令需要处理\n private boolean[] inInstructionsToProcess;\n private int[] instructionsToProcess;\n private int numInstructionsToProcess;\n\n // 数据输出：最终的返回结果\n private Frame[] frames;\n\n public DataFlowAnalyzer(final Interpreter interpreter) {\n this.interpreter = interpreter;\n }\n\n public List getHandlers(final int insnIndex) {\n return handlers[insnIndex];\n }\n\n public Frame[] getFrames() {\n return frames;\n }\n\n\n @SuppressWarnings(\"unchecked\")\n public Frame[] analyze(final String owner, final MethodNode method) throws AnalyzerException {\n if ((method.access & (ACC_ABSTRACT | ACC_NATIVE)) != 0) {\n frames = (Frame[]) new Frame[0];\n return frames;\n }\n insnList = method.instructions;\n insnListSize = insnList.size();\n handlers = (List[]) new List[insnListSize];\n frames = (Frame[]) new Frame[insnListSize];\n inInstructionsToProcess = new boolean[insnListSize];\n instructionsToProcess = new int[insnListSize];\n numInstructionsToProcess = 0;\n\n // For each exception handler, and each instruction within its range, record in 'handlers' the\n // fact that execution can flow from this instruction to the exception handler.\n for (int i = 0; i < method.tryCatchBlocks.size(); ++i) {\n TryCatchBlockNode tryCatchBlock = method.tryCatchBlocks.get(i);\n int startIndex = insnList.indexOf(tryCatchBlock.start);\n int endIndex = insnList.indexOf(tryCatchBlock.end);\n for (int j = startIndex; j < endIndex; ++j) {\n List insnHandlers = handlers[j];\n if (insnHandlers == null) {\n insnHandlers = new ArrayList<>();\n handlers[j] = insnHandlers;\n }\n insnHandlers.add(tryCatchBlock);\n }\n }\n\n // Initializes the data structures for the control flow analysis.\n Frame currentFrame = computeInitialFrame(owner, method);\n merge(0, currentFrame);\n init(owner, method);\n\n // Control flow analysis.\n while (numInstructionsToProcess > 0) {\n // Get and remove one instruction from the list of instructions to process.\n int insnIndex = instructionsToProcess[--numInstructionsToProcess];\n Frame oldFrame = frames[insnIndex];\n inInstructionsToProcess[insnIndex] = false;\n\n // Simulate the execution of this instruction.\n AbstractInsnNode insnNode = null;\n try {\n insnNode = method.instructions.get(insnIndex);\n int insnOpcode = insnNode.getOpcode();\n int insnType = insnNode.getType();\n\n if (insnType == AbstractInsnNode.LABEL\n || insnType == AbstractInsnNode.LINE\n || insnType == AbstractInsnNode.FRAME) {\n merge(insnIndex + 1, oldFrame);\n newControlFlowEdge(insnIndex, insnIndex + 1);\n }\n else {\n currentFrame.init(oldFrame).execute(insnNode, interpreter);\n\n if (insnNode instanceof JumpInsnNode) {\n JumpInsnNode jumpInsn = (JumpInsnNode) insnNode;\n if (insnOpcode != GOTO && insnOpcode != JSR) {\n currentFrame.initJumpTarget(insnOpcode, /* target = */ null);\n merge(insnIndex + 1, currentFrame);\n newControlFlowEdge(insnIndex, insnIndex + 1);\n }\n int jumpInsnIndex = insnList.indexOf(jumpInsn.label);\n currentFrame.initJumpTarget(insnOpcode, jumpInsn.label);\n merge(jumpInsnIndex, currentFrame);\n newControlFlowEdge(insnIndex, jumpInsnIndex);\n }\n else if (insnNode instanceof LookupSwitchInsnNode) {\n LookupSwitchInsnNode lookupSwitchInsn = (LookupSwitchInsnNode) insnNode;\n int targetInsnIndex = insnList.indexOf(lookupSwitchInsn.dflt);\n currentFrame.initJumpTarget(insnOpcode, lookupSwitchInsn.dflt);\n merge(targetInsnIndex, currentFrame);\n newControlFlowEdge(insnIndex, targetInsnIndex);\n for (int i = 0; i < lookupSwitchInsn.labels.size(); ++i) {\n LabelNode label = lookupSwitchInsn.labels.get(i);\n targetInsnIndex = insnList.indexOf(label);\n currentFrame.initJumpTarget(insnOpcode, label);\n merge(targetInsnIndex, currentFrame);\n newControlFlowEdge(insnIndex, targetInsnIndex);\n }\n }\n else if (insnNode instanceof TableSwitchInsnNode) {\n TableSwitchInsnNode tableSwitchInsn = (TableSwitchInsnNode) insnNode;\n int targetInsnIndex = insnList.indexOf(tableSwitchInsn.dflt);\n currentFrame.initJumpTarget(insnOpcode, tableSwitchInsn.dflt);\n merge(targetInsnIndex, currentFrame);\n newControlFlowEdge(insnIndex, targetInsnIndex);\n for (int i = 0; i < tableSwitchInsn.labels.size(); ++i) {\n LabelNode label = tableSwitchInsn.labels.get(i);\n currentFrame.initJumpTarget(insnOpcode, label);\n targetInsnIndex = insnList.indexOf(label);\n merge(targetInsnIndex, currentFrame);\n newControlFlowEdge(insnIndex, targetInsnIndex);\n }\n }\n else if (insnOpcode != ATHROW && (insnOpcode < IRETURN || insnOpcode > RETURN)) {\n merge(insnIndex + 1, currentFrame);\n newControlFlowEdge(insnIndex, insnIndex + 1);\n }\n }\n\n List insnHandlers = handlers[insnIndex];\n if (insnHandlers != null) {\n for (TryCatchBlockNode tryCatchBlock : insnHandlers) {\n Type catchType;\n if (tryCatchBlock.type == null) {\n catchType = Type.getObjectType(\"java/lang/Throwable\");\n }\n else {\n catchType = Type.getObjectType(tryCatchBlock.type);\n }\n if (newControlFlowExceptionEdge(insnIndex, tryCatchBlock)) {\n Frame handler = newFrame(oldFrame);\n handler.clearStack();\n handler.push(interpreter.newExceptionValue(tryCatchBlock, handler, catchType));\n merge(insnList.indexOf(tryCatchBlock.handler), handler);\n }\n }\n }\n }\n catch (AnalyzerException e) {\n throw new AnalyzerException(e.node, \"Error at instruction \" + insnIndex + \": \" + e.getMessage(), e);\n }\n catch (RuntimeException e) {\n // DontCheck(IllegalCatch): can't be fixed, for backward compatibility.\n throw new AnalyzerException(insnNode, \"Error at instruction \" + insnIndex + \": \" + e.getMessage(), e);\n }\n }\n\n return frames;\n }\n\n private Frame computeInitialFrame(final String owner, final MethodNode method) {\n Frame frame = newFrame(method.maxLocals, method.maxStack);\n int currentLocal = 0;\n boolean isInstanceMethod = (method.access & ACC_STATIC) == 0;\n if (isInstanceMethod) {\n Type ownerType = Type.getObjectType(owner);\n frame.setLocal(currentLocal, interpreter.newParameterValue(isInstanceMethod, currentLocal, ownerType));\n currentLocal++;\n }\n Type[] argumentTypes = Type.getArgumentTypes(method.desc);\n for (Type argumentType : argumentTypes) {\n frame.setLocal(\n currentLocal,\n interpreter.newParameterValue(isInstanceMethod, currentLocal, argumentType));\n currentLocal++;\n if (argumentType.getSize() == 2) {\n frame.setLocal(currentLocal, interpreter.newEmptyValue(currentLocal));\n currentLocal++;\n }\n }\n while (currentLocal < method.maxLocals) {\n frame.setLocal(currentLocal, interpreter.newEmptyValue(currentLocal));\n currentLocal++;\n }\n frame.setReturn(interpreter.newReturnTypeValue(Type.getReturnType(method.desc)));\n return frame;\n }\n\n\n protected void init(final String owner, final MethodNode method) throws AnalyzerException {\n // Nothing to do.\n }\n\n protected Frame newFrame(final int numLocals, final int numStack) {\n return new Frame<>(numLocals, numStack);\n }\n\n protected Frame newFrame(final Frame frame) {\n return new Frame<>(frame);\n }\n\n protected void newControlFlowEdge(final int insnIndex, final int successorIndex) {\n // Nothing to do.\n }\n\n protected boolean newControlFlowExceptionEdge(final int insnIndex, final TryCatchBlockNode tryCatchBlock) {\n return newControlFlowExceptionEdge(insnIndex, insnList.indexOf(tryCatchBlock.handler));\n }\n\n protected boolean newControlFlowExceptionEdge(final int insnIndex, final int successorIndex) {\n return true;\n }\n\n // -----------------------------------------------------------------------------------------------\n\n private void merge(final int insnIndex, final Frame frame)\n throws AnalyzerException {\n boolean changed;\n Frame oldFrame = frames[insnIndex];\n if (oldFrame == null) {\n frames[insnIndex] = newFrame(frame);\n changed = true;\n }\n else {\n changed = oldFrame.merge(frame, interpreter);\n }\n\n if (changed && !inInstructionsToProcess[insnIndex]) {\n inInstructionsToProcess[insnIndex] = true;\n instructionsToProcess[numInstructionsToProcess++] = insnIndex;\n }\n }\n\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/InsnText.java",
"content": "package lsieun.asm.analysis;\n\nimport lsieun.utils.OpcodeConst;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.*;\n\nimport java.util.ArrayList;\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\n\nimport static org.objectweb.asm.Opcodes.BIPUSH;\nimport static org.objectweb.asm.Opcodes.SIPUSH;\n\npublic class InsnText {\n private static final String INSTRUCTION_FORMAT = \"%s %s\";\n\n private final Map labelNames = new HashMap<>();\n\n public List toLines(AbstractInsnNode[] array) {\n List resultList = new ArrayList<>();\n if (array == null || array.length < 1) {\n return resultList;\n }\n\n for (AbstractInsnNode node : array) {\n List lines = toLines(node);\n resultList.addAll(lines);\n }\n return resultList;\n }\n\n public List toLines(AbstractInsnNode node) {\n List resultList = new ArrayList<>();\n if (node instanceof TableSwitchInsnNode) {\n TableSwitchInsnNode currentNode = (TableSwitchInsnNode) node;\n List list = table_switch_node_to_str_list(currentNode);\n resultList.addAll(list);\n }\n else if (node instanceof LookupSwitchInsnNode) {\n LookupSwitchInsnNode currentNode = (LookupSwitchInsnNode) node;\n List list = lookup_switch_node_to_str_list(currentNode);\n resultList.addAll(list);\n }\n else {\n String item = node_to_str(node);\n resultList.add(item);\n }\n\n return resultList;\n }\n\n public String node_to_str(AbstractInsnNode currentNode) {\n if (currentNode instanceof InsnNode) {\n return getOpcodeName(currentNode);\n }\n else if (currentNode instanceof IntInsnNode) {\n int opcode = currentNode.getOpcode();\n String opcodeName = getOpcodeName(currentNode);\n IntInsnNode node = (IntInsnNode) currentNode;\n int operand = node.operand;\n if (opcode == BIPUSH || opcode == SIPUSH) {\n return String.format(INSTRUCTION_FORMAT, opcodeName, operand);\n }\n else {\n final String firstArg;\n switch (operand) {\n case 4: {\n firstArg = \"4 (boolean)\";\n break;\n }\n case 5: {\n firstArg = \"5 (char)\";\n break;\n }\n case 6: {\n firstArg = \"6 (float)\";\n break;\n }\n case 7: {\n firstArg = \"7 (double)\";\n break;\n }\n case 8: {\n firstArg = \"8 (byte)\";\n break;\n }\n case 9: {\n firstArg = \"9 (short)\";\n break;\n }\n case 10: {\n firstArg = \"10 (int)\";\n break;\n }\n case 11: {\n firstArg = \"11 (long)\";\n break;\n }\n default:\n throw new RuntimeException(\"atype is not supported: \" + operand);\n }\n return String.format(INSTRUCTION_FORMAT, opcodeName, firstArg);\n }\n }\n else if (currentNode instanceof VarInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n VarInsnNode node = (VarInsnNode) currentNode;\n int var = node.var;\n if (var >= 0 && var <= 3) {\n return String.format(\"%s_%d\", opcodeName, var);\n }\n else {\n return String.format(INSTRUCTION_FORMAT, opcodeName, var);\n }\n }\n else if (currentNode instanceof TypeInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n TypeInsnNode node = (TypeInsnNode) currentNode;\n String type = getSimpleName(node.desc);\n return String.format(INSTRUCTION_FORMAT, opcodeName, type);\n }\n else if (currentNode instanceof FieldInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n FieldInsnNode node = (FieldInsnNode) currentNode;\n String type = getSimpleName(node.owner);\n return String.format(\"%s %s.%s\", opcodeName, type, node.name);\n }\n else if (currentNode instanceof MethodInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n MethodInsnNode node = (MethodInsnNode) currentNode;\n String type = getSimpleName(node.owner);\n return String.format(\"%s %s.%s\", opcodeName, type, node.name);\n }\n else if (currentNode instanceof InvokeDynamicInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n InvokeDynamicInsnNode node = (InvokeDynamicInsnNode) currentNode;\n Type methodType = Type.getMethodType(node.desc);\n Type returnType = methodType.getReturnType();\n String type = getSimpleName(returnType.getInternalName());\n return String.format(\"%s %s.%s\", opcodeName, type, node.name);\n }\n else if (currentNode instanceof JumpInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n JumpInsnNode node = (JumpInsnNode) currentNode;\n String labelName = getLabelName(node.label);\n return String.format(INSTRUCTION_FORMAT, opcodeName, labelName);\n }\n else if (currentNode instanceof LabelNode) {\n LabelNode node = (LabelNode) currentNode;\n return getLabelName(node);\n }\n else if (currentNode instanceof LdcInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n LdcInsnNode node = (LdcInsnNode) currentNode;\n Object cst = node.cst;\n if (cst instanceof Integer) {\n return String.format(\"%s %s(int)\", opcodeName, cst);\n }\n else if (cst instanceof Float) {\n return String.format(\"%s %s(float)\", opcodeName, cst);\n }\n else if (cst instanceof Long) {\n return String.format(\"%s %s(long)\", opcodeName, cst);\n }\n else if (cst instanceof Double) {\n return String.format(\"%s %s(double)\", opcodeName, cst);\n }\n else if (cst instanceof String) {\n return String.format(\"%s \\\"%s\\\"\", opcodeName, cst);\n }\n else if (cst instanceof Class) {\n return String.format(\"%s %s(class)\", opcodeName, cst);\n }\n else {\n return String.format(\"%s %s\", opcodeName, cst);\n }\n }\n else if (currentNode instanceof IincInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n IincInsnNode node = (IincInsnNode) currentNode;\n return String.format(\"%s %d %d\", opcodeName, node.var, node.incr);\n }\n else if (currentNode instanceof MultiANewArrayInsnNode) {\n String opcodeName = getOpcodeName(currentNode);\n MultiANewArrayInsnNode node = (MultiANewArrayInsnNode) currentNode;\n String type = getSimpleName(node.desc);\n return String.format(INSTRUCTION_FORMAT, opcodeName, type);\n }\n else if (currentNode instanceof FrameNode) {\n return \"FrameNode\";\n }\n else if (currentNode instanceof LineNumberNode) {\n return \"LineNumberNode\";\n }\n else {\n System.out.println(currentNode.getClass());\n }\n return currentNode.toString();\n }\n\n public List table_switch_node_to_str_list(TableSwitchInsnNode currentNode) {\n String opcodeName = getOpcodeName(currentNode);\n int min = currentNode.min;\n int max = currentNode.max;\n\n List list = new ArrayList<>();\n list.add(String.format(\"%s {\", opcodeName));\n for (int i = min; i <= max; i++) {\n LabelNode labelNode = currentNode.labels.get(i - min);\n String labelName = getLabelName(labelNode);\n list.add(String.format(\" %d: %s\", i, labelName));\n }\n list.add(String.format(\" default: %s\", getLabelName(currentNode.dflt)));\n list.add(\"}\");\n return list;\n }\n\n public List lookup_switch_node_to_str_list(LookupSwitchInsnNode currentNode) {\n String opcodeName = getOpcodeName(currentNode);\n List keys = currentNode.keys;\n int size = keys.size();\n\n List list = new ArrayList<>();\n list.add(String.format(\"%s {\", opcodeName));\n for (int i = 0; i < size; i++) {\n int caseValue = keys.get(i);\n LabelNode labelNode = currentNode.labels.get(i);\n String labelName = getLabelName(labelNode);\n list.add(String.format(\" %d: %s\", caseValue, labelName));\n }\n list.add(String.format(\" default: %s\", getLabelName(currentNode.dflt)));\n list.add(\"}\");\n return list;\n }\n\n private String getLabelName(LabelNode labelNode) {\n String labelName = labelNames.get(labelNode);\n if (labelName == null) {\n labelName = \"L\" + labelNames.size();\n labelNames.put(labelNode, labelName);\n }\n return labelName;\n }\n\n private static String getOpcodeName(AbstractInsnNode currentNode) {\n int opcode = currentNode.getOpcode();\n return OpcodeConst.getOpcodeName(opcode);\n }\n\n private static String getSimpleName(String descriptor) {\n int squareIndex = descriptor.lastIndexOf(\"[\");\n String prefix = descriptor.substring(0, squareIndex + 1);\n\n String simpleName = descriptor.substring(squareIndex + 1);\n if (simpleName.startsWith(\"L\") && simpleName.endsWith(\";\")) {\n simpleName = simpleName.substring(1, simpleName.length() - 1);\n }\n\n int slashIndex = simpleName.lastIndexOf(\"/\");\n simpleName = simpleName.substring(slashIndex + 1);\n\n return prefix + simpleName;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/MockAnalyzer.java",
"content": "package lsieun.asm.analysis;\n\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.*;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\nimport org.objectweb.asm.tree.analysis.Frame;\nimport org.objectweb.asm.tree.analysis.Interpreter;\nimport org.objectweb.asm.tree.analysis.Value;\n\npublic class MockAnalyzer implements Opcodes {\n private final Interpreter interpreter;\n\n public MockAnalyzer(Interpreter interpreter) {\n this.interpreter = interpreter;\n }\n\n @SuppressWarnings(\"unchecked\")\n public Frame[] analyze(String owner, MethodNode method) throws AnalyzerException {\n // 第一步，如果是abstract或native方法，则直接返回。\n if ((method.access & (ACC_ABSTRACT | ACC_NATIVE)) != 0) {\n return (Frame[]) new Frame[0];\n }\n\n // 第二步，定义局部变量\n // （1）数据输入：获取指令集\n InsnList insnList = method.instructions;\n int size = insnList.size();\n\n // （2）中间状态：记录需要哪一个指令需要处理\n boolean[] instructionsToProcess = new boolean[size];\n\n // （3）数据输出：最终的返回结果\n Frame[] frames = (Frame[]) new Frame[size];\n\n // 第三步，开始计算\n // （1）开始计算：根据方法的参数，计算方法的初始Frame\n Frame currentFrame = computeInitialFrame(owner, method);\n merge(frames, 0, currentFrame, instructionsToProcess);\n\n // （2）开始计算：根据方法的每一条指令，计算相应的Frame\n while (getCount(instructionsToProcess) > 0) {\n // 获取需要处理的指令索引（insnIndex）和旧的Frame（oldFrame）\n int insnIndex = getFirst(instructionsToProcess);\n Frame oldFrame = frames[insnIndex];\n instructionsToProcess[insnIndex] = false;\n\n // 模拟每一条指令的执行\n try {\n AbstractInsnNode insnNode = method.instructions.get(insnIndex);\n int insnOpcode = insnNode.getOpcode();\n int insnType = insnNode.getType();\n\n // 这三者并不是真正的指令，分别表示Label、LineNumberTable和Frame\n if (insnType == AbstractInsnNode.LABEL\n || insnType == AbstractInsnNode.LINE\n || insnType == AbstractInsnNode.FRAME) {\n merge(frames, insnIndex + 1, oldFrame, instructionsToProcess);\n }\n else {\n // 这里是真正的指令\n currentFrame.init(oldFrame).execute(insnNode, interpreter);\n\n if (insnNode instanceof JumpInsnNode) {\n JumpInsnNode jumpInsn = (JumpInsnNode) insnNode;\n // if之后的语句\n if (insnOpcode != GOTO) {\n merge(frames, insnIndex + 1, currentFrame, instructionsToProcess);\n }\n\n // if和goto跳转之后的位置\n int jumpInsnIndex = insnList.indexOf(jumpInsn.label);\n merge(frames, jumpInsnIndex, currentFrame, instructionsToProcess);\n }\n else if (insnNode instanceof LookupSwitchInsnNode) {\n LookupSwitchInsnNode lookupSwitchInsn = (LookupSwitchInsnNode) insnNode;\n\n // lookupswitch的default情况\n int targetInsnIndex = insnList.indexOf(lookupSwitchInsn.dflt);\n merge(frames, targetInsnIndex, currentFrame, instructionsToProcess);\n\n // lookupswitch的各种case情况\n for (int i = 0; i < lookupSwitchInsn.labels.size(); ++i) {\n LabelNode label = lookupSwitchInsn.labels.get(i);\n targetInsnIndex = insnList.indexOf(label);\n merge(frames, targetInsnIndex, currentFrame, instructionsToProcess);\n }\n }\n else if (insnNode instanceof TableSwitchInsnNode) {\n TableSwitchInsnNode tableSwitchInsn = (TableSwitchInsnNode) insnNode;\n\n // tableswitch的default情况\n int targetInsnIndex = insnList.indexOf(tableSwitchInsn.dflt);\n merge(frames, targetInsnIndex, currentFrame, instructionsToProcess);\n\n // tableswitch的各种case情况\n for (int i = 0; i < tableSwitchInsn.labels.size(); ++i) {\n LabelNode label = tableSwitchInsn.labels.get(i);\n targetInsnIndex = insnList.indexOf(label);\n merge(frames, targetInsnIndex, currentFrame, instructionsToProcess);\n }\n }\n else if (insnOpcode != ATHROW && (insnOpcode < IRETURN || insnOpcode > RETURN)) {\n merge(frames, insnIndex + 1, currentFrame, instructionsToProcess);\n }\n }\n }\n catch (AnalyzerException e) {\n throw new AnalyzerException(e.node, \"Error at instruction \" + insnIndex + \": \" + e.getMessage(), e);\n }\n }\n\n return frames;\n }\n\n private int getCount(boolean[] array) {\n int count = 0;\n for (boolean flag : array) {\n if (flag) {\n count++;\n }\n }\n return count;\n }\n\n private int getFirst(boolean[] array) {\n int length = array.length;\n for (int i = 0; i < length; i++) {\n boolean flag = array[i];\n if (flag) {\n return i;\n }\n }\n return -1;\n }\n\n private Frame computeInitialFrame(final String owner, final MethodNode method) {\n Frame frame = new Frame<>(method.maxLocals, method.maxStack);\n int currentLocal = 0;\n\n // 第一步，判断是否需要存储this变量\n boolean isInstanceMethod = (method.access & ACC_STATIC) == 0;\n if (isInstanceMethod) {\n Type ownerType = Type.getObjectType(owner);\n V value = interpreter.newParameterValue(isInstanceMethod, currentLocal, ownerType);\n frame.setLocal(currentLocal, value);\n currentLocal++;\n }\n\n // 第二步，将方法的参数存入到local variable内\n Type[] argumentTypes = Type.getArgumentTypes(method.desc);\n for (Type argumentType : argumentTypes) {\n V value = interpreter.newParameterValue(isInstanceMethod, currentLocal, argumentType);\n frame.setLocal(currentLocal, value);\n currentLocal++;\n if (argumentType.getSize() == 2) {\n frame.setLocal(currentLocal, interpreter.newEmptyValue(currentLocal));\n currentLocal++;\n }\n }\n\n // 第三步，将local variable的剩余位置填补上空值\n while (currentLocal < method.maxLocals) {\n frame.setLocal(currentLocal, interpreter.newEmptyValue(currentLocal));\n currentLocal++;\n }\n\n // 第四步，设置返回值类型\n frame.setReturn(interpreter.newReturnTypeValue(Type.getReturnType(method.desc)));\n return frame;\n }\n\n /**\n * Merge old frame with new frame.\n *\n * @param frames 所有的frame信息。\n * @param insnIndex 当前指令的索引。\n * @param newFrame 新的frame\n * @param instructionsToProcess 记录哪一条指令需要处理\n * @throws AnalyzerException 分析错误，抛出此异常\n */\n private void merge(Frame[] frames, int insnIndex, Frame newFrame, boolean[] instructionsToProcess) throws AnalyzerException {\n boolean changed;\n Frame oldFrame = frames[insnIndex];\n if (oldFrame == null) {\n frames[insnIndex] = new Frame<>(newFrame);\n changed = true;\n }\n else {\n changed = oldFrame.merge(newFrame, interpreter);\n }\n\n if (changed && !instructionsToProcess[insnIndex]) {\n instructionsToProcess[insnIndex] = true;\n }\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/RemoveDeadCodeNode.java",
"content": "package lsieun.asm.analysis;\n\nimport lsieun.asm.tree.transformer.MethodTransformer;\nimport org.objectweb.asm.ClassVisitor;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.ClassNode;\nimport org.objectweb.asm.tree.LabelNode;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.*;\n\npublic class RemoveDeadCodeNode extends ClassNode {\n public RemoveDeadCodeNode(int api, ClassVisitor cv) {\n super(api);\n this.cv = cv;\n }\n\n @Override\n public void visitEnd() {\n // 首先，处理自己的代码逻辑\n MethodTransformer mt = new MethodRemoveDeadCodeTransformer(name, null);\n for (MethodNode mn : methods) {\n mt.transform(mn);\n }\n\n // 其次，调用父类的方法实现\n super.visitEnd();\n\n // 最后，向后续ClassVisitor传递\n if (cv != null) {\n accept(cv);\n }\n }\n\n private static class MethodRemoveDeadCodeTransformer extends MethodTransformer {\n private final String owner;\n\n public MethodRemoveDeadCodeTransformer(String owner, MethodTransformer mt) {\n super(mt);\n this.owner = owner;\n }\n\n @Override\n public void transform(MethodNode mn) {\n // 首先，处理自己的代码逻辑\n Analyzer analyzer = new Analyzer<>(new BasicInterpreter());\n try {\n analyzer.analyze(owner, mn);\n Frame[] frames = analyzer.getFrames();\n AbstractInsnNode[] insnNodes = mn.instructions.toArray();\n for (int i = 0; i < frames.length; i++) {\n if (frames[i] == null && !(insnNodes[i] instanceof LabelNode)) {\n mn.instructions.remove(insnNodes[i]);\n }\n }\n }\n catch (AnalyzerException ex) {\n ex.printStackTrace();\n }\n\n // 其次，调用父类的方法实现\n super.transform(mn);\n }\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/RemoveDeadCodeVisitor.java",
"content": "package lsieun.asm.analysis;\n\nimport org.objectweb.asm.ClassVisitor;\nimport org.objectweb.asm.MethodVisitor;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.LabelNode;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.*;\n\npublic class RemoveDeadCodeVisitor extends ClassVisitor {\n private String owner;\n\n public RemoveDeadCodeVisitor(int api, ClassVisitor cv) {\n super(api, cv);\n }\n\n @Override\n public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {\n super.visit(version, access, name, signature, superName, interfaces);\n this.owner = name;\n }\n\n @Override\n public MethodVisitor visitMethod(int access, String name, String descriptor, String signature, String[] exceptions) {\n MethodVisitor mv = super.visitMethod(access, name, descriptor, signature, exceptions);\n if (mv != null) {\n mv = new MethodRemoveDeadCodeAdapter(api, owner, access, name, descriptor, mv);\n }\n return mv;\n }\n\n private static class MethodRemoveDeadCodeAdapter extends MethodVisitor {\n private final String owner;\n private final MethodVisitor next;\n\n public MethodRemoveDeadCodeAdapter(int api, String owner, int access, String name, String desc, MethodVisitor next) {\n super(api, new MethodNode(access, name, desc, null, null));\n this.owner = owner;\n this.next = next;\n }\n\n @Override\n public void visitEnd() {\n // 首先，处理自己的代码逻辑\n MethodNode mn = (MethodNode) mv;\n Analyzer analyzer = new Analyzer<>(new BasicInterpreter());\n try {\n analyzer.analyze(owner, mn);\n Frame[] frames = analyzer.getFrames();\n AbstractInsnNode[] insnNodes = mn.instructions.toArray();\n for (int i = 0; i < frames.length; i++) {\n if (frames[i] == null && !(insnNodes[i] instanceof LabelNode)) {\n mn.instructions.remove(insnNodes[i]);\n }\n }\n }\n catch (AnalyzerException ex) {\n ex.printStackTrace();\n }\n\n // 其次，调用父类的方法实现\n super.visitEnd();\n\n // 最后，向后续MethodVisitor传递\n if (next != null) {\n mn.accept(next);\n }\n }\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/RemoveUnusedCastNode.java",
"content": "package lsieun.asm.analysis;\n\nimport lsieun.asm.tree.transformer.MethodTransformer;\nimport org.objectweb.asm.ClassVisitor;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.*;\nimport org.objectweb.asm.tree.analysis.*;\n\nimport static org.objectweb.asm.Opcodes.CHECKCAST;\n\npublic class RemoveUnusedCastNode extends ClassNode {\n public RemoveUnusedCastNode(int api, ClassVisitor cv) {\n super(api);\n this.cv = cv;\n }\n\n @Override\n public void visitEnd() {\n // 首先，处理自己的代码逻辑\n MethodTransformer mt = new MethodRemoveUnusedCastTransformer(name, null);\n for (MethodNode mn : methods) {\n if (\"\".equals(mn.name) || \"\".equals(mn.name)) {\n continue;\n }\n InsnList insns = mn.instructions;\n if (insns.size() == 0) {\n continue;\n }\n mt.transform(mn);\n }\n\n // 其次，调用父类的方法实现\n super.visitEnd();\n\n // 最后，向后续ClassVisitor传递\n if (cv != null) {\n accept(cv);\n }\n }\n\n private static class MethodRemoveUnusedCastTransformer extends MethodTransformer {\n private final String owner;\n\n public MethodRemoveUnusedCastTransformer(String owner, MethodTransformer mt) {\n super(mt);\n this.owner = owner;\n }\n\n @Override\n public void transform(MethodNode mn) {\n // 首先，处理自己的代码逻辑\n Analyzer analyzer = new Analyzer<>(new SimpleVerifier());\n try {\n Frame[] frames = analyzer.analyze(owner, mn);\n AbstractInsnNode[] insnNodes = mn.instructions.toArray();\n for (int i = 0; i < insnNodes.length; i++) {\n AbstractInsnNode insn = insnNodes[i];\n if (insn.getOpcode() == CHECKCAST) {\n Frame f = frames[i];\n if (f != null && f.getStackSize() > 0) {\n BasicValue operand = f.getStack(f.getStackSize() - 1);\n Class to = getClass(((TypeInsnNode) insn).desc);\n Class from = getClass(operand.getType());\n if (to.isAssignableFrom(from)) {\n mn.instructions.remove(insn);\n }\n }\n }\n }\n }\n catch (AnalyzerException ex) {\n ex.printStackTrace();\n }\n\n // 其次，调用父类的方法实现\n super.transform(mn);\n }\n\n private static Class getClass(String desc) {\n try {\n return Class.forName(desc.replace('/', '.'));\n }\n catch (ClassNotFoundException ex) {\n throw new RuntimeException(ex.toString());\n }\n }\n\n private static Class getClass(Type t) {\n if (t.getSort() == Type.OBJECT) {\n return getClass(t.getInternalName());\n }\n return getClass(t.getDescriptor());\n }\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/RemoveUnusedCastVisitor.java",
"content": "package lsieun.asm.analysis;\n\nimport org.objectweb.asm.ClassVisitor;\nimport org.objectweb.asm.MethodVisitor;\nimport org.objectweb.asm.commons.AnalyzerAdapter;\n\nimport static org.objectweb.asm.Opcodes.*;\n\npublic class RemoveUnusedCastVisitor extends ClassVisitor {\n private String owner;\n\n public RemoveUnusedCastVisitor(int api, ClassVisitor classVisitor) {\n super(api, classVisitor);\n }\n\n @Override\n public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {\n super.visit(version, access, name, signature, superName, interfaces);\n this.owner = name;\n }\n\n @Override\n public MethodVisitor visitMethod(int access, String name, String descriptor, String signature, String[] exceptions) {\n MethodVisitor mv = super.visitMethod(access, name, descriptor, signature, exceptions);\n if (mv != null && !name.equals(\"\")) {\n boolean isAbstractMethod = (access & ACC_ABSTRACT) != 0;\n boolean isNativeMethod = (access & ACC_NATIVE) != 0;\n if (!isAbstractMethod && !isNativeMethod) {\n RemoveUnusedCastAdapter adapter = new RemoveUnusedCastAdapter(api, mv);\n adapter.aa = new AnalyzerAdapter(owner, access, name, descriptor, adapter);\n return adapter.aa;\n }\n }\n\n return mv;\n }\n\n private static class RemoveUnusedCastAdapter extends MethodVisitor {\n public AnalyzerAdapter aa;\n\n public RemoveUnusedCastAdapter(int api, MethodVisitor methodVisitor) {\n super(api, methodVisitor);\n }\n\n @Override\n public void visitTypeInsn(int opcode, String type) {\n // 首先，处理自己的代码逻辑\n if (opcode == CHECKCAST) {\n Class to = getClass(type);\n if (aa.stack != null && aa.stack.size() > 0) {\n Object operand = aa.stack.get(aa.stack.size() - 1);\n if (operand instanceof String) {\n Class from = getClass((String) operand);\n if (to.isAssignableFrom(from)) {\n return;\n }\n }\n }\n }\n\n // 其次，调用父类的方法实现\n super.visitTypeInsn(opcode, type);\n }\n\n private static Class getClass(String desc) {\n try {\n return Class.forName(desc.replace('/', '.'));\n }\n catch (ClassNotFoundException ex) {\n throw new RuntimeException(ex.toString());\n }\n }\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/cc/CyclomaticComplexity.java",
"content": "package lsieun.asm.analysis.cc;\n\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.*;\n\npublic class CyclomaticComplexity {\n public static int getCyclomaticComplexity(String owner, MethodNode mn) throws AnalyzerException {\n // 第一步，获取Frame信息\n Analyzer analyzer = new CyclomaticComplexityAnalyzer<>(new BasicInterpreter());\n Frame[] frames = analyzer.analyze(owner, mn);\n\n // 第二步，计算复杂度\n int edges = 0;\n int nodes = 0;\n for (Frame frame : frames) {\n if (frame != null) {\n edges += ((CyclomaticComplexityFrame) frame).successors.size();\n nodes += 1;\n }\n }\n return edges - nodes + 2;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/cc/CyclomaticComplexityAnalyzer.java",
"content": "package lsieun.asm.analysis.cc;\n\nimport org.objectweb.asm.tree.analysis.*;\n\npublic class CyclomaticComplexityAnalyzer extends Analyzer {\n public CyclomaticComplexityAnalyzer(Interpreter interpreter) {\n super(interpreter);\n }\n\n @Override\n protected Frame newFrame(int numLocals, int numStack) {\n return new CyclomaticComplexityFrame<>(numLocals, numStack);\n }\n\n @Override\n protected Frame newFrame(Frame frame) {\n return new CyclomaticComplexityFrame<>(frame);\n }\n\n @Override\n protected void newControlFlowEdge(int insnIndex, int successorIndex) {\n CyclomaticComplexityFrame frame = (CyclomaticComplexityFrame) getFrames()[insnIndex];\n frame.successors.add((CyclomaticComplexityFrame) getFrames()[successorIndex]);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/cc/CyclomaticComplexityFrame.java",
"content": "package lsieun.asm.analysis.cc;\n\nimport org.objectweb.asm.tree.analysis.Frame;\nimport org.objectweb.asm.tree.analysis.Value;\n\nimport java.util.HashSet;\nimport java.util.Set;\n\npublic class CyclomaticComplexityFrame extends Frame {\n public Set> successors = new HashSet<>();\n\n public CyclomaticComplexityFrame(int numLocals, int numStack) {\n super(numLocals, numStack);\n }\n\n public CyclomaticComplexityFrame(Frame frame) {\n super(frame);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/cfg/ControlFlowBuilder.java",
"content": "package lsieun.asm.analysis.cfg;\n\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\n\nimport static org.objectweb.asm.Opcodes.ACC_ABSTRACT;\nimport static org.objectweb.asm.Opcodes.ACC_NATIVE;\n\npublic final class ControlFlowBuilder {\n public static ControlFlowGraph buildCFG(String className, MethodNode methodNode) throws AnalyzerException {\n if ((methodNode.access & (ACC_ABSTRACT | ACC_NATIVE)) != 0) {\n throw new RuntimeException(\"method body is empty\");\n }\n\n int size = methodNode.instructions.size();\n SimpleEdgeCreator edgeCreator = new SimpleEdgeCreator(size);\n FramelessAnalyzer myAnalyzer = new FramelessAnalyzer(edgeCreator);\n myAnalyzer.analyze(methodNode);\n\n\n int[][] transitions = new int[edgeCreator.transitions.length][];\n for (int i = 0; i < transitions.length; i++) {\n transitions[i] = edgeCreator.transitions[i].toNativeArray();\n }\n int[][] errorTransitions = new int[edgeCreator.errorTransitions.length][];\n for (int i = 0; i < errorTransitions.length; i++) {\n errorTransitions[i] = edgeCreator.errorTransitions[i].toNativeArray();\n }\n\n return new ControlFlowGraph(transitions, errorTransitions);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/cfg/ControlFlowGraph.java",
"content": "package lsieun.asm.analysis.cfg;\n\nimport java.util.Arrays;\nimport java.util.Formatter;\n\npublic final class ControlFlowGraph {\n public final int[][] transitions;\n public final int[][] errorTransitions;\n\n public ControlFlowGraph(int[][] transitions, int[][] errorTransitions) {\n this.transitions = transitions;\n this.errorTransitions = errorTransitions;\n }\n\n @Override\n public String toString() {\n StringBuilder sb = new StringBuilder();\n Formatter fm = new Formatter(sb);\n fm.format(\"ControlFlowGraph%n\");\n\n fm.format(\"Normal Transitions%n\");\n int normalLength = transitions.length;\n for (int i = 0; i < normalLength; i++) {\n int[] array = transitions[i];\n fm.format(\"%03d %s%n\", i, Arrays.toString(array));\n }\n\n fm.format(\"Error Transitions%n\");\n int errorLength = transitions.length;\n for (int i = 0; i < errorLength; i++) {\n int[] array = errorTransitions[i];\n fm.format(\"%03d %s%n\", i, Arrays.toString(array));\n }\n return sb.toString();\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/cfg/Edge.java",
"content": "package lsieun.asm.analysis.cfg;\n\npublic final class Edge {\n public final int from;\n public final int to;\n\n public Edge(int from, int to) {\n this.from = from;\n this.to = to;\n }\n\n @Override\n public boolean equals(Object obj) {\n if (this == obj) return true;\n if (!(obj instanceof Edge)) {\n return false;\n }\n Edge edge = (Edge) obj;\n return (from == edge.from) && (to == edge.to);\n }\n\n @Override\n public int hashCode() {\n return 31 * from + to;\n }\n}"
},
{
"path": "src/main/java/lsieun/asm/analysis/cfg/EdgeCreator.java",
"content": "package lsieun.asm.analysis.cfg;\n\npublic interface EdgeCreator {\n void newControlFlowEdge(int insn, int successor);\n\n void newControlFlowExceptionEdge(int insn, int successor, boolean npe);\n}"
},
{
"path": "src/main/java/lsieun/asm/analysis/cfg/FramelessAnalyzer.java",
"content": "package lsieun.asm.analysis.cfg;\n\nimport org.objectweb.asm.tree.*;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\n\nimport java.util.ArrayList;\nimport java.util.HashSet;\nimport java.util.List;\nimport java.util.Set;\n\nimport static org.objectweb.asm.Opcodes.*;\n\n/**\n * Specialized version of {@link org.objectweb.asm.tree.analysis.Analyzer}.\n * Calculation of fix-point of frames is removed, since frames are not needed to build control flow graph.\n * So, the main point here is handling of try-catch-finally blocks.\n */\npublic class FramelessAnalyzer {\n private static final Set NPE_HANDLERS = new HashSet<>();\n\n static {\n NPE_HANDLERS.add(\"java/lang/Throwable\");\n NPE_HANDLERS.add(\"java/lang/Exception\");\n NPE_HANDLERS.add(\"java/lang/RuntimeException\");\n NPE_HANDLERS.add(\"java/lang/NullPointerException\");\n }\n\n protected boolean[] wasQueued;\n protected boolean[] queued;\n protected int[] queue;\n protected int top;\n protected final EdgeCreator myEdgeCreator;\n\n public FramelessAnalyzer(EdgeCreator creator) {\n myEdgeCreator = creator;\n }\n\n public void analyze(MethodNode m) throws AnalyzerException {\n int n = m.instructions.size();\n if ((m.access & (ACC_ABSTRACT | ACC_NATIVE)) != 0 || n == 0) {\n return;\n }\n InsnList insns = m.instructions;\n List[] handlers = newListArray(n);\n queued = new boolean[n];\n wasQueued = new boolean[n];\n queue = new int[n];\n top = 0;\n\n // computes exception handlers for each instruction\n for (TryCatchBlockNode tcb : m.tryCatchBlocks) {\n int begin = insns.indexOf(tcb.start);\n int end = insns.indexOf(tcb.end);\n for (int j = begin; j < end; ++j) {\n List insnHandlers = handlers[j];\n if (insnHandlers == null) {\n insnHandlers = new ArrayList<>();\n handlers[j] = insnHandlers;\n }\n insnHandlers.add(tcb);\n }\n }\n\n merge(0);\n // control flow analysis\n while (top > 0) {\n int insn = queue[--top];\n\n queued[insn] = false;\n\n AbstractInsnNode insnNode = null;\n try {\n insnNode = m.instructions.get(insn);\n int insnOpcode = insnNode.getOpcode();\n int insnType = insnNode.getType();\n\n if (insnType == AbstractInsnNode.LABEL || insnType == AbstractInsnNode.LINE || insnType == AbstractInsnNode.FRAME) {\n merge(insn + 1);\n myEdgeCreator.newControlFlowEdge(insn, insn + 1);\n }\n else {\n\n if (insnNode instanceof JumpInsnNode) {\n JumpInsnNode j = (JumpInsnNode) insnNode;\n if (insnOpcode != GOTO && insnOpcode != JSR) {\n merge(insn + 1);\n myEdgeCreator.newControlFlowEdge(insn, insn + 1);\n }\n int jump = insns.indexOf(j.label);\n if (insnOpcode == JSR) {\n throw new AnalyzerException(insnNode, \"ret is not supported\");\n }\n else {\n merge(jump);\n }\n myEdgeCreator.newControlFlowEdge(insn, jump);\n }\n else if (insnNode instanceof LookupSwitchInsnNode) {\n LookupSwitchInsnNode lsi = (LookupSwitchInsnNode) insnNode;\n int jump = insns.indexOf(lsi.dflt);\n merge(jump);\n myEdgeCreator.newControlFlowEdge(insn, jump);\n for (int j = 0; j < lsi.labels.size(); ++j) {\n LabelNode label = lsi.labels.get(j);\n jump = insns.indexOf(label);\n merge(jump);\n myEdgeCreator.newControlFlowEdge(insn, jump);\n }\n }\n else if (insnNode instanceof TableSwitchInsnNode) {\n TableSwitchInsnNode tsi = (TableSwitchInsnNode) insnNode;\n int jump = insns.indexOf(tsi.dflt);\n merge(jump);\n myEdgeCreator.newControlFlowEdge(insn, jump);\n for (int j = 0; j < tsi.labels.size(); ++j) {\n LabelNode label = tsi.labels.get(j);\n jump = insns.indexOf(label);\n merge(jump);\n myEdgeCreator.newControlFlowEdge(insn, jump);\n }\n }\n else if (insnOpcode == RET) {\n throw new AnalyzerException(insnNode, \"ret is not supported\");\n }\n else if (insnOpcode != ATHROW && (insnOpcode < IRETURN || insnOpcode > RETURN)) {\n merge(insn + 1);\n myEdgeCreator.newControlFlowEdge(insn, insn + 1);\n }\n }\n\n List insnHandlers = handlers[insn];\n if (insnHandlers != null) {\n for (TryCatchBlockNode tcb : insnHandlers) {\n myEdgeCreator.newControlFlowExceptionEdge(insn, insns.indexOf(tcb.handler), NPE_HANDLERS.contains(tcb.type));\n merge(insns.indexOf(tcb.handler));\n }\n }\n }\n catch (AnalyzerException e) {\n throw new AnalyzerException(e.node, \"Error at instruction \" + insn + \": \" + e.getMessage(), e);\n }\n catch (Exception e) {\n throw new AnalyzerException(insnNode, \"Error at instruction \" + insn + \": \" + e.getMessage(), e);\n }\n }\n }\n\n // -------------------------------------------------------------------------\n\n protected void merge(int insn) {\n boolean changes = false;\n\n if (!wasQueued[insn]) {\n wasQueued[insn] = true;\n changes = true;\n }\n\n if (changes && !queued[insn]) {\n queued[insn] = true;\n queue[top++] = insn;\n }\n }\n\n public static List[] newListArray(int size) {\n @SuppressWarnings(\"unchecked\")\n List[] a = (List[]) new List[size];\n return a;\n }\n}"
},
{
"path": "src/main/java/lsieun/asm/analysis/cfg/SimpleEdgeCreator.java",
"content": "package lsieun.asm.analysis.cfg;\n\nimport lsieun.trove.TIntArrayList;\n\npublic class SimpleEdgeCreator implements EdgeCreator {\n public final TIntArrayList[] transitions;\n public final TIntArrayList[] errorTransitions;\n\n public SimpleEdgeCreator(int size) {\n this.transitions = new TIntArrayList[size];\n this.errorTransitions = new TIntArrayList[size];\n for (int i = 0; i < transitions.length; i++) {\n this.transitions[i] = new TIntArrayList();\n this.errorTransitions[i] = new TIntArrayList();\n }\n }\n\n @Override\n public void newControlFlowEdge(int insn, int successor) {\n if (!transitions[insn].contains(successor)) {\n transitions[insn].add(successor);\n }\n }\n\n @Override\n public void newControlFlowExceptionEdge(int insn, int successor, boolean npe) {\n if (!errorTransitions[insn].contains(successor)) {\n errorTransitions[insn].add(successor);\n }\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/diagnosis/DeadCodeDiagnosis.java",
"content": "package lsieun.asm.analysis.diagnosis;\n\nimport lsieun.asm.analysis.ControlFlowAnalyzer;\nimport lsieun.trove.TIntArrayList;\nimport org.objectweb.asm.tree.InsnList;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\n\npublic class DeadCodeDiagnosis {\n public static int[] diagnose(String className, MethodNode mn) throws AnalyzerException {\n InsnList instructions = mn.instructions;\n int size = instructions.size();\n\n boolean[] flags = new boolean[size];\n ControlFlowAnalyzer analyzer = new ControlFlowAnalyzer() {\n @Override\n protected void newControlFlowEdge(int insnIndex, int successorIndex) {\n // 首先，处理自己的代码逻辑\n flags[insnIndex] = true;\n flags[successorIndex] = true;\n\n // 其次，调用父类的实现\n super.newControlFlowEdge(insnIndex, successorIndex);\n }\n };\n analyzer.analyze(className, mn);\n\n\n TIntArrayList intArrayList = new TIntArrayList();\n for (int i = 0; i < size; i++) {\n boolean flag = flags[i];\n if (!flag) {\n intArrayList.add(i);\n }\n }\n\n return intArrayList.toNativeArray();\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/diagnosis/NullDereferenceDiagnosis.java",
"content": "package lsieun.asm.analysis.diagnosis;\n\nimport lsieun.asm.analysis.nullability.NullDeferenceInterpreter;\nimport lsieun.trove.TIntArrayList;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.InsnList;\nimport org.objectweb.asm.tree.MethodInsnNode;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.*;\n\nimport java.util.Arrays;\n\nimport static org.objectweb.asm.Opcodes.*;\n\npublic class NullDereferenceDiagnosis {\n public static int[] diagnose(String owner, MethodNode mn) throws AnalyzerException {\n // 第一步，获取Frame信息\n Analyzer analyzer = new Analyzer<>(new NullDeferenceInterpreter(ASM9));\n Frame[] frames = analyzer.analyze(owner, mn);\n\n // 第二步，判断是否为null或maybe-null，收集数据\n TIntArrayList intArrayList = new TIntArrayList();\n InsnList instructions = mn.instructions;\n int size = instructions.size();\n for (int i = 0; i < size; i++) {\n AbstractInsnNode insn = instructions.get(i);\n if (frames[i] != null) {\n Value value = getTarget(insn, frames[i]);\n if (value == NullDeferenceInterpreter.NULL_VALUE || value == NullDeferenceInterpreter.MAYBE_NULL_VALUE) {\n intArrayList.add(i);\n }\n }\n }\n\n // 第三步，将结果转换成int[]形式\n int[] array = intArrayList.toNativeArray();\n Arrays.sort(array);\n return array;\n }\n\n private static BasicValue getTarget(AbstractInsnNode insn, Frame frame) {\n int opcode = insn.getOpcode();\n switch (opcode) {\n case GETFIELD:\n case ARRAYLENGTH:\n case MONITORENTER:\n case MONITOREXIT:\n return getStackValue(frame, 0);\n case PUTFIELD:\n return getStackValue(frame, 1);\n case INVOKEVIRTUAL:\n case INVOKESPECIAL:\n case INVOKEINTERFACE:\n String desc = ((MethodInsnNode) insn).desc;\n return getStackValue(frame, Type.getArgumentTypes(desc).length);\n\n }\n return null;\n }\n\n private static BasicValue getStackValue(Frame frame, int index) {\n int top = frame.getStackSize() - 1;\n return index <= top ? frame.getStack(top - index) : null;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/diagnosis/NullabilityDiagnosis.java",
"content": "package lsieun.asm.analysis.diagnosis;\n\nimport lsieun.asm.analysis.nullability.Nullability;\nimport lsieun.asm.analysis.nullability.NullabilityAnalyzer;\nimport lsieun.asm.analysis.nullability.NullabilityInterpreter;\nimport lsieun.asm.analysis.nullability.NullabilityValue;\nimport lsieun.trove.TIntArrayList;\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.InsnList;\nimport org.objectweb.asm.tree.MethodInsnNode;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.analysis.Analyzer;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\nimport org.objectweb.asm.tree.analysis.Frame;\n\nimport java.util.Arrays;\n\nimport static org.objectweb.asm.Opcodes.*;\n\npublic class NullabilityDiagnosis {\n public static int[] diagnose(String className, MethodNode mn) throws AnalyzerException {\n // 第一步，获取Frame信息\n Analyzer analyzer = new NullabilityAnalyzer(new NullabilityInterpreter(Opcodes.ASM9));\n Frame[] frames = analyzer.analyze(className, mn);\n\n // 第二步，判断是否为null或maybe-null，收集数据\n TIntArrayList intArrayList = new TIntArrayList();\n InsnList instructions = mn.instructions;\n int size = instructions.size();\n for (int i = 0; i < size; i++) {\n AbstractInsnNode insn = instructions.get(i);\n if (frames[i] != null) {\n NullabilityValue value = getTarget(insn, frames[i]);\n if (value == null) continue;\n if (value.getState() == Nullability.NULL || value.getState() == Nullability.NULLABLE) {\n intArrayList.add(i);\n }\n }\n }\n\n // 第三步，将结果转换成int[]形式\n int[] array = intArrayList.toNativeArray();\n Arrays.sort(array);\n return array;\n }\n\n private static NullabilityValue getTarget(AbstractInsnNode insn, Frame frame) {\n int opcode = insn.getOpcode();\n switch (opcode) {\n case GETFIELD:\n case ARRAYLENGTH:\n case MONITORENTER:\n case MONITOREXIT:\n return getStackValue(frame, 0);\n case PUTFIELD:\n return getStackValue(frame, 1);\n case INVOKEVIRTUAL:\n case INVOKESPECIAL:\n case INVOKEINTERFACE:\n String desc = ((MethodInsnNode) insn).desc;\n return getStackValue(frame, Type.getArgumentTypes(desc).length);\n\n }\n return null;\n }\n\n private static NullabilityValue getStackValue(Frame frame, int index) {\n int top = frame.getStackSize() - 1;\n return index <= top ? frame.getStack(top - index) : null;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/diagnosis/RedundantVariableDiagnosis.java",
"content": "package lsieun.asm.analysis.diagnosis;\n\nimport lsieun.trove.TIntArrayList;\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.InsnList;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.VarInsnNode;\nimport org.objectweb.asm.tree.analysis.*;\n\nimport java.util.Arrays;\n\npublic class RedundantVariableDiagnosis {\n public static int[] diagnose(String className, MethodNode mn) throws AnalyzerException {\n // 第一步，准备工作。使用SimpleVerifier进行分析，得到frames信息\n Analyzer analyzer = new Analyzer<>(new SimpleVerifier());\n Frame[] frames = analyzer.analyze(className, mn);\n\n // 第二步，利用frames信息，查看local variable当中哪些slot数据出现了冗余\n TIntArrayList localIndexList = new TIntArrayList();\n for (Frame f : frames) {\n int locals = f.getLocals();\n for (int i = 0; i < locals; i++) {\n BasicValue val1 = f.getLocal(i);\n if (val1 == BasicValue.UNINITIALIZED_VALUE) {\n continue;\n }\n for (int j = i + 1; j < locals; j++) {\n BasicValue val2 = f.getLocal(j);\n if (val2 == BasicValue.UNINITIALIZED_VALUE) {\n continue;\n }\n if (val1 == val2) {\n if (!localIndexList.contains(j)) {\n localIndexList.add(j);\n }\n }\n }\n }\n }\n\n // 第三步，将slot的索引值（local index）转换成instruction的索引值（insn index）\n TIntArrayList insnIndexList = new TIntArrayList();\n InsnList instructions = mn.instructions;\n int size = instructions.size();\n for (int i = 0; i < size; i++) {\n AbstractInsnNode node = instructions.get(i);\n int opcode = node.getOpcode();\n if (opcode >= Opcodes.ISTORE && opcode <= Opcodes.ASTORE) {\n VarInsnNode varInsnNode = (VarInsnNode) node;\n if (localIndexList.contains(varInsnNode.var)) {\n if (!insnIndexList.contains(i)) {\n insnIndexList.add(i);\n }\n }\n }\n }\n\n // 第四步，将insnIndexList转换成int[]形式\n int[] array = insnIndexList.toNativeArray();\n Arrays.sort(array);\n return array;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/diagnosis/RelatedInstructionDiagnosis.java",
"content": "package lsieun.asm.analysis.diagnosis;\n\nimport lsieun.asm.analysis.transition.DestinationInterpreter;\nimport lsieun.trove.TIntArrayList;\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.InsnList;\nimport org.objectweb.asm.tree.MethodNode;\nimport org.objectweb.asm.tree.VarInsnNode;\nimport org.objectweb.asm.tree.analysis.*;\n\nimport java.util.ArrayList;\nimport java.util.Arrays;\nimport java.util.List;\n\npublic class RelatedInstructionDiagnosis {\n public static int[] diagnose(String className, MethodNode mn, int insnIndex) throws AnalyzerException {\n // 第一步，判断insnIndex范围是否合理\n InsnList instructions = mn.instructions;\n int size = instructions.size();\n if (insnIndex < 0 || insnIndex >= size) {\n String message = String.format(\"the 'insnIndex' argument should in range [0, %d]\", size - 1);\n throw new IllegalArgumentException(message);\n }\n\n // 第二步，获取两个Frame\n Frame[] sourceFrames = getSourceFrames(className, mn);\n Frame[] destinationFrames = getDestinationFrames(className, mn);\n\n // 第三步，循环处理，所有结果记录到这个intArrayList变量中\n TIntArrayList intArrayList = new TIntArrayList();\n // 循环tmpInsnList\n List tmpInsnList = new ArrayList<>();\n AbstractInsnNode insnNode = instructions.get(insnIndex);\n tmpInsnList.add(insnNode);\n for (int i = 0; i < tmpInsnList.size(); i++) {\n AbstractInsnNode currentNode = tmpInsnList.get(i);\n int opcode = currentNode.getOpcode();\n\n int index = instructions.indexOf(currentNode);\n intArrayList.add(index);\n\n // 第一种情况，处理load相关的opcode情况\n Frame srcFrame = sourceFrames[index];\n if (opcode >= Opcodes.ILOAD && opcode <= Opcodes.ALOAD) {\n VarInsnNode varInsnNode = (VarInsnNode) currentNode;\n int localIndex = varInsnNode.var;\n SourceValue value = srcFrame.getLocal(localIndex);\n for (AbstractInsnNode insn : value.insns) {\n if (!tmpInsnList.contains(insn)) {\n tmpInsnList.add(insn);\n }\n }\n }\n\n // 第二种情况，从dstFrame到srcFrame查找\n Frame dstFrame = destinationFrames[index];\n int stackSize = dstFrame.getStackSize();\n for (int j = 0; j < stackSize; j++) {\n SourceValue value = dstFrame.getStack(j);\n if (value.insns.contains(currentNode)) {\n for (AbstractInsnNode insn : srcFrame.getStack(j).insns) {\n if (!tmpInsnList.contains(insn)) {\n tmpInsnList.add(insn);\n }\n }\n }\n }\n }\n\n // 第四步，将intArrayList变量转换成int[]，并进行排序\n int[] array = intArrayList.toNativeArray();\n Arrays.sort(array);\n return array;\n }\n\n\n private static Frame[] getSourceFrames(String className, MethodNode mn) throws AnalyzerException {\n Analyzer analyzer = new Analyzer<>(new SourceInterpreter());\n return analyzer.analyze(className, mn);\n }\n\n private static Frame[] getDestinationFrames(String className, MethodNode mn) throws AnalyzerException {\n Analyzer analyzer = new Analyzer<>(new DestinationInterpreter());\n return analyzer.analyze(className, mn);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/diagnosis/ReverseEngineerMethodArgumentsDiagnosis.java",
"content": "package lsieun.asm.analysis.diagnosis;\n\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.*;\nimport org.objectweb.asm.tree.analysis.*;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\npublic class ReverseEngineerMethodArgumentsDiagnosis {\n private static final String UNKNOWN_VARIABLE_NAME = \"unknown\";\n\n public static void diagnose(String className, MethodNode mn) throws AnalyzerException {\n // 第一步，获取Frame信息\n Analyzer analyzer = new Analyzer<>(new SourceInterpreter());\n Frame[] frames = analyzer.analyze(className, mn);\n\n // 第二步，获取LocalVariableTable信息\n List localVariables = mn.localVariables;\n if (localVariables == null || localVariables.size() < 1) {\n System.out.println(\"LocalVariableTable is Empty\");\n return;\n }\n\n // 第三步，获取instructions，并找到与invoke相关的指令\n InsnList instructions = mn.instructions;\n int[] methodInsnArray = findMethodInvokes(instructions);\n\n // 第四步，对invoke相关的指令进行反编译\n for (int methodInsn : methodInsnArray) {\n // (1) 获取方法的参数\n MethodInsnNode methodInsnNode = (MethodInsnNode) instructions.get(methodInsn);\n Type methodType = Type.getMethodType(methodInsnNode.desc);\n Type[] argumentTypes = methodType.getArgumentTypes();\n int argNum = argumentTypes.length;\n\n // (2) 从Frame当中获取指令，并将指令转换LocalVariableTable当中的变量名\n Frame f = frames[methodInsn];\n int stackSize = f.getStackSize();\n List argList = new ArrayList<>();\n for (int i = 0; i < argNum; i++) {\n int stackIndex = stackSize - argNum + i;\n SourceValue stackValue = f.getStack(stackIndex);\n AbstractInsnNode insn = stackValue.insns.iterator().next();\n String argName = getMethodVariableName(insn, localVariables);\n argList.add(argName);\n }\n\n // (3) 将反编译的结果打印出来\n String line = String.format(\"%s.%s(%s)\", methodInsnNode.owner, methodInsnNode.name, argList);\n System.out.println(line);\n }\n }\n\n public static String getMethodVariableName(AbstractInsnNode insn, List localVariables) {\n if (insn instanceof VarInsnNode) {\n VarInsnNode varInsnNode = (VarInsnNode) insn;\n int localIndex = varInsnNode.var;\n\n for (LocalVariableNode node : localVariables) {\n if (node.index == localIndex) {\n return node.name;\n }\n }\n\n return String.format(\"locals[%d]\", localIndex);\n }\n return UNKNOWN_VARIABLE_NAME;\n }\n\n public static int[] findMethodInvokes(InsnList instructions) {\n int size = instructions.size();\n boolean[] methodArray = new boolean[size];\n for (int i = 0; i < size; i++) {\n AbstractInsnNode node = instructions.get(i);\n if (node instanceof MethodInsnNode) {\n methodArray[i] = true;\n }\n }\n\n int count = 0;\n for (boolean flag : methodArray) {\n if (flag) {\n count++;\n }\n }\n\n int[] array = new int[count];\n int j = 0;\n for (int i = 0; i < size; i++) {\n boolean flag = methodArray[i];\n if (flag) {\n array[j] = i;\n j++;\n }\n }\n return array;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/graph/InsnBlock.java",
"content": "package lsieun.asm.analysis.graph;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\npublic class InsnBlock {\n // 文字信息\n public final List lines = new ArrayList<>();\n\n // 关联关系\n public final List nextBlockList = new ArrayList<>();\n public final List jumpBlockList = new ArrayList<>();\n\n public void addLines(List list) {\n lines.addAll(list);\n }\n\n public void addNext(InsnBlock item) {\n if (!nextBlockList.contains(item)) {\n nextBlockList.add(item);\n }\n }\n\n public void addJump(InsnBlock item) {\n if (!jumpBlockList.contains(item)) {\n jumpBlockList.add(item);\n }\n }\n\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/graph/TextBox.java",
"content": "package lsieun.asm.analysis.graph;\n\npublic class TextBox {\n public final int row;\n public final int col;\n public final int width;\n public final int height;\n\n public TextBox(int row, int col, int width, int height) {\n this.row = row;\n this.col = col;\n this.width = width;\n this.height = height;\n }\n\n public static TextBox valueOf(int row, int col, int width, int height) {\n return new TextBox(row, col, width, height);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/graph/TextGraph.java",
"content": "package lsieun.asm.analysis.graph;\n\nimport lsieun.drawing.canvas.Box;\nimport lsieun.drawing.canvas.Canvas;\nimport lsieun.drawing.canvas.TextDirection;\nimport lsieun.drawing.theme.line.ContinuousLine;\nimport lsieun.drawing.theme.text.PlainText;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\npublic class TextGraph {\n private static final int ROW_SPACE = 1;\n private static final int COL_SPACE = 3;\n\n private final InsnBlock[] blockArray;\n private final TextBox[] boxArray;\n private final int boxNum;\n private final int maxInstructionLength;\n private final Canvas canvas = new Canvas();\n\n public TextGraph(InsnBlock[] blockArray) {\n this.blockArray = blockArray;\n this.boxNum = blockArray.length;\n this.boxArray = new TextBox[boxNum];\n this.maxInstructionLength = findMaxStringLength(blockArray);\n }\n\n @SuppressWarnings(\"UnnecessaryLocalVariable\")\n public void draw(int startRow, int startCol) {\n int row = startRow;\n int col = startCol;\n int length = boxArray.length;\n for (int i = 0; i < length; i++) {\n InsnBlock block = blockArray[i];\n\n int width = getOdd(maxInstructionLength + 2);\n int height = block.lines.size();\n\n TextBox box = TextBox.valueOf(row, col, width, height);\n boxArray[i] = box;\n drawBoxAndText(box, block.lines);\n\n row += height + ROW_SPACE + 2;\n }\n\n drawLinks();\n\n System.out.println(canvas);\n }\n\n private void drawBoxAndText(TextBox box, List lines) {\n int row = box.row;\n int col = box.col;\n int width = box.width;\n int height = box.height;\n\n canvas.moveTo(row, col);\n canvas.drawRectangle(width, height);\n canvas.draw(row + 1, col + 2, PlainText.valueOf(lines));\n }\n\n private void drawLinks() {\n for (int i = 0; i < boxNum; i++) {\n InsnBlock currentBlock = blockArray[i];\n TextBox currentBox = boxArray[i];\n\n // draw next\n List nextBoxes = findBoxes(currentBlock.nextBlockList);\n for (TextBox nextBox : nextBoxes) {\n int rowStart = currentBox.row + currentBox.height + 1;\n int rowStop = nextBox.row;\n int col = currentBox.col + currentBox.width / 2;\n\n canvas.moveTo(rowStart, col);\n canvas.drawPixel(Box.DOWN_AND_HORIZONTAL);\n canvas.moveTo(rowStop, col);\n canvas.drawPixel(Box.UP_AND_HORIZONTAL);\n canvas.moveTo(rowStart + 1, col).drawVerticalLine(rowStop - rowStart - 1);\n }\n\n // draw jump\n List jumpBoxes = findBoxes(currentBlock.jumpBlockList);\n for (TextBox nextBox : jumpBoxes) {\n int rowStart = currentBox.row + currentBox.height;\n int rowStop = nextBox.row + 1;\n int colStart = currentBox.col + currentBox.width + 1;\n int colStop = currentBox.col + currentBox.width + 1 + (i + 1) * COL_SPACE;\n\n canvas.moveTo(rowStart, colStart);\n canvas.drawPixel(Box.VERTICAL_AND_RIGHT);\n canvas.moveTo(rowStop, colStart);\n canvas.drawPixel(Box.VERTICAL_AND_RIGHT);\n\n if (rowStart < rowStop) {\n ContinuousLine line = new ContinuousLine();\n line.setDirection(TextDirection.RIGHT);\n line.drawLine(colStop - colStart)\n .turn(TextDirection.DOWN).drawLine(rowStop - rowStart - 1)\n .turn(TextDirection.LEFT).drawLine(colStop - colStart);\n canvas.draw(rowStart, colStart + 1, line);\n\n }\n else {\n ContinuousLine line = new ContinuousLine();\n line.setDirection(TextDirection.RIGHT);\n line.drawLine(colStop - colStart)\n .turn(TextDirection.UP).drawLine(rowStart - rowStop - 1)\n .turn(TextDirection.LEFT).drawLine(colStop - colStart);\n canvas.draw(rowStart, colStart + 1, line);\n }\n }\n }\n }\n\n private List findBoxes(List blockList) {\n List boxList = new ArrayList<>();\n\n for (int i = 0; i < boxNum; i++) {\n InsnBlock block = blockArray[i];\n if (blockList.contains(block)) {\n boxList.add(boxArray[i]);\n }\n }\n\n return boxList;\n }\n\n private int getOdd(int num) {\n int remainder = num % 2;\n if (remainder == 0) {\n return num + 1;\n }\n return num;\n }\n\n private int findMaxStringLength(InsnBlock[] blockArray) {\n int maxLength = 0;\n for (InsnBlock block : blockArray) {\n int length = findMaxStringLength(block.lines);\n if (length > maxLength) {\n maxLength = length;\n }\n }\n return maxLength;\n }\n\n private int findMaxStringLength(List lines) {\n int maxLength = 0;\n for (String item : lines) {\n if (item == null) continue;\n int length = item.length();\n if (length > maxLength) {\n maxLength = length;\n }\n }\n return maxLength;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/nullability/NullDeferenceInterpreter.java",
"content": "package lsieun.asm.analysis.nullability;\n\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\nimport org.objectweb.asm.tree.analysis.BasicInterpreter;\nimport org.objectweb.asm.tree.analysis.BasicValue;\nimport org.objectweb.asm.tree.analysis.Value;\n\npublic class NullDeferenceInterpreter extends BasicInterpreter {\n public final static BasicValue NULL_VALUE = new BasicValue(NULL_TYPE);\n public final static BasicValue MAYBE_NULL_VALUE = new BasicValue(Type.getObjectType(\"may-be-null\"));\n\n public NullDeferenceInterpreter(int api) {\n super(api);\n }\n\n @Override\n public BasicValue newOperation(AbstractInsnNode insn) throws AnalyzerException {\n if (insn.getOpcode() == ACONST_NULL) {\n return NULL_VALUE;\n }\n return super.newOperation(insn);\n }\n\n @Override\n public BasicValue merge(BasicValue value1, BasicValue value2) {\n if (isRef(value1) && isRef(value2) && value1 != value2) {\n return MAYBE_NULL_VALUE;\n }\n return super.merge(value1, value2);\n }\n\n private boolean isRef(Value value) {\n return value == BasicValue.REFERENCE_VALUE || value == NULL_VALUE || value == MAYBE_NULL_VALUE;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/nullability/Nullability.java",
"content": "package lsieun.asm.analysis.nullability;\n\npublic enum Nullability {\n UNKNOWN(0),\n NOT_NULL(1),\n NULL(1),\n NULLABLE(2);\n\n public final int priority;\n\n Nullability(int priority) {\n this.priority = priority;\n }\n\n public static Nullability merge(Nullability value1, Nullability value2) {\n // 第一种情况，两者相等，则直接返回一个\n if (value1 == value2) {\n return value1;\n }\n\n // 第二种情况，两者不相等，比较优先级大小，谁大返回谁\n int priority1 = value1.priority;\n int priority2 = value2.priority;\n if (priority1 > priority2) {\n return value1;\n }\n else if (priority1 < priority2) {\n return value2;\n }\n\n // 第三种情况，两者不相等，但优先级相等，则一个是NOT_NULL，另一个是NULL\n return NULLABLE;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/nullability/NullabilityAnalyzer.java",
"content": "package lsieun.asm.analysis.nullability;\n\nimport org.objectweb.asm.tree.analysis.Analyzer;\nimport org.objectweb.asm.tree.analysis.Frame;\nimport org.objectweb.asm.tree.analysis.Interpreter;\n\npublic class NullabilityAnalyzer extends Analyzer {\n public NullabilityAnalyzer(Interpreter interpreter) {\n super(interpreter);\n }\n\n @Override\n protected Frame newFrame(Frame frame) {\n return new NullabilityFrame((NullabilityFrame) frame);\n }\n\n @Override\n protected Frame newFrame(int numLocals, int numStack) {\n return new NullabilityFrame(numLocals, numStack);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/nullability/NullabilityFrame.java",
"content": "package lsieun.asm.analysis.nullability;\n\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.tree.LabelNode;\nimport org.objectweb.asm.tree.analysis.Frame;\n\npublic class NullabilityFrame extends Frame {\n public NullabilityFrame(int numLocals, int numStack) {\n super(numLocals, numStack);\n }\n\n public NullabilityFrame(NullabilityFrame frame) {\n super(frame);\n }\n\n @Override\n public void initJumpTarget(int opcode, LabelNode target) {\n // 首先，处理自己的代码逻辑\n int stackIndex = getStackSize();\n NullabilityValue oldValue = getStack(stackIndex);\n switch (opcode) {\n case Opcodes.IFNULL: {\n if (target == null) {\n updateFrame(oldValue, Nullability.NOT_NULL);\n }\n else {\n updateFrame(oldValue, Nullability.NULL);\n }\n break;\n }\n case Opcodes.IFNONNULL: {\n if (target == null) {\n updateFrame(oldValue, Nullability.NULL);\n }\n else {\n updateFrame(oldValue, Nullability.NOT_NULL);\n }\n break;\n }\n }\n\n // 其次，调用父类的方法实现\n super.initJumpTarget(opcode, target);\n }\n\n private void updateFrame(NullabilityValue oldValue, Nullability newState) {\n NullabilityValue newValue = new NullabilityValue(oldValue.getType(), newState);\n int numLocals = getLocals();\n for (int i = 0; i < numLocals; i++) {\n NullabilityValue currentValue = getLocal(i);\n if (oldValue == currentValue) {\n setLocal(i, newValue);\n }\n }\n\n int numStack = getMaxStackSize();\n for (int i = 0; i < numStack; i++) {\n NullabilityValue currentValue = getStack(i);\n if (oldValue == currentValue) {\n setStack(i, newValue);\n }\n }\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/nullability/NullabilityInterpreter.java",
"content": "package lsieun.asm.analysis.nullability;\n\nimport org.objectweb.asm.ConstantDynamic;\nimport org.objectweb.asm.Handle;\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.*;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\nimport org.objectweb.asm.tree.analysis.Interpreter;\n\nimport java.util.List;\n\npublic class NullabilityInterpreter extends Interpreter implements Opcodes {\n public static final Type NULL_TYPE = Type.getObjectType(\"null\");\n\n public static final NullabilityValue UNINITIALIZED_VALUE = new NullabilityValue(null);\n public static final NullabilityValue RETURN_ADDRESS_VALUE = new NullabilityValue(Type.VOID_TYPE);\n\n private final ClassLoader loader = getClass().getClassLoader();\n\n public NullabilityInterpreter(int api) {\n super(api);\n }\n\n @Override\n public NullabilityValue newValue(Type type) {\n if (type == null) {\n return UNINITIALIZED_VALUE;\n }\n\n int sort = type.getSort();\n if (sort == Type.VOID) {\n return null;\n }\n return new NullabilityValue(type);\n }\n\n @Override\n public NullabilityValue newOperation(AbstractInsnNode insn) throws AnalyzerException {\n switch (insn.getOpcode()) {\n case ACONST_NULL:\n return new NullabilityValue(NULL_TYPE, Nullability.NULL);\n case ICONST_M1:\n case ICONST_0:\n case ICONST_1:\n case ICONST_2:\n case ICONST_3:\n case ICONST_4:\n case ICONST_5:\n case BIPUSH:\n case SIPUSH:\n return newValue(Type.INT_TYPE);\n case LCONST_0:\n case LCONST_1:\n return newValue(Type.LONG_TYPE);\n case FCONST_0:\n case FCONST_1:\n case FCONST_2:\n return newValue(Type.FLOAT_TYPE);\n case DCONST_0:\n case DCONST_1:\n return newValue(Type.DOUBLE_TYPE);\n case LDC:\n Object value = ((LdcInsnNode) insn).cst;\n if (value instanceof Integer) {\n return newValue(Type.INT_TYPE);\n }\n else if (value instanceof Float) {\n return newValue(Type.FLOAT_TYPE);\n }\n else if (value instanceof Long) {\n return newValue(Type.LONG_TYPE);\n }\n else if (value instanceof Double) {\n return newValue(Type.DOUBLE_TYPE);\n }\n else if (value instanceof String) {\n return new NullabilityValue(Type.getObjectType(\"java/lang/String\"), Nullability.NOT_NULL);\n }\n else if (value instanceof Type) {\n int sort = ((Type) value).getSort();\n if (sort == Type.OBJECT || sort == Type.ARRAY) {\n return newValue(Type.getObjectType(\"java/lang/Class\"));\n }\n else if (sort == Type.METHOD) {\n return newValue(Type.getObjectType(\"java/lang/invoke/MethodType\"));\n }\n else {\n throw new AnalyzerException(insn, \"Illegal LDC value \" + value);\n }\n }\n else if (value instanceof Handle) {\n return newValue(Type.getObjectType(\"java/lang/invoke/MethodHandle\"));\n }\n else if (value instanceof ConstantDynamic) {\n return newValue(Type.getType(((ConstantDynamic) value).getDescriptor()));\n }\n else {\n throw new AnalyzerException(insn, \"Illegal LDC value \" + value);\n }\n case JSR:\n return RETURN_ADDRESS_VALUE;\n case GETSTATIC:\n return newValue(Type.getType(((FieldInsnNode) insn).desc));\n case NEW:\n return newValue(Type.getObjectType(((TypeInsnNode) insn).desc));\n default:\n throw new AssertionError();\n }\n }\n\n @Override\n public NullabilityValue copyOperation(AbstractInsnNode insn, NullabilityValue value) {\n return value;\n }\n\n @Override\n public NullabilityValue unaryOperation(AbstractInsnNode insn, NullabilityValue value) throws AnalyzerException {\n switch (insn.getOpcode()) {\n case INEG:\n case IINC:\n case L2I:\n case F2I:\n case D2I:\n case I2B:\n case I2C:\n case I2S:\n case ARRAYLENGTH:\n case INSTANCEOF:\n return newValue(Type.INT_TYPE);\n case FNEG:\n case I2F:\n case L2F:\n case D2F:\n return newValue(Type.FLOAT_TYPE);\n case LNEG:\n case I2L:\n case F2L:\n case D2L:\n return newValue(Type.LONG_TYPE);\n case DNEG:\n case I2D:\n case L2D:\n case F2D:\n return newValue(Type.DOUBLE_TYPE);\n case GETFIELD:\n return newValue(Type.getType(((FieldInsnNode) insn).desc));\n case NEWARRAY:\n switch (((IntInsnNode) insn).operand) {\n case T_BOOLEAN:\n return newValue(Type.getType(\"[Z\"));\n case T_CHAR:\n return newValue(Type.getType(\"[C\"));\n case T_BYTE:\n return newValue(Type.getType(\"[B\"));\n case T_SHORT:\n return newValue(Type.getType(\"[S\"));\n case T_INT:\n return newValue(Type.getType(\"[I\"));\n case T_FLOAT:\n return newValue(Type.getType(\"[F\"));\n case T_DOUBLE:\n return newValue(Type.getType(\"[D\"));\n case T_LONG:\n return newValue(Type.getType(\"[J\"));\n default:\n break;\n }\n throw new AnalyzerException(insn, \"Invalid array type\");\n case ANEWARRAY:\n return newValue(Type.getType(\"[\" + Type.getObjectType(((TypeInsnNode) insn).desc)));\n case CHECKCAST:\n return value;\n case IFEQ:\n case IFNE:\n case IFLT:\n case IFGE:\n case IFGT:\n case IFLE:\n case TABLESWITCH:\n case LOOKUPSWITCH:\n case IRETURN:\n case LRETURN:\n case FRETURN:\n case DRETURN:\n case ARETURN:\n case PUTSTATIC:\n case ATHROW:\n case MONITORENTER:\n case MONITOREXIT:\n case IFNULL:\n case IFNONNULL:\n return null;\n default:\n throw new AssertionError();\n }\n }\n\n @Override\n public NullabilityValue binaryOperation(AbstractInsnNode insn,\n NullabilityValue value1,\n NullabilityValue value2) {\n switch (insn.getOpcode()) {\n case IALOAD:\n case BALOAD:\n case CALOAD:\n case SALOAD:\n case IADD:\n case ISUB:\n case IMUL:\n case IDIV:\n case IREM:\n case ISHL:\n case ISHR:\n case IUSHR:\n case IAND:\n case IOR:\n case IXOR:\n case LCMP:\n case FCMPL:\n case FCMPG:\n case DCMPL:\n case DCMPG:\n return newValue(Type.INT_TYPE);\n case FALOAD:\n case FADD:\n case FSUB:\n case FMUL:\n case FDIV:\n case FREM:\n return newValue(Type.FLOAT_TYPE);\n case LALOAD:\n case LADD:\n case LSUB:\n case LMUL:\n case LDIV:\n case LREM:\n case LSHL:\n case LSHR:\n case LUSHR:\n case LAND:\n case LOR:\n case LXOR:\n return newValue(Type.LONG_TYPE);\n case DALOAD:\n case DADD:\n case DSUB:\n case DMUL:\n case DDIV:\n case DREM:\n return newValue(Type.LONG_TYPE);\n case AALOAD:\n return getElementValue(value1);\n case IF_ICMPEQ:\n case IF_ICMPNE:\n case IF_ICMPLT:\n case IF_ICMPGE:\n case IF_ICMPGT:\n case IF_ICMPLE:\n case IF_ACMPEQ:\n case IF_ACMPNE:\n case PUTFIELD:\n return null;\n default:\n throw new AssertionError();\n }\n }\n\n @Override\n public NullabilityValue ternaryOperation(AbstractInsnNode insn,\n NullabilityValue value1,\n NullabilityValue value2,\n NullabilityValue value3) {\n return null;\n }\n\n @Override\n public NullabilityValue naryOperation(AbstractInsnNode insn,\n List values) {\n int opcode = insn.getOpcode();\n if (opcode == MULTIANEWARRAY) {\n return newValue(Type.getType(((MultiANewArrayInsnNode) insn).desc));\n }\n else if (opcode == INVOKEDYNAMIC) {\n return newValue(Type.getReturnType(((InvokeDynamicInsnNode) insn).desc));\n }\n else {\n return newValue(Type.getReturnType(((MethodInsnNode) insn).desc));\n }\n }\n\n @Override\n public void returnOperation(AbstractInsnNode insn,\n NullabilityValue value,\n NullabilityValue expected) {\n // Nothing to do.\n }\n\n @Override\n public NullabilityValue merge(NullabilityValue value1, NullabilityValue value2) {\n // 合并两者的状态\n Nullability mergedState = Nullability.merge(value1.getState(), value2.getState());\n\n\n // 第一种情况，两个value的类型相同且状态（state）相同\n if (value1.equals(value2)) {\n return value1;\n }\n\n // 第二种情况，两个value的类型相同，但状态（state）不同，需要合并它们的状态（state）\n Type type1 = value1.getType();\n Type type2 = value2.getType();\n if (type1 != null && type1.equals(type2)) {\n Type type = value1.getType();\n return new NullabilityValue(type, mergedState);\n }\n\n // 第三种情况，两个value的类型不相同的，而且要合并它们的状态（state）\n if (type1 != null\n && (type1.getSort() == Type.OBJECT || type1.getSort() == Type.ARRAY)\n && type2 != null\n && (type2.getSort() == Type.OBJECT || type2.getSort() == Type.ARRAY)) {\n if (type1.equals(NULL_TYPE)) {\n return new NullabilityValue(type2, mergedState);\n }\n if (type2.equals(NULL_TYPE)) {\n return new NullabilityValue(type1, mergedState);\n }\n if (isAssignableFrom(type1, type2)) {\n return new NullabilityValue(type1, mergedState);\n }\n if (isAssignableFrom(type2, type1)) {\n return new NullabilityValue(type2, mergedState);\n }\n int numDimensions = 0;\n if (type1.getSort() == Type.ARRAY\n && type2.getSort() == Type.ARRAY\n && type1.getDimensions() == type2.getDimensions()\n && type1.getElementType().getSort() == Type.OBJECT\n && type2.getElementType().getSort() == Type.OBJECT) {\n numDimensions = type1.getDimensions();\n type1 = type1.getElementType();\n type2 = type2.getElementType();\n }\n\n\n while (true) {\n if (type1 == null || isInterface(type1)) {\n NullabilityValue arrayValue = newArrayValue(Type.getObjectType(\"java/lang/Object\"), numDimensions);\n return new NullabilityValue(arrayValue.getType(), mergedState);\n }\n type1 = getSuperClass(type1);\n if (isAssignableFrom(type1, type2)) {\n NullabilityValue arrayValue = newArrayValue(type1, numDimensions);\n return new NullabilityValue(arrayValue.getType(), mergedState);\n }\n }\n }\n return UNINITIALIZED_VALUE;\n }\n\n protected boolean isInterface(final Type type) {\n return getClass(type).isInterface();\n }\n\n protected Type getSuperClass(final Type type) {\n Class superClass = getClass(type).getSuperclass();\n return superClass == null ? null : Type.getType(superClass);\n }\n\n private NullabilityValue newArrayValue(final Type type, final int dimensions) {\n if (dimensions == 0) {\n return newValue(type);\n }\n else {\n StringBuilder descriptor = new StringBuilder();\n for (int i = 0; i < dimensions; ++i) {\n descriptor.append('[');\n }\n descriptor.append(type.getDescriptor());\n return newValue(Type.getType(descriptor.toString()));\n }\n }\n\n protected NullabilityValue getElementValue(final NullabilityValue objectArrayValue) {\n Type arrayType = objectArrayValue.getType();\n if (arrayType != null) {\n if (arrayType.getSort() == Type.ARRAY) {\n return newValue(Type.getType(arrayType.getDescriptor().substring(1)));\n }\n else if (arrayType.equals(NULL_TYPE)) {\n return objectArrayValue;\n }\n }\n throw new AssertionError();\n }\n\n protected boolean isSubTypeOf(final NullabilityValue value, final NullabilityValue expected) {\n Type expectedType = expected.getType();\n Type type = value.getType();\n switch (expectedType.getSort()) {\n case Type.INT:\n case Type.FLOAT:\n case Type.LONG:\n case Type.DOUBLE:\n return type.equals(expectedType);\n case Type.ARRAY:\n case Type.OBJECT:\n if (type.equals(NULL_TYPE)) {\n return true;\n }\n else if (type.getSort() == Type.OBJECT || type.getSort() == Type.ARRAY) {\n if (isAssignableFrom(expectedType, type)) {\n return true;\n }\n else if (getClass(expectedType).isInterface()) {\n // The merge of class or interface types can only yield class types (because it is not\n // possible in general to find an unambiguous common super interface, due to multiple\n // inheritance). Because of this limitation, we need to relax the subtyping check here\n // if 'value' is an interface.\n return Object.class.isAssignableFrom(getClass(type));\n }\n else {\n return false;\n }\n }\n else {\n return false;\n }\n default:\n throw new AssertionError();\n }\n }\n\n protected boolean isAssignableFrom(final Type type1, final Type type2) {\n if (type1.equals(type2)) {\n return true;\n }\n return getClass(type1).isAssignableFrom(getClass(type2));\n }\n\n protected Class getClass(final Type type) {\n try {\n if (type.getSort() == Type.ARRAY) {\n return Class.forName(type.getDescriptor().replace('/', '.'), false, loader);\n }\n return Class.forName(type.getClassName(), false, loader);\n }\n catch (ClassNotFoundException e) {\n throw new TypeNotPresentException(e.toString(), e);\n }\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/nullability/NullabilityValue.java",
"content": "package lsieun.asm.analysis.nullability;\n\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.analysis.Value;\n\npublic class NullabilityValue implements Value {\n private final Type type;\n private Nullability state;\n\n public NullabilityValue(Type type) {\n this(type, Nullability.UNKNOWN);\n }\n\n public NullabilityValue(Type type, Nullability state) {\n this.type = type;\n this.state = state;\n }\n\n public Type getType() {\n return type;\n }\n\n public void setState(Nullability state) {\n this.state = state;\n }\n\n public Nullability getState() {\n return state;\n }\n\n @Override\n public int getSize() {\n return type == Type.LONG_TYPE || type == Type.DOUBLE_TYPE ? 2 : 1;\n }\n\n public boolean isReference() {\n return type != null && (type.getSort() == Type.OBJECT || type.getSort() == Type.ARRAY);\n }\n\n @Override\n public boolean equals(final Object value) {\n if (value == this) {\n return true;\n }\n else if (value instanceof NullabilityValue) {\n NullabilityValue another = (NullabilityValue) value;\n if (type == null) {\n return ((NullabilityValue) value).type == null;\n }\n else {\n return type.equals(((NullabilityValue) value).type) && state == another.state;\n }\n }\n else {\n return false;\n }\n }\n\n @Override\n public int hashCode() {\n return type == null ? 0 : type.hashCode();\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/state/StateInterpreter.java",
"content": "package lsieun.asm.analysis.state;\n\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.AbstractInsnNode;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\nimport org.objectweb.asm.tree.analysis.Interpreter;\n\nimport java.util.List;\n\npublic class StateInterpreter extends Interpreter implements Opcodes {\n public StateInterpreter() {\n super(ASM9);\n if (getClass() != StateInterpreter.class) {\n throw new IllegalStateException();\n }\n }\n\n public StateInterpreter(int api) {\n super(api);\n }\n\n @Override\n public StateValue newValue(Type type) {\n return null;\n }\n\n @Override\n public StateValue newOperation(AbstractInsnNode insn) throws AnalyzerException {\n return null;\n }\n\n @Override\n public StateValue copyOperation(AbstractInsnNode insn, StateValue value) throws AnalyzerException {\n return null;\n }\n\n @Override\n public StateValue unaryOperation(AbstractInsnNode insn, StateValue value) throws AnalyzerException {\n return null;\n }\n\n @Override\n public StateValue binaryOperation(AbstractInsnNode insn, StateValue value1, StateValue value2) throws AnalyzerException {\n return null;\n }\n\n @Override\n public StateValue ternaryOperation(AbstractInsnNode insn, StateValue value1, StateValue value2, StateValue value3) throws AnalyzerException {\n return null;\n }\n\n @Override\n public StateValue naryOperation(AbstractInsnNode insn, List values) throws AnalyzerException {\n return null;\n }\n\n @Override\n public void returnOperation(AbstractInsnNode insn, StateValue value, StateValue expected) throws AnalyzerException {\n\n }\n\n @Override\n public StateValue merge(StateValue value1, StateValue value2) {\n return null;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/state/StateType.java",
"content": "package lsieun.asm.analysis.state;\n\npublic enum StateType {\n DEFAULT, // default\n TOP,\n INT,\n FLOAT,\n LONG,\n DOUBLE,\n NULL,\n\n CP, // constant pool\n HEAP; // 从堆内存上分配的对象\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/state/StateValue.java",
"content": "package lsieun.asm.analysis.state;\n\nimport org.objectweb.asm.tree.analysis.Value;\n\npublic class StateValue implements Value {\n public final int size;\n\n public StateValue(int size) {\n this.size = size;\n }\n\n @Override\n public int getSize() {\n return size;\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/analysis/transition/DestinationInterpreter.java",
"content": "package lsieun.asm.analysis.transition;\n\nimport org.objectweb.asm.Opcodes;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.tree.*;\nimport org.objectweb.asm.tree.analysis.AnalyzerException;\nimport org.objectweb.asm.tree.analysis.Interpreter;\nimport org.objectweb.asm.tree.analysis.SourceValue;\n\nimport java.util.HashSet;\nimport java.util.List;\n\npublic class DestinationInterpreter extends Interpreter implements Opcodes {\n public DestinationInterpreter() {\n super(ASM9);\n if (getClass() != DestinationInterpreter.class) {\n throw new IllegalStateException();\n }\n }\n\n protected DestinationInterpreter(final int api) {\n super(api);\n }\n\n @Override\n public SourceValue newValue(Type type) {\n if (type == Type.VOID_TYPE) {\n return null;\n }\n return new SourceValue(type == null ? 1 : type.getSize(), new HashSet<>());\n }\n\n @Override\n public SourceValue newOperation(AbstractInsnNode insn) {\n int size;\n switch (insn.getOpcode()) {\n case LCONST_0:\n case LCONST_1:\n case DCONST_0:\n case DCONST_1:\n size = 2;\n break;\n case LDC:\n Object value = ((LdcInsnNode) insn).cst;\n size = value instanceof Long || value instanceof Double ? 2 : 1;\n break;\n case GETSTATIC:\n size = Type.getType(((FieldInsnNode) insn).desc).getSize();\n break;\n default:\n size = 1;\n break;\n }\n return new SourceValue(size, new HashSet<>());\n }\n\n @Override\n public SourceValue copyOperation(AbstractInsnNode insn, SourceValue value) throws AnalyzerException {\n int opcode = insn.getOpcode();\n if (opcode >= ISTORE && opcode <= ASTORE) {\n value.insns.add(insn);\n }\n\n return new SourceValue(value.getSize(), new HashSet<>());\n }\n\n @Override\n public SourceValue unaryOperation(AbstractInsnNode insn, SourceValue value) throws AnalyzerException {\n value.insns.add(insn);\n\n int size;\n switch (insn.getOpcode()) {\n case LNEG:\n case DNEG:\n case I2L:\n case I2D:\n case L2D:\n case F2L:\n case F2D:\n case D2L:\n size = 2;\n break;\n case GETFIELD:\n size = Type.getType(((FieldInsnNode) insn).desc).getSize();\n break;\n default:\n size = 1;\n break;\n }\n return new SourceValue(size, new HashSet<>());\n }\n\n @Override\n public SourceValue binaryOperation(AbstractInsnNode insn, SourceValue value1, SourceValue value2) throws AnalyzerException {\n value1.insns.add(insn);\n value2.insns.add(insn);\n\n int size;\n switch (insn.getOpcode()) {\n case LALOAD:\n case DALOAD:\n case LADD:\n case DADD:\n case LSUB:\n case DSUB:\n case LMUL:\n case DMUL:\n case LDIV:\n case DDIV:\n case LREM:\n case DREM:\n case LSHL:\n case LSHR:\n case LUSHR:\n case LAND:\n case LOR:\n case LXOR:\n size = 2;\n break;\n default:\n size = 1;\n break;\n }\n return new SourceValue(size, new HashSet<>());\n }\n\n @Override\n public SourceValue ternaryOperation(AbstractInsnNode insn, SourceValue value1, SourceValue value2, SourceValue value3) throws AnalyzerException {\n value1.insns.add(insn);\n value2.insns.add(insn);\n value3.insns.add(insn);\n\n return new SourceValue(1, new HashSet<>());\n }\n\n @Override\n public SourceValue naryOperation(AbstractInsnNode insn, List values) throws AnalyzerException {\n if (values != null) {\n for (SourceValue v : values) {\n v.insns.add(insn);\n }\n }\n\n int size;\n int opcode = insn.getOpcode();\n if (opcode == MULTIANEWARRAY) {\n size = 1;\n }\n else if (opcode == INVOKEDYNAMIC) {\n size = Type.getReturnType(((InvokeDynamicInsnNode) insn).desc).getSize();\n }\n else {\n size = Type.getReturnType(((MethodInsnNode) insn).desc).getSize();\n }\n return new SourceValue(size, new HashSet<>());\n }\n\n @Override\n public void returnOperation(AbstractInsnNode insn, SourceValue value, SourceValue expected) throws AnalyzerException {\n // Nothing to do.\n }\n\n @Override\n public SourceValue merge(final SourceValue value1, final SourceValue value2) {\n return new SourceValue(Math.min(value1.size, value2.size), new HashSet<>());\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/commons/ClassRemapperExample01.java",
"content": "package lsieun.asm.commons;\n\nimport lsieun.utils.FileUtils;\nimport org.objectweb.asm.ClassReader;\nimport org.objectweb.asm.ClassVisitor;\nimport org.objectweb.asm.ClassWriter;\nimport org.objectweb.asm.commons.ClassRemapper;\nimport org.objectweb.asm.commons.Remapper;\nimport org.objectweb.asm.commons.SimpleRemapper;\n\npublic class ClassRemapperExample01 {\n public static void main(String[] args) {\n String origin_name = \"sample/HelloWorld\";\n String target_name = \"sample/GoodChild\";\n String origin_filepath = getFilePath(origin_name);\n byte[] bytes1 = FileUtils.readBytes(origin_filepath);\n\n //（1）构建ClassReader\n ClassReader cr = new ClassReader(bytes1);\n\n //（2）构建ClassWriter\n ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);\n\n //（3）串连ClassVisitor\n Remapper remapper = new SimpleRemapper(origin_name, target_name);\n ClassVisitor cv = new ClassRemapper(cw, remapper);\n\n //（4）两者进行结合\n int parsingOptions = ClassReader.SKIP_DEBUG | ClassReader.SKIP_FRAMES;\n cr.accept(cv, parsingOptions);\n\n //（5）重新生成Class\n byte[] bytes2 = cw.toByteArray();\n\n String target_filepath = getFilePath(target_name);\n FileUtils.writeBytes(target_filepath, bytes2);\n }\n\n public static String getFilePath(String internalName) {\n String relative_path = String.format(\"%s.class\", internalName);\n return FileUtils.getFilePath(relative_path);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/commons/ClassRemapperExample02.java",
"content": "package lsieun.asm.commons;\n\nimport lsieun.utils.FileUtils;\nimport org.objectweb.asm.ClassReader;\nimport org.objectweb.asm.ClassVisitor;\nimport org.objectweb.asm.ClassWriter;\nimport org.objectweb.asm.commons.ClassRemapper;\nimport org.objectweb.asm.commons.Remapper;\nimport org.objectweb.asm.commons.SimpleRemapper;\n\nimport java.util.HashMap;\nimport java.util.Map;\n\npublic class ClassRemapperExample02 {\n public static void main(String[] args) {\n String origin_name = \"sample/HelloWorld\";\n String target_name = \"sample/GoodChild\";\n String origin_filepath = getFilePath(origin_name);\n byte[] bytes1 = FileUtils.readBytes(origin_filepath);\n\n //（1）构建ClassReader\n ClassReader cr = new ClassReader(bytes1);\n\n //（2）构建ClassWriter\n ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);\n\n //（3）串连ClassVisitor\n Map mapping = new HashMap<>();\n mapping.put(origin_name, target_name);\n mapping.put(origin_name + \".intValue\", \"a\");\n mapping.put(origin_name + \".test()V\", \"b\");\n Remapper mapper = new SimpleRemapper(mapping);\n ClassVisitor cv = new ClassRemapper(cw, mapper);\n\n //（4）两者进行结合\n int parsingOptions = ClassReader.SKIP_DEBUG | ClassReader.SKIP_FRAMES;\n cr.accept(cv, parsingOptions);\n\n //（5）重新生成Class\n byte[] bytes2 = cw.toByteArray();\n\n String target_filepath = getFilePath(target_name);\n FileUtils.writeBytes(target_filepath, bytes2);\n }\n\n public static String getFilePath(String internalName) {\n String relative_path = String.format(\"%s.class\", internalName);\n return FileUtils.getFilePath(relative_path);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/commons/ClassRemapperExample03.java",
"content": "package lsieun.asm.commons;\n\nimport lsieun.utils.FileUtils;\nimport org.objectweb.asm.ClassReader;\nimport org.objectweb.asm.ClassVisitor;\nimport org.objectweb.asm.ClassWriter;\nimport org.objectweb.asm.commons.ClassRemapper;\nimport org.objectweb.asm.commons.Remapper;\nimport org.objectweb.asm.commons.SimpleRemapper;\n\nimport java.util.HashMap;\nimport java.util.Map;\n\npublic class ClassRemapperExample03 {\n public static void main(String[] args) {\n Map mapping = new HashMap<>();\n mapping.put(\"sample/HelloWorld\", \"sample/AAA\");\n mapping.put(\"sample/GoodChild\", \"sample/BBB\");\n mapping.put(\"sample/HelloWorld.test()V\", \"a\");\n mapping.put(\"sample/GoodChild.study()V\", \"b\");\n obfuscate(\"sample/HelloWorld\", \"sample/AAA\", mapping);\n obfuscate(\"sample/GoodChild\", \"sample/BBB\", mapping);\n }\n\n public static void obfuscate(String origin_name, String target_name, Map mapping) {\n String origin_filepath = getFilePath(origin_name);\n byte[] bytes1 = FileUtils.readBytes(origin_filepath);\n\n //（1）构建ClassReader\n ClassReader cr = new ClassReader(bytes1);\n\n //（2）构建ClassWriter\n ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);\n\n //（3）串连ClassVisitor\n Remapper mapper = new SimpleRemapper(mapping);\n ClassVisitor cv = new ClassRemapper(cw, mapper);\n\n //（4）两者进行结合\n int parsingOptions = ClassReader.SKIP_DEBUG | ClassReader.SKIP_FRAMES;\n cr.accept(cv, parsingOptions);\n\n //（5）重新生成Class\n byte[] bytes2 = cw.toByteArray();\n\n String target_filepath = getFilePath(target_name);\n FileUtils.writeBytes(target_filepath, bytes2);\n }\n\n public static String getFilePath(String internalName) {\n String relative_path = String.format(\"%s.class\", internalName);\n return FileUtils.getFilePath(relative_path);\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/commons/GeneratorAdapterExample01.java",
"content": "package lsieun.asm.commons;\n\nimport lsieun.utils.FileUtils;\nimport org.objectweb.asm.ClassWriter;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.commons.GeneratorAdapter;\nimport org.objectweb.asm.commons.Method;\nimport org.objectweb.asm.util.TraceClassVisitor;\n\nimport java.io.PrintStream;\nimport java.io.PrintWriter;\n\nimport static org.objectweb.asm.Opcodes.*;\n\npublic class GeneratorAdapterExample01 {\n public static void main(String[] args) throws Exception {\n String relative_path = \"sample/HelloWorld.class\";\n String filepath = FileUtils.getFilePath(relative_path);\n\n // (1) 生成byte[]内容\n byte[] bytes = dump();\n\n // (2) 保存byte[]到文件\n FileUtils.writeBytes(filepath, bytes);\n }\n\n public static byte[] dump() throws Exception {\n ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);\n PrintWriter printWriter = new PrintWriter(System.out);\n TraceClassVisitor cv = new TraceClassVisitor(cw, printWriter);\n\n cv.visit(V1_8, ACC_PUBLIC + ACC_SUPER, \"sample/HelloWorld\", null, \"java/lang/Object\", null);\n\n {\n Method m1 = Method.getMethod(\"void ()\");\n GeneratorAdapter mg = new GeneratorAdapter(ACC_PUBLIC, m1, null, null, cv);\n mg.loadThis();\n mg.invokeConstructor(Type.getType(Object.class), m1);\n mg.returnValue();\n mg.endMethod();\n }\n\n {\n Method m2 = Method.getMethod(\"void main (String[])\");\n GeneratorAdapter mg = new GeneratorAdapter(ACC_PUBLIC + ACC_STATIC, m2, null, null, cv);\n mg.getStatic(Type.getType(System.class), \"out\", Type.getType(PrintStream.class));\n mg.push(\"Hello world!\");\n mg.invokeVirtual(Type.getType(PrintStream.class), Method.getMethod(\"void println (String)\"));\n mg.returnValue();\n mg.endMethod();\n }\n\n cv.visitEnd();\n\n return cw.toByteArray();\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/commons/InstructionAdapterExample01.java",
"content": "package lsieun.asm.commons;\n\nimport lsieun.utils.FileUtils;\nimport org.objectweb.asm.ClassWriter;\nimport org.objectweb.asm.MethodVisitor;\nimport org.objectweb.asm.Type;\nimport org.objectweb.asm.commons.InstructionAdapter;\n\nimport static org.objectweb.asm.Opcodes.*;\n\npublic class InstructionAdapterExample01 {\n public static void main(String[] args) throws Exception {\n String relative_path = \"sample/HelloWorld.class\";\n String filepath = FileUtils.getFilePath(relative_path);\n\n // (1) 生成byte[]内容\n byte[] bytes = dump();\n\n // (2) 保存byte[]到文件\n FileUtils.writeBytes(filepath, bytes);\n }\n\n public static byte[] dump() throws Exception {\n // (1) 创建ClassWriter对象\n ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);\n\n // (2) 调用visitXxx()方法\n cw.visit(V1_8, ACC_PUBLIC + ACC_SUPER, \"sample/HelloWorld\",\n null, \"java/lang/Object\", null);\n\n {\n MethodVisitor mv1 = cw.visitMethod(ACC_PUBLIC, \"\", \"()V\", null, null);\n InstructionAdapter ia = new InstructionAdapter(mv1);\n ia.visitCode();\n ia.load(0, InstructionAdapter.OBJECT_TYPE);\n ia.invokespecial(\"java/lang/Object\", \"\", \"()V\", false);\n ia.areturn(Type.VOID_TYPE);\n ia.visitMaxs(1, 1);\n ia.visitEnd();\n }\n\n {\n MethodVisitor mv2 = cw.visitMethod(ACC_PUBLIC, \"test\", \"()V\", null, null);\n InstructionAdapter ia = new InstructionAdapter(mv2);\n ia.visitCode();\n ia.getstatic(\"java/lang/System\", \"out\", \"Ljava/io/PrintStream;\");\n ia.aconst(\"Hello World\");\n ia.invokevirtual(\"java/io/PrintStream\", \"println\", \"(Ljava/lang/String;)V\", false);\n ia.areturn(Type.VOID_TYPE);\n ia.visitMaxs(2, 1);\n ia.visitEnd();\n }\n\n cw.visitEnd();\n\n // (3) 调用toByteArray()方法\n return cw.toByteArray();\n }\n}\n"
},
{
"path": "src/main/java/lsieun/asm/commons/MethodStackMapFrame02Visitor.java",
"content": "package lsieun.asm.commons;\n\nimport lsieun.classfile.ClassFile;\nimport lsieun.classfile.InsnRaw;\nimport org.objectweb.asm.*;\nimport org.objectweb.asm.commons.AnalyzerAdapter;\n\nimport java.util.List;\n\npublic class MethodStackMapFrame02Visitor extends ClassVisitor {\n private String owner;\n private final ClassFile classFile;\n private int methodIndex = 0;\n\n public MethodStackMapFrame02Visitor(int api, ClassVisitor classVisitor, byte[] bytes) {\n super(api, classVisitor);\n this.classFile = new ClassFile(bytes);\n }\n\n @Override\n public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {\n super.visit(version, access, name, signature, superName, interfaces);\n this.owner = name;\n }\n\n @Override\n public MethodVisitor visitMethod(int access, String name, String descriptor, String signature, String[] exceptions) {\n MethodVisitor mv = super.visitMethod(access, name, descriptor, signature, exceptions);\n byte[] code_bytes = classFile.getCode(methodIndex++);\n InsnRaw insnRaw = new InsnRaw(code_bytes);\n List list = insnRaw.getList();\n mv = new MethodStackMapFrame02Adapter(api, owner, access, name, descriptor, mv, list);\n return mv;\n }\n\n private static class MethodStackMapFrame02Adapter extends AnalyzerAdapter {\n private static final String FORMAT = \"%-30s // %s\";\n private static final String START = \"{\";\n private static final String STOP = \"}\";\n private static final String EMPTY = \"{}\";\n private static final String SEPARATOR = \"|\";\n\n private final int methodAccess;\n private final String methodName;\n private final String methodDesc;\n private final List insnList;\n private int insnIndex = 0;\n\n public MethodStackMapFrame02Adapter(int api, String owner, int access, String name, String descriptor, MethodVisitor methodVisitor, List insnList) {\n super(api, owner, access, name, descriptor, methodVisitor);\n this.methodAccess = access;\n this.methodName = name;\n this.methodDesc = descriptor;\n this.insnList = insnList;\n }\n\n @Override\n public void visitCode() {\n super.visitCode();\n\n System.out.println(methodName + \":\" + methodDesc);\n String frame = getStackFrame();\n String line = String.format(FORMAT, \"\", frame);\n System.out.println(line);\n }\n\n @Override\n public void visitFrame(int type, int numLocal, Object[] local, int numStack, Object[] stack) {\n super.visitFrame(type, numLocal, local, numStack, stack);\n\n String frame = getStackFrame();\n String line = String.format(FORMAT, \"\", frame);\n System.out.println(line);\n }\n\n @Override\n public void visitInsn(int opcode) {\n super.visitInsn(opcode);\n printStackMapFrame();\n }\n\n @Override\n public void visitIntInsn(int opcode, int operand) {\n super.visitIntInsn(opcode, operand);\n printStackMapFrame();\n }\n\n @Override\n public void visitVarInsn(int opcode, int var) {\n super.visitVarInsn(opcode, var);\n printStackMapFrame();\n }\n\n @Override\n public void visitTypeInsn(int opcode, String type) {\n super.visitTypeInsn(opcode, type);\n printStackMapFrame();\n }\n\n @Override\n public void visitFieldInsn(int opcode, String owner, String name, String descriptor) {\n super.visitFieldInsn(opcode, owner, name, descriptor);\n printStackMapFrame();\n }\n\n @Override\n public void visitMethodInsn(int opcode, String owner, String name, String descriptor, boolean isInterface) {\n super.visitMethodInsn(opcode, owner, name, descriptor, isInterface);\n printStackMapFrame();\n }\n\n @Override\n public void visitInvokeDynamicInsn(String name, String descriptor, Handle bootstrapMethodHandle, Object... bootstrapMethodArguments) {\n super.visitInvokeDynamicInsn(name, descriptor, bootstrapMethodHandle, bootstrapMethodArguments);\n printStackMapFrame();\n }\n\n @Override\n public void visitJumpInsn(int opcode, Label label) {\n super.visitJumpInsn(opcode, label);\n printStackMapFrame();\n }\n\n @Override\n public void visitLdcInsn(Object value) {\n super.visitLdcInsn(value);\n printStackMapFrame();\n }\n\n @Override\n public void visitIincInsn(int var, int increment) {\n super.visitIincInsn(var, increment);\n printStackMapFrame();\n }\n\n @Override\n public void visitTableSwitchInsn(int min, int max, Label dflt, Label... labels) {\n super.visitTableSwitchInsn(min, max, dflt, labels);\n\n int insnSize = insnList.size();\n String insn;\n if (insnIndex < insnSize) {\n insn = insnList.get(insnIndex);\n insnIndex++;\n }\n else {\n insn = \"\";\n }\n\n String[] array = insn.split(System.lineSeparator(), 2);\n String frame = getStackFrame();\n String line = String.format(FORMAT, array[0], frame);\n System.out.println(line);\n System.out.println(array[1]);\n }\n\n @Override\n public void visitLookupSwitchInsn(Label dflt, int[] keys, Label[] labels) {\n super.visitLookupSwitchInsn(dflt, keys, labels);\n\n int insnSize = insnList.size();\n String insn;\n if (insnIndex < insnSize) {\n insn = insnList.get(insnIndex);\n insnIndex++;\n }\n else {\n insn = \"\";\n }\n\n String[] array = insn.split(System.lineSeparator(), 2);\n String frame = getStackFrame();\n String line = String.format(FORMAT, array[0], frame);\n System.out.println(line);\n System.out.println(array[1]);\n }\n\n @Override\n public void visitMultiANewArrayInsn(String descriptor, int numDimensions) {\n super.visitMultiANewArrayInsn(descriptor, numDimensions);\n printStackMapFrame();\n }\n\n @Override\n public void visitEnd() {\n super.visitEnd();\n\n System.out.println();\n System.out.println();\n }\n\n private void printStackMapFrame() {\n int insnSize = insnList.size();\n String insn;\n if (insnIndex < insnSize) {\n insn = insnList.get(insnIndex);\n insnIndex++;\n }\n else {\n insn = \"\";\n }\n\n if (insn.contains(\" wide \")) {\n System.out.println(insn);\n insn = insnList.get(insnIndex);\n insnIndex++;\n }\n\n String frame = getStackFrame();\n String line = String.format(FORMAT, insn, frame);\n System.out.println(line);\n }\n\n private String getStackFrame() {\n if (locals != null && locals.size() > 0) {\n boolean isStaticMethod = (methodAccess & Opcodes.ACC_STATIC) != 0;\n if (!isStaticMethod) {\n Object obj = locals.get(0);\n if (obj != Opcodes.UNINITIALIZED_THIS) {\n locals.set(0, \"this\");\n }\n }\n }\n String locals_str = locals == null ? EMPTY : list2Str(locals);\n String stack_str = stack == null ? EMPTY : list2Str(stack);\n return String.format(\"%s %s %s\", locals_str, SEPARATOR, stack_str);\n }\n\n private String list2Str(List