select的tcp客户端与服务器

1、作业要求

        select的tcp客户端与服务器

2、实现过程

#include <stdio.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <unistd.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <sys/time.h>
#include <poll.h>
 
#define ERR_MSG(msg) do{\
	fprintf(stderr,"line : %d\n",__LINE__);\
	perror(msg);\
}while(0)
 
#define IP "192.168.8.110" //填服务器的IP
#define PORT 8080			//1024~49151
 
int main(int argc, const char *argv[])
{
	//创建流式套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd < 0){
		ERR_MSG("socket");
		return -1;
	}
 
	//填充地址信息结构体
	//根据地址族的不同，地址信息结构体不一致，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(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0){
		ERR_MSG("connetct");
		return -1;
	}
	printf("连接成功\n");
 
	//创建集合
	fd_set readfds;
	fd_set tempfds;
 
	//清空文件描述符集合
	FD_ZERO(&readfds);
 
	//将需要的文件描述符添加到读集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);
 
	int maxfd = sfd;
	char buf[128] = "";
	ssize_t res = 0;
	int s_res = 0;
 
	while(1){
		tempfds = readfds;
		//让内核监测集合
		s_res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(s_res < 0){
			ERR_MSG("poll");
			return -1;
		}
		else if( 0 == s_res){
			printf("请求超时...\n");
			break;
		}
		for(int i=0;i<=maxfd;i++){
			if(FD_ISSET(i,&tempfds) == 0){
				continue;
			}
			if(0 == i){
				printf("键盘输入事件\n");
				bzero(buf,sizeof(buf));
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]='\0';

				//发送消息
				if(send(sfd,buf,sizeof(buf),0) < 0){
					ERR_MSG("send");
					return -1;
				}
				printf("newfd = %d: %s\n",sfd,buf);
			}
 
			if(sfd == i){
				printf("服务器交互事件\n");
				bzero(buf,sizeof(buf));

				//接收消息
				res = recv(sfd,buf,sizeof(buf),0);
				if(res < 0){
					ERR_MSG("recv");
					return -1;
				}
				else if(0 == res){
					fprintf(stderr,"newfd = %d 服务器已下线\n",sfd);
					break;
				}
				printf("newfd = %d: %s\n",sfd,buf);
			}
		}
	}

	//关闭套接字
	close(sfd);
 
	return 0;
}

3、结果截图