mirror of
https://github.com/oven-sh/bun
synced 2026-02-14 12:51:54 +00:00
81 lines
2.8 KiB
C++
81 lines
2.8 KiB
C++
#include "root.h"
|
|
#include "headers.h"
|
|
#include "ScriptExecutionContext.h"
|
|
|
|
#include "webcore/WebSocket.h"
|
|
|
|
#include <uws/src/App.h>
|
|
|
|
extern "C" void Bun__startLoop(us_loop_t* loop);
|
|
|
|
namespace WebCore {
|
|
|
|
template<bool SSL, bool isServer>
|
|
static void registerHTTPContextForWebSocket(ScriptExecutionContext* script, us_socket_context_t* ctx, us_loop_t* loop)
|
|
{
|
|
if constexpr (!isServer) {
|
|
if constexpr (SSL) {
|
|
Bun__WebSocketHTTPSClient__register(script->jsGlobalObject(), loop, ctx);
|
|
} else {
|
|
Bun__WebSocketHTTPClient__register(script->jsGlobalObject(), loop, ctx);
|
|
}
|
|
} else {
|
|
RELEASE_ASSERT_NOT_REACHED();
|
|
}
|
|
}
|
|
|
|
us_socket_context_t* ScriptExecutionContext::webSocketContextSSL()
|
|
{
|
|
if (!m_ssl_client_websockets_ctx) {
|
|
us_loop_t* loop = (us_loop_t*)uWS::Loop::get();
|
|
us_socket_context_options_t opts;
|
|
memset(&opts, 0, sizeof(us_socket_context_options_t));
|
|
this->m_ssl_client_websockets_ctx = us_create_socket_context(1, loop, sizeof(size_t), opts);
|
|
void** ptr = reinterpret_cast<void**>(us_socket_context_ext(1, m_ssl_client_websockets_ctx));
|
|
*ptr = this;
|
|
registerHTTPContextForWebSocket<true, false>(this, m_ssl_client_websockets_ctx, loop);
|
|
}
|
|
|
|
return m_ssl_client_websockets_ctx;
|
|
}
|
|
|
|
us_socket_context_t* ScriptExecutionContext::webSocketContextNoSSL()
|
|
{
|
|
if (!m_client_websockets_ctx) {
|
|
us_loop_t* loop = (us_loop_t*)uWS::Loop::get();
|
|
us_socket_context_options_t opts;
|
|
memset(&opts, 0, sizeof(us_socket_context_options_t));
|
|
this->m_client_websockets_ctx = us_create_socket_context(0, loop, sizeof(size_t), opts);
|
|
void** ptr = reinterpret_cast<void**>(us_socket_context_ext(0, m_client_websockets_ctx));
|
|
*ptr = this;
|
|
registerHTTPContextForWebSocket<false, false>(this, m_client_websockets_ctx, loop);
|
|
}
|
|
|
|
return m_client_websockets_ctx;
|
|
}
|
|
|
|
template<bool SSL>
|
|
static us_socket_context_t* registerWebSocketClientContext(ScriptExecutionContext* script, us_socket_context_t* parent)
|
|
{
|
|
us_loop_t* loop = (us_loop_t*)uWS::Loop::get();
|
|
if constexpr (SSL) {
|
|
us_socket_context_t* child = us_create_child_socket_context(1, parent, sizeof(size_t));
|
|
Bun__WebSocketClientTLS__register(script->jsGlobalObject(), loop, child);
|
|
return child;
|
|
} else {
|
|
us_socket_context_t* child = us_create_child_socket_context(0, parent, sizeof(size_t));
|
|
Bun__WebSocketClient__register(script->jsGlobalObject(), loop, child);
|
|
return child;
|
|
}
|
|
}
|
|
|
|
us_socket_context_t* ScriptExecutionContext::connectedWebSocketKindClient()
|
|
{
|
|
return registerWebSocketClientContext<false>(this, webSocketContextNoSSL());
|
|
}
|
|
us_socket_context_t* ScriptExecutionContext::connectedWebSocketKindClientSSL()
|
|
{
|
|
return registerWebSocketClientContext<true>(this, webSocketContextSSL());
|
|
}
|
|
|
|
} |