echod_epoll.c

レベルトリガとして使用。まあ、高速なpoll(2)ですね。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <unistd.h>

#include <sys/epoll.h>

#include <errno.h>

#define ECHO_PORT 7
#define MAX_BACKLOG 5
#define RCVBUFSIZE 256
#define MAX_EVENTS 1024
#define EPOLL_TIMEOUT 1000

#define die(s) do { fprintf(stderr, "%s\n", (s)); exit(1); } while(0)
#define die_with_err(s) do { perror((s)); exit(1); } while(0)

#ifdef DEBUG
#define debug(...) do { fprintf(stderr,  __VA_ARGS__); } while(0)
#else
#define debug(...) do {} while(0)
#endif

int echo(int sock) {
  char buf[RCVBUFSIZE + 1];
  size_t len;

  debug("echo(): %d.\n", sock);

  if ((len = recv(sock, buf, RCVBUFSIZE, 0)) < 0) {
    perror("recv(2)");
    return -1;
  }

  if (len == 0) {
    return 0;
  }

  buf[len] = '\0';
  debug("recv: %s.\n", buf);

  if (send(sock, buf, len, 0) < 0) {
    perror("send(2)");
    return -1;
  }

  return len;
}

int main() {
  int sock, epfd;
  struct sockaddr_in addr;
  struct epoll_event event;

  if ((sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
    die_with_err("socket(2)");
  }

  memset(&addr, 0, sizeof(addr));
  addr.sin_family = AF_INET;
  addr.sin_addr.s_addr = htonl(INADDR_ANY);
  addr.sin_port = htons(ECHO_PORT);

  if (bind(sock, (struct sockaddr*) &addr, sizeof(addr)) < 0) {
    die_with_err("bind(2)");
  }

  if (listen(sock, MAX_BACKLOG) < 0) {
    die_with_err("listen(2)");
  }

  if((epfd = epoll_create(MAX_EVENTS)) < 0) {
    die_with_err("epoll_create(2)");
  }

  memset(&event, 0, sizeof(event));
  event.events  = EPOLLIN; // レベルトリガ
  event.data.fd = sock;

  if (epoll_ctl(epfd, EPOLL_CTL_ADD, sock, &event) < 0) {
    die_with_err("epoll_ctl(2)");
  }

  while (1) {
    int i, nfds;
    struct sockaddr_in sa;
    socklen_t len = sizeof(sa);
    struct epoll_event events[MAX_EVENTS];

    if ((nfds = epoll_wait(epfd, events, MAX_EVENTS, EPOLL_TIMEOUT)) < 0) { // nfdsはeventsの数になるのかな？
      die_with_err("epoll_wait(2)");
    }

    for (i = 0; i < nfds; i++) {
      int fd = events[i].data.fd;

      if (sock == fd) {
        int s;

        if ((s = accept(sock, (struct sockaddr*) &sa, &len)) < 0) {
          if (EINTR == errno) { continue; }
          die_with_err("accept(2)");
        }

        debug("accepted.\n");

        // 再利用していいんだろうか…
        event.events  = EPOLLIN; // レベルトリガ
        event.data.fd = s;

        if (epoll_ctl(epfd, EPOLL_CTL_ADD, s, &event) < 0) {
          die_with_err("epoll_ctl(2)");
        }
      } else {
        if (echo(fd) < 1) {
          if (epoll_ctl(epfd, EPOLL_CTL_DEL, fd, &event) < 0) {
            die_with_err("epoll_ctl(2)");
          }
          // EPOLL_CTL_DEL
          //  対象ファイルディスクリプタ fd を epoll ファイルディスクリプタ epfd から削除する。
          //  event は無視されるので、NULL に設定することができる (但し、下記の「バグ」の章を参照のこと)。
          //  (http://www.linux.or.jp/JM/html/LDP_man-pages/man2/epoll_ctl.2.html)

          close(fd); // close(2)はepfdからfdを削除してから
        }
      }
    }
  }

  close(sock);
}