Implements RTP header dumping.

chrome/browser/extensions/api/webrtc_logging_private/webrtc_logging_private_api.cc

 // Copyright 2013 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 <string>
 
 #include "chrome/browser/extensions/api/webrtc_logging_private/webrtc_logging_private_api.h"
 
 #include "base/logging.h"
 #include "base/strings/string_number_conversions.h"
@@ skipping 261 matching lines @@
 WebrtcLoggingPrivateStartRtpDumpFunction::
     WebrtcLoggingPrivateStartRtpDumpFunction() {}
 
 WebrtcLoggingPrivateStartRtpDumpFunction::
     ~WebrtcLoggingPrivateStartRtpDumpFunction() {}
 
 bool WebrtcLoggingPrivateStartRtpDumpFunction::RunAsync() {
   scoped_ptr<StartRtpDump::Params> params(StartRtpDump::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
+  if (!params->incoming && !params->outgoing) {
+    StartRtpDumpCallback(false, "Either incoming or outgoing must be true.");
+    return true;
+  }
+
+  RtpDumpType type =
+      (params->incoming && params->outgoing)
+          ? RTP_DUMP_BOTH
+          : (params->incoming ? RTP_DUMP_INCOMING : RTP_DUMP_OUTGOING);
+
   content::RenderProcessHost* host =
       RphFromTabIdAndSecurityOrigin(params->tab_id, params->security_origin);
   if (!host)
     return false;
 
   scoped_refptr<WebRtcLoggingHandlerHost> webrtc_logging_handler_host(
       base::UserDataAdapter<WebRtcLoggingHandlerHost>::Get(host, host));
 
   WebRtcLoggingHandlerHost::GenericDoneCallback callback = base::Bind(
       &WebrtcLoggingPrivateStartRtpDumpFunction::StartRtpDumpCallback, this);
 
+  // This call cannot fail.
+  content::RenderProcessHost::WebRtcStopRtpDumpCallback stop_callback =
+      host->StartRtpDump(params->incoming,
+                         params->outgoing,
+                         base::Bind(&WebRtcLoggingHandlerHost::OnRtpPacket,
+                                    webrtc_logging_handler_host));
+
   BrowserThread::PostTask(BrowserThread::IO,
                           FROM_HERE,
                           base::Bind(&WebRtcLoggingHandlerHost::StartRtpDump,
                                      webrtc_logging_handler_host,
-                                     params->incoming,
-                                     params->outgoing,
-                                     callback));
-
+                                     type,
+                                     callback,
+                                     stop_callback));
   return true;
 }
 
 void WebrtcLoggingPrivateStartRtpDumpFunction::StartRtpDumpCallback(
     bool success,
     const std::string& error_message) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   if (!success)
     SetError(error_message);
   SendResponse(success);
 }
 
 WebrtcLoggingPrivateStopRtpDumpFunction::
     WebrtcLoggingPrivateStopRtpDumpFunction() {}
 
 WebrtcLoggingPrivateStopRtpDumpFunction::
     ~WebrtcLoggingPrivateStopRtpDumpFunction() {}
 
 bool WebrtcLoggingPrivateStopRtpDumpFunction::RunAsync() {
   scoped_ptr<StopRtpDump::Params> params(StopRtpDump::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
+  if (!params->incoming && !params->outgoing) {
+    StopRtpDumpCallback(false, "Either incoming or outgoing must be true.");
+    return true;
+  }
+
+  RtpDumpType type =
+      (params->incoming && params->outgoing)
+          ? RTP_DUMP_BOTH
+          : (params->incoming ? RTP_DUMP_INCOMING : RTP_DUMP_OUTGOING);
+
   content::RenderProcessHost* host =
       RphFromTabIdAndSecurityOrigin(params->tab_id, params->security_origin);
   if (!host)
     return false;
 
   scoped_refptr<WebRtcLoggingHandlerHost> webrtc_logging_handler_host(
       base::UserDataAdapter<WebRtcLoggingHandlerHost>::Get(host, host));
 
   WebRtcLoggingHandlerHost::GenericDoneCallback callback = base::Bind(
       &WebrtcLoggingPrivateStopRtpDumpFunction::StopRtpDumpCallback, this);
 
   BrowserThread::PostTask(BrowserThread::IO,
                           FROM_HERE,
                           base::Bind(&WebRtcLoggingHandlerHost::StopRtpDump,
                                      webrtc_logging_handler_host,
-                                     params->incoming,
-                                     params->outgoing,
+                                     type,
                                      callback));
-
   return true;
 }
 
 void WebrtcLoggingPrivateStopRtpDumpFunction::StopRtpDumpCallback(
     bool success,
     const std::string& error_message) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   if (!success)
     SetError(error_message);
   SendResponse(success);
 }
 
 }  // namespace extensions