ZeroTierOne/node/OutboundMulticast.hpp

162 lines
4.7 KiB
C++
Raw Normal View History

2014-09-22 20:18:24 +00:00
/*
* ZeroTier One - Network Virtualization Everywhere
2017-04-28 03:47:25 +00:00
* Copyright (C) 2011-2017 ZeroTier, Inc. https://www.zerotier.com/
2014-09-22 20:18:24 +00:00
*
* This program 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.
*
* This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
2017-04-28 03:47:25 +00:00
*
* --
*
* You can be released from the requirements of the license by purchasing
* a commercial license. Buying such a license is mandatory as soon as you
* develop commercial closed-source software that incorporates or links
* directly against ZeroTier software without disclosing the source code
* of your own application.
2014-09-22 20:18:24 +00:00
*/
#ifndef ZT_OUTBOUNDMULTICAST_HPP
#define ZT_OUTBOUNDMULTICAST_HPP
#include <stdint.h>
#include <vector>
#include <algorithm>
#include "Constants.hpp"
#include "MAC.hpp"
#include "MulticastGroup.hpp"
#include "Address.hpp"
#include "Packet.hpp"
namespace ZeroTier {
class CertificateOfMembership;
class RuntimeEnvironment;
2014-09-26 05:08:52 +00:00
2014-09-22 20:18:24 +00:00
/**
* An outbound multicast packet
*
* This object isn't guarded by a mutex; caller must synchronize access.
*/
class OutboundMulticast
{
public:
2014-09-24 20:45:58 +00:00
/**
* Create an uninitialized outbound multicast
*
* It must be initialized with init().
*/
OutboundMulticast() {}
2014-09-22 20:18:24 +00:00
/**
* Initialize outbound multicast
*
* @param RR Runtime environment
2014-09-22 20:18:24 +00:00
* @param timestamp Creation time
* @param nwid Network ID
* @param disableCompression Disable compression of frame payload
* @param limit Multicast limit for desired number of packets to send
2014-09-25 22:57:43 +00:00
* @param gatherLimit Number to lazily/implicitly gather with this frame or 0 for none
* @param src Source MAC address of frame or NULL to imply compute from sender ZT address
2014-09-22 20:18:24 +00:00
* @param dest Destination multicast group (MAC + ADI)
* @param etherType 16-bit Ethernet type ID
* @param payload Data
* @param len Length of data
* @throws std::out_of_range Data too large to fit in a MULTICAST_FRAME
*/
void init(
const RuntimeEnvironment *RR,
uint64_t timestamp,
uint64_t nwid,
bool disableCompression,
unsigned int limit,
unsigned int gatherLimit,
const MAC &src,
const MulticastGroup &dest,
unsigned int etherType,
const void *payload,
unsigned int len);
2014-09-22 20:18:24 +00:00
/**
* @return Multicast creation time
*/
inline uint64_t timestamp() const throw() { return _timestamp; }
2014-09-22 22:03:16 +00:00
/**
* @param now Current time
* @return True if this multicast is expired (has exceeded transmit timeout)
*/
inline bool expired(uint64_t now) const throw() { return ((now - _timestamp) >= ZT_MULTICAST_TRANSMIT_TIMEOUT); }
2014-09-22 20:18:24 +00:00
/**
* @return True if this outbound multicast has been sent to enough peers
2014-09-22 20:18:24 +00:00
*/
inline bool atLimit() const throw() { return (_alreadySentTo.size() >= _limit); }
2014-09-25 22:08:29 +00:00
/**
* Just send without checking log
*
* @param RR Runtime environment
* @param tPtr Thread pointer to be handed through to any callbacks called as a result of this call
2014-09-25 22:08:29 +00:00
* @param toAddr Destination address
*/
void sendOnly(const RuntimeEnvironment *RR,void *tPtr,const Address &toAddr);
2014-09-25 22:08:29 +00:00
/**
* Just send and log but do not check sent log
*
* @param RR Runtime environment
* @param tPtr Thread pointer to be handed through to any callbacks called as a result of this call
2014-09-25 22:08:29 +00:00
* @param toAddr Destination address
*/
inline void sendAndLog(const RuntimeEnvironment *RR,void *tPtr,const Address &toAddr)
2014-09-25 22:08:29 +00:00
{
_alreadySentTo.push_back(toAddr);
sendOnly(RR,tPtr,toAddr);
2014-09-25 22:08:29 +00:00
}
2014-09-22 20:18:24 +00:00
/**
* Try to send this to a given peer if it hasn't been sent to them already
*
* @param RR Runtime environment
* @param tPtr Thread pointer to be handed through to any callbacks called as a result of this call
2014-09-22 20:18:24 +00:00
* @param toAddr Destination address
* @return True if address is new and packet was sent to switch, false if duplicate
*/
inline bool sendIfNew(const RuntimeEnvironment *RR,void *tPtr,const Address &toAddr)
2014-09-22 20:18:24 +00:00
{
2014-11-26 21:14:18 +00:00
if (std::find(_alreadySentTo.begin(),_alreadySentTo.end(),toAddr) == _alreadySentTo.end()) {
sendAndLog(RR,tPtr,toAddr);
2014-11-26 21:14:18 +00:00
return true;
} else {
return false;
}
2014-09-22 20:18:24 +00:00
}
private:
uint64_t _timestamp;
uint64_t _nwid;
MAC _macSrc;
MAC _macDest;
unsigned int _limit;
unsigned int _frameLen;
unsigned int _etherType;
Packet _packet;
2014-09-22 20:18:24 +00:00
std::vector<Address> _alreadySentTo;
uint8_t _frameData[ZT_MAX_MTU];
2014-09-22 20:18:24 +00:00
};
} // namespace ZeroTier
#endif