Add Signals class.
This commit is contained in:
parent
70d6e0e62c
commit
4f188a2cd8
6 changed files with 201 additions and 42 deletions
|
@ -23,9 +23,11 @@ set(HEADERS
|
||||||
src/base/io/Json.h
|
src/base/io/Json.h
|
||||||
src/base/io/Watcher.h
|
src/base/io/Watcher.h
|
||||||
src/base/kernel/interfaces/IConfigListener.h
|
src/base/kernel/interfaces/IConfigListener.h
|
||||||
|
src/base/kernel/interfaces/ISignalListener.h
|
||||||
src/base/kernel/interfaces/IWatcherListener.h
|
src/base/kernel/interfaces/IWatcherListener.h
|
||||||
src/base/kernel/Entry.h
|
src/base/kernel/Entry.h
|
||||||
src/base/kernel/Process.h
|
src/base/kernel/Process.h
|
||||||
|
src/base/kernel/Signals.h
|
||||||
src/base/tools/Arguments.h
|
src/base/tools/Arguments.h
|
||||||
src/base/tools/Handle.h
|
src/base/tools/Handle.h
|
||||||
src/base/tools/String.h
|
src/base/tools/String.h
|
||||||
|
@ -109,6 +111,7 @@ set(SOURCES
|
||||||
src/base/io/Watcher.cpp
|
src/base/io/Watcher.cpp
|
||||||
src/base/kernel/Entry.cpp
|
src/base/kernel/Entry.cpp
|
||||||
src/base/kernel/Process.cpp
|
src/base/kernel/Process.cpp
|
||||||
|
src/base/kernel/Signals.cpp
|
||||||
src/base/tools/Arguments.cpp
|
src/base/tools/Arguments.cpp
|
||||||
src/base/tools/Handle.cpp
|
src/base/tools/Handle.cpp
|
||||||
src/base/tools/String.cpp
|
src/base/tools/String.cpp
|
||||||
|
|
52
src/App.cpp
52
src/App.cpp
|
@ -30,6 +30,7 @@
|
||||||
|
|
||||||
#include "api/Api.h"
|
#include "api/Api.h"
|
||||||
#include "App.h"
|
#include "App.h"
|
||||||
|
#include "base/kernel/Signals.h"
|
||||||
#include "common/Console.h"
|
#include "common/Console.h"
|
||||||
#include "common/cpu/Cpu.h"
|
#include "common/cpu/Cpu.h"
|
||||||
#include "common/log/Log.h"
|
#include "common/log/Log.h"
|
||||||
|
@ -49,16 +50,11 @@
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
||||||
xmrig::App *xmrig::App::m_self = nullptr;
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
xmrig::App::App(Process *process) :
|
xmrig::App::App(Process *process) :
|
||||||
m_console(nullptr),
|
m_console(nullptr),
|
||||||
m_httpd(nullptr)
|
m_httpd(nullptr),
|
||||||
|
m_signals(nullptr)
|
||||||
{
|
{
|
||||||
m_self = this;
|
|
||||||
|
|
||||||
m_controller = new xmrig::Controller(process);
|
m_controller = new xmrig::Controller(process);
|
||||||
if (m_controller->init() != 0) {
|
if (m_controller->init() != 0) {
|
||||||
return;
|
return;
|
||||||
|
@ -67,10 +63,6 @@ xmrig::App::App(Process *process) :
|
||||||
if (!m_controller->config()->isBackground()) {
|
if (!m_controller->config()->isBackground()) {
|
||||||
m_console = new Console(this);
|
m_console = new Console(this);
|
||||||
}
|
}
|
||||||
|
|
||||||
uv_signal_init(uv_default_loop(), &m_sigHUP);
|
|
||||||
uv_signal_init(uv_default_loop(), &m_sigINT);
|
|
||||||
uv_signal_init(uv_default_loop(), &m_sigTERM);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -78,6 +70,7 @@ xmrig::App::~App()
|
||||||
{
|
{
|
||||||
uv_tty_reset_mode();
|
uv_tty_reset_mode();
|
||||||
|
|
||||||
|
delete m_signals;
|
||||||
delete m_console;
|
delete m_console;
|
||||||
delete m_controller;
|
delete m_controller;
|
||||||
|
|
||||||
|
@ -93,9 +86,7 @@ int xmrig::App::exec()
|
||||||
return 2;
|
return 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
uv_signal_start(&m_sigHUP, App::onSignal, SIGHUP);
|
m_signals = new Signals(this);
|
||||||
uv_signal_start(&m_sigINT, App::onSignal, SIGINT);
|
|
||||||
uv_signal_start(&m_sigTERM, App::onSignal, SIGTERM);
|
|
||||||
|
|
||||||
background();
|
background();
|
||||||
|
|
||||||
|
@ -172,21 +163,7 @@ void xmrig::App::onConsoleCommand(char command)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void xmrig::App::close()
|
void xmrig::App::onSignal(int signum)
|
||||||
{
|
|
||||||
m_controller->network()->stop();
|
|
||||||
Workers::stop();
|
|
||||||
|
|
||||||
uv_stop(uv_default_loop());
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void xmrig::App::release()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void xmrig::App::onSignal(uv_signal_t *handle, int signum)
|
|
||||||
{
|
{
|
||||||
switch (signum)
|
switch (signum)
|
||||||
{
|
{
|
||||||
|
@ -206,6 +183,19 @@ void xmrig::App::onSignal(uv_signal_t *handle, int signum)
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
uv_signal_stop(handle);
|
close();
|
||||||
m_self->close();
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void xmrig::App::close()
|
||||||
|
{
|
||||||
|
m_controller->network()->stop();
|
||||||
|
Workers::stop();
|
||||||
|
|
||||||
|
uv_stop(uv_default_loop());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void xmrig::App::release()
|
||||||
|
{
|
||||||
}
|
}
|
||||||
|
|
16
src/App.h
16
src/App.h
|
@ -27,9 +27,7 @@
|
||||||
#define XMRIG_APP_H
|
#define XMRIG_APP_H
|
||||||
|
|
||||||
|
|
||||||
#include <uv.h>
|
#include "base/kernel/interfaces/ISignalListener.h"
|
||||||
|
|
||||||
|
|
||||||
#include "common/interfaces/IConsoleListener.h"
|
#include "common/interfaces/IConsoleListener.h"
|
||||||
|
|
||||||
|
|
||||||
|
@ -43,9 +41,10 @@ namespace xmrig {
|
||||||
|
|
||||||
class Controller;
|
class Controller;
|
||||||
class Process;
|
class Process;
|
||||||
|
class Signals;
|
||||||
|
|
||||||
|
|
||||||
class App : public IConsoleListener
|
class App : public IConsoleListener, public ISignalListener
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
App(Process *process);
|
App(Process *process);
|
||||||
|
@ -55,21 +54,16 @@ public:
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
void onConsoleCommand(char command) override;
|
void onConsoleCommand(char command) override;
|
||||||
|
void onSignal(int signum) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void background();
|
void background();
|
||||||
void close();
|
void close();
|
||||||
void release();
|
void release();
|
||||||
|
|
||||||
static void onSignal(uv_signal_t *handle, int signum);
|
|
||||||
|
|
||||||
static App *m_self;
|
|
||||||
|
|
||||||
Console *m_console;
|
Console *m_console;
|
||||||
Httpd *m_httpd;
|
Httpd *m_httpd;
|
||||||
uv_signal_t m_sigHUP;
|
Signals *m_signals;
|
||||||
uv_signal_t m_sigINT;
|
|
||||||
uv_signal_t m_sigTERM;
|
|
||||||
xmrig::Controller *m_controller;
|
xmrig::Controller *m_controller;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
63
src/base/kernel/Signals.cpp
Normal file
63
src/base/kernel/Signals.cpp
Normal file
|
@ -0,0 +1,63 @@
|
||||||
|
/* XMRig
|
||||||
|
* Copyright 2010 Jeff Garzik <jgarzik@pobox.com>
|
||||||
|
* Copyright 2012-2014 pooler <pooler@litecoinpool.org>
|
||||||
|
* Copyright 2014 Lucas Jones <https://github.com/lucasjones>
|
||||||
|
* Copyright 2014-2016 Wolf9466 <https://github.com/OhGodAPet>
|
||||||
|
* Copyright 2016 Jay D Dee <jayddee246@gmail.com>
|
||||||
|
* Copyright 2017-2018 XMR-Stak <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
|
||||||
|
* Copyright 2018 SChernykh <https://github.com/SChernykh>
|
||||||
|
* Copyright 2016-2018 XMRig <https://github.com/xmrig>, <support@xmrig.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 <uv.h>
|
||||||
|
|
||||||
|
|
||||||
|
#include "base/kernel/interfaces/ISignalListener.h"
|
||||||
|
#include "base/kernel/Signals.h"
|
||||||
|
#include "base/tools/Handle.h"
|
||||||
|
|
||||||
|
|
||||||
|
static const int signums[xmrig::Signals::kSignalsCount] = { SIGHUP, SIGINT, SIGTERM };
|
||||||
|
|
||||||
|
|
||||||
|
xmrig::Signals::Signals(ISignalListener *listener)
|
||||||
|
: m_listener(listener)
|
||||||
|
{
|
||||||
|
for (size_t i = 0; i < kSignalsCount; ++i) {
|
||||||
|
uv_signal_t *signal = new uv_signal_t;
|
||||||
|
signal->data = this;
|
||||||
|
|
||||||
|
m_signals[i] = signal;
|
||||||
|
|
||||||
|
uv_signal_init(uv_default_loop(), signal);
|
||||||
|
uv_signal_start(signal, xmrig::Signals::onSignal, signums[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
xmrig::Signals::~Signals()
|
||||||
|
{
|
||||||
|
for (size_t i = 0; i < kSignalsCount; ++i) {
|
||||||
|
Handle::close(m_signals[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void xmrig::Signals::onSignal(uv_signal_t *handle, int signum)
|
||||||
|
{
|
||||||
|
static_cast<xmrig::Signals *>(handle->data)->m_listener->onSignal(signum);
|
||||||
|
}
|
62
src/base/kernel/Signals.h
Normal file
62
src/base/kernel/Signals.h
Normal file
|
@ -0,0 +1,62 @@
|
||||||
|
/* XMRig
|
||||||
|
* Copyright 2010 Jeff Garzik <jgarzik@pobox.com>
|
||||||
|
* Copyright 2012-2014 pooler <pooler@litecoinpool.org>
|
||||||
|
* Copyright 2014 Lucas Jones <https://github.com/lucasjones>
|
||||||
|
* Copyright 2014-2016 Wolf9466 <https://github.com/OhGodAPet>
|
||||||
|
* Copyright 2016 Jay D Dee <jayddee246@gmail.com>
|
||||||
|
* Copyright 2017-2018 XMR-Stak <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
|
||||||
|
* Copyright 2018-2019 SChernykh <https://github.com/SChernykh>
|
||||||
|
* Copyright 2016-2019 XMRig <https://github.com/xmrig>, <support@xmrig.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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef XMRIG_SIGNALS_H
|
||||||
|
#define XMRIG_SIGNALS_H
|
||||||
|
|
||||||
|
|
||||||
|
#include <stddef.h>
|
||||||
|
|
||||||
|
|
||||||
|
typedef struct uv_signal_s uv_signal_t;
|
||||||
|
|
||||||
|
|
||||||
|
namespace xmrig {
|
||||||
|
|
||||||
|
|
||||||
|
class ISignalListener;
|
||||||
|
|
||||||
|
|
||||||
|
class Signals
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
constexpr static const size_t kSignalsCount = 3;
|
||||||
|
|
||||||
|
Signals(ISignalListener *listener);
|
||||||
|
~Signals();
|
||||||
|
|
||||||
|
private:
|
||||||
|
void close(int signum);
|
||||||
|
|
||||||
|
static void onSignal(uv_signal_t *handle, int signum);
|
||||||
|
|
||||||
|
ISignalListener *m_listener;
|
||||||
|
uv_signal_t *m_signals[kSignalsCount];
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
} /* namespace xmrig */
|
||||||
|
|
||||||
|
|
||||||
|
#endif /* XMRIG_SIGNALS_H */
|
47
src/base/kernel/interfaces/ISignalListener.h
Normal file
47
src/base/kernel/interfaces/ISignalListener.h
Normal file
|
@ -0,0 +1,47 @@
|
||||||
|
/* XMRig
|
||||||
|
* Copyright 2010 Jeff Garzik <jgarzik@pobox.com>
|
||||||
|
* Copyright 2012-2014 pooler <pooler@litecoinpool.org>
|
||||||
|
* Copyright 2014 Lucas Jones <https://github.com/lucasjones>
|
||||||
|
* Copyright 2014-2016 Wolf9466 <https://github.com/OhGodAPet>
|
||||||
|
* Copyright 2016 Jay D Dee <jayddee246@gmail.com>
|
||||||
|
* Copyright 2017-2018 XMR-Stak <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
|
||||||
|
* Copyright 2018-2019 SChernykh <https://github.com/SChernykh>
|
||||||
|
* Copyright 2016-2019 XMRig <https://github.com/xmrig>, <support@xmrig.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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef XMRIG_ISIGNALLISTENER_H
|
||||||
|
#define XMRIG_ISIGNALLISTENER_H
|
||||||
|
|
||||||
|
|
||||||
|
namespace xmrig {
|
||||||
|
|
||||||
|
|
||||||
|
class String;
|
||||||
|
|
||||||
|
|
||||||
|
class ISignalListener
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
virtual ~ISignalListener() = default;
|
||||||
|
|
||||||
|
virtual void onSignal(int signum) = 0;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
} /* namespace xmrig */
|
||||||
|
|
||||||
|
|
||||||
|
#endif // XMRIG_ISIGNALLISTENER_H
|
Loading…
Add table
Add a link
Reference in a new issue