如何使用hook?

水番茄

已于 2023-05-21 17:53:19 修改

阅读量497

点赞数

分类专栏：服务器笔记文章标签： mysql 数据库 hook

于 2023-05-05 13:06:44 首次发布

本文链接：https://blog.csdn.net/qq_51011672/article/details/130504604

版权

笔记同时被 2 个专栏收录

71 篇文章 1 订阅

订阅专栏

服务器

6 篇文章 0 订阅

订阅专栏

文章介绍了如何使用hook技术捕获并修改POSIXAPI的行为，以监控MySQL连接过程。通过定义函数指针，修改原始函数，然后使用dlsym函数动态获取并替换原函数，实现了在调用connect、recv和send等函数时打印信息，无需改动主程序代码。

摘要由CSDN通过智能技术生成

目标：将`posix`函数`hook`住

一个简单的例子

（连接mysql服务），连接成功则打印success

`mysql.c`

#include <mysql/mysql.h>
#include <stdio.h>
int main(){
    
    MYSQL* mysql = mysql_init(NULL);
    if(!mysql){
        printf("mysql_init failed\n");
        return 0;
    }
    
    if(!mysql_real_connect(mysql, "127.0.0.1", "root", "123456","mysql", 3306, NULL, 0)){
        printf("mysql connect failed\n");
        return 0;
    }
    printf("success!\n");
}

编译

gcc -o mysql mysql.c -lmysqlclient

运行

$ ./mysql
success!

成功连接到mysql服务

mysql服务的连接少不了posix API例如connect，recv，send，其中mysql_real_connect函数中一定调用了这些函数，我们就可以使用hook技术来将其hook住捕获原始posixAPI

目标：将posixAPI捕获然后在其调用之前打印一定信息

main函数的主要代码完全不用修改，底层hook了后用户是无法感知的

操作步骤

一、定义要`hook`的`posixAPI`的函数指针

typedef int (*connect_t)(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
typedef ssize_t(*recv_t)(int sockfd, void *buf, size_t len, int flags);
typedef ssize_t(*send_t)(int sockfd, const void *buf, size_t len, int flags);

connect_t connect_f = NULL;
recv_t recv_f = NULL;
send_t send_f = NULL;

返回值和参数必须与原posixAPI相同

二、修改`posixAPI`

int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen){
    printf("CONNECT\n");
    return connect_f(sockfd, addr, addrlen);
}
ssize_t recv(int sockfd, void *buf, size_t len, int flags){
    printf("RECV\n");
    return recv_f(sockfd, buf, len, flags);
}
ssize_t send(int sockfd, const void *buf, size_t len, int flags){
    printf("SEND\n");
    return send_f(sockfd, buf, len, flags);
}

让posixAPI被触发之前能够触发打印

三、添加相关头文件并初始化`hook`

#define _GNU_SOURCE
#include <dlfcn.h>
#include <sys/types.h>    
#include <sys/socket.h>

// ...

void init_hook(void){
    if(!connect_f)
        connect_f = dlsym(RTLD_NEXT, "connect");
    if(!recv_f)
        recv_f = dlsym(RTLD_NEXT, "recv");
    if(!send_f)
        send_f = dlsym(RTLD_NEXT, "send");
}

在上述mysql.c中添加上述代码，并在程序开始前调用init_hook函数

#define _GNU_SOURCE
#include <dlfcn.h>
#include <sys/types.h>    
#include <sys/socket.h>
#include <mysql/mysql.h>
#include <stdio.h>

typedef int (*connect_t)(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
typedef ssize_t(*recv_t)(int sockfd, void *buf, size_t len, int flags);
typedef ssize_t(*send_t)(int sockfd, const void *buf, size_t len, int flags);

connect_t connect_f = NULL;
recv_t recv_f = NULL;
send_t send_f = NULL;

int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen){
    printf("CONNECT\n");
    return connect_f(sockfd, addr, addrlen);
}
ssize_t recv(int sockfd, void *buf, size_t len, int flags){
    printf("RECV\n");
    return recv_f(sockfd, buf, len, flags);
}
ssize_t send(int sockfd, const void *buf, size_t len, int flags){
    printf("SEND\n");
    return send_f(sockfd, buf, len, flags);
}

void init_hook(void){
    if(!connect_f)
        connect_f = dlsym(RTLD_NEXT, "connect");
    if(!recv_f)
        recv_f = dlsym(RTLD_NEXT, "recv");
    if(!send_f)
        send_f = dlsym(RTLD_NEXT, "send");
}


int main(){
    init_hook();
    MYSQL* mysql = mysql_init(NULL);
    if(!mysql){
        printf("mysql_init failed\n");
        return 0;
    }
    if(!mysql_real_connect(mysql, "127.0.0.1", "root", "123456","mysql", 3306, NULL, 0)){
        printf("mysql connect failed\n");
        return 0;
    }
    printf("success!\n");
}