| Index: trunk/src/chrome/browser/media/webrtc_rtp_dump_handler.cc
|
| ===================================================================
|
| --- trunk/src/chrome/browser/media/webrtc_rtp_dump_handler.cc (revision 273764)
|
| +++ trunk/src/chrome/browser/media/webrtc_rtp_dump_handler.cc (working copy)
|
| @@ -1,342 +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 "chrome/browser/media/webrtc_rtp_dump_handler.h"
|
| -
|
| -#include "base/file_util.h"
|
| -#include "base/logging.h"
|
| -#include "base/strings/string_number_conversions.h"
|
| -#include "base/time/time.h"
|
| -#include "chrome/browser/media/webrtc_rtp_dump_writer.h"
|
| -#include "content/public/browser/browser_thread.h"
|
| -
|
| -using content::BrowserThread;
|
| -
|
| -namespace {
|
| -
|
| -static const size_t kMaxOngoingRtpDumpsAllowed = 5;
|
| -
|
| -// The browser process wide total number of ongoing (i.e. started and not
|
| -// released) RTP dumps. Incoming and outgoing in one WebRtcDumpHandler are
|
| -// counted as one dump.
|
| -// Must be accessed on the browser IO thread.
|
| -static size_t g_ongoing_rtp_dumps = 0;
|
| -
|
| -void FireGenericDoneCallback(
|
| - const WebRtcRtpDumpHandler::GenericDoneCallback& callback,
|
| - bool success,
|
| - const std::string& error_message) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - DCHECK(!callback.is_null());
|
| -
|
| - content::BrowserThread::PostTask(
|
| - content::BrowserThread::UI,
|
| - FROM_HERE,
|
| - base::Bind(callback, success, error_message));
|
| -}
|
| -
|
| -bool DumpTypeContainsIncoming(RtpDumpType type) {
|
| - return type == RTP_DUMP_INCOMING || type == RTP_DUMP_BOTH;
|
| -}
|
| -
|
| -bool DumpTypeContainsOutgoing(RtpDumpType type) {
|
| - return type == RTP_DUMP_OUTGOING || type == RTP_DUMP_BOTH;
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -WebRtcRtpDumpHandler::WebRtcRtpDumpHandler(const base::FilePath& dump_dir)
|
| - : dump_dir_(dump_dir),
|
| - incoming_state_(STATE_NONE),
|
| - outgoing_state_(STATE_NONE),
|
| - weak_ptr_factory_(this) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -}
|
| -
|
| -WebRtcRtpDumpHandler::~WebRtcRtpDumpHandler() {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - // Reset dump writer first to stop writing.
|
| - if (dump_writer_) {
|
| - --g_ongoing_rtp_dumps;
|
| - dump_writer_.reset();
|
| - }
|
| -
|
| - if (incoming_state_ != STATE_NONE && !incoming_dump_path_.empty()) {
|
| - BrowserThread::PostTask(
|
| - BrowserThread::FILE,
|
| - FROM_HERE,
|
| - base::Bind(
|
| - base::IgnoreResult(&base::DeleteFile), incoming_dump_path_, false));
|
| - }
|
| -
|
| - if (outgoing_state_ != STATE_NONE && !outgoing_dump_path_.empty()) {
|
| - BrowserThread::PostTask(
|
| - BrowserThread::FILE,
|
| - FROM_HERE,
|
| - base::Bind(
|
| - base::IgnoreResult(&base::DeleteFile), outgoing_dump_path_, false));
|
| - }
|
| -}
|
| -
|
| -bool WebRtcRtpDumpHandler::StartDump(RtpDumpType type,
|
| - std::string* error_message) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - if (!dump_writer_ && g_ongoing_rtp_dumps >= kMaxOngoingRtpDumpsAllowed) {
|
| - *error_message = "Max RTP dump limit reached.";
|
| - DVLOG(2) << *error_message;
|
| - return false;
|
| - }
|
| -
|
| - // Returns an error if any type of dump specified by the caller cannot be
|
| - // started.
|
| - if ((DumpTypeContainsIncoming(type) && incoming_state_ != STATE_NONE) ||
|
| - (DumpTypeContainsOutgoing(type) && outgoing_state_ != STATE_NONE)) {
|
| - *error_message =
|
| - "RTP dump already started for type " + base::IntToString(type);
|
| - return false;
|
| - }
|
| -
|
| - if (DumpTypeContainsIncoming(type))
|
| - incoming_state_ = STATE_STARTED;
|
| -
|
| - if (DumpTypeContainsOutgoing(type))
|
| - outgoing_state_ = STATE_STARTED;
|
| -
|
| - DVLOG(2) << "Start RTP dumping: type = " << type;
|
| -
|
| - if (!dump_writer_) {
|
| - ++g_ongoing_rtp_dumps;
|
| -
|
| - static const char kRecvDumpFilePrefix[] = "rtpdump_recv_";
|
| - static const char kSendDumpFilePrefix[] = "rtpdump_send_";
|
| - static const size_t kMaxDumpSize = 5 * 1024 * 1024; // 5MB
|
| -
|
| - std::string dump_id = base::DoubleToString(base::Time::Now().ToDoubleT());
|
| - incoming_dump_path_ =
|
| - dump_dir_.AppendASCII(std::string(kRecvDumpFilePrefix) + dump_id)
|
| - .AddExtension(FILE_PATH_LITERAL(".gz"));
|
| -
|
| - outgoing_dump_path_ =
|
| - dump_dir_.AppendASCII(std::string(kSendDumpFilePrefix) + dump_id)
|
| - .AddExtension(FILE_PATH_LITERAL(".gz"));
|
| -
|
| - // WebRtcRtpDumpWriter does not support changing the dump path after it's
|
| - // created. So we assign both incoming and outgoing dump path even if only
|
| - // one type of dumping has been started.
|
| - // For "Unretained(this)", see comments StopDump.
|
| - dump_writer_.reset(new WebRtcRtpDumpWriter(
|
| - incoming_dump_path_,
|
| - outgoing_dump_path_,
|
| - kMaxDumpSize,
|
| - base::Bind(&WebRtcRtpDumpHandler::OnMaxDumpSizeReached,
|
| - base::Unretained(this))));
|
| - }
|
| -
|
| - return true;
|
| -}
|
| -
|
| -void WebRtcRtpDumpHandler::StopDump(RtpDumpType type,
|
| - const GenericDoneCallback& callback) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - // Returns an error if any type of dump specified by the caller cannot be
|
| - // stopped.
|
| - if ((DumpTypeContainsIncoming(type) && incoming_state_ != STATE_STARTED) ||
|
| - (DumpTypeContainsOutgoing(type) && outgoing_state_ != STATE_STARTED)) {
|
| - if (!callback.is_null()) {
|
| - FireGenericDoneCallback(
|
| - callback,
|
| - false,
|
| - "RTP dump not started or already stopped for type " +
|
| - base::IntToString(type));
|
| - }
|
| - return;
|
| - }
|
| -
|
| - DVLOG(2) << "Stopping RTP dumping: type = " << type;
|
| -
|
| - if (DumpTypeContainsIncoming(type))
|
| - incoming_state_ = STATE_STOPPING;
|
| -
|
| - if (DumpTypeContainsOutgoing(type))
|
| - outgoing_state_ = STATE_STOPPING;
|
| -
|
| - // Using "Unretained(this)" because the this object owns the writer and the
|
| - // writer is guaranteed to cancel the callback before it goes away. Same for
|
| - // the other posted tasks bound to the writer.
|
| - dump_writer_->EndDump(
|
| - type,
|
| - base::Bind(&WebRtcRtpDumpHandler::OnDumpEnded,
|
| - base::Unretained(this),
|
| - callback.is_null()
|
| - ? base::Closure()
|
| - : base::Bind(&FireGenericDoneCallback, callback, true, ""),
|
| - type));
|
| -}
|
| -
|
| -bool WebRtcRtpDumpHandler::ReadyToRelease() const {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - return incoming_state_ != STATE_STARTED &&
|
| - incoming_state_ != STATE_STOPPING &&
|
| - outgoing_state_ != STATE_STARTED && outgoing_state_ != STATE_STOPPING;
|
| -}
|
| -
|
| -WebRtcRtpDumpHandler::ReleasedDumps WebRtcRtpDumpHandler::ReleaseDumps() {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - DCHECK(ReadyToRelease());
|
| -
|
| - base::FilePath incoming_dump, outgoing_dump;
|
| -
|
| - if (incoming_state_ == STATE_STOPPED) {
|
| - DVLOG(2) << "Incoming RTP dumps released: " << incoming_dump_path_.value();
|
| -
|
| - incoming_state_ = STATE_NONE;
|
| - incoming_dump = incoming_dump_path_;
|
| - }
|
| -
|
| - if (outgoing_state_ == STATE_STOPPED) {
|
| - DVLOG(2) << "Outgoing RTP dumps released: " << outgoing_dump_path_.value();
|
| -
|
| - outgoing_state_ = STATE_NONE;
|
| - outgoing_dump = outgoing_dump_path_;
|
| - }
|
| - return ReleasedDumps(incoming_dump, outgoing_dump);
|
| -}
|
| -
|
| -void WebRtcRtpDumpHandler::OnRtpPacket(const uint8* packet_header,
|
| - size_t header_length,
|
| - size_t packet_length,
|
| - bool incoming) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - if ((incoming && incoming_state_ != STATE_STARTED) ||
|
| - (!incoming && outgoing_state_ != STATE_STARTED)) {
|
| - return;
|
| - }
|
| -
|
| - dump_writer_->WriteRtpPacket(
|
| - packet_header, header_length, packet_length, incoming);
|
| -}
|
| -
|
| -void WebRtcRtpDumpHandler::StopOngoingDumps(const base::Closure& callback) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - DCHECK(!callback.is_null());
|
| -
|
| - // No ongoing dumps, return directly.
|
| - if ((incoming_state_ == STATE_NONE || incoming_state_ == STATE_STOPPED) &&
|
| - (outgoing_state_ == STATE_NONE || outgoing_state_ == STATE_STOPPED)) {
|
| - callback.Run();
|
| - return;
|
| - }
|
| -
|
| - // If the FILE thread is working on stopping the dumps, wait for the FILE
|
| - // thread to return and check the states again.
|
| - if (incoming_state_ == STATE_STOPPING || outgoing_state_ == STATE_STOPPING) {
|
| - BrowserThread::PostTaskAndReply(
|
| - BrowserThread::FILE,
|
| - FROM_HERE,
|
| - base::Bind(&base::DoNothing),
|
| - base::Bind(&WebRtcRtpDumpHandler::StopOngoingDumps,
|
| - weak_ptr_factory_.GetWeakPtr(),
|
| - callback));
|
| - return;
|
| - }
|
| -
|
| - // Either incoming or outgoing dump must be ongoing.
|
| - RtpDumpType type =
|
| - (incoming_state_ == STATE_STARTED)
|
| - ? (outgoing_state_ == STATE_STARTED ? RTP_DUMP_BOTH
|
| - : RTP_DUMP_INCOMING)
|
| - : RTP_DUMP_OUTGOING;
|
| -
|
| - if (incoming_state_ == STATE_STARTED)
|
| - incoming_state_ = STATE_STOPPING;
|
| -
|
| - if (outgoing_state_ == STATE_STARTED)
|
| - outgoing_state_ = STATE_STOPPING;
|
| -
|
| - DVLOG(2) << "Stopping ongoing dumps: type = " << type;
|
| -
|
| - dump_writer_->EndDump(type,
|
| - base::Bind(&WebRtcRtpDumpHandler::OnDumpEnded,
|
| - base::Unretained(this),
|
| - callback,
|
| - type));
|
| -}
|
| -
|
| -void WebRtcRtpDumpHandler::SetDumpWriterForTesting(
|
| - scoped_ptr<WebRtcRtpDumpWriter> writer) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - dump_writer_ = writer.Pass();
|
| - ++g_ongoing_rtp_dumps;
|
| -
|
| - incoming_dump_path_ = dump_dir_.AppendASCII("recv");
|
| - outgoing_dump_path_ = dump_dir_.AppendASCII("send");
|
| -}
|
| -
|
| -void WebRtcRtpDumpHandler::OnMaxDumpSizeReached() {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - RtpDumpType type =
|
| - (incoming_state_ == STATE_STARTED)
|
| - ? (outgoing_state_ == STATE_STARTED ? RTP_DUMP_BOTH
|
| - : RTP_DUMP_INCOMING)
|
| - : RTP_DUMP_OUTGOING;
|
| - StopDump(type, GenericDoneCallback());
|
| -}
|
| -
|
| -void WebRtcRtpDumpHandler::OnDumpEnded(const base::Closure& callback,
|
| - RtpDumpType ended_type,
|
| - bool incoming_success,
|
| - bool outgoing_success) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - if (DumpTypeContainsIncoming(ended_type)) {
|
| - DCHECK_EQ(STATE_STOPPING, incoming_state_);
|
| - incoming_state_ = STATE_STOPPED;
|
| -
|
| - if (!incoming_success) {
|
| - BrowserThread::PostTask(BrowserThread::FILE,
|
| - FROM_HERE,
|
| - base::Bind(base::IgnoreResult(&base::DeleteFile),
|
| - incoming_dump_path_,
|
| - false));
|
| -
|
| - DVLOG(2) << "Deleted invalid incoming dump "
|
| - << incoming_dump_path_.value();
|
| - incoming_dump_path_.clear();
|
| - }
|
| - }
|
| -
|
| - if (DumpTypeContainsOutgoing(ended_type)) {
|
| - DCHECK_EQ(STATE_STOPPING, outgoing_state_);
|
| - outgoing_state_ = STATE_STOPPED;
|
| -
|
| - if (!outgoing_success) {
|
| - BrowserThread::PostTask(BrowserThread::FILE,
|
| - FROM_HERE,
|
| - base::Bind(base::IgnoreResult(&base::DeleteFile),
|
| - outgoing_dump_path_,
|
| - false));
|
| -
|
| - DVLOG(2) << "Deleted invalid outgoing dump "
|
| - << outgoing_dump_path_.value();
|
| - outgoing_dump_path_.clear();
|
| - }
|
| - }
|
| -
|
| - if (!callback.is_null())
|
| - callback.Run();
|
| -
|
| - // Release the writer when it's no longer needed.
|
| - if (incoming_state_ != STATE_STOPPING && outgoing_state_ != STATE_STOPPING &&
|
| - incoming_state_ != STATE_STARTED && outgoing_state_ != STATE_STARTED) {
|
| - dump_writer_.reset();
|
| - --g_ongoing_rtp_dumps;
|
| - }
|
| -}
|
|
|