dup和dup2都是用来复制一个文件的描述符。它们经常用来重定向进程的stdin、stdout和stderr。这两个函数的原形及描述如下:
输出重定向
#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<fcntl.h>
#include<string.h>
int main(int argc,char *argv[])
{
int fd=open("./log",O_CREAT | O_RDWR,0666);
if(fd<0)
{
perror("open");
return 1;
}
//close(1);//关闭想要重定向的fd
//int new_fd=dup(fd);//重定向标准输出到打开文件fd
int new_fd=dup2(fd,1);
if(new_fd<0)
{
perror("dup");
return 2;
}
close(fd);
char buf[1024];
while(1)
{
memset(buf,'\0',sizeof(buf));
fgets(buf,sizeof(buf),stdin);
if(strncmp("hello world",buf,11)==0)
{
break;
}
printf("%s\n",buf);//直接打印到标准输出,即可写入文件
fflush(stdout);
}
close(new_fd);
return 0;
}
使用dup2实现客户端,基于tcp_socket
#include<stdio.h>
#include<stdlib.h>
#include<fcntl.h>
#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<string.h>
#include<netinet/in.h>
#include<arpa/inet.h>
static void usage(const char *proc)
{
printf("%s [server_ip] [server_port]\n",proc);
}
int main(int argc,char *argv[])
{
if(argc!=3)
{
usage(argv[0]);
return 1;
}
//create socket
int sock=socket(AF_INET,SOCK_STREAM,0);
if(sock<0)
{
perror("socket");
return 2;
}
//connect
struct sockaddr_in peer;
peer.sin_family=AF_INET;
peer.sin_port=htons(atoi(argv[2]));
peer.sin_addr.s_addr=inet_addr(argv[1]);
if(connect(sock,(struct sockaddr*)&peer,sizeof(peer))<0)
{
perror("connect");
return 3;
}
close(1);
int new_sock=dup2(sock,1);
close(sock);
char buf[1024];
while(1)
{
ssize_t s=read(0,buf,sizeof(buf)-1);
if(s>0)
{
buf[s-1]=0;
}
if(strncmp("hello",buf,5)==0)
{
break;
}
printf("%s\n",buf);
fflush(stdout);
}
close(new_sock);
return 0;
}