首页 > 代码库 > APUE习题3.2----自己实现的dup2( )函数的源码
APUE习题3.2----自己实现的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!
测试通过。
APUE习题3.2----自己实现的dup2( )函数的源码
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。