Add extracted tools: CitrineOS, OpenOCPP, ShapeShifter
- CitrineOS core extracted (CSMS OCPP 2.0.1) - OpenOCPP extracted (firmware OCPP 1.6J/2.0.1) - ShapeShifter library installed (pip install -e) - ShapeShifter specification extracted - EVerest extracted TODO updated with progress
This commit is contained in:
@@ -0,0 +1,180 @@
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
// Copyright chargebyte GmbH and Contributors to EVerest
|
||||
|
||||
#include "WebSocketTestClient.hpp"
|
||||
|
||||
#include <nlohmann/json.hpp>
|
||||
|
||||
#include "JsonRpcUtils.hpp"
|
||||
|
||||
using namespace json_rpc_utils;
|
||||
|
||||
WebSocketTestClient::WebSocketTestClient(const std::string& address, int port) :
|
||||
m_address(address), m_port(port), m_context(nullptr), m_wsi(nullptr), m_connected(false) {
|
||||
|
||||
struct lws_protocols protocols[] = {{"EVerestRpcApi", callback, 0, 0, 0, NULL, 0}, LWS_PROTOCOL_LIST_TERM};
|
||||
|
||||
struct lws_context_creation_info info;
|
||||
memset(&info, 0, sizeof(info));
|
||||
info.port = CONTEXT_PORT_NO_LISTEN; /* client */
|
||||
info.protocols = protocols;
|
||||
info.gid = -1;
|
||||
info.uid = -1;
|
||||
info.user = this;
|
||||
|
||||
m_context = lws_create_context(&info);
|
||||
if (!m_context) {
|
||||
throw std::runtime_error("Failed to create WebSocket m_context");
|
||||
}
|
||||
|
||||
m_ccinfo.context = m_context;
|
||||
m_ccinfo.address = m_address.c_str();
|
||||
m_ccinfo.port = m_port;
|
||||
m_ccinfo.path = "/";
|
||||
m_ccinfo.host = m_ccinfo.address;
|
||||
m_ccinfo.origin = m_ccinfo.address;
|
||||
m_ccinfo.protocol = "EVerestRpcApi";
|
||||
}
|
||||
|
||||
WebSocketTestClient::~WebSocketTestClient() {
|
||||
close();
|
||||
}
|
||||
|
||||
int WebSocketTestClient::callback(struct lws* wsi, enum lws_callback_reasons reason, void* user, void* in, size_t len) {
|
||||
WebSocketTestClient* client = static_cast<WebSocketTestClient*>(lws_context_user(lws_get_context(wsi)));
|
||||
|
||||
if (client == nullptr) {
|
||||
std::cerr << "Error: WebSocketTestClient instance not found!";
|
||||
return -1;
|
||||
}
|
||||
|
||||
switch (reason) {
|
||||
case LWS_CALLBACK_CLIENT_ESTABLISHED:
|
||||
client->m_connected = true;
|
||||
client->m_cv.notify_all();
|
||||
break;
|
||||
case LWS_CALLBACK_CLIENT_RECEIVE: {
|
||||
std::lock_guard<std::mutex> lock(client->m_cv_mutex);
|
||||
try {
|
||||
client->m_received_data.assign(static_cast<char*>(in), len);
|
||||
client->m_cv.notify_all();
|
||||
} catch (const std::exception& e) {
|
||||
EVLOG_error << "Exception occurred while handling data available: " << e.what();
|
||||
}
|
||||
break;
|
||||
}
|
||||
case LWS_CALLBACK_CLIENT_CLOSED:
|
||||
case LWS_CALLBACK_CLOSED_CLIENT_HTTP: {
|
||||
client->m_connected = false;
|
||||
EVLOG_info << "Client closed connection: " << (in ? static_cast<const char*>(in) : "(null)")
|
||||
<< " reason: " << reason;
|
||||
break;
|
||||
}
|
||||
case LWS_CALLBACK_CLIENT_CONNECTION_ERROR:
|
||||
EVLOG_error << "Client connection error: " << (in ? static_cast<const char*>(in) : "(null)");
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
bool WebSocketTestClient::connect() {
|
||||
if (m_context == nullptr) {
|
||||
EVLOG_error << "Error: WebSocket m_context not found!";
|
||||
return false;
|
||||
}
|
||||
stop_lws_service_thread(); // Stop any existing service thread, otherwise the connect will fail
|
||||
|
||||
m_wsi = lws_client_connect_via_info(&m_ccinfo);
|
||||
|
||||
if (m_wsi == nullptr) {
|
||||
EVLOG_error << "Error while connecting to WebSocket server";
|
||||
} else {
|
||||
EVLOG_info << "Connecting to WebSocket server...";
|
||||
start_lws_service_thread();
|
||||
}
|
||||
return m_wsi != nullptr;
|
||||
}
|
||||
|
||||
void WebSocketTestClient::start_lws_service_thread() {
|
||||
if (m_lws_service_running) {
|
||||
return;
|
||||
}
|
||||
m_lws_service_thread = std::thread([this]() {
|
||||
m_lws_service_running = true;
|
||||
while (m_lws_service_running) {
|
||||
lws_service(m_context, 0);
|
||||
}
|
||||
});
|
||||
|
||||
// Wait for the service thread to start
|
||||
while (!m_lws_service_running) {
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(10)); // Wait for service thread to start
|
||||
}
|
||||
}
|
||||
|
||||
void WebSocketTestClient::stop_lws_service_thread() {
|
||||
if (!m_lws_service_running) {
|
||||
return;
|
||||
}
|
||||
m_lws_service_running = false;
|
||||
lws_cancel_service(m_context);
|
||||
if (m_lws_service_thread.joinable()) {
|
||||
m_lws_service_thread.join();
|
||||
}
|
||||
}
|
||||
|
||||
bool WebSocketTestClient::is_connected() {
|
||||
return m_connected;
|
||||
}
|
||||
|
||||
void WebSocketTestClient::send(const std::string& message) {
|
||||
if (!m_connected)
|
||||
return;
|
||||
|
||||
try {
|
||||
std::vector<unsigned char> buf(LWS_PRE + message.size());
|
||||
memcpy(buf.data() + LWS_PRE, message.c_str(), message.size());
|
||||
lws_write(m_wsi, buf.data() + LWS_PRE, message.size(), LWS_WRITE_TEXT);
|
||||
} catch (const std::exception& e) {
|
||||
EVLOG_error << "Error while sending message: " << e.what();
|
||||
}
|
||||
}
|
||||
|
||||
const std::string& WebSocketTestClient::receive() const {
|
||||
return m_received_data;
|
||||
}
|
||||
|
||||
void WebSocketTestClient::close() {
|
||||
if (m_wsi) {
|
||||
if (m_connected == true) {
|
||||
lws_close_reason(m_wsi, LWS_CLOSE_STATUS_NORMAL, nullptr, 0);
|
||||
}
|
||||
|
||||
if (m_context == nullptr) {
|
||||
EVLOG_error << "Error: WebSocket m_context not found!";
|
||||
return;
|
||||
}
|
||||
|
||||
stop_lws_service_thread();
|
||||
m_wsi = nullptr;
|
||||
|
||||
if (m_lws_service_thread.joinable()) {
|
||||
m_lws_service_thread.join(); // Wait for client thread to finish
|
||||
}
|
||||
}
|
||||
if (m_context) {
|
||||
lws_context_destroy(m_context);
|
||||
m_context = nullptr;
|
||||
}
|
||||
m_connected = false;
|
||||
|
||||
EVLOG_info << "WebSocket client closed";
|
||||
}
|
||||
|
||||
void WebSocketTestClient::send_api_hello_req() {
|
||||
nlohmann::json apiHelloReq = create_json_rpc_request("API.Hello", {}, 1);
|
||||
|
||||
send(apiHelloReq.dump());
|
||||
}
|
||||
Reference in New Issue
Block a user