io 多路复用之 poll

客户端

#include<unistd.h>
#include<arpa/inet.h>
#include<iostream>
#include<string.h>
using namespace std;
int main()
{
	//创建socket
	int fd = socket(AF_INET, SOCK_STREAM, 0);
	if (fd == -1)
	{
		perror("socket error");
		exit(-1);

	}

	struct sockaddr_in seraddr;
	inet_pton(AF_INET, "192.168.76.128", &seraddr.sin_addr.s_addr); //或 127.0.0.1
	seraddr.sin_family = AF_INET;
	seraddr.sin_addr.s_addr = INADDR_ANY;
	seraddr.sin_port = htons(9999);

	//连接服务器
	int ret = connect(fd, (struct sockaddr*)&seraddr, sizeof(seraddr));
	if (ret == -1)
	{
		perror("connect error");
		exit(-1);
	}

	int num = 0;
	while (1)
	{
		char sendBuf[1024] = { 0 };
		 
		sprintf(sendBuf, "send data :%d", num++);
		write(fd, sendBuf, strlen(sendBuf) + 1);
		//接收
		int len = read(fd, sendBuf, sizeof(sendBuf));
		if (len == -1)
		{
			perror("read error");
			exit(-1);

		}
		else if (len > 0)
		{
			cout << "read buf =" << sendBuf << endl;
		}
		else
		{
			cout << "服务器已经断开连接..." << endl;
			break;
		}
		sleep(1);
	}
	close(fd); 
	return 0;
}

服务端

#include <arpa/inet.h>
#include <unistd.h>

#include<iostream>
#include <string.h>
#include <poll.h>

using namespace std;
int main() {

    // 创建 socket
    int lfd = socket(PF_INET, SOCK_STREAM, 0);
    if (lfd == -1)
    {
        perror("socket error");
        exit(-1);
    }
    struct sockaddr_in saddr;
    saddr.sin_family = AF_INET;
    saddr.sin_port = htons(9999);
    saddr.sin_addr.s_addr = INADDR_ANY;

    // 绑定
    bind(lfd, (struct sockaddr*)&saddr, sizeof(saddr));

    // 监听
    listen(lfd, 8);

    // 初始化检测的文件描述符数组
    struct pollfd fds[1024];  //可以不用 1024，数字表示客户端的个数
    for (int i = 0; i < 1024; ++i) 
    {   //初始化
        fds[i].fd = -1; //表示可用
        fds[i].events = POLLIN; //监听读事件
    }
    fds[0].fd = lfd;
    int nfds = 0;

    while (1) {

        // 调用poll系统函数，让内核帮检测哪些文件描述符有数据
        int ret = poll(fds, nfds + 1, -1);
        if (ret == -1) {
            perror("poll error");
            exit(-1);
        }
        else if (ret == 0) {
            continue;
        }
        else if (ret > 0) {
            // 说明检测到了有文件描述符的对应的缓冲区的数据发生了改变
            if (fds[0].revents & POLLIN) {
                // 表示有新的客户端连接进来了
                struct sockaddr_in cliaddr;
                socklen_t len = sizeof(cliaddr);
                int cfd = accept(lfd, (struct sockaddr*)&cliaddr, &len);

                // 将新的文件描述符加入到集合中
                for (int i = 1; i < 1024; i++) {
                    if (fds[i].fd == -1) {
                        fds[i].fd = cfd;
                        fds[i].events = POLLIN;
                        break;
                    }
                }

                // 更新最大的文件描述符的索引
                nfds = nfds > cfd ? nfds : cfd;
            }

            for (int i = 1; i <= nfds; i++) {
                if (fds[i].revents & POLLIN) {
                    // 说明这个文件描述符对应的客户端发来了数据
                    char buf[1024] = { 0 };
                    int len = read(fds[i].fd, buf, sizeof(buf));
                    if (len == -1) 
                    {
                        perror("read error");
                        exit(-1);
                    }
                    else if (len == 0) 
                    {
                        cout << "client closed...(客户端断开连接)" << endl;
                        close(fds[i].fd);//关闭当前文件描述符后，该位置置为可用
                        fds[i].fd = -1;
                    }
                    else if (len > 0) {
                        cout << "read buf = " << buf << endl;
                        write(fds[i].fd, buf, strlen(buf) + 1);
                    }
                }
            }

        }

    }
    close(lfd); //关闭监听文件描述符
    return 0;
}