summaryrefslogtreecommitdiff
path: root/osdep
diff options
context:
space:
mode:
authorGrant Limberg <glimberg@gmail.com>2015-09-26 13:47:55 -0700
committerGrant Limberg <glimberg@gmail.com>2015-09-26 13:47:55 -0700
commite8cdff3eafd8096da22122eabddf57a09fe2bb90 (patch)
treed231aa6d9ccccc8ced6e1ead606ce16ff551cab9 /osdep
parent53d98343b7b444508259f6f1643e8d6724fb11e9 (diff)
parentf69454ec9879a0b0a424f743ca144d1123ef7e99 (diff)
downloadinfinitytier-e8cdff3eafd8096da22122eabddf57a09fe2bb90.tar.gz
infinitytier-e8cdff3eafd8096da22122eabddf57a09fe2bb90.zip
Merge branch 'adamierymenko-dev' into android-jni-dev
also update for changed function calls that now accept a local address # Conflicts: # include/ZeroTierOne.h # java/CMakeLists.txt # java/jni/Android.mk # java/jni/ZT1_jnicache.cpp # java/jni/ZT1_jnilookup.h # java/jni/ZT1_jniutils.cpp # java/jni/com_zerotierone_sdk_Node.cpp
Diffstat (limited to 'osdep')
-rw-r--r--osdep/Arp.cpp134
-rw-r--r--osdep/Arp.hpp156
-rw-r--r--osdep/Http.cpp6
-rw-r--r--osdep/Phy.hpp310
4 files changed, 584 insertions, 22 deletions
diff --git a/osdep/Arp.cpp b/osdep/Arp.cpp
new file mode 100644
index 00000000..f71dfb54
--- /dev/null
+++ b/osdep/Arp.cpp
@@ -0,0 +1,134 @@
+/*
+ * ZeroTier One - Network Virtualization Everywhere
+ * Copyright (C) 2011-2015 ZeroTier, Inc.
+ *
+ * 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/>.
+ *
+ * --
+ *
+ * ZeroTier may be used and distributed under the terms of the GPLv3, which
+ * are available at: http://www.gnu.org/licenses/gpl-3.0.html
+ *
+ * If you would like to embed ZeroTier into a commercial application or
+ * redistribute it in a modified binary form, please contact ZeroTier Networks
+ * LLC. Start here: http://www.zerotier.com/
+ */
+
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+
+#include "Arp.hpp"
+#include "OSUtils.hpp"
+
+namespace ZeroTier {
+
+static const uint8_t ARP_REQUEST_HEADER[8] = { 0x00,0x01,0x08,0x00,0x06,0x04,0x00,0x01 };
+static const uint8_t ARP_RESPONSE_HEADER[8] = { 0x00,0x01,0x08,0x00,0x06,0x04,0x00,0x02 };
+
+Arp::Arp() :
+ _cache(256),
+ _lastCleaned(OSUtils::now())
+{
+}
+
+void Arp::addLocal(uint32_t ip,const MAC &mac)
+{
+ _ArpEntry &e = _cache[ip];
+ e.lastQuerySent = 0; // local IP
+ e.lastResponseReceived = 0; // local IP
+ e.mac = mac;
+ e.local = true;
+}
+
+void Arp::remove(uint32_t ip)
+{
+ _cache.erase(ip);
+}
+
+uint32_t Arp::processIncomingArp(const void *arp,unsigned int len,void *response,unsigned int &responseLen,MAC &responseDest)
+{
+ const uint64_t now = OSUtils::now();
+ uint32_t ip = 0;
+
+ responseLen = 0;
+ responseDest.zero();
+
+ if (len > 28) {
+ if (!memcmp(arp,ARP_REQUEST_HEADER,8)) {
+ // Respond to ARP requests for locally-known IPs
+ _ArpEntry *targetEntry = _cache.get(reinterpret_cast<const uint32_t *>(arp)[6]);
+ if ((targetEntry)&&(targetEntry->local)) {
+ memcpy(response,ARP_RESPONSE_HEADER,8);
+ targetEntry->mac.copyTo(reinterpret_cast<uint8_t *>(response) + 8,6);
+ memcpy(reinterpret_cast<uint8_t *>(response) + 14,reinterpret_cast<const uint8_t *>(arp) + 24,4);
+ memcpy(reinterpret_cast<uint8_t *>(response) + 18,reinterpret_cast<const uint8_t *>(arp) + 8,10);
+ responseLen = 28;
+ responseDest.setTo(reinterpret_cast<const uint8_t *>(arp) + 8,6);
+ }
+ } else if (!memcmp(arp,ARP_RESPONSE_HEADER,8)) {
+ // Learn cache entries for remote IPs from relevant ARP replies
+ uint32_t responseIp = 0;
+ memcpy(&responseIp,reinterpret_cast<const uint8_t *>(arp) + 14,4);
+ _ArpEntry *queryEntry = _cache.get(responseIp);
+ if ((queryEntry)&&(!queryEntry->local)&&((now - queryEntry->lastQuerySent) <= ZT_ARP_QUERY_MAX_TTL)) {
+ queryEntry->lastResponseReceived = now;
+ queryEntry->mac.setTo(reinterpret_cast<const uint8_t *>(arp) + 8,6);
+ ip = responseIp;
+ }
+ }
+ }
+
+ if ((now - _lastCleaned) >= ZT_ARP_EXPIRE) {
+ _lastCleaned = now;
+ Hashtable< uint32_t,_ArpEntry >::Iterator i(_cache);
+ uint32_t *k = (uint32_t *)0;
+ _ArpEntry *v = (_ArpEntry *)0;
+ while (i.next(k,v)) {
+ if ((!v->local)&&((now - v->lastResponseReceived) >= ZT_ARP_EXPIRE))
+ _cache.erase(*k);
+ }
+ }
+
+ return ip;
+}
+
+MAC Arp::query(const MAC &localMac,uint32_t ip,void *query,unsigned int &queryLen,MAC &queryDest)
+{
+ const uint64_t now = OSUtils::now();
+
+ _ArpEntry &e = _cache[ip];
+
+ if ( ((e.mac)&&((now - e.lastResponseReceived) >= (ZT_ARP_EXPIRE / 3))) ||
+ ((!e.mac)&&((now - e.lastQuerySent) >= ZT_ARP_QUERY_INTERVAL)) ) {
+ e.lastQuerySent = now;
+
+ uint8_t *q = reinterpret_cast<uint8_t *>(query);
+ memcpy(q,ARP_REQUEST_HEADER,8); q += 8; // ARP request header information, always the same
+ localMac.copyTo(q,6); q += 6; // sending host address
+ memset(q,0,10); q += 10; // sending IP and target media address are ignored in requests
+ memcpy(q,&ip,4); // target IP address for resolution (IP already in big-endian byte order)
+ queryLen = 28;
+ if (e.mac)
+ queryDest = e.mac; // confirmation query, send directly to address holder
+ else queryDest = (uint64_t)0xffffffffffffULL; // broadcast query
+ } else {
+ queryLen = 0;
+ queryDest.zero();
+ }
+
+ return e.mac;
+}
+
+} // namespace ZeroTier
diff --git a/osdep/Arp.hpp b/osdep/Arp.hpp
new file mode 100644
index 00000000..b747cf85
--- /dev/null
+++ b/osdep/Arp.hpp
@@ -0,0 +1,156 @@
+/*
+ * ZeroTier One - Network Virtualization Everywhere
+ * Copyright (C) 2011-2015 ZeroTier, Inc.
+ *
+ * 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/>.
+ *
+ * --
+ *
+ * ZeroTier may be used and distributed under the terms of the GPLv3, which
+ * are available at: http://www.gnu.org/licenses/gpl-3.0.html
+ *
+ * If you would like to embed ZeroTier into a commercial application or
+ * redistribute it in a modified binary form, please contact ZeroTier Networks
+ * LLC. Start here: http://www.zerotier.com/
+ */
+
+#ifndef ZT_ARP_HPP
+#define ZT_ARP_HPP
+
+#include <stdint.h>
+
+#include <utility>
+
+#include "../node/Constants.hpp"
+#include "../node/Hashtable.hpp"
+#include "../node/MAC.hpp"
+
+/**
+ * Maximum possible ARP length
+ *
+ * ARPs are 28 bytes in length, but specify a 128 byte buffer since
+ * some weird extensions we may support in the future can pad them
+ * out to as long as 72 bytes.
+ */
+#define ZT_ARP_BUF_LENGTH 128
+
+/**
+ * Minimum permitted interval between sending ARP queries for a given IP
+ */
+#define ZT_ARP_QUERY_INTERVAL 2000
+
+/**
+ * Maximum time between query and response, otherwise responses are discarded to prevent poisoning
+ */
+#define ZT_ARP_QUERY_MAX_TTL 5000
+
+/**
+ * ARP expiration time
+ */
+#define ZT_ARP_EXPIRE 600000
+
+namespace ZeroTier {
+
+/**
+ * ARP cache and resolver
+ *
+ * To implement ARP:
+ *
+ * (1) Call processIncomingArp() on all ARP packets received and then always
+ * check responseLen after calling. If it is non-zero, send the contents
+ * of response to responseDest.
+ *
+ * (2) Call query() to look up IP addresses, and then check queryLen. If it
+ * is non-zero, send the contents of query to queryDest (usually broadcast).
+ *
+ * Note that either of these functions can technically generate a response or
+ * a query at any time, so their result parameters for sending ARPs should
+ * always be checked.
+ *
+ * This class is not thread-safe and must be guarded if used in multi-threaded
+ * code.
+ */
+class Arp
+{
+public:
+ Arp();
+
+ /**
+ * Set a local IP entry that we should respond to ARPs for
+ *
+ * @param mac Our local MAC address
+ * @param ip IP in big-endian byte order (sin_addr.s_addr)
+ */
+ void addLocal(uint32_t ip,const MAC &mac);
+
+ /**
+ * Delete a local IP entry or a cached ARP entry
+ *
+ * @param ip IP in big-endian byte order (sin_addr.s_addr)
+ */
+ void remove(uint32_t ip);
+
+ /**
+ * Process ARP packets
+ *
+ * For ARP queries, a response is generated and responseLen is set to its
+ * frame payload length in bytes.
+ *
+ * For ARP responses, the cache is populated and the IP address entry that
+ * was learned is returned.
+ *
+ * @param arp ARP frame data
+ * @param len Length of ARP frame (usually 28)
+ * @param response Response buffer -- MUST be a minimum of ZT_ARP_BUF_LENGTH in size
+ * @param responseLen Response length, or set to 0 if no response
+ * @param responseDest Destination of response, or set to null if no response
+ * @return IP address learned or 0 if no new IPs in cache
+ */
+ uint32_t processIncomingArp(const void *arp,unsigned int len,void *response,unsigned int &responseLen,MAC &responseDest);
+
+ /**
+ * Get the MAC corresponding to an IP, generating a query if needed
+ *
+ * This returns a MAC for a remote IP. The local MAC is returned for local
+ * IPs as well. It may also generate a query if the IP is not known or the
+ * entry needs to be refreshed. In this case queryLen will be set to a
+ * non-zero value, so this should always be checked on return even if the
+ * MAC returned is non-null.
+ *
+ * @param localMac Local MAC address of host interface
+ * @param ip IP to look up
+ * @param query Buffer for generated query -- MUST be a minimum of ZT_ARP_BUF_LENGTH in size
+ * @param queryLen Length of generated query, or set to 0 if no query generated
+ * @param queryDest Destination of query, or set to null if no query generated
+ * @return MAC or 0 if no cached entry for this IP
+ */
+ MAC query(const MAC &localMac,uint32_t ip,void *query,unsigned int &queryLen,MAC &queryDest);
+
+private:
+ struct _ArpEntry
+ {
+ _ArpEntry() : lastQuerySent(0),lastResponseReceived(0),mac(),local(false) {}
+ uint64_t lastQuerySent; // Time last query was sent or 0 for local IP
+ uint64_t lastResponseReceived; // Time of last ARP response or 0 for local IP
+ MAC mac; // MAC address of device responsible for IP or null if not known yet
+ bool local; // True if this is a local ARP entry
+ };
+
+ Hashtable< uint32_t,_ArpEntry > _cache;
+ uint64_t _lastCleaned;
+};
+
+} // namespace ZeroTier
+
+#endif
diff --git a/osdep/Http.cpp b/osdep/Http.cpp
index d491b062..0eb7c4c6 100644
--- a/osdep/Http.cpp
+++ b/osdep/Http.cpp
@@ -68,7 +68,7 @@ struct HttpPhyHandler
inline void phyOnTcpConnect(PhySocket *sock,void **uptr,bool success)
{
if (success) {
- phy->tcpSetNotifyWritable(sock,true);
+ phy->setNotifyWritable(sock,true);
} else {
*responseBody = "connection failed";
error = true;
@@ -92,12 +92,12 @@ struct HttpPhyHandler
inline void phyOnTcpWritable(PhySocket *sock,void **uptr)
{
if (writePtr < writeSize) {
- long n = phy->tcpSend(sock,writeBuf + writePtr,writeSize - writePtr,true);
+ long n = phy->streamSend(sock,writeBuf + writePtr,writeSize - writePtr,true);
if (n > 0)
writePtr += n;
}
if (writePtr >= writeSize)
- phy->tcpSetNotifyWritable(sock,false);
+ phy->setNotifyWritable(sock,false);
}
http_parser parser;
diff --git a/osdep/Phy.hpp b/osdep/Phy.hpp
index 2ea68b9d..7f790e5d 100644
--- a/osdep/Phy.hpp
+++ b/osdep/Phy.hpp
@@ -46,6 +46,7 @@
#define ZT_PHY_SOCKFD_VALID(s) ((s) != INVALID_SOCKET)
#define ZT_PHY_CLOSE_SOCKET(s) ::closesocket(s)
#define ZT_PHY_MAX_SOCKETS (FD_SETSIZE)
+#define ZT_PHY_MAX_INTERCEPTS ZT_PHY_MAX_SOCKETS
#define ZT_PHY_SOCKADDR_STORAGE_TYPE struct sockaddr_storage
#else // not Windows
@@ -58,6 +59,7 @@
#include <sys/types.h>
#include <sys/select.h>
#include <sys/socket.h>
+#include <sys/un.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
@@ -67,8 +69,14 @@
#define ZT_PHY_SOCKFD_VALID(s) ((s) > -1)
#define ZT_PHY_CLOSE_SOCKET(s) ::close(s)
#define ZT_PHY_MAX_SOCKETS (FD_SETSIZE)
+#define ZT_PHY_MAX_INTERCEPTS ZT_PHY_MAX_SOCKETS
#define ZT_PHY_SOCKADDR_STORAGE_TYPE struct sockaddr_storage
+#if defined(__linux__) || defined(linux) || defined(__LINUX__) || defined(__linux)
+#define ZT_PHY_HAVE_EVENTFD 1
+#include <sys/eventfd.h>
+#endif
+
#endif // Windows or not
namespace ZeroTier {
@@ -87,6 +95,8 @@ typedef void PhySocket;
* This class is templated on a pointer to a handler class which must
* implement the following functions:
*
+ * For all platforms:
+ *
* phyOnDatagram(PhySocket *sock,void **uptr,const struct sockaddr *from,void *data,unsigned long len)
* phyOnTcpConnect(PhySocket *sock,void **uptr,bool success)
* phyOnTcpAccept(PhySocket *sockL,PhySocket *sockN,void **uptrL,void **uptrN,const struct sockaddr *from)
@@ -94,6 +104,16 @@ typedef void PhySocket;
* phyOnTcpData(PhySocket *sock,void **uptr,void *data,unsigned long len)
* phyOnTcpWritable(PhySocket *sock,void **uptr)
*
+ * On Linux/OSX/Unix only (not required/used on Windows or elsewhere):
+ *
+ * phyOnUnixAccept(PhySocket *sockL,PhySocket *sockN,void **uptrL,void **uptrN)
+ * phyOnUnixClose(PhySocket *sock,void **uptr)
+ * phyOnUnixData(PhySocket *sock,void **uptr,void *data,unsigned long len)
+ * phyOnUnixWritable(PhySocket *sock,void **uptr)
+ * phyOnSocketPairEndpointClose(PhySocket *sock,void **uptr)
+ * phyOnSocketPairEndpointData(PhySocket *sock,void **uptr,void *data,unsigned long len)
+ * phyOnSocketPairEndpointWritable(PhySocket *sock,void **uptr)
+ *
* These templates typically refer to function objects. Templates are used to
* avoid the call overhead of indirection, which is surprisingly high for high
* bandwidth applications pushing a lot of packets.
@@ -129,7 +149,10 @@ private:
ZT_PHY_SOCKET_TCP_IN = 0x03,
ZT_PHY_SOCKET_TCP_LISTEN = 0x04,
ZT_PHY_SOCKET_RAW = 0x05,
- ZT_PHY_SOCKET_UDP = 0x06
+ ZT_PHY_SOCKET_UDP = 0x06,
+ ZT_PHY_SOCKET_UNIX_IN = 0x07,
+ ZT_PHY_SOCKET_UNIX_LISTEN = 0x08,
+ ZT_PHY_SOCKET_PAIR_ENDPOINT = 0x09
};
struct PhySocketImpl
@@ -218,7 +241,16 @@ public:
}
/**
+ * @param s Socket object
+ * @return Underlying OS-type (usually int or long) file descriptor associated with object
+ */
+ static inline ZT_PHY_SOCKFD_TYPE getDescriptor(PhySocket *s) throw() { return reinterpret_cast<PhySocketImpl *>(s)->sock; }
+
+ /**
* Cause poll() to stop waiting immediately
+ *
+ * This can be used to reset the polling loop after changes that require
+ * attention, or to shut down a background thread that is waiting, etc.
*/
inline void whack()
{
@@ -239,6 +271,58 @@ public:
*/
inline unsigned long maxCount() const throw() { return ZT_PHY_MAX_SOCKETS; }
+#ifdef __UNIX_LIKE__
+ /**
+ * Create a two-way socket pair
+ *
+ * This uses socketpair() to create a local domain pair. The returned
+ * PhySocket holds the local side of the socket pair, while the
+ * supplied fd variable is set to the descriptor for the remote side.
+ *
+ * The local side is set to O_NONBLOCK to work with our poll loop, but
+ * the remote descriptor is left untouched. It's up to the caller to
+ * set any required fcntl(), ioctl(), or setsockopt() settings there.
+ * It's also up to the caller to close the remote descriptor when
+ * done, if necessary.
+ *
+ * @param remoteSocketDescriptor Result parameter set to remote end of socket pair's socket FD
+ * @param uptr Pointer to associate with local side of socket pair
+ * @return PhySocket for local side of socket pair
+ */
+ inline PhySocket *createSocketPair(ZT_PHY_SOCKFD_TYPE &remoteSocketDescriptor,void *uptr = (void *)0)
+ {
+ if (_socks.size() >= ZT_PHY_MAX_SOCKETS)
+ return (PhySocket *)0;
+
+ int fd[2]; fd[0] = -1; fd[1] = -1;
+ if ((::socketpair(PF_LOCAL,SOCK_STREAM,0,fd) != 0)||(fd[0] <= 0)||(fd[1] <= 0))
+ return (PhySocket *)0;
+ fcntl(fd[0],F_SETFL,O_NONBLOCK);
+
+ try {
+ _socks.push_back(PhySocketImpl());
+ } catch ( ... ) {
+ ZT_PHY_CLOSE_SOCKET(fd[0]);
+ ZT_PHY_CLOSE_SOCKET(fd[1]);
+ return (PhySocket *)0;
+ }
+ PhySocketImpl &sws = _socks.back();
+
+ if ((long)fd[0] > _nfds)
+ _nfds = (long)fd[0];
+ FD_SET(fd[0],&_readfds);
+ sws.type = ZT_PHY_SOCKET_PAIR_ENDPOINT;
+ sws.sock = fd[0];
+ sws.uptr = uptr;
+ memset(&(sws.saddr),0,sizeof(struct sockaddr_storage));
+ // no sockaddr for this socket type, leave saddr null
+
+ remoteSocketDescriptor = fd[1];
+
+ return (PhySocket *)&sws;
+ }
+#endif // __UNIX_LIKE__
+
/**
* Bind a UDP socket
*
@@ -358,6 +442,64 @@ public:
#endif
}
+#ifdef __UNIX_LIKE__
+ /**
+ * Listen for connections on a Unix domain socket
+ *
+ * @param path Path to Unix domain socket
+ * @param uptr Arbitrary pointer to associate
+ * @return PhySocket or NULL if cannot bind
+ */
+ inline PhySocket *unixListen(const char *path,void *uptr = (void *)0)
+ {
+ struct sockaddr_un sun;
+
+ if (_socks.size() >= ZT_PHY_MAX_SOCKETS)
+ return (PhySocket *)0;
+
+ memset(&sun,0,sizeof(sun));
+ sun.sun_family = AF_UNIX;
+ if (strlen(path) >= sizeof(sun.sun_path))
+ return (PhySocket *)0;
+ strcpy(sun.sun_path,path);
+
+ ZT_PHY_SOCKFD_TYPE s = ::socket(PF_UNIX,SOCK_STREAM,0);
+ if (!ZT_PHY_SOCKFD_VALID(s))
+ return (PhySocket *)0;
+
+ ::fcntl(s,F_SETFL,O_NONBLOCK);
+
+ ::unlink(path);
+ if (::bind(s,(struct sockaddr *)&sun,sizeof(struct sockaddr_un)) != 0) {
+ ZT_PHY_CLOSE_SOCKET(s);
+ return (PhySocket *)0;
+ }
+ if (::listen(s,128) != 0) {
+ ZT_PHY_CLOSE_SOCKET(s);
+ return (PhySocket *)0;
+ }
+
+ try {
+ _socks.push_back(PhySocketImpl());
+ } catch ( ... ) {
+ ZT_PHY_CLOSE_SOCKET(s);
+ return (PhySocket *)0;
+ }
+ PhySocketImpl &sws = _socks.back();
+
+ if ((long)s > _nfds)
+ _nfds = (long)s;
+ FD_SET(s,&_readfds);
+ sws.type = ZT_PHY_SOCKET_UNIX_LISTEN;
+ sws.sock = s;
+ sws.uptr = uptr;
+ memset(&(sws.saddr),0,sizeof(struct sockaddr_storage));
+ memcpy(&(sws.saddr),&sun,sizeof(struct sockaddr_un));
+
+ return (PhySocket *)&sws;
+ }
+#endif // __UNIX_LIKE__
+
/**
* Bind a local listen socket to listen for new TCP connections
*
@@ -523,19 +665,21 @@ public:
}
/**
- * Attempt to send data to a TCP connection (non-blocking)
+ * Attempt to send data to a stream socket (non-blocking)
*
* If -1 is returned, the socket should no longer be used as it is now
* destroyed. If callCloseHandler is true, the close handler will be
* called before the function returns.
*
- * @param sock An open TCP socket (other socket types will fail)
+ * This can be used with TCP, Unix, or socket pair sockets.
+ *
+ * @param sock An open stream socket (other socket types will fail)
* @param data Data to send
* @param len Length of data
* @param callCloseHandler If true, call close handler on socket closing failure condition (default: true)
* @return Number of bytes actually sent or -1 on fatal error (socket closure)
*/
- inline long tcpSend(PhySocket *sock,const void *data,unsigned long len,bool callCloseHandler = true)
+ inline long streamSend(PhySocket *sock,const void *data,unsigned long len,bool callCloseHandler = true)
{
PhySocketImpl &sws = *(reinterpret_cast<PhySocketImpl *>(sock));
#if defined(_WIN32) || defined(_WIN64)
@@ -573,17 +717,58 @@ public:
return n;
}
+#ifdef __UNIX_LIKE__
/**
- * Set whether we want to be notified via the TCP writability handler when a socket is writable
+ * Attempt to send data to a Unix domain socket connection (non-blocking)
+ *
+ * If -1 is returned, the socket should no longer be used as it is now
+ * destroyed. If callCloseHandler is true, the close handler will be
+ * called before the function returns.
+ *
+ * @param sock An open Unix socket (other socket types will fail)
+ * @param data Data to send
+ * @param len Length of data
+ * @param callCloseHandler If true, call close handler on socket closing failure condition (default: true)
+ * @return Number of bytes actually sent or -1 on fatal error (socket closure)
+ */
+ inline long unixSend(PhySocket *sock,const void *data,unsigned long len,bool callCloseHandler = true)
+ {
+ PhySocketImpl &sws = *(reinterpret_cast<PhySocketImpl *>(sock));
+ long n = (long)::write(sws.sock,data,len);
+ if (n < 0) {
+ switch(errno) {
+#ifdef EAGAIN
+ case EAGAIN:
+#endif
+#if defined(EWOULDBLOCK) && ( !defined(EAGAIN) || (EWOULDBLOCK != EAGAIN) )
+ case EWOULDBLOCK:
+#endif
+#ifdef EINTR
+ case EINTR:
+#endif
+ return 0;
+ default:
+ this->close(sock,callCloseHandler);
+ return -1;
+ }
+ }
+ return n;
+ }
+#endif // __UNIX_LIKE__
+
+ /**
+ * For streams, sets whether we want to be notified that the socket is writable
+ *
+ * This can be used with TCP, Unix, or socket pair sockets.
*
* Call whack() if this is being done from another thread and you want
* it to take effect immediately. Otherwise it is only guaranteed to
* take effect on the next poll().
*
- * @param sock TCP connection socket (other types are not valid)
+ * @param sock Stream connection socket
* @param notifyWritable Want writable notifications?
*/
- inline const void tcpSetNotifyWritable(PhySocket *sock,bool notifyWritable)
+ inline const void setNotifyWritable(PhySocket *sock,bool notifyWritable)
{
PhySocketImpl &sws = *(reinterpret_cast<PhySocketImpl *>(sock));
if (notifyWritable) {
@@ -727,6 +912,77 @@ public:
}
break;
+ case ZT_PHY_SOCKET_UNIX_IN: {
+#ifdef __UNIX_LIKE__
+ ZT_PHY_SOCKFD_TYPE sock = s->sock; // if closed, s->sock becomes invalid as s is no longer dereferencable
+ if (FD_ISSET(sock,&rfds)) {
+ long n = (long)::read(sock,buf,sizeof(buf));
+ if (n <= 0) {
+ this->close((PhySocket *)&(*s),true);
+ } else {
+ try {
+ _handler->phyOnUnixData((PhySocket *)&(*s),&(s->uptr),(void *)buf,(unsigned long)n);
+ } catch ( ... ) {}
+ }
+ }
+ if ((FD_ISSET(sock,&wfds))&&(FD_ISSET(sock,&_writefds))) {
+ try {
+ _handler->phyOnUnixWritable((PhySocket *)&(*s),&(s->uptr));
+ } catch ( ... ) {}
+ }
+#endif // __UNIX_LIKE__
+ } break;
+
+ case ZT_PHY_SOCKET_UNIX_LISTEN:
+#ifdef __UNIX_LIKE__
+ if (FD_ISSET(s->sock,&rfds)) {
+ memset(&ss,0,sizeof(ss));
+ socklen_t slen = sizeof(ss);
+ ZT_PHY_SOCKFD_TYPE newSock = ::accept(s->sock,(struct sockaddr *)&ss,&slen);
+ if (ZT_PHY_SOCKFD_VALID(newSock)) {
+ if (_socks.size() >= ZT_PHY_MAX_SOCKETS) {
+ ZT_PHY_CLOSE_SOCKET(newSock);
+ } else {
+ fcntl(newSock,F_SETFL,O_NONBLOCK);
+ _socks.push_back(PhySocketImpl());
+ PhySocketImpl &sws = _socks.back();
+ FD_SET(newSock,&_readfds);
+ if ((long)newSock > _nfds)
+ _nfds = (long)newSock;
+ sws.type = ZT_PHY_SOCKET_UNIX_IN;
+ sws.sock = newSock;
+ sws.uptr = (void *)0;
+ memcpy(&(sws.saddr),&ss,sizeof(struct sockaddr_storage));
+ try {
+ _handler->phyOnUnixAccept((PhySocket *)&(*s),(PhySocket *)&(_socks.back()),&(s->uptr),&(sws.uptr));
+ } catch ( ... ) {}
+ }
+ }
+ }
+#endif // __UNIX_LIKE__
+ break;
+
+ case ZT_PHY_SOCKET_PAIR_ENDPOINT: {
+#ifdef __UNIX_LIKE__
+ ZT_PHY_SOCKFD_TYPE sock = s->sock; // if closed, s->sock becomes invalid as s is no longer dereferencable
+ if (FD_ISSET(sock,&rfds)) {
+ long n = (long)::read(sock,buf,sizeof(buf));
+ if (n <= 0) {
+ this->close((PhySocket *)&(*s),true);
+ } else {
+ try {
+ _handler->phyOnSocketPairEndpointData((PhySocket *)&(*s),&(s->uptr),(void *)buf,(unsigned long)n);
+ } catch ( ... ) {}
+ }
+ }
+ if ((FD_ISSET(sock,&wfds))&&(FD_ISSET(sock,&_writefds))) {
+ try {
+ _handler->phyOnSocketPairEndpointWritable((PhySocket *)&(*s),&(s->uptr));
+ } catch ( ... ) {}
+ }
+#endif // __UNIX_LIKE__
+ } break;
+
default:
break;
@@ -758,24 +1014,40 @@ public:
ZT_PHY_CLOSE_SOCKET(sws.sock);
- switch(sws.type) {
- case ZT_PHY_SOCKET_TCP_OUT_PENDING:
- if (callHandlers) {
+#ifdef __UNIX_LIKE__
+ if (sws.type == ZT_PHY_SOCKET_UNIX_LISTEN)
+ ::unlink(((struct sockaddr_un *)(&(sws.saddr)))->sun_path);
+#endif // __UNIX_LIKE__
+
+ if (callHandlers) {
+ switch(sws.type) {
+ case ZT_PHY_SOCKET_TCP_OUT_PENDING:
try {
_handler->phyOnTcpConnect(sock,&(sws.uptr),false);
} catch ( ... ) {}
- }
- break;
- case ZT_PHY_SOCKET_TCP_OUT_CONNECTED:
- case ZT_PHY_SOCKET_TCP_IN:
- if (callHandlers) {
+ break;
+ case ZT_PHY_SOCKET_TCP_OUT_CONNECTED:
+ case ZT_PHY_SOCKET_TCP_IN:
try {
_handler->phyOnTcpClose(sock,&(sws.uptr));
} catch ( ... ) {}
- }
- break;
- default:
- break;
+ break;
+ case ZT_PHY_SOCKET_UNIX_IN:
+#ifdef __UNIX_LIKE__
+ try {
+ _handler->phyOnUnixClose(sock,&(sws.uptr));
+ } catch ( ... ) {}
+#endif // __UNIX_LIKE__
+ break;
+ case ZT_PHY_SOCKET_PAIR_ENDPOINT:
+#ifdef __UNIX_LIKE__
+ try {
+ _handler->phyOnSocketPairEndpointClose(sock,&(sws.uptr));
+ } catch ( ... ) {}
+#endif // __UNIX_LIKE__
+ default:
+ break;
+ }
}
// Causes entry to be deleted from list in poll(), ignored elsewhere