Bladeren bron

Merge branch 'v0.18'

Paul-Louis Ageneau 2 jaren geleden
bovenliggende
commit
d9a8dd4ca9
3 gewijzigde bestanden met toevoegingen van 40 en 27 verwijderingen
  1. 38 25
      include/rtc/nalunit.hpp
  2. 1 1
      src/impl/dtlstransport.cpp
  3. 1 1
      src/impl/tlstransport.cpp

+ 38 - 25
include/rtc/nalunit.hpp

@@ -23,9 +23,9 @@ namespace rtc {
 struct RTC_CPP_EXPORT NalUnitHeader {
 struct RTC_CPP_EXPORT NalUnitHeader {
 	uint8_t _first = 0;
 	uint8_t _first = 0;
 
 
-	bool forbiddenBit() { return _first >> 7; }
-	uint8_t nri() { return _first >> 5 & 0x03; }
-	uint8_t unitType() { return _first & 0x1F; }
+	bool forbiddenBit() const { return _first >> 7; }
+	uint8_t nri() const { return _first >> 5 & 0x03; }
+	uint8_t unitType() const { return _first & 0x1F; }
 
 
 	void setForbiddenBit(bool isSet) { _first = (_first & 0x7F) | (isSet << 7); }
 	void setForbiddenBit(bool isSet) { _first = (_first & 0x7F) | (isSet << 7); }
 	void setNRI(uint8_t nri) { _first = (_first & 0x9F) | ((nri & 0x03) << 5); }
 	void setNRI(uint8_t nri) { _first = (_first & 0x9F) | ((nri & 0x03) << 5); }
@@ -36,10 +36,10 @@ struct RTC_CPP_EXPORT NalUnitHeader {
 struct RTC_CPP_EXPORT NalUnitFragmentHeader {
 struct RTC_CPP_EXPORT NalUnitFragmentHeader {
 	uint8_t _first = 0;
 	uint8_t _first = 0;
 
 
-	bool isStart() { return _first >> 7; }
-	bool reservedBit6() { return (_first >> 5) & 0x01; }
-	bool isEnd() { return (_first >> 6) & 0x01; }
-	uint8_t unitType() { return _first & 0x1F; }
+	bool isStart() const { return _first >> 7; }
+	bool reservedBit6() const { return (_first >> 5) & 0x01; }
+	bool isEnd() const { return (_first >> 6) & 0x01; }
+	uint8_t unitType() const { return _first & 0x1F; }
 
 
 	void setStart(bool isSet) { _first = (_first & 0x7F) | (isSet << 7); }
 	void setStart(bool isSet) { _first = (_first & 0x7F) | (isSet << 7); }
 	void setEnd(bool isSet) { _first = (_first & 0xBF) | (isSet << 6); }
 	void setEnd(bool isSet) { _first = (_first & 0xBF) | (isSet << 6); }
@@ -53,17 +53,15 @@ struct RTC_CPP_EXPORT NalUnitFragmentHeader {
 struct RTC_CPP_EXPORT NalUnit : binary {
 struct RTC_CPP_EXPORT NalUnit : binary {
 	NalUnit(const NalUnit &unit) = default;
 	NalUnit(const NalUnit &unit) = default;
 	NalUnit(size_t size, bool includingHeader = true) : binary(size + (includingHeader ? 0 : 1)) {}
 	NalUnit(size_t size, bool includingHeader = true) : binary(size + (includingHeader ? 0 : 1)) {}
-
-	template <typename Iterator> NalUnit(Iterator begin_, Iterator end_) : binary(begin_, end_) {}
-
 	NalUnit(binary &&data) : binary(std::move(data)) {}
 	NalUnit(binary &&data) : binary(std::move(data)) {}
-
 	NalUnit() : binary(1) {}
 	NalUnit() : binary(1) {}
+	template <typename Iterator> NalUnit(Iterator begin_, Iterator end_) : binary(begin_, end_) {}
+
+	bool forbiddenBit() const { return header()->forbiddenBit(); }
+	uint8_t nri() const { return header()->nri(); }
+	uint8_t unitType() const { return header()->unitType(); }
 
 
-	bool forbiddenBit() { return header()->forbiddenBit(); }
-	uint8_t nri() { return header()->nri(); }
-	uint8_t unitType() { return header()->unitType(); }
-	binary payload() {
+	binary payload() const {
 		assert(size() >= 1);
 		assert(size() >= 1);
 		return {begin() + 1, end()};
 		return {begin() + 1, end()};
 	}
 	}
@@ -71,6 +69,7 @@ struct RTC_CPP_EXPORT NalUnit : binary {
 	void setForbiddenBit(bool isSet) { header()->setForbiddenBit(isSet); }
 	void setForbiddenBit(bool isSet) { header()->setForbiddenBit(isSet); }
 	void setNRI(uint8_t nri) { header()->setNRI(nri); }
 	void setNRI(uint8_t nri) { header()->setNRI(nri); }
 	void setUnitType(uint8_t type) { header()->setUnitType(type); }
 	void setUnitType(uint8_t type) { header()->setUnitType(type); }
+
 	void setPayload(binary payload) {
 	void setPayload(binary payload) {
 		assert(size() >= 1);
 		assert(size() >= 1);
 		erase(begin() + 1, end());
 		erase(begin() + 1, end());
@@ -78,30 +77,35 @@ struct RTC_CPP_EXPORT NalUnit : binary {
 	}
 	}
 
 
 protected:
 protected:
+	const NalUnitHeader *header() const {
+		assert(size() >= 1);
+		return reinterpret_cast<const NalUnitHeader *>(data());
+	}
+
 	NalUnitHeader *header() {
 	NalUnitHeader *header() {
 		assert(size() >= 1);
 		assert(size() >= 1);
-		return (NalUnitHeader *)data();
+		return reinterpret_cast<NalUnitHeader *>(data());
 	}
 	}
 };
 };
 
 
 /// Nal unit fragment A
 /// Nal unit fragment A
 struct RTC_CPP_EXPORT NalUnitFragmentA : NalUnit {
 struct RTC_CPP_EXPORT NalUnitFragmentA : NalUnit {
+	static std::vector<shared_ptr<NalUnitFragmentA>> fragmentsFrom(shared_ptr<NalUnit> nalu,
+	                                                               uint16_t maximumFragmentSize);
+
 	enum class FragmentType { Start, Middle, End };
 	enum class FragmentType { Start, Middle, End };
 
 
 	NalUnitFragmentA(FragmentType type, bool forbiddenBit, uint8_t nri, uint8_t unitType,
 	NalUnitFragmentA(FragmentType type, bool forbiddenBit, uint8_t nri, uint8_t unitType,
 	                 binary data);
 	                 binary data);
 
 
-	static std::vector<shared_ptr<NalUnitFragmentA>> fragmentsFrom(shared_ptr<NalUnit> nalu,
-	                                                               uint16_t maximumFragmentSize);
-
-	uint8_t unitType() { return fragmentHeader()->unitType(); }
+	uint8_t unitType() const { return fragmentHeader()->unitType(); }
 
 
-	binary payload() {
+	binary payload() const {
 		assert(size() >= 2);
 		assert(size() >= 2);
 		return {begin() + 2, end()};
 		return {begin() + 2, end()};
 	}
 	}
 
 
-	FragmentType type() {
+	FragmentType type() const {
 		if (fragmentHeader()->isStart()) {
 		if (fragmentHeader()->isStart()) {
 			return FragmentType::Start;
 			return FragmentType::Start;
 		} else if (fragmentHeader()->isEnd()) {
 		} else if (fragmentHeader()->isEnd()) {
@@ -122,19 +126,28 @@ struct RTC_CPP_EXPORT NalUnitFragmentA : NalUnit {
 	void setFragmentType(FragmentType type);
 	void setFragmentType(FragmentType type);
 
 
 protected:
 protected:
-	NalUnitHeader *fragmentIndicator() { return (NalUnitHeader *)data(); }
+	const uint8_t nal_type_fu_A = 28;
+
+	NalUnitHeader *fragmentIndicator() { return reinterpret_cast<NalUnitHeader *>(data()); }
+
+	const NalUnitHeader *fragmentIndicator() const {
+		return reinterpret_cast<const NalUnitHeader *>(data());
+	}
 
 
 	NalUnitFragmentHeader *fragmentHeader() {
 	NalUnitFragmentHeader *fragmentHeader() {
-		return (NalUnitFragmentHeader *)fragmentIndicator() + 1;
+		return reinterpret_cast<NalUnitFragmentHeader *>(fragmentIndicator() + 1);
 	}
 	}
 
 
-	const uint8_t nal_type_fu_A = 28;
+	const NalUnitFragmentHeader *fragmentHeader() const {
+		return reinterpret_cast<const NalUnitFragmentHeader *>(fragmentIndicator() + 1);
+	}
 };
 };
 
 
 class RTC_CPP_EXPORT NalUnits : public std::vector<shared_ptr<NalUnit>> {
 class RTC_CPP_EXPORT NalUnits : public std::vector<shared_ptr<NalUnit>> {
 public:
 public:
 	static const uint16_t defaultMaximumFragmentSize =
 	static const uint16_t defaultMaximumFragmentSize =
 	    uint16_t(RTC_DEFAULT_MTU - 12 - 8 - 40); // SRTP/UDP/IPv6
 	    uint16_t(RTC_DEFAULT_MTU - 12 - 8 - 40); // SRTP/UDP/IPv6
+
 	std::vector<shared_ptr<binary>> generateFragments(uint16_t maximumFragmentSize);
 	std::vector<shared_ptr<binary>> generateFragments(uint16_t maximumFragmentSize);
 };
 };
 
 

+ 1 - 1
src/impl/dtlstransport.cpp

@@ -261,7 +261,7 @@ void DtlsTransport::doRecv() {
 		PLOG_ERROR << "DTLS recv: " << e.what();
 		PLOG_ERROR << "DTLS recv: " << e.what();
 	}
 	}
 
 
-	gnutls_bye(mSession, GNUTLS_SHUT_RDWR);
+	gnutls_bye(mSession, GNUTLS_SHUT_WR);
 
 
 	PLOG_INFO << "DTLS closed";
 	PLOG_INFO << "DTLS closed";
 	changeState(State::Disconnected);
 	changeState(State::Disconnected);

+ 1 - 1
src/impl/tlstransport.cpp

@@ -208,7 +208,7 @@ void TlsTransport::doRecv() {
 		PLOG_ERROR << "TLS recv: " << e.what();
 		PLOG_ERROR << "TLS recv: " << e.what();
 	}
 	}
 
 
-	gnutls_bye(mSession, GNUTLS_SHUT_RDWR);
+	gnutls_bye(mSession, GNUTLS_SHUT_WR);
 
 
 	PLOG_INFO << "TLS closed";
 	PLOG_INFO << "TLS closed";
 	changeState(State::Disconnected);
 	changeState(State::Disconnected);