华清远见第五课程day5作业

select TCP服务器

#include<myhead.h>

#define IP "192.168.114.52"   //本机IP
#define PORT 25565           //服务器端口
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)          //输出错误信息和位置

int keybord_events(void);
int cliConnect_events(int,struct sockaddr_in*,fd_set*,int *);
int cliRcvSnd_events(int,struct sockaddr_in*,fd_set*,int *);

int main(int argc, const char *argv[])
{
	//创建套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0){
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create sfd=%d\n",sfd);

	//允许端口快速的被复用
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0){
			ERR_MSG("set sockopt");
			return -1;
	}
	printf("允许端口快速的被复用成功\n");

	//填充地址信息结构体给bind函数绑定,
	//真实的地址信息结构体根据地址族指定 AF_INET
	struct sockaddr_in sin;
	sin.sin_family = AF_INET; //必须填AF_INET
	sin.sin_port = htons(PORT);//将端口号转变为网络字节序
	sin.sin_addr.s_addr = inet_addr(IP);//本机IP
	
	//绑定服务器的地址信息 ——>必须绑定bind
	//bind 用来绑定服务器的地址信息，绑定到监听套接字文件对应的空间中
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//将套接字设置为被动监听状态 listen
	//将通信套接字转换成监听套接字
	//listen让内核维护量队列:未完成队列，已完成队列
	//其中未完成队列是正在连接的客户端信息；
	//已完成的是已经连接上的客户端信息
	if(listen(sfd,128)<0){    //数字为未完成连接队列的容量不能少于１
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

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

	FD_ZERO(&readfds);

	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);
	
	int maxfd = sfd;
	int s_res = -1;
	char buf[128]="";
	struct sockaddr_in saveCin[1024];
	ssize_t res=0; //获取接收函数recv的返回值
	while(1){
		tempfds = readfds;

		s_res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(s_res <0){
			ERR_MSG("select");
			return -1;
		}
		else if(0==s_res){
			printf("time out \n");
			break;
		}
		for(int i =0;i<=maxfd;i++){
			if(FD_ISSET(i,&tempfds)==0)
				continue;
				if(0==i){
					printf("触发键盘输入事件\n");
					keybord_events();
				}
				else if(sfd ==i){
					printf("触发客户端连接事件\n");
					cliConnect_events(sfd,saveCin,&readfds,&maxfd);
				}
				else{
					printf("触发客户端交互事件\n");
					cliRcvSnd_events(i,saveCin,&readfds,&maxfd);
				}
			}
	}

	//关闭文件名描述符
	if(close(sfd)<0){
		ERR_MSG("close");
		return -1;
	}

	return 0;
}

//键盘输入事件
int keybord_events(void)
{
	char buf[128]="";
	bzero(buf,sizeof(buf));
	fgets(buf,sizeof(buf),stdin);
	buf[strlen(buf)-1]=0;

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

	return 0;
}

//客户端连接事件
int cliConnect_events(int sfd,struct sockaddr_in saveCin[],fd_set * preadfds,int *pmaxfd)
{
	int newfd =-1;
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);

	newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
	if(newfd <0){
		ERR_MSG("newfd");
		return -1;
	}
	printf("[%s:%d]客户端连接成功 newfd=%d\n",\
			inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
	saveCin[newfd] = cin;

	FD_SET(newfd,preadfds);
	*pmaxfd = *pmaxfd>newfd ? *pmaxfd:newfd;
	
	return 0;
}

//客户端交互事件
int cliRcvSnd_events(int fd ,struct sockaddr_in* saveCin,fd_set *preadfds,int* pmaxfd)
{
	char buf[128]="";
	//清空字符串
	bzero(buf,sizeof(buf));

	//接收
	ssize_t res =recv(fd,buf,sizeof(buf),0);
	if(res <0){
		ERR_MSG("recv");
		return -1;
	}
	else if( 0 == res){
		printf("[%s:%d]客户端下线 newfd=%d\n",\
				inet_ntoa(saveCin[fd].sin_addr),ntohs(saveCin[fd].sin_port),fd);
		close(fd);
		FD_CLR(fd,preadfds);
	while(FD_ISSET(*pmaxfd,preadfds)==1 &&(*pmaxfd)-->=0);
		return 0;
	}
	printf("[%s:%d] newfd=%d:%s\n",\
			inet_ntoa(saveCin[fd].sin_addr),ntohs(saveCin[fd].sin_port),fd,buf);

	//发送
	strcat(buf,"nb");
	if(send(fd,buf,sizeof(buf),0)<0){
		ERR_MSG("send");
		return -1;
	}
	printf("send success\n");
	
	return 0;
}