Giulio Grassi | 624f6c6 | 2014-02-18 19:42:14 +0100 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
| 2 | /** |
Alexander Afanasyev | 29d1fab | 2014-07-07 19:27:16 -0700 | [diff] [blame^] | 3 | * Copyright (c) 2014, Regents of the University of California, |
| 4 | * Arizona Board of Regents, |
| 5 | * Colorado State University, |
| 6 | * University Pierre & Marie Curie, Sorbonne University, |
| 7 | * Washington University in St. Louis, |
| 8 | * Beijing Institute of Technology, |
| 9 | * The University of Memphis |
Alexander Afanasyev | 9bcbc7c | 2014-04-06 19:37:37 -0700 | [diff] [blame] | 10 | * |
| 11 | * This file is part of NFD (Named Data Networking Forwarding Daemon). |
| 12 | * See AUTHORS.md for complete list of NFD authors and contributors. |
| 13 | * |
| 14 | * NFD is free software: you can redistribute it and/or modify it under the terms |
| 15 | * of the GNU General Public License as published by the Free Software Foundation, |
| 16 | * either version 3 of the License, or (at your option) any later version. |
| 17 | * |
| 18 | * NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; |
| 19 | * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR |
| 20 | * PURPOSE. See the GNU General Public License for more details. |
| 21 | * |
| 22 | * You should have received a copy of the GNU General Public License along with |
| 23 | * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>. |
Alexander Afanasyev | 29d1fab | 2014-07-07 19:27:16 -0700 | [diff] [blame^] | 24 | */ |
Giulio Grassi | 624f6c6 | 2014-02-18 19:42:14 +0100 | [diff] [blame] | 25 | |
| 26 | #include "udp-face.hpp" |
| 27 | |
Davide Pesavento | 0e76f62 | 2014-06-25 09:58:20 +0200 | [diff] [blame] | 28 | #ifdef __linux__ |
| 29 | #include <netinet/in.h> // for IP_MTU_DISCOVER and IP_PMTUDISC_DONT |
| 30 | #include <sys/socket.h> // for setsockopt() |
| 31 | #endif |
| 32 | |
Giulio Grassi | 624f6c6 | 2014-02-18 19:42:14 +0100 | [diff] [blame] | 33 | namespace nfd { |
| 34 | |
| 35 | NFD_LOG_INCLASS_TEMPLATE_SPECIALIZATION_DEFINE(DatagramFace, UdpFace::protocol, "UdpFace"); |
| 36 | |
Alexander Afanasyev | e515f0a | 2014-06-30 15:28:10 -0700 | [diff] [blame] | 37 | UdpFace::UdpFace(const shared_ptr<UdpFace::protocol::socket>& socket, |
| 38 | bool isOnDemand, |
| 39 | const time::seconds& idleTimeout) |
Junxiao Shi | 7949416 | 2014-04-02 18:25:11 -0700 | [diff] [blame] | 40 | : DatagramFace<protocol>(FaceUri(socket->remote_endpoint()), |
| 41 | FaceUri(socket->local_endpoint()), |
| 42 | socket, isOnDemand) |
Alexander Afanasyev | e515f0a | 2014-06-30 15:28:10 -0700 | [diff] [blame] | 43 | , m_idleTimeout(idleTimeout) |
| 44 | , m_lastIdleCheck(time::steady_clock::now()) |
Giulio Grassi | 624f6c6 | 2014-02-18 19:42:14 +0100 | [diff] [blame] | 45 | { |
Davide Pesavento | 0e76f62 | 2014-06-25 09:58:20 +0200 | [diff] [blame] | 46 | #ifdef __linux__ |
| 47 | // |
| 48 | // By default, Linux does path MTU discovery on IPv4 sockets, |
| 49 | // and sets the DF (Don't Fragment) flag on datagrams smaller |
| 50 | // than the interface MTU. However this does not work for us, |
| 51 | // because we cannot properly respond to ICMP "packet too big" |
| 52 | // messages by fragmenting the packet at the application level, |
| 53 | // since we want to rely on IP for fragmentation and reassembly. |
| 54 | // |
| 55 | // Therefore, we disable PMTU discovery, which prevents the kernel |
| 56 | // from setting the DF flag on outgoing datagrams, and thus allows |
| 57 | // routers along the path to perform fragmentation as needed. |
| 58 | // |
| 59 | const int value = IP_PMTUDISC_DONT; |
| 60 | if (::setsockopt(socket->native_handle(), IPPROTO_IP, |
| 61 | IP_MTU_DISCOVER, &value, sizeof(value)) < 0) |
| 62 | { |
| 63 | NFD_LOG_WARN("[id:" << this->getId() |
Alexander Afanasyev | 29d1fab | 2014-07-07 19:27:16 -0700 | [diff] [blame^] | 64 | << ",uri:" << this->getRemoteUri() |
Davide Pesavento | 0e76f62 | 2014-06-25 09:58:20 +0200 | [diff] [blame] | 65 | << "] Failed to disable path MTU discovery"); |
| 66 | } |
| 67 | #endif |
Alexander Afanasyev | e515f0a | 2014-06-30 15:28:10 -0700 | [diff] [blame] | 68 | |
| 69 | if (isOnDemand && m_idleTimeout > time::seconds::zero()) { |
| 70 | m_closeIfIdleEvent = scheduler::schedule(m_idleTimeout, |
| 71 | bind(&UdpFace::closeIfIdle, this)); |
| 72 | } |
| 73 | } |
| 74 | |
| 75 | UdpFace::~UdpFace() |
| 76 | { |
| 77 | scheduler::cancel(m_closeIfIdleEvent); |
Giulio Grassi | 624f6c6 | 2014-02-18 19:42:14 +0100 | [diff] [blame] | 78 | } |
| 79 | |
Alexander Afanasyev | e515f0a | 2014-06-30 15:28:10 -0700 | [diff] [blame] | 80 | ndn::nfd::FaceStatus |
| 81 | UdpFace::getFaceStatus() const |
| 82 | { |
| 83 | ndn::nfd::FaceStatus status = Face::getFaceStatus(); |
| 84 | if (isOnDemand()) { |
Alexander Afanasyev | 40c61f7 | 2014-06-30 17:21:01 -0700 | [diff] [blame] | 85 | time::milliseconds left = m_idleTimeout - time::duration_cast<time::milliseconds>( |
Alexander Afanasyev | e515f0a | 2014-06-30 15:28:10 -0700 | [diff] [blame] | 86 | time::steady_clock::now() - m_lastIdleCheck); |
Alexander Afanasyev | 40c61f7 | 2014-06-30 17:21:01 -0700 | [diff] [blame] | 87 | if (left < time::milliseconds::zero()) |
| 88 | left = time::milliseconds::zero(); |
Alexander Afanasyev | e515f0a | 2014-06-30 15:28:10 -0700 | [diff] [blame] | 89 | |
| 90 | if (hasBeenUsedRecently()) { |
| 91 | status.setExpirationPeriod(left + m_idleTimeout); |
| 92 | } |
| 93 | else { |
| 94 | status.setExpirationPeriod(left); |
| 95 | } |
| 96 | } |
| 97 | return status; |
| 98 | } |
| 99 | |
| 100 | void |
| 101 | UdpFace::closeIfIdle() |
| 102 | { |
| 103 | // Face can be switched from on-demand to non-on-demand mode |
| 104 | // (non-on-demand -> on-demand transition is not allowed) |
| 105 | if (isOnDemand()) { |
| 106 | if (!hasBeenUsedRecently()) { |
| 107 | // face has been idle since the last time closeIfIdle |
| 108 | // has been called. Going to close it |
| 109 | NFD_LOG_DEBUG("Found idle face id: " << getId()); |
| 110 | |
| 111 | NFD_LOG_INFO("[id:" << this->getId() |
Alexander Afanasyev | 29d1fab | 2014-07-07 19:27:16 -0700 | [diff] [blame^] | 112 | << ",uri:" << this->getRemoteUri() |
Alexander Afanasyev | e515f0a | 2014-06-30 15:28:10 -0700 | [diff] [blame] | 113 | << "] Idle for more than " << m_idleTimeout << ", closing"); |
| 114 | close(); |
| 115 | } |
| 116 | else { |
| 117 | resetRecentUsage(); |
| 118 | |
| 119 | m_lastIdleCheck = time::steady_clock::now(); |
| 120 | m_closeIfIdleEvent = scheduler::schedule(m_idleTimeout, |
| 121 | bind(&UdpFace::closeIfIdle, this)); |
| 122 | } |
| 123 | } |
| 124 | // else do nothing and do not reschedule the event |
| 125 | } |
| 126 | |
Giulio Grassi | 624f6c6 | 2014-02-18 19:42:14 +0100 | [diff] [blame] | 127 | } // namespace nfd |