mirror of
https://github.com/SatDump/SatDump
synced 2026-08-13 17:47:30 -04:00
345 lines
No EOL
16 KiB
C++
345 lines
No EOL
16 KiB
C++
#include "module_stdc_parser.h"
|
|
#include <fstream>
|
|
#include "logger.h"
|
|
#include <filesystem>
|
|
#include "imgui/imgui.h"
|
|
#include "common/utils.h"
|
|
#include "pkt_parser.h"
|
|
#include "msg_parser.h"
|
|
#include "egc_parser.h"
|
|
|
|
namespace inmarsat
|
|
{
|
|
namespace stdc
|
|
{
|
|
STDCParserModule::STDCParserModule(std::string input_file, std::string output_file_hint, nlohmann::json parameters) : ProcessingModule(input_file, output_file_hint, parameters)
|
|
{
|
|
buffer = new uint8_t[FRAME_SIZE_BYTES];
|
|
}
|
|
|
|
std::vector<ModuleDataType> STDCParserModule::getInputTypes()
|
|
{
|
|
return {DATA_FILE, DATA_STREAM};
|
|
}
|
|
|
|
std::vector<ModuleDataType> STDCParserModule::getOutputTypes()
|
|
{
|
|
return {DATA_FILE};
|
|
}
|
|
|
|
STDCParserModule::~STDCParserModule()
|
|
{
|
|
delete[] buffer;
|
|
}
|
|
|
|
std::string timestampToTod(time_t time_v)
|
|
{
|
|
std::tm *timeReadable = gmtime(&time_v);
|
|
return (timeReadable->tm_hour > 9 ? std::to_string(timeReadable->tm_hour) : "0" + std::to_string(timeReadable->tm_hour)) + ":" + // Hour HH
|
|
(timeReadable->tm_min > 9 ? std::to_string(timeReadable->tm_min) : "0" + std::to_string(timeReadable->tm_min)) + ":" + // Minutes mm
|
|
(timeReadable->tm_sec > 9 ? std::to_string(timeReadable->tm_sec) : "0" + std::to_string(timeReadable->tm_sec)); // Seconds ss
|
|
}
|
|
|
|
void STDCParserModule::write_pkt_file_out(nlohmann::json &msg)
|
|
{
|
|
std::string directory = d_output_file_hint.substr(0, d_output_file_hint.rfind('/')) + "/" + msg["pkt_type"].get<std::string>();
|
|
|
|
if (!std::filesystem::exists(directory))
|
|
std::filesystem::create_directory(directory);
|
|
|
|
double time_d = msg["timestamp"].get<double>();
|
|
time_t time_v = time_d;
|
|
std::tm *timeReadable = gmtime(&time_v);
|
|
|
|
std::string utc_filename = std::to_string(timeReadable->tm_year + 1900) + // Year yyyy
|
|
(timeReadable->tm_mon + 1 > 9 ? std::to_string(timeReadable->tm_mon + 1) : "0" + std::to_string(timeReadable->tm_mon + 1)) + // Month MM
|
|
(timeReadable->tm_mday > 9 ? std::to_string(timeReadable->tm_mday) : "0" + std::to_string(timeReadable->tm_mday)) + "T" + // Day dd
|
|
(timeReadable->tm_hour > 9 ? std::to_string(timeReadable->tm_hour) : "0" + std::to_string(timeReadable->tm_hour)) + // Hour HH
|
|
(timeReadable->tm_min > 9 ? std::to_string(timeReadable->tm_min) : "0" + std::to_string(timeReadable->tm_min)) + // Minutes mm
|
|
(timeReadable->tm_sec > 9 ? std::to_string(timeReadable->tm_sec) : "0" + std::to_string(timeReadable->tm_sec)) + "Z"; // Seconds ss
|
|
|
|
std::ofstream outf(directory + "/" + utc_filename + ".json", std::ios::binary);
|
|
outf << msg.dump(4);
|
|
outf.close();
|
|
}
|
|
|
|
void STDCParserModule::process()
|
|
{
|
|
if (input_data_type == DATA_FILE)
|
|
filesize = getFilesize(d_input_file);
|
|
else
|
|
filesize = 0;
|
|
if (input_data_type == DATA_FILE)
|
|
data_in = std::ifstream(d_input_file, std::ios::binary);
|
|
|
|
logger->info("Using input frames " + d_input_file);
|
|
|
|
int msg_id = 0;
|
|
|
|
STDPacketParser pkt_parser;
|
|
MessageParser msg_parser;
|
|
EGCMessageParser egc_parser;
|
|
|
|
pkt_parser.on_packet = [&](nlohmann::json msg)
|
|
{
|
|
double time_d = msg["timestamp"].get<double>();
|
|
if (msg["pkt_id"].get<int>() == 0x7D)
|
|
{
|
|
msg_parser.push_current_time(time_d);
|
|
egc_parser.push_current_time(time_d);
|
|
}
|
|
|
|
if (msg["pkt_id"].get<int>() != 0xAA)
|
|
write_pkt_file_out(msg);
|
|
|
|
if (msg["pkt_id"].get<int>() == 0xAA)
|
|
{
|
|
std::string m = msg["message"].get<std::string>();
|
|
logger->info("Message : \n" + m);
|
|
if (m != "abcdefghijklmnopqrstuvwxyz1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ-!") // Remove those periodic ones
|
|
msg_parser.push_message(msg);
|
|
}
|
|
else if (msg["pkt_id"].get<int>() == 0xb1)
|
|
{
|
|
logger->info("EGC Double Header 1 : \n" + msg["message"].get<std::string>());
|
|
egc_parser.push_message(msg);
|
|
}
|
|
else if (msg["pkt_id"].get<int>() == 0xb2)
|
|
{
|
|
logger->info("EGC Double Header 2 : \n" + msg["message"].get<std::string>());
|
|
egc_parser.push_message(msg);
|
|
}
|
|
else
|
|
logger->info("Packet : " + msg["pkt_type"].get<std::string>());
|
|
|
|
pkt_history_mtx.lock();
|
|
pkt_history.push_back(msg);
|
|
if (pkt_history.size() > 1000)
|
|
pkt_history.erase(pkt_history.begin());
|
|
pkt_history_mtx.unlock();
|
|
|
|
msg_id++;
|
|
};
|
|
|
|
msg_parser.on_message = [&](nlohmann::json msg)
|
|
{
|
|
write_pkt_file_out(msg);
|
|
|
|
logger->info("Full Message : \n" + msg["message"].get<std::string>());
|
|
|
|
pkt_history_mtx.lock();
|
|
pkt_history_msg.push_back(msg);
|
|
if (pkt_history_msg.size() > 1000)
|
|
pkt_history_msg.erase(pkt_history_msg.begin());
|
|
pkt_history_mtx.unlock();
|
|
|
|
msg_id++;
|
|
};
|
|
|
|
egc_parser.on_message = [&](nlohmann::json msg)
|
|
{
|
|
msg["pkt_type"] = "EGC Message";
|
|
|
|
write_pkt_file_out(msg);
|
|
|
|
logger->info("EGC Message : \n" + msg["message"].get<std::string>());
|
|
|
|
pkt_history_mtx.lock();
|
|
pkt_history_msg.push_back(msg);
|
|
if (pkt_history_msg.size() > 1000)
|
|
pkt_history_msg.erase(pkt_history_msg.begin());
|
|
pkt_history_mtx.unlock();
|
|
|
|
msg_id++;
|
|
};
|
|
|
|
time_t lastTime = 0;
|
|
while (input_data_type == DATA_FILE ? !data_in.eof() : input_active.load())
|
|
{
|
|
// Read a buffer
|
|
if (input_data_type == DATA_FILE)
|
|
data_in.read((char *)buffer, FRAME_SIZE_BYTES);
|
|
else
|
|
input_fifo->read((uint8_t *)buffer, FRAME_SIZE_BYTES);
|
|
|
|
last_pkt_count = buffer[2] << 8 | buffer[3];
|
|
|
|
try
|
|
{
|
|
// logger->critical("new pkt");
|
|
pkt_parser.parse_main_pkt(buffer);
|
|
}
|
|
catch (std::exception &e)
|
|
{
|
|
logger->error("Error processing STD-C frame {:s}", e.what());
|
|
}
|
|
|
|
if (input_data_type == DATA_FILE)
|
|
progress = data_in.tellg();
|
|
|
|
if (time(NULL) % 10 == 0 && lastTime != time(NULL))
|
|
{
|
|
lastTime = time(NULL);
|
|
logger->info("Progress " + std::to_string(round(((float)progress / (float)filesize) * 1000.0f) / 10.0f) + "%");
|
|
}
|
|
}
|
|
|
|
if (input_data_type == DATA_FILE)
|
|
data_in.close();
|
|
}
|
|
|
|
void STDCParserModule::drawUI(bool window)
|
|
{
|
|
ImGui::Begin("Inmarsat STD-C Parser", NULL, window ? 0 : NOWINDOW_FLAGS);
|
|
|
|
ImGui::Text("Decoded packets can be seen in a floating window.");
|
|
ImGui::Text("Credits go to Paul Maxan (microp11) for the \nreverse-engineering work that went into Scytale-C!");
|
|
ImGui::Spacing();
|
|
ImGui::TextColored(ImColor(255, 0, 0), "Note : EGC Parsing is untested!");
|
|
|
|
ImGui::Spacing();
|
|
ImGui::Text("Last packet count : ");
|
|
ImGui::SameLine();
|
|
ImGui::TextColored(ImColor(0, 255, 0), "%d", last_pkt_count);
|
|
|
|
if (input_data_type == DATA_FILE)
|
|
ImGui::ProgressBar((float)progress / (float)filesize, ImVec2(ImGui::GetWindowWidth() - 10, 20 * ui_scale));
|
|
|
|
ImGui::End();
|
|
|
|
if (input_data_type != DATA_FILE)
|
|
{
|
|
ImGui::Begin("STD-C Packets", NULL, ImGuiWindowFlags_HorizontalScrollbar);
|
|
|
|
pkt_history_mtx.lock();
|
|
ImGui::BeginTabBar("##sdtmessagestabbar");
|
|
if (ImGui::BeginTabItem("Messages"))
|
|
{
|
|
ImGui::BeginTable("##stdmessagetable", 3, ImGuiTableFlags_Borders | ImGuiTableFlags_RowBg | ImGuiTableFlags_SizingFixedFit);
|
|
ImGui::TableSetupColumn("Type", ImGuiTableColumnFlags_NoResize, 150 * ui_scale);
|
|
ImGui::TableSetupColumn("Timestamp", ImGuiTableColumnFlags_NoResize, 75 * ui_scale);
|
|
ImGui::TableSetupColumn("Contents", 0, -1);
|
|
ImGui::TableHeadersRow();
|
|
|
|
for (int i = pkt_history_msg.size() - 1; i >= 0; i--)
|
|
{
|
|
auto &msg = pkt_history_msg[i];
|
|
try
|
|
{
|
|
ImGui::TableNextRow();
|
|
ImGui::TableSetColumnIndex(0);
|
|
ImGui::TextColored(ImColor(160, 160, 255), "%s", msg["pkt_type"].get<std::string>().c_str());
|
|
ImGui::TableSetColumnIndex(1);
|
|
ImGui::TextColored(ImColor(255, 255, 0), "%s", timestampToTod(msg["timestamp"].get<double>()).c_str());
|
|
ImGui::TableSetColumnIndex(2);
|
|
ImGui::TextColored(ImColor(0, 255, 0), "%s", msg["message"].get<std::string>().c_str());
|
|
}
|
|
catch (std::exception &e)
|
|
{
|
|
}
|
|
}
|
|
ImGui::EndTable();
|
|
ImGui::EndTabItem();
|
|
}
|
|
if (ImGui::BeginTabItem("EGC Messages"))
|
|
{
|
|
ImGui::BeginTable("##stdmessagetable", 3, ImGuiTableFlags_Borders | ImGuiTableFlags_RowBg | ImGuiTableFlags_SizingFixedFit);
|
|
ImGui::TableSetupColumn("Type", ImGuiTableColumnFlags_NoResize, 150 * ui_scale);
|
|
ImGui::TableSetupColumn("Timestamp", ImGuiTableColumnFlags_NoResize, 75 * ui_scale);
|
|
ImGui::TableSetupColumn("Contents", 0, -1);
|
|
ImGui::TableHeadersRow();
|
|
|
|
for (int i = pkt_history_egc.size() - 1; i >= 0; i--)
|
|
{
|
|
auto &msg = pkt_history_egc[i];
|
|
try
|
|
{
|
|
ImGui::TableNextRow();
|
|
ImGui::TableSetColumnIndex(0);
|
|
ImGui::TextColored(ImColor(160, 160, 255), "%s", msg["pkt_type"].get<std::string>().c_str());
|
|
ImGui::TableSetColumnIndex(1);
|
|
ImGui::TextColored(ImColor(255, 255, 0), "%s", timestampToTod(msg["timestamp"].get<double>()).c_str());
|
|
ImGui::TableSetColumnIndex(2);
|
|
ImGui::TextColored(ImColor(0, 255, 0), "%s", msg["message"].get<std::string>().c_str());
|
|
}
|
|
catch (std::exception &e)
|
|
{
|
|
}
|
|
}
|
|
ImGui::EndTable();
|
|
ImGui::EndTabItem();
|
|
}
|
|
if (ImGui::BeginTabItem("Packets"))
|
|
{
|
|
ImGui::BeginTable("##stdmessagetable", 3, ImGuiTableFlags_Borders | ImGuiTableFlags_RowBg | ImGuiTableFlags_SizingFixedFit);
|
|
ImGui::TableSetupColumn("Type", ImGuiTableColumnFlags_NoResize, 150 * ui_scale);
|
|
ImGui::TableSetupColumn("Timestamp", ImGuiTableColumnFlags_NoResize, 75 * ui_scale);
|
|
ImGui::TableSetupColumn("Contents", 0, -1);
|
|
ImGui::TableHeadersRow();
|
|
|
|
for (int i = pkt_history.size() - 1; i >= 0; i--)
|
|
{
|
|
auto &msg = pkt_history[i];
|
|
try
|
|
{
|
|
if (msg["pkt_id"].get<int>() == 0xAA)
|
|
{
|
|
ImGui::TableNextRow();
|
|
ImGui::TableSetColumnIndex(0);
|
|
ImGui::TextColored(ImColor(160, 160, 255), "%s", msg["pkt_type"].get<std::string>().c_str());
|
|
ImGui::TableSetColumnIndex(1);
|
|
ImGui::TextColored(ImColor(255, 255, 0), "%s", timestampToTod(msg["timestamp"].get<double>()).c_str());
|
|
ImGui::TableSetColumnIndex(2);
|
|
ImGui::TextColored(ImColor(0, 255, 0), "%s", msg["message"].get<std::string>().c_str());
|
|
}
|
|
else if (msg["pkt_id"].get<int>() == 0xb1 || msg["pkt_id"].get<int>() == 0xb2)
|
|
{
|
|
ImGui::TableNextRow();
|
|
ImGui::TableSetColumnIndex(0);
|
|
ImGui::TextColored(ImColor(160, 160, 255), "%s", msg["pkt_type"].get<std::string>().c_str());
|
|
ImGui::TableSetColumnIndex(1);
|
|
ImGui::TextColored(ImColor(255, 255, 0), "%s", timestampToTod(msg["timestamp"].get<double>()).c_str());
|
|
ImGui::TableSetColumnIndex(2);
|
|
ImGui::TextColored(ImColor(255, 0, 0), "%s", msg["message"].get<std::string>().c_str());
|
|
}
|
|
else
|
|
{
|
|
ImGui::TableNextRow();
|
|
ImGui::TableSetColumnIndex(0);
|
|
ImGui::TextColored(ImColor(160, 160, 255), "%s", msg["pkt_type"].get<std::string>().c_str());
|
|
ImGui::TableSetColumnIndex(1);
|
|
ImGui::TextColored(ImColor(255, 255, 0), "%s", timestampToTod(msg["timestamp"].get<double>()).c_str());
|
|
ImGui::TableSetColumnIndex(2);
|
|
ImGui::TextColored(ImColor(255, 255, 255), "%s", msg.dump().c_str());
|
|
}
|
|
}
|
|
catch (std::exception &e)
|
|
{
|
|
}
|
|
}
|
|
ImGui::EndTable();
|
|
ImGui::EndTabItem();
|
|
}
|
|
ImGui::EndTabBar();
|
|
pkt_history_mtx.unlock();
|
|
|
|
ImGui::End();
|
|
}
|
|
}
|
|
|
|
std::string STDCParserModule::getID()
|
|
{
|
|
return "inmarsat_stdc_parser";
|
|
}
|
|
|
|
std::vector<std::string> STDCParserModule::getParameters()
|
|
{
|
|
return {};
|
|
}
|
|
|
|
std::shared_ptr<ProcessingModule> STDCParserModule::getInstance(std::string input_file, std::string output_file_hint, nlohmann::json parameters)
|
|
{
|
|
return std::make_shared<STDCParserModule>(input_file, output_file_hint, parameters);
|
|
}
|
|
}
|
|
} |