2022.8.16 作业

1、用select实现TCP客户端的收发功能

#include <stdio.h>	
#include <sys/select.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>

#define PORT 8888
#define IP "192.168.31.200"

#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%s__ __%d__\n",__func__,__LINE__);\
	perror(msg);\
}while(0)

int main(int argc, const char *argv[])
{
	int cfd=socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0){
		ERR_MSG("socket");
		return -1;
	}
	
	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;
	}

	//创建读集合
	fd_set readfds,tempfds;

	//将读集合清空
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	
	//将文件描述符添加进读集合
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);
	
	int maxfd=cfd;
	int res;
	int select_res;
	char buf[128]={0};
	while(1){
		
		tempfds=readfds;
		
		//检测读集合
		select_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(select_res<0){
			perror("select");
			return -1;
		}else if(0==select_res){
			fprintf(stderr,"select time out\n");
			return -1;
		}
		
		if(FD_ISSET(0,&tempfds)){
			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");
			}
		}

		if(FD_ISSET(cfd,&tempfds)){
			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");
				return -1;
			}else{
				printf("服务器:%s\n",buf);
			}
		}
	}

	return 0;
}