*/
+#include "digester.h"
#include <boost/asio.hpp>
+#include <boost/scoped_ptr.hpp>
/** @class Socket
* @brief A class to wrap a boost::asio::ip::tcp::socket with some things
* This class wraps some things that I could not work out how to do easily with boost;
* most notably, sync read/write calls with timeouts.
*/
-class Socket : public boost::noncopyable
+class Socket
{
public:
- Socket (int timeout = 30);
+ explicit Socket (int timeout = 30);
+
+ Socket (Socket const&) = delete;
+ Socket& operator= (Socket const&) = delete;
/** @return Our underlying socket */
boost::asio::ip::tcp::socket& socket () {
return _socket;
}
+ void set_send_buffer_size (int size);
void connect (boost::asio::ip::tcp::endpoint);
void write (uint32_t n);
void read (uint8_t* data, int size);
uint32_t read_uint32 ();
+ class ReadDigestScope
+ {
+ public:
+ ReadDigestScope (std::shared_ptr<Socket> socket);
+ bool check ();
+ private:
+ std::weak_ptr<Socket> _socket;
+ };
+
+ /** After one of these is created everything that is sent from the socket will be
+ * added to a digest. When the DigestScope is destroyed the digest will be sent
+ * from the socket.
+ */
+ class WriteDigestScope
+ {
+ public:
+ WriteDigestScope (std::shared_ptr<Socket> socket);
+ ~WriteDigestScope ();
+ private:
+ std::weak_ptr<Socket> _socket;
+ };
+
private:
- void check ();
+ friend class DigestScope;
- Socket (Socket const &);
+ void check ();
+ void start_read_digest ();
+ bool check_read_digest ();
+ void start_write_digest ();
+ void finish_write_digest ();
boost::asio::io_service _io_service;
boost::asio::deadline_timer _deadline;
boost::asio::ip::tcp::socket _socket;
int _timeout;
+ boost::scoped_ptr<Digester> _read_digester;
+ boost::scoped_ptr<Digester> _write_digester;
+ boost::optional<int> _send_buffer_size;
};