X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnet%2Fcommunicator.h;h=e9f646a59a62ba1f0796a36cf57f2d27454d849d;hb=394c9a732192fce9b3b453dfdb9e92400af2a4f8;hp=5921767e6a37ada58ba91876f5a2b80a30c3c59a;hpb=3c2a877580e234df5fcbe06bf2850cd29f875e28;p=libs%2Fnet.git diff --git a/source/net/communicator.h b/source/net/communicator.h index 5921767..e9f646a 100644 --- a/source/net/communicator.h +++ b/source/net/communicator.h @@ -1,21 +1,31 @@ #ifndef MSP_NET_COMMUNICATOR_H_ #define MSP_NET_COMMUNICATOR_H_ +#include +#include +#include +#include "mspnet_api.h" #include "protocol.h" -#include "streamsocket.h" namespace Msp { namespace Net { -class sequence_error: public std::logic_error +class StreamSocket; + +class MSPNET_API sequence_error: public invalid_state +{ +public: + sequence_error(const std::string &w): invalid_state(w) { } +}; + +class MSPNET_API incompatible_protocol: public std::runtime_error { public: - sequence_error(const std::string &w): std::logic_error(w) { } - virtual ~sequence_error() throw() { } + incompatible_protocol(const std::string &w): std::runtime_error(w) { } }; -class Communicator +class MSPNET_API Communicator: public NonCopyable { public: sigc::signal signal_handshake_done; @@ -25,13 +35,13 @@ private: StreamSocket &socket; const Protocol &protocol; ReceiverBase &receiver; - int handshake_status; - unsigned buf_size; - char *in_buf; - char *in_begin; - char *in_end; - char *out_buf; - bool good; + int handshake_status = 0; + std::size_t buf_size = 65536; + char *in_buf = nullptr; + char *in_begin = nullptr; + char *in_end = nullptr; + char *out_buf = nullptr; + bool good = true; public: Communicator(StreamSocket &, const Protocol &, ReceiverBase &); @@ -41,32 +51,22 @@ public: bool is_handshake_done() const { return handshake_status==2; } template - void send(const P &pkt) - { - if(!good) - throw sequence_error("connection aborted"); - if(handshake_status!=2) - throw sequence_error("handshaking not done"); - unsigned size = protocol.serialize(pkt, out_buf, buf_size); - try - { - socket.write(out_buf, size); - } - catch(const std::exception &e) - { - good = false; - if(signal_error.empty()) - throw; - signal_error.emit(e); - } - } + void send(const P &); private: + void send_data(std::size_t); + void data_available(); bool receive_packet(const Protocol &, ReceiverBase &); void send_handshake(); }; +template +void Communicator::send(const P &pkt) +{ + send_data(protocol.serialize(pkt, out_buf, buf_size)); +} + } // namespace Net } // namespace Msp