aa6882b3b7
and writes. This was previously unsafe because we almost always passed read/write buffers to boost that were on the stack. Canceling the task deleted the stack and therefore the buffer, but couldn't reliably prevent boost from writing to the buffer if data came in after the cancel. This commit adds variants of the read and write functions that take a shared_ptr<char> instead of a raw char* as the buffer, and these variants will ensure the shared_ptr will outlive the boost::asio read/write.
85 lines
2.1 KiB
C++
85 lines
2.1 KiB
C++
#pragma once
|
|
#include <fc/utility.hpp>
|
|
#include <fc/fwd.hpp>
|
|
#include <fc/io/iostream.hpp>
|
|
#include <fc/time.hpp>
|
|
|
|
namespace fc {
|
|
namespace ip { class endpoint; }
|
|
|
|
class tcp_socket_io_hooks;
|
|
|
|
class tcp_socket : public virtual iostream
|
|
{
|
|
public:
|
|
tcp_socket();
|
|
~tcp_socket();
|
|
|
|
void connect_to( const fc::ip::endpoint& remote_endpoint );
|
|
void bind( const fc::ip::endpoint& local_endpoint );
|
|
void enable_keep_alives(const fc::microseconds& interval);
|
|
void set_io_hooks(tcp_socket_io_hooks* new_hooks);
|
|
void set_reuse_address(bool enable = true); // set SO_REUSEADDR
|
|
fc::ip::endpoint remote_endpoint() const;
|
|
fc::ip::endpoint local_endpoint() const;
|
|
|
|
void get( char& c )
|
|
{
|
|
read( &c, 1 );
|
|
}
|
|
|
|
|
|
/// istream interface
|
|
/// @{
|
|
virtual size_t readsome( char* buffer, size_t max );
|
|
virtual size_t readsome( const std::shared_ptr<char>& buffer, size_t max );
|
|
virtual bool eof()const;
|
|
/// @}
|
|
|
|
/// ostream interface
|
|
/// @{
|
|
virtual size_t writesome( const char* buffer, size_t len );
|
|
virtual size_t writesome( const std::shared_ptr<const char>& buffer, size_t len );
|
|
virtual void flush();
|
|
virtual void close();
|
|
/// @}
|
|
|
|
void open();
|
|
bool is_open()const;
|
|
|
|
private:
|
|
friend class tcp_server;
|
|
class impl;
|
|
#ifdef _WIN64
|
|
fc::fwd<impl,0x81> my;
|
|
#else
|
|
fc::fwd<impl,0x54> my;
|
|
#endif
|
|
};
|
|
typedef std::shared_ptr<tcp_socket> tcp_socket_ptr;
|
|
|
|
|
|
class tcp_server
|
|
{
|
|
public:
|
|
tcp_server();
|
|
~tcp_server();
|
|
|
|
void close();
|
|
void accept( tcp_socket& s );
|
|
void set_reuse_address(bool enable = true); // set SO_REUSEADDR, call before listen
|
|
void listen( uint16_t port );
|
|
void listen( const fc::ip::endpoint& ep );
|
|
fc::ip::endpoint get_local_endpoint() const;
|
|
uint16_t get_port()const;
|
|
private:
|
|
// non copyable
|
|
tcp_server( const tcp_server& );
|
|
tcp_server& operator=(const tcp_server& s );
|
|
|
|
class impl;
|
|
impl* my;
|
|
};
|
|
|
|
} // namesapce fc
|
|
|