113 lines
4.2 KiB
C
113 lines
4.2 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
|
|
*/
|
|
|
|
#ifndef _MT_SYS_HOOK___
|
|
#define _MT_SYS_HOOK___
|
|
|
|
#include <poll.h>
|
|
#include <dlfcn.h>
|
|
|
|
#include "ff_api.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
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);
|
|
|
|
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;
|
|
func_select real_select;
|
|
func_poll real_poll;
|
|
|
|
func_accept real_accept;
|
|
} MtSyscallFuncTab;
|
|
|
|
extern MtSyscallFuncTab g_mt_syscall_tab;
|
|
extern int g_mt_hook_flag;
|
|
extern int g_ff_hook_flag;
|
|
|
|
#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
|
|
|
|
|