26 #ifndef NFD_DAEMON_FACE_STREAM_TRANSPORT_HPP
27 #define NFD_DAEMON_FACE_STREAM_TRANSPORT_HPP
41 template<
class Protocol>
68 handleSend(
const boost::system::error_code& error,
73 size_t nBytesReceived);
84 uint8_t m_receiveBuffer[ndn::MAX_NDN_PACKET_SIZE];
85 size_t m_receiveBufferSize;
86 std::queue<Block> m_sendQueue;
92 : m_socket(
std::move(socket))
93 , m_receiveBufferSize(0)
95 m_socket.async_receive(boost::asio::buffer(m_receiveBuffer, ndn::MAX_NDN_PACKET_SIZE),
97 boost::asio::placeholders::error,
98 boost::asio::placeholders::bytes_transferred));
107 if (m_socket.is_open()) {
111 boost::system::error_code error;
112 m_socket.cancel(error);
113 m_socket.shutdown(protocol::socket::shutdown_both, error);
140 std::queue<Block> emptyQueue;
141 std::swap(emptyQueue, m_sendQueue);
144 boost::system::error_code error;
145 m_socket.close(error);
156 bool wasQueueEmpty = m_sendQueue.empty();
157 m_sendQueue.push(packet.packet);
167 boost::asio::async_write(m_socket, boost::asio::buffer(m_sendQueue.front()),
169 boost::asio::placeholders::error,
170 boost::asio::placeholders::bytes_transferred));
179 return processErrorCode(error);
183 BOOST_ASSERT(!m_sendQueue.empty());
186 if (!m_sendQueue.empty())
193 size_t nBytesReceived)
196 return processErrorCode(error);
200 m_receiveBufferSize += nBytesReceived;
206 while (m_receiveBufferSize - offset > 0) {
207 std::tie(isOk, element) = Block::fromBuffer(m_receiveBuffer + offset, m_receiveBufferSize - offset);
211 offset += element.size();
212 BOOST_ASSERT(offset <= m_receiveBufferSize);
217 if (!isOk && m_receiveBufferSize == ndn::MAX_NDN_PACKET_SIZE && offset == 0) {
218 NFD_LOG_FACE_WARN(
"Failed to parse incoming packet or packet too large to process");
225 if (offset != m_receiveBufferSize) {
226 std::copy(m_receiveBuffer + offset, m_receiveBuffer + m_receiveBufferSize, m_receiveBuffer);
227 m_receiveBufferSize -= offset;
230 m_receiveBufferSize = 0;
234 m_socket.async_receive(boost::asio::buffer(m_receiveBuffer + m_receiveBufferSize,
235 ndn::MAX_NDN_PACKET_SIZE - m_receiveBufferSize),
237 boost::asio::placeholders::error,
238 boost::asio::placeholders::bytes_transferred));
250 error == boost::asio::error::operation_aborted ||
251 error == boost::asio::error::shut_down)
255 if (error != boost::asio::error::eof)
265 #endif // NFD_DAEMON_FACE_STREAM_TRANSPORT_HPP
#define NFD_LOG_FACE_TRACE(msg)
Log a message at TRACE level.
NFD_LOG_INCLASS_DECLARE()
stores a packet along with the remote endpoint
the transport is being closed due to a failure
Implements Transport for stream-based protocols.
void handleReceive(const boost::system::error_code &error, size_t nBytesReceived)
#define NFD_LOG_FACE_WARN(msg)
Log a message at WARN level.
Copyright (c) 2014-2015, Regents of the University of California, Arizona Board of Regents...
virtual void doSend(Transport::Packet &&packet) override
performs Transport specific operations to send a packet
the transport is closed, and can be safely deallocated
StreamTransport(typename protocol::socket &&socket)
Construct stream transport.
the transport is requested to be closed
void processErrorCode(const boost::system::error_code &error)
virtual void doClose() override
performs Transport specific operations to close the transport
void handleSend(const boost::system::error_code &error, size_t nBytesSent)
protocol::socket m_socket
boost::asio::io_service & getGlobalIoService()