linux 下hook函数

hook在windows下可以说是知名度相当高的一种"高级“技术
想在linux下面实现像windows下的那种hook的功能，不过网上的资料很少（LD_PRELOAD 也可以做类似的事）
自己研究了下，写了个类似功能的函数
思想很简单，就在运行时把一个函数的入口改成jmp到另一个地址

点击(此处)折叠或打开

1. #include <iostream>
   
2. #include <stdlib.h>
   
3. #include <stdio.h>
   
4. #include <stdint.h>
   
5. #include <string.h>
   
6. #include <sys/mman.h>
   
7. #include <errno.h>
   
8. #include <unistd.h>
   
9. 
   
10. void set_hook(void *to_mock_func, void *mock_func)
    
11. {
    
12.     uint8_t machine_code[] = {
    
13.             //movq $0x0, %rax 后面8个字节的0为64位立即数
    
14.             0x48, 0xb8, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
    
15.             //jmpq *%rax
    
16.             0xff, 0xe0
    
17.         };
    
18. 
    
19.     int pagesize = sysconf(_SC_PAGE_SIZE);
    
20.     if (pagesize == -1)
    
21.     {
    
22.         exit(errno);
    
23.     }
    
24. 
    
25.     uint8_t *mem = (uint8_t *) to_mock_func;
    
26.     void *p = (uint8_t*) (mem - ((uint64_t) mem % pagesize));
    
27.     if (mprotect(p, pagesize, PROT_READ | PROT_WRITE | PROT_EXEC))
    
28.     {
    
29.         perror("mprotect error");
    
30.         exit(errno);
    
31.     }
    
32. 
    
33.     //改写立即数为mock的地址,写入函数入口处
    
34.     memcpy(machine_code + 2, &mock_func, sizeof(mock_func));
    
35.     memcpy(mem, machine_code, sizeof(machine_code));
    
36. 
    
37.     if (mprotect(p, pagesize, PROT_EXEC))
    
38.     {
    
39.         perror("mprotect error");
    
40.         exit(errno);
    
41.     }
    
42. }
    
43. 
    
44. using namespace std;
    
45. class Cal
    
46. {
    
47. public:
    
48.     int Sum(int a, int b, long c, long d, long e)
    
49.     {
    
50.         std::cout << "Cal::Sum called... " << std::endl;
    
51.         return 0;
    
52.     }
    
53.     int m;
    
54. };
    
55. 
    
56. int MockCalSum(Cal *p, int a, int b, long c, long d, long e)
    
57. {
    
58.     int x = 0;
    
59.     x += a + b + c + d;
    
60.     std::cout << "MockCalSum called... " << std::endl;
    
61.     p->m = x;
    
62.     return x;
    
63. }
    
64. 
    
65. int main()
    
66. {
    
67.     int a = 0;
    
68.     int b = 1;
    
69.     long c = 2;
    
70.     long d = 3;
    
71.     long e = 4;
    
72.     Cal cal;
    
73.     cal.Sum(a, b, c, d, e);
    
74.     MockCalSum(&cal, a, b, c, d, e);
    
75.     set_hook((void*)&Cal::Sum, (void*)MockCalSum);
    
76.     cal.Sum(a, b, c, d, e);
    
77.     return 0;
    
78. }