Add rtppacketuil.h/cc

webrtc/p2p/base/rtppacketutil_unittest.cc

+/*
+ *  Copyright 2016 The WebRTC Project Authors. All rights reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "webrtc/p2p/base/rtppacketutil.h"
+
+#include <stddef.h>
+
+#include <vector>
+
+#include "webrtc/base/asyncpacketsocket.h"
+#include "webrtc/base/gunit.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 rtc {
+
+// This test verifies parsing of all invalid raw packets.
+TEST(RtpPacketUtil, TestInvalidRawRtpMessages) {
+  size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX;
+  EXPECT_FALSE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kRtpMsgWithInvalidLength),
+      sizeof(kRtpMsgWithInvalidLength), &start_pos, &rtp_length));
+  EXPECT_EQ(SIZE_MAX, start_pos);
+  EXPECT_EQ(SIZE_MAX, rtp_length);
+
+  EXPECT_FALSE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kRtpMsgWithInvalidExtnLength),
+      sizeof(kRtpMsgWithInvalidExtnLength), &start_pos, &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(RtpPacketUtil, TestInvalidTurnSendIndicationMessages) {
+  // Initializing out params to very large value.
+  size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX;
+  EXPECT_FALSE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnSendIndicationMsgWithNoAttributes),
+      sizeof(kTurnSendIndicationMsgWithNoAttributes), &start_pos, &rtp_length));
+  EXPECT_EQ(SIZE_MAX, start_pos);
+  EXPECT_EQ(SIZE_MAX, rtp_length);
+
+  EXPECT_FALSE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnSendIndicationMsgWithInvalidLength),
+      sizeof(kTurnSendIndicationMsgWithInvalidLength), &start_pos,
+      &rtp_length));
+  EXPECT_EQ(SIZE_MAX, start_pos);
+  EXPECT_EQ(SIZE_MAX, rtp_length);
+
+  EXPECT_FALSE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnSendIndicatinMsgWithNoDataAttribute),
+      sizeof(kTurnSendIndicatinMsgWithNoDataAttribute), &start_pos,
+      &rtp_length));
+  EXPECT_EQ(SIZE_MAX, start_pos);
+  EXPECT_EQ(SIZE_MAX, rtp_length);
+}
+
+// This test verifies incorrectly formed TURN channel messages.
+TEST(RtpPacketUtil, TestInvalidTurnChannelMessages) {
+  size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX;
+  EXPECT_FALSE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnChannelMsgNoRtpPacket),
+      sizeof(kTurnChannelMsgNoRtpPacket), &start_pos, &rtp_length));
+  EXPECT_EQ(SIZE_MAX, start_pos);
+  EXPECT_EQ(SIZE_MAX, rtp_length);
+
+  EXPECT_FALSE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnChannelMsgWithZeroLength),
+      sizeof(kTurnChannelMsgWithZeroLength), &start_pos, &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(RtpPacketUtil, TestValid2ByteExtnHdrRtpMessage) {
+  size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX;
+  EXPECT_TRUE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kRtpMsgWith2ByteExtnHeader),
+      sizeof(kRtpMsgWith2ByteExtnHeader), &start_pos, &rtp_length));
+  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(RtpPacketUtil, TestValidRtpPacketWithAbsSendTimeExtension) {
+  size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX;
+  EXPECT_TRUE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kRtpMsgWithAbsSendTimeExtension),
+      sizeof(kRtpMsgWithAbsSendTimeExtension), &start_pos, &rtp_length));
+  EXPECT_EQ(24U, rtp_length);
+  EXPECT_EQ(0U, start_pos);
+}
+
+// This test verifies parsing of a valid TURN Send Indication messages.
+TEST(RtpPacketUtil, TestValidTurnSendIndicationMessages) {
+  size_t start_pos = SIZE_MAX, rtp_length = SIZE_MAX;
+  EXPECT_TRUE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnSendIndicationMsgWithoutRtpExtension),
+      sizeof(kTurnSendIndicationMsgWithoutRtpExtension), &start_pos,
+      &rtp_length));
+  EXPECT_EQ(12U, rtp_length);
+  EXPECT_EQ(32U, start_pos);
+
+  start_pos = SIZE_MAX, rtp_length = SIZE_MAX;
+  EXPECT_TRUE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnSendIndicationMsgWithAbsSendTimeExtension),
+      sizeof(kTurnSendIndicationMsgWithAbsSendTimeExtension), &start_pos,
+      &rtp_length));
+  EXPECT_EQ(24U, rtp_length);
+  EXPECT_EQ(32U, start_pos);
+}
+
+// This test verifies parsing of valid TURN Channel Messages.
+TEST(RtpPacketUtil, TestValidTurnChannelMessages) {
+  size_t start_pos = 0, rtp_length = 0;
+  EXPECT_TRUE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnChannelMsgWithRtpPacket),
+      sizeof(kTurnChannelMsgWithRtpPacket), &start_pos, &rtp_length));
+  EXPECT_EQ(12U, rtp_length);
+  EXPECT_EQ(4U, start_pos);
+
+  start_pos = 0, rtp_length = 0;
+  EXPECT_TRUE(GetRtpPacketStartPositionAndLength(
+      reinterpret_cast<char*>(kTurnChannelMsgWithAbsSendTimeExtension),
+      sizeof(kTurnChannelMsgWithAbsSendTimeExtension), &start_pos,
+      &rtp_length));
+  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(RtpPacketUtil, TestUpdateAbsSendTimeExtensionIn2ByteHeaderExtn) {
+  EXPECT_FALSE(UpdateRtpAbsSendTimeExtension(
+      reinterpret_cast<char*>(kRtpMsgWith2ByteExtnHeader),
+      sizeof(kRtpMsgWith2ByteExtnHeader), 3, 0));
+}
+
+// Verify finding an extension ID in the TURN send indication message.
+TEST(RtpPacketUtil, TestUpdateAbsSendTimeExtensionInTurnSendIndication) {
+  EXPECT_TRUE(UpdateRtpAbsSendTimeExtension(
+      reinterpret_cast<char*>(kTurnSendIndicationMsgWithoutRtpExtension),
+      sizeof(kTurnSendIndicationMsgWithoutRtpExtension), 3, 0));
+
+  EXPECT_TRUE(UpdateRtpAbsSendTimeExtension(
+      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(RtpPacketUtil, 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(
+      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(RtpPacketUtil, 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(
+      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(RtpPacketUtil, TestUpdateAbsSendTimeExtensionInRtpPacket) {
+  EXPECT_TRUE(UpdateRtpAbsSendTimeExtension(
+      reinterpret_cast<char*>(kRtpMsgWithAbsSendTimeExtension),
+      sizeof(kRtpMsgWithAbsSendTimeExtension), 3, 0));
+}
+
+// Verify we update both AbsSendTime extension header and HMAC.
+TEST(RtpPacketUtil, 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(
+      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 rtc