Jigger with shutdown method to avoid a crash on CTRL+C in Windows. Feels a big hacky, might revisit later.

This commit is contained in:
Adam Ierymenko 2013-08-27 18:00:07 -04:00
parent 0afcf4877c
commit 01a70d09db
4 changed files with 27 additions and 31 deletions

View File

@ -210,7 +210,8 @@ Demarc::Port Demarc::send(Demarc::Port fromPort,const InetAddress &to,const void
void Demarc::_CBudpSocketPacketHandler(UdpSocket *sock,void *arg,const InetAddress &from,const void *data,unsigned int len)
{
((DemarcPortObj *)arg)->parent->_r->sw->onRemotePacket(((DemarcPortObj *)arg)->port,from,Buffer<4096>(data,len));
if (!((DemarcPortObj *)arg)->parent->_r->shutdownInProgress)
((DemarcPortObj *)arg)->parent->_r->sw->onRemotePacket(((DemarcPortObj *)arg)->port,from,Buffer<4096>(data,len));
}
} // namespace ZeroTier

View File

@ -282,6 +282,8 @@ void Network::_CBhandleTapData(void *arg,const MAC &from,const MAC &to,unsigned
if (!((Network *)arg)->_ready)
return;
const RuntimeEnvironment *_r = ((Network *)arg)->_r;
if (_r->shutdownInProgress)
return;
try {
_r->sw->onLocalEthernet(SharedPtr<Network>((Network *)arg),from,to,etherType,data);
} catch (std::exception &exc) {

View File

@ -188,15 +188,31 @@ struct _NodeImpl
volatile bool running;
volatile bool terminateNow;
// Helper used to rapidly terminate from run()
// run() calls this on all return paths
inline Node::ReasonForTermination terminateBecause(Node::ReasonForTermination r,const char *rstr)
{
RuntimeEnvironment *_r = &renv;
LOG("terminating: %s",rstr);
renv.shutdownInProgress = true;
Thread::sleep(500);
#ifndef __WINDOWS__
delete renv.netconfService;
#endif
delete renv.nc;
delete renv.sysEnv;
delete renv.topology;
delete renv.demarc;
delete renv.sw;
delete renv.multicaster;
delete renv.prng;
delete renv.log;
reasonForTerminationStr = rstr;
reasonForTermination = r;
running = false;
return r;
}
};
@ -257,7 +273,6 @@ Node::Node(const char *hp)
_impl(new _NodeImpl)
{
_NodeImpl *impl = (_NodeImpl *)_impl;
if ((hp)&&(strlen(hp) > 0))
impl->renv.homePath = hp;
else impl->renv.homePath = ZT_DEFAULTS.defaultHomePath;
@ -269,34 +284,9 @@ Node::Node(const char *hp)
Node::~Node()
{
_NodeImpl *impl = (_NodeImpl *)_impl;
#ifndef __WINDOWS__
delete impl->renv.netconfService;
#endif
delete impl->renv.nc;
delete impl->renv.sysEnv;
delete impl->renv.topology;
delete impl->renv.sw;
delete impl->renv.multicaster;
delete impl->renv.demarc;
delete impl->renv.prng;
delete impl->renv.log;
delete impl;
delete (_NodeImpl *)_impl;
}
/**
* Execute node in current thread
*
* This does not return until the node shuts down. Shutdown may be caused
* by an internally detected condition such as a new upgrade being
* available or a fatal error, or it may be signaled externally using
* the terminate() method.
*
* @return Reason for termination
*/
Node::ReasonForTermination Node::run()
throw()
{
@ -375,9 +365,9 @@ Node::ReasonForTermination Node::run()
// Create the core objects in RuntimeEnvironment: node config, demarcation
// point, switch, network topology database, and system environment
// watcher.
_r->demarc = new Demarc(_r);
_r->multicaster = new Multicaster();
_r->sw = new Switch(_r);
_r->demarc = new Demarc(_r);
_r->topology = new Topology(_r,(_r->homePath + ZT_PATH_SEPARATOR_S + "peer.db").c_str());
_r->sysEnv = new SysEnv(_r);
try {

View File

@ -61,6 +61,7 @@ class RuntimeEnvironment
{
public:
RuntimeEnvironment() :
shutdownInProgress(false),
log((Logger *)0),
prng((CMWC4096 *)0),
demarc((Demarc *)0),
@ -82,14 +83,16 @@ public:
Identity identity;
volatile bool shutdownInProgress;
// Order matters a bit here. These are constructed in this order
// and then deleted in the opposite order on Node exit.
Logger *log; // may be null
CMWC4096 *prng;
Demarc *demarc;
Multicaster *multicaster;
Switch *sw;
Demarc *demarc;
Topology *topology;
SysEnv *sysEnv;
NodeConfig *nc;