129 lines
4.2 KiB
C++
Executable File
129 lines
4.2 KiB
C++
Executable File
#include "main_handler.h"
|
|
#include <NGLLogger/NGLLoggerBootstrap.h>
|
|
#include <DDS/DDSConnector/dds_connector.h>
|
|
#include <QtCore/QThread>
|
|
#include <QtCore/QCoreApplication>
|
|
#include <QtCore/QDateTime>
|
|
#include <QtCore/QElapsedTimer>
|
|
#include "proxy_emulator.h"
|
|
|
|
#include <QTimer>
|
|
|
|
VulnerabilityStateGenerator::MainHandler::MainHandler(QObject *parent)
|
|
: QObject(parent)
|
|
, m_domainId(0)
|
|
, m_exerciseId(0)
|
|
, m_proxyEmulator(nullptr)
|
|
, m_genTimer(new QTimer())
|
|
, m_logger(NGL::NGLLoggerBootstrap::getNGLRootLogger().getNGLLogger("MainHandler"))
|
|
{
|
|
}
|
|
|
|
VulnerabilityStateGenerator::MainHandler::~MainHandler()
|
|
{
|
|
}
|
|
|
|
bool VulnerabilityStateGenerator::MainHandler::init(int32_t domainId, int32_t exerciseId)
|
|
{
|
|
m_domainId = domainId;
|
|
m_exerciseId = exerciseId;
|
|
|
|
delete m_proxyEmulator;
|
|
m_proxyEmulator = nullptr;
|
|
|
|
// DdsConnector
|
|
DEBUG_LOG(m_logger) << "Creating and initializing DdsConnector ...";
|
|
m_ddsConnector.reset(new DDSConnector::DdsConnector(m_domainId, IDL::PartitionType::Exercise, m_exerciseId));
|
|
if (!m_ddsConnector->init())
|
|
{
|
|
ERROR_LOG(m_logger) << "Could not init DdsConnector";
|
|
return false;
|
|
}
|
|
m_ddsConnector->initPublisherSubscriber();
|
|
DEBUG_LOG(m_logger) << "... done";
|
|
|
|
// dtc
|
|
DEBUG_LOG(m_logger) << "Creating and initializing PPM DTC ...";
|
|
m_ppmDtc = ParticipantManagement::Interface::Dtc::create(m_ddsConnector);
|
|
QThread::msleep(100);
|
|
if (!m_ppmDtc->enableAll())
|
|
{
|
|
ERROR_LOG(m_logger) << "Could not init PPM DTC";
|
|
return false;
|
|
}
|
|
DEBUG_LOG(m_logger) << "... done";
|
|
|
|
fillMapFromEnum();
|
|
|
|
DEBUG_LOG(m_logger) << "m_vsMap.size(): " << m_vsMap.size();
|
|
|
|
// ProxyEmulator
|
|
DEBUG_LOG(m_logger) << "Creating and initializing ProxyEmulator ...";
|
|
m_proxyEmulator = new ProxyEmulator(m_ddsConnector, this);
|
|
QThread::msleep(100);
|
|
if (!m_proxyEmulator->init())
|
|
{
|
|
ERROR_LOG(m_logger) << "Could not init ProxyEmulator";
|
|
return false;
|
|
}
|
|
|
|
m_partMap = m_ppmDtc->participants();
|
|
|
|
m_proxyEmulator->giveInitState(m_partMap);
|
|
|
|
connect(m_genTimer, &QTimer::timeout, this, &MainHandler::setNewState);
|
|
m_genTimer->start(1000);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
void VulnerabilityStateGenerator::MainHandler::quit()
|
|
{
|
|
ERROR_LOG(m_logger) << "Are the following procs running?\n"
|
|
<< "svcParticipantManagement -d " << m_domainId << " -e 0\n"
|
|
<< "svcParticipantManagement -d " << m_domainId << " -e " << m_exerciseId << "\n"
|
|
<< "... shutting down";
|
|
QCoreApplication::exit(-1);
|
|
}
|
|
|
|
|
|
ProjectDataManagement::VulnerabilityState VulnerabilityStateGenerator::MainHandler::randomVulnerabilityState()
|
|
{
|
|
ProjectDataManagement::VulnerabilityState result;
|
|
|
|
qsrand(QDateTime::currentMSecsSinceEpoch());
|
|
unsigned int index = 0;
|
|
if (m_vsMap.size() > 0)
|
|
index = qrand() % m_vsMap.size();
|
|
|
|
|
|
return m_vsMap.value(index);
|
|
|
|
}
|
|
|
|
void VulnerabilityStateGenerator::MainHandler::setNewState()
|
|
{
|
|
for (auto it = m_partMap.begin(); it != m_partMap.end(); ++it)
|
|
{
|
|
if (it.value().equipmentNumber().is_set() && it.value().equipmentNumber().get() > 0)
|
|
m_proxyEmulator->publishNewState(randomVulnerabilityState(), it.value());
|
|
}
|
|
}
|
|
|
|
void VulnerabilityStateGenerator::MainHandler::fillMapFromEnum()
|
|
{
|
|
m_vsMap.insert(0, ProjectDataManagement::VulnerabilityState_def::CombatReady);
|
|
m_vsMap.insert(1, ProjectDataManagement::VulnerabilityState_def::LightDamage);
|
|
m_vsMap.insert(2, ProjectDataManagement::VulnerabilityState_def::MediumDamage);
|
|
m_vsMap.insert(3, ProjectDataManagement::VulnerabilityState_def::HeavyDamage);
|
|
m_vsMap.insert(4, ProjectDataManagement::VulnerabilityState_def::MobilityKill);
|
|
m_vsMap.insert(5, ProjectDataManagement::VulnerabilityState_def::VisibleMobilityKill);
|
|
m_vsMap.insert(6, ProjectDataManagement::VulnerabilityState_def::WeaponSystemDamage);
|
|
m_vsMap.insert(7, ProjectDataManagement::VulnerabilityState_def::VisibleWeaponSystemDamage);
|
|
m_vsMap.insert(8, ProjectDataManagement::VulnerabilityState_def::CatastrophicKill);
|
|
m_vsMap.insert(9, ProjectDataManagement::VulnerabilityState_def::TemporarySeverance);
|
|
m_vsMap.insert(10, ProjectDataManagement::VulnerabilityState_def::NearMiss);
|
|
m_vsMap.insert(11, ProjectDataManagement::VulnerabilityState_def::HitWithNoEffect);
|
|
}
|