X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;ds=sidebyside;f=source%2Fnet%2Fcommunicator.cpp;h=07c3be4f6b1d7c25623a957217a98f5bccea21a6;hb=df1b7561be0d4a57e964c783c01060c54864ec60;hp=b30a8e965c388c8457b26ae454ce6c399fe5f6fc;hpb=3c2a877580e234df5fcbe06bf2850cd29f875e28;p=libs%2Fnet.git

diff --git a/source/net/communicator.cpp b/source/net/communicator.cpp
index b30a8e9..07c3be4 100644
--- a/source/net/communicator.cpp
+++ b/source/net/communicator.cpp
@@ -1,5 +1,6 @@
 #include <cstring>
 #include "communicator.h"
+#include "streamsocket.h"
 
 using namespace std;
 
@@ -9,7 +10,7 @@ using namespace Msp::Net;
 
 struct Handshake
 {
-	unsigned hash;
+	Msp::UInt64 hash;
 };
 
 
@@ -29,11 +30,11 @@ HandshakeProtocol::HandshakeProtocol():
 class HandshakeReceiver: public PacketReceiver<Handshake>
 {
 private:
-	unsigned hash;
+	Msp::UInt64 hash;
 
 public:
 	HandshakeReceiver();
-	unsigned get_hash() const { return hash; }
+	Msp::UInt64 get_hash() const { return hash; }
 	virtual void receive(const Handshake &);
 };
 
@@ -82,6 +83,26 @@ void Communicator::initiate_handshake()
 	handshake_status = 1;
 }
 
+void Communicator::send_data(unsigned size)
+{
+	if(!good)
+		throw sequence_error("connection aborted");
+	if(handshake_status!=2)
+		throw sequence_error("handshake incomplete");
+
+	try
+	{
+		socket.write(out_buf, size);
+	}
+	catch(const std::exception &e)
+	{
+		good = false;
+		if(signal_error.empty())
+			throw;
+		signal_error.emit(e);
+	}
+}
+
 void Communicator::data_available()
 {
 	if(!good)
@@ -104,15 +125,16 @@ void Communicator::data_available()
 				HandshakeReceiver hsrecv;
 				if((more = receive_packet(hsproto, hsrecv)))
 				{
+					if(handshake_status==0)
+						send_handshake();
+
 					if(hsrecv.get_hash()==protocol.get_hash())
 					{
-						if(handshake_status==0)
-							send_handshake();
 						handshake_status = 2;
 						signal_handshake_done.emit();
 					}
 					else
-						good = false;
+						throw incompatible_protocol("hash mismatch");
 				}
 			}
 		}