Unverified Commit 92137b06 authored by kladko's avatar kladko

bug/SKALE-3751-enable-zeromq

parent 89a789c1
...@@ -88,7 +88,7 @@ void systemHealthCheck() { ...@@ -88,7 +88,7 @@ void systemHealthCheck() {
} }
} }
static ZMQServer *zmqServer = nullptr;
void initUserSpace() { void initUserSpace() {
...@@ -107,12 +107,6 @@ void initUserSpace() { ...@@ -107,12 +107,6 @@ void initUserSpace() {
} }
void exitZMQServer() {
spdlog::info("Exiting zmq server ...");
zmqServer->exitWorkers();
spdlog::info("Exited zmq server ...");
zmqServer = nullptr;
}
uint64_t initEnclave() { uint64_t initEnclave() {
...@@ -210,14 +204,14 @@ void initAll(uint32_t _logLevel, bool _checkCert, bool _autoSign, bool _generate ...@@ -210,14 +204,14 @@ void initAll(uint32_t _logLevel, bool _checkCert, bool _autoSign, bool _generate
SGXWalletServer::initHttpsServer(_checkCert); SGXWalletServer::initHttpsServer(_checkCert);
SGXRegistrationServer::initRegistrationServer(_autoSign); SGXRegistrationServer::initRegistrationServer(_autoSign);
CSRManagerServer::initCSRManagerServer(); CSRManagerServer::initCSRManagerServer();
zmqServer = new ZMQServer(); ZMQServer::zmqServer = new ZMQServer();
static std::thread serverThread(std::bind(&ZMQServer::run, zmqServer)); static std::thread serverThread(std::bind(&ZMQServer::run, ZMQServer::zmqServer));
serverThread.detach(); serverThread.detach();
} else { } else {
SGXWalletServer::initHttpServer(); SGXWalletServer::initHttpServer();
zmqServer = new ZMQServer(); ZMQServer::zmqServer = new ZMQServer();
static std::thread serverThread(std::bind(&ZMQServer::run, zmqServer)); static std::thread serverThread(std::bind(&ZMQServer::run, ZMQServer::zmqServer));
serverThread.detach(); serverThread.detach();
} }
SGXInfoServer::initInfoServer(_logLevel, _checkCert, _autoSign, _generateTestKeys); SGXInfoServer::initInfoServer(_logLevel, _checkCert, _autoSign, _generateTestKeys);
......
...@@ -30,6 +30,8 @@ ...@@ -30,6 +30,8 @@
using namespace std; using namespace std;
ZMQServer *ZMQServer::zmqServer = nullptr;
ZMQServer::ZMQServer() ZMQServer::ZMQServer()
: isExitRequested(false), ctx_(make_shared<zmq::context_t>(1)), : isExitRequested(false), ctx_(make_shared<zmq::context_t>(1)),
frontend_(*ctx_, ZMQ_ROUTER), frontend_(*ctx_, ZMQ_ROUTER),
...@@ -141,3 +143,9 @@ void ZMQServer::exitWorkers() { ...@@ -141,3 +143,9 @@ void ZMQServer::exitWorkers() {
} }
void ZMQServer::exitZMQServer() {
spdlog::info("Exiting zmq server ...");
zmqServer->exitWorkers();
spdlog::info("Exited zmq server ...");
zmqServer = nullptr;
}
...@@ -44,6 +44,8 @@ using namespace std; ...@@ -44,6 +44,8 @@ using namespace std;
class ZMQServer { class ZMQServer {
public: public:
static ZMQServer *zmqServer;
ZMQServer(); ZMQServer();
enum { enum {
...@@ -54,6 +56,8 @@ public: ...@@ -54,6 +56,8 @@ public:
void exitWorkers(); void exitWorkers();
static void exitZMQServer();
private: private:
shared_ptr<zmq::context_t> ctx_; shared_ptr<zmq::context_t> ctx_;
zmq::socket_t frontend_; zmq::socket_t frontend_;
......
...@@ -58,6 +58,7 @@ ...@@ -58,6 +58,7 @@
#include "SGXRegistrationServer.h" #include "SGXRegistrationServer.h"
#include "SGXWalletServer.h" #include "SGXWalletServer.h"
#include "ZMQClient.h" #include "ZMQClient.h"
#include "ZMQServer.h"
#include "sgxwallet.h" #include "sgxwallet.h"
#include "TestUtils.h" #include "TestUtils.h"
#include "testw.h" #include "testw.h"
...@@ -77,7 +78,7 @@ public: ...@@ -77,7 +78,7 @@ public:
} }
~TestFixture() { ~TestFixture() {
exitZMQServer(); ZMQServer::exitZMQServer();
TestUtils::destroyEnclave(); TestUtils::destroyEnclave();
} }
}; };
...@@ -91,7 +92,7 @@ public: ...@@ -91,7 +92,7 @@ public:
} }
~TestFixtureHTTPS() { ~TestFixtureHTTPS() {
exitZMQServer(); ZMQServer::exitZMQServer();
TestUtils::destroyEnclave(); TestUtils::destroyEnclave();
} }
}; };
...@@ -104,7 +105,7 @@ public: ...@@ -104,7 +105,7 @@ public:
} }
~TestFixtureNoResetFromBackup() { ~TestFixtureNoResetFromBackup() {
exitZMQServer(); ZMQServer::exitZMQServer();
TestUtils::destroyEnclave(); TestUtils::destroyEnclave();
} }
}; };
...@@ -118,7 +119,7 @@ public: ...@@ -118,7 +119,7 @@ public:
} }
~TestFixtureNoReset() { ~TestFixtureNoReset() {
exitZMQServer(); ZMQServer::exitZMQServer();
TestUtils::destroyEnclave(); TestUtils::destroyEnclave();
} }
}; };
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment