首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >使用ASM重写Java本机方法

使用ASM重写Java本机方法
EN

Stack Overflow用户
提问于 2012-11-08 01:16:18
回答 2查看 2K关注 0票数 17

我试图通过使用ASM 4.0重写类的字节码来实现这一点,用非native存根替换所有native方法。

到目前为止,我有这样的想法:

代码语言:javascript
复制
class ClassAdapter extends ClassVisitor {

    public ClassAdapter(ClassVisitor cv) {
        super(Opcodes.ASM4, cv);
    }

    @Override
    public MethodVisitor visitMethod(int access, String base, String desc, String signature, String[] exceptions) {
        return cv.visitMethod(access & ~Opcodes.ACC_NATIVE, base, desc, signature, exceptions);
    }

}

,它由执行

代码语言:javascript
复制
private static byte[] instrument(byte[] originalBytes, ClassLoader loader) {
    ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
    ClassAdapter adapter = new ClassAdapter(cw);

    ClassReader cr = new ClassReader(originalBytes);
    cr.accept(adapter, ClassReader.SKIP_FRAMES);

    return cw.toByteArray();
}

这似乎很简单:我从visitMethod()中的方法中剥离了ACC_NATIVE,并保持其他所有内容不变。但是,当我对java.lang.Object执行此操作时,它将以

代码语言:javascript
复制
Exception in thread "main" 
Exception: java.lang.StackOverflowError thrown from the UncaughtExceptionHandler in thread "main"

StackOverflow在检测时发生,而不是在运行时,我认为这是相当不寻常的。但是,如果我删除了& ~Opcodes.ACC_NATIVE修饰符,java.lang.Object会被重写(在本例中没有变化),并且可以完美地执行。

很明显,我没有做正确的事情,并且用非native方法替换native方法并不像去掉该方法上的native修饰符那么简单,但是我不知道从哪里开始。ASM Docs根本不讨论使用native方法。有使用ASM的经验的人知道我需要做什么才能让native方法重写工作吗?

编辑

对不起,那条简短的、无用的消息是e.printStackTrace()给我的,但使用e.getStackTrace()我设法得到了一些有用的东西:

代码语言:javascript
复制
java.util.concurrent.ConcurrentHashMap.hash(ConcurrentHashMap.java:332)
java.util.concurrent.ConcurrentHashMap.put(ConcurrentHashMap.java:1124)
java.util.Collections$SetFromMap.add(Collections.java:3903)
sandbox.classloader.MyClassLoader.instrument(Unknown Source)
sandbox.classloader.MyClassLoader.loadClass(Unknown Source)
java.lang.ClassLoader.defineClass1(Native Method)
java.lang.ClassLoader.defineClass(ClassLoader.java:791)
java.lang.ClassLoader.defineClass(ClassLoader.java:634)
sandbox.classloader.MyClassLoader.findClass(Unknown Source)
sandbox.classloader.MyClassLoader.loadClass(Unknown Source)
sandbox.Tester.main(Unknown Source)
sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:57)
sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
java.lang.reflect.Method.invoke(Method.java:601)
com.intellij.rt.execution.application.AppMain.main(AppMain.java:120)

因此,在我看来,错误实际上是在执行时发生的(例如,我错误地认为它是在检测时发生的),并且是调用hashCode()的结果。碰巧的是,我(可能不正确地)从native修饰符中剥离了一个原生方法hashCode()。因此,很明显,它调用的是native-stripped方法,这是导致问题的原因。

看起来奇怪的是,堆栈跟踪只有16帧深;如果它是一个StackOverflowError,我会期望更多。

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2012-11-12 03:36:57

票数 2
EN

Stack Overflow用户

发布于 2018-02-09 20:55:10

为了详细说明公认的答案,这里是一个完整的插装代理示例,它使用ASM将本机方法java.net.NetworkInterface#getHardwareAddress()替换为返回固定值的存根。

代码语言:javascript
复制
public class MacModifyAgent {

    private static final String TARGET = "java/net/NetworkInterface";

    public static void premain(String agentArgs, Instrumentation inst) {
        inst.addTransformer(new ClassFileTransformer() {
            @Override
            public byte[] transform(ClassLoader l, String name, Class<?> c, ProtectionDomain d, byte[] b)
                    throws IllegalClassFormatException {
                if (TARGET.equals(name)) {
                    return instrument(b);
                }
                return b;
            }
        });
    }

    private static byte[] instrument(byte[] originalBytes) {
        ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
        ClassAdapter adapter = new ClassAdapter(cw);

        ClassReader cr = new ClassReader(originalBytes);
        cr.accept(adapter, ClassReader.SKIP_FRAMES);

        return cw.toByteArray();
    }

    public static class ClassAdapter extends ClassVisitor implements Opcodes {
        public ClassAdapter(ClassVisitor cv) {
            super(ASM4, cv);
        }

        @Override
        public MethodVisitor visitMethod(int access, String name, String descriptor, String signature,
                String[] exceptions) {

            if ("getHardwareAddress".equals(name)) {
                MethodVisitor mv = super.visitMethod(access & ~ACC_NATIVE, name, descriptor, signature, exceptions);

                MethodVisitor special = new StubReturnValue(mv, new byte[] { 1, 2, 3, 4, 5, 6 });
                return special;
            } else {
                return super.visitMethod(access, name, descriptor, signature, exceptions);
            }
        }
    }

    public static class StubReturnValue extends MethodVisitor implements Opcodes {
        private final MethodVisitor target;
        private byte[] mac;

        public StubReturnValue(MethodVisitor target, byte [] mac) {
            super(ASM4, null);
            this.target = target;
        }

        @Override
        public void visitCode() {
            target.visitCode();
            target.visitVarInsn(BIPUSH, 6);
            target.visitIntInsn(NEWARRAY, T_BYTE);

            for (int i = 0; i < 6; i++) {
                target.visitInsn(DUP);
                target.visitIntInsn(BIPUSH, i);
                target.visitIntInsn(BIPUSH, mac[i]);
                target.visitInsn(BASTORE);
            }

            target.visitInsn(ARETURN);
            target.visitEnd();
        }
    }
}
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/13274771

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档