APUE习题3.2----自己实现的dup2( )函数的源码

APUE习题3.2----自己实现的dup2( )函数的源码

转载的，不清楚出处在哪里，对于我这个菜鸟来说，写的很好。

原 dup2（）函数：

#include <unistd.h>

int dup2( int fd, int fd2 );

对于 dup2，可以用 fd2 参数指定新描述符的值。如果 fd2 已经打开，则先将其关闭。如若 fd 等于 fd2，则 dup2 返回 fd2，而不关闭它。否则，fd2 的 FD_CLOEXEC 文件描述符标志就被清除，这样 fd2 在进程调用 exec 时是打开状态。该函数返回的新文件描述符与参数 fd 共享同一个文件表项。

下面是自己实现的 dup2函数：

#include <unistd.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h> 
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>

//检查文件描述符是否有效
int isFileDescriptor( int fd )
{
    struct stat st;
    if( (-1 == fstat(fd,&st)) && (EBADF == errno) )
        return -1;
    return 0;
}

int my_dup2( int oldfd, int newfd )
{
    int tempfd;
    int fd_count;
    int fdarray[newfd];
    int res;

    if( -1 == isFileDescriptor( oldfd ) ) {
        printf("the file descriptor is invalid.\n");
        return -1;
    }

    //如果newfd等于oldfd，则直接返回newfd，而不关闭它
    if( oldfd == newfd ) 
        return newfd;
    
    //否则，关闭newfd
    if( 0 == isFileDescriptor( newfd ) ) {
        res = close( newfd );  
        if( -1 == res ) {
            perror("close file descriptor failed");
            return -1;
        }
    }

    //复制文件描述符
    for( fd_count=0; fd_count<newfd; fd_count++ )
        fdarray[fd_count] = 0;
    
    for( fd_count=0; fd_count<newfd; fd_count++ ) {
        tempfd = dup( oldfd );
        if( -1 == tempfd )
            return -1;
        if( tempfd == newfd )
            break;
        else
            fdarray[fd_count] = 1;

    }

    //关闭之前寻找指定描述符时打开的描述符
    for( fd_count=0; fd_count<newfd; fd_count++ ) {
        if( 1 == fdarray[fd_count] ) {
            res = close( fd_count );
            if( -1 ==res ) {
                perror("close file descriptor failed");
                return -1;
            }
        }
    }

    return tempfd;
}

//测试代码
int main()
{
    int fd;
    int testfd;
    int res;
    char *buffer = (char *)malloc(sizeof(char)*32);
    fd = open("/tmp/dup2test1.txt", O_RDWR | O_CREAT, 0666);
    if( -1 == fd ) {
        perror("file open failed");
        exit( EXIT_SUCCESS );
    }
    
    testfd = my_dup2( fd, 5 );

    res = write( testfd, "Hey man!", strlen("Hey man!") ); //通过复制得到的文件描述符 testfd 写数据
    if( -1 == res ) {
        perror("write to testfd failed");
        exit( EXIT_FAILURE );
    }

    printf("write to testfd %d successfully\n", testfd);

    memset( buffer, '\0', 32 );
    lseek( testfd, 0, SEEK_SET );
    res = read( fd, buffer, 30 );  //通过初始的文件描述符 fd 读取数据
    if( -1 == res ) {
        perror("read from testfd failed");
        exit( EXIT_FAILURE );
    }
    printf("read from initial fd %d is: %s\n", fd, buffer );
    exit( EXIT_SUCCESS );
}

程序运行结果：

[zhang@localhost APUE]$ ./my_dup2
write to testfd 5 successfully
read from initial fd 3 is: Hey man!

测试通过。