网编day5

 select的TCP服务器模型

#include<myhead.h>

#define PORT 8888
#define IP "192.168.125.61"
void handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>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[])
{
		if(signal(17,handler)==SIG_ERR)
	{
		ERR_MSG("signal");
		return -1;
	}
	//创建流式套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d\n",sfd);

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

	//填充地址信息结构体给bind
	//真实的地址信息结构体根据地质组指定AF_INET:man 7 ip
	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

	//绑定服务器的地址信息ip、端口--->必须绑定bind//绑定地址服务器的信息
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//将套接字设置为被动监听状态 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;

	//获取一个已经完成的客户端信息，生成一个新的文件描述符accept
	int s_res=-1;
	char buf[128]="";	
	ssize_t res=-1;

	struct sockaddr_in saveCin[1024];
	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 error");
		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);
	//更新maxfd
	*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)==0 && (*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,"*_*");
	if(send(fd,buf,sizeof(buf),0)<0)
	{
		ERR_MSG("send");
		return -1;
	}
	printf("发送成功\n");

	return 0;
}