2021-10-19 15:30:39 +08:00

112 lines
4.1 KiB
C

/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache 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
*
* http://www.apache.org/licenses/LICENSE-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.
*/
#ifndef LIBCSYMTABLE_H_
#define LIBCSYMTABLE_H_
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <stddef.h>
#include <dlfcn.h>
#include <cassert>
#include <poll.h>
#include <unistd.h> // needed for _POSIX_MONOTONIC_CLOCK
#ifdef THREADED
#include <pthread.h>
#endif
#include "config.h"
// TODO: move all these macros to config.h (generated by autoconf)
#ifdef __CYGWIN__
#if (CYGWIN_VERSION_DLL_MAJOR < 1007)
#define RANDOM_RET_TYPE int
#else
#define RANDOM_RET_TYPE long int
#endif
#define GETTIMEOFDAY_ARG2_TYPE void*
#else
#define RANDOM_RET_TYPE long int
#define GETTIMEOFDAY_ARG2_TYPE struct timezone*
#endif
#define DECLARE_SYM(ret,sym,sig) \
typedef ret (*sym##_sig)sig; \
static sym##_sig preload_##sym () { \
static sym##_sig ptr=0;\
if(!ptr){ void* h=getHandle(); ptr=(sym##_sig)dlsym(h,#sym); } \
assert("Unable to load "#sym" from libc"&&ptr); \
return ptr; \
} \
sym##_sig sym
#define LIBC_SYMBOLS LibCSymTable::instance()
//******************************************************************************
// preload original libc symbols
struct LibCSymTable
{
DECLARE_SYM(hostent*,gethostbyname,(const char*));
DECLARE_SYM(void*,calloc,(size_t, size_t));
DECLARE_SYM(void*,realloc,(void*, size_t));
DECLARE_SYM(void,free,(void*));
DECLARE_SYM(RANDOM_RET_TYPE,random,(void));
DECLARE_SYM(void,srandom,(unsigned long));
DECLARE_SYM(int,printf,(const char*, ...));
DECLARE_SYM(int,socket,(int,int,int));
DECLARE_SYM(int,close,(int));
DECLARE_SYM(int,getsockopt,(int,int,int,void*,socklen_t*));
DECLARE_SYM(int,setsockopt,(int,int,int,const void*,socklen_t));
DECLARE_SYM(int,fcntl,(int,int,...));
DECLARE_SYM(int,connect,(int,const struct sockaddr*,socklen_t));
DECLARE_SYM(ssize_t,send,(int,const void*,size_t,int));
DECLARE_SYM(ssize_t,recv,(int,const void*,size_t,int));
DECLARE_SYM(int,select,(int,fd_set*,fd_set*,fd_set*,struct timeval*));
DECLARE_SYM(int,poll,(struct pollfd*,POLL_NFDS_TYPE,int));
DECLARE_SYM(int,gettimeofday,(struct timeval*,GETTIMEOFDAY_ARG2_TYPE));
#ifdef _POSIX_MONOTONIC_CLOCK
DECLARE_SYM(int,clock_gettime,(clockid_t clk_id, struct timespec*));
#endif
#ifdef THREADED
DECLARE_SYM(int,pthread_create,(pthread_t *, const pthread_attr_t *,
void *(*)(void *), void *));
DECLARE_SYM(int,pthread_detach,(pthread_t));
DECLARE_SYM(int,pthread_cond_broadcast,(pthread_cond_t *));
DECLARE_SYM(int,pthread_cond_destroy,(pthread_cond_t *));
DECLARE_SYM(int,pthread_cond_init,(pthread_cond_t *, const pthread_condattr_t *));
DECLARE_SYM(int,pthread_cond_signal,(pthread_cond_t *));
DECLARE_SYM(int,pthread_cond_timedwait,(pthread_cond_t *,
pthread_mutex_t *, const struct timespec *));
DECLARE_SYM(int,pthread_cond_wait,(pthread_cond_t *, pthread_mutex_t *));
DECLARE_SYM(int,pthread_join,(pthread_t, void **));
DECLARE_SYM(int,pthread_mutex_destroy,(pthread_mutex_t *));
DECLARE_SYM(int,pthread_mutex_init,(pthread_mutex_t *, const pthread_mutexattr_t *));
DECLARE_SYM(int,pthread_mutex_lock,(pthread_mutex_t *));
DECLARE_SYM(int,pthread_mutex_trylock,(pthread_mutex_t *));
DECLARE_SYM(int,pthread_mutex_unlock,(pthread_mutex_t *));
#endif
LibCSymTable();
static void* getHandle();
static LibCSymTable& instance();
};
#endif /*LIBCSYMTABLE_H_*/