diff options
author | jiayl <jiayl@chromium.org> | 2014-09-23 15:44:43 -0700 |
---|---|---|
committer | Commit bot <commit-bot@chromium.org> | 2014-09-23 22:44:56 +0000 |
commit | 08f0eac49f32b8d24c82bd5cc884938eb494f85e (patch) | |
tree | 90ec394b18fa0ba3bc4898f4bfeba1aff7f75522 /content/browser | |
parent | 85b87978351f4449aba1aa68f459342600fafdfc (diff) | |
download | chromium_src-08f0eac49f32b8d24c82bd5cc884938eb494f85e.zip chromium_src-08f0eac49f32b8d24c82bd5cc884938eb494f85e.tar.gz chromium_src-08f0eac49f32b8d24c82bd5cc884938eb494f85e.tar.bz2 |
Fix boundary check problems in socket_host.cc.
See the bug for the full list of security issues fixed.
BUG=416528
Review URL: https://codereview.chromium.org/589183002
Cr-Commit-Position: refs/heads/master@{#296287}
Diffstat (limited to 'content/browser')
-rw-r--r-- | content/browser/renderer_host/p2p/socket_host.cc | 267 | ||||
-rw-r--r-- | content/browser/renderer_host/p2p/socket_host.h | 20 | ||||
-rw-r--r-- | content/browser/renderer_host/p2p/socket_host_unittest.cc | 96 |
3 files changed, 228 insertions, 155 deletions
diff --git a/content/browser/renderer_host/p2p/socket_host.cc b/content/browser/renderer_host/p2p/socket_host.cc index fd26298..1693918 100644 --- a/content/browser/renderer_host/p2p/socket_host.cc +++ b/content/browser/renderer_host/p2p/socket_host.cc @@ -19,12 +19,14 @@ namespace { const uint32 kStunMagicCookie = 0x2112A442; -const int kMinRtpHdrLen = 12; -const int kRtpExtnHdrLen = 4; -const int kDtlsRecordHeaderLen = 13; -const int kTurnChannelHdrLen = 4; -const int kAbsSendTimeExtnLen = 3; -const int kOneByteHdrLen = 1; +const size_t kMinRtpHeaderLength = 12; +const size_t kMinRtcpHeaderLength = 8; +const size_t kRtpExtensionHeaderLength = 4; +const size_t kDtlsRecordHeaderLength = 13; +const size_t kTurnChannelHeaderLength = 4; +const size_t kAbsSendTimeExtensionLength = 3; +const size_t kOneByteHeaderLength = 1; +const size_t kMaxRtpPacketLength = 2048; // Fake auth tag written by the render process if external authentication is // enabled. HMAC in packet will be compared against this value before updating @@ -33,37 +35,50 @@ static const unsigned char kFakeAuthTag[10] = { 0xba, 0xdd, 0xba, 0xdd, 0xba, 0xdd, 0xba, 0xdd, 0xba, 0xdd }; -bool IsTurnChannelData(const char* data) { - return ((*data & 0xC0) == 0x40); +bool IsTurnChannelData(const char* data, size_t length) { + return length >= kTurnChannelHeaderLength && ((*data & 0xC0) == 0x40); } -bool IsDtlsPacket(const char* data, int len) { +bool IsDtlsPacket(const char* data, size_t length) { const uint8* u = reinterpret_cast<const uint8*>(data); - return (len >= kDtlsRecordHeaderLen && (u[0] > 19 && u[0] < 64)); + return (length >= kDtlsRecordHeaderLength && (u[0] > 19 && u[0] < 64)); } -bool IsRtcpPacket(const char* data) { +bool IsRtcpPacket(const char* data, size_t length) { + if (length < kMinRtcpHeaderLength) { + return false; + } + int type = (static_cast<uint8>(data[1]) & 0x7F); return (type >= 64 && type < 96); } -bool IsTurnSendIndicationPacket(const char* data) { +bool IsTurnSendIndicationPacket(const char* data, size_t length) { + if (length < content::P2PSocketHost::kStunHeaderSize) { + return false; + } + uint16 type = rtc::GetBE16(data); return (type == cricket::TURN_SEND_INDICATION); } -bool IsRtpPacket(const char* data, int len) { - return ((*data & 0xC0) == 0x80); +bool IsRtpPacket(const char* data, size_t length) { + return (length >= kMinRtpHeaderLength) && ((*data & 0xC0) == 0x80); } // Verifies rtp header and message length. -bool ValidateRtpHeader(const char* rtp, int length, size_t* header_length) { - if (header_length) +bool ValidateRtpHeader(const char* rtp, size_t length, size_t* header_length) { + if (header_length) { *header_length = 0; + } - int cc_count = rtp[0] & 0x0F; - int rtp_hdr_len_without_extn = kMinRtpHdrLen + 4 * cc_count; - if (rtp_hdr_len_without_extn > length) { + if (length < kMinRtpHeaderLength) { + return false; + } + + size_t cc_count = rtp[0] & 0x0F; + size_t header_length_without_extension = kMinRtpHeaderLength + 4 * cc_count; + if (header_length_without_extension > length) { return false; } @@ -71,29 +86,40 @@ bool ValidateRtpHeader(const char* rtp, int length, size_t* header_length) { // length is verified above. if (!(rtp[0] & 0x10)) { if (header_length) - *header_length = rtp_hdr_len_without_extn; + *header_length = header_length_without_extension; return true; } - rtp += rtp_hdr_len_without_extn; + rtp += header_length_without_extension; + + if (header_length_without_extension + kRtpExtensionHeaderLength > length) { + return false; + } // Getting extension profile length. // Length is in 32 bit words. - uint16 extn_length = rtc::GetBE16(rtp + 2) * 4; + uint16 extension_length_in_32bits = rtc::GetBE16(rtp + 2); + size_t extension_length = extension_length_in_32bits * 4; + + size_t rtp_header_length = extension_length + + header_length_without_extension + + kRtpExtensionHeaderLength; // Verify input length against total header size. - if (rtp_hdr_len_without_extn + kRtpExtnHdrLen + extn_length > length) { + if (rtp_header_length > length) { return false; } - if (header_length) - *header_length = rtp_hdr_len_without_extn + kRtpExtnHdrLen + extn_length; + if (header_length) { + *header_length = rtp_header_length; + } return true; } -void UpdateAbsSendTimeExtnValue(char* extn_data, int len, - uint32 abs_send_time) { +void UpdateAbsSendTimeExtensionValue(char* extension_data, + size_t length, + uint32 abs_send_time) { // Absolute send time in RTP streams. // // The absolute send time is signaled to the receiver in-band using the @@ -109,7 +135,11 @@ void UpdateAbsSendTimeExtnValue(char* extn_data, int len, // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | ID | len=2 | absolute send time | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ - DCHECK_EQ(len, kAbsSendTimeExtnLen); + if (length != kAbsSendTimeExtensionLength) { + NOTREACHED(); + return; + } + // Now() has resolution ~1-15ms, using HighResNow(). But it is warned not to // use it unless necessary, as it is expensive than Now(). uint32 now_second = abs_send_time; @@ -121,24 +151,29 @@ void UpdateAbsSendTimeExtnValue(char* extn_data, int len, ((now_us << 18) / base::Time::kMicrosecondsPerSecond) & 0x00FFFFFF; } // TODO(mallinath) - Add SetBE24 to byteorder.h in libjingle. - extn_data[0] = static_cast<uint8>(now_second >> 16); - extn_data[1] = static_cast<uint8>(now_second >> 8); - extn_data[2] = static_cast<uint8>(now_second); + extension_data[0] = static_cast<uint8>(now_second >> 16); + extension_data[1] = static_cast<uint8>(now_second >> 8); + extension_data[2] = static_cast<uint8>(now_second); } -// Assumes |len| is actual packet length + tag length. Updates HMAC at end of +// Assumes |length| is actual packet length + tag length. Updates HMAC at end of // the RTP packet. -void UpdateRtpAuthTag(char* rtp, int len, +void UpdateRtpAuthTag(char* rtp, + size_t length, const rtc::PacketOptions& options) { // If there is no key, return. - if (options.packet_time_params.srtp_auth_key.empty()) + if (options.packet_time_params.srtp_auth_key.empty()) { return; + } size_t tag_length = options.packet_time_params.srtp_auth_tag_len; - char* auth_tag = rtp + (len - tag_length); - // We should have a fake HMAC value @ auth_tag. - DCHECK_EQ(0, memcmp(auth_tag, kFakeAuthTag, tag_length)); + // ROC (rollover counter) is at the beginning of the auth tag. + const size_t kRocLength = 4; + if (tag_length < kRocLength || tag_length > length) { + NOTREACHED(); + return; + } crypto::HMAC hmac(crypto::HMAC::SHA1); if (!hmac.Init(reinterpret_cast<const unsigned char*>( @@ -148,15 +183,20 @@ void UpdateRtpAuthTag(char* rtp, int len, return; } - if (hmac.DigestLength() < tag_length) { + if (tag_length > hmac.DigestLength()) { NOTREACHED(); return; } + char* auth_tag = rtp + (length - tag_length); + + // We should have a fake HMAC value @ auth_tag. + DCHECK_EQ(0, memcmp(auth_tag, kFakeAuthTag, tag_length)); + // Copy ROC after end of rtp packet. - memcpy(auth_tag, &options.packet_time_params.srtp_packet_index, 4); + memcpy(auth_tag, &options.packet_time_params.srtp_packet_index, kRocLength); // Authentication of a RTP packet will have RTP packet + ROC size. - int auth_required_length = len - tag_length + 4; + int auth_required_length = length - tag_length + kRocLength; unsigned char output[64]; if (!hmac.Sign(base::StringPiece(rtp, auth_required_length), @@ -175,7 +215,8 @@ namespace content { namespace packet_processing_helpers { -bool ApplyPacketOptions(char* data, int length, +bool ApplyPacketOptions(char* data, + size_t length, const rtc::PacketOptions& options, uint32 abs_send_time) { DCHECK(data != NULL); @@ -188,13 +229,13 @@ bool ApplyPacketOptions(char* data, int length, } DCHECK(!IsDtlsPacket(data, length)); - DCHECK(!IsRtcpPacket(data)); + DCHECK(!IsRtcpPacket(data, length)); // If there is a srtp auth key present then packet must be a RTP packet. // RTP packet may have been wrapped in a TURN Channel Data or // TURN send indication. - int rtp_start_pos; - int rtp_length; + size_t rtp_start_pos; + size_t rtp_length; if (!GetRtpPacketStartPositionAndLength( data, length, &rtp_start_pos, &rtp_length)) { // This method should never return false. @@ -208,9 +249,11 @@ bool ApplyPacketOptions(char* data, int length, // then we should parse the rtp packet to update the timestamp. Otherwise // just calculate HMAC and update packet with it. if (options.packet_time_params.rtp_sendtime_extension_id != -1) { - UpdateRtpAbsSendTimeExtn( - start, rtp_length, - options.packet_time_params.rtp_sendtime_extension_id, abs_send_time); + UpdateRtpAbsSendTimeExtension( + start, + rtp_length, + options.packet_time_params.rtp_sendtime_extension_id, + abs_send_time); } UpdateRtpAuthTag(start, rtp_length, options); @@ -218,12 +261,16 @@ bool ApplyPacketOptions(char* data, int length, } bool GetRtpPacketStartPositionAndLength(const char* packet, - int length, - int* rtp_start_pos, - int* rtp_packet_length) { - int rtp_begin; - int rtp_length = 0; - if (IsTurnChannelData(packet)) { + size_t length, + size_t* rtp_start_pos, + size_t* rtp_packet_length) { + if (length < kMinRtpHeaderLength || length > kMaxRtpPacketLength) { + return false; + } + + size_t rtp_begin; + size_t rtp_length = 0; + if (IsTurnChannelData(packet, length)) { // Turn Channel Message header format. // 0 1 2 3 // 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 @@ -234,32 +281,23 @@ bool GetRtpPacketStartPositionAndLength(const char* packet, // / Application Data / // / / // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ - if (length < kTurnChannelHdrLen) { - return false; - } - - rtp_begin = kTurnChannelHdrLen; + rtp_begin = kTurnChannelHeaderLength; rtp_length = rtc::GetBE16(&packet[2]); - if (length < rtp_length + kTurnChannelHdrLen) { - return false; - } - } else if (IsTurnSendIndicationPacket(packet)) { - if (length <= P2PSocketHost::kStunHeaderSize) { - // Message must be greater than 20 bytes, if it's carrying any payload. + if (length < rtp_length + kTurnChannelHeaderLength) { return false; } + } else if (IsTurnSendIndicationPacket(packet, length)) { // Validate STUN message length. - int stun_msg_len = rtc::GetBE16(&packet[2]); - if (stun_msg_len + P2PSocketHost::kStunHeaderSize != length) { + const size_t stun_message_length = rtc::GetBE16(&packet[2]); + if (stun_message_length + P2PSocketHost::kStunHeaderSize != length) { return false; } // First skip mandatory stun header which is of 20 bytes. rtp_begin = P2PSocketHost::kStunHeaderSize; // Loop through STUN attributes until we find STUN DATA attribute. - const char* start = packet + rtp_begin; bool data_attr_present = false; - while ((packet + rtp_begin) - start < stun_msg_len) { + while (rtp_begin < length) { // Keep reading STUN attributes until we hit DATA attribute. // Attribute will be a TLV structure. // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ @@ -274,17 +312,26 @@ bool GetRtpPacketStartPositionAndLength(const char* packet, // padding so that its value contains a multiple of 4 bytes. The // padding bits are ignored, and may be any value. uint16 attr_type, attr_length; + const int kAttrHeaderLength = sizeof(attr_type) + sizeof(attr_length); + + if (length < rtp_begin + kAttrHeaderLength) { + return false; + } + // Getting attribute type and length. attr_type = rtc::GetBE16(&packet[rtp_begin]); attr_length = rtc::GetBE16( &packet[rtp_begin + sizeof(attr_type)]); + + rtp_begin += kAttrHeaderLength; // Skip STUN_DATA_ATTR header. + // Checking for bogus attribute length. - if (length < attr_length + rtp_begin) { + if (length < rtp_begin + attr_length) { return false; } if (attr_type != cricket::STUN_ATTR_DATA) { - rtp_begin += sizeof(attr_type) + sizeof(attr_length) + attr_length; + rtp_begin += attr_length; if ((attr_length % 4) != 0) { rtp_begin += (4 - (attr_length % 4)); } @@ -292,12 +339,8 @@ bool GetRtpPacketStartPositionAndLength(const char* packet, } data_attr_present = true; - rtp_begin += 4; // Skip STUN_DATA_ATTR header. rtp_length = attr_length; - // One final check of length before exiting. - if (length < rtp_length + rtp_begin) { - return false; - } + // We found STUN_DATA_ATTR. We can skip parsing rest of the packet. break; } @@ -315,8 +358,7 @@ bool GetRtpPacketStartPositionAndLength(const char* packet, } // Making sure we have a valid RTP packet at the end. - if ((rtp_length >= kMinRtpHdrLen) && - IsRtpPacket(packet + rtp_begin, rtp_length) && + if (IsRtpPacket(packet + rtp_begin, rtp_length) && ValidateRtpHeader(packet + rtp_begin, rtp_length, NULL)) { *rtp_start_pos = rtp_begin; *rtp_packet_length = rtp_length; @@ -327,8 +369,10 @@ bool GetRtpPacketStartPositionAndLength(const char* packet, // ValidateRtpHeader must be called before this method to make sure, we have // a sane rtp packet. -bool UpdateRtpAbsSendTimeExtn(char* rtp, int length, - int extension_id, uint32 abs_send_time) { +bool UpdateRtpAbsSendTimeExtension(char* rtp, + size_t length, + int extension_id, + uint32 abs_send_time) { // 0 1 2 3 // 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ @@ -347,17 +391,18 @@ bool UpdateRtpAbsSendTimeExtn(char* rtp, int length, return true; } - int cc_count = rtp[0] & 0x0F; - int rtp_hdr_len_without_extn = kMinRtpHdrLen + 4 * cc_count; + size_t cc_count = rtp[0] & 0x0F; + size_t header_length_without_extension = kMinRtpHeaderLength + 4 * cc_count; - rtp += rtp_hdr_len_without_extn; + rtp += header_length_without_extension; // Getting extension profile ID and length. uint16 profile_id = rtc::GetBE16(rtp); // Length is in 32 bit words. - uint16 extn_length = rtc::GetBE16(rtp + 2) * 4; + uint16 extension_length_in_32bits = rtc::GetBE16(rtp + 2); + size_t extension_length = extension_length_in_32bits * 4; - rtp += kRtpExtnHdrLen; // Moving past extn header. + rtp += kRtpExtensionHeaderLength; // Moving past extension header. bool found = false; // WebRTC is using one byte header extension. @@ -366,7 +411,7 @@ bool UpdateRtpAbsSendTimeExtn(char* rtp, int length, // 0 // 0 1 2 3 4 5 6 7 // +-+-+-+-+-+-+-+-+ - // | ID | len | + // | ID |length | // +-+-+-+-+-+-+-+-+ // 0 1 2 3 @@ -380,20 +425,26 @@ bool UpdateRtpAbsSendTimeExtn(char* rtp, int length, // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | data | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ - char* extn_start = rtp; - while (rtp - extn_start < extn_length) { + const char* extension_start = rtp; + const char* extension_end = extension_start + extension_length; + + while (rtp < extension_end) { const int id = (*rtp & 0xF0) >> 4; - const int len = (*rtp & 0x0F) + 1; + const size_t length = (*rtp & 0x0F) + 1; + if (rtp + kOneByteHeaderLength + length > extension_end) { + return false; + } // The 4-bit length is the number minus one of data bytes of this header // extension element following the one-byte header. if (id == extension_id) { - UpdateAbsSendTimeExtnValue(rtp + kOneByteHdrLen, len, abs_send_time); + UpdateAbsSendTimeExtensionValue( + rtp + kOneByteHeaderLength, length, abs_send_time); found = true; break; } - rtp += kOneByteHdrLen + len; + rtp += kOneByteHeaderLength + length; // Counting padding bytes. - while ((*rtp == 0) && (rtp - extn_start < extn_length)) { + while ((rtp < extension_end) && (*rtp == 0)) { ++rtp; } } @@ -419,16 +470,19 @@ P2PSocketHost::~P2PSocketHost() { } bool P2PSocketHost::GetStunPacketType( const char* data, int data_size, StunMessageType* type) { - if (data_size < kStunHeaderSize) + if (data_size < kStunHeaderSize) { return false; + } uint32 cookie = base::NetToHost32(*reinterpret_cast<const uint32*>(data + 4)); - if (cookie != kStunMagicCookie) + if (cookie != kStunMagicCookie) { return false; + } uint16 length = base::NetToHost16(*reinterpret_cast<const uint16*>(data + 2)); - if (length != data_size - kStunHeaderSize) + if (length != data_size - kStunHeaderSize) { return false; + } int message_type = base::NetToHost16(*reinterpret_cast<const uint16*>(data)); @@ -502,11 +556,13 @@ void P2PSocketHost::StartRtpDump( DCHECK(!packet_callback.is_null()); DCHECK(incoming || outgoing); - if (incoming) + if (incoming) { dump_incoming_rtp_packet_ = true; + } - if (outgoing) + if (outgoing) { dump_outgoing_rtp_packet_ = true; + } packet_dump_callback_ = packet_callback; } @@ -515,27 +571,32 @@ void P2PSocketHost::StopRtpDump(bool incoming, bool outgoing) { DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); DCHECK(incoming || outgoing); - if (incoming) + if (incoming) { dump_incoming_rtp_packet_ = false; + } - if (outgoing) + if (outgoing) { dump_outgoing_rtp_packet_ = false; + } - if (!dump_incoming_rtp_packet_ && !dump_outgoing_rtp_packet_) + if (!dump_incoming_rtp_packet_ && !dump_outgoing_rtp_packet_) { packet_dump_callback_.Reset(); + } } void P2PSocketHost::DumpRtpPacket(const char* packet, size_t length, bool incoming) { - if (IsDtlsPacket(packet, length) || IsRtcpPacket(packet)) + if (IsDtlsPacket(packet, length) || IsRtcpPacket(packet, length)) { return; + } - int rtp_packet_pos = 0; - int rtp_packet_length = length; + size_t rtp_packet_pos = 0; + size_t rtp_packet_length = length; if (!packet_processing_helpers::GetRtpPacketStartPositionAndLength( - packet, length, &rtp_packet_pos, &rtp_packet_length)) + packet, length, &rtp_packet_pos, &rtp_packet_length)) { return; + } packet += rtp_packet_pos; diff --git a/content/browser/renderer_host/p2p/socket_host.h b/content/browser/renderer_host/p2p/socket_host.h index 86eff5d..8a12898 100644 --- a/content/browser/renderer_host/p2p/socket_host.h +++ b/content/browser/renderer_host/p2p/socket_host.h @@ -33,20 +33,24 @@ namespace packet_processing_helpers { // called. It will try to do, 1. update absolute send time extension header // if present with current time and 2. update HMAC in RTP packet. // If abs_send_time is 0, ApplyPacketOption will get current time from system. -CONTENT_EXPORT bool ApplyPacketOptions(char* data, int length, +CONTENT_EXPORT bool ApplyPacketOptions(char* data, + size_t length, const rtc::PacketOptions& options, uint32 abs_send_time); // Helper method which finds RTP ofset and length if the packet is encapsulated // in a TURN Channel Message or TURN Send Indication message. -CONTENT_EXPORT bool GetRtpPacketStartPositionAndLength(const char* data, - int length, - int* rtp_start_pos, - int* rtp_packet_length); +CONTENT_EXPORT bool GetRtpPacketStartPositionAndLength( + const char* data, + size_t length, + size_t* rtp_start_pos, + size_t* rtp_packet_length); + // Helper method which updates absoulute send time extension if present. -CONTENT_EXPORT bool UpdateRtpAbsSendTimeExtn(char* rtp, int length, - int extension_id, - uint32 abs_send_time); +CONTENT_EXPORT bool UpdateRtpAbsSendTimeExtension(char* rtp, + size_t length, + int extension_id, + uint32 abs_send_time); } // packet_processing_helpers diff --git a/content/browser/renderer_host/p2p/socket_host_unittest.cc b/content/browser/renderer_host/p2p/socket_host_unittest.cc index fc96e4a..7e9393f 100644 --- a/content/browser/renderer_host/p2p/socket_host_unittest.cc +++ b/content/browser/renderer_host/p2p/socket_host_unittest.cc @@ -152,145 +152,151 @@ namespace content { // This test verifies parsing of all invalid raw packets. TEST(P2PSocketHostTest, TestInvalidRawRtpMessages) { - int start_pos = INT_MAX, rtp_length = INT_MAX; + size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX; EXPECT_FALSE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kRtpMsgWithInvalidLength), sizeof(kRtpMsgWithInvalidLength), &start_pos, &rtp_length)); - EXPECT_EQ(INT_MAX, start_pos); - EXPECT_EQ(INT_MAX, rtp_length); + EXPECT_EQ(SIZE_MAX, start_pos); + EXPECT_EQ(SIZE_MAX, rtp_length); EXPECT_FALSE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kRtpMsgWithInvalidExtnLength), sizeof(kRtpMsgWithInvalidExtnLength), &start_pos, &rtp_length)); - EXPECT_EQ(INT_MAX, start_pos); - EXPECT_EQ(INT_MAX, rtp_length); + EXPECT_EQ(SIZE_MAX, start_pos); + EXPECT_EQ(SIZE_MAX, rtp_length); } // Verify invalid TURN send indication messages. Messages are proper STUN // messages with incorrect values in attributes. TEST(P2PSocketHostTest, TestInvalidTurnSendIndicationMessages) { // Initializing out params to very large value. - int start_pos = INT_MAX, rtp_length = INT_MAX; + size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX; EXPECT_FALSE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnSendIndicationMsgWithNoAttributes), sizeof(kTurnSendIndicationMsgWithNoAttributes), &start_pos, &rtp_length)); - EXPECT_EQ(INT_MAX, start_pos); - EXPECT_EQ(INT_MAX, rtp_length); + EXPECT_EQ(SIZE_MAX, start_pos); + EXPECT_EQ(SIZE_MAX, rtp_length); EXPECT_FALSE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnSendIndicationMsgWithInvalidLength), sizeof(kTurnSendIndicationMsgWithInvalidLength), &start_pos, &rtp_length)); - EXPECT_EQ(INT_MAX, start_pos); - EXPECT_EQ(INT_MAX, rtp_length); + EXPECT_EQ(SIZE_MAX, start_pos); + EXPECT_EQ(SIZE_MAX, rtp_length); EXPECT_FALSE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnSendIndicatinMsgWithNoDataAttribute), sizeof(kTurnSendIndicatinMsgWithNoDataAttribute), &start_pos, &rtp_length)); - EXPECT_EQ(INT_MAX, start_pos); - EXPECT_EQ(INT_MAX, rtp_length); + EXPECT_EQ(SIZE_MAX, start_pos); + EXPECT_EQ(SIZE_MAX, rtp_length); } // This test verifies incorrectly formed TURN channel messages. TEST(P2PSocketHostTest, TestInvalidTurnChannelMessages) { - int start_pos = INT_MAX, rtp_length = INT_MAX; + size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX; EXPECT_FALSE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnChannelMsgNoRtpPacket), sizeof(kTurnChannelMsgNoRtpPacket), &start_pos, &rtp_length)); - EXPECT_EQ(INT_MAX, start_pos); - EXPECT_EQ(INT_MAX, rtp_length); + EXPECT_EQ(SIZE_MAX, start_pos); + EXPECT_EQ(SIZE_MAX, rtp_length); EXPECT_FALSE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnChannelMsgWithZeroLength), sizeof(kTurnChannelMsgWithZeroLength), &start_pos, &rtp_length)); - EXPECT_EQ(INT_MAX, start_pos); - EXPECT_EQ(INT_MAX, rtp_length); + EXPECT_EQ(SIZE_MAX, start_pos); + EXPECT_EQ(SIZE_MAX, rtp_length); } // This test verifies parsing of a valid RTP packet which has 2byte header // extension instead of a 1 byte header extension. TEST(P2PSocketHostTest, TestValid2ByteExtnHdrRtpMessage) { - int start_pos = INT_MAX, rtp_length = INT_MAX; + size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX; EXPECT_TRUE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kRtpMsgWith2ByteExtnHeader), sizeof(kRtpMsgWith2ByteExtnHeader), &start_pos, &rtp_length)); - EXPECT_EQ(20, rtp_length); - EXPECT_EQ(0, start_pos); + EXPECT_EQ(20U, rtp_length); + EXPECT_EQ(0U, start_pos); } // This test verifies parsing of a valid RTP packet which has 1 byte header // AbsSendTime extension in it. TEST(P2PSocketHostTest, TestValidRtpPacketWithAbsSendTimeExtension) { - int start_pos = INT_MAX, rtp_length = INT_MAX; + size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX; EXPECT_TRUE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kRtpMsgWithAbsSendTimeExtension), sizeof(kRtpMsgWithAbsSendTimeExtension), &start_pos, &rtp_length)); - EXPECT_EQ(24, rtp_length); - EXPECT_EQ(0, start_pos); + EXPECT_EQ(24U, rtp_length); + EXPECT_EQ(0U, start_pos); } // This test verifies parsing of a valid TURN Send Indication messages. TEST(P2PSocketHostTest, TestValidTurnSendIndicationMessages) { - int start_pos = INT_MAX, rtp_length = INT_MAX; + size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX; EXPECT_TRUE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnSendIndicationMsgWithoutRtpExtension), sizeof(kTurnSendIndicationMsgWithoutRtpExtension), &start_pos, &rtp_length)); - EXPECT_EQ(12, rtp_length); - EXPECT_EQ(32, start_pos); + EXPECT_EQ(12U, rtp_length); + EXPECT_EQ(32U, start_pos); - start_pos = INT_MAX, rtp_length = INT_MAX; + start_pos = SIZE_MAX, rtp_length = SIZE_MAX; EXPECT_TRUE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnSendIndicationMsgWithAbsSendTimeExtension), sizeof(kTurnSendIndicationMsgWithAbsSendTimeExtension), &start_pos, &rtp_length)); - EXPECT_EQ(24, rtp_length); - EXPECT_EQ(32, start_pos); + EXPECT_EQ(24U, rtp_length); + EXPECT_EQ(32U, start_pos); } // This test verifies parsing of valid TURN Channel Messages. TEST(P2PSocketHostTest, TestValidTurnChannelMessages) { - int start_pos = -1, rtp_length = -1; + size_t start_pos = 0, rtp_length = 0; EXPECT_TRUE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnChannelMsgWithRtpPacket), sizeof(kTurnChannelMsgWithRtpPacket), &start_pos, &rtp_length)); - EXPECT_EQ(12, rtp_length); - EXPECT_EQ(4, start_pos); + EXPECT_EQ(12U, rtp_length); + EXPECT_EQ(4U, start_pos); - start_pos = -1, rtp_length = -1; + start_pos = 0, rtp_length = 0; EXPECT_TRUE(packet_processing_helpers::GetRtpPacketStartPositionAndLength( reinterpret_cast<char*>(kTurnChannelMsgWithAbsSendTimeExtension), sizeof(kTurnChannelMsgWithAbsSendTimeExtension), &start_pos, &rtp_length)); - EXPECT_EQ(20, rtp_length); - EXPECT_EQ(4, start_pos); + EXPECT_EQ(20U, rtp_length); + EXPECT_EQ(4U, start_pos); } // Verify handling of a 2 byte extension header RTP messsage. Currently we don't // handle this kind of message. TEST(P2PSocketHostTest, TestUpdateAbsSendTimeExtensionIn2ByteHeaderExtn) { - EXPECT_FALSE(packet_processing_helpers::UpdateRtpAbsSendTimeExtn( - reinterpret_cast<char*>(kRtpMsgWith2ByteExtnHeader), - sizeof(kRtpMsgWith2ByteExtnHeader), 3, 0)); + EXPECT_FALSE(packet_processing_helpers::UpdateRtpAbsSendTimeExtension( + reinterpret_cast<char*>(kRtpMsgWith2ByteExtnHeader), + sizeof(kRtpMsgWith2ByteExtnHeader), + 3, + 0)); } // Verify finding an extension ID in the TURN send indication message. TEST(P2PSocketHostTest, TestUpdateAbsSendTimeExtensionInTurnSendIndication) { - EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtn( + EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtension( reinterpret_cast<char*>(kTurnSendIndicationMsgWithoutRtpExtension), - sizeof(kTurnSendIndicationMsgWithoutRtpExtension), 3, 0)); + sizeof(kTurnSendIndicationMsgWithoutRtpExtension), + 3, + 0)); - EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtn( + EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtension( reinterpret_cast<char*>(kTurnSendIndicationMsgWithAbsSendTimeExtension), - sizeof(kTurnSendIndicationMsgWithAbsSendTimeExtension), 3, 0)); + sizeof(kTurnSendIndicationMsgWithAbsSendTimeExtension), + 3, + 0)); } // Test without any packet options variables set. This method should return @@ -341,9 +347,11 @@ TEST(P2PSocketHostTest, TestApplyPacketOptionsWithAuthParams) { // Verify finding an extension ID in a raw rtp message. TEST(P2PSocketHostTest, TestUpdateAbsSendTimeExtensionInRtpPacket) { - EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtn( + EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtension( reinterpret_cast<char*>(kRtpMsgWithAbsSendTimeExtension), - sizeof(kRtpMsgWithAbsSendTimeExtension), 3, 0)); + sizeof(kRtpMsgWithAbsSendTimeExtension), + 3, + 0)); } // Verify we update both AbsSendTime extension header and HMAC. |