EpollTest2.cpp

/*
 * =====================================================================================
 *
 *       Filename:  EpollTest.cpp
 *
 *    Description:
 *
 *        Version:  1.0
 *        Created:  08/31/2017 02:46:29 PM
 *  Last Modified:  08/31/2017 02:46:29 PM
 *       Revision:  none
 *       Compiler:  gcc
 *
 *         Author:  zt (),
 *   Organization:
 *
 * =====================================================================================
 */

#include <stdio.h>
#include <stdlib.h>
#include <sys/epoll.h>
#include <map>
#include <memory>
#include <iostream>

#include "Network.cpp"

#define MAX_CNT   (2048)

typedef int ( *Fun ) ( int fd );

struct my_epoll_t
{
  my_epoll_t ( int _fd, Fun _pFun )
  {
    fd = _fd;
    pFun = _pFun;
  }
  ~my_epoll_t()
  {
    shutdown ( fd, SHUT_RDWR );
    close ( fd );
  }

  int fd;
  Fun pFun;
};

int F1 ( int fd )
{
  int ret = 0;
  char buffer[2048];
  ret = read ( fd, buffer, 2048 );
  //ret = Network::Recvn ( fd, buffer, 2048 );

  if ( ret > 0 )
    printf ( "fd=%d recv=%d %s\n", fd, ret, buffer );

  return ret;
}

int F2 ( int fd )
{
  char buffer[2048];
  return Network::Sendn ( fd, buffer, 2048 );
}

int main ( int argc, char* argv[] )
{
  int listensock = Network::MakeTCPServer ( 12345 );
  int ret = -1;

  epoll_event ev;
  epoll_event events[MAX_CNT];

  int epollfd = epoll_create1 ( 0 );

  ev.data.fd = listensock;
  ev.events = EPOLLIN;
  ret = epoll_ctl ( epollfd, EPOLL_CTL_ADD, listensock, &ev );
  printf ( "epoll_ctl, add listensock ret %d\n", ret );

  int nfd = 0;

  while ( 1 )
  {
    nfd = epoll_wait ( epollfd, events, MAX_CNT, 1000 );

    for ( int i = 0; i < nfd; ++i )
    {
      if ( events[i].data.fd == listensock )
      {
        sockaddr saddr;
        int sk = Network::MakeAccept ( listensock, &saddr );
        Network::SetNonBlock ( sk );

        ev.data.ptr = new my_epoll_t ( sk, F1 );

        ev.events = EPOLLIN | EPOLLET;
        ret = epoll_ctl ( epollfd, EPOLL_CTL_ADD, sk, &ev );

        printf ( "incomming connect %d %s ret=%d \n",
                 sk, inet_ntoa ( reinterpret_cast<sockaddr_in*> ( &saddr )->sin_addr ), ret );
      }
      else
      {
        my_epoll_t* p = reinterpret_cast< my_epoll_t*> ( events[i].data.ptr );
        ret = p->pFun ( p->fd );

        if ( ret <= 0 )
        {
          ev.data.fd = p->fd;
          ev.events = 0;
          epoll_ctl ( epollfd, EPOLL_CTL_DEL, p->fd, &ev );
          delete p;

          printf ( "line=%d del connect %d ret=%d\n",
                   __LINE__, ev.data.fd, ret );
        }
        else
        {
          //printf ( "line=%d ret=%d\n", __LINE__, ret );
        }
      }
    }
  }

  close ( epollfd );
  close ( listensock );

  return 0;
}