[置顶] 重写dup2函数

     在unix环境高级编程的第三章中有一道题目是这样的:
重写dup2函数,不能使用fnctl函数,要有规范的容错处理

刚开始没有什么思路,后来在网上查了一下,发现大多数的处理都是,一直调用dup直到得到我们所想要的文件描述符为止,然后关闭不需要的文件描述符。代码如下:

#include <unistd.h>
#include <stdio.h>

int mydup2(int fd,int fd2)
{
        int fdtmp[fd2+1];
        int loop;
        int loop1;

        // 判断fd的范围
        if( (fd<0)||(fd>63) )
        {
                printf("fd=%d is out of range!\n",fd);
                return -1;
        }

        if( (fd2<0)||(fd2>63) )
        {
                printf("fd2=%d is out of range!\n",fd2);
                return -2;
        }

        //判断fd是否被打开
        if( (fdtmp[0]=dup(fd))==-1 )
        {
                printf("fd=%d is not a file descriptor!\n",fd);
                return -3;
        }

        if( (fdtmp[0]==fd2)||(fd==fd2) )
        {
                return fd2;
        }

        //关闭fd2
        close(fd2);
        close(fdtmp[0]);

        //一直dup到所期望的文件描述符为止
        for(loop=0;loop<=fd2;loop++)
        {
                if( (fdtmp[loop]=dup(fd))<0 )
                {
                        printf("dup error!\n");
                        return -4;
                }
                else
                {
                        if( fdtmp[loop]==fd2 )
                        {
                                break;
                        }
                }
        }
        //关闭不符合要求的文件描述符
        for(loop1=0;loop1<loop;loop1++)
        {
                if( close(fdtmp[loop1])<0 )
                {
                        printf("close fd error!\n");
                        return -5;
                }
        }

        return fd2;
}

int main(void)
{
        int fd;
        char buff[10]={'a','b','c','d','e','f','g','h','i','j'};
        if( (fd=mydup2(0,20))>0 )
        {
                printf("fd=%d\n",fd);
        }
        write(fd,buff,10);


        return 0;
}
                     

需要说明的一点是,有的系统对文件描述符的上界没有限制,不需要进行上界的检查。



你可能感兴趣的:(linux,dup2,unix高级编程,重写dup2)