APUE习题[实现dup2函数功能，不使用fcntl]

学习，记录。

int dup2(int oldhandle, int newhandle);

函数功能：

    复制文件句柄，newhandle指定的dup2和dup的区别就是可以用newfd参数指定新描述符的数值，如果newfd已经打开，则先将其关闭。如果newfd等于oldfd，则dup2返回newfd, 而不关闭它。dup2函数返回的新文件描述符同样与参数oldfd共享同一文件表项。

关键部分实现思路：   
    先close关闭需要复制到的文件描述符newdup。
    连续dup,每dup一次产生的新的fd记录下来。
    当新产生的fd等于需要产生的fd的时候，跳出循环，并把前面产生的fd全都close掉，返回该描述符。

注释挺详细，看注释吧

执行结果:

1. //Code by Pnig0s1992 
2. //Date:2012,3,28 
3. #include <unistd.h> 
4. #include <fcntl.h> 
5. #include <string.h> 
6. #include <sys/types.h> 
7. #include <sys/stat.h> 
8.  
9.  
10. int my_dup(int olddup,int newdup); 
11.  
12. int main(int argc,char ** argv) 
13. { 
14.     int newdup = 3; 
15.     const char * filename = "newfile.txt"; 
16.     int fd = open(filename,O_RDWR); 
17.     int newfd = my_dup(fd,newdup); 
18.     if(write(newfd,"Test new fd.",strlen("Test new fd.")) < 0) 
19.     { 
20.         printf("Use new fd write file failed."); 
21.         exit(2); 
22.     }else 
23.     { 
24.         printf("Write successfully."); 
25.     } 
26.     exit(0); 
27. } 
28.  
29. int my_dup(int olddup,int newdup) 
30. { 
31.     int tempdup; 
32.     int icount = 0; 
33.     int filedesarr[newdup]; 
34.     if((tempdup = dup(olddup)) == -1) //判断原文件描述服是否有效 
35.     { 
36.         printf("the file desp is invalid."); 
37.         exit(1); 
38.     }else 
39.     { 
40.         close(tempdup); 
41.     } 
42.  
43.     if(newdup == olddup) //若新旧文件描述符相等则直接返回 
44.     { 
45.         return olddup; 
46.     } 
47.     close(newdup);//关闭要复制的文件描述符 
48.     for(icount = 0;icount<newdup+1;icount++) //循环复制文件描述符 
49.     { 
50.         filedesarr[icount] = 0; 
51.         tempdup = dup(newdup); 
52.         if(tempdup < 0) 
53.         { 
54.             return -1; 
55.         }else 
56.         { 
57.             if(tempdup == newdup) 
58.             { //若复制后的文件描述符于指定的相等则跳出 
59.                 break; 
60.             }else{ 
61.                 filedesarr[icount] = 1; //否则将对应下标的数组元素置为1 
62.             } 
63.         } 
64.     } 
65.     for(icount = 0;icount<newdup+1;icount++) //关闭之前打开的非指定描述符 
66.     { 
67.         if(filedesarr[icount] == 1) 
68.         { 
69.             close(icount); 
70.         } 
71.     } 
72.     return tempdup; 
73. } 

 

本文转hackfreer51CTO博客，原文链接：http://blog.51cto.com/pnig0s1992/819830，如需转载请自行联系原作者