parent
8a03f16aa5
commit
4d277701cb
@ -1,266 +0,0 @@
|
|||||||
#include "webfuse/utils/ws_server.hpp"
|
|
||||||
#include "webfuse/utils/timeout_watcher.hpp"
|
|
||||||
|
|
||||||
#include "webfuse/core/util.h"
|
|
||||||
#include "webfuse/core/protocol_names.h"
|
|
||||||
#include <libwebsockets.h>
|
|
||||||
#include <cstring>
|
|
||||||
#include <vector>
|
|
||||||
#include <queue>
|
|
||||||
#include <string>
|
|
||||||
|
|
||||||
using webfuse_test::TimeoutWatcher;
|
|
||||||
|
|
||||||
#define DEFAULT_TIMEOUT (std::chrono::milliseconds(5 * 1000))
|
|
||||||
|
|
||||||
namespace
|
|
||||||
{
|
|
||||||
|
|
||||||
class IServer
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
virtual ~IServer() = default;
|
|
||||||
virtual void onConnectionEstablished(struct lws * wsi) = 0;
|
|
||||||
virtual void onConnectionClosed(struct lws * wsi) = 0;
|
|
||||||
virtual void onMessageReceived(struct lws * wsi, char const * data, size_t length) = 0;
|
|
||||||
virtual void onWritable(struct lws * wsi) = 0;
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
extern "C"
|
|
||||||
{
|
|
||||||
|
|
||||||
static int wf_test_utils_ws_server_callback(
|
|
||||||
struct lws * wsi,
|
|
||||||
enum lws_callback_reasons reason,
|
|
||||||
void * WF_UNUSED_PARAM(user),
|
|
||||||
void * in,
|
|
||||||
size_t len)
|
|
||||||
{
|
|
||||||
struct lws_protocols const * ws_protocol = lws_get_protocol(wsi);
|
|
||||||
if (NULL == ws_protocol)
|
|
||||||
{
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
auto * server = reinterpret_cast<IServer*>(ws_protocol->user);
|
|
||||||
switch(reason)
|
|
||||||
{
|
|
||||||
case LWS_CALLBACK_ESTABLISHED:
|
|
||||||
server->onConnectionEstablished(wsi);
|
|
||||||
break;
|
|
||||||
case LWS_CALLBACK_CLOSED:
|
|
||||||
server->onConnectionClosed(wsi);
|
|
||||||
break;
|
|
||||||
case LWS_CALLBACK_RECEIVE:
|
|
||||||
{
|
|
||||||
auto * data = reinterpret_cast<char const *>(in);
|
|
||||||
server->onMessageReceived(wsi, data, len);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case LWS_CALLBACK_SERVER_WRITEABLE:
|
|
||||||
server->onWritable(wsi);
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
namespace webfuse_test
|
|
||||||
{
|
|
||||||
|
|
||||||
class WebsocketServer::Private: public IServer
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
Private(int port, struct lws_protocols * additionalProtocols, size_t additionalProtocolsCount)
|
|
||||||
: client_wsi(nullptr)
|
|
||||||
{
|
|
||||||
ws_protocols = new struct lws_protocols[2 + additionalProtocolsCount];
|
|
||||||
memset(ws_protocols, 0, sizeof(struct lws_protocols) * (2 + additionalProtocolsCount));
|
|
||||||
|
|
||||||
ws_protocols[0].name = WF_PROTOCOL_NAME_ADAPTER_SERVER;
|
|
||||||
ws_protocols[0].callback = &wf_test_utils_ws_server_callback;
|
|
||||||
ws_protocols[0].per_session_data_size = 0;
|
|
||||||
ws_protocols[0].user = reinterpret_cast<void*>(this);
|
|
||||||
|
|
||||||
if (0 < additionalProtocolsCount)
|
|
||||||
{
|
|
||||||
memcpy(&ws_protocols[additionalProtocolsCount], additionalProtocols, sizeof(struct lws_protocols) * additionalProtocolsCount);
|
|
||||||
}
|
|
||||||
|
|
||||||
memset(&info, 0, sizeof(struct lws_context_creation_info));
|
|
||||||
info.port = port;
|
|
||||||
info.mounts = NULL;
|
|
||||||
info.protocols =ws_protocols;
|
|
||||||
info.vhost_name = "localhost";
|
|
||||||
info.ws_ping_pong_interval = 10;
|
|
||||||
info.options = LWS_SERVER_OPTION_HTTP_HEADERS_SECURITY_BEST_PRACTICES_ENFORCE;
|
|
||||||
|
|
||||||
context = lws_create_context(&info);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
virtual ~Private()
|
|
||||||
{
|
|
||||||
lws_context_destroy(context);
|
|
||||||
delete[] ws_protocols;
|
|
||||||
}
|
|
||||||
|
|
||||||
struct lws_context * getContext()
|
|
||||||
{
|
|
||||||
return context;
|
|
||||||
}
|
|
||||||
|
|
||||||
void waitForConnection()
|
|
||||||
{
|
|
||||||
TimeoutWatcher watcher(DEFAULT_TIMEOUT);
|
|
||||||
|
|
||||||
while (nullptr == client_wsi)
|
|
||||||
{
|
|
||||||
watcher.check();
|
|
||||||
lws_service(context, 100);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void sendMessage(json_t * message)
|
|
||||||
{
|
|
||||||
char* message_text = json_dumps(message, JSON_COMPACT);
|
|
||||||
writeQueue.push(message_text);
|
|
||||||
json_decref(message);
|
|
||||||
free(message_text);
|
|
||||||
|
|
||||||
if (nullptr != client_wsi)
|
|
||||||
{
|
|
||||||
lws_callback_on_writable(client_wsi);
|
|
||||||
|
|
||||||
TimeoutWatcher watcher(DEFAULT_TIMEOUT);
|
|
||||||
while (!writeQueue.empty())
|
|
||||||
{
|
|
||||||
watcher.check();
|
|
||||||
lws_service(context, 100);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
json_t * receiveMessage()
|
|
||||||
{
|
|
||||||
TimeoutWatcher watcher(DEFAULT_TIMEOUT);
|
|
||||||
|
|
||||||
while (recvQueue.empty())
|
|
||||||
{
|
|
||||||
watcher.check();
|
|
||||||
lws_service(context, 100);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string const & message_text = recvQueue.front();
|
|
||||||
json_t * message = json_loads(message_text.c_str(), JSON_DECODE_ANY, nullptr);
|
|
||||||
recvQueue.pop();
|
|
||||||
|
|
||||||
return message;
|
|
||||||
}
|
|
||||||
|
|
||||||
void onConnectionEstablished(struct lws * wsi) override
|
|
||||||
{
|
|
||||||
client_wsi = wsi;
|
|
||||||
}
|
|
||||||
|
|
||||||
void onConnectionClosed(struct lws * wsi) override
|
|
||||||
{
|
|
||||||
if (wsi == client_wsi)
|
|
||||||
{
|
|
||||||
client_wsi = nullptr;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void onMessageReceived(struct lws * wsi, char const * data, size_t length) override
|
|
||||||
{
|
|
||||||
if (wsi == client_wsi)
|
|
||||||
{
|
|
||||||
recvQueue.push(std::string(data, length));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void onWritable(struct lws * wsi) override
|
|
||||||
{
|
|
||||||
if (!writeQueue.empty())
|
|
||||||
{
|
|
||||||
std::string const & message = writeQueue.front();
|
|
||||||
|
|
||||||
unsigned char * data = new unsigned char[LWS_PRE + message.size()];
|
|
||||||
memcpy(&data[LWS_PRE], message.c_str(), message.size());
|
|
||||||
lws_write(wsi, &data[LWS_PRE], message.size(), LWS_WRITE_TEXT);
|
|
||||||
delete[] data;
|
|
||||||
|
|
||||||
writeQueue.pop();
|
|
||||||
if (!writeQueue.empty())
|
|
||||||
{
|
|
||||||
lws_callback_on_writable(wsi);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
private:
|
|
||||||
void send(std::string const & message)
|
|
||||||
{
|
|
||||||
if (nullptr != client_wsi)
|
|
||||||
{
|
|
||||||
writeQueue.push(message);
|
|
||||||
lws_callback_on_writable(client_wsi);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct lws * client_wsi;
|
|
||||||
|
|
||||||
struct lws_protocols * ws_protocols;
|
|
||||||
struct lws_context_creation_info info;
|
|
||||||
struct lws_context * context;
|
|
||||||
std::queue<std::string> writeQueue;
|
|
||||||
std::queue<std::string> recvQueue;
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
WebsocketServer::WebsocketServer(int port)
|
|
||||||
: d(new Private(port, nullptr, 0))
|
|
||||||
{
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
WebsocketServer::WebsocketServer(int port, struct lws_protocols * additionalProtocols, std::size_t additionalProtocolsCount)
|
|
||||||
: d(new Private(port, additionalProtocols, additionalProtocolsCount))
|
|
||||||
{
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
WebsocketServer::~WebsocketServer()
|
|
||||||
{
|
|
||||||
delete d;
|
|
||||||
}
|
|
||||||
|
|
||||||
struct lws_context * WebsocketServer::getContext()
|
|
||||||
{
|
|
||||||
return d->getContext();
|
|
||||||
}
|
|
||||||
|
|
||||||
void WebsocketServer::waitForConnection()
|
|
||||||
{
|
|
||||||
d->waitForConnection();
|
|
||||||
}
|
|
||||||
|
|
||||||
void WebsocketServer::sendMessage(json_t * message)
|
|
||||||
{
|
|
||||||
d->sendMessage(message);
|
|
||||||
}
|
|
||||||
|
|
||||||
json_t * WebsocketServer::receiveMessage()
|
|
||||||
{
|
|
||||||
return d->receiveMessage();
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
@ -1,29 +0,0 @@
|
|||||||
#ifndef WF_TEST_UTILS_WS_SERVER_HPP
|
|
||||||
#define WF_TEST_UTILS_WS_SERVER_HPP
|
|
||||||
|
|
||||||
#include <libwebsockets.h>
|
|
||||||
#include <jansson.h>
|
|
||||||
|
|
||||||
namespace webfuse_test
|
|
||||||
{
|
|
||||||
|
|
||||||
class WebsocketServer
|
|
||||||
{
|
|
||||||
WebsocketServer(WebsocketServer const &) = delete;
|
|
||||||
WebsocketServer & operator=(WebsocketServer const &) = delete;
|
|
||||||
public:
|
|
||||||
explicit WebsocketServer(int port);
|
|
||||||
WebsocketServer(int port, struct lws_protocols * additionalProtocols, std::size_t additionalProtocolsCount);
|
|
||||||
~WebsocketServer();
|
|
||||||
struct lws_context * getContext();
|
|
||||||
void waitForConnection();
|
|
||||||
void sendMessage(json_t * message);
|
|
||||||
json_t * receiveMessage();
|
|
||||||
private:
|
|
||||||
class Private;
|
|
||||||
Private * d;
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
Loading…
Reference in new issue