trigger.cc

#include "trigger.h"

#include <lace/try.h>
#include "reactor.h"

#include <cassert>
#include <unistd.h>

trigger::trigger(int fd)
  : fd_(fd), r_(NULL)
{
  ev_.events = EPOLLONESHOT;
  ev_.data.ptr = NULL;
}

trigger::trigger(const file_descriptor & fd)
  : fd_(fd), r_(NULL)
{
  ev_.events = EPOLLONESHOT;
  ev_.data.ptr = NULL;
}

trigger::~trigger() { }

int trigger::fd() const { return fd_; }
int trigger::dup() const { return file_descriptor::dup(fd_); }

basic_context *
trigger::wait_for_read() {
  arm();
  ev_.events = EPOLLRDHUP|EPOLLIN|EPOLLONESHOT;
  TRY(epoll_ctl, r_->fd, EPOLL_CTL_MOD, fd_, &ev_);
  return static_cast<basic_context*>(this);
}

basic_context *
trigger::wait_for_write() {
  arm();
  ev_.events = EPOLLOUT|EPOLLONESHOT;
  TRY(epoll_ctl, r_->fd, EPOLL_CTL_MOD, fd_, &ev_);
  return static_cast<basic_context*>(this);
}

basic_context *
trigger::wait_for_error() {
  arm();
  ev_.events = EPOLLRDHUP|EPOLLONESHOT;
  TRY(epoll_ctl, r_->fd, EPOLL_CTL_MOD, fd_, &ev_);
  return static_cast<basic_context*>(this);
}

basic_context *
trigger::wait_for_nothing() {
  disarm();
  ev_.events = EPOLLONESHOT;
  TRY(epoll_ctl, r_->fd, EPOLL_CTL_MOD, fd_, &ev_);
  return static_cast<basic_context*>(this);
}

basic_context *
trigger::wait_for_anything() {
  arm();
  ev_.events = EPOLLRDHUP|EPOLLIN|EPOLLOUT|EPOLLONESHOT;
  TRY(epoll_ctl, r_->fd, EPOLL_CTL_MOD, fd_, &ev_);
  return static_cast<basic_context*>(this);
}

bool
trigger::readable() const {
  assert(!armed());
  return ev_.events & EPOLLIN;
}

bool
trigger::writable() const {
  assert(!armed());
  return ev_.events & EPOLLOUT;
}

basic_context *
trigger::fire() {
  assert(armed());
  disarm();
  return static_cast<basic_context*>(this);
}

void
trigger::arm() {
  assert(!armed());
  ev_.data.ptr = this;
  if (!r_) {
    r_ = &reactor::instance();
    TRY_ERR(EPERM, epoll_ctl, r_->fd, EPOLL_CTL_ADD, fd_, &ev_);
  }
  ++r_->armed_triggers;
  assert(armed());
}

void
trigger::disarm() {
  assert(armed());
  ev_.data.ptr = NULL;
  assert(r_);
  --r_->armed_triggers;
  assert(!armed());
}

bool
trigger::armed() const {
  return ev_.data.ptr == static_cast<const trigger*>(this);
}

//