| // Copyright (c) 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "content/browser/renderer_host/p2p/socket_host.h" |
| |
| #include <vector> |
| |
| #include "base/memory/scoped_ptr.h" |
| #include "content/browser/renderer_host/p2p/socket_host_test_utils.h" |
| #include "net/base/ip_endpoint.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| static unsigned char kFakeTag[4] = { 0xba, 0xdd, 0xba, 0xdd }; |
| static unsigned char kTestKey[] = "12345678901234567890"; |
| static unsigned char kTestAstValue[3] = { 0xaa, 0xbb, 0xcc }; |
| |
| // Rtp message with invalid length. |
| static unsigned char kRtpMsgWithInvalidLength[] = { |
| 0x94, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| 0xAA, 0xBB, 0xCC, 0XDD, // SSRC |
| 0xDD, 0xCC, 0xBB, 0xAA // Only 1 CSRC, but CC count is 4. |
| }; |
| |
| // Rtp message with single byte header extension, invalid extension length. |
| static unsigned char kRtpMsgWithInvalidExtnLength[] = { |
| 0x90, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| 0xBE, 0xDE, 0x0A, 0x00 // Extn length - 0x0A00 |
| }; |
| |
| // Valid rtp Message with 2 byte header extension. |
| static unsigned char kRtpMsgWith2ByteExtnHeader[] = { |
| 0x90, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| 0xAA, 0xBB, 0xCC, 0XDD, // SSRC |
| 0x10, 0x00, 0x00, 0x01, // 2 Byte header extension |
| 0x01, 0x00, 0x00, 0x00 |
| }; |
| |
| // Stun Indication message with Zero length |
| static unsigned char kTurnSendIndicationMsgWithNoAttributes[] = { |
| 0x00, 0x16, 0x00, 0x00, // Zero length |
| 0x21, 0x12, 0xA4, 0x42, // magic cookie |
| '0', '1', '2', '3', // transaction id |
| '4', '5', '6', '7', |
| '8', '9', 'a', 'b', |
| }; |
| |
| // Stun Send Indication message with invalid length in stun header. |
| static unsigned char kTurnSendIndicationMsgWithInvalidLength[] = { |
| 0x00, 0x16, 0xFF, 0x00, // length of 0xFF00 |
| 0x21, 0x12, 0xA4, 0x42, // magic cookie |
| '0', '1', '2', '3', // transaction id |
| '4', '5', '6', '7', |
| '8', '9', 'a', 'b', |
| }; |
| |
| // Stun Send Indication message with no DATA attribute in message. |
| static unsigned char kTurnSendIndicatinMsgWithNoDataAttribute[] = { |
| 0x00, 0x16, 0x00, 0x08, // length of |
| 0x21, 0x12, 0xA4, 0x42, // magic cookie |
| '0', '1', '2', '3', // transaction id |
| '4', '5', '6', '7', |
| '8', '9', 'a', 'b', |
| 0x00, 0x20, 0x00, 0x04, // Mapped address. |
| 0x00, 0x00, 0x00, 0x00 |
| }; |
| |
| // A valid STUN indication message with a valid RTP header in data attribute |
| // payload field and no extension bit set. |
| static unsigned char kTurnSendIndicationMsgWithoutRtpExtension[] = { |
| 0x00, 0x16, 0x00, 0x18, // length of |
| 0x21, 0x12, 0xA4, 0x42, // magic cookie |
| '0', '1', '2', '3', // transaction id |
| '4', '5', '6', '7', |
| '8', '9', 'a', 'b', |
| 0x00, 0x20, 0x00, 0x04, // Mapped address. |
| 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x13, 0x00, 0x0C, // Data attribute. |
| 0x80, 0x00, 0x00, 0x00, // RTP packet. |
| 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| }; |
| |
| // A valid STUN indication message with a valid RTP header and a extension |
| // header. |
| static unsigned char kTurnSendIndicationMsgWithAbsSendTimeExtension[] = { |
| 0x00, 0x16, 0x00, 0x24, // length of |
| 0x21, 0x12, 0xA4, 0x42, // magic cookie |
| '0', '1', '2', '3', // transaction id |
| '4', '5', '6', '7', |
| '8', '9', 'a', 'b', |
| 0x00, 0x20, 0x00, 0x04, // Mapped address. |
| 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x13, 0x00, 0x18, // Data attribute. |
| 0x90, 0x00, 0x00, 0x00, // RTP packet. |
| 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| 0xBE, 0xDE, 0x00, 0x02, |
| 0x22, 0xaa, 0xbb, 0xcc, |
| 0x32, 0xaa, 0xbb, 0xcc, |
| }; |
| |
| // A valid TURN channel header, but not RTP packet. |
| static unsigned char kTurnChannelMsgNoRtpPacket[] = { |
| 0x40, 0x00, 0x00, 0x04, |
| 0xaa, 0xbb, 0xcc, 0xdd, |
| }; |
| |
| // Turn ChannelMessage with zero length of payload. |
| static unsigned char kTurnChannelMsgWithZeroLength[] = { |
| 0x40, 0x00, 0x00, 0x00 |
| }; |
| |
| // Turn ChannelMessage, wrapping a RTP packet without extension. |
| static unsigned char kTurnChannelMsgWithRtpPacket[] = { |
| 0x40, 0x00, 0x00, 0x0C, |
| 0x80, 0x00, 0x00, 0x00, // RTP packet. |
| 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| }; |
| |
| // Turn ChannelMessage, wrapping a RTP packet with AbsSendTime Extension. |
| static unsigned char kTurnChannelMsgWithAbsSendTimeExtension[] = { |
| 0x40, 0x00, 0x00, 0x14, |
| 0x90, 0x00, 0x00, 0x00, // RTP packet. |
| 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| 0xBE, 0xDE, 0x00, 0x01, |
| 0x32, 0xaa, 0xbb, 0xcc, |
| }; |
| |
| // RTP packet with single byte extension header of length 4 bytes. |
| // Extension id = 3 and length = 3 |
| static unsigned char kRtpMsgWithAbsSendTimeExtension[] = { |
| 0x90, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, |
| 0xBE, 0xDE, 0x00, 0x02, |
| 0x22, 0x00, 0x02, 0x1c, |
| 0x32, 0xaa, 0xbb, 0xcc, |
| }; |
| |
| // Index of AbsSendTimeExtn data in message |kRtpMsgWithAbsSendTimeExtension|. |
| static const int kAstIndexInRtpMsg = 21; |
| |
| namespace content { |
| |
| // This test verifies parsing of all invalid raw packets. |
| TEST(P2PSocketHostTest, TestInvalidRawRtpMessages) { |
| int start_pos = INT_MAX, rtp_length = INT_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_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); |
| } |
| |
| // 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; |
| 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_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_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); |
| } |
| |
| // This test verifies incorrectly formed TURN channel messages. |
| TEST(P2PSocketHostTest, TestInvalidTurnChannelMessages) { |
| int start_pos = INT_MAX, rtp_length = INT_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_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); |
| } |
| |
| // 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; |
| 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); |
| } |
| |
| // 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; |
| 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); |
| } |
| |
| // This test verifies parsing of a valid TURN Send Indication messages. |
| TEST(P2PSocketHostTest, TestValidTurnSendIndicationMessages) { |
| int start_pos = INT_MAX, rtp_length = INT_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); |
| |
| start_pos = INT_MAX, rtp_length = INT_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); |
| } |
| |
| // This test verifies parsing of valid TURN Channel Messages. |
| TEST(P2PSocketHostTest, TestValidTurnChannelMessages) { |
| int start_pos = -1, rtp_length = -1; |
| 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); |
| |
| start_pos = -1, rtp_length = -1; |
| 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); |
| } |
| |
| // 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)); |
| } |
| |
| // Verify finding an extension ID in the TURN send indication message. |
| TEST(P2PSocketHostTest, TestUpdateAbsSendTimeExtensionInTurnSendIndication) { |
| EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtn( |
| reinterpret_cast<char*>(kTurnSendIndicationMsgWithoutRtpExtension), |
| sizeof(kTurnSendIndicationMsgWithoutRtpExtension), 3, 0)); |
| |
| EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtn( |
| reinterpret_cast<char*>(kTurnSendIndicationMsgWithAbsSendTimeExtension), |
| sizeof(kTurnSendIndicationMsgWithAbsSendTimeExtension), 3, 0)); |
| } |
| |
| // Test without any packet options variables set. This method should return |
| // without HMAC value in the packet. |
| TEST(P2PSocketHostTest, TestApplyPacketOptionsWithDefaultValues) { |
| unsigned char fake_tag[4] = { 0xba, 0xdd, 0xba, 0xdd }; |
| rtc::PacketOptions options; |
| std::vector<char> rtp_packet; |
| rtp_packet.resize(sizeof(kRtpMsgWithAbsSendTimeExtension) + 4); // tag length |
| memcpy(&rtp_packet[0], kRtpMsgWithAbsSendTimeExtension, |
| sizeof(kRtpMsgWithAbsSendTimeExtension)); |
| memcpy(&rtp_packet[sizeof(kRtpMsgWithAbsSendTimeExtension)], fake_tag, 4); |
| EXPECT_TRUE( |
| packet_processing_helpers::ApplyPacketOptions( |
| &rtp_packet[0], rtp_packet.size(), options, 0)); |
| // Making sure we have't updated the HMAC. |
| EXPECT_EQ(0, memcmp(&rtp_packet[sizeof(kRtpMsgWithAbsSendTimeExtension)], |
| fake_tag, 4)); |
| |
| // Verify AbsouluteSendTime extension field is not modified. |
| EXPECT_EQ(0, memcmp(&rtp_packet[kAstIndexInRtpMsg], |
| kTestAstValue, sizeof(kTestAstValue))); |
| } |
| |
| // Veirfy HMAC is updated when packet option parameters are set. |
| TEST(P2PSocketHostTest, TestApplyPacketOptionsWithAuthParams) { |
| rtc::PacketOptions options; |
| options.packet_time_params.srtp_auth_key.resize(20); |
| options.packet_time_params.srtp_auth_key.assign( |
| kTestKey, kTestKey + sizeof(kTestKey)); |
| options.packet_time_params.srtp_auth_tag_len = 4; |
| |
| std::vector<char> rtp_packet; |
| rtp_packet.resize(sizeof(kRtpMsgWithAbsSendTimeExtension) + 4); // tag length |
| memcpy(&rtp_packet[0], kRtpMsgWithAbsSendTimeExtension, |
| sizeof(kRtpMsgWithAbsSendTimeExtension)); |
| memcpy(&rtp_packet[sizeof(kRtpMsgWithAbsSendTimeExtension)], kFakeTag, 4); |
| EXPECT_TRUE(packet_processing_helpers::ApplyPacketOptions( |
| &rtp_packet[0], rtp_packet.size(), options, 0)); |
| // HMAC should be different from fake_tag. |
| EXPECT_NE(0, memcmp(&rtp_packet[sizeof(kRtpMsgWithAbsSendTimeExtension)], |
| kFakeTag, sizeof(kFakeTag))); |
| |
| // Verify AbsouluteSendTime extension field is not modified. |
| EXPECT_EQ(0, memcmp(&rtp_packet[kAstIndexInRtpMsg], |
| kTestAstValue, sizeof(kTestAstValue))); |
| } |
| |
| // Verify finding an extension ID in a raw rtp message. |
| TEST(P2PSocketHostTest, TestUpdateAbsSendTimeExtensionInRtpPacket) { |
| EXPECT_TRUE(packet_processing_helpers::UpdateRtpAbsSendTimeExtn( |
| reinterpret_cast<char*>(kRtpMsgWithAbsSendTimeExtension), |
| sizeof(kRtpMsgWithAbsSendTimeExtension), 3, 0)); |
| } |
| |
| // Verify we update both AbsSendTime extension header and HMAC. |
| TEST(P2PSocketHostTest, TestApplyPacketOptionsWithAuthParamsAndAbsSendTime) { |
| rtc::PacketOptions options; |
| options.packet_time_params.srtp_auth_key.resize(20); |
| options.packet_time_params.srtp_auth_key.assign( |
| kTestKey, kTestKey + sizeof(kTestKey)); |
| options.packet_time_params.srtp_auth_tag_len = 4; |
| options.packet_time_params.rtp_sendtime_extension_id = 3; |
| // 3 is also present in the test message. |
| |
| std::vector<char> rtp_packet; |
| rtp_packet.resize(sizeof(kRtpMsgWithAbsSendTimeExtension) + 4); // tag length |
| memcpy(&rtp_packet[0], kRtpMsgWithAbsSendTimeExtension, |
| sizeof(kRtpMsgWithAbsSendTimeExtension)); |
| memcpy(&rtp_packet[sizeof(kRtpMsgWithAbsSendTimeExtension)], kFakeTag, 4); |
| EXPECT_TRUE(packet_processing_helpers::ApplyPacketOptions( |
| &rtp_packet[0], rtp_packet.size(), options, 0xccbbaa)); |
| // HMAC should be different from fake_tag. |
| EXPECT_NE(0, memcmp(&rtp_packet[sizeof(kRtpMsgWithAbsSendTimeExtension)], |
| kFakeTag, sizeof(kFakeTag))); |
| |
| // ApplyPackets should have the new timestamp passed as input. |
| unsigned char timestamp_array[3] = { 0xcc, 0xbb, 0xaa }; |
| EXPECT_EQ(0, memcmp(&rtp_packet[kAstIndexInRtpMsg], |
| timestamp_array, sizeof(timestamp_array))); |
| } |
| |
| } // namespace content |