1 files changed, 10 insertions, 5 deletions
diff --git a/dom/media/webrtc/jsapi/MediaTransportHandler.h b/dom/media/webrtc/jsapi/MediaTransportHandler.h
index a776cb6fd7..100eff019e 100644
--- a/dom/media/webrtc/jsapi/MediaTransportHandler.h
+++ b/dom/media/webrtc/jsapi/MediaTransportHandler.h
@@ -12,7 +12,8 @@
 #include "transport/dtlsidentity.h"    // For DtlsDigest
 #include "mozilla/dom/RTCPeerConnectionBinding.h"
 #include "mozilla/dom/RTCConfigurationBinding.h"
-#include "transport/nricectx.h"  // Need some enums
+#include "mozilla/dom/RTCIceTransportBinding.h"  // RTCIceTransportState
+#include "transport/nricectx.h"                  // Need some enums
 #include "common/CandidateInfo.h"
 #include "transport/nr_socket_proxy_config.h"
 #include "RTCStatsReport.h"
@@ -122,8 +123,10 @@ class MediaTransportHandler {
 
   sigslot::signal2<const std::string&, const CandidateInfo&> SignalCandidate;
   sigslot::signal2<const std::string&, bool> SignalAlpnNegotiated;
-  sigslot::signal1<dom::RTCIceGatheringState> SignalGatheringStateChange;
-  sigslot::signal1<dom::RTCIceConnectionState> SignalConnectionStateChange;
+  sigslot::signal2<const std::string&, dom::RTCIceGathererState>
+      SignalGatheringStateChange;
+  sigslot::signal2<const std::string&, dom::RTCIceTransportState>
+      SignalConnectionStateChange;
 
   sigslot::signal2<const std::string&, const MediaPacket&> SignalPacketReceived;
   sigslot::signal2<const std::string&, const MediaPacket&>
@@ -142,8 +145,10 @@ class MediaTransportHandler {
   void OnCandidate(const std::string& aTransportId,
                    const CandidateInfo& aCandidateInfo);
   void OnAlpnNegotiated(const std::string& aAlpn);
-  void OnGatheringStateChange(dom::RTCIceGatheringState aState);
-  void OnConnectionStateChange(dom::RTCIceConnectionState aState);
+  void OnGatheringStateChange(const std::string& aTransportId,
+                              dom::RTCIceGathererState aState);
+  void OnConnectionStateChange(const std::string& aTransportId,
+                               dom::RTCIceTransportState aState);
   void OnPacketReceived(const std::string& aTransportId,
                         const MediaPacket& aPacket);
   void OnEncryptedSending(const std::string& aTransportId,