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>
  10. int my_dup(int olddup,int newdup);
  12. int main(int argc,char ** argv)
  13. {
  14. int newdup = 3;
  15. constchar * 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. }
  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. }
  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. }

相关内容