28 #ifndef WEBSOCKETPP_TRANSPORT_IOSTREAM_CON_HPP
29 #define WEBSOCKETPP_TRANSPORT_IOSTREAM_CON_HPP
31 #include <websocketpp/common/connection_hdl.hpp>
32 #include <websocketpp/common/memory.hpp>
33 #include <websocketpp/common/platforms.hpp>
34 #include <websocketpp/logger/levels.hpp>
36 #include <websocketpp/transport/base/connection.hpp>
37 #include <websocketpp/transport/iostream/base.hpp>
52 template <
typename config>
53 class connection :
public lib::enable_shared_from_this< connection<config> > {
58 typedef lib::shared_ptr<type>
ptr;
68 typedef typename concurrency_type::scoped_lock_type scoped_lock_type;
69 typedef typename concurrency_type::mutex_type mutex_type;
71 typedef lib::shared_ptr<timer> timer_ptr;
73 explicit connection(
bool is_server, alog_type & alog, elog_type & elog)
74 : m_output_stream(NULL)
76 , m_is_server(is_server)
80 , m_remote_endpoint(
"iostream transport")
87 return type::shared_from_this();
99 scoped_lock_type lock(m_read_mutex);
122 friend std::istream &
operator>> (std::istream & in, type & t) {
124 scoped_lock_type lock(t.m_read_mutex);
147 scoped_lock_type lock(m_read_mutex);
149 return this->read_some_impl(buf,len);
170 scoped_lock_type lock(m_read_mutex);
172 size_t total_read = 0;
173 size_t temp_read = 0;
176 temp_read = this->read_some_impl(buf+total_read,len-total_read);
177 total_read += temp_read;
178 }
while (temp_read != 0 && total_read < len);
201 scoped_lock_type lock(m_read_mutex);
217 scoped_lock_type lock(m_read_mutex);
270 m_remote_endpoint =
value;
286 return m_remote_endpoint;
294 return m_connection_hdl;
319 handler(lib::error_code());
350 s <<
"iostream_con async_read_at_least: " << num_bytes;
353 if (num_bytes > len) {
358 if (m_reading ==
true) {
363 if (num_bytes == 0 || len == 0) {
364 handler(lib::error_code(),
size_t(0));
370 m_bytes_needed = num_bytes;
371 m_read_handler = handler;
394 if (!m_output_stream) {
399 m_output_stream->write(buf,len);
401 if (m_output_stream->bad()) {
404 handler(lib::error_code());
425 if (!m_output_stream) {
430 std::vector<buffer>::const_iterator it;
431 for (it = bufs.begin(); it != bufs.end(); it++) {
432 m_output_stream->write((*it).buf,(*it).len);
434 if (m_output_stream->bad()) {
439 handler(lib::error_code());
447 m_connection_hdl = hdl;
463 return lib::error_code();
471 handler(lib::error_code());
474 void read(std::istream &in) {
483 in.read(m_buf+m_cursor,static_cast<std::streamsize>(m_len-m_cursor));
485 if (in.gcount() == 0) {
490 m_cursor +=
static_cast<size_t>(in.gcount());
498 if (m_cursor >= m_bytes_needed) {
500 complete_read(lib::error_code());
505 size_t read_some_impl(
char const * buf,
size_t len) {
513 size_t bytes_to_copy = (std::min)(len,m_len-m_cursor);
515 std::copy(buf,buf+bytes_to_copy,m_buf+m_cursor);
517 m_cursor += bytes_to_copy;
519 if (m_cursor >= m_bytes_needed) {
520 complete_read(lib::error_code());
523 return bytes_to_copy;
542 void complete_read(lib::error_code
const & ec) {
548 handler(ec,m_cursor);
554 size_t m_bytes_needed;
559 std::ostream * m_output_stream;
563 bool const m_is_server;
567 std::string m_remote_endpoint;
574 mutex_type m_read_mutex;
582 #endif // WEBSOCKETPP_TRANSPORT_IOSTREAM_CON_HPP
void register_ostream(std::ostream *o)
Register a std::ostream with the transport for writing output.
uint16_t value
The type of a close code value.
void set_handle(connection_hdl hdl)
Set Connection Handle.
friend std::istream & operator>>(std::istream &in, type &t)
Overloaded stream input operator.
lib::error_code dispatch(dispatch_handler handler)
Call given handler back within the transport's event system (if present)
void set_remote_endpoint(std::string value)
Set human readable remote endpoint address.
lib::function< void(lib::error_code const &)> write_handler
The type and signature of the callback passed to the write method.
static level const devel
Low level debugging information (warning: very chatty)
lib::weak_ptr< void > connection_hdl
A handle to uniquely identify a connection.
bool is_secure() const
Tests whether or not the underlying transport is secure.
size_t readsome(char const *buf, size_t len)
Manual input supply (DEPRECATED)
async_read called while another async_read was in progress
lib::function< void(lib::error_code const &, size_t)> read_handler
The type and signature of the callback passed to the read method.
size_t read_all(char const *buf, size_t len)
Manual input supply (read all)
underlying transport pass through
std::string get_remote_endpoint() const
Get human readable remote endpoint address.
static level const devel
Development messages (warning: very chatty)
void async_write(char const *buf, size_t len, write_handler handler)
Asyncronous Transport Write.
size_t read_some(char const *buf, size_t len)
Manual input supply (read some)
lib::function< void()> dispatch_handler
The type and signature of the callback passed to the dispatch method.
async_read_at_least call requested more bytes than buffer can store
config::alog_type alog_type
Type of this transport's access logging policy.
lib::function< void(lib::error_code const &)> timer_handler
The type and signature of the callback passed to the read method.
lib::function< void(lib::error_code const &)> shutdown_handler
The type and signature of the callback passed to the shutdown method.
Namespace for the WebSocket++ project.
config::elog_type elog_type
Type of this transport's error logging policy.
lib::function< void(lib::error_code const &)> init_handler
The type and signature of the callback passed to the init hook.
connection< config > type
Type of this connection transport component.
ptr get_shared()
Get a shared pointer to this component.
void init(init_handler handler)
Initialize the connection transport.
connection_hdl get_handle() const
Get the connection handle.
void async_shutdown(shutdown_handler handler)
Perform cleanup on socket shutdown_handler.
void set_secure(bool value)
Set whether or not this connection is secure.
timer_ptr set_timer(long, timer_handler)
Call back a function after a period of time.
void async_write(std::vector< buffer > const &bufs, write_handler handler)
Asyncronous Transport Write (scatter-gather)
void fatal_error()
Signal transport error.
void async_read_at_least(size_t num_bytes, char *buf, size_t len, read_handler handler)
Initiate an async_read for at least num_bytes bytes into buf.
config::concurrency_type concurrency_type
transport concurrency policy
lib::shared_ptr< type > ptr
Type of a shared pointer to this connection transport component.