poll服务端程序例子

以下程序是poll服务端程序。

注意：如果是自己判断状态会有问题，现在还不清楚是哪里的问题。

#include <stdlib.h>
#include <stdio.h>
#include <iostream>
#include <string>
#include <fstream>
#include    <cstring>
#include    <errno.h>
#include    <sys/socket.h>
#include    <sys/types.h>
#include    <sys/time.h>
#include    <arpa/inet.h>
#include    <sys/select.h>
#include    <poll.h>
#include    <unistd.h>
#include    <fcntl.h>
#include    "Define.h"
using namespace std;
/**
 * @brief
 */
void do_read(CONN* conn);
void do_send(CONN* conn);
#ifdef SELECT
int select_fd_queue[1024]; // fd---->queueIndex
int select_queue_fd[1024]; // queueIndex -> fd
#endif
struct pollfd poll_queue_fd [1024];
int poll_fd_queue[1024];

int queue_num = 0;
CONN conn[1024];
int add_fd(int fd, int rw )
{
    poll_queue_fd[queue_num].fd = fd;
    poll_fd_queue[fd] = queue_num;
    switch ( rw )
    {
        case FDW_READ: poll_queue_fd[queue_num].events = POLLIN; break;
        case FDW_WRITE: poll_queue_fd[queue_num].events = POLLOUT; break;
        default: break;
    }

    queue_num++;
}
int del_fd(int fd)
{
    if(queue_num > 1){
        queue_num--;
        poll_queue_fd[poll_fd_queue[fd]].fd = poll_queue_fd[queue_num].fd;
        poll_fd_queue[ poll_queue_fd[queue_num].fd ]  =  poll_fd_queue[fd];
        poll_queue_fd[queue_num].fd = -1;
        poll_fd_queue[fd] = -1;

    }else{
        poll_queue_fd[poll_fd_queue[fd]].fd = -1;
        queue_num--;
    }
}
 int main ( int argc, char *argv[] )
{
    int listenfd, connectfd ;
    struct sockaddr_in serveaddr, clientaddr;
    int ret = 0;
    char buff[64]={0};
    int maxfd = -1;

    bzero(&serveaddr, sizeof(serveaddr));
    serveaddr.sin_family = AF_INET;
    serveaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    serveaddr.sin_port = htons(atoi(argv[1]));

    listenfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if(listenfd < 0)
    {
        printf ( "listen fd < 0\n" );
        exit(0);
    }

    ret = bind(listenfd, (struct sockaddr*)&serveaddr, sizeof(serveaddr));
    if(ret < 0){
        printf ( "bind error\n" );
        exit(0);
    }

    ret = listen(listenfd,1024);

    if ( fcntl( listenfd, F_SETFL,  O_NDELAY ) < 0 )
    {

        printf ( "fcntl error = [%s]\n", strerror(errno) );
    }
    struct timeval tm;
    tm.tv_usec = 1;
    tm.tv_sec = 10;
    add_fd(listenfd, FDW_READ);
                printf ( "queueNum = [%d]\n", queue_num);
    
    for ( ; ; )     
    {
        int nready= poll(poll_queue_fd, queue_num,3000);

        if( poll_queue_fd[0].revents & POLLIN )
        {
            ret = sizeof(clientaddr);
            connectfd = accept(listenfd, (struct sockaddr*)&clientaddr, (socklen_t*)&ret);
            printf ( "connectfd = [%d]\n", connectfd );
            add_fd(connectfd, FDW_READ);
            bzero(&conn[connectfd], sizeof(conn[0]));
            conn[connectfd].fd = connectfd;
            conn[connectfd].conn_state = CNST_READING;
            if ( fcntl( connectfd, F_SETFL,  O_NDELAY ) < 0 )
            {

                printf ( "fcntl error = [%s]\n", strerror(errno) );
            }
            if(-- nready <= 0)
                continue;
        }
//        for ( int i = 1; i < queue_num; i++) {
//            printf ( "fd = [%d] | %d %d | %d %d |\n", poll_queue_fd[i].fd, conn[poll_queue_fd[i].fd].read_off, conn[poll_queue_fd[i].fd].read_size, conn[poll_queue_fd[i].fd].write_off, conn[poll_queue_fd[i].fd].write_size);
            if(queue_num > 1)
                printf ( "queueNum = [%d] error = [%s]\n", queue_num, strerror(errno));
//        }

        for ( int i = 1; i < queue_num; i++) {
    //        if(conn[poll_queue_fd[i].fd].conn_state == CNST_READING )
    //        {
                if(poll_queue_fd[i].revents  & POLLIN ){
                    do_read(&conn[poll_queue_fd[i].fd]);
                    if(conn[poll_queue_fd[i].fd].conn_state == CNST_SENDING)
                    {
                        poll_queue_fd[i].events = POLLOUT;
                    }
                }

///            }else if(conn[poll_queue_fd[i].fd].conn_state == CNST_SENDING)
//            {
                if(poll_queue_fd[i].revents  & POLLOUT ){
            //        printf ( "fd=[%d] len=[%d]\n", poll_queue_fd[i].fd , conn[poll_queue_fd[i].fd].read_size );
                    do_send(&conn[poll_queue_fd[i].fd]);
                }

//            }else
            if(conn[poll_queue_fd[i].fd].conn_state == CNST_PAUSING){
                bzero(&conn[poll_queue_fd[i].fd], sizeof(conn[0]));
                close(poll_queue_fd[i].fd);
                del_fd(poll_queue_fd[i].fd);
            }/* else{
                printf ( "Error\n" );

            } */

        }

    }
    return EXIT_SUCCESS;
}                // ----------  end of function main  ----------
void do_read(CONN* conn)
{
    conn->read_off = read(conn->fd, &conn->read_buf[conn->read_size], 100);
    if(conn->read_off > 0)
        conn->read_size += conn->read_off;
    if(strstr(conn->read_buf, "#|#hanshaoenIloveYou#|") != NULL)
    {
        conn->conn_state = CNST_SENDING;
        strcpy(conn->write_buf, "Hanwei I love you forever !!!!!");
        conn->write_size = strlen(conn->write_buf);
        conn->write_off = 0;
    }

}
void do_send(CONN* conn)
{
        if( conn->write_off < conn->write_size)
        {
            int rt = write(conn->fd, &conn->write_buf[conn->write_off], conn->write_size-conn->write_off);
            if(rt > 0){
                conn->write_off += rt;
            }
        }
        if(conn->write_off == conn->write_size)
            conn->conn_state = CNST_PAUSING;

}