zookeeper/c/tests/LibCSymTable.cc

/*
 * Copyright 2008, Yahoo! Inc.
 *
 * Licensed 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.
 */

#include "LibCSymTable.h" 

#define LOAD_SYM(sym) \
    sym=(sym##_sig)dlsym(handle,#sym); \
    assert("Unable to load "#sym" from libc"&&sym)      
    

LibCSymTable& LibCSymTable::instance(){
    static LibCSymTable tbl;
    return tbl;
}

//******************************************************************************
// preload original libc symbols
LibCSymTable::LibCSymTable()
{
    void* handle=getHandle();
    LOAD_SYM(gethostbyname);
    LOAD_SYM(calloc);
    LOAD_SYM(realloc);
    LOAD_SYM(free);
    LOAD_SYM(random);
    LOAD_SYM(srandom);
    LOAD_SYM(printf);
    LOAD_SYM(socket);
    LOAD_SYM(close);
    LOAD_SYM(getsockopt);
    LOAD_SYM(setsockopt);
    LOAD_SYM(fcntl);
    LOAD_SYM(connect);
    LOAD_SYM(send);
    LOAD_SYM(recv);
    LOAD_SYM(select);
    LOAD_SYM(poll);
    LOAD_SYM(gettimeofday);
#ifdef THREADED
    LOAD_SYM(pthread_create);
    LOAD_SYM(pthread_detach);
    LOAD_SYM(pthread_cond_broadcast);
    LOAD_SYM(pthread_cond_destroy);
    LOAD_SYM(pthread_cond_init);
    LOAD_SYM(pthread_cond_signal);
    LOAD_SYM(pthread_cond_timedwait);
    LOAD_SYM(pthread_cond_wait);
    LOAD_SYM(pthread_join);
    LOAD_SYM(pthread_mutex_destroy);
    LOAD_SYM(pthread_mutex_init);
    LOAD_SYM(pthread_mutex_lock);
    LOAD_SYM(pthread_mutex_trylock);
    LOAD_SYM(pthread_mutex_unlock);
#endif
}

void* LibCSymTable::getHandle(){
    static void* handle=0;
    if(!handle){
#ifdef __CYGWIN__
        handle=dlopen("cygwin1.dll",RTLD_LAZY);
        assert("Unable to dlopen global sym table"&&handle);
#else
        handle=RTLD_NEXT;
#endif
    }
    return handle;
}