blob: 2127cd6afe7dc4b19eb0fb7cc70abbb994b1f5b3 [file] [log] [blame]
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/**
* Copyright (C) 2014 Named Data Networking Project
* See COPYING for copyright and distribution information.
*/
#include "udp-channel.hpp"
#include "core/global-io.hpp"
namespace nfd {
NFD_LOG_INIT("UdpChannel");
using namespace boost::asio;
UdpChannel::UdpChannel(const udp::Endpoint& localEndpoint,
const time::Duration& timeout)
: m_localEndpoint(localEndpoint)
, m_isListening(false)
{
/// \todo the reuse_address works as we want in Linux, but in other system could be different.
/// We need to check this
/// (SO_REUSEADDR doesn't behave uniformly in different OS)
m_socket = make_shared<ip::udp::socket>(boost::ref(getGlobalIoService()));
m_socket->open(m_localEndpoint.protocol());
m_socket->set_option(boost::asio::ip::udp::socket::reuse_address(true));
try {
m_socket->bind(m_localEndpoint);
}
catch (boost::system::system_error& e) {
//The bind failed, so the socket is useless now
m_socket->close();
throw Error("Failed to properly configure the socket. "
"UdpChannel creation aborted, check the address (" + std::string(e.what()) + ")");
}
}
void
UdpChannel::listen(const FaceCreatedCallback& onFaceCreated,
const ConnectFailedCallback& onListenFailed)
{
if (m_isListening) {
throw Error("Listen already called on this channel");
}
m_isListening = true;
onFaceCreatedNewPeerCallback = onFaceCreated;
onConnectFailedNewPeerCallback = onListenFailed;
m_socket->async_receive_from(boost::asio::buffer(m_inputBuffer, MAX_NDN_PACKET_SIZE),
m_newRemoteEndpoint,
bind(&UdpChannel::newPeer, this,
boost::asio::placeholders::error,
boost::asio::placeholders::bytes_transferred));
}
void
UdpChannel::connect(const udp::Endpoint& remoteEndpoint,
const FaceCreatedCallback& onFaceCreated)
{
ChannelFaceMap::iterator i = m_channelFaces.find(remoteEndpoint);
if (i != m_channelFaces.end()) {
onFaceCreated(i->second);
return;
}
//creating a new socket for the face that will be created soon
shared_ptr<ip::udp::socket> clientSocket =
make_shared<ip::udp::socket>(boost::ref(getGlobalIoService()));
clientSocket->open(m_localEndpoint.protocol());
clientSocket->set_option(ip::udp::socket::reuse_address(true));
try {
clientSocket->bind(m_localEndpoint);
clientSocket->connect(remoteEndpoint); //@todo connect or async_connect
//(since there is no handshake the connect shouldn't block). If we go for
//async_connect, make sure that if in the meantime we receive a UDP pkt from
//that endpoint nothing bad happen (it's difficult, but it could happen)
}
catch (boost::system::system_error& e) {
clientSocket->close();
throw Error("Failed to properly configure the socket. Check the address ("
+ std::string(e.what()) + ")");
}
createFace(clientSocket, onFaceCreated);
}
void
UdpChannel::connect(const std::string& remoteHost,
const std::string& remotePort,
const FaceCreatedCallback& onFaceCreated,
const ConnectFailedCallback& onConnectFailed)
{
ip::udp::resolver::query query(remoteHost, remotePort);
shared_ptr<ip::udp::resolver> resolver =
make_shared<ip::udp::resolver>(boost::ref(getGlobalIoService()));
resolver->async_resolve(query,
bind(&UdpChannel::handleEndpointResolution, this, _1, _2,
onFaceCreated, onConnectFailed,
resolver));
}
void
UdpChannel::handleEndpointResolution(const boost::system::error_code& error,
ip::udp::resolver::iterator remoteEndpoint,
const FaceCreatedCallback& onFaceCreated,
const ConnectFailedCallback& onConnectFailed,
const shared_ptr<ip::udp::resolver>& resolver)
{
if (error != 0 ||
remoteEndpoint == ip::udp::resolver::iterator())
{
if (error == boost::system::errc::operation_canceled) // when socket is closed by someone
return;
NFD_LOG_DEBUG("Remote endpoint hostname or port cannot be resolved: "
<< error.category().message(error.value()));
onConnectFailed("Remote endpoint hostname or port cannot be resolved: " +
error.category().message(error.value()));
return;
}
connect(*remoteEndpoint, onFaceCreated);
}
size_t
UdpChannel::size() const
{
return m_channelFaces.size();
}
shared_ptr<UdpFace>
UdpChannel::createFace(const shared_ptr<ip::udp::socket>& socket,
const FaceCreatedCallback& onFaceCreated)
{
udp::Endpoint remoteEndpoint = socket->remote_endpoint();
shared_ptr<UdpFace> face = make_shared<UdpFace>(boost::cref(socket));
face->onFail += bind(&UdpChannel::afterFaceFailed, this, remoteEndpoint);
onFaceCreated(face);
m_channelFaces[remoteEndpoint] = face;
return face;
}
void
UdpChannel::newPeer(const boost::system::error_code& error,
std::size_t nBytesReceived)
{
NFD_LOG_DEBUG("UdpChannel::newPeer from " << m_newRemoteEndpoint);
ChannelFaceMap::iterator i = m_channelFaces.find(m_newRemoteEndpoint);
if (i != m_channelFaces.end()) {
//The face already exists.
//Usually this shouldn't happen, because the channel creates a Udpface
//as soon as it receives a pkt from a new endpoint and then the
//traffic is dispatched by the kernel directly to the face.
//However, if the node receives multiple packets from the same endpoint
//"at the same time", while the channel is creating the face the kernel
//could dispatch the other pkts to the channel because the face is not yet
//ready. In this case, the channel has to pass the pkt to the face
NFD_LOG_DEBUG("The creation of the face for the remote endpoint "
<< m_newRemoteEndpoint
<< " is in progress");
//Passing the message to the correspondent face
i->second->handleFirstReceive(m_inputBuffer, nBytesReceived, error);
return;
}
shared_ptr<ip::udp::socket> clientSocket =
make_shared<ip::udp::socket>(boost::ref(getGlobalIoService()));
clientSocket->open(m_localEndpoint.protocol());
clientSocket->set_option(ip::udp::socket::reuse_address(true));
clientSocket->bind(m_localEndpoint);
clientSocket->connect(m_newRemoteEndpoint);
shared_ptr<UdpFace> newFace = createFace(clientSocket, onFaceCreatedNewPeerCallback);
//Passing the message to the correspondent face
newFace->handleFirstReceive(m_inputBuffer, nBytesReceived, error);
m_socket->async_receive_from(boost::asio::buffer(m_inputBuffer, MAX_NDN_PACKET_SIZE),
m_newRemoteEndpoint,
bind(&UdpChannel::newPeer, this,
boost::asio::placeholders::error,
boost::asio::placeholders::bytes_transferred));
}
void UdpChannel::afterFaceFailed(udp::Endpoint &endpoint)
{
NFD_LOG_DEBUG("afterFaceFailed: " << endpoint);
m_channelFaces.erase(endpoint);
}
} // namespace nfd