diff options
Diffstat (limited to 'src/bun.js/bindings/ScriptExecutionContext.cpp')
| -rw-r--r-- | src/bun.js/bindings/ScriptExecutionContext.cpp | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/src/bun.js/bindings/ScriptExecutionContext.cpp b/src/bun.js/bindings/ScriptExecutionContext.cpp new file mode 100644 index 000000000..73b8b4c01 --- /dev/null +++ b/src/bun.js/bindings/ScriptExecutionContext.cpp @@ -0,0 +1,81 @@ +#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()); +} + +}
\ No newline at end of file |