32网络通信之Poll模型

#include<stdio.h>
#include<unistd.h>
#include<string.h>

#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>

#include<sys/poll.h>
#include<sys/time.h>

#define SRV_PORT 0xabcd
#define CONN_MAX 10000

void PollProcss(struct  pollfd *pfds,int *plen)
{
	int i;
	int fd;
	int iRet;
	struct sockaddr_in addr;
	socklen_t addrlen = sizeof(addr);
	char szMsg[1000] = { 0 };
	char szBuff[1000] = { 0 };

	//monitor POLLIN is 1 or 0
	if (pfds[0].revents & POLLIN) 
	{
		read(0,szMsg,1000);
		for (i = 2; i < *plen; ++i)
		{
			write(pfds[i].fd, szMsg, strlen(szMsg));
		}
	}
	if (pfds[1].revents & POLLIN)
	{
		fd = accept(pfds[1].fd, (struct sockaddr*)&addr, &addrlen);
		if (fd < 0)
		{
			perror("Fail to accept!");
			return;
		}
		if (*plen == CONN_MAX)
		{
			printf("\rConnect over limit\n");
			write(fd, "Connect over limit", 20);
			close(fd);
		}
		else
		{
			write(fd, "Welcome", 8);
			printf("\r[%d]New connect from %s[%d]\n", fd, inet_ntoa(addr.sin_addr), ntohs(addr.sin_port));

			pfds[*plen].fd = fd;
			pfds[*plen].events = POLLIN;

			(*plen)++;
		}
	}
	//TranslateMessage DispatchMessage
	for (i = 3; i < *plen; ++i)
	{
		if (pfds[i].fd.revents & POLLIN)
		{
			memset(szBuff, 0, 1000);

			iRet = read((pfds[i].fd, szBuff, 1000);
			if (iRet < 0)
			{
				perror("Fail to read!");
				break;
			}
			//Disconnect
			else if (iRet == 0)
			{
				int j;
				//Back cover front
				for (j = i; j < *plen; ++j)
				{
					pfds[j].fd = pfds[j + 1].fd;
				}
				(*plen++);
				//
				i--;

				/*  That‘s the same thing as the top
					pfds.fd=pfds[*plen-1].fd; //replace pfds[i].fd to last fd
					(*plen)--;
					i--;
				*/

			}
			//Normal processing data
			else
			{
				printf("\r[%d]Recv:%s\n", pfds[i].fd, szBuff);
			}

		}
	}

	return;
}

void PollServer()
{
	int fd;
	int iRet;
	struct sockaddr_in addr;
	socklen_t addrlen = sizeof(addr);

	fd = socket(PF_INET, SOCK_STREAM, 0);
	if (fd < 0)
	{
		perror("Fail to socket!");
		return;
	}

	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = htonl(INADDR_ANY);
	addr.sin_port = htons(SRV_PORT);

	iRet = bind(fd, (struct sockaddr*)&addr, addrlen);
	if (iRet)
	{
		perror("Fail to bind!");
		close(fd);
		return;
	}

	iRet = listen(fd, 100);
	if (iRet)
	{
		perror("Fail to listen!");
		close(fd);
		return;
	}

	// /////////////init pollfd
	//struct pollfd pfds[CONN_MAX];
	struct  pollfd *pfds = (struct pollfd*)malloc(CONN_MAX *sizeof(struct pollfd));
	int nfds = 0;	//fd num of array:pfds

	//monitor stdin
	pfds[0].fd = 0;
	pfds[0].events = POLLIN;
	nfds++;

	//monitor tcp server fd
	pfds[1].fd = fd;
	pfds[1].events = POLLIN;
	nfds++;
	////////////////

	printf("Start server ok..\n");

	while (1)
	{
		fprintf(stderr, "Send:");
		iRet = poll(pfds, nfds, -1);
		if (iRet == 0)
		{
			//timeout
			continue;
		}
		eles if (iRet < 0)
		{
			perror("Fail to poll!");
			break;
		}
		else
		{
			PollProcss(pfds, &nfds);
		}
	}
	close(fd);
	return;
}

int main()
{
	TcpServer();
	
	return 0;
}