使用poll实现echo服务器

#define POLLSIZE 100
#define BUFFERSIZE 512

char ch_char(char c) {
    if(c >= 'a' && c <= 'z')
    return c - 32;
    return c;
}


int main(int argc, char **argv){
    if(argc != 2) {
        fprintf(stderr, "usage : %s port!\n", argv[0]);
        ecit(1);
    }
    int server_listen, fd;

    if((server_listen = socket_create(atoi(argv[1]))) < 0) {
        perror("socket_create");
        exit(1);
    }

    struct pollfd event_set[POLLSIZE];

    for(int i = 0; i < POLLSIZE; i++){
        event_set[i].fd = -1;//忽略，不关注套接字
    }
    event_set[0].fd = server_listen;//0号元素专门检测listen
    event_set[0].events = POLLIN;//普通或优先级带数据可读

    while(1) {
        int retval;
        if((retval = poll(event_set, POLLSIZE, -1)) < 0){
            perror("poll");
            return 1;
        }

        if(event_set[0].revents & POLLIN){
            if((fd = accept(server_listen, NULL, NULL)) < 0){
                perror("accept");
                continue;
            }
            retval--;
            int i;
            for(i = 1; i < POLLSIZE; i++){
                if(event_set[i].fd < 0){
                    event_set[i].fd = fd;
                    event_set[i].events = POLLIN;
                    break;
                }
            }
            if(i == POLLSIZE) {
                printf("too many clients!\n");
            }
        }
        for(int i = 0; i < POLLSIZE; i++){
            if(event_set[i] < 0) continue;
            if(event_set[i].revents &(POLLIN | POLLHUP | POLLERR)) {
                retval--;
                char buffer[BUFFERSIZE] = {0};
                if(recv(event_set[i].fd, buffer, BUFFSIZE, 0)) {
                    printf("recv :%s\n", buffer);
                    for(int j = 0; j < strlen(buffer); j++) {
                        buffer[j] = ch_char(buffer[j]);
                    }
                    send(event_set[i].fd, buffer, strlen(buffer), 0);
                }else{
                    close(event_set[i].fd);
                    event_set[i].fd = -1;
                }
            }
            if(retval <= 0) break;
        }
    }


    return 0;
}