node-datachannel 0.28.0 → 0.30.0

package/src/cpp/data-channel-wrapper.h

@@ -1,9 +1,7 @@
 #ifndef DATA_CHANNEL_WRAPPER_H
 #define DATA_CHANNEL_WRAPPER_H
 
-#include <iostream>
 #include <string>
-#include <variant>
 #include <memory>
 #include <unordered_set>
 

package/src/cpp/ice-udp-mux-listener-wrapper.cpp

@@ -0,0 +1,157 @@
+#include "ice-udp-mux-listener-wrapper.h"
+
+#include "plog/Log.h"
+
+#include <cctype>
+
+Napi::FunctionReference IceUdpMuxListenerWrapper::constructor = Napi::FunctionReference();
+std::unordered_set<IceUdpMuxListenerWrapper *> IceUdpMuxListenerWrapper::instances;
+
+void IceUdpMuxListenerWrapper::StopAll()
+{
+  PLOG_DEBUG << "IceUdpMuxListenerWrapper StopAll() called";
+  auto copy(instances);
+  for (auto inst : copy)
+    inst->doCleanup();
+}
+
+Napi::Object IceUdpMuxListenerWrapper::Init(Napi::Env env, Napi::Object exports)
+{
+  Napi::HandleScope scope(env);
+
+  Napi::Function func =
+      DefineClass(env, "IceUdpMuxListener",
+                  {InstanceMethod("stop", &IceUdpMuxListenerWrapper::stop),
+                   InstanceMethod("onUnhandledStunRequest", &IceUdpMuxListenerWrapper::onUnhandledStunRequest),
+                   InstanceMethod("port", &IceUdpMuxListenerWrapper::port),
+                   InstanceMethod("address", &IceUdpMuxListenerWrapper::address)});
+
+  // If this is not the first call, we don't want to reassign the constructor (hot-reload problem)
+  if (constructor.IsEmpty())
+  {
+    constructor = Napi::Persistent(func);
+    constructor.SuppressDestruct();
+  }
+
+  exports.Set("IceUdpMuxListener", func);
+  return exports;
+}
+
+IceUdpMuxListenerWrapper::IceUdpMuxListenerWrapper(const Napi::CallbackInfo &info)
+    : Napi::ObjectWrap<IceUdpMuxListenerWrapper>(info)
+{
+  PLOG_DEBUG << "IceUdpMuxListenerWrapper Constructor called";
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  // We expect (Number, String?) as param
+  if (length > 0 && info[0].IsNumber())
+  {
+    // Port
+    mPort = info[0].As<Napi::Number>().ToNumber().Uint32Value();
+  }
+  else
+  {
+    Napi::TypeError::New(env, "Port (Number) and optional Address (String) expected").ThrowAsJavaScriptException();
+    return;
+  }
+
+  if (length > 1 && info[1].IsString())
+  {
+    // Address
+    mAddress = info[1].As<Napi::String>().ToString();
+  }
+
+  iceUdpMuxListenerPtr = std::make_unique<rtc::IceUdpMuxListener>(mPort, mAddress);
+  instances.insert(this);
+}
+
+IceUdpMuxListenerWrapper::~IceUdpMuxListenerWrapper()
+{
+  PLOG_DEBUG << "IceUdpMuxListenerWrapper Destructor called";
+  doCleanup();
+}
+
+void IceUdpMuxListenerWrapper::doCleanup()
+{
+  PLOG_DEBUG << "IceUdpMuxListenerWrapper::doCleanup() called";
+
+  if (iceUdpMuxListenerPtr)
+  {
+    iceUdpMuxListenerPtr->stop();
+    iceUdpMuxListenerPtr.reset();
+  }
+
+  mOnUnhandledStunRequestCallback.reset();
+  instances.erase(this);
+}
+
+Napi::Value IceUdpMuxListenerWrapper::port(const Napi::CallbackInfo &info)
+{
+  Napi::Env env = info.Env();
+
+  return Napi::Number::New(env, mPort);
+}
+
+Napi::Value IceUdpMuxListenerWrapper::address(const Napi::CallbackInfo &info)
+{
+  Napi::Env env = info.Env();
+
+  if (!mAddress.has_value())
+  {
+    return env.Undefined();
+  }
+
+  return Napi::String::New(env, mAddress.value());
+}
+
+void IceUdpMuxListenerWrapper::stop(const Napi::CallbackInfo &info)
+{
+  PLOG_DEBUG << "IceUdpMuxListenerWrapper::stop() called";
+  doCleanup();
+}
+
+void IceUdpMuxListenerWrapper::onUnhandledStunRequest(const Napi::CallbackInfo &info)
+{
+  PLOG_DEBUG << "IceUdpMuxListenerWrapper::onUnhandledStunRequest() called";
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  if (!iceUdpMuxListenerPtr)
+  {
+    Napi::Error::New(env, "IceUdpMuxListenerWrapper::onUnhandledStunRequest() called on destroyed IceUdpMuxListener")
+        .ThrowAsJavaScriptException();
+    return;
+  }
+
+  if (length < 1 || !info[0].IsFunction())
+  {
+    Napi::TypeError::New(env, "Function expected").ThrowAsJavaScriptException();
+    return;
+  }
+
+  // Callback
+  mOnUnhandledStunRequestCallback = std::make_unique<ThreadSafeCallback>(info[0].As<Napi::Function>());
+
+  iceUdpMuxListenerPtr->OnUnhandledStunRequest(
+      [&](rtc::IceUdpMuxRequest request)
+      {
+    PLOG_DEBUG << "IceUdpMuxListenerWrapper::onUnhandledStunRequest() IceUdpMuxCallback call(1)";
+
+    if (mOnUnhandledStunRequestCallback)
+    {
+      mOnUnhandledStunRequestCallback->call(
+          [request = std::move(request)](Napi::Env env, std::vector<napi_value> &args)
+          {
+        Napi::Object reqObj = Napi::Object::New(env);
+        reqObj.Set("ufrag", request.remoteUfrag.c_str());
+        reqObj.Set("host", request.remoteAddress.c_str());
+        reqObj.Set("port", request.remotePort);
+
+        args = {reqObj};
+      });
+    }
+
+    PLOG_DEBUG << "IceUdpMuxListenerWrapper::onUnhandledStunRequest() IceUdpMuxCallback call(2)";
+  });
+}

package/src/cpp/ice-udp-mux-listener-wrapper.h

@@ -0,0 +1,43 @@
+#ifndef ICE_UDP_MUX_LISTENER_WRAPPER_H
+#define ICE_UDP_MUX_LISTENER_WRAPPER_H
+
+#include <napi.h>
+#include <rtc/rtc.hpp>
+#include <unordered_set>
+
+#include "thread-safe-callback.h"
+
+class IceUdpMuxListenerWrapper : public Napi::ObjectWrap<IceUdpMuxListenerWrapper>
+{
+public:
+  static Napi::Object Init(Napi::Env env, Napi::Object exports);
+  IceUdpMuxListenerWrapper(const Napi::CallbackInfo &info);
+  ~IceUdpMuxListenerWrapper();
+
+  // Functions
+  void stop(const Napi::CallbackInfo &info);
+  void onUnhandledStunRequest(const Napi::CallbackInfo &info);
+
+  // Stop listening on all ports
+  static void StopAll();
+
+  // Properties
+  Napi::Value port(const Napi::CallbackInfo &info);
+  Napi::Value address(const Napi::CallbackInfo &info);
+  Napi::Value unhandledStunRequestCallback(const Napi::CallbackInfo &info);
+
+  // Callback Ptrs
+  std::unique_ptr<ThreadSafeCallback> mOnUnhandledStunRequestCallback = nullptr;
+
+private:
+  static Napi::FunctionReference constructor;
+  static std::unordered_set<IceUdpMuxListenerWrapper *> instances;
+
+  void doCleanup();
+
+  std::optional<std::string> mAddress;
+  uint16_t mPort;
+  std::unique_ptr<rtc::IceUdpMuxListener> iceUdpMuxListenerPtr = nullptr;
+};
+
+#endif // ICE_UDP_MUX_LISTENER_WRAPPER_H

package/src/cpp/main.cpp

@@ -2,6 +2,7 @@
 #include "rtc-wrapper.h"
 #include "peer-connection-wrapper.h"
 #include "data-channel-wrapper.h"
+#include "ice-udp-mux-listener-wrapper.h"
 
 #if RTC_ENABLE_MEDIA == 1
 #include "media-rtcpreceivingsession-wrapper.h"
@@ -17,24 +18,25 @@
 
 Napi::Object InitAll(Napi::Env env, Napi::Object exports)
 {
-    RtcWrapper::Init(env, exports);
+  RtcWrapper::Init(env, exports);
 
 #if RTC_ENABLE_MEDIA == 1
-    RtcpReceivingSessionWrapper::Init(env, exports);
-    TrackWrapper::Init(env, exports);
-    VideoWrapper::Init(env, exports);
-    AudioWrapper::Init(env, exports);
+  RtcpReceivingSessionWrapper::Init(env, exports);
+  TrackWrapper::Init(env, exports);
+  VideoWrapper::Init(env, exports);
+  AudioWrapper::Init(env, exports);
 #endif
 
-    DataChannelWrapper::Init(env, exports);
-    PeerConnectionWrapper::Init(env, exports);
+  DataChannelWrapper::Init(env, exports);
+  IceUdpMuxListenerWrapper::Init(env, exports);
+  PeerConnectionWrapper::Init(env, exports);
 
 #if RTC_ENABLE_WEBSOCKET == 1
-    WebSocketWrapper::Init(env, exports);
-    WebSocketServerWrapper::Init(env, exports);
+  WebSocketWrapper::Init(env, exports);
+  WebSocketServerWrapper::Init(env, exports);
 #endif
 
-    return exports;
+  return exports;
 }
 
 NODE_API_MODULE(nodeDataChannel, InitAll)