linux poll使用

#include <poll.h>

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

struct pollfd {
    int fd;         // 文件描述符
    short events;   // 关注的事件
    short revents;  // 发生的事件
};

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

int main() {
    int sockfd;
    int result;
    struct sockaddr_in serv_addr;
    socklen_t length;

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

    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(8080);

    result = connect(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr));
    if (result < 0) {
        perror("connect");
        exit(1);
    }

    struct pollfd fds[2];
    fds[0].fd = STDIN_FILENO;
    fds[0].events = POLLIN;
    fds[1].fd = sockfd;
    fds[1].events = POLLIN;

    while (1) {
        result = poll(fds, 2, -1);
        if (result < 0) {
            perror("poll");
            exit(1);
        }

        if (fds[0].revents & POLLIN) {
            char buf[1024];
            ssize_t n = read(STDIN_FILENO, buf, sizeof(buf));
            if (n > 0) {
                printf("stdin: %s", buf);
            }
        }

        if (fds[1].revents & POLLIN) {
            char buf[1024];
            ssize_t n = read(sockfd, buf, sizeof(buf));
            if (n > 0) {
                printf("socket: %s", buf);
            }
        }
    }

    close(sockfd);
    return 0;
}