Merge branch 'master' into fix-uwp

This commit is contained in:
Paul-Louis Ageneau
2020-11-03 09:14:29 +01:00
19 changed files with 659 additions and 272 deletions

View File

@ -11,6 +11,7 @@ option(NO_WEBSOCKET "Disable WebSocket support" OFF)
option(NO_EXAMPLES "Disable examples" OFF) option(NO_EXAMPLES "Disable examples" OFF)
option(NO_TESTS "Disable tests build" OFF) option(NO_TESTS "Disable tests build" OFF)
option(WARNINGS_AS_ERRORS "Treat warnings as errors" OFF) option(WARNINGS_AS_ERRORS "Treat warnings as errors" OFF)
option(RSA_KEY_BITS_2048 "Use 2048-bit RSA key instead of 3072-bit" OFF)
option(CAPI_STDCALL "Set calling convention of C API callbacks stdcall" OFF) option(CAPI_STDCALL "Set calling convention of C API callbacks stdcall" OFF)
# Option USE_SRTP defaults to AUTO (enabled if libSRTP is found, else disabled) # Option USE_SRTP defaults to AUTO (enabled if libSRTP is found, else disabled)
set(USE_SRTP AUTO CACHE STRING "Use libSRTP and enable media support") set(USE_SRTP AUTO CACHE STRING "Use libSRTP and enable media support")
@ -229,6 +230,11 @@ else()
target_link_libraries(datachannel-static PRIVATE LibJuice::LibJuiceStatic) target_link_libraries(datachannel-static PRIVATE LibJuice::LibJuiceStatic)
endif() endif()
if(RSA_KEY_BITS_2048)
target_compile_definitions(datachannel PUBLIC RSA_KEY_BITS_2048)
target_compile_definitions(datachannel-static PUBLIC RSA_KEY_BITS_2048)
endif()
if(CAPI_STDCALL) if(CAPI_STDCALL)
target_compile_definitions(datachannel PUBLIC CAPI_STDCALL) target_compile_definitions(datachannel PUBLIC CAPI_STDCALL)
target_compile_definitions(datachannel-static PUBLIC CAPI_STDCALL) target_compile_definitions(datachannel-static PUBLIC CAPI_STDCALL)

View File

@ -34,8 +34,8 @@ namespace rtc {
class Description { class Description {
public: public:
enum class Type { Unspec = 0, Offer = 1, Answer = 2 }; enum class Type { Unspec, Offer, Answer, Pranswer, Rollback };
enum class Role { ActPass = 0, Passive = 1, Active = 2 }; enum class Role { ActPass, Passive, Active };
enum class Direction { SendOnly, RecvOnly, SendRecv, Inactive, Unknown }; enum class Direction { SendOnly, RecvOnly, SendRecv, Inactive, Unknown };
Description(const string &sdp, const string &typeString = ""); Description(const string &sdp, const string &typeString = "");
@ -45,10 +45,9 @@ public:
Type type() const; Type type() const;
string typeString() const; string typeString() const;
Role role() const; Role role() const;
string roleString() const;
string bundleMid() const; string bundleMid() const;
string iceUfrag() const; std::optional<string> iceUfrag() const;
string icePwd() const; std::optional<string> icePwd() const;
std::optional<string> fingerprint() const; std::optional<string> fingerprint() const;
bool ended() const; bool ended() const;
@ -56,6 +55,7 @@ public:
void setFingerprint(string fingerprint); void setFingerprint(string fingerprint);
void addCandidate(Candidate candidate); void addCandidate(Candidate candidate);
void addCandidates(std::vector<Candidate> candidates);
void endCandidates(); void endCandidates();
std::vector<Candidate> extractCandidates(); std::vector<Candidate> extractCandidates();
@ -94,8 +94,7 @@ public:
struct Application : public Entry { struct Application : public Entry {
public: public:
Application(string mid = "data"); Application(string mid = "data");
Application(const Application &other) = default; virtual ~Application() = default;
Application(Application &&other) = default;
string description() const override; string description() const override;
Application reciprocate() const; Application reciprocate() const;
@ -121,8 +120,6 @@ public:
public: public:
Media(const string &sdp); Media(const string &sdp);
Media(const string &mline, string mid, Direction dir = Direction::SendOnly); Media(const string &mline, string mid, Direction dir = Direction::SendOnly);
Media(const Media &other) = default;
Media(Media &&other) = default;
virtual ~Media() = default; virtual ~Media() = default;
string description() const override; string description() const override;
@ -180,6 +177,7 @@ public:
bool hasApplication() const; bool hasApplication() const;
bool hasAudioOrVideo() const; bool hasAudioOrVideo() const;
bool hasMid(string_view mid) const;
int addMedia(Media media); int addMedia(Media media);
int addMedia(Application application); int addMedia(Application application);
@ -193,6 +191,9 @@ public:
Application *application(); Application *application();
static Type stringToType(const string &typeString);
static string typeToString(Type type);
private: private:
std::optional<Candidate> defaultCandidate() const; std::optional<Candidate> defaultCandidate() const;
std::shared_ptr<Entry> createEntry(string mline, string mid, Direction dir); std::shared_ptr<Entry> createEntry(string mline, string mid, Direction dir);
@ -204,7 +205,7 @@ private:
Role mRole; Role mRole;
string mUsername; string mUsername;
string mSessionId; string mSessionId;
string mIceUfrag, mIcePwd; std::optional<string> mIceUfrag, mIcePwd;
std::optional<string> mFingerprint; std::optional<string> mFingerprint;
// Entries // Entries
@ -214,14 +215,12 @@ private:
// Candidates // Candidates
std::vector<Candidate> mCandidates; std::vector<Candidate> mCandidates;
bool mEnded = false; bool mEnded = false;
static Type stringToType(const string &typeString);
static string typeToString(Type type);
static string roleToString(Role role);
}; };
} // namespace rtc } // namespace rtc
std::ostream &operator<<(std::ostream &out, const rtc::Description &description); std::ostream &operator<<(std::ostream &out, const rtc::Description &description);
std::ostream &operator<<(std::ostream &out, rtc::Description::Type type);
std::ostream &operator<<(std::ostream &out, rtc::Description::Role role);
#endif #endif

View File

@ -62,7 +62,7 @@ using std::uint8_t;
const size_t MAX_NUMERICNODE_LEN = 48; // Max IPv6 string representation length const size_t MAX_NUMERICNODE_LEN = 48; // Max IPv6 string representation length
const size_t MAX_NUMERICSERV_LEN = 6; // Max port string representation length const size_t MAX_NUMERICSERV_LEN = 6; // Max port string representation length
const uint16_t DEFAULT_SCTP_PORT = 5000; // SCTP port to use by default const uint16_t DEFAULT_SCTP_PORT = 5000; // SCTP port to use by default
const size_t DEFAULT_MAX_MESSAGE_SIZE = 65536; // Remote max message size if not specified in SDP const size_t DEFAULT_MAX_MESSAGE_SIZE = 65536; // Remote max message size if not specified in SDP
const size_t LOCAL_MAX_MESSAGE_SIZE = 256 * 1024; // Local max message size const size_t LOCAL_MAX_MESSAGE_SIZE = 256 * 1024; // Local max message size
@ -72,7 +72,7 @@ const int THREADPOOL_SIZE = 4; // Number of threads in the global thread pool
// overloaded helper // overloaded helper
template <class... Ts> struct overloaded : Ts... { using Ts::operator()...; }; template <class... Ts> struct overloaded : Ts... { using Ts::operator()...; };
template <class... Ts> overloaded(Ts...)->overloaded<Ts...>; template <class... Ts> overloaded(Ts...) -> overloaded<Ts...>;
// weak_ptr bind helper // weak_ptr bind helper
template <typename F, typename T, typename... Args> auto weak_bind(F &&f, T *t, Args &&... _args) { template <typename F, typename T, typename... Args> auto weak_bind(F &&f, T *t, Args &&... _args) {
@ -85,6 +85,23 @@ template <typename F, typename T, typename... Args> auto weak_bind(F &&f, T *t,
}; };
} }
// scope_guard helper
class scope_guard {
public:
scope_guard(std::function<void()> func) : function(std::move(func)) {}
scope_guard(scope_guard &&other) = delete;
scope_guard(const scope_guard &) = delete;
void operator=(const scope_guard &) = delete;
~scope_guard() {
if (function)
function();
}
private:
std::function<void()> function;
};
template <typename... P> class synchronized_callback { template <typename... P> class synchronized_callback {
public: public:
synchronized_callback() = default; synchronized_callback() = default;
@ -127,6 +144,6 @@ private:
std::function<void(P...)> callback; std::function<void(P...)> callback;
mutable std::recursive_mutex mutex; mutable std::recursive_mutex mutex;
}; };
} } // namespace rtc
#endif #endif

View File

@ -67,6 +67,14 @@ public:
Complete = RTC_GATHERING_COMPLETE Complete = RTC_GATHERING_COMPLETE
}; };
enum class SignalingState : int {
Stable = RTC_SIGNALING_STABLE,
HaveLocalOffer = RTC_SIGNALING_HAVE_LOCAL_OFFER,
HaveRemoteOffer = RTC_SIGNALING_HAVE_REMOTE_OFFER,
HaveLocalPranswer = RTC_SIGNALING_HAVE_LOCAL_PRANSWER,
HaveRemotePranswer = RTC_SIGNALING_HAVE_REMOTE_PRANSWER,
} rtcSignalingState;
PeerConnection(void); PeerConnection(void);
PeerConnection(const Configuration &config); PeerConnection(const Configuration &config);
~PeerConnection(); ~PeerConnection();
@ -76,6 +84,7 @@ public:
const Configuration *config() const; const Configuration *config() const;
State state() const; State state() const;
GatheringState gatheringState() const; GatheringState gatheringState() const;
SignalingState signalingState() const;
bool hasLocalDescription() const; bool hasLocalDescription() const;
bool hasRemoteDescription() const; bool hasRemoteDescription() const;
bool hasMedia() const; bool hasMedia() const;
@ -83,8 +92,9 @@ public:
std::optional<Description> remoteDescription() const; std::optional<Description> remoteDescription() const;
std::optional<string> localAddress() const; std::optional<string> localAddress() const;
std::optional<string> remoteAddress() const; std::optional<string> remoteAddress() const;
bool getSelectedCandidatePair(Candidate *local, Candidate *remote);
void setLocalDescription(); void setLocalDescription(Description::Type type = Description::Type::Unspec);
void setRemoteDescription(Description description); void setRemoteDescription(Description description);
void addRemoteCandidate(Candidate candidate); void addRemoteCandidate(Candidate candidate);
@ -100,6 +110,7 @@ public:
void onLocalCandidate(std::function<void(Candidate candidate)> callback); void onLocalCandidate(std::function<void(Candidate candidate)> callback);
void onStateChange(std::function<void(State state)> callback); void onStateChange(std::function<void(State state)> callback);
void onGatheringStateChange(std::function<void(GatheringState state)> callback); void onGatheringStateChange(std::function<void(GatheringState state)> callback);
void onSignalingStateChange(std::function<void(SignalingState state)> callback);
// Stats // Stats
void clearStats(); void clearStats();
@ -111,9 +122,6 @@ public:
std::shared_ptr<Track> addTrack(Description::Media description); std::shared_ptr<Track> addTrack(Description::Media description);
void onTrack(std::function<void(std::shared_ptr<Track> track)> callback); void onTrack(std::function<void(std::shared_ptr<Track> track)> callback);
// libnice only
bool getSelectedCandidatePair(Candidate *local, Candidate *remote);
private: private:
std::shared_ptr<IceTransport> initIceTransport(Description::Role role); std::shared_ptr<IceTransport> initIceTransport(Description::Role role);
std::shared_ptr<DtlsTransport> initDtlsTransport(); std::shared_ptr<DtlsTransport> initDtlsTransport();
@ -137,12 +145,16 @@ private:
void incomingTrack(Description::Media description); void incomingTrack(Description::Media description);
void openTracks(); void openTracks();
void validateRemoteDescription(const Description &description);
void processLocalDescription(Description description); void processLocalDescription(Description description);
void processLocalCandidate(Candidate candidate); void processLocalCandidate(Candidate candidate);
void processRemoteDescription(Description description);
void processRemoteCandidate(Candidate candidate);
void triggerDataChannel(std::weak_ptr<DataChannel> weakDataChannel); void triggerDataChannel(std::weak_ptr<DataChannel> weakDataChannel);
void triggerTrack(std::shared_ptr<Track> track); void triggerTrack(std::shared_ptr<Track> track);
bool changeState(State state); bool changeState(State state);
bool changeGatheringState(GatheringState state); bool changeGatheringState(GatheringState state);
bool changeSignalingState(SignalingState state);
void resetCallbacks(); void resetCallbacks();
@ -154,6 +166,7 @@ private:
const std::unique_ptr<Processor> mProcessor; const std::unique_ptr<Processor> mProcessor;
std::optional<Description> mLocalDescription, mRemoteDescription; std::optional<Description> mLocalDescription, mRemoteDescription;
std::optional<Description> mCurrentLocalDescription;
mutable std::mutex mLocalDescriptionMutex, mRemoteDescriptionMutex; mutable std::mutex mLocalDescriptionMutex, mRemoteDescriptionMutex;
std::shared_ptr<IceTransport> mIceTransport; std::shared_ptr<IceTransport> mIceTransport;
@ -168,18 +181,22 @@ private:
std::atomic<State> mState; std::atomic<State> mState;
std::atomic<GatheringState> mGatheringState; std::atomic<GatheringState> mGatheringState;
std::atomic<SignalingState> mSignalingState;
std::atomic<bool> mNegotiationNeeded;
synchronized_callback<std::shared_ptr<DataChannel>> mDataChannelCallback; synchronized_callback<std::shared_ptr<DataChannel>> mDataChannelCallback;
synchronized_callback<Description> mLocalDescriptionCallback; synchronized_callback<Description> mLocalDescriptionCallback;
synchronized_callback<Candidate> mLocalCandidateCallback; synchronized_callback<Candidate> mLocalCandidateCallback;
synchronized_callback<State> mStateChangeCallback; synchronized_callback<State> mStateChangeCallback;
synchronized_callback<GatheringState> mGatheringStateChangeCallback; synchronized_callback<GatheringState> mGatheringStateChangeCallback;
synchronized_callback<SignalingState> mSignalingStateChangeCallback;
synchronized_callback<std::shared_ptr<Track>> mTrackCallback; synchronized_callback<std::shared_ptr<Track>> mTrackCallback;
}; };
} // namespace rtc } // namespace rtc
std::ostream &operator<<(std::ostream &out, const rtc::PeerConnection::State &state); std::ostream &operator<<(std::ostream &out, rtc::PeerConnection::State state);
std::ostream &operator<<(std::ostream &out, const rtc::PeerConnection::GatheringState &state); std::ostream &operator<<(std::ostream &out, rtc::PeerConnection::GatheringState state);
std::ostream &operator<<(std::ostream &out, rtc::PeerConnection::SignalingState state);
#endif #endif

View File

@ -59,6 +59,14 @@ typedef enum {
RTC_GATHERING_COMPLETE = 2 RTC_GATHERING_COMPLETE = 2
} rtcGatheringState; } rtcGatheringState;
typedef enum {
RTC_SIGNALING_STABLE = 0,
RTC_SIGNALING_HAVE_LOCAL_OFFER = 1,
RTC_SIGNALING_HAVE_REMOTE_OFFER = 2,
RTC_SIGNALING_HAVE_LOCAL_PRANSWER = 3,
RTC_SIGNALING_HAVE_REMOTE_PRANSWER = 4,
} rtcSignalingState;
typedef enum { // Don't change, it must match plog severity typedef enum { // Don't change, it must match plog severity
RTC_LOG_NONE = 0, RTC_LOG_NONE = 0,
RTC_LOG_FATAL = 1, RTC_LOG_FATAL = 1,
@ -92,6 +100,7 @@ typedef void (RTC_API *rtcDescriptionCallbackFunc)(int pc, const char *sdp, cons
typedef void (RTC_API *rtcCandidateCallbackFunc)(int pc, const char *cand, const char *mid, void *ptr); typedef void (RTC_API *rtcCandidateCallbackFunc)(int pc, const char *cand, const char *mid, void *ptr);
typedef void (RTC_API *rtcStateChangeCallbackFunc)(int pc, rtcState state, void *ptr); typedef void (RTC_API *rtcStateChangeCallbackFunc)(int pc, rtcState state, void *ptr);
typedef void (RTC_API *rtcGatheringStateCallbackFunc)(int pc, rtcGatheringState state, void *ptr); typedef void (RTC_API *rtcGatheringStateCallbackFunc)(int pc, rtcGatheringState state, void *ptr);
typedef void (RTC_API *rtcSignalingStateCallbackFunc)(int pc, rtcSignalingState state, void *ptr);
typedef void (RTC_API *rtcDataChannelCallbackFunc)(int pc, int dc, void *ptr); typedef void (RTC_API *rtcDataChannelCallbackFunc)(int pc, int dc, void *ptr);
typedef void (RTC_API *rtcTrackCallbackFunc)(int pc, int tr, void *ptr); typedef void (RTC_API *rtcTrackCallbackFunc)(int pc, int tr, void *ptr);
typedef void (RTC_API *rtcOpenCallbackFunc)(int id, void *ptr); typedef void (RTC_API *rtcOpenCallbackFunc)(int id, void *ptr);
@ -116,8 +125,9 @@ RTC_EXPORT int rtcSetLocalDescriptionCallback(int pc, rtcDescriptionCallbackFunc
RTC_EXPORT int rtcSetLocalCandidateCallback(int pc, rtcCandidateCallbackFunc cb); RTC_EXPORT int rtcSetLocalCandidateCallback(int pc, rtcCandidateCallbackFunc cb);
RTC_EXPORT int rtcSetStateChangeCallback(int pc, rtcStateChangeCallbackFunc cb); RTC_EXPORT int rtcSetStateChangeCallback(int pc, rtcStateChangeCallbackFunc cb);
RTC_EXPORT int rtcSetGatheringStateChangeCallback(int pc, rtcGatheringStateCallbackFunc cb); RTC_EXPORT int rtcSetGatheringStateChangeCallback(int pc, rtcGatheringStateCallbackFunc cb);
RTC_EXPORT int rtcSetSignalingStateChangeCallback(int pc, rtcSignalingStateCallbackFunc cb);
RTC_EXPORT int rtcSetLocalDescription(int pc); RTC_EXPORT int rtcSetLocalDescription(int pc, const char *type);
RTC_EXPORT int rtcSetRemoteDescription(int pc, const char *sdp, const char *type); RTC_EXPORT int rtcSetRemoteDescription(int pc, const char *sdp, const char *type);
RTC_EXPORT int rtcAddRemoteCandidate(int pc, const char *cand, const char *mid); RTC_EXPORT int rtcAddRemoteCandidate(int pc, const char *cand, const char *mid);

View File

@ -43,6 +43,8 @@ public:
string mid() const; string mid() const;
Description::Media description() const; Description::Media description() const;
void setDescription(Description::Media description);
void close(void) override; void close(void) override;
bool send(message_variant data) override; bool send(message_variant data) override;
bool send(const byte *data, size_t size); bool send(const byte *data, size_t size);

View File

@ -317,7 +317,7 @@ int rtcCreateDataChannel(int pc, const char *label) {
int rtcCreateDataChannelExt(int pc, const char *label, const char *protocol, int rtcCreateDataChannelExt(int pc, const char *label, const char *protocol,
const rtcReliability *reliability) { const rtcReliability *reliability) {
int dc = rtcAddDataChannelExt(pc, label, protocol, reliability); int dc = rtcAddDataChannelExt(pc, label, protocol, reliability);
rtcSetLocalDescription(pc); rtcSetLocalDescription(pc, NULL);
return dc; return dc;
} }
@ -468,6 +468,19 @@ int rtcSetGatheringStateChangeCallback(int pc, rtcGatheringStateCallbackFunc cb)
}); });
} }
int rtcSetSignalingStateChangeCallback(int pc, rtcSignalingStateCallbackFunc cb) {
return WRAP({
auto peerConnection = getPeerConnection(pc);
if (cb)
peerConnection->onSignalingStateChange([pc, cb](PeerConnection::SignalingState state) {
if (auto ptr = getUserPointer(pc))
cb(pc, static_cast<rtcSignalingState>(state), *ptr);
});
else
peerConnection->onGatheringStateChange(nullptr);
});
}
int rtcSetDataChannelCallback(int pc, rtcDataChannelCallbackFunc cb) { int rtcSetDataChannelCallback(int pc, rtcDataChannelCallbackFunc cb) {
return WRAP({ return WRAP({
auto peerConnection = getPeerConnection(pc); auto peerConnection = getPeerConnection(pc);
@ -500,10 +513,11 @@ int rtcSetTrackCallback(int pc, rtcTrackCallbackFunc cb) {
}); });
} }
int rtcSetLocalDescription(int pc) { int rtcSetLocalDescription(int pc, const char *type) {
return WRAP({ return WRAP({
auto peerConnection = getPeerConnection(pc); auto peerConnection = getPeerConnection(pc);
peerConnection->setLocalDescription(); peerConnection->setLocalDescription(type ? Description::stringToType(type)
: Description::Type::Unspec);
}); });
} }

View File

@ -99,7 +99,11 @@ certificate_ptr make_certificate_impl(string commonName) {
unique_ptr<gnutls_x509_crt_t, decltype(&free_crt)> crt(new_crt(), free_crt); unique_ptr<gnutls_x509_crt_t, decltype(&free_crt)> crt(new_crt(), free_crt);
unique_ptr<gnutls_x509_privkey_t, decltype(&free_privkey)> privkey(new_privkey(), free_privkey); unique_ptr<gnutls_x509_privkey_t, decltype(&free_privkey)> privkey(new_privkey(), free_privkey);
#ifdef RSA_KEY_BITS_2048
const unsigned int bits = 2048;
#else
const unsigned int bits = gnutls_sec_param_to_pk_bits(GNUTLS_PK_RSA, GNUTLS_SEC_PARAM_HIGH); const unsigned int bits = gnutls_sec_param_to_pk_bits(GNUTLS_PK_RSA, GNUTLS_SEC_PARAM_HIGH);
#endif
gnutls::check(gnutls_x509_privkey_generate(*privkey, GNUTLS_PK_RSA, bits, 0), gnutls::check(gnutls_x509_privkey_generate(*privkey, GNUTLS_PK_RSA, bits, 0),
"Unable to generate key pair"); "Unable to generate key pair");
@ -190,7 +194,11 @@ certificate_ptr make_certificate_impl(string commonName) {
if (!x509 || !pkey || !rsa || !exponent || !serial_number || !name) if (!x509 || !pkey || !rsa || !exponent || !serial_number || !name)
throw std::runtime_error("Unable allocate structures for certificate generation"); throw std::runtime_error("Unable allocate structures for certificate generation");
const int bits = 4096; #ifdef RSA_KEY_BITS_2048
const int bits = 2048;
#else
const int bits = 3072;
#endif
const unsigned int e = 65537; // 2^16 + 1 const unsigned int e = 65537; // 2^16 + 1
if (!pkey || !rsa || !exponent || !BN_set_word(exponent.get(), e) || if (!pkey || !rsa || !exponent || !BN_set_word(exponent.get(), e) ||

View File

@ -26,6 +26,7 @@
#include <iostream> #include <iostream>
#include <random> #include <random>
#include <sstream> #include <sstream>
#include <unordered_map>
using std::shared_ptr; using std::shared_ptr;
using std::size_t; using std::size_t;
@ -129,12 +130,6 @@ Description::Description(const string &sdp, Type type, Role role)
} }
} }
if (mIceUfrag.empty())
throw std::invalid_argument("Missing ice-ufrag parameter in SDP description");
if (mIcePwd.empty())
throw std::invalid_argument("Missing ice-pwd parameter in SDP description");
if (mUsername.empty()) if (mUsername.empty())
mUsername = "rtc"; mUsername = "rtc";
@ -152,16 +147,14 @@ string Description::typeString() const { return typeToString(mType); }
Description::Role Description::role() const { return mRole; } Description::Role Description::role() const { return mRole; }
string Description::roleString() const { return roleToString(mRole); }
string Description::bundleMid() const { string Description::bundleMid() const {
// Get the mid of the first media // Get the mid of the first media
return !mEntries.empty() ? mEntries[0]->mid() : "0"; return !mEntries.empty() ? mEntries[0]->mid() : "0";
} }
string Description::iceUfrag() const { return mIceUfrag; } std::optional<string> Description::iceUfrag() const { return mIceUfrag; }
string Description::icePwd() const { return mIcePwd; } std::optional<string> Description::icePwd() const { return mIcePwd; }
std::optional<string> Description::fingerprint() const { return mFingerprint; } std::optional<string> Description::fingerprint() const { return mFingerprint; }
@ -183,6 +176,11 @@ void Description::addCandidate(Candidate candidate) {
mCandidates.emplace_back(std::move(candidate)); mCandidates.emplace_back(std::move(candidate));
} }
void Description::addCandidates(std::vector<Candidate> candidates) {
for(auto candidate : candidates)
mCandidates.emplace_back(std::move(candidate));
}
void Description::endCandidates() { mEnded = true; } void Description::endCandidates() { mEnded = true; }
std::vector<Candidate> Description::extractCandidates() { std::vector<Candidate> Description::extractCandidates() {
@ -222,13 +220,14 @@ string Description::generateSdp(string_view eol) const {
// Session-level attributes // Session-level attributes
sdp << "a=msid-semantic:WMS *" << eol; sdp << "a=msid-semantic:WMS *" << eol;
sdp << "a=setup:" << roleToString(mRole) << eol; sdp << "a=setup:" << mRole << eol;
sdp << "a=ice-ufrag:" << mIceUfrag << eol;
sdp << "a=ice-pwd:" << mIcePwd << eol;
if (mIceUfrag)
sdp << "a=ice-ufrag:" << *mIceUfrag << eol;
if (mIcePwd)
sdp << "a=ice-pwd:" << *mIcePwd << eol;
if (!mEnded) if (!mEnded)
sdp << "a=ice-options:trickle" << eol; sdp << "a=ice-options:trickle" << eol;
if (mFingerprint) if (mFingerprint)
sdp << "a=fingerprint:sha-256 " << *mFingerprint << eol; sdp << "a=fingerprint:sha-256 " << *mFingerprint << eol;
@ -281,13 +280,14 @@ string Description::generateApplicationSdp(string_view eol) const {
// Session-level attributes // Session-level attributes
sdp << "a=msid-semantic:WMS *" << eol; sdp << "a=msid-semantic:WMS *" << eol;
sdp << "a=setup:" << roleToString(mRole) << eol; sdp << "a=setup:" << mRole << eol;
sdp << "a=ice-ufrag:" << mIceUfrag << eol;
sdp << "a=ice-pwd:" << mIcePwd << eol;
if (mIceUfrag)
sdp << "a=ice-ufrag:" << *mIceUfrag << eol;
if (mIcePwd)
sdp << "a=ice-pwd:" << *mIcePwd << eol;
if (!mEnded) if (!mEnded)
sdp << "a=ice-options:trickle" << eol; sdp << "a=ice-options:trickle" << eol;
if (mFingerprint) if (mFingerprint)
sdp << "a=fingerprint:sha-256 " << *mFingerprint << eol; sdp << "a=fingerprint:sha-256 " << *mFingerprint << eol;
@ -351,6 +351,14 @@ bool Description::hasAudioOrVideo() const {
return false; return false;
} }
bool Description::hasMid(string_view mid) const {
for (const auto &entry : mEntries)
if (entry->mid() == mid)
return true;
return false;
}
int Description::addMedia(Media media) { int Description::addMedia(Media media) {
mEntries.emplace_back(std::make_shared<Media>(std::move(media))); mEntries.emplace_back(std::make_shared<Media>(std::move(media)));
return int(mEntries.size()) - 1; return int(mEntries.size()) - 1;
@ -767,33 +775,30 @@ Description::Video::Video(string mid, Direction dir)
: Media("video 9 UDP/TLS/RTP/SAVPF", std::move(mid), dir) {} : Media("video 9 UDP/TLS/RTP/SAVPF", std::move(mid), dir) {}
Description::Type Description::stringToType(const string &typeString) { Description::Type Description::stringToType(const string &typeString) {
if (typeString == "offer") using TypeMap_t = std::unordered_map<string, Type>;
return Type::Offer; static const TypeMap_t TypeMap = {{"unspec", Type::Unspec},
else if (typeString == "answer") {"offer", Type::Offer},
return Type::Answer; {"answer", Type::Pranswer},
else {"pranswer", Type::Pranswer},
return Type::Unspec; {"rollback", Type::Rollback}};
auto it = TypeMap.find(typeString);
return it != TypeMap.end() ? it->second : Type::Unspec;
} }
string Description::typeToString(Type type) { string Description::typeToString(Type type) {
switch (type) { switch (type) {
case Type::Unspec:
return "unspec";
case Type::Offer: case Type::Offer:
return "offer"; return "offer";
case Type::Answer: case Type::Answer:
return "answer"; return "answer";
case Type::Pranswer:
return "pranswer";
case Type::Rollback:
return "rollback";
default: default:
return ""; return "unknown";
}
}
string Description::roleToString(Role role) {
switch (role) {
case Role::Active:
return "active";
case Role::Passive:
return "passive";
default:
return "actpass";
} }
} }
@ -802,3 +807,25 @@ string Description::roleToString(Role role) {
std::ostream &operator<<(std::ostream &out, const rtc::Description &description) { std::ostream &operator<<(std::ostream &out, const rtc::Description &description) {
return out << std::string(description); return out << std::string(description);
} }
std::ostream &operator<<(std::ostream &out, rtc::Description::Type type) {
return out << rtc::Description::typeToString(type);
}
std::ostream &operator<<(std::ostream &out, rtc::Description::Role role) {
using Role = rtc::Description::Role;
const char *str;
// Used for SDP generation, do not change
switch (role) {
case Role::Active:
str = "active";
break;
case Role::Passive:
str = "passive";
break;
default:
str = "actpass";
break;
}
return out << str;
}

View File

@ -44,7 +44,8 @@ PeerConnection::PeerConnection() : PeerConnection(Configuration()) {}
PeerConnection::PeerConnection(const Configuration &config) PeerConnection::PeerConnection(const Configuration &config)
: mConfig(config), mCertificate(make_certificate()), mProcessor(std::make_unique<Processor>()), : mConfig(config), mCertificate(make_certificate()), mProcessor(std::make_unique<Processor>()),
mState(State::New), mGatheringState(GatheringState::New) { mState(State::New), mGatheringState(GatheringState::New),
mSignalingState(SignalingState::Stable), mNegotiationNeeded(false) {
PLOG_VERBOSE << "Creating PeerConnection"; PLOG_VERBOSE << "Creating PeerConnection";
if (config.portRangeEnd && config.portRangeBegin > config.portRangeEnd) if (config.portRangeEnd && config.portRangeBegin > config.portRangeEnd)
@ -60,6 +61,8 @@ PeerConnection::~PeerConnection() {
void PeerConnection::close() { void PeerConnection::close() {
PLOG_VERBOSE << "Closing PeerConnection"; PLOG_VERBOSE << "Closing PeerConnection";
mNegotiationNeeded = false;
// Close data channels asynchronously // Close data channels asynchronously
mProcessor->enqueue(std::bind(&PeerConnection::closeDataChannels, this)); mProcessor->enqueue(std::bind(&PeerConnection::closeDataChannels, this));
@ -72,6 +75,8 @@ PeerConnection::State PeerConnection::state() const { return mState; }
PeerConnection::GatheringState PeerConnection::gatheringState() const { return mGatheringState; } PeerConnection::GatheringState PeerConnection::gatheringState() const { return mGatheringState; }
PeerConnection::SignalingState PeerConnection::signalingState() const { return mSignalingState; }
std::optional<Description> PeerConnection::localDescription() const { std::optional<Description> PeerConnection::localDescription() const {
std::lock_guard lock(mLocalDescriptionMutex); std::lock_guard lock(mLocalDescriptionMutex);
return mLocalDescription; return mLocalDescription;
@ -97,88 +102,178 @@ bool PeerConnection::hasMedia() const {
return local && local->hasAudioOrVideo(); return local && local->hasAudioOrVideo();
} }
void PeerConnection::setLocalDescription() { void PeerConnection::setLocalDescription(Description::Type type) {
PLOG_VERBOSE << "Setting local description"; PLOG_VERBOSE << "Setting local description, type=" << Description::typeToString(type);
if (std::atomic_load(&mIceTransport)) { SignalingState signalingState = mSignalingState.load();
PLOG_DEBUG << "Local description is already set, ignoring"; if (type == Description::Type::Rollback) {
if (signalingState == SignalingState::HaveLocalOffer ||
signalingState == SignalingState::HaveLocalPranswer) {
PLOG_DEBUG << "Rolling back pending local description";
std::unique_lock lock(mLocalDescriptionMutex);
if (mCurrentLocalDescription) {
std::vector<Candidate> existingCandidates;
if (mLocalDescription)
existingCandidates = mLocalDescription->extractCandidates();
mLocalDescription.emplace(std::move(*mCurrentLocalDescription));
mLocalDescription->addCandidates(std::move(existingCandidates));
mCurrentLocalDescription.reset();
}
lock.unlock();
changeSignalingState(SignalingState::Stable);
}
return; return;
} }
// RFC 5763: The endpoint that is the offerer MUST use the setup attribute value of // Guess the description type if unspecified
// setup:actpass. if (type == Description::Type::Unspec) {
// See https://tools.ietf.org/html/rfc5763#section-5 if (mSignalingState == SignalingState::HaveRemoteOffer)
auto iceTransport = initIceTransport(Description::Role::ActPass); type = Description::Type::Answer;
Description localDescription = iceTransport->getLocalDescription(Description::Type::Offer); else
processLocalDescription(localDescription); type = Description::Type::Offer;
iceTransport->gatherLocalCandidates(); }
// Only a local offer resets the negotiation needed flag
if (type == Description::Type::Offer && !mNegotiationNeeded.exchange(false)) {
PLOG_DEBUG << "No negotiation needed";
return;
}
// Get the new signaling state
SignalingState newSignalingState;
switch (signalingState) {
case SignalingState::Stable:
if (type != Description::Type::Offer) {
std::ostringstream oss;
oss << "Unexpected local desciption type " << type << " in signaling state "
<< signalingState;
throw std::logic_error(oss.str());
}
newSignalingState = SignalingState::HaveLocalOffer;
break;
case SignalingState::HaveRemoteOffer:
case SignalingState::HaveLocalPranswer:
if (type != Description::Type::Answer && type != Description::Type::Pranswer) {
std::ostringstream oss;
oss << "Unexpected local description type " << type
<< " description in signaling state " << signalingState;
throw std::logic_error(oss.str());
}
newSignalingState = SignalingState::Stable;
break;
default: {
std::ostringstream oss;
oss << "Unexpected local description in signaling state " << signalingState << ", ignoring";
LOG_WARNING << oss.str();
return;
}
}
auto iceTransport = std::atomic_load(&mIceTransport);
if (!iceTransport) {
// RFC 5763: The endpoint that is the offerer MUST use the setup attribute value of
// setup:actpass.
// See https://tools.ietf.org/html/rfc5763#section-5
iceTransport = initIceTransport(Description::Role::ActPass);
}
Description localDescription = iceTransport->getLocalDescription(type);
processLocalDescription(std::move(localDescription));
changeSignalingState(newSignalingState);
if (mGatheringState == GatheringState::New)
iceTransport->gatherLocalCandidates();
} }
void PeerConnection::setRemoteDescription(Description description) { void PeerConnection::setRemoteDescription(Description description) {
PLOG_VERBOSE << "Setting remote description: " << string(description); PLOG_VERBOSE << "Setting remote description: " << string(description);
if (hasRemoteDescription()) if (description.type() == Description::Type::Rollback) {
throw std::logic_error("Remote description is already set"); // This is mostly useless because we accept any offer
PLOG_VERBOSE << "Rolling back pending remote description";
changeSignalingState(SignalingState::Stable);
return;
}
if (description.mediaCount() == 0) validateRemoteDescription(description);
throw std::invalid_argument("Remote description has no media line");
int activeMediaCount = 0; // Get the new signaling state
for (int i = 0; i < description.mediaCount(); ++i) SignalingState signalingState = mSignalingState.load();
std::visit( // reciprocate each media SignalingState newSignalingState;
rtc::overloaded{[&](Description::Application *) { ++activeMediaCount; }, switch (signalingState) {
[&](Description::Media *media) { case SignalingState::Stable:
if (media->direction() != Description::Direction::Inactive) description.hintType(Description::Type::Offer);
++activeMediaCount; if (description.type() != Description::Type::Offer) {
}}, std::ostringstream oss;
description.media(i)); oss << "Unexpected remote " << description.type() << " description in signaling state "
<< signalingState;
if (activeMediaCount == 0) throw std::logic_error(oss.str());
throw std::invalid_argument("Remote description has no active media");
if (!description.fingerprint())
throw std::invalid_argument("Remote description has no fingerprint");
description.hintType(hasLocalDescription() ? Description::Type::Answer
: Description::Type::Offer);
if (description.type() == Description::Type::Offer) {
if (hasLocalDescription()) {
PLOG_ERROR << "Got a remote offer description while an answer was expected";
throw std::logic_error("Got an unexpected remote offer description");
} }
} else { // Answer newSignalingState = SignalingState::HaveRemoteOffer;
if (auto local = localDescription()) { break;
if (description.iceUfrag() == local->iceUfrag() &&
description.icePwd() == local->icePwd()) case SignalingState::HaveLocalOffer:
throw std::logic_error("Got the local description as remote description"); description.hintType(Description::Type::Answer);
} else { if (description.type() == Description::Type::Offer) {
PLOG_ERROR << "Got a remote answer description while an offer was expected"; // The ICE agent will automatically initiate a rollback when a peer that had previously
throw std::logic_error("Got an unexpected remote answer description"); // created an offer receives an offer from the remote peer
setLocalDescription(Description::Type::Rollback);
newSignalingState = SignalingState::HaveRemoteOffer;
break;
} }
if (description.type() != Description::Type::Answer &&
description.type() != Description::Type::Pranswer) {
std::ostringstream oss;
oss << "Unexpected remote " << description.type() << " description in signaling state "
<< signalingState;
throw std::logic_error(oss.str());
}
newSignalingState = SignalingState::Stable;
break;
case SignalingState::HaveRemotePranswer:
description.hintType(Description::Type::Answer);
if (description.type() != Description::Type::Answer &&
description.type() != Description::Type::Pranswer) {
std::ostringstream oss;
oss << "Unexpected remote " << description.type() << " description in signaling state "
<< signalingState;
throw std::logic_error(oss.str());
}
newSignalingState = SignalingState::Stable;
break;
default: {
std::ostringstream oss;
oss << "Unexpected remote description in signaling state " << signalingState;
throw std::logic_error(oss.str());
}
} }
// Candidates will be added at the end, extract them for now // Candidates will be added at the end, extract them for now
auto remoteCandidates = description.extractCandidates(); auto remoteCandidates = description.extractCandidates();
auto type = description.type();
auto iceTransport = std::atomic_load(&mIceTransport); auto iceTransport = std::atomic_load(&mIceTransport);
if (!iceTransport) if (!iceTransport)
iceTransport = initIceTransport(Description::Role::ActPass); iceTransport = initIceTransport(Description::Role::ActPass);
iceTransport->setRemoteDescription(description); iceTransport->setRemoteDescription(description);
processRemoteDescription(std::move(description));
{ changeSignalingState(newSignalingState);
// Set as remote description
std::lock_guard lock(mRemoteDescriptionMutex);
mRemoteDescription.emplace(std::move(description));
}
if (description.type() == Description::Type::Offer) { if (type == Description::Type::Offer) {
// This is an offer and we are the answerer. // This is an offer, we need to answer
Description localDescription = iceTransport->getLocalDescription(Description::Type::Answer); setLocalDescription(Description::Type::Answer);
processLocalDescription(localDescription);
iceTransport->gatherLocalCandidates();
} else { } else {
// This is an answer and we are the offerer. // This is an answer
auto sctpTransport = std::atomic_load(&mSctpTransport); auto sctpTransport = std::atomic_load(&mSctpTransport);
if (!sctpTransport && iceTransport->role() == Description::Role::Active) { if (!sctpTransport && iceTransport->role() == Description::Role::Active) {
// Since we assumed passive role during DataChannel creation, we need to shift the // Since we assumed passive role during DataChannel creation, we need to shift the
@ -203,27 +298,7 @@ void PeerConnection::setRemoteDescription(Description description) {
void PeerConnection::addRemoteCandidate(Candidate candidate) { void PeerConnection::addRemoteCandidate(Candidate candidate) {
PLOG_VERBOSE << "Adding remote candidate: " << string(candidate); PLOG_VERBOSE << "Adding remote candidate: " << string(candidate);
processRemoteCandidate(std::move(candidate));
auto iceTransport = std::atomic_load(&mIceTransport);
if (!mRemoteDescription || !iceTransport)
throw std::logic_error("Remote candidate set without remote description");
if (candidate.resolve(Candidate::ResolveMode::Simple)) {
iceTransport->addRemoteCandidate(candidate);
} else {
// OK, we might need a lookup, do it asynchronously
// We don't use the thread pool because we have no control on the timeout
weak_ptr<IceTransport> weakIceTransport{iceTransport};
std::thread t([weakIceTransport, candidate]() mutable {
if (candidate.resolve(Candidate::ResolveMode::Lookup))
if (auto iceTransport = weakIceTransport.lock())
iceTransport->addRemoteCandidate(candidate);
});
t.detach();
}
std::lock_guard lock(mRemoteDescriptionMutex);
mRemoteDescription->addCandidate(candidate);
} }
std::optional<string> PeerConnection::localAddress() const { std::optional<string> PeerConnection::localAddress() const {
@ -238,11 +313,6 @@ std::optional<string> PeerConnection::remoteAddress() const {
shared_ptr<DataChannel> PeerConnection::addDataChannel(string label, string protocol, shared_ptr<DataChannel> PeerConnection::addDataChannel(string label, string protocol,
Reliability reliability) { Reliability reliability) {
if (auto local = localDescription(); local && !local->hasApplication()) {
PLOG_ERROR << "The PeerConnection was negociated without DataChannel support.";
throw std::runtime_error("No DataChannel support on the PeerConnection");
}
// RFC 5763: The answerer MUST use either a setup attribute value of setup:active or // RFC 5763: The answerer MUST use either a setup attribute value of setup:active or
// setup:passive. [...] Thus, setup:active is RECOMMENDED. // setup:passive. [...] Thus, setup:active is RECOMMENDED.
// See https://tools.ietf.org/html/rfc5763#section-5 // See https://tools.ietf.org/html/rfc5763#section-5
@ -257,6 +327,11 @@ shared_ptr<DataChannel> PeerConnection::addDataChannel(string label, string prot
if (transport->state() == SctpTransport::State::Connected) if (transport->state() == SctpTransport::State::Connected)
channel->open(transport); channel->open(transport);
// Renegotiation is needed iff the current local description does not have application
std::lock_guard lock(mLocalDescriptionMutex);
if (!mLocalDescription || !mLocalDescription->hasApplication())
mNegotiationNeeded = true;
return channel; return channel;
} }
@ -288,21 +363,30 @@ void PeerConnection::onGatheringStateChange(std::function<void(GatheringState st
mGatheringStateChangeCallback = callback; mGatheringStateChangeCallback = callback;
} }
void PeerConnection::onSignalingStateChange(std::function<void(SignalingState state)> callback) {
mSignalingStateChangeCallback = callback;
}
std::shared_ptr<Track> PeerConnection::addTrack(Description::Media description) { std::shared_ptr<Track> PeerConnection::addTrack(Description::Media description) {
if (hasLocalDescription())
throw std::logic_error("Tracks must be created before local description");
if (auto it = mTracks.find(description.mid()); it != mTracks.end())
if (auto track = it->second.lock())
return track;
#if !RTC_ENABLE_MEDIA #if !RTC_ENABLE_MEDIA
if (mTracks.empty()) { if (mTracks.empty()) {
PLOG_WARNING << "Tracks will be inative (not compiled with SRTP support)"; PLOG_WARNING << "Tracks will be inative (not compiled with SRTP support)";
} }
#endif #endif
auto track = std::make_shared<Track>(std::move(description));
mTracks.emplace(std::make_pair(track->mid(), track)); std::shared_ptr<Track> track;
if (auto it = mTracks.find(description.mid()); it != mTracks.end())
if (track = it->second.lock(); track)
track->setDescription(std::move(description));
if (!track) {
track = std::make_shared<Track>(std::move(description));
mTracks.emplace(std::make_pair(track->mid(), track));
}
// Renegotiation is needed for the new or updated track
mNegotiationNeeded = true;
return track; return track;
} }
@ -311,6 +395,7 @@ void PeerConnection::onTrack(std::function<void(std::shared_ptr<Track>)> callbac
} }
shared_ptr<IceTransport> PeerConnection::initIceTransport(Description::Role role) { shared_ptr<IceTransport> PeerConnection::initIceTransport(Description::Role role) {
PLOG_VERBOSE << "Starting ICE transport";
try { try {
if (auto transport = std::atomic_load(&mIceTransport)) if (auto transport = std::atomic_load(&mIceTransport))
return transport; return transport;
@ -373,6 +458,7 @@ shared_ptr<IceTransport> PeerConnection::initIceTransport(Description::Role role
} }
shared_ptr<DtlsTransport> PeerConnection::initDtlsTransport() { shared_ptr<DtlsTransport> PeerConnection::initDtlsTransport() {
PLOG_VERBOSE << "Starting DTLS transport";
try { try {
if (auto transport = std::atomic_load(&mDtlsTransport)) if (auto transport = std::atomic_load(&mDtlsTransport))
return transport; return transport;
@ -388,12 +474,12 @@ shared_ptr<DtlsTransport> PeerConnection::initDtlsTransport() {
switch (state) { switch (state) {
case DtlsTransport::State::Connected: case DtlsTransport::State::Connected:
if (auto local = localDescription(); local && local->hasApplication()) if (auto remote = remoteDescription(); remote && remote->hasApplication())
initSctpTransport(); initSctpTransport();
else else
changeState(State::Connected); changeState(State::Connected);
openTracks(); mProcessor->enqueue(std::bind(&PeerConnection::openTracks, this));
break; break;
case DtlsTransport::State::Failed: case DtlsTransport::State::Failed:
changeState(State::Failed); changeState(State::Failed);
@ -443,42 +529,43 @@ shared_ptr<DtlsTransport> PeerConnection::initDtlsTransport() {
} }
shared_ptr<SctpTransport> PeerConnection::initSctpTransport() { shared_ptr<SctpTransport> PeerConnection::initSctpTransport() {
PLOG_VERBOSE << "Starting SCTP transport";
try { try {
if (auto transport = std::atomic_load(&mSctpTransport)) if (auto transport = std::atomic_load(&mSctpTransport))
return transport; return transport;
auto remote = remoteDescription(); auto remote = remoteDescription();
if (!remote || !remote->application()) if (!remote || !remote->application())
throw std::logic_error("Initializing SCTP transport without application description"); throw std::logic_error("Starting SCTP transport without application description");
uint16_t sctpPort = remote->application()->sctpPort().value_or(DEFAULT_SCTP_PORT); uint16_t sctpPort = remote->application()->sctpPort().value_or(DEFAULT_SCTP_PORT);
auto lower = std::atomic_load(&mDtlsTransport); auto lower = std::atomic_load(&mDtlsTransport);
auto transport = std::make_shared<SctpTransport>( auto transport = std::make_shared<SctpTransport>(
lower, sctpPort, weak_bind(&PeerConnection::forwardMessage, this, _1), lower, sctpPort, weak_bind(&PeerConnection::forwardMessage, this, _1),
weak_bind(&PeerConnection::forwardBufferedAmount, this, _1, _2), weak_bind(&PeerConnection::forwardBufferedAmount, this, _1, _2),
[this, weak_this = weak_from_this()](SctpTransport::State state) { [this, weak_this = weak_from_this()](SctpTransport::State state) {
auto shared_this = weak_this.lock(); auto shared_this = weak_this.lock();
if (!shared_this) if (!shared_this)
return; return;
switch (state) { switch (state) {
case SctpTransport::State::Connected: case SctpTransport::State::Connected:
changeState(State::Connected); changeState(State::Connected);
mProcessor->enqueue(std::bind(&PeerConnection::openDataChannels, this)); mProcessor->enqueue(std::bind(&PeerConnection::openDataChannels, this));
break; break;
case SctpTransport::State::Failed: case SctpTransport::State::Failed:
LOG_WARNING << "SCTP transport failed"; LOG_WARNING << "SCTP transport failed";
changeState(State::Failed); changeState(State::Failed);
mProcessor->enqueue(std::bind(&PeerConnection::remoteCloseDataChannels, this)); mProcessor->enqueue(std::bind(&PeerConnection::remoteCloseDataChannels, this));
break; break;
case SctpTransport::State::Disconnected: case SctpTransport::State::Disconnected:
changeState(State::Disconnected); changeState(State::Disconnected);
mProcessor->enqueue(std::bind(&PeerConnection::remoteCloseDataChannels, this)); mProcessor->enqueue(std::bind(&PeerConnection::remoteCloseDataChannels, this));
break; break;
default: default:
// Ignore // Ignore
break; break;
} }
}); });
std::atomic_store(&mSctpTransport, transport); std::atomic_store(&mSctpTransport, transport);
if (mState == State::Closed) { if (mState == State::Closed) {
@ -499,7 +586,8 @@ void PeerConnection::closeTransports() {
PLOG_VERBOSE << "Closing transports"; PLOG_VERBOSE << "Closing transports";
// Change state to sink state Closed // Change state to sink state Closed
changeState(State::Closed); if (!changeState(State::Closed))
return; // already closed
// Reset callbacks now that state is changed // Reset callbacks now that state is changed
resetCallbacks(); resetCallbacks();
@ -723,40 +811,105 @@ void PeerConnection::openTracks() {
std::shared_lock lock(mTracksMutex); // read-only std::shared_lock lock(mTracksMutex); // read-only
for (auto it = mTracks.begin(); it != mTracks.end(); ++it) for (auto it = mTracks.begin(); it != mTracks.end(); ++it)
if (auto track = it->second.lock()) if (auto track = it->second.lock())
track->open(srtpTransport); if (!track->isOpen())
track->open(srtpTransport);
} }
#endif #endif
} }
void PeerConnection::validateRemoteDescription(const Description &description) {
if (!description.iceUfrag())
throw std::invalid_argument("Remote description has no ICE user fragment");
if (!description.icePwd())
throw std::invalid_argument("Remote description has no ICE password");
if (!description.fingerprint())
throw std::invalid_argument("Remote description has no fingerprint");
if (description.mediaCount() == 0)
throw std::invalid_argument("Remote description has no media line");
int activeMediaCount = 0;
for (int i = 0; i < description.mediaCount(); ++i)
std::visit(rtc::overloaded{[&](const Description::Application *) { ++activeMediaCount; },
[&](const Description::Media *media) {
if (media->direction() != Description::Direction::Inactive)
++activeMediaCount;
}},
description.media(i));
if (activeMediaCount == 0)
throw std::invalid_argument("Remote description has no active media");
if (auto local = localDescription(); local && local->iceUfrag() && local->icePwd())
if (*description.iceUfrag() == *local->iceUfrag() &&
*description.icePwd() == *local->icePwd())
throw std::logic_error("Got the local description as remote description");
PLOG_VERBOSE << "Remote description looks valid";
}
void PeerConnection::processLocalDescription(Description description) { void PeerConnection::processLocalDescription(Description description) {
int activeMediaCount = 0;
if (hasLocalDescription())
throw std::logic_error("Local description is already set");
if (auto remote = remoteDescription()) { if (auto remote = remoteDescription()) {
// Reciprocate remote description // Reciprocate remote description
for (int i = 0; i < remote->mediaCount(); ++i) for (int i = 0; i < remote->mediaCount(); ++i)
std::visit( // reciprocate each media std::visit( // reciprocate each media
rtc::overloaded{ rtc::overloaded{
[&](Description::Application *app) { [&](Description::Application *remoteApp) {
auto reciprocated = app->reciprocate(); std::shared_lock lock(mDataChannelsMutex);
if (!mDataChannels.empty()) {
// Prefer local description
Description::Application app(remoteApp->mid());
app.setSctpPort(DEFAULT_SCTP_PORT);
app.setMaxMessageSize(LOCAL_MAX_MESSAGE_SIZE);
PLOG_DEBUG << "Adding application to local description, mid=\""
<< app.mid() << "\"";
description.addMedia(std::move(app));
return;
}
auto reciprocated = remoteApp->reciprocate();
reciprocated.hintSctpPort(DEFAULT_SCTP_PORT); reciprocated.hintSctpPort(DEFAULT_SCTP_PORT);
reciprocated.setMaxMessageSize(LOCAL_MAX_MESSAGE_SIZE); reciprocated.setMaxMessageSize(LOCAL_MAX_MESSAGE_SIZE);
++activeMediaCount;
PLOG_DEBUG << "Reciprocating application in local description, mid=\"" PLOG_DEBUG << "Reciprocating application in local description, mid=\""
<< reciprocated.mid() << "\""; << reciprocated.mid() << "\"";
description.addMedia(std::move(reciprocated)); description.addMedia(std::move(reciprocated));
}, },
[&](Description::Media *media) { [&](Description::Media *remoteMedia) {
auto reciprocated = media->reciprocate(); std::shared_lock lock(mTracksMutex);
#if RTC_ENABLE_MEDIA if (auto it = mTracks.find(remoteMedia->mid()); it != mTracks.end()) {
if (reciprocated.direction() != Description::Direction::Inactive) // Prefer local description
++activeMediaCount; if (auto track = it->second.lock()) {
#else auto media = track->description();
#if !RTC_ENABLE_MEDIA
// No media support, mark as inactive
media.setDirection(Description::Direction::Inactive);
#endif
PLOG_DEBUG
<< "Adding media to local description, mid=\"" << media.mid()
<< "\", active=" << std::boolalpha
<< (media.direction() != Description::Direction::Inactive);
description.addMedia(std::move(media));
} else {
auto reciprocated = remoteMedia->reciprocate();
reciprocated.setDirection(Description::Direction::Inactive);
PLOG_DEBUG << "Adding inactive media to local description, mid=\""
<< reciprocated.mid() << "\"";
description.addMedia(std::move(reciprocated));
}
return;
}
auto reciprocated = remoteMedia->reciprocate();
#if !RTC_ENABLE_MEDIA
// No media support, mark as inactive // No media support, mark as inactive
reciprocated.setDirection(Description::Direction::Inactive); reciprocated.setDirection(Description::Direction::Inactive);
#endif #endif
@ -771,15 +924,17 @@ void PeerConnection::processLocalDescription(Description description) {
}, },
}, },
remote->media(i)); remote->media(i));
} else { }
if (description.type() == Description::Type::Offer) {
// This is an offer, add locally created data channels and tracks
// Add application for data channels // Add application for data channels
{ if (!description.hasApplication()) {
std::shared_lock lock(mDataChannelsMutex); std::shared_lock lock(mDataChannelsMutex);
if (!mDataChannels.empty()) { if (!mDataChannels.empty()) {
Description::Application app("data"); Description::Application app("data");
app.setSctpPort(DEFAULT_SCTP_PORT); app.setSctpPort(DEFAULT_SCTP_PORT);
app.setMaxMessageSize(LOCAL_MAX_MESSAGE_SIZE); app.setMaxMessageSize(LOCAL_MAX_MESSAGE_SIZE);
++activeMediaCount;
PLOG_DEBUG << "Adding application to local description, mid=\"" << app.mid() PLOG_DEBUG << "Adding application to local description, mid=\"" << app.mid()
<< "\""; << "\"";
@ -789,45 +944,52 @@ void PeerConnection::processLocalDescription(Description description) {
} }
// Add media for local tracks // Add media for local tracks
{ std::shared_lock lock(mTracksMutex);
std::shared_lock lock(mTracksMutex); for (auto it = mTracks.begin(); it != mTracks.end(); ++it) {
for (auto it = mTracks.begin(); it != mTracks.end(); ++it) { if (description.hasMid(it->first))
if (auto track = it->second.lock()) { continue;
auto media = track->description();
#if RTC_ENABLE_MEDIA
if (media.direction() != Description::Direction::Inactive)
++activeMediaCount;
#else
// No media support, mark as inactive
media.setDirection(Description::Direction::Inactive);
#endif
PLOG_DEBUG << "Adding media to local description, mid=\"" << media.mid()
<< "\", active=" << std::boolalpha
<< (media.direction() != Description::Direction::Inactive);
description.addMedia(std::move(media)); if (auto track = it->second.lock()) {
} auto media = track->description();
#if !RTC_ENABLE_MEDIA
// No media support, mark as inactive
media.setDirection(Description::Direction::Inactive);
#endif
PLOG_DEBUG << "Adding media to local description, mid=\"" << media.mid()
<< "\", active=" << std::boolalpha
<< (media.direction() != Description::Direction::Inactive);
description.addMedia(std::move(media));
} }
} }
} }
// There must be at least one active media to negociate
if (activeMediaCount == 0)
throw std::runtime_error("Nothing to negociate");
// Set local fingerprint (wait for certificate if necessary) // Set local fingerprint (wait for certificate if necessary)
description.setFingerprint(mCertificate.get()->fingerprint()); description.setFingerprint(mCertificate.get()->fingerprint());
{ {
// Set as local description // Set as local description
std::lock_guard lock(mLocalDescriptionMutex); std::lock_guard lock(mLocalDescriptionMutex);
std::vector<Candidate> existingCandidates;
if (mLocalDescription) {
existingCandidates = mLocalDescription->extractCandidates();
mCurrentLocalDescription.emplace(std::move(*mLocalDescription));
}
mLocalDescription.emplace(std::move(description)); mLocalDescription.emplace(std::move(description));
mLocalDescription->addCandidates(std::move(existingCandidates));
} }
mProcessor->enqueue([this, description = *mLocalDescription]() { mProcessor->enqueue([this, description = *mLocalDescription]() {
PLOG_VERBOSE << "Issuing local description: " << description; PLOG_VERBOSE << "Issuing local description: " << description;
mLocalDescriptionCallback(std::move(description)); mLocalDescriptionCallback(std::move(description));
}); });
// Reciprocated tracks might need to be open
if (auto dtlsTransport = std::atomic_load(&mDtlsTransport);
dtlsTransport && dtlsTransport->state() == Transport::State::Connected)
mProcessor->enqueue(std::bind(&PeerConnection::openTracks, this));
} }
void PeerConnection::processLocalCandidate(Candidate candidate) { void PeerConnection::processLocalCandidate(Candidate candidate) {
@ -844,6 +1006,56 @@ void PeerConnection::processLocalCandidate(Candidate candidate) {
}); });
} }
void PeerConnection::processRemoteDescription(Description description) {
{
// Set as remote description
std::lock_guard lock(mRemoteDescriptionMutex);
std::vector<Candidate> existingCandidates;
if (mRemoteDescription)
existingCandidates = mRemoteDescription->extractCandidates();
mRemoteDescription.emplace(std::move(description));
mRemoteDescription->addCandidates(std::move(existingCandidates));
}
if (description.hasApplication()) {
auto dtlsTransport = std::atomic_load(&mDtlsTransport);
auto sctpTransport = std::atomic_load(&mSctpTransport);
if (!sctpTransport && dtlsTransport &&
dtlsTransport->state() == Transport::State::Connected)
initSctpTransport();
}
}
void PeerConnection::processRemoteCandidate(Candidate candidate) {
auto iceTransport = std::atomic_load(&mIceTransport);
if (!iceTransport)
throw std::logic_error("Remote candidate set without remote description");
if (candidate.resolve(Candidate::ResolveMode::Simple)) {
iceTransport->addRemoteCandidate(candidate);
} else {
// OK, we might need a lookup, do it asynchronously
// We don't use the thread pool because we have no control on the timeout
weak_ptr<IceTransport> weakIceTransport{iceTransport};
std::thread t([weakIceTransport, candidate]() mutable {
if (candidate.resolve(Candidate::ResolveMode::Lookup))
if (auto iceTransport = weakIceTransport.lock())
iceTransport->addRemoteCandidate(candidate);
});
t.detach();
}
{
std::lock_guard lock(mRemoteDescriptionMutex);
if (!mRemoteDescription)
throw std::logic_error("Got a remote candidate without remote description");
mRemoteDescription->addCandidate(candidate);
}
}
void PeerConnection::triggerDataChannel(weak_ptr<DataChannel> weakDataChannel) { void PeerConnection::triggerDataChannel(weak_ptr<DataChannel> weakDataChannel) {
auto dataChannel = weakDataChannel.lock(); auto dataChannel = weakDataChannel.lock();
if (!dataChannel) if (!dataChannel)
@ -861,10 +1073,10 @@ bool PeerConnection::changeState(State state) {
State current; State current;
do { do {
current = mState.load(); current = mState.load();
if (current == state)
return true;
if (current == State::Closed) if (current == State::Closed)
return false; return false;
if (current == state)
return false;
} while (!mState.compare_exchange_weak(current, state)); } while (!mState.compare_exchange_weak(current, state));
@ -882,12 +1094,24 @@ bool PeerConnection::changeState(State state) {
} }
bool PeerConnection::changeGatheringState(GatheringState state) { bool PeerConnection::changeGatheringState(GatheringState state) {
if (mGatheringState.exchange(state) != state) { if (mGatheringState.exchange(state) == state)
std::ostringstream s; return false;
s << state;
PLOG_INFO << "Changed gathering state to " << s.str(); std::ostringstream s;
mProcessor->enqueue([this, state] { mGatheringStateChangeCallback(state); }); s << state;
} PLOG_INFO << "Changed gathering state to " << s.str();
mProcessor->enqueue([this, state] { mGatheringStateChangeCallback(state); });
return true;
}
bool PeerConnection::changeSignalingState(SignalingState state) {
if (mSignalingState.exchange(state) == state)
return false;
std::ostringstream s;
s << state;
PLOG_INFO << "Changed signaling state to " << s.str();
mProcessor->enqueue([this, state] { mSignalingStateChangeCallback(state); });
return true; return true;
} }
@ -930,15 +1154,14 @@ std::optional<std::chrono::milliseconds> PeerConnection::rtt() {
auto sctpTransport = std::atomic_load(&mSctpTransport); auto sctpTransport = std::atomic_load(&mSctpTransport);
if (sctpTransport) if (sctpTransport)
return sctpTransport->rtt(); return sctpTransport->rtt();
PLOG_WARNING << "Could not load sctpTransport";
return std::nullopt; return std::nullopt;
} }
} // namespace rtc } // namespace rtc
std::ostream &operator<<(std::ostream &out, const rtc::PeerConnection::State &state) { std::ostream &operator<<(std::ostream &out, rtc::PeerConnection::State state) {
using State = rtc::PeerConnection::State; using State = rtc::PeerConnection::State;
std::string str; const char *str;
switch (state) { switch (state) {
case State::New: case State::New:
str = "new"; str = "new";
@ -965,15 +1188,15 @@ std::ostream &operator<<(std::ostream &out, const rtc::PeerConnection::State &st
return out << str; return out << str;
} }
std::ostream &operator<<(std::ostream &out, const rtc::PeerConnection::GatheringState &state) { std::ostream &operator<<(std::ostream &out, rtc::PeerConnection::GatheringState state) {
using GatheringState = rtc::PeerConnection::GatheringState; using GatheringState = rtc::PeerConnection::GatheringState;
std::string str; const char *str;
switch (state) { switch (state) {
case GatheringState::New: case GatheringState::New:
str = "new"; str = "new";
break; break;
case GatheringState::InProgress: case GatheringState::InProgress:
str = "in_progress"; str = "in-progress";
break; break;
case GatheringState::Complete: case GatheringState::Complete:
str = "complete"; str = "complete";
@ -984,3 +1207,29 @@ std::ostream &operator<<(std::ostream &out, const rtc::PeerConnection::Gathering
} }
return out << str; return out << str;
} }
std::ostream &operator<<(std::ostream &out, rtc::PeerConnection::SignalingState state) {
using SignalingState = rtc::PeerConnection::SignalingState;
const char *str;
switch (state) {
case SignalingState::Stable:
str = "stable";
break;
case SignalingState::HaveLocalOffer:
str = "have-local-offer";
break;
case SignalingState::HaveRemoteOffer:
str = "have-remote-offer";
break;
case SignalingState::HaveLocalPranswer:
str = "have-local-pranswer";
break;
case SignalingState::HaveRemotePranswer:
str = "have-remote-pranswer";
break;
default:
str = "unknown";
break;
}
return out << str;
}

View File

@ -45,7 +45,7 @@ public:
void join(); void join();
template <class F, class... Args> template <class F, class... Args>
auto enqueue(F &&f, Args &&... args) -> invoke_future_t<F, Args...>; void enqueue(F &&f, Args &&... args);
protected: protected:
void schedule(); void schedule();
@ -60,31 +60,20 @@ protected:
std::condition_variable mCondition; std::condition_variable mCondition;
}; };
template <class F, class... Args> template <class F, class... Args> void Processor::enqueue(F &&f, Args &&... args) {
auto Processor::enqueue(F &&f, Args &&... args) -> invoke_future_t<F, Args...> {
std::unique_lock lock(mMutex); std::unique_lock lock(mMutex);
using R = std::invoke_result_t<std::decay_t<F>, std::decay_t<Args>...>; auto bound = std::bind(std::forward<F>(f), std::forward<Args>(args)...);
auto task = std::make_shared<std::packaged_task<R()>>( auto task = [this, bound = std::move(bound)]() mutable {
std::bind(std::forward<F>(f), std::forward<Args>(args)...)); scope_guard guard(std::bind(&Processor::schedule, this)); // chain the next task
std::future<R> result = task->get_future(); return bound();
auto bundle = [this, task = std::move(task)]() {
try {
(*task)();
} catch (const std::exception &e) {
PLOG_WARNING << "Unhandled exception in task: " << e.what();
}
schedule(); // chain the next task
}; };
if (!mPending) { if (!mPending) {
ThreadPool::Instance().enqueue(std::move(bundle)); ThreadPool::Instance().enqueue(std::move(task));
mPending = true; mPending = true;
} else { } else {
mTasks.emplace(std::move(bundle)); mTasks.emplace(std::move(task));
} }
return result;
} }
} // namespace rtc } // namespace rtc

View File

@ -58,11 +58,7 @@ void ThreadPool::run() {
bool ThreadPool::runOne() { bool ThreadPool::runOne() {
if (auto task = dequeue()) { if (auto task = dequeue()) {
try { task();
task();
} catch (const std::exception &e) {
PLOG_WARNING << "Unhandled exception in task: " << e.what();
}
return true; return true;
} }
return false; return false;

View File

@ -73,8 +73,15 @@ template <class F, class... Args>
auto ThreadPool::enqueue(F &&f, Args &&... args) -> invoke_future_t<F, Args...> { auto ThreadPool::enqueue(F &&f, Args &&... args) -> invoke_future_t<F, Args...> {
std::unique_lock lock(mMutex); std::unique_lock lock(mMutex);
using R = std::invoke_result_t<std::decay_t<F>, std::decay_t<Args>...>; using R = std::invoke_result_t<std::decay_t<F>, std::decay_t<Args>...>;
auto task = std::make_shared<std::packaged_task<R()>>( auto bound = std::bind(std::forward<F>(f), std::forward<Args>(args)...);
std::bind(std::forward<F>(f), std::forward<Args>(args)...)); auto task = std::make_shared<std::packaged_task<R()>>([bound = std::move(bound)]() mutable {
try {
return bound();
} catch (const std::exception &e) {
PLOG_WARNING << e.what();
throw;
}
});
std::future<R> result = task->get_future(); std::future<R> result = task->get_future();
mTasks.emplace([task = std::move(task), token = Init::Token()]() { return (*task)(); }); mTasks.emplace([task = std::move(task), token = Init::Token()]() { return (*task)(); });

View File

@ -32,6 +32,13 @@ string Track::mid() const { return mMediaDescription.mid(); }
Description::Media Track::description() const { return mMediaDescription; } Description::Media Track::description() const { return mMediaDescription; }
void Track::setDescription(Description::Media description) {
if(description.mid() != mMediaDescription.mid())
throw std::logic_error("Media description mid does not match track mid");
mMediaDescription = std::move(description);
}
void Track::close() { void Track::close() {
mIsClosed = true; mIsClosed = true;
resetCallbacks(); resetCallbacks();

View File

@ -159,6 +159,7 @@ void WebSocket::incoming(message_ptr message) {
} }
shared_ptr<TcpTransport> WebSocket::initTcpTransport() { shared_ptr<TcpTransport> WebSocket::initTcpTransport() {
PLOG_VERBOSE << "Starting TCP transport";
using State = TcpTransport::State; using State = TcpTransport::State;
try { try {
std::lock_guard lock(mInitMutex); std::lock_guard lock(mInitMutex);
@ -205,6 +206,7 @@ shared_ptr<TcpTransport> WebSocket::initTcpTransport() {
} }
shared_ptr<TlsTransport> WebSocket::initTlsTransport() { shared_ptr<TlsTransport> WebSocket::initTlsTransport() {
PLOG_VERBOSE << "Starting TLS transport";
using State = TlsTransport::State; using State = TlsTransport::State;
try { try {
std::lock_guard lock(mInitMutex); std::lock_guard lock(mInitMutex);
@ -262,6 +264,7 @@ shared_ptr<TlsTransport> WebSocket::initTlsTransport() {
} }
shared_ptr<WsTransport> WebSocket::initWsTransport() { shared_ptr<WsTransport> WebSocket::initWsTransport() {
PLOG_VERBOSE << "Starting WebSocket transport";
using State = WsTransport::State; using State = WsTransport::State;
try { try {
std::lock_guard lock(mInitMutex); std::lock_guard lock(mInitMutex);
@ -340,6 +343,6 @@ void WebSocket::closeTransports() {
}); });
} }
} // namespace rtc } // namespace rtc
#endif #endif

View File

@ -34,6 +34,7 @@ static void sleep(unsigned int secs) { Sleep(secs * 1000); }
typedef struct { typedef struct {
rtcState state; rtcState state;
rtcGatheringState gatheringState; rtcGatheringState gatheringState;
rtcSignalingState signalingState;
int pc; int pc;
int dc; int dc;
bool connected; bool connected;
@ -68,6 +69,12 @@ static void RTC_API gatheringStateCallback(int pc, rtcGatheringState state, void
printf("Gathering state %d: %d\n", peer == peer1 ? 1 : 2, (int)state); printf("Gathering state %d: %d\n", peer == peer1 ? 1 : 2, (int)state);
} }
static void RTC_API signalingStateCallback(int pc, rtcSignalingState state, void *ptr) {
Peer *peer = (Peer *)ptr;
peer->signalingState = state;
printf("Signaling state %d: %d\n", peer == peer1 ? 1 : 2, (int)state);
}
static void RTC_API openCallback(int id, void *ptr) { static void RTC_API openCallback(int id, void *ptr) {
Peer *peer = (Peer *)ptr; Peer *peer = (Peer *)ptr;
peer->connected = true; peer->connected = true;
@ -180,6 +187,12 @@ int test_capi_connectivity_main() {
goto error; goto error;
} }
if (peer1->signalingState != RTC_SIGNALING_STABLE ||
peer2->signalingState != RTC_SIGNALING_STABLE) {
fprintf(stderr, "Signaling state is not stable\n");
goto error;
}
if (!peer1->connected || !peer2->connected) { if (!peer1->connected || !peer2->connected) {
fprintf(stderr, "DataChannel is not connected\n"); fprintf(stderr, "DataChannel is not connected\n");
goto error; goto error;
@ -236,7 +249,6 @@ int test_capi_connectivity_main() {
} }
printf("Remote address 2: %s\n", buffer); printf("Remote address 2: %s\n", buffer);
if (rtcGetSelectedCandidatePair(peer1->pc, buffer, BUFFER_SIZE, buffer2, BUFFER_SIZE) < 0) { if (rtcGetSelectedCandidatePair(peer1->pc, buffer, BUFFER_SIZE, buffer2, BUFFER_SIZE) < 0) {
fprintf(stderr, "rtcGetSelectedCandidatePair failed\n"); fprintf(stderr, "rtcGetSelectedCandidatePair failed\n");
goto error; goto error;
@ -251,7 +263,6 @@ int test_capi_connectivity_main() {
printf("Local candidate 2: %s\n", buffer); printf("Local candidate 2: %s\n", buffer);
printf("Remote candidate 2: %s\n", buffer2); printf("Remote candidate 2: %s\n", buffer2);
deletePeer(peer1); deletePeer(peer1);
sleep(1); sleep(1);
deletePeer(peer2); deletePeer(peer2);

View File

@ -156,7 +156,7 @@ int test_capi_track_main() {
rtcSetClosedCallback(peer1->tr, closedCallback); rtcSetClosedCallback(peer1->tr, closedCallback);
// Initiate the handshake // Initiate the handshake
rtcSetLocalDescription(peer1->pc); rtcSetLocalDescription(peer1->pc, NULL);
attempts = 10; attempts = 10;
while ((!peer2->connected || !peer1->connected) && attempts--) while ((!peer2->connected || !peer1->connected) && attempts--)

View File

@ -69,6 +69,10 @@ void test_connectivity() {
cout << "Gathering state 1: " << state << endl; cout << "Gathering state 1: " << state << endl;
}); });
pc1->onSignalingStateChange([](PeerConnection::SignalingState state) {
cout << "Signaling state 1: " << state << endl;
});
pc2->onLocalDescription([wpc1 = make_weak_ptr(pc1)](Description sdp) { pc2->onLocalDescription([wpc1 = make_weak_ptr(pc1)](Description sdp) {
auto pc1 = wpc1.lock(); auto pc1 = wpc1.lock();
if (!pc1) if (!pc1)
@ -91,6 +95,10 @@ void test_connectivity() {
cout << "Gathering state 2: " << state << endl; cout << "Gathering state 2: " << state << endl;
}); });
pc2->onSignalingStateChange([](PeerConnection::SignalingState state) {
cout << "Signaling state 2: " << state << endl;
});
shared_ptr<DataChannel> dc2; shared_ptr<DataChannel> dc2;
pc2->onDataChannel([&dc2](shared_ptr<DataChannel> dc) { pc2->onDataChannel([&dc2](shared_ptr<DataChannel> dc) {
cout << "DataChannel 2: Received with label \"" << dc->label() << "\"" << endl; cout << "DataChannel 2: Received with label \"" << dc->label() << "\"" << endl;

View File

@ -92,9 +92,10 @@ void test_track() {
}); });
shared_ptr<Track> t2; shared_ptr<Track> t2;
pc2->onTrack([&t2](shared_ptr<Track> t) { string newTrackMid;
pc2->onTrack([&t2, &newTrackMid](shared_ptr<Track> t) {
cout << "Track 2: Received with mid \"" << t->mid() << "\"" << endl; cout << "Track 2: Received with mid \"" << t->mid() << "\"" << endl;
if (t->mid() != "test") { if (t->mid() != newTrackMid) {
cerr << "Wrong track mid" << endl; cerr << "Wrong track mid" << endl;
return; return;
} }
@ -102,7 +103,9 @@ void test_track() {
std::atomic_store(&t2, t); std::atomic_store(&t2, t);
}); });
auto t1 = pc1->addTrack(Description::Video("test")); // Test opening a track
newTrackMid = "test";
auto t1 = pc1->addTrack(Description::Video(newTrackMid));
pc1->setLocalDescription(); pc1->setLocalDescription();
@ -118,6 +121,20 @@ void test_track() {
if (!at2 || !at2->isOpen() || !t1->isOpen()) if (!at2 || !at2->isOpen() || !t1->isOpen())
throw runtime_error("Track is not open"); throw runtime_error("Track is not open");
// Test renegotiation
newTrackMid = "added";
t1 = pc1->addTrack(Description::Video(newTrackMid));
pc1->setLocalDescription();
attempts = 10;
t2.reset();
while ((!(at2 = std::atomic_load(&t2)) || !at2->isOpen() || !t1->isOpen()) && attempts--)
this_thread::sleep_for(1s);
if (!at2 || !at2->isOpen() || !t1->isOpen())
throw runtime_error("Renegociated track is not open");
// TODO: Test sending RTP packets in track // TODO: Test sending RTP packets in track
// Delay close of peer 2 to check closing works properly // Delay close of peer 2 to check closing works properly