Linux作业

2023-5-21作业补

用poll搭建TCP客户端

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <poll.h>
//封装错误宏
#define ERR_MSG(msg) do{printf("line:%d\n",__LINE__);perror(msg);}while(0)
#define PORT 6666    //端口号
#define IP "192.168.8.62"    //主机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 __%d__\n",cfd,__LINE__);

	int reuse=1;
	if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
        //客户端IP和端口号创建
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
        //重用端口
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");return -1;
	}printf("Server link success! __%d__\n",__LINE__);
        
	struct pollfd fds[2];

	fds[0].fd=0;
	fds[0].events=POLLIN;

	fds[1].fd=cfd;
	fds[1].events=POLLIN;

	char buf[128]="";
	ssize_t res=0;
	int p_res=0;
	while(1)
	{
                //poll监听描述符事件
		p_res=poll(fds,2,-1);
		if(p_res<0)
		{
			ERR_MSG("poll");return -1;
		}
		else if(0==p_res)
		{
			printf("Overrun.....\n");break;
		}

            
		if((fds[0].revents&POLLIN)!=0)
		{
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]=0;

			if(send(cfd,buf,sizeof(buf),0)<0)
			{
				ERR_MSG("send");return -1;
			}printf("Transmit success!\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)
			{
				printf("[%s:%d] Server offline __%d__\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),__LINE__);break;
			}printf("[%s:%d] cfd=%d:%s,__%d__\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),cfd,buf,__LINE__);
		}
	}
        //关闭套接字
	if(close(cfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}

	return 0;
}