ZLMediaKit/webrtc/SctpAssociation.cpp

1007 lines
34 KiB
C++
Raw Normal View History

2022-06-18 14:34:58 +08:00
#ifdef ENABLE_SCTP
2022-05-04 22:15:21 +08:00
#define MS_CLASS "RTC::SctpAssociation"
// #define MS_LOG_DEV_LEVEL 3
#include "SctpAssociation.hpp"
#include "logger.h"
#include <stdarg.h>
2022-05-04 22:15:21 +08:00
#include <cstdlib> // std::malloc(), std::free()
#include <cstring> // std::memset(), std::memcpy()
#include <string>
#include <memory>
#include <Util/util.h>
using namespace std;
using namespace toolkit;
/* Static. */
static constexpr size_t SctpMtu{ 1200 };
static constexpr uint16_t MaxSctpStreams{ 65535 };
/* SCTP events to which we are subscribing. */
/* clang-format off */
static constexpr uint16_t EventTypes[] =
{
2023-03-03 15:24:12 +08:00
SCTP_ADAPTATION_INDICATION,
SCTP_ASSOC_CHANGE,
SCTP_ASSOC_RESET_EVENT,
SCTP_REMOTE_ERROR,
SCTP_SHUTDOWN_EVENT,
SCTP_SEND_FAILED_EVENT,
SCTP_STREAM_RESET_EVENT,
SCTP_STREAM_CHANGE_EVENT
2022-05-04 22:15:21 +08:00
};
/* clang-format on */
/* Static methods for usrsctp callbacks. */
inline static int onRecvSctpData(
struct socket* /*sock*/,
union sctp_sockstore /*addr*/,
void* data,
size_t len,
struct sctp_rcvinfo rcv,
int flags,
void* ulpInfo)
{
2023-03-03 15:24:12 +08:00
auto* sctpAssociation = static_cast<RTC::SctpAssociation*>(ulpInfo);
if (sctpAssociation == nullptr)
{
std::free(data);
return 0;
}
if (flags & MSG_NOTIFICATION)
{
sctpAssociation->OnUsrSctpReceiveSctpNotification(
static_cast<union sctp_notification*>(data), len);
}
else
{
uint16_t streamId = rcv.rcv_sid;
uint32_t ppid = ntohl(rcv.rcv_ppid);
uint16_t ssn = rcv.rcv_ssn;
MS_DEBUG_TAG(
sctp,
"data chunk received [length:%zu, streamId:%" PRIu16 ", SSN:%" PRIu16 ", TSN:%" PRIu32
", PPID:%" PRIu32 ", context:%" PRIu32 ", flags:%d]",
len,
rcv.rcv_sid,
rcv.rcv_ssn,
rcv.rcv_tsn,
ntohl(rcv.rcv_ppid),
rcv.rcv_context,
flags);
sctpAssociation->OnUsrSctpReceiveSctpData(
streamId, ssn, ppid, flags, static_cast<uint8_t*>(data), len);
}
std::free(data);
return 1;
2022-05-04 22:15:21 +08:00
}
/* Static methods for usrsctp global callbacks. */
inline static int onSendSctpData(void *addr, void *data, size_t len, uint8_t /*tos*/, uint8_t /*setDf*/) {
auto* sctpAssociation = static_cast<RTC::SctpAssociation*>(addr);
if (sctpAssociation == nullptr)
return -1;
sctpAssociation->OnUsrSctpSendSctpData(data, len);
// NOTE: Must not free data, usrsctp lib does it.
return 0;
}
// Static method for printing usrsctp debug.
inline static void sctpDebug(const char *format, ...) {
va_list ap;
va_start(ap, format);
LoggerWrapper::printLogV(getLogger(), LTrace, __FILE__, __FUNCTION__, __LINE__, format, ap);
va_end(ap);
}
namespace RTC
{
class SctpEnv : public enable_shared_from_this<SctpEnv> {
public:
~SctpEnv();
static SctpEnv &Instance();
private:
SctpEnv();
};
INSTANCE_IMP(SctpEnv)
SctpEnv::SctpEnv() {
usrsctp_init(0, onSendSctpData, sctpDebug);
// Disable explicit congestion notifications (ecn).
usrsctp_sysctl_set_sctp_ecn_enable(0);
#ifdef SCTP_DEBUG
usrsctp_sysctl_set_sctp_debug_on(SCTP_DEBUG_ALL);
#endif
}
SctpEnv::~SctpEnv() {
usrsctp_finish();
}
////////////////////////////////////////////////////////////////////////////////////
2023-03-03 15:24:12 +08:00
/* Instance methods. */
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
SctpAssociation::SctpAssociation(
Listener* listener, uint16_t os, uint16_t mis, size_t maxSctpMessageSize, bool isDataChannel)
: listener(listener), os(os), mis(mis), maxSctpMessageSize(maxSctpMessageSize),
isDataChannel(isDataChannel)
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
_env = SctpEnv::Instance().shared_from_this();
2023-03-03 15:24:12 +08:00
// Register ourselves in usrsctp.
usrsctp_register_address(static_cast<void*>(this));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
int ret;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
this->socket = usrsctp_socket(
AF_CONN, SOCK_STREAM, IPPROTO_SCTP, onRecvSctpData, nullptr, 0, static_cast<void*>(this));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (this->socket == nullptr)
MS_THROW_ERROR("usrsctp_socket() failed: %s", std::strerror(errno));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
usrsctp_set_ulpinfo(this->socket, static_cast<void*>(this));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Make the socket non-blocking.
ret = usrsctp_set_non_blocking(this->socket, 1);
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
MS_THROW_ERROR("usrsctp_set_non_blocking() failed: %s", std::strerror(errno));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Set SO_LINGER.
// This ensures that the usrsctp close call deletes the association. This
// prevents usrsctp from calling the global send callback with references to
// this class as the address.
struct linger lingerOpt; // NOLINT(cppcoreguidelines-pro-type-member-init)
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
lingerOpt.l_onoff = 1;
lingerOpt.l_linger = 0;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
ret = usrsctp_setsockopt(this->socket, SOL_SOCKET, SO_LINGER, &lingerOpt, sizeof(lingerOpt));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
MS_THROW_ERROR("usrsctp_setsockopt(SO_LINGER) failed: %s", std::strerror(errno));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Set SCTP_ENABLE_STREAM_RESET.
struct sctp_assoc_value av; // NOLINT(cppcoreguidelines-pro-type-member-init)
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
av.assoc_value =
SCTP_ENABLE_RESET_STREAM_REQ | SCTP_ENABLE_RESET_ASSOC_REQ | SCTP_ENABLE_CHANGE_ASSOC_REQ;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
ret = usrsctp_setsockopt(this->socket, IPPROTO_SCTP, SCTP_ENABLE_STREAM_RESET, &av, sizeof(av));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
{
MS_THROW_ERROR("usrsctp_setsockopt(SCTP_ENABLE_STREAM_RESET) failed: %s", std::strerror(errno));
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Set SCTP_NODELAY.
uint32_t noDelay = 1;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
ret = usrsctp_setsockopt(this->socket, IPPROTO_SCTP, SCTP_NODELAY, &noDelay, sizeof(noDelay));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
MS_THROW_ERROR("usrsctp_setsockopt(SCTP_NODELAY) failed: %s", std::strerror(errno));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Enable events.
struct sctp_event event; // NOLINT(cppcoreguidelines-pro-type-member-init)
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
std::memset(&event, 0, sizeof(event));
event.se_on = 1;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
for (size_t i{ 0 }; i < sizeof(EventTypes) / sizeof(uint16_t); ++i)
{
event.se_type = EventTypes[i];
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
ret = usrsctp_setsockopt(this->socket, IPPROTO_SCTP, SCTP_EVENT, &event, sizeof(event));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
MS_THROW_ERROR("usrsctp_setsockopt(SCTP_EVENT) failed: %s", std::strerror(errno));
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Init message.
struct sctp_initmsg initmsg; // NOLINT(cppcoreguidelines-pro-type-member-init)
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
std::memset(&initmsg, 0, sizeof(initmsg));
initmsg.sinit_num_ostreams = this->os;
initmsg.sinit_max_instreams = this->mis;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
ret = usrsctp_setsockopt(this->socket, IPPROTO_SCTP, SCTP_INITMSG, &initmsg, sizeof(initmsg));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
MS_THROW_ERROR("usrsctp_setsockopt(SCTP_INITMSG) failed: %s", std::strerror(errno));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Server side.
struct sockaddr_conn sconn; // NOLINT(cppcoreguidelines-pro-type-member-init)
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
std::memset(&sconn, 0, sizeof(sconn));
sconn.sconn_family = AF_CONN;
sconn.sconn_port = htons(5000);
sconn.sconn_addr = static_cast<void*>(this);
2022-05-04 22:15:21 +08:00
#ifdef HAVE_SCONN_LEN
sconn.sconn_len = sizeof(sconn);
#endif
2023-03-03 15:24:12 +08:00
ret = usrsctp_bind(this->socket, reinterpret_cast<struct sockaddr*>(&sconn), sizeof(sconn));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
MS_THROW_ERROR("usrsctp_bind() failed: %s", std::strerror(errno));
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
SctpAssociation::~SctpAssociation()
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
usrsctp_set_ulpinfo(this->socket, nullptr);
usrsctp_close(this->socket);
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Deregister ourselves from usrsctp.
usrsctp_deregister_address(static_cast<void*>(this));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
delete[] this->messageBuffer;
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
void SctpAssociation::TransportConnected()
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Just run the SCTP stack if our state is 'new'.
if (this->state != SctpState::NEW)
return;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
try
{
int ret;
struct sockaddr_conn rconn; // NOLINT(cppcoreguidelines-pro-type-member-init)
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
std::memset(&rconn, 0, sizeof(rconn));
rconn.sconn_family = AF_CONN;
rconn.sconn_port = htons(5000);
rconn.sconn_addr = static_cast<void*>(this);
2022-05-04 22:15:21 +08:00
#ifdef HAVE_SCONN_LEN
2023-03-03 15:24:12 +08:00
rconn.sconn_len = sizeof(rconn);
2022-05-04 22:15:21 +08:00
#endif
2023-03-03 15:24:12 +08:00
ret = usrsctp_connect(this->socket, reinterpret_cast<struct sockaddr*>(&rconn), sizeof(rconn));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0 && errno != EINPROGRESS)
MS_THROW_ERROR("usrsctp_connect() failed: %s", std::strerror(errno));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Disable MTU discovery.
sctp_paddrparams peerAddrParams; // NOLINT(cppcoreguidelines-pro-type-member-init)
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
std::memset(&peerAddrParams, 0, sizeof(peerAddrParams));
std::memcpy(&peerAddrParams.spp_address, &rconn, sizeof(rconn));
peerAddrParams.spp_flags = SPP_PMTUD_DISABLE;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// The MTU value provided specifies the space available for chunks in the
// packet, so let's subtract the SCTP header size.
peerAddrParams.spp_pathmtu = SctpMtu - sizeof(peerAddrParams);
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
ret = usrsctp_setsockopt(
this->socket, IPPROTO_SCTP, SCTP_PEER_ADDR_PARAMS, &peerAddrParams, sizeof(peerAddrParams));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
MS_THROW_ERROR("usrsctp_setsockopt(SCTP_PEER_ADDR_PARAMS) failed: %s", std::strerror(errno));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Announce connecting state.
this->state = SctpState::CONNECTING;
this->listener->OnSctpAssociationConnecting(this);
}
catch (... /*error*/)
{
this->state = SctpState::FAILED;
this->listener->OnSctpAssociationFailed(this);
2022-05-04 22:15:21 +08:00
throw;
2023-03-03 15:24:12 +08:00
}
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
void SctpAssociation::ProcessSctpData(const uint8_t* data, size_t len)
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
#if MS_LOG_DEV_LEVEL == 3
2023-03-03 15:24:12 +08:00
MS_DUMP_DATA(data, len);
2022-05-04 22:15:21 +08:00
#endif
2023-03-03 15:24:12 +08:00
usrsctp_conninput(static_cast<void*>(this), data, len, 0);
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
void SctpAssociation::SendSctpMessage(
2022-05-04 22:15:21 +08:00
const RTC::SctpStreamParameters &parameters, uint32_t ppid, const uint8_t* msg, size_t len)
2023-03-03 15:24:12 +08:00
{
MS_TRACE();
// This must be controlled by the DataConsumer.
MS_ASSERT(
len <= this->maxSctpMessageSize,
"given message exceeds max allowed message size [message size:%zu, max message size:%zu]",
len,
this->maxSctpMessageSize);
// Fill stcp_sendv_spa.
struct sctp_sendv_spa spa; // NOLINT(cppcoreguidelines-pro-type-member-init)
std::memset(&spa, 0, sizeof(spa));
spa.sendv_flags = SCTP_SEND_SNDINFO_VALID;
spa.sendv_sndinfo.snd_sid = parameters.streamId;
spa.sendv_sndinfo.snd_ppid = htonl(ppid);
spa.sendv_sndinfo.snd_flags = SCTP_EOR;
// If ordered it must be reliable.
if (parameters.ordered)
{
spa.sendv_prinfo.pr_policy = SCTP_PR_SCTP_NONE;
spa.sendv_prinfo.pr_value = 0;
}
// Configure reliability: https://tools.ietf.org/html/rfc3758
else
{
spa.sendv_flags |= SCTP_SEND_PRINFO_VALID;
spa.sendv_sndinfo.snd_flags |= SCTP_UNORDERED;
if (parameters.maxPacketLifeTime != 0)
{
spa.sendv_prinfo.pr_policy = SCTP_PR_SCTP_TTL;
spa.sendv_prinfo.pr_value = parameters.maxPacketLifeTime;
}
else if (parameters.maxRetransmits != 0)
{
spa.sendv_prinfo.pr_policy = SCTP_PR_SCTP_RTX;
spa.sendv_prinfo.pr_value = parameters.maxRetransmits;
}
}
int ret = usrsctp_sendv(
this->socket, msg, len, nullptr, 0, &spa, static_cast<socklen_t>(sizeof(spa)), SCTP_SENDV_SPA, 0);
if (ret < 0)
{
MS_WARN_TAG(
sctp,
"error sending SCTP message [sid:%" PRIu16 ", ppid:%" PRIu32 ", message size:%zu]: %s",
parameters.streamId,
ppid,
len,
std::strerror(errno));
}
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
void SctpAssociation::HandleDataConsumer(const RTC::SctpStreamParameters &params)
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
auto streamId = params.streamId;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// We need more OS.
if (streamId > this->os - 1)
AddOutgoingStreams(/*force*/ false);
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
void SctpAssociation::DataProducerClosed(const RTC::SctpStreamParameters &params)
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
auto streamId = params.streamId;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Send SCTP_RESET_STREAMS to the remote.
// https://tools.ietf.org/html/draft-ietf-rtcweb-data-channel-13#section-6.7
if (this->isDataChannel)
ResetSctpStream(streamId, StreamDirection::OUTGOING);
else
ResetSctpStream(streamId, StreamDirection::INCOMING);
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
void SctpAssociation::DataConsumerClosed(const RTC::SctpStreamParameters &params)
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
auto streamId = params.streamId;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Send SCTP_RESET_STREAMS to the remote.
ResetSctpStream(streamId, StreamDirection::OUTGOING);
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
void SctpAssociation::ResetSctpStream(uint16_t streamId, StreamDirection direction)
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Do nothing if an outgoing stream that could not be allocated by us.
if (direction == StreamDirection::OUTGOING && streamId > this->os - 1)
return;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
int ret;
struct sctp_assoc_value av; // NOLINT(cppcoreguidelines-pro-type-member-init)
socklen_t len = sizeof(av);
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
#ifndef SCTP_RECONFIG_SUPPORTED
#define SCTP_RECONFIG_SUPPORTED 0x00000029
#endif
ret = usrsctp_getsockopt(this->socket, IPPROTO_SCTP, SCTP_RECONFIG_SUPPORTED, &av, &len);
if (ret == 0)
{
if (av.assoc_value != 1)
{
MS_DEBUG_TAG(sctp, "stream reconfiguration not negotiated");
return;
}
}
else
{
MS_WARN_TAG(
sctp,
"could not retrieve whether stream reconfiguration has been negotiated: %s\n",
std::strerror(errno));
return;
}
// As per spec: https://tools.ietf.org/html/rfc6525#section-4.1
len = sizeof(sctp_assoc_t) + (2 + 1) * sizeof(uint16_t);
auto* srs = static_cast<struct sctp_reset_streams*>(std::malloc(len));
switch (direction)
{
case StreamDirection::INCOMING:
srs->srs_flags = SCTP_STREAM_RESET_INCOMING;
break;
case StreamDirection::OUTGOING:
srs->srs_flags = SCTP_STREAM_RESET_OUTGOING;
break;
}
srs->srs_number_streams = 1;
srs->srs_stream_list[0] = streamId; // No need for htonl().
ret = usrsctp_setsockopt(this->socket, IPPROTO_SCTP, SCTP_RESET_STREAMS, srs, len);
if (ret == 0)
{
MS_DEBUG_TAG(sctp, "SCTP_RESET_STREAMS sent [streamId:%" PRIu16 "]", streamId);
}
else
{
MS_WARN_TAG(sctp, "usrsctp_setsockopt(SCTP_RESET_STREAMS) failed: %s", std::strerror(errno));
}
std::free(srs);
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
void SctpAssociation::AddOutgoingStreams(bool force)
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
uint16_t additionalOs{ 0 };
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (MaxSctpStreams - this->os >= 32)
additionalOs = 32;
else
additionalOs = MaxSctpStreams - this->os;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (additionalOs == 0)
{
MS_WARN_TAG(sctp, "cannot add more outgoing streams [OS:%" PRIu16 "]", this->os);
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
return;
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
auto nextDesiredOs = this->os + additionalOs;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Already in progress, ignore (unless forced).
if (!force && nextDesiredOs == this->desiredOs)
return;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// Update desired value.
this->desiredOs = nextDesiredOs;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
// If not connected, defer it.
if (this->state != SctpState::CONNECTED)
{
MS_DEBUG_TAG(sctp, "SCTP not connected, deferring OS increase");
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
return;
}
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
struct sctp_add_streams sas; // NOLINT(cppcoreguidelines-pro-type-member-init)
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
std::memset(&sas, 0, sizeof(sas));
sas.sas_instrms = 0;
sas.sas_outstrms = additionalOs;
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
MS_DEBUG_TAG(sctp, "adding %" PRIu16 " outgoing streams", additionalOs);
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
int ret = usrsctp_setsockopt(
this->socket, IPPROTO_SCTP, SCTP_ADD_STREAMS, &sas, static_cast<socklen_t>(sizeof(sas)));
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
if (ret < 0)
MS_WARN_TAG(sctp, "usrsctp_setsockopt(SCTP_ADD_STREAMS) failed: %s", std::strerror(errno));
}
void SctpAssociation::OnUsrSctpSendSctpData(void* buffer, size_t len)
{
MS_TRACE();
2022-05-04 22:15:21 +08:00
2023-03-03 15:24:12 +08:00
const uint8_t* data = static_cast<uint8_t*>(buffer);
2022-05-04 22:15:21 +08:00
#if MS_LOG_DEV_LEVEL == 3
2023-03-03 15:24:12 +08:00
MS_DUMP_DATA(data, len);
2022-05-04 22:15:21 +08:00
#endif
2023-03-03 15:24:12 +08:00
this->listener->OnSctpAssociationSendData(this, data, len);
}
void SctpAssociation::OnUsrSctpReceiveSctpData(
uint16_t streamId, uint16_t ssn, uint32_t ppid, int flags, const uint8_t* data, size_t len)
{
// Ignore WebRTC DataChannel Control DATA chunks.
if (ppid == 50)
{
MS_WARN_TAG(sctp, "ignoring SCTP data with ppid:50 (WebRTC DataChannel Control)");
return;
}
if (this->messageBufferLen != 0 && ssn != this->lastSsnReceived)
{
MS_WARN_TAG(
sctp,
"message chunk received with different SSN while buffer not empty, buffer discarded [ssn:%" PRIu16
", last ssn received:%" PRIu16 "]",
ssn,
this->lastSsnReceived);
this->messageBufferLen = 0;
}
// Update last SSN received.
this->lastSsnReceived = ssn;
auto eor = static_cast<bool>(flags & MSG_EOR);
if (this->messageBufferLen + len > this->maxSctpMessageSize)
{
MS_WARN_TAG(
sctp,
"ongoing received message exceeds max allowed message size [message size:%zu, max message size:%zu, eor:%u]",
this->messageBufferLen + len,
this->maxSctpMessageSize,
eor ? 1 : 0);
this->lastSsnReceived = 0;
return;
}
// If end of message and there is no buffered data, notify it directly.
if (eor && this->messageBufferLen == 0)
{
MS_DEBUG_DEV("directly notifying listener [eor:1, buffer len:0]");
this->listener->OnSctpAssociationMessageReceived(this, streamId, ppid, data, len);
}
// If end of message and there is buffered data, append data and notify buffer.
else if (eor && this->messageBufferLen != 0)
{
std::memcpy(this->messageBuffer + this->messageBufferLen, data, len);
this->messageBufferLen += len;
MS_DEBUG_DEV("notifying listener [eor:1, buffer len:%zu]", this->messageBufferLen);
this->listener->OnSctpAssociationMessageReceived(
this, streamId, ppid, this->messageBuffer, this->messageBufferLen);
this->messageBufferLen = 0;
}
// If non end of message, append data to the buffer.
else if (!eor)
{
// Allocate the buffer if not already done.
if (!this->messageBuffer)
this->messageBuffer = new uint8_t[this->maxSctpMessageSize];
std::memcpy(this->messageBuffer + this->messageBufferLen, data, len);
this->messageBufferLen += len;
MS_DEBUG_DEV("data buffered [eor:0, buffer len:%zu]", this->messageBufferLen);
}
}
void SctpAssociation::OnUsrSctpReceiveSctpNotification(union sctp_notification* notification, size_t len)
{
if (notification->sn_header.sn_length != (uint32_t)len)
return;
switch (notification->sn_header.sn_type)
{
case SCTP_ADAPTATION_INDICATION:
{
MS_DEBUG_TAG(
sctp,
"SCTP adaptation indication [%x]",
notification->sn_adaptation_event.sai_adaptation_ind);
break;
}
case SCTP_ASSOC_CHANGE:
{
switch (notification->sn_assoc_change.sac_state)
{
case SCTP_COMM_UP:
{
MS_DEBUG_TAG(
sctp,
"SCTP association connected, streams [out:%" PRIu16 ", in:%" PRIu16 "]",
notification->sn_assoc_change.sac_outbound_streams,
notification->sn_assoc_change.sac_inbound_streams);
// Update our OS.
this->os = notification->sn_assoc_change.sac_outbound_streams;
// Increase if requested before connected.
if (this->desiredOs > this->os)
AddOutgoingStreams(/*force*/ true);
if (this->state != SctpState::CONNECTED)
{
this->state = SctpState::CONNECTED;
this->listener->OnSctpAssociationConnected(this);
}
break;
}
case SCTP_COMM_LOST:
{
if (notification->sn_header.sn_length > 0)
{
static const size_t BufferSize{ 1024 };
static char buffer[BufferSize];
uint32_t len = notification->sn_header.sn_length;
for (uint32_t i{ 0 }; i < len; ++i)
{
std::snprintf(
buffer, BufferSize, " 0x%02x", notification->sn_assoc_change.sac_info[i]);
}
MS_DEBUG_TAG(sctp, "SCTP communication lost [info:%s]", buffer);
}
else
{
MS_DEBUG_TAG(sctp, "SCTP communication lost");
}
if (this->state != SctpState::CLOSED)
{
this->state = SctpState::CLOSED;
this->listener->OnSctpAssociationClosed(this);
}
break;
}
case SCTP_RESTART:
{
MS_DEBUG_TAG(
sctp,
"SCTP remote association restarted, streams [out:%" PRIu16 ", int:%" PRIu16 "]",
notification->sn_assoc_change.sac_outbound_streams,
notification->sn_assoc_change.sac_inbound_streams);
// Update our OS.
this->os = notification->sn_assoc_change.sac_outbound_streams;
// Increase if requested before connected.
if (this->desiredOs > this->os)
AddOutgoingStreams(/*force*/ true);
if (this->state != SctpState::CONNECTED)
{
this->state = SctpState::CONNECTED;
this->listener->OnSctpAssociationConnected(this);
}
break;
}
case SCTP_SHUTDOWN_COMP:
{
MS_DEBUG_TAG(sctp, "SCTP association gracefully closed");
if (this->state != SctpState::CLOSED)
{
this->state = SctpState::CLOSED;
this->listener->OnSctpAssociationClosed(this);
}
break;
}
case SCTP_CANT_STR_ASSOC:
{
if (notification->sn_header.sn_length > 0)
{
static const size_t BufferSize{ 1024 };
static char buffer[BufferSize];
uint32_t len = notification->sn_header.sn_length;
for (uint32_t i{ 0 }; i < len; ++i)
{
std::snprintf(
buffer, BufferSize, " 0x%02x", notification->sn_assoc_change.sac_info[i]);
}
MS_WARN_TAG(sctp, "SCTP setup failed: %s", buffer);
}
if (this->state != SctpState::FAILED)
{
this->state = SctpState::FAILED;
this->listener->OnSctpAssociationFailed(this);
}
break;
}
default:;
}
break;
}
// https://tools.ietf.org/html/rfc6525#section-6.1.2.
case SCTP_ASSOC_RESET_EVENT:
{
MS_DEBUG_TAG(sctp, "SCTP association reset event received");
break;
}
// An Operation Error is not considered fatal in and of itself, but may be
// used with an ABORT chunk to report a fatal condition.
case SCTP_REMOTE_ERROR:
{
static const size_t BufferSize{ 1024 };
static char buffer[BufferSize];
uint32_t len = notification->sn_remote_error.sre_length - sizeof(struct sctp_remote_error);
for (uint32_t i{ 0 }; i < len; i++)
{
std::snprintf(buffer, BufferSize, "0x%02x", notification->sn_remote_error.sre_data[i]);
}
MS_WARN_TAG(
sctp,
"remote SCTP association error [type:0x%04x, data:%s]",
notification->sn_remote_error.sre_error,
buffer);
break;
}
// When a peer sends a SHUTDOWN, SCTP delivers this notification to
// inform the application that it should cease sending data.
case SCTP_SHUTDOWN_EVENT:
{
MS_DEBUG_TAG(sctp, "remote SCTP association shutdown");
if (this->state != SctpState::CLOSED)
{
this->state = SctpState::CLOSED;
this->listener->OnSctpAssociationClosed(this);
}
break;
}
case SCTP_SEND_FAILED_EVENT:
{
static const size_t BufferSize{ 1024 };
static char buffer[BufferSize];
uint32_t len =
notification->sn_send_failed_event.ssfe_length - sizeof(struct sctp_send_failed_event);
for (uint32_t i{ 0 }; i < len; ++i)
{
std::snprintf(buffer, BufferSize, "0x%02x", notification->sn_send_failed_event.ssfe_data[i]);
}
MS_WARN_TAG(
sctp,
"SCTP message sent failure [streamId:%" PRIu16 ", ppid:%" PRIu32
", sent:%s, error:0x%08x, info:%s]",
notification->sn_send_failed_event.ssfe_info.snd_sid,
ntohl(notification->sn_send_failed_event.ssfe_info.snd_ppid),
(notification->sn_send_failed_event.ssfe_flags & SCTP_DATA_SENT) ? "yes" : "no",
notification->sn_send_failed_event.ssfe_error,
buffer);
break;
}
case SCTP_STREAM_RESET_EVENT:
{
bool incoming{ false };
bool outgoing{ false };
uint16_t numStreams =
(notification->sn_strreset_event.strreset_length - sizeof(struct sctp_stream_reset_event)) /
sizeof(uint16_t);
if (notification->sn_strreset_event.strreset_flags & SCTP_STREAM_RESET_INCOMING_SSN)
incoming = true;
if (notification->sn_strreset_event.strreset_flags & SCTP_STREAM_RESET_OUTGOING_SSN)
outgoing = true;
2022-05-04 22:15:21 +08:00
//todo 打印sctp调试信息
2023-03-03 15:24:12 +08:00
if (false /*MS_HAS_DEBUG_TAG(sctp)*/)
{
std::string streamIds;
for (uint16_t i{ 0 }; i < numStreams; ++i)
{
auto streamId = notification->sn_strreset_event.strreset_stream_list[i];
// Don't log more than 5 stream ids.
if (i > 4)
{
streamIds.append("...");
break;
}
if (i > 0)
streamIds.append(",");
streamIds.append(std::to_string(streamId));
}
MS_DEBUG_TAG(
sctp,
"SCTP stream reset event [flags:%x, i|o:%s|%s, num streams:%" PRIu16 ", stream ids:%s]",
notification->sn_strreset_event.strreset_flags,
incoming ? "true" : "false",
outgoing ? "true" : "false",
numStreams,
streamIds.c_str());
}
// Special case for WebRTC DataChannels in which we must also reset our
// outgoing SCTP stream.
if (incoming && !outgoing && this->isDataChannel)
{
for (uint16_t i{ 0 }; i < numStreams; ++i)
{
auto streamId = notification->sn_strreset_event.strreset_stream_list[i];
ResetSctpStream(streamId, StreamDirection::OUTGOING);
}
}
break;
}
case SCTP_STREAM_CHANGE_EVENT:
{
if (notification->sn_strchange_event.strchange_flags == 0)
{
MS_DEBUG_TAG(
sctp,
"SCTP stream changed, streams [out:%" PRIu16 ", in:%" PRIu16 ", flags:%x]",
notification->sn_strchange_event.strchange_outstrms,
notification->sn_strchange_event.strchange_instrms,
notification->sn_strchange_event.strchange_flags);
}
else if (notification->sn_strchange_event.strchange_flags & SCTP_STREAM_RESET_DENIED)
{
MS_WARN_TAG(
sctp,
"SCTP stream change denied, streams [out:%" PRIu16 ", in:%" PRIu16 ", flags:%x]",
notification->sn_strchange_event.strchange_outstrms,
notification->sn_strchange_event.strchange_instrms,
notification->sn_strchange_event.strchange_flags);
break;
}
else if (notification->sn_strchange_event.strchange_flags & SCTP_STREAM_RESET_FAILED)
{
MS_WARN_TAG(
sctp,
"SCTP stream change failed, streams [out:%" PRIu16 ", in:%" PRIu16 ", flags:%x]",
notification->sn_strchange_event.strchange_outstrms,
notification->sn_strchange_event.strchange_instrms,
notification->sn_strchange_event.strchange_flags);
break;
}
// Update OS.
this->os = notification->sn_strchange_event.strchange_outstrms;
break;
}
default:
{
MS_WARN_TAG(
sctp, "unhandled SCTP event received [type:%" PRIu16 "]", notification->sn_header.sn_type);
}
}
}
2022-05-04 22:15:21 +08:00
////////////////////////////////////////////////////////////////////////////////////////
void SctpAssociationImp::OnUsrSctpSendSctpData(void* buffer, size_t len) {
if (_poller->isCurrentThread()) {
SctpAssociation::OnUsrSctpSendSctpData(buffer, len);
} else {
weak_ptr<SctpAssociationImp> weak_self = shared_from_this();
string copy((const char *)buffer, len);
_poller->async([weak_self, copy]() {
auto strong_self = weak_self.lock();
if (strong_self) {
strong_self->SctpAssociation::OnUsrSctpSendSctpData((void *)copy.data(), copy.size());
}
});
}
}
void SctpAssociationImp::OnUsrSctpReceiveSctpData(uint16_t streamId, uint16_t ssn, uint32_t ppid, int flags, const uint8_t* data, size_t len) {
if (_poller->isCurrentThread()) {
SctpAssociation::OnUsrSctpReceiveSctpData(streamId, ssn, ppid, flags, data, len);
} else {
weak_ptr<SctpAssociationImp> weak_self = shared_from_this();
string copy((const char *)data, len);
_poller->async([weak_self, copy, streamId, ssn, ppid, flags]() {
auto strong_self = weak_self.lock();
if (strong_self) {
strong_self->SctpAssociation::OnUsrSctpReceiveSctpData(streamId, ssn, ppid, flags, (const uint8_t* )copy.data(), copy.size());
}
});
}
}
void SctpAssociationImp::OnUsrSctpReceiveSctpNotification(union sctp_notification* notification, size_t len) {
if (_poller->isCurrentThread()) {
SctpAssociation::OnUsrSctpReceiveSctpNotification(notification, len);
} else {
weak_ptr<SctpAssociationImp> weak_self = shared_from_this();
string copy((const char *)notification, len);
_poller->async([weak_self, copy]() {
auto strong_self = weak_self.lock();
if (strong_self) {
strong_self->SctpAssociation::OnUsrSctpReceiveSctpNotification((union sctp_notification *)copy.data(), copy.size());
}
});
}
}
} // namespace RTC
#endif //ENABLE_SCTP