Adam Ierymenko
10 years ago
13 changed files with 45 additions and 315 deletions
@ -1,100 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere |
||||
* Copyright (C) 2011-2016 ZeroTier, Inc. https://www.zerotier.com/
|
||||
* |
||||
* 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/>.
|
||||
*/ |
||||
|
||||
#include "Constants.hpp" |
||||
#include "DeferredPackets.hpp" |
||||
#include "IncomingPacket.hpp" |
||||
#include "RuntimeEnvironment.hpp" |
||||
#include "Node.hpp" |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
DeferredPackets::DeferredPackets(const RuntimeEnvironment *renv) : |
||||
RR(renv), |
||||
_waiting(0), |
||||
_die(false) |
||||
{ |
||||
} |
||||
|
||||
DeferredPackets::~DeferredPackets() |
||||
{ |
||||
_q_m.lock(); |
||||
_die = true; |
||||
_q_m.unlock(); |
||||
|
||||
for(;;) { |
||||
_q_s.post(); |
||||
|
||||
_q_m.lock(); |
||||
if (_waiting <= 0) { |
||||
_q_m.unlock(); |
||||
break; |
||||
} else { |
||||
_q_m.unlock(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
bool DeferredPackets::enqueue(IncomingPacket *pkt) |
||||
{ |
||||
{ |
||||
Mutex::Lock _l(_q_m); |
||||
if (_q.size() >= ZT_DEFFEREDPACKETS_MAX) |
||||
return false; |
||||
_q.push_back(*pkt); |
||||
} |
||||
_q_s.post(); |
||||
return true; |
||||
} |
||||
|
||||
int DeferredPackets::process() |
||||
{ |
||||
std::list<IncomingPacket> pkt; |
||||
|
||||
_q_m.lock(); |
||||
|
||||
if (_die) { |
||||
_q_m.unlock(); |
||||
return -1; |
||||
} |
||||
|
||||
while (_q.empty()) { |
||||
++_waiting; |
||||
_q_m.unlock(); |
||||
_q_s.wait(); |
||||
_q_m.lock(); |
||||
--_waiting; |
||||
if (_die) { |
||||
_q_m.unlock(); |
||||
return -1; |
||||
} |
||||
} |
||||
|
||||
// Move item from _q list to a dummy list here to avoid copying packet
|
||||
pkt.splice(pkt.end(),_q,_q.begin()); |
||||
|
||||
_q_m.unlock(); |
||||
|
||||
try { |
||||
pkt.front().tryDecode(RR,true); |
||||
} catch ( ... ) {} // drop invalids
|
||||
|
||||
return 1; |
||||
} |
||||
|
||||
} // namespace ZeroTier
|
||||
@ -1,85 +0,0 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere |
||||
* Copyright (C) 2011-2016 ZeroTier, Inc. https://www.zerotier.com/
|
||||
* |
||||
* 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/>.
|
||||
*/ |
||||
|
||||
#ifndef ZT_DEFERREDPACKETS_HPP |
||||
#define ZT_DEFERREDPACKETS_HPP |
||||
|
||||
#include <list> |
||||
|
||||
#include "Constants.hpp" |
||||
#include "SharedPtr.hpp" |
||||
#include "Mutex.hpp" |
||||
#include "DeferredPackets.hpp" |
||||
#include "BinarySemaphore.hpp" |
||||
|
||||
/**
|
||||
* Maximum number of deferred packets |
||||
*/ |
||||
#define ZT_DEFFEREDPACKETS_MAX 256 |
||||
|
||||
namespace ZeroTier { |
||||
|
||||
class IncomingPacket; |
||||
class RuntimeEnvironment; |
||||
|
||||
/**
|
||||
* Deferred packets |
||||
* |
||||
* IncomingPacket can defer its decoding this way by enqueueing itself here. |
||||
* When this is done, deferredDecode() is called later. This is done for |
||||
* operations that may be expensive to allow them to potentially be handled |
||||
* in the background or rate limited to maintain quality of service for more |
||||
* routine operations. |
||||
*/ |
||||
class DeferredPackets |
||||
{ |
||||
public: |
||||
DeferredPackets(const RuntimeEnvironment *renv); |
||||
~DeferredPackets(); |
||||
|
||||
/**
|
||||
* Enqueue a packet |
||||
* |
||||
* @param pkt Packet to process later (possibly in the background) |
||||
* @return False if queue is full |
||||
*/ |
||||
bool enqueue(IncomingPacket *pkt); |
||||
|
||||
/**
|
||||
* Wait for and then process a deferred packet |
||||
* |
||||
* If we are shutting down (in destructor), this returns -1 and should |
||||
* not be called again. Otherwise it returns the number of packets |
||||
* processed. |
||||
* |
||||
* @return Number processed or -1 if shutting down |
||||
*/ |
||||
int process(); |
||||
|
||||
private: |
||||
std::list<IncomingPacket> _q; |
||||
const RuntimeEnvironment *const RR; |
||||
volatile int _waiting; |
||||
volatile bool _die; |
||||
Mutex _q_m; |
||||
BinarySemaphore _q_s; |
||||
}; |
||||
|
||||
} // namespace ZeroTier
|
||||
|
||||
#endif |
||||
Loading…
Reference in new issue