]> git.tdb.fi Git - libs/net.git/blobdiff - source/net/communicator.cpp
Use standard fixed-width integer types
[libs/net.git] / source / net / communicator.cpp
index 07c3be4f6b1d7c25623a957217a98f5bccea21a6..c473d53a7e06a5ead67d4e8ca76cdbd826cc478d 100644 (file)
@@ -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
 {
-       Msp::UInt64 hash;
+       uint64_t hash;
 };
 
 
@@ -30,11 +31,11 @@ HandshakeProtocol::HandshakeProtocol():
 class HandshakeReceiver: public PacketReceiver<Handshake>
 {
 private:
-       Msp::UInt64 hash;
+       uint64_t hash;
 
 public:
        HandshakeReceiver();
-       Msp::UInt64 get_hash() const { return hash; }
+       uint64_t 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),
@@ -83,7 +84,7 @@ void Communicator::initiate_handshake()
        handshake_status = 1;
 }
 
-void Communicator::send_data(unsigned size)
+void Communicator::send_data(size_t size)
 {
        if(!good)
                throw sequence_error("connection aborted");
@@ -116,9 +117,7 @@ void Communicator::data_available()
                while(more)
                {
                        if(handshake_status==2)
-                       {
                                more = receive_packet(protocol, receiver);
-                       }
                        else
                        {
                                HandshakeProtocol hsproto;
@@ -162,7 +161,7 @@ bool Communicator::receive_packet(const Protocol &proto, ReceiverBase &recv)
        {
                if(in_end==in_buf+buf_size)
                {
-                       unsigned used = in_end-in_begin;
+                       size_t used = in_end-in_begin;
                        memmove(in_buf, in_begin, used);
                        in_begin = in_buf;
                        in_end = in_begin+used;
@@ -177,7 +176,7 @@ void Communicator::send_handshake()
        shake.hash = protocol.get_hash();
 
        HandshakeProtocol hsproto;
-       unsigned size = hsproto.serialize(shake, out_buf, buf_size);
+       size_t size = hsproto.serialize(shake, out_buf, buf_size);
        socket.write(out_buf, size);
 }