1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
#include "module_network_server.h"
#include "logger.h"
#include "imgui/imgui.h"
#include "common/utils.h"
#include <nng/nng.h>
#include <nng/protocol/pubsub0/pub.h>
#include "common/net/udp.h"
#include "core/exception.h"
namespace network
{
NetworkServerModule::NetworkServerModule(std::string input_file, std::string output_file_hint, nlohmann::json parameters)
: ProcessingModule(input_file, output_file_hint, parameters)
{
if (parameters.contains("server_mode"))
mode = parameters["server_mode"].get<std::string>();
if (parameters.count("pkt_size") > 0)
pkt_size = parameters["pkt_size"].get<int>();
else
throw satdump_exception("pkt_size parameter must be present!");
if (parameters.count("server_address") > 0)
address = parameters["server_address"].get<std::string>();
else
throw satdump_exception("server_address parameter must be present!");
if (parameters.count("server_port") > 0)
port = parameters["server_port"].get<int>();
else
throw satdump_exception("server_port parameter must be present!");
buffer = new uint8_t[pkt_size];
}
std::vector<ModuleDataType> NetworkServerModule::getInputTypes()
{
return {DATA_FILE, DATA_STREAM};
}
std::vector<ModuleDataType> NetworkServerModule::getOutputTypes()
{
return {DATA_FILE};
}
NetworkServerModule::~NetworkServerModule()
{
delete[] buffer;
}
void NetworkServerModule::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);
if (mode == "default")
{
nng_socket sock;
nng_listener listener;
logger->info("Opening TCP socket on " + std::string("tcp://" + address + ":" + std::to_string(port)));
nng_pub0_open_raw(&sock);
nng_listener_create(&listener, sock, std::string("tcp://" + address + ":" + std::to_string(port)).c_str());
nng_listener_start(listener, NNG_FLAG_NONBLOCK);
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, pkt_size);
else
input_fifo->read((uint8_t *)buffer, pkt_size);
nng_send(sock, buffer, pkt_size, NNG_FLAG_NONBLOCK);
if (input_data_type == DATA_FILE)
progress = data_in.tellg();
}
nng_listener_close(listener);
}
else if (mode == "udp_send")
{
net::UDPClient udp_send((char *)address.c_str(), port);
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, pkt_size);
else
input_fifo->read((uint8_t *)buffer, pkt_size);
udp_send.send(buffer, pkt_size);
if (input_data_type == DATA_FILE)
progress = data_in.tellg();
}
}
if (input_data_type == DATA_FILE)
data_in.close();
}
void NetworkServerModule::drawUI(bool window)
{
ImGui::Begin("Network Server", NULL, window ? 0 : NOWINDOW_FLAGS);
ImGui::Text("Address : ");
ImGui::SameLine();
ImGui::TextColored(style::theme.green, "%s", address.c_str());
ImGui::Text("Port : ");
ImGui::SameLine();
ImGui::TextColored(style::theme.green, UITO_C_STR(port));
if (!streamingInput)
ImGui::ProgressBar((double)progress / (double)filesize, ImVec2(ImGui::GetContentRegionAvail().x, 20 * ui_scale));
ImGui::End();
}
std::string NetworkServerModule::getID()
{
return "network_server";
}
std::vector<std::string> NetworkServerModule::getParameters()
{
return {"server_address", "server_port", "pkt_size"};
}
std::shared_ptr<ProcessingModule> NetworkServerModule::getInstance(std::string input_file, std::string output_file_hint, nlohmann::json parameters)
{
return std::make_shared<NetworkServerModule>(input_file, output_file_hint, parameters);
}
}
|
