summaryrefslogtreecommitdiffstats
path: root/media/mtransport/nricectx.cpp
diff options
context:
space:
mode:
authorMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
committerMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
commit5f8de423f190bbb79a62f804151bc24824fa32d8 (patch)
tree10027f336435511475e392454359edea8e25895d /media/mtransport/nricectx.cpp
parent49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff)
downloadUXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip
Add m-esr52 at 52.6.0
Diffstat (limited to 'media/mtransport/nricectx.cpp')
-rw-r--r--media/mtransport/nricectx.cpp1017
1 files changed, 1017 insertions, 0 deletions
diff --git a/media/mtransport/nricectx.cpp b/media/mtransport/nricectx.cpp
new file mode 100644
index 000000000..abd8870ee
--- /dev/null
+++ b/media/mtransport/nricectx.cpp
@@ -0,0 +1,1017 @@
+/* -*- mode: c++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+
+// Original author: ekr@rtfm.com
+
+// Some of this code is cut-and-pasted from nICEr. Copyright is:
+
+/*
+Copyright (c) 2007, Adobe Systems, Incorporated
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are
+met:
+
+* Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+
+* Neither the name of Adobe Systems, Network Resonance nor the names of its
+ contributors may be used to endorse or promote products derived from
+ this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+*/
+
+#include <string>
+#include <vector>
+
+#include "mozilla/UniquePtr.h"
+
+#include "logging.h"
+#include "nspr.h"
+#include "nss.h"
+#include "pk11pub.h"
+#include "plbase64.h"
+
+#include "nsCOMPtr.h"
+#include "nsComponentManagerUtils.h"
+#include "nsError.h"
+#include "nsIEventTarget.h"
+#include "nsNetCID.h"
+#include "nsComponentManagerUtils.h"
+#include "nsServiceManagerUtils.h"
+#include "ScopedNSSTypes.h"
+#include "runnable_utils.h"
+#include "nsIPrefService.h"
+#include "nsIPrefBranch.h"
+
+// nICEr includes
+extern "C" {
+#include "nr_api.h"
+#include "registry.h"
+#include "async_timer.h"
+#include "r_crc32.h"
+#include "r_memory.h"
+#include "ice_reg.h"
+#include "ice_util.h"
+#include "transport_addr.h"
+#include "nr_crypto.h"
+#include "nr_socket.h"
+#include "nr_socket_local.h"
+#include "nr_proxy_tunnel.h"
+#include "stun_client_ctx.h"
+#include "stun_reg.h"
+#include "stun_server_ctx.h"
+#include "ice_codeword.h"
+#include "ice_ctx.h"
+#include "ice_candidate.h"
+#include "ice_handler.h"
+}
+
+// Local includes
+#include "nricectx.h"
+#include "nricemediastream.h"
+#include "nr_socket_prsock.h"
+#include "nrinterfaceprioritizer.h"
+#include "rlogconnector.h"
+#include "test_nr_socket.h"
+
+namespace mozilla {
+
+TimeStamp nr_socket_short_term_violation_time() {
+ return NrSocketBase::short_term_violation_time();
+}
+
+TimeStamp nr_socket_long_term_violation_time() {
+ return NrSocketBase::long_term_violation_time();
+}
+
+MOZ_MTLOG_MODULE("mtransport")
+
+const char kNrIceTransportUdp[] = "udp";
+const char kNrIceTransportTcp[] = "tcp";
+
+static bool initialized = false;
+
+// Implement NSPR-based crypto algorithms
+static int nr_crypto_nss_random_bytes(UCHAR *buf, int len) {
+ ScopedPK11SlotInfo slot(PK11_GetInternalSlot());
+ if (!slot)
+ return R_INTERNAL;
+
+ SECStatus rv = PK11_GenerateRandomOnSlot(slot, buf, len);
+ if (rv != SECSuccess)
+ return R_INTERNAL;
+
+ return 0;
+}
+
+static int nr_crypto_nss_hmac(UCHAR *key, int keyl, UCHAR *buf, int bufl,
+ UCHAR *result) {
+ CK_MECHANISM_TYPE mech = CKM_SHA_1_HMAC;
+ PK11SlotInfo *slot = 0;
+ MOZ_ASSERT(keyl > 0);
+ SECItem keyi = { siBuffer, key, static_cast<unsigned int>(keyl)};
+ PK11SymKey *skey = 0;
+ PK11Context *hmac_ctx = 0;
+ SECStatus status;
+ unsigned int hmac_len;
+ SECItem param = { siBuffer, nullptr, 0 };
+ int err = R_INTERNAL;
+
+ slot = PK11_GetInternalKeySlot();
+ if (!slot)
+ goto abort;
+
+ skey = PK11_ImportSymKey(slot, mech, PK11_OriginUnwrap,
+ CKA_SIGN, &keyi, nullptr);
+ if (!skey)
+ goto abort;
+
+
+ hmac_ctx = PK11_CreateContextBySymKey(mech, CKA_SIGN,
+ skey, &param);
+ if (!hmac_ctx)
+ goto abort;
+
+ status = PK11_DigestBegin(hmac_ctx);
+ if (status != SECSuccess)
+ goto abort;
+
+ status = PK11_DigestOp(hmac_ctx, buf, bufl);
+ if (status != SECSuccess)
+ goto abort;
+
+ status = PK11_DigestFinal(hmac_ctx, result, &hmac_len, 20);
+ if (status != SECSuccess)
+ goto abort;
+
+ MOZ_ASSERT(hmac_len == 20);
+
+ err = 0;
+
+ abort:
+ if(hmac_ctx) PK11_DestroyContext(hmac_ctx, PR_TRUE);
+ if (skey) PK11_FreeSymKey(skey);
+ if (slot) PK11_FreeSlot(slot);
+
+ return err;
+}
+
+static int nr_crypto_nss_md5(UCHAR *buf, int bufl, UCHAR *result) {
+ int err = R_INTERNAL;
+ SECStatus rv;
+
+ const SECHashObject *ho = HASH_GetHashObject(HASH_AlgMD5);
+ MOZ_ASSERT(ho);
+ if (!ho)
+ goto abort;
+
+ MOZ_ASSERT(ho->length == 16);
+
+ rv = HASH_HashBuf(ho->type, result, buf, bufl);
+ if (rv != SECSuccess)
+ goto abort;
+
+ err = 0;
+abort:
+ return err;
+}
+
+static nr_ice_crypto_vtbl nr_ice_crypto_nss_vtbl = {
+ nr_crypto_nss_random_bytes,
+ nr_crypto_nss_hmac,
+ nr_crypto_nss_md5
+};
+
+nsresult NrIceStunServer::ToNicerStunStruct(nr_ice_stun_server *server) const {
+ int r;
+
+ memset(server, 0, sizeof(nr_ice_stun_server));
+ if (transport_ == kNrIceTransportUdp) {
+ server->transport = IPPROTO_UDP;
+ } else if (transport_ == kNrIceTransportTcp) {
+ server->transport = IPPROTO_TCP;
+ } else {
+ MOZ_MTLOG(ML_ERROR, "Unsupported STUN server transport: " << transport_);
+ return NS_ERROR_FAILURE;
+ }
+
+ if (has_addr_) {
+ r = nr_praddr_to_transport_addr(&addr_, &server->u.addr,
+ server->transport, 0);
+ if (r) {
+ return NS_ERROR_FAILURE;
+ }
+ server->type=NR_ICE_STUN_SERVER_TYPE_ADDR;
+ }
+ else {
+ MOZ_ASSERT(sizeof(server->u.dnsname.host) > host_.size());
+ PL_strncpyz(server->u.dnsname.host, host_.c_str(),
+ sizeof(server->u.dnsname.host));
+ server->u.dnsname.port = port_;
+ server->type=NR_ICE_STUN_SERVER_TYPE_DNSNAME;
+ }
+
+ return NS_OK;
+}
+
+
+nsresult NrIceTurnServer::ToNicerTurnStruct(nr_ice_turn_server *server) const {
+ memset(server, 0, sizeof(nr_ice_turn_server));
+
+ nsresult rv = ToNicerStunStruct(&server->turn_server);
+ if (NS_FAILED(rv))
+ return rv;
+
+ if (!(server->username=r_strdup(username_.c_str())))
+ return NS_ERROR_OUT_OF_MEMORY;
+
+ // TODO(ekr@rtfm.com): handle non-ASCII passwords somehow?
+ // STUN requires they be SASLpreped, but we don't know if
+ // they are at this point.
+
+ // C++03 23.2.4, Paragraph 1 stipulates that the elements
+ // in std::vector must be contiguous, and can therefore be
+ // used as input to functions expecting C arrays.
+ int r = r_data_create(&server->password,
+ const_cast<UCHAR *>(&password_[0]),
+ password_.size());
+ if (r) {
+ RFREE(server->username);
+ return NS_ERROR_OUT_OF_MEMORY;
+ }
+
+ return NS_OK;
+}
+
+NrIceCtx::NrIceCtx(const std::string& name,
+ bool offerer,
+ Policy policy)
+ : connection_state_(ICE_CTX_INIT),
+ gathering_state_(ICE_CTX_GATHER_INIT),
+ name_(name),
+ offerer_(offerer),
+ ice_controlling_set_(false),
+ streams_(),
+ ctx_(nullptr),
+ peer_(nullptr),
+ ice_handler_vtbl_(nullptr),
+ ice_handler_(nullptr),
+ trickle_(true),
+ policy_(policy),
+ nat_ (nullptr) {
+ // XXX: offerer_ will be used eventually; placate clang in the meantime.
+ (void)offerer_;
+}
+
+// Handler callbacks
+int NrIceCtx::select_pair(void *obj,nr_ice_media_stream *stream,
+ int component_id, nr_ice_cand_pair **potentials,
+ int potential_ct) {
+ MOZ_MTLOG(ML_DEBUG, "select pair called: potential_ct = "
+ << potential_ct);
+
+ return 0;
+}
+
+int NrIceCtx::stream_ready(void *obj, nr_ice_media_stream *stream) {
+ MOZ_MTLOG(ML_DEBUG, "stream_ready called");
+
+ // Get the ICE ctx.
+ NrIceCtx *ctx = static_cast<NrIceCtx *>(obj);
+
+ RefPtr<NrIceMediaStream> s = ctx->FindStream(stream);
+
+ // Streams which do not exist should never be ready.
+ MOZ_ASSERT(s);
+
+ s->Ready();
+
+ return 0;
+}
+
+int NrIceCtx::stream_failed(void *obj, nr_ice_media_stream *stream) {
+ MOZ_MTLOG(ML_DEBUG, "stream_failed called");
+
+ // Get the ICE ctx
+ NrIceCtx *ctx = static_cast<NrIceCtx *>(obj);
+ RefPtr<NrIceMediaStream> s = ctx->FindStream(stream);
+
+ // Streams which do not exist should never fail.
+ MOZ_ASSERT(s);
+
+ ctx->SetConnectionState(ICE_CTX_FAILED);
+ s -> SignalFailed(s);
+ return 0;
+}
+
+int NrIceCtx::ice_checking(void *obj, nr_ice_peer_ctx *pctx) {
+ MOZ_MTLOG(ML_DEBUG, "ice_checking called");
+
+ // Get the ICE ctx
+ NrIceCtx *ctx = static_cast<NrIceCtx *>(obj);
+
+ ctx->SetConnectionState(ICE_CTX_CHECKING);
+
+ return 0;
+}
+
+int NrIceCtx::ice_connected(void *obj, nr_ice_peer_ctx *pctx) {
+ MOZ_MTLOG(ML_DEBUG, "ice_connected called");
+
+ // Get the ICE ctx
+ NrIceCtx *ctx = static_cast<NrIceCtx *>(obj);
+
+ // This is called even on failed contexts.
+ if (ctx->connection_state() != ICE_CTX_FAILED) {
+ ctx->SetConnectionState(ICE_CTX_CONNECTED);
+ }
+
+ return 0;
+}
+
+int NrIceCtx::ice_disconnected(void *obj, nr_ice_peer_ctx *pctx) {
+ MOZ_MTLOG(ML_DEBUG, "ice_disconnected called");
+
+ // Get the ICE ctx
+ NrIceCtx *ctx = static_cast<NrIceCtx *>(obj);
+
+ ctx->SetConnectionState(ICE_CTX_DISCONNECTED);
+
+ return 0;
+}
+
+int NrIceCtx::msg_recvd(void *obj, nr_ice_peer_ctx *pctx,
+ nr_ice_media_stream *stream, int component_id,
+ UCHAR *msg, int len) {
+ // Get the ICE ctx
+ NrIceCtx *ctx = static_cast<NrIceCtx *>(obj);
+ RefPtr<NrIceMediaStream> s = ctx->FindStream(stream);
+
+ // Streams which do not exist should never have packets.
+ MOZ_ASSERT(s);
+
+ s->SignalPacketReceived(s, component_id, msg, len);
+
+ return 0;
+}
+
+void NrIceCtx::trickle_cb(void *arg, nr_ice_ctx *ice_ctx,
+ nr_ice_media_stream *stream,
+ int component_id,
+ nr_ice_candidate *candidate) {
+ // Get the ICE ctx
+ NrIceCtx *ctx = static_cast<NrIceCtx *>(arg);
+ RefPtr<NrIceMediaStream> s = ctx->FindStream(stream);
+
+ if (!s) {
+ // This stream has been removed because it is inactive
+ return;
+ }
+
+ // Format the candidate.
+ char candidate_str[NR_ICE_MAX_ATTRIBUTE_SIZE];
+ int r = nr_ice_format_candidate_attribute(candidate, candidate_str,
+ sizeof(candidate_str));
+ MOZ_ASSERT(!r);
+ if (r)
+ return;
+
+ MOZ_MTLOG(ML_INFO, "NrIceCtx(" << ctx->name_ << "): trickling candidate "
+ << candidate_str);
+
+ s->SignalCandidate(s, candidate_str);
+}
+
+
+void
+NrIceCtx::InitializeGlobals(bool allow_loopback,
+ bool tcp_enabled,
+ bool allow_link_local) {
+ // Initialize the crypto callbacks and logging stuff
+ if (!initialized) {
+ NR_reg_init(NR_REG_MODE_LOCAL);
+ nr_crypto_vtbl = &nr_ice_crypto_nss_vtbl;
+ initialized = true;
+
+ // Set the priorites for candidate type preferences.
+ // These numbers come from RFC 5245 S. 4.1.2.2
+ NR_reg_set_uchar((char *)NR_ICE_REG_PREF_TYPE_SRV_RFLX, 100);
+ NR_reg_set_uchar((char *)NR_ICE_REG_PREF_TYPE_PEER_RFLX, 110);
+ NR_reg_set_uchar((char *)NR_ICE_REG_PREF_TYPE_HOST, 126);
+ NR_reg_set_uchar((char *)NR_ICE_REG_PREF_TYPE_RELAYED, 5);
+ NR_reg_set_uchar((char *)NR_ICE_REG_PREF_TYPE_SRV_RFLX_TCP, 99);
+ NR_reg_set_uchar((char *)NR_ICE_REG_PREF_TYPE_PEER_RFLX_TCP, 109);
+ NR_reg_set_uchar((char *)NR_ICE_REG_PREF_TYPE_HOST_TCP, 125);
+ NR_reg_set_uchar((char *)NR_ICE_REG_PREF_TYPE_RELAYED_TCP, 0);
+
+ int32_t stun_client_maximum_transmits = 7;
+ int32_t ice_trickle_grace_period = 5000;
+ int32_t ice_tcp_so_sock_count = 3;
+ int32_t ice_tcp_listen_backlog = 10;
+ nsAutoCString force_net_interface;
+ nsresult res;
+ nsCOMPtr<nsIPrefService> prefs =
+ do_GetService("@mozilla.org/preferences-service;1", &res);
+
+ if (NS_SUCCEEDED(res)) {
+ nsCOMPtr<nsIPrefBranch> branch = do_QueryInterface(prefs);
+ if (branch) {
+ branch->GetIntPref(
+ "media.peerconnection.ice.stun_client_maximum_transmits",
+ &stun_client_maximum_transmits);
+ branch->GetIntPref(
+ "media.peerconnection.ice.trickle_grace_period",
+ &ice_trickle_grace_period);
+ branch->GetIntPref(
+ "media.peerconnection.ice.tcp_so_sock_count",
+ &ice_tcp_so_sock_count);
+ branch->GetIntPref(
+ "media.peerconnection.ice.tcp_listen_backlog",
+ &ice_tcp_listen_backlog);
+ branch->GetCharPref(
+ "media.peerconnection.ice.force_interface",
+ getter_Copies(force_net_interface));
+ }
+ }
+
+ NR_reg_set_uint4((char *)"stun.client.maximum_transmits",
+ stun_client_maximum_transmits);
+ NR_reg_set_uint4((char *)NR_ICE_REG_TRICKLE_GRACE_PERIOD,
+ ice_trickle_grace_period);
+ NR_reg_set_int4((char *)NR_ICE_REG_ICE_TCP_SO_SOCK_COUNT,
+ ice_tcp_so_sock_count);
+ NR_reg_set_int4((char *)NR_ICE_REG_ICE_TCP_LISTEN_BACKLOG,
+ ice_tcp_listen_backlog);
+
+ NR_reg_set_char((char *)NR_ICE_REG_ICE_TCP_DISABLE, !tcp_enabled);
+
+ if (allow_loopback) {
+ NR_reg_set_char((char *)NR_STUN_REG_PREF_ALLOW_LOOPBACK_ADDRS, 1);
+ }
+
+ if (allow_link_local) {
+ NR_reg_set_char((char *)NR_STUN_REG_PREF_ALLOW_LINK_LOCAL_ADDRS, 1);
+ }
+ if (force_net_interface.Length() > 0) {
+ // Stupid cast.... but needed
+ const nsCString& flat = PromiseFlatCString(static_cast<nsACString&>(force_net_interface));
+ NR_reg_set_string((char *)NR_ICE_REG_PREF_FORCE_INTERFACE_NAME, const_cast<char*>(flat.get()));
+ }
+ }
+}
+
+std::string
+NrIceCtx::GetNewUfrag()
+{
+ char* ufrag;
+ int r;
+
+ if ((r=nr_ice_get_new_ice_ufrag(&ufrag))) {
+ MOZ_CRASH("Unable to get new ice ufrag");
+ return "";
+ }
+
+ std::string ufragStr = ufrag;
+ RFREE(ufrag);
+
+ return ufragStr;
+}
+
+std::string
+NrIceCtx::GetNewPwd()
+{
+ char* pwd;
+ int r;
+
+ if ((r=nr_ice_get_new_ice_pwd(&pwd))) {
+ MOZ_CRASH("Unable to get new ice pwd");
+ return "";
+ }
+
+ std::string pwdStr = pwd;
+ RFREE(pwd);
+
+ return pwdStr;
+}
+
+bool
+NrIceCtx::Initialize()
+{
+ std::string ufrag = GetNewUfrag();
+ std::string pwd = GetNewPwd();
+
+ return Initialize(ufrag, pwd);
+}
+
+bool
+NrIceCtx::Initialize(const std::string& ufrag,
+ const std::string& pwd)
+{
+ MOZ_ASSERT(!ufrag.empty());
+ MOZ_ASSERT(!pwd.empty());
+ if (ufrag.empty() || pwd.empty()) {
+ return false;
+ }
+
+ // Create the ICE context
+ int r;
+
+ UINT4 flags = offerer_ ? NR_ICE_CTX_FLAGS_OFFERER:
+ NR_ICE_CTX_FLAGS_ANSWERER;
+ flags |= NR_ICE_CTX_FLAGS_AGGRESSIVE_NOMINATION;
+ switch (policy_) {
+ case ICE_POLICY_RELAY:
+ flags |= NR_ICE_CTX_FLAGS_RELAY_ONLY;
+ break;
+ case ICE_POLICY_NO_HOST:
+ flags |= NR_ICE_CTX_FLAGS_HIDE_HOST_CANDIDATES;
+ break;
+ case ICE_POLICY_ALL:
+ break;
+ }
+
+ r = nr_ice_ctx_create_with_credentials(const_cast<char *>(name_.c_str()),
+ flags,
+ const_cast<char *>(ufrag.c_str()),
+ const_cast<char *>(pwd.c_str()),
+ &ctx_);
+ MOZ_ASSERT(ufrag == ctx_->ufrag);
+ MOZ_ASSERT(pwd == ctx_->pwd);
+
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't create ICE ctx for '" << name_ << "'");
+ return false;
+ }
+
+ nr_interface_prioritizer *prioritizer = CreateInterfacePrioritizer();
+ if (!prioritizer) {
+ MOZ_MTLOG(LogLevel::Error, "Couldn't create interface prioritizer.");
+ return false;
+ }
+
+ r = nr_ice_ctx_set_interface_prioritizer(ctx_, prioritizer);
+ if (r) {
+ MOZ_MTLOG(LogLevel::Error, "Couldn't set interface prioritizer.");
+ return false;
+ }
+
+ if (generating_trickle()) {
+ r = nr_ice_ctx_set_trickle_cb(ctx_, &NrIceCtx::trickle_cb, this);
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't set trickle cb for '" << name_ << "'");
+ return false;
+ }
+ }
+
+ nsCString mapping_type;
+ nsCString filtering_type;
+ bool block_udp = false;
+
+ nsresult rv;
+ nsCOMPtr<nsIPrefService> pref_service =
+ do_GetService(NS_PREFSERVICE_CONTRACTID, &rv);
+
+ if (NS_SUCCEEDED(rv)) {
+ nsCOMPtr<nsIPrefBranch> pref_branch;
+ rv = pref_service->GetBranch(nullptr, getter_AddRefs(pref_branch));
+ if (NS_SUCCEEDED(rv)) {
+ rv = pref_branch->GetCharPref(
+ "media.peerconnection.nat_simulator.mapping_type",
+ getter_Copies(mapping_type));
+ rv = pref_branch->GetCharPref(
+ "media.peerconnection.nat_simulator.filtering_type",
+ getter_Copies(filtering_type));
+ rv = pref_branch->GetBoolPref(
+ "media.peerconnection.nat_simulator.block_udp",
+ &block_udp);
+ }
+ }
+
+ if (!mapping_type.IsEmpty() && !filtering_type.IsEmpty()) {
+ MOZ_MTLOG(ML_DEBUG, "NAT filtering type: " << filtering_type.get());
+ MOZ_MTLOG(ML_DEBUG, "NAT mapping type: " << mapping_type.get());
+ TestNat* test_nat = new TestNat;
+ test_nat->filtering_type_ = TestNat::ToNatBehavior(filtering_type.get());
+ test_nat->mapping_type_ = TestNat::ToNatBehavior(mapping_type.get());
+ test_nat->block_udp_ = block_udp;
+ test_nat->enabled_ = true;
+ SetNat(test_nat);
+ }
+
+ // Create the handler objects
+ ice_handler_vtbl_ = new nr_ice_handler_vtbl();
+ ice_handler_vtbl_->select_pair = &NrIceCtx::select_pair;
+ ice_handler_vtbl_->stream_ready = &NrIceCtx::stream_ready;
+ ice_handler_vtbl_->stream_failed = &NrIceCtx::stream_failed;
+ ice_handler_vtbl_->ice_connected = &NrIceCtx::ice_connected;
+ ice_handler_vtbl_->msg_recvd = &NrIceCtx::msg_recvd;
+ ice_handler_vtbl_->ice_checking = &NrIceCtx::ice_checking;
+ ice_handler_vtbl_->ice_disconnected = &NrIceCtx::ice_disconnected;
+
+ ice_handler_ = new nr_ice_handler();
+ ice_handler_->vtbl = ice_handler_vtbl_;
+ ice_handler_->obj = this;
+
+ // Create the peer ctx. Because we do not support parallel forking, we
+ // only have one peer ctx.
+ std::string peer_name = name_ + ":default";
+ r = nr_ice_peer_ctx_create(ctx_, ice_handler_,
+ const_cast<char *>(peer_name.c_str()),
+ &peer_);
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't create ICE peer ctx for '" << name_ << "'");
+ return false;
+ }
+
+ sts_target_ = do_GetService(NS_SOCKETTRANSPORTSERVICE_CONTRACTID, &rv);
+
+ if (!NS_SUCCEEDED(rv))
+ return false;
+
+ return true;
+}
+
+int NrIceCtx::SetNat(const RefPtr<TestNat>& aNat) {
+ nat_ = aNat;
+ nr_socket_factory *fac;
+ int r = nat_->create_socket_factory(&fac);
+ if (r) {
+ return r;
+ }
+ nr_ice_ctx_set_socket_factory(ctx_, fac);
+ return 0;
+}
+
+// ONLY USE THIS FOR TESTING. Will cause totally unpredictable and possibly very
+// bad effects if ICE is still live.
+void NrIceCtx::internal_DeinitializeGlobal() {
+ NR_reg_del((char *)"stun");
+ NR_reg_del((char *)"ice");
+ RLogConnector::DestroyInstance();
+ nr_crypto_vtbl = nullptr;
+ initialized = false;
+}
+
+void NrIceCtx::internal_SetTimerAccelarator(int divider) {
+ ctx_->test_timer_divider = divider;
+}
+
+NrIceCtx::~NrIceCtx() {
+ MOZ_MTLOG(ML_DEBUG, "Destroying ICE ctx '" << name_ <<"'");
+ for (auto stream = streams_.begin(); stream != streams_.end(); stream++) {
+ if (*stream) {
+ (*stream)->Close();
+ }
+ }
+ nr_ice_peer_ctx_destroy(&peer_);
+ nr_ice_ctx_destroy(&ctx_);
+ delete ice_handler_vtbl_;
+ delete ice_handler_;
+}
+
+void
+NrIceCtx::SetStream(size_t index, NrIceMediaStream* stream) {
+ if (index >= streams_.size()) {
+ streams_.resize(index + 1);
+ }
+
+ RefPtr<NrIceMediaStream> oldStream(streams_[index]);
+ streams_[index] = stream;
+
+ if (oldStream) {
+ oldStream->Close();
+ }
+}
+
+std::string NrIceCtx::ufrag() const {
+ return ctx_->ufrag;
+}
+
+std::string NrIceCtx::pwd() const {
+ return ctx_->pwd;
+}
+
+void NrIceCtx::destroy_peer_ctx() {
+ nr_ice_peer_ctx_destroy(&peer_);
+}
+
+nsresult NrIceCtx::SetControlling(Controlling controlling) {
+ if (!ice_controlling_set_) {
+ peer_->controlling = (controlling == ICE_CONTROLLING)? 1 : 0;
+ ice_controlling_set_ = true;
+
+ MOZ_MTLOG(ML_DEBUG, "ICE ctx " << name_ << " setting controlling to" <<
+ controlling);
+ }
+ return NS_OK;
+}
+
+NrIceCtx::Controlling NrIceCtx::GetControlling() {
+ return (peer_->controlling) ? ICE_CONTROLLING : ICE_CONTROLLED;
+}
+
+nsresult NrIceCtx::SetPolicy(Policy policy) {
+ policy_ = policy;
+ return NS_OK;
+}
+
+nsresult NrIceCtx::SetStunServers(const std::vector<NrIceStunServer>&
+ stun_servers) {
+ if (stun_servers.empty())
+ return NS_OK;
+
+ auto servers = MakeUnique<nr_ice_stun_server[]>(stun_servers.size());
+
+ for (size_t i=0; i < stun_servers.size(); ++i) {
+ nsresult rv = stun_servers[i].ToNicerStunStruct(&servers[i]);
+ if (NS_FAILED(rv)) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't set STUN server for '" << name_ << "'");
+ return NS_ERROR_FAILURE;
+ }
+ }
+
+ int r = nr_ice_ctx_set_stun_servers(ctx_, servers.get(), stun_servers.size());
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't set STUN server for '" << name_ << "'");
+ return NS_ERROR_FAILURE;
+ }
+
+ return NS_OK;
+}
+
+// TODO(ekr@rtfm.com): This is just SetStunServers with s/Stun/Turn
+// Could we do a template or something?
+nsresult NrIceCtx::SetTurnServers(const std::vector<NrIceTurnServer>&
+ turn_servers) {
+ if (turn_servers.empty())
+ return NS_OK;
+
+ auto servers = MakeUnique<nr_ice_turn_server[]>(turn_servers.size());
+
+ for (size_t i=0; i < turn_servers.size(); ++i) {
+ nsresult rv = turn_servers[i].ToNicerTurnStruct(&servers[i]);
+ if (NS_FAILED(rv)) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't set TURN server for '" << name_ << "'");
+ return NS_ERROR_FAILURE;
+ }
+ }
+
+ int r = nr_ice_ctx_set_turn_servers(ctx_, servers.get(), turn_servers.size());
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't set TURN server for '" << name_ << "'");
+ return NS_ERROR_FAILURE;
+ }
+
+ // TODO(ekr@rtfm.com): This leaks the username/password. Need to free that.
+
+ return NS_OK;
+}
+
+nsresult NrIceCtx::SetResolver(nr_resolver *resolver) {
+ int r = nr_ice_ctx_set_resolver(ctx_, resolver);
+
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't set resolver for '" << name_ << "'");
+ return NS_ERROR_FAILURE;
+ }
+
+ return NS_OK;
+}
+
+nsresult NrIceCtx::SetProxyServer(const NrIceProxyServer& proxy_server) {
+ int r,_status;
+ nr_proxy_tunnel_config *config = nullptr;
+ nr_socket_wrapper_factory *wrapper = nullptr;
+
+ if ((r = nr_proxy_tunnel_config_create(&config))) {
+ ABORT(r);
+ }
+
+ if ((r = nr_proxy_tunnel_config_set_proxy(config,
+ proxy_server.host().c_str(),
+ proxy_server.port()))) {
+ ABORT(r);
+ }
+
+ if ((r = nr_proxy_tunnel_config_set_resolver(config, ctx_->resolver))) {
+ ABORT(r);
+ }
+
+ if ((r = nr_socket_wrapper_factory_proxy_tunnel_create(config, &wrapper))) {
+ MOZ_MTLOG(LogLevel::Error, "Couldn't create proxy tunnel wrapper.");
+ ABORT(r);
+ }
+
+ // nr_ice_ctx will own the wrapper after this call
+ if ((r = nr_ice_ctx_set_turn_tcp_socket_wrapper(ctx_, wrapper))) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't set proxy for '" << name_ << "': " << r);
+ ABORT(r);
+ }
+
+ _status = 0;
+abort:
+ nr_proxy_tunnel_config_destroy(&config);
+ if (_status) {
+ nr_socket_wrapper_factory_destroy(&wrapper);
+ return NS_ERROR_FAILURE;
+ }
+ return NS_OK;
+}
+
+nsresult NrIceCtx::StartGathering(bool default_route_only, bool proxy_only) {
+ ASSERT_ON_THREAD(sts_target_);
+ SetGatheringState(ICE_CTX_GATHER_STARTED);
+
+ if (default_route_only) {
+ nr_ice_ctx_add_flags(ctx_, NR_ICE_CTX_FLAGS_ONLY_DEFAULT_ADDRS);
+ } else {
+ nr_ice_ctx_remove_flags(ctx_, NR_ICE_CTX_FLAGS_ONLY_DEFAULT_ADDRS);
+ }
+
+ if (proxy_only) {
+ nr_ice_ctx_add_flags(ctx_, NR_ICE_CTX_FLAGS_ONLY_PROXY);
+ } else {
+ nr_ice_ctx_remove_flags(ctx_, NR_ICE_CTX_FLAGS_ONLY_PROXY);
+ }
+
+ // This might start gathering for the first time, or again after
+ // renegotiation, or might do nothing at all if gathering has already
+ // finished.
+ int r = nr_ice_gather(ctx_, &NrIceCtx::gather_cb, this);
+
+ if (!r) {
+ SetGatheringState(ICE_CTX_GATHER_COMPLETE);
+ } else if (r != R_WOULDBLOCK) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't gather ICE candidates for '"
+ << name_ << "', error=" << r);
+ SetConnectionState(ICE_CTX_FAILED);
+ return NS_ERROR_FAILURE;
+ }
+
+ return NS_OK;
+}
+
+RefPtr<NrIceMediaStream> NrIceCtx::FindStream(
+ nr_ice_media_stream *stream) {
+ for (size_t i=0; i<streams_.size(); ++i) {
+ if (streams_[i] && (streams_[i]->stream() == stream)) {
+ return streams_[i];
+ }
+ }
+
+ return nullptr;
+}
+
+std::vector<std::string> NrIceCtx::GetGlobalAttributes() {
+ char **attrs = 0;
+ int attrct;
+ int r;
+ std::vector<std::string> ret;
+
+ r = nr_ice_get_global_attributes(ctx_, &attrs, &attrct);
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't get ufrag and password for '"
+ << name_ << "'");
+ return ret;
+ }
+
+ for (int i=0; i<attrct; i++) {
+ ret.push_back(std::string(attrs[i]));
+ RFREE(attrs[i]);
+ }
+ RFREE(attrs);
+
+ return ret;
+}
+
+nsresult NrIceCtx::ParseGlobalAttributes(std::vector<std::string> attrs) {
+ std::vector<char *> attrs_in;
+
+ for (size_t i=0; i<attrs.size(); ++i) {
+ attrs_in.push_back(const_cast<char *>(attrs[i].c_str()));
+ }
+
+ int r = nr_ice_peer_ctx_parse_global_attributes(peer_,
+ attrs_in.size() ?
+ &attrs_in[0] : nullptr,
+ attrs_in.size());
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't parse global attributes for "
+ << name_ << "'");
+ return NS_ERROR_FAILURE;
+ }
+
+ return NS_OK;
+}
+
+nsresult NrIceCtx::StartChecks() {
+ int r;
+
+ r=nr_ice_peer_ctx_pair_candidates(peer_);
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't pair candidates on "
+ << name_ << "'");
+ SetConnectionState(ICE_CTX_FAILED);
+ return NS_ERROR_FAILURE;
+ }
+
+ r = nr_ice_peer_ctx_start_checks2(peer_,1);
+ if (r) {
+ if (r == R_NOT_FOUND) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't start peer checks on "
+ << name_ << "' assuming trickle ICE");
+ } else {
+ MOZ_MTLOG(ML_ERROR, "Couldn't start peer checks on "
+ << name_ << "'");
+ SetConnectionState(ICE_CTX_FAILED);
+ return NS_ERROR_FAILURE;
+ }
+ }
+
+ return NS_OK;
+}
+
+
+void NrIceCtx::gather_cb(NR_SOCKET s, int h, void *arg) {
+ NrIceCtx *ctx = static_cast<NrIceCtx *>(arg);
+
+ ctx->SetGatheringState(ICE_CTX_GATHER_COMPLETE);
+}
+
+nsresult NrIceCtx::Finalize() {
+ int r = nr_ice_ctx_finalize(ctx_, peer_);
+
+ if (r) {
+ MOZ_MTLOG(ML_ERROR, "Couldn't finalize "
+ << name_ << "'");
+ return NS_ERROR_FAILURE;
+ }
+
+ return NS_OK;
+}
+
+void NrIceCtx::SetConnectionState(ConnectionState state) {
+ if (state == connection_state_)
+ return;
+
+ MOZ_MTLOG(ML_INFO, "NrIceCtx(" << name_ << "): state " <<
+ connection_state_ << "->" << state);
+ connection_state_ = state;
+
+ if (connection_state_ == ICE_CTX_FAILED) {
+ MOZ_MTLOG(ML_INFO, "NrIceCtx(" << name_ << "): dumping r_log ringbuffer... ");
+ std::deque<std::string> logs;
+ RLogConnector::GetInstance()->GetAny(0, &logs);
+ for (auto l = logs.begin(); l != logs.end(); ++l) {
+ MOZ_MTLOG(ML_INFO, *l);
+ }
+ }
+
+ SignalConnectionStateChange(this, state);
+}
+
+void NrIceCtx::SetGatheringState(GatheringState state) {
+ if (state == gathering_state_)
+ return;
+
+ MOZ_MTLOG(ML_DEBUG, "NrIceCtx(" << name_ << "): gathering state " <<
+ gathering_state_ << "->" << state);
+ gathering_state_ = state;
+
+ SignalGatheringStateChange(this, state);
+}
+
+} // close namespace
+
+// Reimplement nr_ice_compute_codeword to avoid copyright issues
+void nr_ice_compute_codeword(char *buf, int len,char *codeword) {
+ UINT4 c;
+
+ r_crc32(buf,len,&c);
+
+ PL_Base64Encode(reinterpret_cast<char*>(&c), 3, codeword);
+ codeword[4] = 0;
+
+ return;
+}