From 4a680a89e60cd7eda7aa07302471cfe81fcfa425 Mon Sep 17 00:00:00 2001
From: Davin McCall <davmac@davmac.org>
Date: Mon, 13 Jun 2016 00:57:52 +0100
Subject: [PATCH] Use a BidiFdWatcher for control connections. One step closer
 to working with kqueue...

---
 src/control.h | 41 +++++++++++++++++++++++++++--------------
 1 file changed, 27 insertions(+), 14 deletions(-)

diff --git a/src/control.h b/src/control.h
index bf50be5..943f7b9 100644
--- a/src/control.h
+++ b/src/control.h
@@ -21,9 +21,6 @@
 using namespace dasync;
 using EventLoop_t = EventLoop<NullMutex>;
 
-// TODO: Use the input buffer as a circular buffer, instead of chomping data from
-// the front using a data move.
-
 class ControlConn;
 class ControlConnWatcher;
 
@@ -50,26 +47,47 @@ extern int active_control_conns;
 class ServiceSet;
 class ServiceRecord;
 
-class ControlConnWatcher : public PosixFdWatcher<NullMutex>
+class ControlConnWatcher : public PosixBidiFdWatcher<NullMutex>
 {
-    Rearm gotEvent(EventLoop_t * loop, int fd, int flags) override
+    inline Rearm receiveEvent(EventLoop_t * loop, int fd, int flags) noexcept;
+
+    Rearm readReady(EventLoop_t * loop, int fd) noexcept override
+    {
+        return receiveEvent(loop, fd, in_events);
+    }
+    
+    Rearm writeReady(EventLoop_t * loop, int fd) noexcept override
     {
-        control_conn_cb(loop, this, flags);
-        return Rearm::REARM;
+        return receiveEvent(loop, fd, out_events);
     }
     
     public:
     int fd; // TODO this is already stored, find a better way to access it.
+    EventLoop_t * eventLoop;
+    int watch_flags;
     
-    using PosixFdWatcher<NullMutex>::setWatchFlags;
+    void setWatchFlags(int flags) noexcept
+    {
+        PosixBidiFdWatcher<NullMutex>::setWatchFlags(eventLoop, flags);
+        watch_flags = flags;
+    }
     
     void registerWith(EventLoop_t *loop, int fd, int flags)
     {
         this->fd = fd;
-        PosixFdWatcher<NullMutex>::registerWith(loop, fd, flags);
+        PosixBidiFdWatcher<NullMutex>::registerWith(loop, fd, flags);
+        watch_flags = flags;
     }
 };
 
+inline Rearm ControlConnWatcher::receiveEvent(EventLoop_t * loop, int fd, int flags) noexcept
+{
+    PosixBidiFdWatcher<NullMutex>::setWatchFlags(loop, watch_flags);
+    control_conn_cb(loop, this, flags);
+    return Rearm::NOOP;
+}
+
+
 class ControlConn : private ServiceListener
 {
     friend void control_conn_cb(EventLoop_t *loop, ControlConnWatcher *watcher, int revents);
@@ -149,7 +167,6 @@ class ControlConn : private ServiceListener
         bad_conn_close = true;
         oom_close = true;
         iob.setWatchFlags(out_events);
-        //ev_io_set(&iob, iob.fd, EV_WRITE);
     }
     
     // Process service event broadcast.
@@ -184,11 +201,7 @@ class ControlConn : private ServiceListener
     public:
     ControlConn(EventLoop_t * loop, ServiceSet * service_set, int fd) : loop(loop), service_set(service_set), chklen(0)
     {
-        //ev_io_init(&iob, control_conn_cb, fd, EV_READ);
-        //iob.data = this;
-        //ev_io_start(loop, &iob);
         iob.registerWith(loop, fd, in_events);
-        
         active_control_conns++;
     }
     
-- 
2.25.1