linux 客户端服务器模型

/* ************************************************************************
 *       Filename:  t06_epoll.cpp
 *    Description:  
 *        Version:  1.0
 *        Created:  11/05/2016 07:06:58 PM
 *       Revision:  none
 *       Compiler:  gcc
 *         Author:  YOUR NAME (), 
 *        Company:  
 * ************************************************************************/

#include "../h.h"

void set_non_block(int fd)
{
	uint32_t flags;
	flags=fcntl(fd,F_GETFL);
	flags|=O_NONBLOCK;
	fcntl(fd,F_SETFL,flags);
}

int main()
{
	int fd=socket(AF_INET,SOCK_STREAM,0);

	struct sockaddr_in addr;
	addr.sin_family=AF_INET;
	addr.sin_port=htons(9988);
	addr.sin_addr.s_addr=INADDR_ANY;

	int ret=bind(fd,(struct sockaddr*)&addr,sizeof(addr));
	if(ret<0)
	{
		perror("bind error");
		return 1;
	}
	listen(fd,20);

	set_non_block(fd);
	int epollfd=epoll_create(512);

	struct epoll_event ev;
	ev.data.fd=fd;
	ev.events=EPOLLIN;

	epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&ev);

	while(1)
	{
		struct epoll_event out_ev[8];
		int ret=epoll_wait(epollfd,out_ev,8,2000);
		if(ret>0)
		{
			for(int i=0;i<ret;i++)
			{
				struct epoll_event*p=&out_ev[i];
				if(p->data.fd==fd)
				{
					while(1)
					{
						int newfd=accept(fd,NULL,NULL);
						if(newfd<0)
						{
							if(errno==EAGAIN||errno==EINTR)
								break;
							else
								exit(1);
						}
						printf("has new client...\n");
						send(newfd,"hello client...\n",20,0);
						ev.data.fd=newfd;
						set_non_block(newfd);
						ev.events=EPOLLIN|EPOLLET|EPOLLOUT;
						epoll_ctl(epollfd,EPOLL_CTL_ADD,newfd,&ev);
					}
				}
				else
				{
					char buf[1024];
					while(1)
					{
						int ret=read(p->data.fd,buf,sizeof(buf));
						if(ret>0)
						{
							printf("Servrecv:%s\n",buf);
							send(p->data.fd,buf,sizeof(buf),0);
						}
						else if(ret<=0)
						{
							if(errno==EAGAIN||EINTR) break;
							epoll_ctl(epollfd,EPOLL_CTL_DEL,p->data.fd,&ev);
							close(p->data.fd);
							break;
						}
					}
				}
			}
		}
	}

}





/*****************************************华丽分割线*************************************/


/* ************************************************************************
 *       Filename:  to3_send.cpp
 *    Description:  
 *        Version:  1.0
 *        Created:  10/31/2016 06:44:08 PM
 *       Revision:  none
 *       Compiler:  gcc
 *         Author:  YOUR NAME (), 
 *        Company:  
 * ************************************************************************/

#include "../h.h"

int main(int argc,char*argv[])
{
	int fd=socket(AF_INET,SOCK_STREAM,0);
	if(argc!=3)
	{
		printf("usage [%s] [addr] [port]...\n",argv[0]);
		return -1;
	}
	int port=atoi(argv[2]);
	char *address=argv[1];

	struct sockaddr_in addr;
	addr.sin_family=AF_INET;
	addr.sin_port=htons(port);
	addr.sin_addr.s_addr=inet_addr(address);

	int ret=connect(fd,(struct sockaddr*)&addr,sizeof(addr));
	
	if(ret<0)
	{
		printf("server is no ACK...\n");
		return -1;
	}

	fd_set set;
	while(1)
	{
		FD_ZERO(&set);
		FD_SET(STDIN_FILENO,&set);
		FD_SET(fd,&set);

		select(fd+1,&set,NULL,NULL,0);

		if(FD_ISSET(STDIN_FILENO,&set))
		{
			char temp[1024]={0};
			fgets(temp,sizeof(temp),stdin);
			send(fd,temp,sizeof(temp),0);

		}
		if(FD_ISSET(fd,&set))
		{
			char buf[1024]={0};
			int ret=recv(fd,buf,sizeof(buf),0);
			if(ret==0) break;
			printf("Clientrecv:%s\n",buf);
		}
	}
	return 0;

}

 

转载于:https://my.oschina.net/urlove/blog/1573241