多路复用（四）poll

服务端

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

int main()
{
	int sockfd = socket(AF_INET,SOCK_STREAM,0);
	if(0 > sockfd)
	{
		perror("socket");
		return -1;
	}

	struct sockaddr_in addr = {};
	addr.sin_family = AF_INET;
	addr.sin_port = htons(5678);
	addr.sin_addr.s_addr = inet_addr("192.168.0.111");
	socklen_t addrlen = sizeof(addr);

	if(bind(sockfd,(struct sockaddr*)&addr,addrlen))
	{
		perror("bind");
		return -1;
	}

	if(listen(sockfd,10))
	{
		perror("listen");
		return -1;
	}

	nfds_t nfds = 10,maxi=1;
	struct pollfd* fds = calloc(sizeof(struct pollfd),10);
	fds[0].fd = sockfd;
	fds[0].events = POLLIN;

	for(;;)
	{
		int ret = poll(fds,nfds,100000);
		if(fds[0].revents & POLLIN)
		{
			int clifd = accept(sockfd,(struct sockaddr*)&addr,&addrlen);
			if(clifd < 0) continue;
			for(int i=0; i<nfds; i++)
			{
				if(0 == fds[i].fd)
				{
					fds[i].fd = clifd;
					fds[i].events = POLLIN;
					maxi = i+1;
					break;
				}
			}
		}
		for(int i=1; i<maxi; i++)
		{
			if(fds[i].revents & (POLLIN))
			{
				char buf[4096] = {};
				int ret = recv(fds[i].fd,buf,sizeof(buf),0);
				if(ret <= 0)
				{
					fds[i].fd = 0;
					continue;
				}
				printf("clifd:%d recv:%s\n",fds[i].fd,buf);
				send(fds[i].fd,buf,strlen(buf)+1,1);
			}
		}
	}
}

客户端

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

int main()
{
	printf("服务器创建socket...\n");
	int sockfd = socket(AF_INET,SOCK_STREAM,0);
	if(0 > sockfd)
	{
		perror("socket");
		return -1;
	}

	printf("准备地址...\n");
	struct sockaddr_in addr = {};
	addr.sin_family = AF_INET;
	addr.sin_port = htons(7777);
	addr.sin_addr.s_addr = inet_addr("192.168.0.111");
	socklen_t len = sizeof(addr);

	printf("绑定连接服务器...\n");
	if(connect(sockfd,(struct sockaddr*)&addr,len))
	{
		perror("bind");
		return -1;
	}

	for(;;)
	{
			char buf[1024] = {};
			printf(">");
			gets(buf);
			write(sockfd,buf,strlen(buf)+1);
			if(0 == strcmp("quit",buf))
			{
				printf("通信结束!\n");
				break;
			}
			printf("read:");
			read(sockfd,buf,sizeof(buf));
			printf("%s\n",buf);
	}
	close(sockfd);
}