[DO NOT COMMIT] mojo datapipe performance measurement

content/browser/renderer_host/websocket_host.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 "content/browser/renderer_host/websocket_host.h"
 
 #include <inttypes.h>
 
 #include <utility>
 
@@ skipping 353 matching lines @@
 WebSocketHost::WebSocketHost(int routing_id,
                              WebSocketDispatcherHost* dispatcher,
                              net::URLRequestContext* url_request_context,
                              base::TimeDelta delay)
     : dispatcher_(dispatcher),
       url_request_context_(url_request_context),
       routing_id_(routing_id),
       delay_(delay),
       pending_flow_control_quota_(0),
       handshake_succeeded_(false),
+      loader_test_num_read_bytes_(0),
       weak_ptr_factory_(this) {
   DVLOG(1) << "WebSocketHost: created routing_id=" << routing_id;
 }
 
 WebSocketHost::~WebSocketHost() {}
 
 void WebSocketHost::GoAway() {
   OnDropChannel(false, static_cast<uint16_t>(net::kWebSocketErrorGoingAway),
                 "");
 }
 
 bool WebSocketHost::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(WebSocketHost, message)
     IPC_MESSAGE_HANDLER(WebSocketHostMsg_AddChannelRequest, OnAddChannelRequest)
     IPC_MESSAGE_HANDLER(WebSocketHostMsg_SendBlob, OnSendBlob)
     IPC_MESSAGE_HANDLER(WebSocketMsg_SendFrame, OnSendFrame)
     IPC_MESSAGE_HANDLER(WebSocketMsg_FlowControl, OnFlowControl)
     IPC_MESSAGE_HANDLER(WebSocketMsg_DropChannel, OnDropChannel)
+    IPC_MESSAGE_HANDLER(WebSocketHostMsg_LoaderTransferTest_Setup,
+                        OnLoaderTransferTest_Setup)
+    IPC_MESSAGE_HANDLER(WebSocketHostMsg_LoaderTransferTest_Send,
+                        OnLoaderTransferTest_Send)
+    IPC_MESSAGE_HANDLER(WebSocketHostMsg_LoaderTransferTest_Close,
+                        OnLoaderTransferTest_Close)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void WebSocketHost::OnAddChannelRequest(
     const GURL& socket_url,
     const std::vector<std::string>& requested_protocols,
     const url::Origin& origin,
     int render_frame_id) {
@@ skipping 170 matching lines @@
 
     default:
       ignore_result(dispatcher_->NotifyFailure(
           routing_id_,
           "Failed to load Blob: error code = " + net::ErrorToString(result)));
       // |this| is destroyed here.
       return;
   }
 }
 
+void WebSocketHost::OnLoaderTransferTest_Setup(int buffer_size) {
+  //  fprintf(stderr, "%s\n", __PRETTY_FUNCTION__);
+  loader_test_num_read_bytes_ = 0;
+  base::SharedMemoryHandle handle;
+  shared_memory_.reset(new base::SharedMemory);
+  shared_memory_->CreateAndMapAnonymous(buffer_size);
+  shared_memory_->ShareToProcess(dispatcher_->PeerHandle(), &handle);
+  dispatcher_->SendLoaderTransferTest_SetDataBuffer(routing_id_, handle);
+}
+
+void WebSocketHost::OnLoaderTransferTest_Send(int offset, int size) {
+  // fprintf(stderr, "%s offset = %d, size = %d\n", __PRETTY_FUNCTION__, offset, size);
+  loader_test_num_read_bytes_ += size;
+  dispatcher_->SendLoaderTransferTest_Ack(routing_id_);
+
+  // const char* p = static_cast<const char*>(shared_memory_->memory()) + offset;
+  // fprintf(stderr, "*p = 0x%02x\n", *p);
+}
+
+void WebSocketHost::OnLoaderTransferTest_Close() {
+  //  fprintf(stderr, "%s\n", __PRETTY_FUNCTION__);
+  shared_memory_ = nullptr;
+  dispatcher_->SendLoaderTransferTest_Done(routing_id_);
+  fprintf(stderr, "%s done: read %zu bytes\n", __PRETTY_FUNCTION__,
+          loader_test_num_read_bytes_);
+}
+
+
 }  // namespace content