瀏覽代碼

Simplify packet critical path. Plus more platform fixes

Joseph Henry 11 月之前
父節點
當前提交
b813ea70a5
共有 7 個文件被更改,包括 31 次插入27 次删除
  1. 2 12
      node/IncomingPacket.cpp
  2. 1 2
      node/Node.cpp
  3. 1 7
      node/Node.hpp
  4. 15 3
      node/PacketMultiplexer.cpp
  5. 1 0
      node/PacketMultiplexer.hpp
  6. 1 1
      osdep/EthernetTap.cpp
  7. 10 2
      service/OneService.cpp

+ 2 - 12
node/IncomingPacket.cpp

@@ -869,12 +869,7 @@ bool IncomingPacket::_doFRAME(const RuntimeEnvironment *RR,void *tPtr,const Shar
 				const unsigned int frameLen = size() - ZT_PROTO_VERB_FRAME_IDX_PAYLOAD;
 				const uint8_t *const frameData = reinterpret_cast<const uint8_t *>(data()) + ZT_PROTO_VERB_FRAME_IDX_PAYLOAD;
 				if (network->filterIncomingPacket(tPtr,peer,RR->identity.address(),sourceMac,network->mac(),frameData,frameLen,etherType,0) > 0) {
-					if (RR->node->getMultithreadingEnabled()) {
-						RR->pm->putFrame(tPtr,nwid,network->userPtr(),sourceMac,network->mac(),etherType,0,(const void *)frameData,frameLen, _flowId);
-					}
-					else {
-						RR->node->putFrame(tPtr,nwid,network->userPtr(),sourceMac,network->mac(),etherType,0,(const void *)frameData,frameLen);
-					}
+					RR->pm->putFrame(tPtr,nwid,network->userPtr(),sourceMac,network->mac(),etherType,0,(const void *)frameData,frameLen, _flowId);
 				}
 			}
 		} else {
@@ -947,12 +942,7 @@ bool IncomingPacket::_doEXT_FRAME(const RuntimeEnvironment *RR,void *tPtr,const
 					}
 					// fall through -- 2 means accept regardless of bridging checks or other restrictions
 				case 2:
-					if (RR->node->getMultithreadingEnabled()) {
-						RR->pm->putFrame(tPtr,nwid,network->userPtr(),from,to,etherType,0,(const void *)frameData,frameLen, flowId);
-					}
-					else {
-						RR->node->putFrame(tPtr,nwid,network->userPtr(),from,to,etherType,0,(const void *)frameData,frameLen);
-					}
+					RR->pm->putFrame(tPtr,nwid,network->userPtr(),from,to,etherType,0,(const void *)frameData,frameLen, flowId);
 					break;
 			}
 		}

+ 1 - 2
node/Node.cpp

@@ -240,9 +240,8 @@ ZT_ResultCode Node::processVirtualNetworkFrame(
 	}
 }
 
-void Node::initMultithreading(bool isEnabled, unsigned int concurrency, bool cpuPinningEnabled)
+void Node::initMultithreading(unsigned int concurrency, bool cpuPinningEnabled)
 {
-	_multithreadingEnabled = isEnabled;
 	RR->pm->setUpPostDecodeReceiveThreads(concurrency, cpuPinningEnabled);
 }
 

+ 1 - 7
node/Node.hpp

@@ -283,12 +283,7 @@ public:
 		return _lowBandwidthMode;
 	}
 
-	inline bool getMultithreadingEnabled()
-	{
-		return _multithreadingEnabled;
-	}
-
-	void initMultithreading(bool isEnabled, unsigned int concurrency, bool cpuPinningEnabled);
+	void initMultithreading(unsigned int concurrency, bool cpuPinningEnabled);
 
 
 public:
@@ -339,7 +334,6 @@ public:
 	volatile int64_t _prngState[2];
 	bool _online;
 	bool _lowBandwidthMode;
-	bool _multithreadingEnabled;
 };
 
 } // namespace ZeroTier

+ 15 - 3
node/PacketMultiplexer.cpp

@@ -15,6 +15,7 @@
 
 #include "Node.hpp"
 #include "RuntimeEnvironment.hpp"
+#include "Constants.hpp"
 
 #include <stdio.h>
 #include <stdlib.h>
@@ -28,6 +29,16 @@ PacketMultiplexer::PacketMultiplexer(const RuntimeEnvironment* renv)
 
 void PacketMultiplexer::putFrame(void* tPtr, uint64_t nwid, void** nuptr, const MAC& source, const MAC& dest, unsigned int etherType, unsigned int vlanId, const void* data, unsigned int len, unsigned int flowId)
 {
+#if defined(__APPLE__) || defined(__OpenBSD__) || defined(__NetBSD__) || defined(__WINDOWS__)
+	RR->node->putFrame(tPtr,nwid,nuptr,source,dest,etherType,vlanId,(const void *)data,len);
+	return;
+#endif
+
+	if (!_enabled) {
+		RR->node->putFrame(tPtr,nwid,nuptr,source,dest,etherType,vlanId,(const void *)data,len);
+		return;
+	}
+
 	PacketRecord* packet;
 	_rxPacketVector_m.lock();
 	if (_rxPacketVector.empty()) {
@@ -56,9 +67,10 @@ void PacketMultiplexer::putFrame(void* tPtr, uint64_t nwid, void** nuptr, const
 
 void PacketMultiplexer::setUpPostDecodeReceiveThreads(unsigned int concurrency, bool cpuPinningEnabled)
 {
-	if (! RR->node->getMultithreadingEnabled()) {
-		return;
-	}
+#if defined(__APPLE__) || defined(__OpenBSD__) || defined(__NetBSD__) || defined(__WINDOWS__)
+	return;
+#endif
+	_enabled = true;
 	_concurrency = concurrency;
 	bool _enablePinning = cpuPinningEnabled;
 

+ 1 - 0
node/PacketMultiplexer.hpp

@@ -57,6 +57,7 @@ class PacketMultiplexer {
 
 	std::vector<std::thread> _rxThreads;
 	unsigned int _rxThreadCount;
+	bool _enabled;
 };
 
 }	// namespace ZeroTier

+ 1 - 1
osdep/EthernetTap.cpp

@@ -140,7 +140,7 @@ std::shared_ptr<EthernetTap> EthernetTap::newInstance(
 #endif // __NetBSD__
 
 #ifdef __OpenBSD__
-	return std::shared_ptr<EthernetTap>(new BSDEthernetTap(homePath,concurrency,mac,mtu,metric,nwid,friendlyName,handler,arg));
+	return std::shared_ptr<EthernetTap>(new BSDEthernetTap(homePath,mac,mtu,metric,nwid,friendlyName,handler,arg));
 #endif // __OpenBSD__
 
 #endif // ZT_SDK?

+ 10 - 2
service/OneService.cpp

@@ -991,7 +991,10 @@ public:
 
 	void setUpMultithreading()
 	{
-		_node->initMultithreading(true, _concurrency, _cpuPinningEnabled);
+#if defined(__APPLE__) || defined(__OpenBSD__) || defined(__NetBSD__) || defined(__WINDOWS__)
+		return;
+#endif
+		_node->initMultithreading(_concurrency, _cpuPinningEnabled);
 		bool pinning = _cpuPinningEnabled;
 
 		fprintf(stderr, "Starting %d RX threads\n", _concurrency);
@@ -2648,6 +2651,7 @@ public:
 			fprintf(stderr,"WARNING: using manually-specified secondary and/or tertiary ports. This can cause NAT issues." ZT_EOL_S);
 		}
 		_portMappingEnabled = OSUtils::jsonBool(settings["portMappingEnabled"],true);
+#if defined(__LINUX__) || defined(__FreeBSD__)
 		_multicoreEnabled = OSUtils::jsonBool(settings["multicoreEnabled"],false);
 		_concurrency = OSUtils::jsonInt(settings["concurrency"],0);
 		_cpuPinningEnabled = OSUtils::jsonBool(settings["cpuPinningEnabled"],false);
@@ -2660,6 +2664,7 @@ public:
 			}
 			setUpMultithreading();
 		}
+#endif
 
 #ifndef ZT_SDK
 		const std::string up(OSUtils::jsonString(settings["softwareUpdate"],ZT_SOFTWARE_UPDATE_DEFAULT));
@@ -2987,7 +2992,7 @@ public:
 		if ((len >= 16) && (reinterpret_cast<const InetAddress*>(from)->ipScope() == InetAddress::IP_SCOPE_GLOBAL)) {
 			_lastDirectReceiveFromGlobal = now;
 		}
-
+#if defined(__LINUX__) || defined(__FreeBSD__)
 		if (_multicoreEnabled) {
 			PacketRecord* packet;
 			_rxPacketVector_m.lock();
@@ -3008,6 +3013,7 @@ public:
 			_rxPacketQueue.postLimit(packet, 256 * _concurrency);
 		}
 		else {
+#endif
 			const ZT_ResultCode rc = _node->processWirePacket(nullptr,now,reinterpret_cast<int64_t>(sock),reinterpret_cast<const struct sockaddr_storage *>(from),data,len,&_nextBackgroundTaskDeadline);
 			if (ZT_ResultCode_isFatal(rc)) {
 				char tmp[256];
@@ -3017,7 +3023,9 @@ public:
 				_fatalErrorMessage = tmp;
 				this->terminate();
 			}
+#if defined(__LINUX__) || defined(__FreeBSD__)
 		}
+#endif
 	}