poll

12_30 I/O复用poll

1.如果一个一个字符读取，有几个字符就读去几次，因为接收缓冲区内还有数据

2.如果直接结束客户端，select就会收到这个描述符上有读事件，recv就不会受到数据0，就会知道已经关闭，然后就关闭它。

int poll(struct pollfd* fds,nfds_t nfds,int timeout);

struct pollfd
{
	int fd;//文件描述符
	short events;//注册的事件
	short revents;//实际发生的事件，由内核填充
}

poll的文件描述符没有上限

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

#define MAX 10
int socket_init();

void poll_fds_init(struct pollfd fds[])
{
	for(int i=0;i<MAX;i++)
	{
		fds[i].fd=-1;
		fds[i].events=0;
		fds[i].revents=0;
	}
}

void poll_fds_add(int fd,struct pollfd fds[])
{
	for(int i=0;i<MAX;i++)
	{
		if(fds[i].fd==-1)
		{
			fds[i].fd=fd;
			fds[i],events=POLLIN;
			fds[i].revents=0;
			break;
		}
	}
}

void poll_fds_del(int fd,struct pollfd fds[])
{
	for(int i=0;i<MAX;i++)
	{
		if(fds[i].fd==fd)
		{
			fds[i].fd==-1;
			fds[i].events=0;
			fds[i].revents=0;
			break;
		}
	}
}


int main()
{
	int sockfd=socket_init();
	assert(sockfd!=-1);
	
	struct pollfd fds[MAX];
	poll_fds_init(poll_fds);
	poll_fds_add(sockfd,poll_fds);
	
	while(1)
	{
		int n=poll(poll_fds,MAX,5000);
		if(n<0)
		{
			printf("poll error\n");
		}
		else if(n==0)
		{
			printf("time out\n");
		}
		else
		{
			for(int i=0;i<MAX;i++)
			{
				if(poll_fds[i].fd==-1)
				{
					continue;
				}
				
				if(poll_fds[i]_revents&POLLIN)
				{
					if(poll_fds[i].fd===sockfd)
					{
						struct sockaddr_in caddr;
						int len=sizeof(caddr);
						int c=accept(sockfd,(struct sockaddr*)&caddr,&len);
						if(c<0)
						{
							continue;
						}
						printf("accept:%d\n",c);
						
						poll_fds_add(poll_fds);
					}
					else
					{
						char buff[128]={0};
						int num=recv(poll_fds[i].fd,buff,127,0);
						if(num<=0)
						{
							close(poll_fds[i].fd);
							poll_fds_del(poll_fds[i].fd,poll_fds);
							printf("client close\n");
						}
						else
						{
							printf("recv(%d)=%s\n",poll_fds[i].fd,buff);
							send(poll_fds[i].fd,"ok",2,0);
						}
					}
				}
				
			}
		}
	}
	
}

int socket_init()
{
	int sockfd=socket(AF_INET,SOCK_STREAM,0);
	if(sockfd!=-1)
	{
		return -1;
	}
	
	struct sockaddr_in saddr;
	memset(saddr,0,sizeof(saddr));
	saddr.sin_family=AF_INET;
	saddr.sin_port=htons(6000);
	saddr.sin_saddr.s_addr=inet_addr("127.0.0,1");
	
	int res=bind(sockfd,(struct sockaddr*)&saddr,sizeof(saddr));
	if(res==-1)
	{
		return -1;
	}
	
	res=listen(sockfd,5);
	if(res==-1)
	{
		return -1;
	}
	return sockfd;
}