poll实现tcp通信客户端

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>                             
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <poll.h>
#define ERR_MSG(msg) do{\
    fprintf(stderr,"line:%d",__LINE__);\
    perror(msg);\
}while(0)
#define PORT 6666      //1026~49151
#define ip "192.168.31.226"  //本机IP
int main(int argc, const char *argv[])
{
    //创建流式套接字
    int cfd=socket(AF_INET,SOCK_STREAM,0);
    if(cfd<0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("socket create success cfd=%d\n",cfd);
    //填充客户端的地址信息结构体
    //真是的地址信息结构体根据地址族执行，AF_INET:man 7 ip
    struct sockaddr_in sin;
    sin.sin_family =AF_INET;//必须填AF_INET
    sin.sin_port = htons(PORT);//端口号的网络字节序 1024~49151
    sin.sin_addr.s_addr = inet_addr(ip);         //IP地址的网络字节序,ifconfig查看

   	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect sunccess\n");


	//创建集合
	struct pollfd fds[2];
	fds[0].fd=0;          	//将0号添加到集合中
	fds[0].events=POLLIN;   //读事件
	fds[1].fd=cfd;
	fds[1].events=POLLIN;

    char buf[128]="";
    ssize_t res=-1;
	int p_res=0;
	while(1)
	{
		p_res=poll(fds,2,-1);
		if(p_res<0)
		{
			ERR_MSG("poll");
			return -1;
		}
		else if(0==p_res)
		{
			printf("time out...\n");
			return -1;
		}

		//能运行到此位置，代表有文件描述符产生事件
		//判断所有文件描述符实际产生的事件(revents)中是否有POLLIN
		//有可能同时产生多个事件，其中包含了POLLIN
		//通过按位与提取出revents中是否有POLLIN
		if(fds[0].revents & POLLIN)
		{

			bzero(buf,sizeof(buf));

			//发送 	
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]='\0';
			if(send(cfd,buf, sizeof(buf),0)<0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send sunccess\n");
		}
		if(fds[1].revents & POLLIN)
		{
			//接受       
			bzero(buf,sizeof(buf));

			res=recv(cfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(0==res)
			{
				fprintf(stderr,"服务器关闭\n");
				break;

			}
			printf("%s\n",buf);
		}
	}

	//关闭所有套接字文件描述符
	close(cfd);

	return 0;
}