X-Git-Url: http://git.tdb.fi/?p=libs%2Fnet.git;a=blobdiff_plain;f=source%2Fnet%2Fcommunicator.cpp;h=22a79631d9a214741fdbf71721d8e6322986ea4e;hp=92279c24e642bc7c6f9c2b80eba60f69d1b9c8a2;hb=8248f2912454229f4ded72b8fb4db3e76ea3ad15;hpb=400f3e031834eee9d96dbd5cda25655102b75805

diff --git a/source/net/communicator.cpp b/source/net/communicator.cpp
index 92279c2..22a7963 100644
--- a/source/net/communicator.cpp
+++ b/source/net/communicator.cpp
@@ -1,5 +1,6 @@
 #include <cstring>
 #include "communicator.h"
+#include "protocol_impl.h"
 #include "streamsocket.h"
 
 using namespace std;
@@ -10,7 +11,7 @@ using namespace Msp::Net;
 
 struct Handshake
 {
-	unsigned hash;
+	Msp::UInt64 hash;
 };
 
 
@@ -30,11 +31,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 &);
 };
 
@@ -58,7 +59,7 @@ Communicator::Communicator(StreamSocket &s, const Protocol &p, ReceiverBase &r):
 	protocol(p),
 	receiver(r),
 	handshake_status(0),
-	buf_size(1024),
+	buf_size(65536),
 	in_buf(new char[buf_size]),
 	in_begin(in_buf),
 	in_end(in_buf),
@@ -116,24 +117,23 @@ void Communicator::data_available()
 		while(more)
 		{
 			if(handshake_status==2)
-			{
 				more = receive_packet(protocol, receiver);
-			}
 			else
 			{
 				HandshakeProtocol hsproto;
 				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");
 				}
 			}
 		}