温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

实现的dup2( )函数的源码怎么写

发布时间:2021-10-14 14:16:42 来源:亿速云 阅读:191 作者:柒染 栏目:编程语言

这篇文章将为大家详细讲解有关实现的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( )函数的源码怎么写就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI