ASM学习思路

关于AOP、Plugin、Transform的概念，大家都卷到这个份上了，就不再细说了。撸个经典的demo，方法耗时检测，提供一下学习思路。相信按照下面这个流程走下来，肯定是能入门了，毕竟我都能学会。

进入正题，说实话直接用asm api织入字节码还是很懵逼的。工欲善其事必先利其器，下载字节码插件ASM Bytecode Viewer。

首先需要一个辅助类，保存方法起始时间、结束时间，还需要计算耗时的方法current。

public class TimeManager {
    private static final HashMap<String, Long> startMap = new HashMap<>();
    private static final HashMap<String, Long> endMap = new HashMap<>();

    public static void start(String key, long timeMills) {
        startMap.put(key, timeMills);
    }

    public static void end(String key, long timeMills) {
        endMap.put(key, timeMills);
    }

    public static long current(String key) {
        long startTimeMills = 0L;
        long endTimeMills = 0L;
        if (startMap.get(key) != null) {
            startTimeMills = startMap.get(key);
        }
        if (endMap.get(key) != null) {
            endTimeMills = endMap.get(key);
        }
        return endTimeMills - startTimeMills;
    }
}

没什么好多说的，以方法名为key。那么接下来就是ClassVisitor、MethodVisitor。

class TrackClassVisitor(classVisitor: ClassVisitor) : ClassVisitor(Opcodes.ASM9, classVisitor) {
    
    override fun visitMethod(
        access: Int,
        name: String?,
        descriptor: String?,
        signature: String?,
        exceptions: Array<out String>?
    ): MethodVisitor {
        val mv = cv.visitMethod(access, name, descriptor, signature, exceptions)
        return TrackMethodVisitor(mv, access, name, descriptor)
    }
}

object TrackMethodVisitor {

    operator fun invoke(
        mv: MethodVisitor,
        access: Int,
        name: String?,
        descriptor: String?,
    ): MethodVisitor {
        return object : AdviceAdapter(Opcodes.ASM9, mv, access, name, descriptor) {

            override fun visitAnnotation(descriptor: String?, visible: Boolean): AnnotationVisitor {
                return super.visitAnnotation(descriptor, visible)
            }

            override fun onMethodEnter() {
                super.onMethodEnter()
            }

            override fun onMethodExit(opcode: Int) {
                super.onMethodExit(opcode)
            }
        }
    }
}

asm的代码怎么写呢，别急，弄个工具类TimeUtil，启动插件。

public class TimeUtil {

    public void handler() {
        TimeManager.start("", System.currentTimeMillis());
        TimeManager.end("", System.currentTimeMillis());
        long currentTimeMills = TimeManager.current("");
    }
}

点个锤子编译一下，查看build中编译后的TimeUtil，右键ASM Bytecode Viewer。

TimeUtil.class

直接看生成的asm api还是看不太明白

ASMified

没关系，对应字节码看就很清晰了，切换到Bytecodetab

Bytecode

仔细对照一下，从LDC ""开始对比，asm api和字节码能一一对应。接下来的工作就很简单了，对照字节码去掉无用的label，然后cv大法。

因为这里MethodVisitor使用的AdviceAdapter，onMethodEnter()、onMethodExit()就可以很方便的在方法前后插桩。

方法前插入

        TimeManager.start("", System.currentTimeMillis());

onMethodEnter()

            override fun onMethodEnter() {
                mv.visitLdcInsn(name);//方法名
                mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false);
                mv.visitMethodInsn(INVOKESTATIC, "com/chenxuan/annotation/TimeManager", "start", "(Ljava/lang/String;J)V", false);
                super.onMethodEnter()
            }

方法后插入

        TimeManager.end("", System.currentTimeMillis());
        long currentTimeMills = TimeManager.current("");
        Log.d("chenxuan----->", "Method name total time: " + currentTimeMills + "ms");

onMethodExit()

            override fun onMethodExit(opcode: Int) {
                mv.visitLdcInsn(name)
                mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false)
                mv.visitMethodInsn(INVOKESTATIC, "com/chenxuan/annotation/TimeManager", "end", "(Ljava/lang/String;J)V", false)
                mv.visitLdcInsn(name)
                mv.visitMethodInsn(INVOKESTATIC, "com/chenxuan/annotation/TimeManager", "current", "(Ljava/lang/String;)J", false)
                mv.visitVarInsn(LSTORE, 1)
                mv.visitLdcInsn("chenxuan----->")
                mv.visitTypeInsn(NEW, "java/lang/StringBuilder")
                mv.visitInsn(DUP)
                mv.visitMethodInsn(INVOKESPECIAL, "java/lang/StringBuilder", "<init>", "()V", false)
                mv.visitLdcInsn("Method $name total time: ")
                mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(Ljava/lang/String;)Ljava/lang/StringBuilder;", false)
                mv.visitVarInsn(LLOAD, 1)
                mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(J)Ljava/lang/StringBuilder;", false)
                mv.visitLdcInsn("ms")
                mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(Ljava/lang/String;)Ljava/lang/StringBuilder;", false)
                mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "toString", "()Ljava/lang/String;", false)
                mv.visitMethodInsn(INVOKESTATIC, "android/util/Log", "d", "(Ljava/lang/String;Ljava/lang/String;)I", false)
                mv.visitInsn(POP)
                super.onMethodExit(opcode)
            }

加个注解Track，只对注解过的方法进行插桩。

@Target({ElementType.METHOD})
public @interface Track {
}

TrackMethodVisitor最终代码

object TrackMethodVisitor {

    operator fun invoke(
        mv: MethodVisitor,
        access: Int,
        name: String?,
        descriptor: String?,
    ): MethodVisitor {
        return object : AdviceAdapter(Opcodes.ASM9, mv, access, name, descriptor) {
            var needTrack = false

            override fun visitAnnotation(descriptor: String?, visible: Boolean): AnnotationVisitor {
                if (Type.getDescriptor(Track::class.java) == descriptor) needTrack = true
                return super.visitAnnotation(descriptor, visible)
            }

            override fun onMethodEnter() {
                if (needTrack){
                    mv.visitLdcInsn("");
                    mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false);
                    mv.visitMethodInsn(INVOKESTATIC, "com/chenxuan/annotation/TimeManager", "start", "(Ljava/lang/String;J)V", false);
                }
                super.onMethodEnter()
            }

            override fun onMethodExit(opcode: Int) {
                if (needTrack){
                    mv.visitLdcInsn(name)
                    mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false)
                    mv.visitMethodInsn(INVOKESTATIC, "com/chenxuan/annotation/TimeManager", "end", "(Ljava/lang/String;J)V", false)
                    mv.visitLdcInsn(name)
                    mv.visitMethodInsn(INVOKESTATIC, "com/chenxuan/annotation/TimeManager", "current", "(Ljava/lang/String;)J", false)
                    mv.visitVarInsn(LSTORE, 1)
                    mv.visitLdcInsn("chenxuan----->")
                    mv.visitTypeInsn(NEW, "java/lang/StringBuilder")
                    mv.visitInsn(DUP)
                    mv.visitMethodInsn(INVOKESPECIAL, "java/lang/StringBuilder", "<init>", "()V", false)
                    mv.visitLdcInsn("Method $name total time: ")
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(Ljava/lang/String;)Ljava/lang/StringBuilder;", false)
                    mv.visitVarInsn(LLOAD, 1)
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(J)Ljava/lang/StringBuilder;", false)
                    mv.visitLdcInsn("ms")
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(Ljava/lang/String;)Ljava/lang/StringBuilder;", false)
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "toString", "()Ljava/lang/String;", false)
                    mv.visitMethodInsn(INVOKESTATIC, "android/util/Log", "d", "(Ljava/lang/String;Ljava/lang/String;)I", false)
                    mv.visitInsn(POP)
                }
                super.onMethodExit(opcode)
            }
        }
    }
}

测试一下MainActivity

class MainActivity : AppCompatActivity() {
    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContentView(R.layout.activity_main)
        trackMethod()
    }

    @Track
    private fun trackMethod() {
        val data = mutableListOf<String>()
        Thread.sleep(50)
    }
}

编译后查看build下transforms文件夹

MainActivity.class

Tools->kotlin反编译成Java

MainActivity

插桩成功，但是有个问题，TimeManager工具类以方法名为key保存方法调用的起始时间、结束时间，不太优雅。而且方法名会重复，当然也可以做一下优化，key拼接下类名，但是同一个类多个对象还是会重复，虽然可以继续优化，但是最好的做法肯定是用局部变量。

public class TimeUtil {

    public void handler() {
        long start = System.currentTimeMillis();
        long end = System.currentTimeMillis();
        long currentTimeMills = end - start;
        Log.d("chenxuan----->", "Method name total time: " + currentTimeMills + "ms");
    }
}

老样子，编译后查看TimeUtil.class，然后使用插件。

ASMified

嗯，对照Bytecode，cv到TrackMethodVisitor

object TrackMethodVisitor {

    operator fun invoke(
        mv: MethodVisitor,
        access: Int,
        name: String?,
        descriptor: String?,
    ): MethodVisitor {
        return object : AdviceAdapter(Opcodes.ASM9, mv, access, name, descriptor) {
            var needTrack = false

            override fun visitAnnotation(descriptor: String?, visible: Boolean): AnnotationVisitor {
                if (Type.getDescriptor(Track::class.java) == descriptor) needTrack = true
                return super.visitAnnotation(descriptor, visible)
            }

            override fun onMethodEnter() {
                if (needTrack) {
                    mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false);
                    mv.visitVarInsn(LSTORE, 1);
                }
                super.onMethodEnter()
            }

            override fun onMethodExit(opcode: Int) {
                if (needTrack) {
                    mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false);
                    mv.visitVarInsn(LSTORE, 3);
                    mv.visitVarInsn(LLOAD, 3);
                    mv.visitVarInsn(LLOAD, 1);
                    mv.visitInsn(LSUB);
                    mv.visitVarInsn(LSTORE, 5);
                    mv.visitLdcInsn("chenxuan----->");
                    mv.visitTypeInsn(NEW, "java/lang/StringBuilder");
                    mv.visitInsn(DUP);
                    mv.visitMethodInsn(INVOKESPECIAL, "java/lang/StringBuilder", "<init>", "()V", false);
                    mv.visitLdcInsn("Method name total time: ");
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(Ljava/lang/String;)Ljava/lang/StringBuilder;", false);
                    mv.visitVarInsn(LLOAD, 5);
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(J)Ljava/lang/StringBuilder;", false);
                    mv.visitLdcInsn("ms");
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(Ljava/lang/String;)Ljava/lang/StringBuilder;", false);
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "toString", "()Ljava/lang/String;", false);
                    mv.visitMethodInsn(INVOKESTATIC, "android/util/Log", "d", "(Ljava/lang/String;Ljava/lang/String;)I", false);
                    mv.visitInsn(POP);
                }
                super.onMethodExit(opcode)
            }
        }
    }
}

build不出意外会报错。MainActivity中的track()有一行

val data = mutableListOf<String>()

回看上面的asm中插入两个局部变量的代码

            override fun onMethodEnter() {
                if (needTrack) {
                    mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false);
                    mv.visitVarInsn(LSTORE, 1);
                }
                super.onMethodEnter()
            }

             override fun onMethodExit(opcode: Int) {
                if (needTrack) {
                    mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false);
                    mv.visitVarInsn(LSTORE, 3);
                }
                super.onMethodExit(opcode)
            }
        }

LSTORE在方法前插入1的位置，方法后插入3，这个可没有考虑到方法体本身也有局部变量，在方法后插入局部变量表的理想操作是先拿到表的长度N，然后mv.visitVarInsn(LSTORE, N+2);。理想很丰满现实很骨感，搜索一番没有找到获取局部变量表长度的api，有点裂开了。好在之前有不少大佬写过这种东西，面向google不丢人，“借鉴”了大佬们的写法，最后用newLocal搞定了，先上代码。

object TrackMethodVisitor {

    operator fun invoke(
        mv: MethodVisitor,
        access: Int,
        name: String?,
        descriptor: String?,
    ): MethodVisitor {
        return object : AdviceAdapter(Opcodes.ASM9, mv, access, name, descriptor) {
            var needTrack = false
            var startTimeMills = -1

            override fun visitAnnotation(descriptor: String?, visible: Boolean): AnnotationVisitor {
                if (Type.getDescriptor(Track::class.java) == descriptor) needTrack = true
                return super.visitAnnotation(descriptor, visible)
            }

            override fun onMethodEnter() {
                if (needTrack) {
                    startTimeMills = newLocal(Type.LONG_TYPE)
                    mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false)
                    mv.visitIntInsn(LSTORE, startTimeMills)
                }
                super.onMethodEnter()
            }

            override fun onMethodExit(opcode: Int) {
                if (needTrack) {
                    val durationTimeMills = newLocal(Type.LONG_TYPE)
                    mv.visitMethodInsn(INVOKESTATIC, "java/lang/System", "currentTimeMillis", "()J", false)
                    mv.visitVarInsn(LLOAD, startTimeMills)
                    mv.visitInsn(LSUB)
                    mv.visitVarInsn(LSTORE, durationTimeMills)
                    mv.visitLdcInsn("chenxuan----->")
                    mv.visitTypeInsn(NEW, "java/lang/StringBuilder")
                    mv.visitInsn(DUP)
                    mv.visitMethodInsn(INVOKESPECIAL, "java/lang/StringBuilder", "<init>", "()V", false)
                    mv.visitLdcInsn("Method $name total time: ")
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(Ljava/lang/String;)Ljava/lang/StringBuilder;", false)
                    mv.visitVarInsn(LLOAD, durationTimeMills)
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(J)Ljava/lang/StringBuilder;", false)
                    mv.visitLdcInsn("ms")
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "append", "(Ljava/lang/String;)Ljava/lang/StringBuilder;", false)
                    mv.visitMethodInsn(INVOKEVIRTUAL, "java/lang/StringBuilder", "toString", "()Ljava/lang/String;", false)
                    mv.visitMethodInsn(INVOKESTATIC, "android/util/Log", "d", "(Ljava/lang/String;Ljava/lang/String;)I", false)
                    mv.visitInsn(POP);
                }
                super.onMethodExit(opcode)
            }
        }
    }
}

说人话：startTimeMills记录了插入的位置mv.visitIntInsn(LSTORE, startTimeMills)，然后计算结束时间end；取出startTimeMills，end与其相减，插入局部变量表，记录一下durationTimeMills；后面拼接string的时候mv.visitVarInsn(LLOAD, durationTimeMills)取出duration。好起来了，接下来编译看一下成果。

MainActivity

没啥问题，按照这个思路卷一卷岂不是美滋滋。

最后编辑于：2021.11.19 14:27:14

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 158,847评论 4赞 362
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 67,208评论 1赞 292
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 108,587评论 0赞 243
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 43,942评论 0赞 205
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 52,332评论 3赞 287
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 40,587评论 1赞 218
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 31,853评论 2赞 312
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 30,568评论 0赞 198
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 34,273评论 1赞 242
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 30,542评论 2赞 246
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 32,033评论 1赞 260
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 28,373评论 2赞 253
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 33,031评论 3赞 236
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 26,073评论 0赞 8
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 26,830评论 0赞 195
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 35,628评论 2赞 274
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 35,537评论 2赞 269

ASM学习思路

推荐阅读更多精彩内容