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
|
#include "sol_manager.hpp"
#include "main.hpp"
#include "sol_context.hpp"
#include <sys/socket.h>
#include <sys/un.h>
#include <boost/asio/basic_stream_socket.hpp>
#include <boost/asio/io_context.hpp>
#include <boost/asio/local/stream_protocol.hpp>
#include <boost/asio/write.hpp>
#include <chrono>
#include <cmath>
#include <phosphor-logging/log.hpp>
namespace sol
{
using namespace phosphor::logging;
void Manager::initConsoleSocket()
{
// explicit length constructor for NUL-prefixed abstract path
std::string path(CONSOLE_SOCKET_PATH, CONSOLE_SOCKET_PATH_LEN);
boost::asio::local::stream_protocol::endpoint ep(path);
consoleSocket =
std::make_unique<boost::asio::local::stream_protocol::socket>(*io);
consoleSocket->connect(ep);
}
void Manager::consoleInputHandler()
{
boost::system::error_code ec;
boost::asio::socket_base::bytes_readable cmd(true);
consoleSocket->io_control(cmd, ec);
size_t readSize;
if (!ec)
{
readSize = cmd.get();
}
else
{
log<level::ERR>("Reading ready count from host console socket failed:",
entry("EXCEPTION=%s", ec.message().c_str()));
return;
}
std::vector<uint8_t> buffer(readSize);
ec.clear();
size_t readDataLen =
consoleSocket->read_some(boost::asio::buffer(buffer), ec);
if (ec)
{
log<level::ERR>("Reading from host console socket failed:",
entry("EXCEPTION=%s", ec.message().c_str()));
return;
}
// Update the Console buffer with data read from the socket
buffer.resize(readDataLen);
dataBuffer.write(buffer);
}
int Manager::writeConsoleSocket(const std::vector<uint8_t>& input) const
{
boost::system::error_code ec;
boost::asio::write(*consoleSocket, boost::asio::buffer(input), ec);
return ec.value();
}
void Manager::startHostConsole()
{
if (!consoleSocket)
{
initConsoleSocket();
}
consoleSocket->async_wait(boost::asio::socket_base::wait_read,
[this](const boost::system::error_code& ec) {
if (!ec)
{
consoleInputHandler();
startHostConsole();
}
});
}
void Manager::stopHostConsole()
{
if (consoleSocket)
{
consoleSocket->cancel();
consoleSocket.reset();
}
}
void Manager::startPayloadInstance(uint8_t payloadInstance,
session::SessionID sessionID)
{
if (payloadMap.empty())
{
startHostConsole();
}
// Create the SOL Context data for payload instance
auto context = std::make_unique<Context>(io, retryCount, sendThreshold,
payloadInstance, sessionID);
payloadMap.emplace(payloadInstance, std::move(context));
}
void Manager::stopPayloadInstance(uint8_t payloadInstance)
{
auto iter = payloadMap.find(payloadInstance);
if (iter == payloadMap.end())
{
throw std::runtime_error("SOL Payload instance not found ");
}
payloadMap.erase(iter);
if (payloadMap.empty())
{
stopHostConsole();
dataBuffer.erase(dataBuffer.size());
}
}
} // namespace sol
|
