Unidbg中Hook和Patch

ゝ一世哀愁。 2023-09-26 16:22 28阅读 0赞

hook之前之后类似于xposed  
patch两种方式，直接修改hex,以及指令转化hex(keystone 将汇编指令转化为hex)

hex to arm | arm to hex 推荐网站：  
[https://armconverter.com/][https_armconverter.com]

package com.dta.lesson5;
    
    import com.github.unidbg.AndroidEmulator;
    import com.github.unidbg.Emulator;
    import com.github.unidbg.Module;
    import com.github.unidbg.arm.HookStatus;
    import com.github.unidbg.arm.context.Arm32RegisterContext;
    import com.github.unidbg.arm.context.RegisterContext;
    import com.github.unidbg.hook.Arm32HookContext;
    import com.github.unidbg.hook.HookContext;
    import com.github.unidbg.hook.IHook;
    import com.github.unidbg.hook.ReplaceCallback;
    import com.github.unidbg.hook.hookzz.HookEntryInfo;
    import com.github.unidbg.hook.hookzz.HookZz;
    import com.github.unidbg.hook.hookzz.WrapCallback;
    import com.github.unidbg.linux.android.AndroidEmulatorBuilder;
    import com.github.unidbg.linux.android.AndroidResolver;
    import com.github.unidbg.linux.android.dvm.DalvikModule;
    import com.github.unidbg.linux.android.dvm.DvmObject;
    import com.github.unidbg.linux.android.dvm.StringObject;
    import com.github.unidbg.linux.android.dvm.VM;
    import com.github.unidbg.linux.android.dvm.jni.ProxyDvmObject;
    import com.github.unidbg.memory.Memory;
    import com.github.unidbg.pointer.UnidbgPointer;
    import com.sun.jna.Pointer;
    import keystone.Keystone;
    import keystone.KeystoneArchitecture;
    import keystone.KeystoneMode;
    import unicorn.ArmConst;
    import unicorn.Unicorn;
    import unicorn.UnicornConst;
    
    import java.io.File;
    import java.util.ArrayList;
    import java.util.List;
    
    public class MainActivity {
        
        private final AndroidEmulator emulator;
        private final VM vm;
        private final Memory memory;
        private final Module module;
    
        public MainActivity(){
        
            emulator = AndroidEmulatorBuilder
                    .for32Bit()
                    //.setRootDir(new File("target/rootfs/default"))
                    //.addBackendFactory(new DynarmicFactory(true))
                    .build();
    
            memory = emulator.getMemory();
            memory.setLibraryResolver(new AndroidResolver(23));
    
            vm = emulator.createDalvikVM();
    
            DalvikModule dalvikModule = vm.loadLibrary(new File("unidbg-android/src/test/java/com/dta/lesson5/libnative-lib.so"), true);
            module = dalvikModule.getModule();
    
            vm.callJNI_OnLoad(emulator,module);
        }
    
        public void callAdd(){
        
            DvmObject obj = ProxyDvmObject.createObject(vm,this);
            int result = obj.callJniMethodInt(emulator, "add(II)I", 1,2);
            System.out.println("[symble] Call the so add function result is ==> "+ result);
        }
    
        public void hook(){
        
            HookZz hook = HookZz.getInstance(emulator);
            hook.replace(module.base + 0x3DC + 1, new ReplaceCallback() {
        
                @Override
                // hook之前
                public HookStatus onCall(Emulator<?> emulator, HookContext context, long originFunction) {
        
                    System.out.println(String.format("R2: %d, R3: %d",context.getIntArg(2),context.getIntArg(3)));
                    emulator.getBackend().reg_write(Unicorn.UC_ARM_REG_R3,5);
                    return super.onCall(emulator, context, originFunction);
                }
    
                @Override
                // hook之后
                public void postCall(Emulator<?> emulator, HookContext context) {
        
                    emulator.getBackend().reg_write(Unicorn.UC_ARM_REG_R0,10);
                    super.postCall(emulator, context);
                }
            }, true);
        }
    
    	//直接更改hex值
        public void patch(){
        
            UnidbgPointer pointer = UnidbgPointer.pointer(emulator,module.base + 0x3E8);
            byte[] code = new byte[]{
        (byte) 0xd0, 0x1a};
            pointer.write(code);
        }
    	//keystone 将汇编指令转化为hex
        public void patch2(){
        
            UnidbgPointer pointer = UnidbgPointer.pointer(emulator,module.base + 0x3E8);
            Keystone keystone = new Keystone(KeystoneArchitecture.Arm, KeystoneMode.ArmThumb);
            String s = "subs r0, r2, r3";
            byte[] machineCode = keystone.assemble(s).getMachineCode();
            //byte[] code = ;
            pointer.write(machineCode);
        }
    
        public static void main(String[] args) {
        
            long start = System.currentTimeMillis();
            MainActivity mainActivity = new MainActivity();
            System.out.println("load the vm "+( System.currentTimeMillis() - start )+ "ms");
            //mainActivity.hook();
            //mainActivity.patch();
            mainActivity.patch2();
            mainActivity.callAdd();
        }
    
    }

[https_armconverter.com]: https://armconverter.com/