external/srs
1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-03-09 15:49:59 +00:00
Code Issues Releases Wiki Activity

RTC: Support ARQ for DTLS client

Browse source
This commit is contained in:
winlin 2020-08-13 16:21:01 +08:00
parent a5cefbf092
commit 92b59e308c
2 changed files with 159 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

34
trunk/src/app/srs_app_rtc_dtls.hpp
View file

@ -27,12 +27,15 @@
#include <srs_core.hpp>
#include <string>
class SrsRequest;
#include <vector>
#include <openssl/ssl.h>
#include <srtp2/srtp.h>
#include <srs_app_st.hpp>
class SrsRequest;
class SrsDtlsCertificate
{
private:
@ -92,7 +95,17 @@ public:
virtual srs_error_t write_dtls_data(void* data, int size) = 0;
};
class SrsDtls
// The state for DTLS client.
enum SrsDtlsState {
SrsDtlsStateInit,
SrsDtlsStateClientHello,
SrsDtlsStateServerHello,
SrsDtlsStateClientCertificate,
SrsDtlsStateServerDone,
SrsDtlsStateClientDone,
};
class SrsDtls : public ISrsCoroutineHandler
{
private:
SSL_CTX* dtls_ctx;
@ -110,6 +123,12 @@ private:
uint8_t* last_outgoing_packet_cache;
int nn_last_outgoing_packet;
// ARQ thread, for role active(DTLS client).
// @note If passive(DTLS server), the ARQ is driven by DTLS client.
SrsCoroutine* trd;
// The DTLS-client state to drive the ARQ thread.
SrsDtlsState client_state_;
// @remark: dtls_role_ default value is DTLS_SERVER.
SrsDtlsRole role_;
// @remark: dtls_version_ default value is SrsDtlsVersionAuto.
@ -130,7 +149,14 @@ public:
private:
srs_error_t do_on_dtls(char* data, int nb_data);
srs_error_t do_handshake();
void trace(uint8_t* data, int length, bool incoming, int ssl_err, bool cache);
// interface ISrsCoroutineHandler
public:
virtual srs_error_t cycle();
private:
void state_trace(uint8_t* data, int length, bool incoming, int ssl_err, bool cache, bool arq);
private:
srs_error_t start_arq();
void stop_arq();
public:
srs_error_t get_srtp_key(std::string& recv_key, std::string& send_key);
};