| Index: trunk/src/chrome/browser/media/webrtc_rtp_dump_writer_unittest.cc
|
| ===================================================================
|
| --- trunk/src/chrome/browser/media/webrtc_rtp_dump_writer_unittest.cc (revision 273764)
|
| +++ trunk/src/chrome/browser/media/webrtc_rtp_dump_writer_unittest.cc (working copy)
|
| @@ -1,362 +0,0 @@
|
| -// Copyright 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 "base/big_endian.h"
|
| -#include "base/bind.h"
|
| -#include "base/file_util.h"
|
| -#include "base/files/scoped_temp_dir.h"
|
| -#include "base/macros.h"
|
| -#include "base/memory/scoped_ptr.h"
|
| -#include "base/run_loop.h"
|
| -#include "chrome/browser/media/webrtc_rtp_dump_writer.h"
|
| -#include "content/public/browser/browser_thread.h"
|
| -#include "content/public/test/test_browser_thread_bundle.h"
|
| -#include "content/public/test/test_utils.h"
|
| -#include "testing/gmock/include/gmock/gmock.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -#include "third_party/zlib/zlib.h"
|
| -
|
| -static const size_t kMinimumRtpHeaderLength = 12;
|
| -
|
| -static void CreateFakeRtpPacketHeader(size_t csrc_count,
|
| - size_t extension_header_count,
|
| - std::vector<uint8>* packet_header) {
|
| - packet_header->resize(kMinimumRtpHeaderLength + csrc_count * sizeof(uint32) +
|
| - (extension_header_count + 1) * sizeof(uint32));
|
| -
|
| - memset(&(*packet_header)[0], 0, packet_header->size());
|
| -
|
| - // First byte format: vvpxcccc, where 'vv' is the version, 'p' is padding, 'x'
|
| - // is the extension bit, 'cccc' is the CSRC count.
|
| - (*packet_header)[0] = 0;
|
| - (*packet_header)[0] |= (0x2 << 6); // version.
|
| - // The extension bit.
|
| - (*packet_header)[0] |= (extension_header_count > 0 ? (0x1 << 4) : 0);
|
| - (*packet_header)[0] |= (csrc_count & 0xf);
|
| -
|
| - // Set extension length.
|
| - size_t offset = kMinimumRtpHeaderLength +
|
| - (csrc_count & 0xf) * sizeof(uint32) + sizeof(uint16);
|
| - base::WriteBigEndian(reinterpret_cast<char*>(&(*packet_header)[offset]),
|
| - static_cast<uint16>(extension_header_count));
|
| -}
|
| -
|
| -class WebRtcRtpDumpWriterTest : public testing::Test {
|
| - public:
|
| - WebRtcRtpDumpWriterTest()
|
| - : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP |
|
| - content::TestBrowserThreadBundle::REAL_FILE_THREAD),
|
| - temp_dir_(new base::ScopedTempDir()) {}
|
| -
|
| - virtual void SetUp() {
|
| - ASSERT_TRUE(temp_dir_->CreateUniqueTempDir());
|
| -
|
| - incoming_dump_path_ = temp_dir_->path().AppendASCII("rtpdump_recv");
|
| - outgoing_dump_path_ = temp_dir_->path().AppendASCII("rtpdump_send");
|
| - writer_.reset(new WebRtcRtpDumpWriter(
|
| - incoming_dump_path_,
|
| - outgoing_dump_path_,
|
| - 4 * 1024 * 1024,
|
| - base::Bind(&WebRtcRtpDumpWriterTest::OnMaxSizeReached,
|
| - base::Unretained(this))));
|
| - }
|
| -
|
| - // Verifies that the dump contains records of |rtp_packet| repeated
|
| - // |packet_count| times.
|
| - void VerifyDumps(size_t incoming_packet_count, size_t outgoing_packet_count) {
|
| - std::string incoming_dump;
|
| - std::string outgoing_dump;
|
| -
|
| - if (incoming_packet_count) {
|
| - EXPECT_TRUE(base::ReadFileToString(incoming_dump_path_, &incoming_dump));
|
| - EXPECT_TRUE(VerifyCompressedDump(&incoming_dump, incoming_packet_count));
|
| - } else {
|
| - EXPECT_FALSE(base::PathExists(incoming_dump_path_));
|
| - }
|
| -
|
| - if (outgoing_packet_count) {
|
| - EXPECT_TRUE(base::ReadFileToString(outgoing_dump_path_, &outgoing_dump));
|
| - EXPECT_TRUE(VerifyCompressedDump(&outgoing_dump, outgoing_packet_count));
|
| - } else {
|
| - EXPECT_FALSE(base::PathExists(outgoing_dump_path_));
|
| - }
|
| - }
|
| -
|
| - MOCK_METHOD2(OnEndDumpDone, void(bool, bool));
|
| - MOCK_METHOD0(OnMaxSizeReached, void(void));
|
| -
|
| - protected:
|
| - // Verifies the compressed dump file contains the expected number of packets.
|
| - bool VerifyCompressedDump(std::string* dump, size_t expected_packet_count) {
|
| - EXPECT_GT(dump->size(), 0U);
|
| -
|
| - std::vector<uint8> decompressed_dump;
|
| - EXPECT_TRUE(Decompress(dump, &decompressed_dump));
|
| -
|
| - size_t actual_packet_count = 0;
|
| - EXPECT_TRUE(ReadDecompressedDump(decompressed_dump, &actual_packet_count));
|
| - EXPECT_EQ(expected_packet_count, actual_packet_count);
|
| -
|
| - return true;
|
| - }
|
| -
|
| - // Decompresses the |input| into |output|.
|
| - bool Decompress(std::string* input, std::vector<uint8>* output) {
|
| - z_stream stream = {0};
|
| -
|
| - int result = inflateInit2(&stream, 15 + 16);
|
| - EXPECT_EQ(Z_OK, result);
|
| -
|
| - output->resize(input->size() * 100);
|
| -
|
| - stream.next_in =
|
| - reinterpret_cast<unsigned char*>(const_cast<char*>(&(*input)[0]));
|
| - stream.avail_in = input->size();
|
| - stream.next_out = &(*output)[0];
|
| - stream.avail_out = output->size();
|
| -
|
| - result = inflate(&stream, Z_FINISH);
|
| - DCHECK_EQ(Z_STREAM_END, result);
|
| - result = inflateEnd(&stream);
|
| - DCHECK_EQ(Z_OK, result);
|
| -
|
| - output->resize(output->size() - stream.avail_out);
|
| - return true;
|
| - }
|
| -
|
| - // Tries to read |dump| as a rtpplay dump file and returns the number of
|
| - // packets found in the dump.
|
| - bool ReadDecompressedDump(const std::vector<uint8>& dump,
|
| - size_t* packet_count) {
|
| - static const char kFirstLine[] = "#!rtpplay1.0 0.0.0.0/0\n";
|
| - static const size_t kDumpFileHeaderSize = 4 * sizeof(uint32);
|
| -
|
| - *packet_count = 0;
|
| - size_t dump_pos = 0;
|
| -
|
| - // Verifies the first line.
|
| - EXPECT_EQ(memcmp(&dump[0], kFirstLine, arraysize(kFirstLine) - 1), 0);
|
| -
|
| - dump_pos += arraysize(kFirstLine) - 1;
|
| - EXPECT_GT(dump.size(), dump_pos);
|
| -
|
| - // Skips the file header.
|
| - dump_pos += kDumpFileHeaderSize;
|
| - EXPECT_GT(dump.size(), dump_pos);
|
| -
|
| - // Reads each packet dump.
|
| - while (dump_pos < dump.size()) {
|
| - size_t packet_dump_length = 0;
|
| - if (!VerifyPacketDump(&dump[dump_pos],
|
| - dump.size() - dump_pos,
|
| - &packet_dump_length)) {
|
| - DVLOG(0) << "Failed to read the packet dump for packet "
|
| - << *packet_count << ", dump_pos = " << dump_pos
|
| - << ", dump_length = " << dump.size();
|
| - return false;
|
| - }
|
| -
|
| - EXPECT_GE(dump.size(), dump_pos + packet_dump_length);
|
| - dump_pos += packet_dump_length;
|
| -
|
| - (*packet_count)++;
|
| - }
|
| - return true;
|
| - }
|
| -
|
| - // Tries to read one packet dump starting at |dump| and returns the size of
|
| - // the packet dump.
|
| - bool VerifyPacketDump(const uint8* dump,
|
| - size_t dump_length,
|
| - size_t* packet_dump_length) {
|
| - static const size_t kDumpHeaderLength = 8;
|
| -
|
| - size_t dump_pos = 0;
|
| - base::ReadBigEndian(reinterpret_cast<const char*>(dump + dump_pos),
|
| - reinterpret_cast<uint16*>(packet_dump_length));
|
| - if (*packet_dump_length < kDumpHeaderLength + kMinimumRtpHeaderLength)
|
| - return false;
|
| -
|
| - EXPECT_GE(dump_length, *packet_dump_length);
|
| - dump_pos += sizeof(uint16);
|
| -
|
| - uint16 rtp_packet_length = 0;
|
| - base::ReadBigEndian(reinterpret_cast<const char*>(dump + dump_pos),
|
| - &rtp_packet_length);
|
| - if (rtp_packet_length < kMinimumRtpHeaderLength)
|
| - return false;
|
| -
|
| - dump_pos += sizeof(uint16);
|
| -
|
| - // Skips the elapsed time field.
|
| - dump_pos += sizeof(uint32);
|
| -
|
| - return IsValidRtpHeader(dump + dump_pos,
|
| - *packet_dump_length - kDumpHeaderLength);
|
| - }
|
| -
|
| - // Returns true if |header| is a valid RTP header.
|
| - bool IsValidRtpHeader(const uint8* header, size_t length) {
|
| - if ((header[0] & 0xC0) != 0x80)
|
| - return false;
|
| -
|
| - size_t cc_count = header[0] & 0x0F;
|
| - size_t header_length_without_extn = kMinimumRtpHeaderLength + 4 * cc_count;
|
| -
|
| - if (length < header_length_without_extn)
|
| - return false;
|
| -
|
| - uint16 extension_count = 0;
|
| - base::ReadBigEndian(
|
| - reinterpret_cast<const char*>(header + header_length_without_extn + 2),
|
| - &extension_count);
|
| -
|
| - if (length < (extension_count + 1) * 4 + header_length_without_extn)
|
| - return false;
|
| -
|
| - return true;
|
| - }
|
| -
|
| - content::TestBrowserThreadBundle thread_bundle_;
|
| - scoped_ptr<base::ScopedTempDir> temp_dir_;
|
| - base::FilePath incoming_dump_path_;
|
| - base::FilePath outgoing_dump_path_;
|
| - scoped_ptr<WebRtcRtpDumpWriter> writer_;
|
| -};
|
| -
|
| -TEST_F(WebRtcRtpDumpWriterTest, NoDumpFileIfNoPacketDumped) {
|
| - // The scope is used to make sure the EXPECT_CALL is checked before exiting
|
| - // the scope.
|
| - {
|
| - EXPECT_CALL(*this, OnEndDumpDone(false, false));
|
| -
|
| - writer_->EndDump(RTP_DUMP_BOTH,
|
| - base::Bind(&WebRtcRtpDumpWriterTest::OnEndDumpDone,
|
| - base::Unretained(this)));
|
| -
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - }
|
| - EXPECT_FALSE(base::PathExists(incoming_dump_path_));
|
| - EXPECT_FALSE(base::PathExists(outgoing_dump_path_));
|
| -}
|
| -
|
| -TEST_F(WebRtcRtpDumpWriterTest, WriteAndFlushSmallSizeDump) {
|
| - std::vector<uint8> packet_header;
|
| - CreateFakeRtpPacketHeader(1, 2, &packet_header);
|
| -
|
| - writer_->WriteRtpPacket(
|
| - &packet_header[0], packet_header.size(), 100, true);
|
| - writer_->WriteRtpPacket(
|
| - &packet_header[0], packet_header.size(), 100, false);
|
| -
|
| - // The scope is used to make sure the EXPECT_CALL is checked before exiting
|
| - // the scope.
|
| - {
|
| - EXPECT_CALL(*this, OnEndDumpDone(true, true));
|
| -
|
| - writer_->EndDump(RTP_DUMP_BOTH,
|
| - base::Bind(&WebRtcRtpDumpWriterTest::OnEndDumpDone,
|
| - base::Unretained(this)));
|
| -
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - }
|
| -
|
| - VerifyDumps(1, 1);
|
| -}
|
| -
|
| -TEST_F(WebRtcRtpDumpWriterTest, WriteOverMaxLimit) {
|
| - // Reset the writer with a small max size limit.
|
| - writer_.reset(new WebRtcRtpDumpWriter(
|
| - incoming_dump_path_,
|
| - outgoing_dump_path_,
|
| - 100,
|
| - base::Bind(&WebRtcRtpDumpWriterTest::OnMaxSizeReached,
|
| - base::Unretained(this))));
|
| -
|
| - std::vector<uint8> packet_header;
|
| - CreateFakeRtpPacketHeader(3, 4, &packet_header);
|
| -
|
| - const size_t kPacketCount = 200;
|
| - // The scope is used to make sure the EXPECT_CALL is checked before exiting
|
| - // the scope.
|
| - {
|
| - EXPECT_CALL(*this, OnMaxSizeReached()).Times(testing::AtLeast(1));
|
| -
|
| - // Write enough packets to overflow the in-memory buffer and max limit.
|
| - for (size_t i = 0; i < kPacketCount; ++i) {
|
| - writer_->WriteRtpPacket(
|
| - &packet_header[0], packet_header.size(), 100, true);
|
| -
|
| - writer_->WriteRtpPacket(
|
| - &packet_header[0], packet_header.size(), 100, false);
|
| - }
|
| -
|
| - EXPECT_CALL(*this, OnEndDumpDone(true, true));
|
| -
|
| - writer_->EndDump(RTP_DUMP_BOTH,
|
| - base::Bind(&WebRtcRtpDumpWriterTest::OnEndDumpDone,
|
| - base::Unretained(this)));
|
| -
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - }
|
| - VerifyDumps(kPacketCount, kPacketCount);
|
| -}
|
| -
|
| -TEST_F(WebRtcRtpDumpWriterTest, DestroyWriterBeforeEndDumpCallback) {
|
| - EXPECT_CALL(*this, OnEndDumpDone(testing::_, testing::_)).Times(0);
|
| -
|
| - writer_->EndDump(RTP_DUMP_BOTH,
|
| - base::Bind(&WebRtcRtpDumpWriterTest::OnEndDumpDone,
|
| - base::Unretained(this)));
|
| -
|
| - writer_.reset();
|
| -
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| -}
|
| -
|
| -TEST_F(WebRtcRtpDumpWriterTest, EndDumpsSeparately) {
|
| - std::vector<uint8> packet_header;
|
| - CreateFakeRtpPacketHeader(1, 2, &packet_header);
|
| -
|
| - writer_->WriteRtpPacket(
|
| - &packet_header[0], packet_header.size(), 100, true);
|
| - writer_->WriteRtpPacket(
|
| - &packet_header[0], packet_header.size(), 100, true);
|
| - writer_->WriteRtpPacket(
|
| - &packet_header[0], packet_header.size(), 100, false);
|
| -
|
| - // The scope is used to make sure the EXPECT_CALL is checked before exiting
|
| - // the scope.
|
| - {
|
| - EXPECT_CALL(*this, OnEndDumpDone(true, false));
|
| - EXPECT_CALL(*this, OnEndDumpDone(false, true));
|
| -
|
| - writer_->EndDump(RTP_DUMP_INCOMING,
|
| - base::Bind(&WebRtcRtpDumpWriterTest::OnEndDumpDone,
|
| - base::Unretained(this)));
|
| -
|
| - writer_->EndDump(RTP_DUMP_OUTGOING,
|
| - base::Bind(&WebRtcRtpDumpWriterTest::OnEndDumpDone,
|
| - base::Unretained(this)));
|
| -
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - content::RunAllPendingInMessageLoop(content::BrowserThread::FILE);
|
| - base::RunLoop().RunUntilIdle();
|
| - }
|
| -
|
| - VerifyDumps(2, 1);
|
| -}
|
|
|