New IPC work...

This commit is contained in:
Adam Ierymenko 2014-03-18 17:19:32 -07:00
parent 0b75992737
commit 6a1bd52854
3 changed files with 280 additions and 0 deletions

76
node/IpcConnection.hpp Normal file
View File

@ -0,0 +1,76 @@
/*
* ZeroTier One - Global Peer to Peer Ethernet
* Copyright (C) 2011-2014 ZeroTier Networks LLC
*
* 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_IPCCONNECTION_HPP
#define ZT_IPCCONNECTION_HPP
#include "Constants.hpp"
#include "Thread.hpp"
#include "NonCopyable.hpp"
#include "Mutex.hpp"
#include "SharedPtr.hpp"
#include "AtomicCounter.hpp"
namespace ZeroTier {
class IpcListener;
/**
* Interprocess communication connection
*/
class IpcConnection : NonCopyable
{
friend class IpcListener;
friend class SharedPtr<IpcConnection>;
public:
IpcConnection(const char *endpoint);
~IpcConnection();
void writeln(const char *format,...);
void close();
void threadMain()
throw();
private:
// Used by IpcListener to construct connections from incoming attempts
IpcConnection(int s);
void (*_handler)(void *,const SharedPtr<IpcConnection> &,const char *);
void *_arg;
volatile int _sock;
Thread _thread;
Mutex _writeLock;
AtomicCounter __refCount;
};
} // namespace ZeroTier
#endif

127
node/IpcListener.cpp Normal file
View File

@ -0,0 +1,127 @@
/*
* ZeroTier One - Global Peer to Peer Ethernet
* Copyright (C) 2011-2014 ZeroTier Networks LLC
*
* 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 <stdlib.h>
#include <string.h>
#include <errno.h>
#include <set>
#include "IpcListener.hpp"
#ifdef __WINDOWS__
#include <WinSock2.h>
#include <Windows.h>
#else
#include <sys/socket.h>
#include <sys/ud.h>
#include <unistd.h>
#endif
namespace ZeroTier {
IpcListener::IpcListener(cosnt char *ep,void (*commandHandler)(void *,const SharedPtr<IpcConnection> &,const char *),void *arg) :
_endpoint(ep),
_handler(commandHandler),
_arg(arg),
_sock(0)
{
struct sockaddr_un unaddr;
unaddr.sun_family = AF_UNIX;
if ((long)_endpoint.length() > (long)(sizeof(unaddr.sun_path) - 1))
throw std::runtime_error("IPC endpoint path too long");
strncpy(unaddr.sun_path,_endpoint.c_str(),sizeof(unaddr.sun_path));
for(int tries=0;tries<3;++tries) {
_sock = socket(AF_UNIX,SOCK_STREAM,0);
if (_sock <= 0)
throw std::runtime_error("unable to create socket of type AF_UNIX");
if (bind(_sock,(struct sockaddr *)&unaddr,sizeof(unaddr))) {
::close(_sock);
if (errno == EADDRINUSE) {
int testSock = socket(AF_UNIX,SOCK_STREAM,0);
if (testSock <= 0)
throw std::runtime_error("unable to create socket of type AF_UNIX");
if (connect(testSock,(struct sockaddr *)&unaddr,sizeof(unaddr))) {
// error indicates nothing is listening on other end, so unlink and try again
::close(testSock);
unlink(_endpoint.c_str());
} else {
// success means endpoint is being actively listened to by a process
::close(testSock);
throw std::runtime_error("IPC endpoint address in use");
}
} else throw std::runtime_error("IPC endpoint could not be bound");
}
}
if (listen(_sock,8)) {
::close(_sock);
throw std::runtime_error("listen() failed for bound AF_UNIX socket");
}
_thread = Thread::start(this);
}
IpcListener::~IpcListener()
{
int s = _sock;
_sock = 0;
if (s > 0) {
::shutdown(s,SHUT_RDWR);
::close(s);
}
Thread::join(_thread);
unlink(_endpoint.c_str());
}
void IpcListener::threadMain()
throw()
{
struct sockaddr_un unaddr;
socklen_t socklen;
int s;
unaddr.sun_family = AF_UNIX;
strncpy(unaddr.sun_path,_endpoint.c_str(),sizeof(unaddr.sun_path));
while (_sock > 0) {
socklen = sizeof(unaddr);
s = accept(_sock,(struct sockaddr *)unaddr,&socklen);
if (s <= 0)
break;
if (!_sock) {
::close(s);
break;
}
try {
_handler(_arg,SharedPtr<IpcConnection>(new IpcConnection(s)),(const char *)0);
} catch ( ... ) {} // handlers should not throw
}
}
} // namespace ZeroTier

77
node/IpcListener.hpp Normal file
View File

@ -0,0 +1,77 @@
/*
* ZeroTier One - Global Peer to Peer Ethernet
* Copyright (C) 2011-2014 ZeroTier Networks LLC
*
* 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_IPCLISTENER_HPP
#define ZT_IPCLISTENER_HPP
#include "Constants.hpp"
#include "Thread.hpp"
#include "NonCopyable.hpp"
#include "SharedPtr.hpp"
#include <string>
#include <stdexcept>
#ifdef __WINDOWS__
#define ZT_IPC_ENDPOINT "\\\\.\\pipe\\ZeroTierOne-control"
#else
#define ZT_IPC_ENDPOINT "/tmp/.ZeroTierOne-control"
#endif
namespace ZeroTier {
class IpcConnection;
/**
* IPC incoming connection listener (Unix domain sockets or named pipes on Windows)
*/
class IpcListener : NonCopyable
{
public:
/**
* @param commandHandler Function to call for each command
* @param arg First argument to pass to handler
* @throws std::runtime_error Unable to bind to endpoint
*/
IpcListener(cosnt char *ep,void (*commandHandler)(void *,const SharedPtr<IpcConnection> &,const char *),void *arg);
~IpcListener();
void threadMain()
throw();
private:
std::string _endpoint;
void (*_handler)(void *,const SharedPtr<IpcConnection> &,const char *);
void *_arg;
volatile int _sock;
Thread _thread;
};
} // namespace ZeroTier
#endif