Android arm64(aarch64)中的so注入(inject) - 相容x86 and arm

標籤：android   arm64   inject   注入   

實現Android arm64(aarch64)中的so注入(inject) ，並且相容x86和arm。如果沒有搞錯，這是國內外第一份公開的arm64注入的針對性完整資料～～

代碼基於 ariesjzj 的 《Android中的so注入(inject)和掛鈎(hook) - For both x86 and arm 》，增加了對arm64的支援。（目前已經開始有64位安卓手機）

同樣，代碼由jni中以下三個檔案構成。

    inject.c
    Android.mk
    Application.mk

其中Android.mk一樣，Application.mk中

APP_ABI := arm64-v8a

inject.c如下：

#include <stdio.h>    #include <stdlib.h>    #include <asm/user.h>    #include <asm/ptrace.h>    #include <sys/ptrace.h>    #include <sys/wait.h>    #include <sys/mman.h>    #include <dlfcn.h>    #include <dirent.h>    #include <unistd.h>    #include <string.h>    #include <elf.h>    #include <android/log.h>    #include <sys/uio.h>    #if defined(__i386__)    #define pt_regs         user_regs_struct    #elif defined(__aarch64__)#define pt_regs         user_pt_regs  #define uregsregs#define ARM_pcpc#define ARM_spsp#define ARM_cpsrpstate#define ARM_lrregs[30]#define ARM_r0regs[0]  #define PTRACE_GETREGS PTRACE_GETREGSET#define PTRACE_SETREGS PTRACE_SETREGSET#endif        #define ENABLE_DEBUG 1        #if ENABLE_DEBUG    #define  LOG_TAG "INJECT"    #define  LOGD(fmt, args...)  __android_log_print(ANDROID_LOG_DEBUG,LOG_TAG, fmt, ##args)    #define DEBUG_PRINT(format,args...) \        LOGD(format, ##args)    #else    #define DEBUG_PRINT(format,args...)    #endif        #define CPSR_T_MASK     ( 1u << 5 )        #if defined(__aarch64__)    const char *libc_path = "/system/lib64/libc.so";    const char *linker_path = "/system/bin/linker64";    #elseconst char *libc_path = "/system/lib/libc.so";    const char *linker_path = "/system/bin/linker";    #endif    int ptrace_readdata(pid_t pid,  uint8_t *src, uint8_t *buf, size_t size)    {        long i, j, remain;        uint8_t *laddr;           size_t bytes_width = sizeof(long);    union u {            long val;            char chars[bytes_width];        } d;            j = size / bytes_width;        remain = size % bytes_width;            laddr = buf;            for (i = 0; i < j; i ++) {            d.val = ptrace(PTRACE_PEEKTEXT, pid, src, 0);            memcpy(laddr, d.chars, bytes_width);            src += bytes_width;            laddr += bytes_width;        }            if (remain > 0) {            d.val = ptrace(PTRACE_PEEKTEXT, pid, src, 0);            memcpy(laddr, d.chars, remain);        }            return 0;    }        int ptrace_writedata(pid_t pid, uint8_t *dest, uint8_t *data, size_t size)    {        long i, j, remain;        uint8_t *laddr;        size_t bytes_width = sizeof(long);    union u {            long val;            char chars[bytes_width];        } d;            j = size / bytes_width;        remain = size % bytes_width;            laddr = data;            for (i = 0; i < j; i ++) {            memcpy(d.chars, laddr, bytes_width);            ptrace(PTRACE_POKETEXT, pid, dest, d.val);                dest  += bytes_width;            laddr += bytes_width;        }            if (remain > 0) {            d.val = ptrace(PTRACE_PEEKTEXT, pid, dest, 0);            for (i = 0; i < remain; i ++) {                d.chars[i] = *laddr ++;            }                ptrace(PTRACE_POKETEXT, pid, dest, d.val);       }            return 0;    }        #if defined(__arm__) || defined(__aarch64__)int ptrace_call(pid_t pid, uintptr_t addr, long *params, int num_params, struct pt_regs* regs)    {        int i;   #if defined(__arm__)     int num_param_registers = 4;#elif defined(__aarch64__)     int num_param_registers = 8;#endif    for (i = 0; i < num_params && i < num_param_registers; i ++) {            regs->uregs[i] = params[i];        }            //        // push remained params onto stack        //        if (i < num_params) {            regs->ARM_sp -= (num_params - i) * sizeof(long) ;            ptrace_writedata(pid, (void *)regs->ARM_sp, (uint8_t *)¶ms[i], (num_params - i) * sizeof(long));        }            regs->ARM_pc = addr;        if (regs->ARM_pc & 1) {            /* thumb */            regs->ARM_pc &= (~1u);            regs->ARM_cpsr |= CPSR_T_MASK;        } else {            /* arm */            regs->ARM_cpsr &= ~CPSR_T_MASK;        }            regs->ARM_lr = 0;                if (ptrace_setregs(pid, regs) == -1                 || ptrace_continue(pid) == -1) {            printf("error\n");            return -1;        }            int stat = 0;      waitpid(pid, &stat, WUNTRACED);      while (stat != 0xb7f) {          if (ptrace_continue(pid) == -1) {              printf("error\n");              return -1;          }          waitpid(pid, &stat, WUNTRACED);      }          return 0;    }    #elif defined(__i386__)    long ptrace_call(pid_t pid, uintptr_t addr, long *params, int num_params, struct user_regs_struct * regs)    {        regs->esp -= (num_params) * sizeof(long) ;        ptrace_writedata(pid, (void *)regs->esp, (uint8_t *)params, (num_params) * sizeof(long));            long tmp_addr = 0x00;        regs->esp -= sizeof(long);        ptrace_writedata(pid, regs->esp, (char *)&tmp_addr, sizeof(tmp_addr));             regs->eip = addr;            if (ptrace_setregs(pid, regs) == -1                 || ptrace_continue( pid) == -1) {            printf("error\n");            return -1;        }            int stat = 0;      waitpid(pid, &stat, WUNTRACED);      while (stat != 0xb7f) {          if (ptrace_continue(pid) == -1) {              printf("error\n");              return -1;          }          waitpid(pid, &stat, WUNTRACED);      }          return 0;    }    #else     #error "Not supported"    #endif        int ptrace_getregs(pid_t pid, struct pt_regs * regs)    {    #if defined (__aarch64__)int regset = NT_PRSTATUS;struct iovec ioVec;ioVec.iov_base = regs;ioVec.iov_len = sizeof(*regs);    if (ptrace(PTRACE_GETREGSET, pid, (void*)regset, &ioVec) < 0) {            perror("ptrace_getregs: Can not get register values");           printf(" io %llx, %d", ioVec.iov_base, ioVec.iov_len);         return -1;        }            return 0;   #else    if (ptrace(PTRACE_GETREGS, pid, NULL, regs) < 0) {            perror("ptrace_getregs: Can not get register values");            return -1;        }            return 0;   #endif     }        int ptrace_setregs(pid_t pid, struct pt_regs * regs)    {     #if defined (__aarch64__)int regset = NT_PRSTATUS;struct iovec ioVec;ioVec.iov_base = regs;ioVec.iov_len = sizeof(*regs);    if (ptrace(PTRACE_SETREGSET, pid, (void*)regset, &ioVec) < 0) {            perror("ptrace_setregs: Can not get register values");            return -1;        }            return 0;   #else    if (ptrace(PTRACE_SETREGS, pid, NULL, regs) < 0) {            perror("ptrace_setregs: Can not set register values");            return -1;        }            return 0;   #endif     }        int ptrace_continue(pid_t pid)    {        if (ptrace(PTRACE_CONT, pid, NULL, 0) < 0) {            perror("ptrace_cont");            return -1;        }            return 0;    }        int ptrace_attach(pid_t pid)    {        if (ptrace(PTRACE_ATTACH, pid, NULL, 0) < 0) {            perror("ptrace_attach");            return -1;        }            int status = 0;        waitpid(pid, &status , WUNTRACED);            return 0;    }        int ptrace_detach(pid_t pid)    {        if (ptrace(PTRACE_DETACH, pid, NULL, 0) < 0) {            perror("ptrace_detach");            return -1;        }            return 0;    }        void* get_module_base(pid_t pid, const char* module_name)    {        FILE *fp;        long addr = 0;        char *pch;        char filename[32];        char line[1024];            if (pid < 0) {            /* self process */            snprintf(filename, sizeof(filename), "/proc/self/maps", pid);        } else {            snprintf(filename, sizeof(filename), "/proc/%d/maps", pid);        }            fp = fopen(filename, "r");            if (fp != NULL) {            while (fgets(line, sizeof(line), fp)) {                if (strstr(line, module_name)) {                    pch = strtok( line, "-" );                    addr = strtoull( pch, NULL, 16 );                       if (addr == 0x8000)                        addr = 0;                        break;                }            }                fclose(fp) ;        }            return (void *)addr;    }        void* get_remote_addr(pid_t target_pid, const char* module_name, void* local_addr)    {        void* local_handle, *remote_handle;            local_handle = get_module_base(-1, module_name);        remote_handle = get_module_base(target_pid, module_name);            DEBUG_PRINT("[+] get_remote_addr: local[%llx], remote[%llx]\n", local_handle, remote_handle);            void * ret_addr = (void *)((uintptr_t)local_addr + (uintptr_t)remote_handle - (uintptr_t)local_handle);        #if defined(__i386__)        if (!strcmp(module_name, libc_path)) {            ret_addr += 2;        }    #endif        return ret_addr;    }        int find_pid_of(const char *process_name)    {        int id;        pid_t pid = -1;        DIR* dir;        FILE *fp;        char filename[32];        char cmdline[256];            struct dirent * entry;            if (process_name == NULL)            return -1;            dir = opendir("/proc");        if (dir == NULL)            return -1;            while((entry = readdir(dir)) != NULL) {            id = atoi(entry->d_name);            if (id != 0) {                sprintf(filename, "/proc/%d/cmdline", id);                fp = fopen(filename, "r");                if (fp) {                    fgets(cmdline, sizeof(cmdline), fp);                    fclose(fp);                        if (strcmp(process_name, cmdline) == 0) {                        /* process found */                        pid = id;                        break;                    }                }            }        }            closedir(dir);        return pid;    }        uint64_t ptrace_retval(struct pt_regs * regs)    {    #if defined(__arm__) || defined(__aarch64__)    return regs->ARM_r0;    #elif defined(__i386__)        return regs->eax;    #else    #error "Not supported"    #endif    }        uint64_t ptrace_ip(struct pt_regs * regs)    {    #if defined(__arm__) || defined(__aarch64__)     return regs->ARM_pc;   #elif defined(__i386__)        return regs->eip;    #else    #error "Not supported"    #endif    }        int ptrace_call_wrapper(pid_t target_pid, const char * func_name, void * func_addr, long * parameters, int param_num, struct pt_regs * regs)     {        DEBUG_PRINT("[+] Calling %s in target process.\n", func_name);        if (ptrace_call(target_pid, (uintptr_t)func_addr, parameters, param_num, regs) == -1)            return -1;            if (ptrace_getregs(target_pid, regs) == -1)            return -1;        DEBUG_PRINT("[+] Target process returned from %s, return value=%llx, pc=%llx \n",                 func_name, ptrace_retval(regs), ptrace_ip(regs));        return 0;    }        int inject_remote_process(pid_t target_pid, const char *library_path, const char *function_name, const char *param, size_t param_size)    {        int ret = -1;        void *mmap_addr, *dlopen_addr, *dlsym_addr, *dlclose_addr, *dlerror_addr;        void *local_handle, *remote_handle, *dlhandle;        uint8_t *map_base = 0;        uint8_t *dlopen_param1_ptr, *dlsym_param2_ptr, *saved_r0_pc_ptr, *inject_param_ptr, *remote_code_ptr, *local_code_ptr;            struct pt_regs regs, original_regs;         long parameters[10];            DEBUG_PRINT("[+] Injecting process: %d\n", target_pid);            if (ptrace_attach(target_pid) == -1)            goto exit;            if (ptrace_getregs(target_pid, &regs) == -1)            goto exit;            /* save original registers */        memcpy(&original_regs, &regs, sizeof(regs));            mmap_addr = get_remote_addr(target_pid, libc_path, (void *)mmap);        DEBUG_PRINT("[+] Remote mmap address: %llx\n", mmap_addr);            /* call mmap */        parameters[0] = 0;  // addr        parameters[1] = 0x4000; // size        parameters[2] = PROT_READ | PROT_WRITE | PROT_EXEC;  // prot        parameters[3] = MAP_ANONYMOUS | MAP_PRIVATE; // flags        parameters[4] = 0; //fd        parameters[5] = 0; //offset            if (ptrace_call_wrapper(target_pid, "mmap", mmap_addr, parameters, 6, &regs) == -1)            goto exit;            map_base = ptrace_retval(&regs);          dlopen_addr = get_remote_addr( target_pid, linker_path, (void *)dlopen );        dlsym_addr = get_remote_addr( target_pid, linker_path, (void *)dlsym );        dlclose_addr = get_remote_addr( target_pid, linker_path, (void *)dlclose );        dlerror_addr = get_remote_addr( target_pid, linker_path, (void *)dlerror );            DEBUG_PRINT("[+] Get imports: dlopen: %llx, dlsym: %llx, dlclose: %llx, dlerror: %llx\n",                dlopen_addr, dlsym_addr, dlclose_addr, dlerror_addr);            printf("library path = %s\n", library_path);        ptrace_writedata(target_pid, map_base, library_path, strlen(library_path) + 1);              parameters[0] = map_base;           parameters[1] = RTLD_NOW| RTLD_GLOBAL;             if (ptrace_call_wrapper(target_pid, "dlopen", dlopen_addr, parameters, 2, &regs) == -1)            goto exit;            void * sohandle = ptrace_retval(&regs);        if(!sohandle) {    if (ptrace_call_wrapper(target_pid, "dlerror", dlerror_addr, 0, 0, &regs) == -1)            goto exit;                uint8_t *errret = ptrace_retval(&regs);      uint8_t errbuf[100];    ptrace_readdata(target_pid, errret, errbuf, 100);  }        #define FUNCTION_NAME_ADDR_OFFSET       0x100        ptrace_writedata(target_pid, map_base + FUNCTION_NAME_ADDR_OFFSET, function_name, strlen(function_name) + 1);        parameters[0] = sohandle;           parameters[1] = map_base + FUNCTION_NAME_ADDR_OFFSET;             if (ptrace_call_wrapper(target_pid, "dlsym", dlsym_addr, parameters, 2, &regs) == -1)            goto exit;            void * hook_entry_addr = ptrace_retval(&regs);        DEBUG_PRINT("hook_entry_addr = %p\n", hook_entry_addr);        #define FUNCTION_PARAM_ADDR_OFFSET      0x200        ptrace_writedata(target_pid, map_base + FUNCTION_PARAM_ADDR_OFFSET, param, strlen(param) + 1);        parameters[0] = map_base + FUNCTION_PARAM_ADDR_OFFSET;            if (ptrace_call_wrapper(target_pid, "hook_entry", hook_entry_addr, parameters, 1, &regs) == -1)            goto exit;                printf("Press enter to dlclose and detach\n");        getchar();        parameters[0] = sohandle;               if (ptrace_call_wrapper(target_pid, "dlclose", dlclose, parameters, 1, &regs) == -1)            goto exit;            /* restore */        ptrace_setregs(target_pid, &original_regs);        ptrace_detach(target_pid);        ret = 0;        exit:        return ret;    }        int main(int argc, char** argv) {        pid_t target_pid;        target_pid = find_pid_of("system_server");    if (-1 == target_pid) {          printf("Can't find the process\n");          return -1;      }      //target_pid = find_pid_of("/data/test");        inject_remote_process(target_pid, "/data/libhello.so", "hook_entry",  "I'm parameter!", strlen("I'm parameter!"));        return 0;  }    

另外，也用到libhello.so，代碼基本不變，只是其中Application.mk中也指定產生arm64。

APP_ABI := arm64-v8a

最後要注意一點，目前arm64的手機應該都是Android 5.0+的，可能會因selinux的許可權控製造成注入失敗，比如例子中的system_server會失敗，但注入calendar沒問題。

此時可以暫時關閉掉selinux（執行setenforce 0），要針對解決就需要配置selinux了。

Android arm64(aarch64)中的so注入(inject) - 相容x86 and arm