實現的dup2( )函數的源碼怎么寫

這篇文章將為大家詳細講解有關實現的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!

測試通過。

關于實現的dup2( )函數的源碼怎么寫就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。