diff options
Diffstat (limited to 'chromium/content/browser/renderer_host/p2p/socket_host_unittest.cc')
| -rw-r--r-- | chromium/content/browser/renderer_host/p2p/socket_host_unittest.cc | 376 |
1 files changed, 376 insertions, 0 deletions
diff --git a/chromium/content/browser/renderer_host/p2p/socket_host_unittest.cc b/chromium/content/browser/renderer_host/p2p/socket_host_unittest.cc new file mode 100644 index 00000000000..1404ced5bc6 --- /dev/null +++ b/chromium/content/browser/renderer_host/p2p/socket_host_unittest.cc @@ -0,0 +1,376 @@ +// 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 }; + talk_base::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) { + talk_base::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) { + talk_base::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 |