Explicit refcounting removed from connection object

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

diff --git a/rpc/connection.h b/rpc/connection.h
index da48cf4..7032b8c 100644 (file)
--- a/rpc/connection.h
+++ b/rpc/connection.h
@@ -1,101 +1,93 @@
 #ifndef connection_h
 #ifndef connection_h

-#define connection_h 1
+#define connection_h

 
 

+#include "types.h"

 #include <sys/types.h>
 #include <sys/types.h>

-#include <sys/socket.h>

 #include <arpa/inet.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <netinet/in.h>

-#include <cstddef>
+#include "pollmgr.h"
+#include "file.h"

 
 

-#include <map>
+constexpr size_t size_t_max = numeric_limits<size_t>::max();

 
 

-#include "pollmgr.h"
+class thread_exit_exception : exception {};

 
 class connection;
 
 class chanmgr {
 
 class connection;
 
 class chanmgr {

-       public:
-               virtual bool got_pdu(connection *c, char *b, int sz) = 0;
-               virtual ~chanmgr() {}
+    public:
+        virtual bool got_pdu(const shared_ptr<connection> & c, const string & b) = 0;
+        virtual ~chanmgr() {}

 };
 
 };
 

-class connection : public aio_callback {
-       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
-               };
-
-               connection(chanmgr *m1, int f1, int lossytest=0);
-               ~connection();
-
-               int channo() { return fd_; }
-               bool isdead();
-               void closeconn();
-
-               bool send(char *b, int sz);
-               void write_cb(int s);
-               void read_cb(int s);
-
-               void incref();
-               void decref();
-               int ref();
-                
-                int compare(connection *another);
-       private:
-
-               bool readpdu();
-               bool writepdu();
-
-               chanmgr *mgr_;
-               const int fd_;
-               bool dead_;
-
-               charbuf wpdu_;
-               charbuf rpdu_;
-                
-                struct timeval create_time_;
-
-               int waiters_;
-               int refno_;
-               const int lossy_;
-
-               pthread_mutex_t m_;
-               pthread_mutex_t ref_m_;
-               pthread_cond_t send_complete_;
-               pthread_cond_t send_wait_;
+class connection : public aio_callback, public enable_shared_from_this<connection> {
+    public:
+        struct charbuf {
+            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() { lock ml(m_); return dead_; }
+        void closeconn();
+
+        bool send(const string & b);
+        void write_cb(int s);
+        void read_cb(int s);
+
+        time_point<steady_clock> create_time() const { return create_time_; }
+
+    private:
+
+        bool readpdu();
+        bool writepdu();
+
+        chanmgr *mgr_;
+        const file_t fd_;
+        bool dead_ = false;
+
+        charbuf wpdu_;
+        charbuf rpdu_;
+
+        time_point<steady_clock> create_time_;
+
+        int waiters_ = 0;
+        int lossy_ = 0;
+
+        mutex m_;
+        cond send_complete_;
+        cond send_wait_;

 };
 
 class tcpsconn {
 };
 
 class tcpsconn {

-       public:
-               tcpsconn(chanmgr *m1, int port, int lossytest=0);
-               ~tcpsconn();
-                inline int port() { return port_; }
-               void accept_conn();
-       private:
-                int port_;
-               pthread_mutex_t m_;
-               pthread_t th_;
-               int pipe_[2];
-
-               int tcp_; //file desciptor for accepting connection
-               chanmgr *mgr_;
-               int lossy_;
-               std::map<int, connection *> conns_;
-
-               void process_accept();
+    public:
+        tcpsconn(chanmgr *m1, in_port_t port, int lossytest=0);
+        ~tcpsconn();
+        inline in_port_t port() { return port_; }
+        void accept_conn();
+    private:
+        in_port_t port_;
+        mutex m_;
+        thread th_;
+        file_t pipe_[2];
+
+        socket_t tcp_; // listens for connections
+        chanmgr *mgr_;
+        int lossy_;
+        map<int, shared_ptr<connection>> conns_;
+
+        void process_accept();

 };
 
 struct bundle {
 };
 
 struct bundle {

-       bundle(chanmgr *m, int s, int l):mgr(m),tcp(s),lossy(l) {}
-       chanmgr *mgr;
-       int tcp;
-       int lossy;
+    bundle(chanmgr *m, int s, int l):mgr(m),tcp(s),lossy(l) {}
+    chanmgr *mgr;
+    int tcp;
+    int lossy;

 };
 
 };
 

-void start_accept_thread(chanmgr *mgr, int port, pthread_t *th, int *fd = NULL, int lossy=0);
-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
 #endif