Renamed SocketTap to VirtualTap and Connection to VirtualSocket to prevent confusion

2017-08-15 18:15:06 -07:00
parent cb93c752f0
commit 7d551d6198
15 changed files with 671 additions and 672 deletions
--- a/src/VirtualSocket.hpp
+++ b/src/VirtualSocket.hpp
@@ -0,0 +1,115 @@
+/*
+ * ZeroTier SDK - Network Virtualization Everywhere
+ * Copyright (C) 2011-2017  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/>.
+ *
+ * --
+ *
+ * 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.
+ */
+
+// General connection object used by VirtualTap and network stack drivers
+
+#ifndef ZT_CONNECTION_HPP
+#define ZT_CONNECTION_HPP
+
+#include <ctime>
+#include <sys/socket.h>
+
+#if defined(STACK_PICO)
+#include "pico_socket.h"
+#endif
+
+#include "Phy.hpp"
+
+#include "libzt.h"
+#include "VirtualTap.hpp"
+#include "RingBuffer.hpp"
+
+namespace ZeroTier {
+	
+	class VirtualTap;
+
+	/*
+	 * Something analogous to a socket. This is a common object used by the 
+	 * libzt API, VirtualTap, and the userspace network stack driver implementations. 
+	 * In some situations the word 'Connection' would capture the meaning and
+	 * function of this object, however I'd like to discourage this since this 
+	 * object also handles non-connection-based traffic as well.
+	 */
+	struct VirtualSocket
+	{
+		RingBuffer<unsigned char> *TXbuf;
+		RingBuffer<unsigned char> *RXbuf;
+		Mutex _tx_m, _rx_m;
+
+		PhySocket *sock;	
+
+#if defined(STACK_PICO)			
+		struct pico_socket *picosock;
+#endif
+#if defined(STACK_LWIP)
+		void *pcb;
+#endif
+
+		// TODO: For getsockname, etc
+		struct sockaddr_storage *local_addr; // Address we've bound to locally
+		struct sockaddr_storage *peer_addr;  // Address of connection call to remote host
+
+		int socket_family, socket_type, protocol;
+
+		int app_fd; // used by app for I/O
+		int sdk_fd; // used by lib for I/O
+
+		std::queue<VirtualSocket*> _AcceptedConnections;
+		VirtualTap *tap;
+		int state;      // See libzt.h for (ZT_SOCK_STATE_*)
+
+		// timestamp for closure event
+		std::time_t closure_ts;
+
+		VirtualSocket() {
+			TXbuf = new RingBuffer<unsigned char>(ZT_TCP_TX_BUF_SZ);
+			RXbuf = new RingBuffer<unsigned char>(ZT_TCP_RX_BUF_SZ);
+
+			closure_ts = -1;
+			ZT_PHY_SOCKFD_TYPE fdpair[2];
+			if(socketpair(PF_LOCAL, SOCK_STREAM, 0, fdpair) < 0) {
+				if(errno < 0) {
+					DEBUG_ERROR("unable to create socketpair");
+					return;
+				}
+			}
+			sdk_fd = fdpair[0];
+			app_fd = fdpair[1];
+		}
+		~VirtualSocket() { }
+	};
+
+	/*
+	 * A helper object for passing VirtualTap(s) and VirtualSocket(s) through the stack
+	 */
+	struct VirtualBindingPair
+	{
+	  VirtualTap *tap;
+	  VirtualSocket *vs;
+	  VirtualBindingPair(VirtualTap *_tap, VirtualSocket *_vs) : tap(_tap), vs(_vs) {}
+	};
+}
+#endif