mirror of https://github.com/axmolengine/axmol.git
141 lines
4.4 KiB
C++
141 lines
4.4 KiB
C++
#include "network/WebSocket-wasm.h"
|
|
#include "yasio/errc.hpp"
|
|
#include "base/Console.h"
|
|
|
|
NS_AX_BEGIN
|
|
|
|
namespace network
|
|
{
|
|
|
|
EM_BOOL WebSocket::em_ws_onopen(int eventType, const EmscriptenWebSocketOpenEvent* websocketEvent, void* userData)
|
|
{
|
|
auto ws = static_cast<WebSocket*>(userData);
|
|
if (!ws || !ws->_delegate)
|
|
return EM_TRUE;
|
|
ws->_state = WebSocket::State::OPEN;
|
|
ws->_delegate->onOpen(ws);
|
|
return EM_TRUE;
|
|
}
|
|
|
|
EM_BOOL WebSocket::em_ws_onerror(int eventType, const EmscriptenWebSocketErrorEvent* websocketEvent, void* userData)
|
|
{
|
|
auto ws = static_cast<WebSocket*>(userData);
|
|
if (!ws || !ws->_delegate)
|
|
return EM_TRUE;
|
|
ws->_state = WebSocket::State::CLOSED;
|
|
ws->_delegate->onError(ws, ErrorCode::CONNECTION_FAILURE);
|
|
return EM_TRUE;
|
|
}
|
|
|
|
EM_BOOL WebSocket::em_ws_onclose(int eventType, const EmscriptenWebSocketCloseEvent* websocketEvent, void* userData)
|
|
{
|
|
auto ws = static_cast<WebSocket*>(userData);
|
|
if (!ws || !ws->_delegate)
|
|
return EM_TRUE;
|
|
ws->_state = WebSocket::State::CLOSED;
|
|
ws->_delegate->onClose(ws);
|
|
return EM_TRUE;
|
|
}
|
|
|
|
EM_BOOL WebSocket::em_ws_onmessage(int eventType, const EmscriptenWebSocketMessageEvent* websocketEvent, void* userData)
|
|
{
|
|
auto ws = static_cast<WebSocket*>(userData);
|
|
if (!ws || !ws->_delegate)
|
|
return EM_TRUE;
|
|
WebSocket::Data dataView;
|
|
dataView.bytes = reinterpret_cast<const char*>(websocketEvent->data);
|
|
dataView.isBinary = !websocketEvent->isText;
|
|
dataView.len = websocketEvent->numBytes;
|
|
ws->_delegate->onMessage(ws, dataView);
|
|
return EM_TRUE;
|
|
}
|
|
|
|
WebSocket::WebSocket() {}
|
|
|
|
WebSocket::~WebSocket() {}
|
|
|
|
bool WebSocket::open(Delegate* delegate, std::string_view url, std::string_view caFilePath, const char* protocols)
|
|
{
|
|
if (url.empty())
|
|
{
|
|
AXLOG("ws open fail, url is empty!");
|
|
return false;
|
|
}
|
|
|
|
_delegate = delegate;
|
|
|
|
EmscriptenWebSocketCreateAttributes ws_attrs = {url.data(), protocols, EM_TRUE};
|
|
|
|
AXLOG("ws open url: %s, protocols: %s", ws_attrs.url, ws_attrs.protocols);
|
|
|
|
_state = WebSocket::State::CONNECTING;
|
|
_wsfd = emscripten_websocket_new(&ws_attrs);
|
|
|
|
// chrome/edge can't connect
|
|
// firefox works with "Sec-Fetch-Site: cross-site" in request header
|
|
// refer: https://github.com/emscripten-core/emscripten/issues/19100
|
|
// wasm websocket callback thread same with axmol render thread
|
|
emscripten_websocket_set_onopen_callback(_wsfd, this, em_ws_onopen);
|
|
emscripten_websocket_set_onerror_callback(_wsfd, this, em_ws_onerror);
|
|
emscripten_websocket_set_onclose_callback(_wsfd, this, em_ws_onclose);
|
|
emscripten_websocket_set_onmessage_callback(_wsfd, this, em_ws_onmessage);
|
|
return true;
|
|
}
|
|
/**
|
|
* @brief Sends string data to websocket server.
|
|
*
|
|
* @param message string data.
|
|
* @lua sendstring
|
|
*/
|
|
void WebSocket::send(std::string_view message)
|
|
{
|
|
auto error = emscripten_websocket_send_utf8_text(_wsfd, message.data());
|
|
if (error)
|
|
AXLOG("Failed to emscripten_websocket_send_binary(): %d", error);
|
|
}
|
|
|
|
/**
|
|
* @brief Sends binary data to websocket server.
|
|
*
|
|
* @param binaryMsg binary string data.
|
|
* @param len the size of binary string data.
|
|
* @lua sendstring
|
|
*/
|
|
void WebSocket::send(const void* data, unsigned int len)
|
|
{
|
|
auto error = emscripten_websocket_send_binary(_wsfd, const_cast<void*>(data), len);
|
|
if (error)
|
|
AXLOG("Failed to emscripten_websocket_send_binary(): %d", error);
|
|
}
|
|
|
|
/**
|
|
* @brief Closes the connection to server synchronously.
|
|
* @note It's a synchronous method, it will not return until websocket thread exits.
|
|
*/
|
|
void WebSocket::close()
|
|
{
|
|
closeAsync(); // TODO
|
|
}
|
|
|
|
/**
|
|
* @brief Closes the connection to server asynchronously.
|
|
* @note It's an asynchronous method, it just notifies websocket thread to exit and returns directly,
|
|
* If using 'closeAsync' to close websocket connection,
|
|
* be careful of not using destructed variables in the callback of 'onClose'.
|
|
*/
|
|
void WebSocket::closeAsync()
|
|
{
|
|
// close code: Uncaught DOMException: Failed to execute 'close' on 'WebSocket':
|
|
// The code must be either 1000, or between 3000 and 4999. 1024 is neither.
|
|
EMSCRIPTEN_RESULT error =
|
|
emscripten_websocket_close(_wsfd, 3000 - yasio::errc::shutdown_by_localhost, "shutdown by localhost");
|
|
if (!error)
|
|
_state = WebSocket::State::CLOSING;
|
|
else
|
|
AXLOG("Failed to emscripten_websocket_close(): %d", result);
|
|
}
|
|
|
|
} // namespace network
|
|
|
|
NS_AX_END
|