mirror of https://github.com/F-Stack/f-stack.git
131 lines
5.3 KiB
C
131 lines
5.3 KiB
C
|
|
|||
|
|
/**
|
|||
|
|
* Tencent is pleased to support the open source community by making MSEC available.
|
|||
|
|
*
|
|||
|
|
* Copyright (C) 2016 THL A29 Limited, a Tencent company. All rights reserved.
|
|||
|
|
*
|
|||
|
|
* Licensed under the GNU General Public License, Version 2.0 (the "License");
|
|||
|
|
* you may not use this file except in compliance with the License. You may
|
|||
|
|
* obtain a copy of the License at
|
|||
|
|
*
|
|||
|
|
* https://opensource.org/licenses/GPL-2.0
|
|||
|
|
*
|
|||
|
|
* Unless required by applicable law or agreed to in writing, software distributed under the
|
|||
|
|
* License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
|
|||
|
|
* either express or implied. See the License for the specific language governing permissions
|
|||
|
|
* and limitations under the License.
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @filename mt_sys_hook.h
|
|||
|
|
* @info <EFBFBD>߳<EFBFBD>hookϵͳapi, <EFBFBD>Բ<EFBFBD><EFBFBD>ö<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>, תͬ<EFBFBD><EFBFBD>Ϊ<EFBFBD>첽<EFBFBD><EFBFBD>
|
|||
|
|
* HOOK <EFBFBD><EFBFBD><EFBFBD><EFBFBD>, <EFBFBD>ο<EFBFBD>pth<EFBFBD><EFBFBD>libcoʵ<EFBFBD><EFBFBD>
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
#ifndef _MT_SYS_HOOK___
|
|||
|
|
#define _MT_SYS_HOOK___
|
|||
|
|
|
|||
|
|
#include <poll.h>
|
|||
|
|
#include <dlfcn.h>
|
|||
|
|
|
|||
|
|
#include "ff_api.h"
|
|||
|
|
|
|||
|
|
#ifdef __cplusplus
|
|||
|
|
extern "C" {
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
/******************************************************************************/
|
|||
|
|
/* 1. HOOK <20>ĺ<EFBFBD><C4BA><EFBFBD><EFBFBD><EFBFBD><EFBFBD>岿<EFBFBD><E5B2BF> */
|
|||
|
|
/******************************************************************************/
|
|||
|
|
|
|||
|
|
typedef int (*func_socket)(int domain, int type, int protocol);
|
|||
|
|
typedef int (*func_bind)(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
|
|||
|
|
typedef int (*func_listen)(int sockfd, int backlog);
|
|||
|
|
typedef int (*func_close)(int fd);
|
|||
|
|
typedef int (*func_connect)(int socket, const struct sockaddr *address, socklen_t address_len);
|
|||
|
|
typedef int (*func_accept)(int socket, struct sockaddr *address, socklen_t *addrlen);
|
|||
|
|
typedef ssize_t (*func_read)(int fildes, void *buf, size_t nbyte);
|
|||
|
|
typedef ssize_t (*func_write)(int fildes, const void *buf, size_t nbyte);
|
|||
|
|
typedef ssize_t (*func_sendto)(int socket, const void *message, size_t length,
|
|||
|
|
int flags, const struct sockaddr *dest_addr, socklen_t dest_len);
|
|||
|
|
typedef ssize_t (*func_recvfrom)(int socket, void *buffer, size_t length,
|
|||
|
|
int flags, struct sockaddr *address, socklen_t *address_len);
|
|||
|
|
typedef size_t (*func_send)(int socket, const void *buffer, size_t length, int flags);
|
|||
|
|
typedef ssize_t (*func_recv)(int socket, void *buffer, size_t length, int flags);
|
|||
|
|
typedef int (*func_select)(int nfds, fd_set *readfds, fd_set *writefds,
|
|||
|
|
fd_set *exceptfds, struct timeval *timeout);
|
|||
|
|
typedef int (*func_poll)(struct pollfd fds[], nfds_t nfds, int timeout);
|
|||
|
|
typedef int (*func_setsockopt)(int socket, int level, int option_name,
|
|||
|
|
const void *option_value, socklen_t option_len);
|
|||
|
|
typedef int (*func_ioctl)(int fd, unsigned long cmd, ...);
|
|||
|
|
typedef int (*func_fcntl)(int fd, int cmd, ...);
|
|||
|
|
|
|||
|
|
typedef unsigned int (*func_sleep)(unsigned int seconds);
|
|||
|
|
|
|||
|
|
|
|||
|
|
/******************************************************************************/
|
|||
|
|
/* 2. ȫ<>ֵ<EFBFBD>hook<6F><6B><EFBFBD><EFBFBD><EFBFBD>ṹ */
|
|||
|
|
/******************************************************************************/
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief Hook<EFBFBD><EFBFBD>ԭʼ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>й<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>, ֧<EFBFBD>ֶ<EFBFBD>̬<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
|
*/
|
|||
|
|
typedef struct mt_syscall_func_tab
|
|||
|
|
{
|
|||
|
|
func_socket real_socket;
|
|||
|
|
func_bind real_bind;
|
|||
|
|
func_listen real_listen;
|
|||
|
|
func_close real_close;
|
|||
|
|
func_connect real_connect;
|
|||
|
|
func_read real_read;
|
|||
|
|
func_write real_write;
|
|||
|
|
func_sendto real_sendto;
|
|||
|
|
func_recvfrom real_recvfrom;
|
|||
|
|
func_send real_send;
|
|||
|
|
func_recv real_recv;
|
|||
|
|
func_setsockopt real_setsockopt;
|
|||
|
|
func_fcntl real_fcntl;
|
|||
|
|
func_ioctl real_ioctl;
|
|||
|
|
|
|||
|
|
func_sleep real_sleep; // <20>ݲ<EFBFBD>֧<EFBFBD>֣<EFBFBD><D6A3><EFBFBD>Ϊû<CEAA><C3BB><EFBFBD><EFBFBD>fd<66><64><EFBFBD><EFBFBD>, <20><>ֹ<EFBFBD><D6B9><EFBFBD><EFBFBD>
|
|||
|
|
func_select real_select; // <20>ݲ<EFBFBD>֧<EFBFBD><D6A7>, 1024<32><34><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
|
func_poll real_poll; // <20>ݲ<EFBFBD>֧<EFBFBD><D6A7>, ȷ<><C8B7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʵʩ
|
|||
|
|
|
|||
|
|
func_accept real_accept;
|
|||
|
|
}MtSyscallFuncTab;
|
|||
|
|
|
|||
|
|
|
|||
|
|
/******************************************************************************/
|
|||
|
|
/* 3. ֱ<>ӵ<EFBFBD><D3B5><EFBFBD>ԭʼϵͳapi<70>Ľӿ<C4BD> */
|
|||
|
|
/******************************************************************************/
|
|||
|
|
extern MtSyscallFuncTab g_mt_syscall_tab; // ȫ<>ַ<EFBFBD><D6B7>ű<EFBFBD>
|
|||
|
|
extern int g_mt_hook_flag; // ȫ<>ֿ<EFBFBD><D6BF>Ʊ<EFBFBD><C6B1><EFBFBD>
|
|||
|
|
extern int g_ff_hook_flag; // ȫ<>ֿ<EFBFBD><D6BF>Ʊ<EFBFBD><C6B1><EFBFBD>
|
|||
|
|
|
|||
|
|
#define mt_hook_syscall(name) \
|
|||
|
|
do { \
|
|||
|
|
if (!g_mt_syscall_tab.real_##name) { \
|
|||
|
|
g_mt_syscall_tab.real_##name = (func_##name)dlsym(RTLD_NEXT, #name);\
|
|||
|
|
} \
|
|||
|
|
} while (0)
|
|||
|
|
|
|||
|
|
#define mt_real_func(name) g_mt_syscall_tab.real_##name
|
|||
|
|
|
|||
|
|
#define mt_set_hook_flag() (g_mt_hook_flag = 1)
|
|||
|
|
#define mt_unset_hook_flag() (g_mt_hook_flag = 0)
|
|||
|
|
|
|||
|
|
#define mt_hook_active() (g_mt_hook_flag == 1)
|
|||
|
|
|
|||
|
|
#define ff_set_hook_flag() (g_ff_hook_flag = 1)
|
|||
|
|
#define ff_unset_hook_flag() (g_ff_hook_flag = 0)
|
|||
|
|
#define ff_hook_active() (g_ff_hook_flag == 1)
|
|||
|
|
|
|||
|
|
#ifdef __cplusplus
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
|