libt2n: (tomj) small fix to support data transfers over 2GB

[libt2n] / src / command_client.hxx

diff --git a/src/command_client.hxx b/src/command_client.hxx
index 257b0a8..59d4755 100644 (file)
--- a/src/command_client.hxx
+++ b/src/command_client.hxx
@@ -19,6 +19,9 @@
 #ifndef __LIBT2N_COMMAND_CLIENT
 #define __LIBT2N_COMMAND_CLIENT
 
+#include <functional>
+#include <string>
+
 #include "client.hxx"
 #include "container.hxx"
 
@@ -33,7 +36,7 @@ class command_client
         static const long long hello_timeout_usec_default=30000000;
 
     private:
-        client_connection &c;
+        client_connection *c;
 
         long long hello_timeout_usec;
         long long command_timeout_usec;
@@ -42,12 +45,16 @@ class command_client
         std::string read_packet(const long long &usec_timeout);
         bool check_hello(const std::string& hellostr);
 
+        std::auto_ptr<t2n_exception> constructorException;
+
     public:
-        command_client(client_connection& _c,
+        command_client(client_connection* _c,
             long long _command_timeout_usec=command_timeout_usec_default,
             long long _hello_timeout_usec=hello_timeout_usec_default);
         virtual ~command_client() {}
 
+        void replace_connection(client_connection* _c);
+
         void send_command(command* cmd, result_container &res);
 
         void set_command_timeout_usec(long long _command_timeout_usec=command_timeout_usec_default)
@@ -58,6 +65,10 @@ class command_client
             { return command_timeout_usec; }
         long long get_hello_timeout_usec(void)
             { return hello_timeout_usec; }
+        bool is_connection_closed(void)
+            { return c->is_closed(); }
+        t2n_exception* get_constuctor_exception(void)
+            { return constructorException.get(); }
 };
 
 }