X-Git-Url: http://git.tdb.fi/?p=libs%2Fnet.git;a=blobdiff_plain;f=source%2Fnet%2Fcommunicator.cpp;h=3d6353a2158fb2d62fbe6df164085a1b2ea7a417;hp=92279c24e642bc7c6f9c2b80eba60f69d1b9c8a2;hb=2dfa05663dd67d4d7c68f96df0b1ab733b2063c2;hpb=400f3e031834eee9d96dbd5cda25655102b75805 diff --git a/source/net/communicator.cpp b/source/net/communicator.cpp index 92279c2..3d6353a 100644 --- a/source/net/communicator.cpp +++ b/source/net/communicator.cpp @@ -10,7 +10,7 @@ using namespace Msp::Net; struct Handshake { - unsigned hash; + Msp::UInt64 hash; }; @@ -30,11 +30,11 @@ HandshakeProtocol::HandshakeProtocol(): class HandshakeReceiver: public PacketReceiver { 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 +58,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 +116,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"); } } }