|
|
@@ -0,0 +1,194 @@
|
|
|
+/*
|
|
|
+ * 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 "../osdep/Thread.hpp"
|
|
|
+#include "ButtFlare.hpp"
|
|
|
+
|
|
|
+namespace ZeroTier {
|
|
|
+
|
|
|
+ typedef void PhySocket;
|
|
|
+
|
|
|
+ ButtFlare::ButtFlare(
|
|
|
+ void (*handler)(void *,uint64_t,const MAC &,const MAC &,unsigned int,unsigned int,const void *,unsigned int), void *arg)
|
|
|
+ :
|
|
|
+ _handler(handler),
|
|
|
+ _phy(this,false,true),
|
|
|
+ _enabled(true),
|
|
|
+ _run(true)
|
|
|
+ {
|
|
|
+ // HTTP listen socket
|
|
|
+ struct sockaddr_in in4;
|
|
|
+ memset(&in4,0,sizeof(in4));
|
|
|
+ in4.sin_family = AF_INET;
|
|
|
+ in4.sin_addr.s_addr = Utils::hton((uint32_t)(0x7f000001)); // right now we just listen for TCP @127.0.0.1
|
|
|
+ in4.sin_port = Utils::hton((uint16_t)http_listen_port);
|
|
|
+ _tcpHttpListenSocket = _phy.tcpListen((const struct sockaddr *)&in4,this);
|
|
|
+ // SSL listen socket
|
|
|
+ in4.sin_port = Utils::hton((uint16_t)ssl_listen_port);
|
|
|
+ _tcpSSLListenSocket = _phy.tcpListen((const struct sockaddr *)&in4,this);
|
|
|
+
|
|
|
+ /*
|
|
|
+ struct sockaddr_in6 in6;
|
|
|
+ memset((void *)&in6,0,sizeof(in6));
|
|
|
+ in6.sin6_family = AF_INET6;
|
|
|
+ in6.sin6_port = in4.sin_port;
|
|
|
+ if (_allowManagementFrom.size() == 0)
|
|
|
+ in6.sin6_addr.s6_addr[15] = 1; // IPv6 localhost == ::1
|
|
|
+ _v6TcpControlSocket = _phy.tcpListen((const struct sockaddr *)&in6,this);
|
|
|
+
|
|
|
+ // We must bind one of IPv4 or IPv6 -- support either failing to support hosts that
|
|
|
+ // have only IPv4 or only IPv6 stacks.
|
|
|
+ if ((_v4TcpControlSocket)||(_v6TcpControlSocket)) {
|
|
|
+ _ports[0] = _primaryPort;
|
|
|
+ break;
|
|
|
+ } else {
|
|
|
+ if (_v4TcpControlSocket)
|
|
|
+ _phy.close(_v4TcpControlSocket,false);
|
|
|
+ if (_v6TcpControlSocket)
|
|
|
+ _phy.close(_v6TcpControlSocket,false);
|
|
|
+ _primaryPort = 0;
|
|
|
+ }
|
|
|
+ */
|
|
|
+
|
|
|
+ if(!_tcpHttpListenSocket)
|
|
|
+ printf("Error binding on port %d for HTTP listen socket\n", http_listen_port);
|
|
|
+ if(!_tcpSSLListenSocket)
|
|
|
+ printf("Error binding on port %d for SSL listen socket\n", ssl_listen_port);
|
|
|
+ _thread = Thread::start(this);
|
|
|
+ }
|
|
|
+
|
|
|
+ ButtFlare::~ButtFlare()
|
|
|
+ {
|
|
|
+ _run = false;
|
|
|
+ _phy.whack();
|
|
|
+ _phy.whack(); // TODO: Rationale?
|
|
|
+ Thread::join(_thread);
|
|
|
+ _phy.close(_tcpHttpListenSocket,false);
|
|
|
+ _phy.close(_tcpSSLListenSocket,false);
|
|
|
+ }
|
|
|
+ void ButtFlare::threadMain()
|
|
|
+ throw()
|
|
|
+ {
|
|
|
+ while(_run) {
|
|
|
+ _phy.poll(50); // in ms
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ void ButtFlare::phyOnTcpData(PhySocket *sock,void **uptr,void *data,unsigned long len)
|
|
|
+ {
|
|
|
+ TcpConnection *conn = cmap[sock];
|
|
|
+ unsigned char *buf = (unsigned char*)data;
|
|
|
+ std::string host = "";
|
|
|
+
|
|
|
+ for(int i=0; i<len; i++) { printf("buf[%d] = %d, (char) = %c\n", i, buf[i], (char*)buf[i]); }
|
|
|
+ printf("phyOnTcpData(len=%lu)\n", len);
|
|
|
+
|
|
|
+ if(conn == NULL) {
|
|
|
+ printf("No existing connection for this socket\n");
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ if(!conn->destination_sock) { // no connection yet
|
|
|
+
|
|
|
+ // Determine if HTTP or TLS/SSL
|
|
|
+ if(buf[0] == 22 && len > 100) // naive and incomplete way of checking for TLS, just for stub code
|
|
|
+ {
|
|
|
+ printf("TLS/SSL\n");
|
|
|
+ host = "127.0.0.1";
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ printf("HTTP\n");
|
|
|
+ host = "127.0.0.1";
|
|
|
+ }
|
|
|
+
|
|
|
+ if(host != "")
|
|
|
+ {
|
|
|
+ bool connected;
|
|
|
+ struct sockaddr_in in4;
|
|
|
+ memset(&in4,0,sizeof(in4));
|
|
|
+ in4.sin_family = AF_INET;
|
|
|
+ in4.sin_addr.s_addr = Utils::hton((uint32_t)(0x7f000001)); // right now we just listen for TCP @127.0.0.1
|
|
|
+ in4.sin_port = Utils::hton((uint16_t)http_listen_port);
|
|
|
+ conn->destination_sock = _phy.tcpConnect((const struct sockaddr *)&in4, connected, this);
|
|
|
+
|
|
|
+ if(!connected) {
|
|
|
+ printf("instant connect has occured\n");
|
|
|
+ }
|
|
|
+ if(!conn->destination_sock) {
|
|
|
+ printf("there was an error connecting to the remote host\n");
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ else // connection already established, just forward the data
|
|
|
+ {
|
|
|
+ int n = _phy.streamSend(conn->destination_sock, buf, len);
|
|
|
+ printf("wrote %d bytes (%p -> %p)\n", conn->origin_sock, conn->destination_sock);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ void ButtFlare::phyOnDatagram(PhySocket *sock,void **uptr,const struct sockaddr *localAddr,const struct sockaddr *from,void *data,unsigned long len)
|
|
|
+ {
|
|
|
+ }
|
|
|
+ void ButtFlare::phyOnTcpWritable(PhySocket *sock,void **uptr)
|
|
|
+ {
|
|
|
+ }
|
|
|
+ void ButtFlare::phyOnFileDescriptorActivity(PhySocket *sock,void **uptr,bool readable,bool writable)
|
|
|
+ {
|
|
|
+ }
|
|
|
+
|
|
|
+ void ButtFlare::phyOnTcpConnect(PhySocket *sock,void **uptr,bool success)
|
|
|
+ {
|
|
|
+ printf("phyOnTcpConnect()\n");
|
|
|
+ }
|
|
|
+
|
|
|
+ void ButtFlare::phyOnTcpAccept(PhySocket *sockL,PhySocket *sockN,void **uptrL,void **uptrN,const struct sockaddr *from)
|
|
|
+ {
|
|
|
+ TcpConnection *conn = new TcpConnection();
|
|
|
+ conn->origin_sock = sockN;
|
|
|
+ cmap[sockN]=conn; // add new connection
|
|
|
+ }
|
|
|
+
|
|
|
+ void ButtFlare::phyOnUnixClose(PhySocket *sock,void **uptr)
|
|
|
+ {
|
|
|
+ }
|
|
|
+ void ButtFlare::phyOnUnixData(PhySocket *sock,void **uptr,void *data,ssize_t len)
|
|
|
+ {
|
|
|
+ }
|
|
|
+ void ButtFlare::phyOnUnixWritable(PhySocket *sock,void **uptr,bool lwip_invoked)
|
|
|
+ {
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ void ButtFlare::phyOnTcpClose(PhySocket *sock,void **uptr)
|
|
|
+ {
|
|
|
+ TcpConnection *conn = cmap[sock];
|
|
|
+ if(conn == NULL) {
|
|
|
+ printf("No existing connection for this sock. Only closing origin sock\n");
|
|
|
+ // _phy.close(sock);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ // Close both ends
|
|
|
+ // _phy.close(conn->destination_sock);
|
|
|
+ // _phy.close(conn->origin_sock);
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+//#endif
|
Joseph Henry