最近终于沉下心来对着书把hook跟注入方面的代码敲了一遍，打算写几个博客把它们记录下来。
第一次介绍一下我感觉难度最大的inline hook，实现代码参考了腾讯GAD的游戏安全入门。
inline hook的大致流程如下：
首先将目标指令替换为跳转指令，跳转地址为一段我们自己编写的汇编代码，这段汇编代码先是执行用户指定的代码，如修改寄存器的值，然后执行被替换掉的原指令2，最后再跳转回原指令3处，恢复程序的正常运行。
为了避开注入过程，我们通过hook自己进程加载的动态连接库进行演示。
1、实现目标注入程序
我们将这个程序编译为动态连接库，然后在主程序中加载，作为hook的目标。
target.h
#ifndef TARGET_H_INCLUDED
#define TARGET_H_INCLUDED void target_foo(); #endif // TARGET_H_INCLUDED
target.c
#include "target.h"
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h> void target_foo()
{
int a = ;
int b = ;
while(a--) {
sleep();
b = a * b;
printf("[INFO] b is %d\n", b);
}
b = b + ;
b = b - ;
printf("[INFO] finally, b is %d\n", b);
}
Android.mk
include $(CLEAR_VARS)
LOCAL_ARM_MODE := arm
LOCAL_MODULE := target
LOCAL_CFLAGS += -pie -fPIE -std=c11
LOCAL_LDFLAGS += -pie -fPIE -shared -llog
APP_ABI := armeabi-v7a
LOCAL_SRC_FILES := target.c
include $(BUILD_SHARED_LIBRARY)
注意Android.mk中LOCAL_ARM_MODE := arm代表编译时使用4字节的arm指令集，而不是2字节的thumb指令集。
2、实现主程序
在主程序中我们首先加载之前编写的动态链接库，进行hook之后再对其中的函数target_foo进行调用。
main.c
#include <stdio.h>
#include <stdlib.h>
#include <dlfcn.h>
#include <unistd.h>
#include <stdbool.h>
#include "hook_inline.h" typedef void (*target_foo)(void);
void my_func(struct hook_reg *reg)
{
puts("here we go!");
}
void main()
{
void *handler = dlopen("/data/local/tmp/libtarget.so", RTLD_NOW);
target_foo foo = (target_foo)dlsym(handler, "target_foo");
hook_inline_make("/data/local/tmp/libtarget.so", 0xde2, my_func, true);
foo();
}
hook_inline.h
#ifndef HOOK_INLINE_H_INCLUDED
#define HOOK_INLINE_H_INCLUDED
#include <stdbool.h>
struct hook_reg {
long ARM_r0; long ARM_r1; long ARM_r2; long ARM_r3;
long ARM_r4; long ARM_r5; long ARM_r6; long ARM_r7;
long ARM_r8; long ARM_r9; long ARM_r10;long ARM_r11;
long ARM_r12;long ARM_sp; long ARM_lr; long ARM_cpsr;
};
typedef void (*hook_func)(struct hook_reg *reg);
bool hook_inline_make(const char *library, long address, hook_func func, bool isArm);
#endif // HOOK_INLINE_H_INCLUDED
这里我们hook功能的实现函数为hook_inline_make，4个参数分别为动态库路径，目标地址，用户函数，目标地址处指令集。
当程序执行到目标地址处时会回调我们传入的用户函数，可通过参数hook_reg来更改寄存器的值（不包括寄存器pc）。因为之前在动态链接库的Android.mk文件指定了使用arm指令集进行编译，所以此处指定最后一个参数为true。
3、实现注入函数
现在到了最为关键的地方，为了实现这个功能还需要了解几个知识。
（1）、获取内存中动态链接库的基址
Linux系统中各个进程的内存加载信息可以在/proc/pid/maps文件中到，通过它我们可以获取到动态链接库在内存中的加载基址。
long get_module_addr(pid_t pid, const char *module_name)
{
char file_path[];
char file_line[];
if (pid < ) {
snprintf(file_path, sizeof(file_path), "/proc/self/maps");
} else {
snprintf(file_path, sizeof(file_path), "/proc/%d/maps", pid);
}
FILE *fp = fopen(file_path, "r");
if (fp == NULL) {
return -;
}
long addr_start = -, addr_end = ;
while (fgets(file_line, sizeof(file_line), fp)) {
if (strstr(file_line, module_name)) {
if ( == sscanf(file_line, "%8lx-%8lx", &addr_start, &addr_end)) {
break;
}
}
}
fclose(fp);
printf("library :%s %lx-%lx, pid : %d\n", module_name, addr_start, addr_end, pid);
return addr_start;
}
（2）、更改内存中的二进制代码
现在的计算机系统中一般对内存进行分段式管理，不同的段有不同的读、写、执行的属性。一般来讲代码段只有读和执行的属性，不允许对代码段进行写操作。Linux系统中通过函数mprotect对内存的属性进行更改，需要注意的一点是需要以内存页的大小进行对齐。
bool change_addr_writable(long address, bool writable) {
long page_size = sysconf(_SC_PAGESIZE);
//align address by page size
long page_start = (address) & (~(page_size - ));
//change memory attribute
if (writable == true) {
return mprotect((void*)page_start, page_size, PROT_READ | PROT_WRITE | PROT_EXEC) != -;
} else {
return mprotect((void*)page_start, page_size, PROT_READ | PROT_EXEC) != -;
}
}
接下来就可以着手实现功能了，inline hook跟指令集密切相关，此处我们先演示arm指令集的情况，之后对thumb指令集进行讨论。这里实现的功能是用户可在自己注册的回调函数中对hook点寄存器的值进行修改。
为了实现32位地址空间的长跳转，我们需要两条指令的长度（8个字节）来实现。一般手机上的arm处理器为3级流水，所以pc寄存器的值总是指向当前执行指令后的第二条指令，因而使用ldr pc, [pc, #-4]来加载该指令之后的跳转地址。当程序跳转到shellcode后，首先对寄存器组进行备份，然后调用用户注册的回调函数，用户可在回调函数中修改备份中各个寄存器（pc寄存器除外）的值，然后从备份中恢复寄存器组再跳转到stubcode，stubcode的功能是执行被hook点的跳转指令替换掉的两条指令，最后跳回原程序。
shellcode.S
1 .global _shellcode_start_s
.global _shellcode_end_s
.global _hook_func_addr_s
.global _stub_func_addr_s
.data
_shellcode_start_s:
@ 备份各个寄存器
push {r0, r1, r2, r3}
mrs r0, cpsr
str r0, [sp, #0xc]
str r14, [sp, #0x8]
add r14, sp, #0x10
str r14, [sp, #0x4]
pop {r0}
push {r0-r12}
@ 此时寄存器被备份在栈中，将栈顶地址作为回调函数的参数（struct hook_reg）
mov r0, sp
ldr r3, _hook_func_addr_s
blx r3
@ 恢复寄存器值
ldr r0, [sp, #0x3c]
msr cpsr, r0
ldmfd sp!, {r0-r12}
ldr r14, [sp, #0x4]
ldr sp, [r13]
ldr pc, _stub_func_addr_s
_hook_func_addr_s:
.word 0x0
_stub_func_addr_s:
.word 0x0
_shellcode_end_s:
.end
shellcode使用汇编实现，在使用时需要对里边的两个地址进行修复，用户回调函数地址（_hook_func_addr_s）跟stubcode地址（_stub_func_addr_s）。
接下来我们可以看一下函数hook_inline_make的具体实现了
void hook_inline_make(const char *library, long address, hook_func func)
{
//获取hook点在内存中的地址
long base_addr = get_module_addr(-, library);
long hook_addr = base_addr + address;
//获取shellcode中的符号地址
extern long _shellcode_start_s;
extern long _shellcode_end_s;
extern long _hook_func_addr_s;
extern long _stub_func_addr_s;
void *p_shellcode_start = &_shellcode_start_s;
void *p_shellcdoe_end = &_shellcode_end_s;
void *p_hook_func = &_hook_func_addr_s;
void *p_stub_func = &_stub_func_addr_s;
//计算shellcode大小
int shellcode_size = (int)(p_shellcdoe_end - p_shellcode_start);
//新建shellcode
void *shellcode = malloc(shellcode_size);
memcpy(shellcode, p_shellcode_start, shellcode_size);
//添加执行属性
change_addr_writable((long)shellcode, true);
//在32bit的arm指令集中，stubcode中的4条指令占用16个字节的空间
//前两条指令为hook点被替换的两条指令
//后两条指令跳转回原程序
void *stubcode = malloc();
memcpy(stubcode, (void*)hook_addr, );
//ldr pc, [pc, #-4]
//[address]
//手动填充stubcode
char jump_ins[] = {0x04, 0xF0, 0x1F, 0xE5};
uint32_t jmp_address = hook_addr + ;
memcpy(jump_ins + , &jmp_address, );
memcpy(stubcode + , jump_ins, );
//添加执行属性
change_addr_writable((long)stubcode, true);
//修复shellcode中的两个地址值
uint32_t *shell_hook = shellcode + (p_hook_func - p_shellcode_start);
*shell_hook = (uint32_t)func;
uint32_t *shell_stub = shellcode + (p_stub_func - p_shellcode_start);
*shell_stub = (uint32_t)stubcode;
//为hook点添加写属性
change_addr_writable(hook_addr, true);
//替换hook点指令为跳转指令，跳转至shellcode
jmp_address = (uint32_t)shellcode;
memcpy(jump_ins + , &jmp_address, );
memcpy((void*)hook_addr, jump_ins, );
change_addr_writable(hook_addr, false);
//刷新cache
cacheflush(hook_addr, , );
}
注意这里的change_addr_writable函数无论传入false还是true对应地址都会添加上执行属性。由于处理器采用流水线跟多级缓存，在更改代码后我们需要手动刷新cache，即函数cacheflush(第三个参数无意义)。
4、thumb指令集实现
由于thumb指令集的功能受到限制，虽然思路上跟arm指令集一致，但在实现上需要用更多条指令，下面是我自己想的一种实现方式，欢迎交流。
需要注意的是由于每条thumb指令为16bit，所以32位的跳转地址需要占用两条指令的空间，而且跳转时会污染r0寄存器所以要对其进行保护。我在实现程序时将shellcode编译为了arm指令集，所以在原程序、shellcode、stubcode之间相互跳转时需要使用bx指令进行处理器状态切换（需要跳转的地址代码为thumb指令集时，需要将地址的第1个bit位置位）。