Changeset 1648 for zzuf/trunk


Ignore:
Timestamp:
Jan 11, 2007, 1:39:59 AM (16 years ago)
Author:
Sam Hocevar
Message:
  • Minor code factorisation into LOADSYM().
Location:
zzuf/trunk/src
Files:
6 edited

Legend:

Unmodified
Added
Removed

  • zzuf/trunk/src/libzzuf.c

    r1641 r1648  
    5353    char *tmp;
    5454
     55    _zz_load_mem();
     56    _zz_load_signal();
     57    _zz_load_fd();
     58    _zz_load_stream();
     59
    5560    tmp = getenv("ZZUF_DEBUG");
    5661    if(tmp && *tmp == '1')
    … …  
    99104        _zz_register(0);
    100105
    101     _zz_load_fd();
    102     _zz_load_mem();
    103     _zz_load_signal();
    104     _zz_load_stream();
    105 
    106106    _zz_ready = 1;
    107107

  • zzuf/trunk/src/load-fd.c

    r1639 r1648  
    8787    { \
    8888        int mode = 0; \
    89         if(!_zz_ready) \
    90             LOADSYM(fn); \
     89        LOADSYM(fn); \
    9190        if(oflag & O_CREAT) \
    9291        { \
    … …  
    132131    int ret;
    133132
    134     if(!_zz_ready)
    135         LOADSYM(accept);
     133    LOADSYM(accept);
    136134    ret = accept_orig(sockfd, addr, addrlen);
    137135    if(!_zz_ready || _zz_disabled || !_zz_network)
    … …  
    151149    int ret;
    152150
    153     if(!_zz_ready)
    154         LOADSYM(socket);
     151    LOADSYM(socket);
    155152    ret = socket_orig(domain, type, protocol);
    156153    if(!_zz_ready || _zz_disabled || !_zz_network)
    … …  
    170167    int ret;
    171168
    172     if(!_zz_ready)
    173         LOADSYM(read);
     169    LOADSYM(read);
    174170    ret = read_orig(fd, buf, count);
    175171    if(!_zz_ready || !_zz_iswatched(fd) || _zz_disabled)
    … …  
    207203    do \
    208204    { \
    209         if(!_zz_ready) \
    210             LOADSYM(fn); \
     205        LOADSYM(fn); \
    211206        ret = ORIG(fn)(fd, offset, whence); \
    212207        if(!_zz_ready || !_zz_iswatched(fd) || _zz_disabled) \
    … …  
    238233    int ret;
    239234
    240     if(!_zz_ready)
    241         LOADSYM(close);
     235    LOADSYM(close);
    242236
    243237    /* Hey, it’s our debug channel! Silently pretend we closed it. */

  • zzuf/trunk/src/load-mem.c

    r1646 r1648  
    115115{
    116116    void *ret;
    117     if(!_zz_ready)
     117    if(!calloc_orig)
    118118    {
    119119        int i = (nmemb * size + 7) / 8;
    … …  
    133133{
    134134    void *ret;
    135     if(!_zz_ready)
     135    if(!malloc_orig)
    136136    {
    137137        int i = (size + 7) / 8;
    … …  
    151151       && (uintptr_t)ptr <= (uintptr_t)dummy_buffer + sizeof(dummy_buffer))
    152152        return;
    153     if(!_zz_ready)
    154         LOADSYM(free);
     153    LOADSYM(free);
    155154    free_orig(ptr);
    156155}
    … …  
    162161       && (uintptr_t)ptr <= (uintptr_t)dummy_buffer + sizeof(dummy_buffer))
    163162        return ptr; /* FIXME: who would call realloc() so early? */
    164     if(!_zz_ready)
    165         LOADSYM(realloc);
     163    LOADSYM(realloc);
    166164    ret = realloc_orig(ptr, size);
    167165    if(ret == NULL && _zz_memory && errno == ENOMEM)
    … …  
    173171{
    174172    void *ret;
    175     if(!_zz_ready)
    176         LOADSYM(valloc);
     173    LOADSYM(valloc);
    177174    ret = valloc_orig(size);
    178175    if(ret == NULL && _zz_memory && errno == ENOMEM)
    … …  
    185182{
    186183    void *ret;
    187     if(!_zz_ready)
    188         LOADSYM(memalign);
     184    LOADSYM(memalign);
    189185    ret = memalign_orig(boundary, size);
    190186    if(ret == NULL && _zz_memory && errno == ENOMEM)
    … …  
    198194{
    199195    int ret;
    200     if(!_zz_ready)
    201         LOADSYM(posix_memalign);
     196    LOADSYM(posix_memalign);
    202197    ret = posix_memalign_orig(memptr, alignment, size);
    203198    if(ret == ENOMEM && _zz_memory)
    … …  
    213208#define MMAP(fn, off_t) \
    214209    do { \
    215         if(!_zz_ready) \
    216             LOADSYM(fn); \
     210        LOADSYM(fn); \
    217211        ret = ORIG(fn)(start, length, prot, flags, fd, offset); \
    218212        if(!_zz_ready || !_zz_iswatched(fd) || _zz_disabled) \
    … …  
    271265    int ret, i;
    272266
    273     if(!_zz_ready)
    274         LOADSYM(munmap);
     267    LOADSYM(munmap);
    275268    for(i = 0; i < nbmaps; i++)
    276269    {
    … …  
    295288    kern_return_t ret;
    296289
    297     if(!_zz_ready)
    298         LOADSYM(map_fd);
     290    LOADSYM(map_fd);
    299291    ret = map_fd_orig(fd, offset, addr, find_space, numbytes);
    300292    if(!_zz_ready || !_zz_iswatched(fd) || _zz_disabled)

  • zzuf/trunk/src/load-signal.c

    r1627 r1648  
    9494    SIG_T ret;
    9595
    96     if(!_zz_ready)
    97         LOADSYM(signal);
     96    LOADSYM(signal);
    9897
    9998    if(!_zz_signal)
    … …  
    111110    int ret;
    112111
    113     if(!_zz_ready)
    114         LOADSYM(sigaction);
     112    LOADSYM(sigaction);
    115113
    116114    if(!_zz_signal)

  • zzuf/trunk/src/load-stream.c

    r1621 r1648  
    132132    do \
    133133    { \
     134        LOADSYM(fn); \
    134135        if(!_zz_ready) \
    135         { \
    136             LOADSYM(fn); \
    137136            return ORIG(fn)(path, mode); \
    138         } \
    139137        _zz_disabled = 1; \
    140138        ret = ORIG(fn)(path, mode); \
    … …  
    165163    int fd0 = -1, fd1 = -1, disp = 0;
    166164
    167     if(!_zz_ready)
    168         LOADSYM(freopen);
     165    LOADSYM(freopen);
    169166    if(_zz_ready && (fd0 = fileno(stream)) >= 0 && _zz_iswatched(fd0))
    170167    {
    … …  
    216213    { \
    217214        int fd; \
    218         if(!_zz_ready) \
    219             LOADSYM(fn); \
     215        LOADSYM(fn); \
    220216        fd = fileno(stream); \
    221217        if(!_zz_ready || !_zz_iswatched(fd)) \
    … …  
    245241    int fd;
    246242
    247     if(!_zz_ready)
    248         LOADSYM(rewind);
     243    LOADSYM(rewind);
    249244    fd = fileno(stream);
    250245    if(!_zz_ready || !_zz_iswatched(fd))
    … …  
    276271    int fd;
    277272
    278     if(!_zz_ready)
    279         LOADSYM(fread);
     273    LOADSYM(fread);
    280274    fd = fileno(stream);
    281275    if(!_zz_ready || !_zz_iswatched(fd))
    … …  
    324318    do { \
    325319        int fd; \
    326         if(!_zz_ready) \
    327             LOADSYM(fn); \
     320        LOADSYM(fn); \
    328321        fd = fileno(stream); \
    329322        if(!_zz_ready || !_zz_iswatched(fd)) \
    … …  
    362355    int fd;
    363356
    364     if(!_zz_ready)
    365         LOADSYM(fgets);
     357    LOADSYM(fgets);
    366358    fd = fileno(stream);
    367359    if(!_zz_ready || !_zz_iswatched(fd))
    … …  
    417409    int ret, fd;
    418410
    419     if(!_zz_ready)
    420         LOADSYM(ungetc);
     411    LOADSYM(ungetc);
    421412    fd = fileno(stream);
    422413    if(!_zz_ready || !_zz_iswatched(fd))
    … …  
    451442    int ret, fd;
    452443
    453     if(!_zz_ready)
    454         LOADSYM(fclose);
     444    LOADSYM(fclose);
    455445    fd = fileno(fp);
    456446    if(!_zz_ready || !_zz_iswatched(fd))
    … …  
    471461        ssize_t done, size; \
    472462        int fd, finished = 0; \
    473         if(!_zz_ready) \
    474             LOADSYM(fn); \
     463        LOADSYM(fn); \
    475464        fd = fileno(stream); \
    476465        if(!_zz_ready || !_zz_iswatched(fd)) \
    … …  
    553542    int fd;
    554543
    555     if(!_zz_ready)
    556         LOADSYM(fgetln);
     544    LOADSYM(fgetln);
    557545    fd = fileno(stream);
    558546    if(!_zz_ready || !_zz_iswatched(fd))
    … …  
    603591    int ret, fd, tmp;
    604592
    605     if(!_zz_ready)
    606         LOADSYM(__srefill);
     593    LOADSYM(__srefill);
    607594    fd = fileno(fp);
    608595    if(!_zz_ready || !_zz_iswatched(fd))

  • zzuf/trunk/src/load.h

    r1639 r1648  
    2222#define LOADSYM(x) \
    2323    do { \
    24         ORIG(x) = dlsym(RTLD_NEXT, STR(x)); \
     24        if(!ORIG(x)) \
     25            ORIG(x) = dlsym(RTLD_NEXT, STR(x)); \
    2526        if(!ORIG(x)) \
    2627            abort(); \
Note: See TracChangeset for help on using the changeset viewer.