1 files changed, 89 insertions, 0 deletions
diff --git a/src/networktransport.h b/src/networktransport.h
new file mode 100644
index 0000000..0a9d11b
--- /dev/null
+++ b/src/networktransport.h
@@ -0,0 +1,89 @@
+#ifndef NETWORK_TRANSPORT_HPP
+#define NETWORK_TRANSPORT_HPP
+
+#include <string>
+#include <signal.h>
+#include <time.h>
+#include <list>
+#include <vector>
+
+#include "network.h"
+#include "transportsender.h"
+#include "transportfragment.h"
+
+using namespace std;
+
+namespace Network {
+  template <class MyState, class RemoteState>
+  class Transport
+  {
+  private:
+    /* the underlying, encrypted network connection */
+    Connection connection;
+
+    /* sender side */
+    TransportSender<MyState> sender;
+
+    /* helper methods for recv() */
+    void process_throwaway_until( uint64_t throwaway_num );
+
+    /* simple receiver */
+    list< TimestampedState<RemoteState> > received_states;
+    RemoteState last_receiver_state; /* the state we were in when user last queried state */
+    uint64_t sent_state_late_acked;
+    FragmentAssembly fragments;
+    bool verbose;
+
+  public:
+    Transport( MyState &initial_state, RemoteState &initial_remote, const char *desired_ip );
+    Transport( MyState &initial_state, RemoteState &initial_remote,
+	       const char *key_str, const char *ip, int port );
+
+    /* Send data or an ack if necessary. */
+    void tick( void ) { sender.tick(); }
+
+    /* Returns the number of ms to wait until next possible event. */
+    int wait_time( void ) { return sender.wait_time(); }
+
+    /* Blocks waiting for a packet. */
+    void recv( void );
+
+    /* Find diff between last receiver state and current remote state, then rationalize states. */
+    string get_remote_diff( void );
+
+    /* Shut down other side of connection. */
+    /* Illegal to change current_state after this. */
+    void start_shutdown( void ) { sender.start_shutdown(); }
+    bool shutdown_in_progress( void ) const { return sender.get_shutdown_in_progress(); }
+    bool shutdown_acknowledged( void ) const { return sender.get_shutdown_acknowledged(); }
+    bool shutdown_ack_timed_out( void ) const { return sender.shutdown_ack_timed_out(); }
+    bool attached( void ) const { return connection.get_attached(); }
+
+    /* Other side has requested shutdown and we have sent one ACK */
+    bool counterparty_shutdown_ack_sent( void ) const { return sender.get_counterparty_shutdown_acknowledged(); }
+
+    int port( void ) const { return connection.port(); }
+    string get_key( void ) const { return connection.get_key(); }
+
+    MyState &get_current_state( void ) { return sender.get_current_state(); }
+    void set_current_state( const MyState &x ) { sender.set_current_state( x ); }
+
+    uint64_t get_remote_state_num( void ) const { return received_states.back().num; }
+
+    const TimestampedState<RemoteState> & get_latest_remote_state( void ) const { return received_states.back(); }
+
+    int fd( void ) const { return connection.fd(); }
+
+    void set_verbose( void ) { sender.set_verbose(); verbose = true; }
+
+    void set_send_delay( int new_delay ) { sender.set_send_delay( new_delay ); }
+
+    uint64_t get_sent_state_acked( void ) const { return sender.get_sent_state_acked(); }
+    uint64_t get_sent_state_last( void ) const { return sender.get_sent_state_last(); }
+    uint64_t get_sent_state_late_acked( void ) const { return sent_state_late_acked; }
+
+    unsigned int send_interval( void ) const { return sender.send_interval(); }
+  };
+}
+
+#endif