libevでechoサーバを作る

C言語

パフォーマンスがよいよいと評判のlibevでechoサーバを書いてみる。

echod_libev.c

#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 <fcntl.h>

#include <errno.h>

#include <ev.h>

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

#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;
}

void event_client(EV_P_ struct ev_io *w, int revents) {
  if (echo(w->fd) < 1) {
    close(w->fd);
    ev_io_stop(EV_A_ w);
    free(w);
  }
}

void event_server(EV_P_ struct ev_io *w, int revents) {
  int sock, flags;
  struct sockaddr_in addr;
  socklen_t len = sizeof(addr);
  struct ev_loop *l;
  ev_io *client_watcher;

  if ((sock = accept(w->fd, (struct sockaddr*) &addr, &len)) < 0) {
    if (EINTR == errno) { return; }
    die_with_err("accept(2)");
  }

  debug("accepted.\n");

  if ((flags = fcntl(sock, F_GETFL, 0)) < 0 || fcntl(sock, F_SETFL, flags | O_NONBLOCK) < 0) {
    die_with_err("fcntl(2)");
  }

  client_watcher = calloc(1, sizeof(ev_io));
  l = w->data;
  ev_io_init(client_watcher, event_client, sock, EV_READ);
  ev_io_start(l, client_watcher);
}

int main() {
  int sock, epfd, flags;
  struct sockaddr_in addr;
  struct ev_loop *loop;
  ev_io watcher;

  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 ((flags = fcntl(sock, F_GETFL, 0)) < 0 || fcntl(sock, F_SETFL, flags | O_NONBLOCK) < 0) {
    die_with_err("fcntl(2)");
  }

  loop = ev_default_loop(0);
  watcher.data = loop;
  ev_io_init(&watcher, event_server, sock, EV_READ);
  ev_io_start(loop, &watcher);
  ev_loop(loop, 0);

  close(sock);

  return 0;
}

所感

  • APIの設計はlibeventとほとんど同じ。なので乗り換えは楽かも
  • ドキュメントが充実しているのは素晴らしい
    • ドキュメントを読んでみると、いろいろとできそうな感じがする