blob: 3c6536e06ef7f0bfa7ac938bd124437bd6a72b78 [file] [log] [blame]
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
* Copyright (c) 2014-2021, Regents of the University of California,
* Arizona Board of Regents,
* Colorado State University,
* University Pierre & Marie Curie, Sorbonne University,
* Washington University in St. Louis,
* Beijing Institute of Technology,
* The University of Memphis.
*
* This file is part of NFD (Named Data Networking Forwarding Daemon).
* See AUTHORS.md for complete list of NFD authors and contributors.
*
* NFD is free software: you can redistribute it and/or modify it under the terms
* of the GNU General Public License as published by the Free Software Foundation,
* either version 3 of the License, or (at your option) any later version.
*
* NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
* PURPOSE. See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along with
* NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef NFD_DAEMON_FW_STRATEGY_HPP
#define NFD_DAEMON_FW_STRATEGY_HPP
#include "forwarder.hpp"
#include "table/measurements-accessor.hpp"
namespace nfd {
namespace fw {
/** \brief Represents a forwarding strategy
*/
class Strategy : noncopyable
{
public: // registry
/** \brief Register a strategy type
* \tparam S subclass of Strategy
* \param strategyName strategy program name, must contain version
* \note It is permitted to register the same strategy type under multiple names,
* which is useful in tests and for creating aliases.
*/
template<typename S>
static void
registerType(const Name& strategyName = S::getStrategyName())
{
BOOST_ASSERT(strategyName.size() > 1);
BOOST_ASSERT(strategyName.at(-1).isVersion());
Registry& registry = getRegistry();
BOOST_ASSERT(registry.count(strategyName) == 0);
registry[strategyName] = [] (auto&&... args) {
return make_unique<S>(std::forward<decltype(args)>(args)...);
};
}
/** \return Whether a strategy instance can be created from \p instanceName
* \param instanceName strategy instance name, may contain version and parameters
* \note This function finds a strategy type using same rules as \p create ,
* but does not attempt to construct an instance.
*/
static bool
canCreate(const Name& instanceName);
/** \return A strategy instance created from \p instanceName
* \retval nullptr if !canCreate(instanceName)
* \throw std::invalid_argument strategy type constructor does not accept
* specified version or parameters
*/
static unique_ptr<Strategy>
create(const Name& instanceName, Forwarder& forwarder);
/** \return Whether \p instanceNameA and \p instanceNameA will initiate same strategy type
*/
static bool
areSameType(const Name& instanceNameA, const Name& instanceNameB);
/** \return Registered versioned strategy names
*/
static std::set<Name>
listRegistered();
public: // constructor, destructor, strategy info
/** \brief Construct a strategy instance.
* \param forwarder a reference to the forwarder, used to enable actions and accessors.
* \note Strategy subclass constructor must not retain a reference to \p forwarder.
*/
explicit
Strategy(Forwarder& forwarder);
virtual
~Strategy();
#ifdef DOXYGEN
/** \return Strategy program name
*
* The strategy name is defined by the strategy program.
* It must end with a version component.
*/
static const Name&
getStrategyName();
#endif
/** \return Strategy instance name
*
* The instance name is assigned during instantiation.
* It contains a version component, and may have extra parameter components.
*/
const Name&
getInstanceName() const
{
return m_name;
}
public: // triggers
/** \brief Trigger after Interest is received
*
* The Interest:
* - has not exceeded HopLimit
* - does not violate Scope
* - is not looped
* - cannot be satisfied by ContentStore
* - is under a namespace managed by this strategy
*
* The PIT entry is set to expire after InterestLifetime has elapsed at each downstream.
*
* The strategy should decide whether and where to forward this Interest.
* - If the strategy decides to forward this Interest,
* invoke \c sendInterest for each upstream, either now or shortly after via a scheduler event,
* but before PIT entry expires.
* Optionally, the strategy can invoke \c setExpiryTimer to adjust how long it would wait for a response.
* - If the strategy has already forwarded this Interest previously and decides to continue waiting,
* do nothing.
* Optionally, the strategy can invoke \c setExpiryTimer to adjust how long it would wait for a response.
* - If the strategy concludes that this Interest cannot be satisfied,
* invoke \c rejectPendingInterest to erase the PIT entry.
*
* \warning The strategy must not retain shared_ptr<pit::Entry>, otherwise undefined behavior
* may occur. However, the strategy is allowed to store weak_ptr<pit::Entry>.
*/
virtual void
afterReceiveInterest(const FaceEndpoint& ingress, const Interest& interest,
const shared_ptr<pit::Entry>& pitEntry) = 0;
/** \brief Trigger before PIT entry is satisfied
*
* This trigger is invoked when an incoming Data satisfies more than one PIT entry.
* The strategy can collect measurements information, but cannot manipulate Data forwarding.
* When an incoming Data satisfies only one PIT entry, \c afterReceiveData is invoked instead
* and given full control over Data forwarding. If a strategy does not override \c afterReceiveData,
* the default implementation invokes \c beforeSatisfyInterest.
*
* Normally, PIT entries would be erased after receiving the first matching Data.
* If the strategy wishes to collect responses from additional upstream nodes,
* it should invoke \c setExpiryTimer within this function to prolong the PIT entry lifetime.
* If a Data arrives from another upstream during the extended PIT entry lifetime, this trigger will be invoked again.
* At that time, this function must invoke \c setExpiryTimer again to continue collecting more responses.
*
* In this base class this method does nothing.
*
* \warning The strategy must not retain shared_ptr<pit::Entry>, otherwise undefined behavior
* may occur. However, the strategy is allowed to store weak_ptr<pit::Entry>.
*/
virtual void
beforeSatisfyInterest(const shared_ptr<pit::Entry>& pitEntry,
const FaceEndpoint& ingress, const Data& data);
/** \brief Trigger after a Data is matched in CS
*
* In the base class this method sends \p data to \p ingress
*/
virtual void
afterContentStoreHit(const shared_ptr<pit::Entry>& pitEntry,
const FaceEndpoint& ingress, const Data& data);
/** \brief Trigger after Data is received
*
* This trigger is invoked when an incoming Data satisfies exactly one PIT entry,
* and gives the strategy full control over Data forwarding.
*
* When this trigger is invoked:
* - The Data has been verified to satisfy the PIT entry.
* - The PIT entry expiry timer is set to now
*
* Within this function:
* - A strategy should return Data to downstream nodes via \c sendData or \c sendDataToAll.
* - A strategy can modify the Data as long as it still satisfies the PIT entry, such as
* adding or removing congestion marks.
* - A strategy can delay Data forwarding by prolonging the PIT entry lifetime via \c setExpiryTimer,
* and forward Data before the PIT entry is erased.
* - A strategy can collect measurements about the upstream.
* - A strategy can collect responses from additional upstream nodes by prolonging the PIT entry
* lifetime via \c setExpiryTimer every time a Data is received. Note that only one Data should
* be returned to each downstream node.
*
* In the base class this method invokes \c beforeSatisfyInterest trigger and then returns
* the Data to downstream faces via \c sendDataToAll.
*/
virtual void
afterReceiveData(const shared_ptr<pit::Entry>& pitEntry,
const FaceEndpoint& ingress, const Data& data);
/** \brief Trigger after Nack is received
*
* This trigger is invoked when an incoming Nack is received in response to
* an forwarded Interest.
* The Nack has been confirmed to be a response to the last Interest forwarded
* to that upstream, i.e. the PIT out-record exists and has a matching Nonce.
* The NackHeader has been recorded in the PIT out-record.
*
* If the PIT entry is not yet satisfied, its expiry timer remains unchanged.
* Otherwise, the PIT entry normally would expire immediately after this function returns.
*
* If the strategy wishes to collect responses from additional upstream nodes,
* it should invoke \c setExpiryTimer within this function to retain the PIT entry.
* If a Nack arrives from another upstream during the extended PIT entry lifetime, this trigger will be invoked again.
* At that time, this function must invoke \c setExpiryTimer again to continue collecting more responses.
*
* In the base class this method does nothing.
*
* \warning The strategy must not retain shared_ptr<pit::Entry>, otherwise undefined behavior
* may occur. However, the strategy is allowed to store weak_ptr<pit::Entry>.
*/
virtual void
afterReceiveNack(const FaceEndpoint& ingress, const lp::Nack& nack,
const shared_ptr<pit::Entry>& pitEntry);
/** \brief Trigger after Interest dropped (e.g., for exceeding allowed retransmissions)
*
* In the base class this method does nothing.
*/
virtual void
onDroppedInterest(const Face& egress, const Interest& interest);
/** \brief Trigger after new nexthop is added
*
* The strategy should decide whether to send the buffered Interests to the new nexthop.
* In the base class, this method does nothing.
*/
virtual void
afterNewNextHop(const fib::NextHop& nextHop, const shared_ptr<pit::Entry>& pitEntry);
protected: // actions
/** \brief Send an Interest packet.
* \param pitEntry the PIT entry
* \param egress face through which to send out the Interest
* \param interest the Interest packet
* \return A pointer to the out-record created or nullptr if the Interest was dropped
*/
NFD_VIRTUAL_WITH_TESTS pit::OutRecord*
sendInterest(const shared_ptr<pit::Entry>& pitEntry, Face& egress,
const Interest& interest);
/** \brief Send a Data packet.
* \param pitEntry the PIT entry
* \param data the Data packet
* \param egress face through which to send out the Data
* \return Whether the Data was sent (true) or dropped (false)
*/
NFD_VIRTUAL_WITH_TESTS bool
sendData(const shared_ptr<pit::Entry>& pitEntry, const Data& data, Face& egress);
/** \brief Send a Data packet to all matched and qualified faces.
*
* A matched face is qualified if it is ad-hoc or it is NOT \p inFace.
*
* \param pitEntry the PIT entry
* \param inFace face on which the Data arrived
* \param data the Data packet
*/
NFD_VIRTUAL_WITH_TESTS void
sendDataToAll(const shared_ptr<pit::Entry>& pitEntry, const Face& inFace, const Data& data);
/** \brief Schedule the PIT entry for immediate deletion.
*
* This helper function sets the PIT entry expiry time to zero.
* The strategy should invoke this function when it concludes that the Interest cannot
* be forwarded and it does not want to wait for responses from existing upstream nodes.
*/
NFD_VIRTUAL_WITH_TESTS void
rejectPendingInterest(const shared_ptr<pit::Entry>& pitEntry)
{
this->setExpiryTimer(pitEntry, 0_ms);
}
/** \brief Send a Nack packet.
*
* The egress face must have a PIT in-record, otherwise this method has no effect.
*
* \param pitEntry the PIT entry
* \param egress face through which to send out the Nack
* \param header the Nack header
* \return Whether the Nack was sent (true) or dropped (false)
*/
NFD_VIRTUAL_WITH_TESTS bool
sendNack(const shared_ptr<pit::Entry>& pitEntry, Face& egress,
const lp::NackHeader& header)
{
return m_forwarder.onOutgoingNack(pitEntry, egress, header);
}
/** \brief Send Nack to every face that has an in-record, except those in \p exceptFaces
* \param pitEntry the PIT entry
* \param header the Nack header
* \param exceptFaces list of faces that should be excluded from sending Nacks
* \note This is not an action, but a helper that invokes the sendNack() action.
*/
void
sendNacks(const shared_ptr<pit::Entry>& pitEntry, const lp::NackHeader& header,
std::initializer_list<const Face*> exceptFaces = {});
/** \brief Schedule the PIT entry to be erased after \p duration.
*/
void
setExpiryTimer(const shared_ptr<pit::Entry>& pitEntry, time::milliseconds duration)
{
m_forwarder.setExpiryTimer(pitEntry, duration);
}
protected: // accessors
/** \brief Performs a FIB lookup, considering Link object if present.
*/
const fib::Entry&
lookupFib(const pit::Entry& pitEntry) const;
MeasurementsAccessor&
getMeasurements()
{
return m_measurements;
}
Face*
getFace(FaceId id) const
{
return getFaceTable().get(id);
}
const FaceTable&
getFaceTable() const
{
return m_forwarder.m_faceTable;
}
protected: // instance name
struct ParsedInstanceName
{
Name strategyName; ///< strategy name without parameters
optional<uint64_t> version; ///< whether strategyName contains a version component
PartialName parameters; ///< parameter components
};
/** \brief Parse a strategy instance name
* \param input strategy instance name, may contain version and parameters
* \throw std::invalid_argument input format is unacceptable
*/
static ParsedInstanceName
parseInstanceName(const Name& input);
/** \brief Construct a strategy instance name
* \param input strategy instance name, may contain version and parameters
* \param strategyName strategy name with version but without parameters;
* typically this should be \p getStrategyName()
*
* If \p input contains a version component, return \p input unchanged.
* Otherwise, return \p input plus the version component taken from \p strategyName.
* This allows a strategy instance to be constructed with an unversioned name,
* but its final instance name should contain the version.
*/
static Name
makeInstanceName(const Name& input, const Name& strategyName);
/** \brief Set strategy instance name
* \note This must be called by strategy subclass constructor.
*/
void
setInstanceName(const Name& name)
{
m_name = name;
}
private: // registry
typedef std::function<unique_ptr<Strategy>(Forwarder& forwarder, const Name& strategyName)> CreateFunc;
typedef std::map<Name, CreateFunc> Registry; // indexed by strategy name
static Registry&
getRegistry();
static Registry::const_iterator
find(const Name& instanceName);
protected: // accessors
signal::Signal<FaceTable, Face>& afterAddFace;
signal::Signal<FaceTable, Face>& beforeRemoveFace;
private: // instance fields
Name m_name;
/** \brief Reference to the forwarder
*
* Triggers can access forwarder indirectly via actions.
*/
Forwarder& m_forwarder;
MeasurementsAccessor m_measurements;
};
} // namespace fw
} // namespace nfd
/** \brief Registers a strategy
*
* This macro should appear once in .cpp of each strategy.
*/
#define NFD_REGISTER_STRATEGY(S) \
static class NfdAuto ## S ## StrategyRegistrationClass \
{ \
public: \
NfdAuto ## S ## StrategyRegistrationClass() \
{ \
::nfd::fw::Strategy::registerType<S>(); \
} \
} g_nfdAuto ## S ## StrategyRegistrationVariable
#endif // NFD_DAEMON_FW_STRATEGY_HPP