linux poll实例

#include <poll.h>

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

struct pollfd {
    int fd;         // 文件描述符
    short events;   // 请求的事件
    short revents;  // 返回的事件
};

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

#define MAX_FDS 2

int main() {
    int sockfd1, sockfd2;
    struct sockaddr_in serv_addr1, serv_addr2;
    socklen_t length;

    if ((sockfd1 = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        perror("socket");
        exit(1);
    }

    if ((sockfd2 = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        perror("socket");
        exit(1);
    }

    // 设置服务器地址和端口...

    struct pollfd fds[MAX_FDS];
    fds[0].fd = sockfd1;
    fds[0].events = POLLIN;
    fds[1].fd = sockfd2;
    fds[1].events = POLLIN;

    int ret = poll(fds, MAX_FDS, -1);
    if (ret < 0) {
        perror("poll");
        exit(1);
    }

    for (int i = 0; i < MAX_FDS; i++) {
        if (fds[i].revents & POLLIN) {
            char buf[1024];
            ssize_t n = read(fds[i].fd, buf, sizeof(buf));
            if (n < 0) {
                perror("read");
                exit(1);
            }
            printf("Received data from socket %d: %s
", fds[i].fd, buf);
        }
    }

    close(sockfd1);
    close(sockfd2);

    return 0;
}