Includes cleanups

[invirt/third/libt4.git] / rpc / connection.h

diff --git a/rpc/connection.h b/rpc/connection.h
index 16b0398..b1df8a1 100644 (file)
--- a/rpc/connection.h
+++ b/rpc/connection.h
@@ -1,94 +1,82 @@
 #ifndef connection_h
 #define connection_h
 
-#include <sys/types.h>
-#include <sys/socket.h>
+#include "types.h"
 #include <arpa/inet.h>
 #include <netinet/in.h>
-#include <cstddef>
-#include <thread>
-
-#include <map>
-
 #include "pollmgr.h"
+#include "file.h"
 
-class thread_exit_exception : std::exception {
-};
+constexpr size_t size_t_max = numeric_limits<size_t>::max();
+
+class thread_exit_exception : exception {};
 
 class connection;
 
 class chanmgr {
     public:
-        virtual bool got_pdu(connection *c, char *b, int sz) = 0;
+        virtual bool got_pdu(const shared_ptr<connection> & c, const string & b) = 0;
         virtual ~chanmgr() {}
 };
 
-class connection : public aio_callback {
+class connection : public aio_callback, public enable_shared_from_this<connection> {
     public:
         struct charbuf {
-            charbuf(): buf(NULL), sz(0), solong(0) {}
-            charbuf (char *b, int s) : buf(b), sz(s), solong(0){}
-            char *buf;
-            int sz;
-            int solong; //amount of bytes written or read so far
+            string buf;
+            size_t solong = 0; // number of bytes written or read so far
         };
 
         connection(chanmgr *m1, int f1, int lossytest=0);
         ~connection();
 
         int channo() { return fd_; }
-        bool isdead();
+        bool isdead() { lock ml(m_); return dead_; }
         void closeconn();
 
-        bool send(char *b, int sz);
+        bool send(const string & b);
         void write_cb(int s);
         void read_cb(int s);
 
-        void incref();
-        void decref();
-        int ref();
+        time_point<steady_clock> create_time() const { return create_time_; }
 
-        int compare(connection *another);
     private:
 
         bool readpdu();
         bool writepdu();
 
         chanmgr *mgr_;
-        const int fd_;
-        bool dead_;
+        const file_t fd_;
+        bool dead_ = false;
 
         charbuf wpdu_;
         charbuf rpdu_;
 
-        std::chrono::time_point<std::chrono::steady_clock> create_time_;
+        time_point<steady_clock> create_time_;
 
-        int waiters_;
-        int refno_;
-        const int lossy_;
+        int waiters_ = 0;
+        int lossy_ = 0;
 
-        std::mutex m_;
-        std::mutex ref_m_;
-        std::condition_variable send_complete_;
-        std::condition_variable send_wait_;
+        mutex m_;
+        cond send_complete_;
+        cond send_wait_;
 };
 
 class tcpsconn {
     public:
-        tcpsconn(chanmgr *m1, int port, int lossytest=0);
+        tcpsconn(chanmgr *m1, in_port_t port, int lossytest=0);
         ~tcpsconn();
-        inline int port() { return port_; }
+        inline in_port_t port() { return port_; }
         void accept_conn();
     private:
-        int port_;
-        std::mutex m_;
-        std::thread th_;
-        int pipe_[2];
+        in_port_t port_;
+        mutex m_;
+        thread th_;
+        file_t pipe_[2];
 
-        int tcp_; //file desciptor for accepting connection
+        socket_t tcp_; // listens for connections
         chanmgr *mgr_;
         int lossy_;
-        std::map<int, connection *> conns_;
+        map<int, shared_ptr<connection>> conns_;
 
         void process_accept();
 };
@@ -100,5 +88,5 @@ struct bundle {
     int lossy;
 };
 
-connection *connect_to_dst(const sockaddr_in &dst, chanmgr *mgr, int lossy=0);
+shared_ptr<connection> connect_to_dst(const sockaddr_in &dst, chanmgr *mgr, int lossy=0);
 #endif