net-internals: Add buttons to close idle and flush all sockets

chrome/browser/ui/webui/net_internals_ui.cc

 // Copyright (c) 2011 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/ui/webui/net_internals_ui.h"
 
 #include <algorithm>
 #include <string>
 #include <utility>
 #include <vector>
@@ skipping 262 matching lines @@
   void OnClearBadProxies(const ListValue* list);
   void OnGetHostResolverInfo(const ListValue* list);
   void OnClearHostResolverCache(const ListValue* list);
   void OnEnableIPv6(const ListValue* list);
   void OnStartConnectionTests(const ListValue* list);
   void OnHSTSQuery(const ListValue* list);
   void OnHSTSAdd(const ListValue* list);
   void OnHSTSDelete(const ListValue* list);
   void OnGetHttpCacheInfo(const ListValue* list);
   void OnGetSocketPoolInfo(const ListValue* list);
+  void OnCloseIdleSockets(const ListValue* list);
+  void OnFlushSocketPools(const ListValue* list);
   void OnGetSpdySessionInfo(const ListValue* list);
   void OnGetSpdyStatus(const ListValue* list);
   void OnGetSpdyAlternateProtocolMappings(const ListValue* list);
 #ifdef OS_WIN
   void OnGetServiceProviders(const ListValue* list);
 #endif
 
   void OnSetLogLevel(const ListValue* list);
 
   // ChromeNetLog::ThreadSafeObserver implementation:
@@ skipping 257 matching lines @@
   web_ui_->RegisterMessageCallback(
       "hstsDelete",
       proxy_->CreateCallback(&IOThreadImpl::OnHSTSDelete));
   web_ui_->RegisterMessageCallback(
       "getHttpCacheInfo",
       proxy_->CreateCallback(&IOThreadImpl::OnGetHttpCacheInfo));
   web_ui_->RegisterMessageCallback(
       "getSocketPoolInfo",
       proxy_->CreateCallback(&IOThreadImpl::OnGetSocketPoolInfo));
   web_ui_->RegisterMessageCallback(
+      "closeIdleSockets",
+      proxy_->CreateCallback(&IOThreadImpl::OnCloseIdleSockets));
+  web_ui_->RegisterMessageCallback(
+      "flushSocketPools",
+      proxy_->CreateCallback(&IOThreadImpl::OnFlushSocketPools));
+  web_ui_->RegisterMessageCallback(
       "getSpdySessionInfo",
       proxy_->CreateCallback(&IOThreadImpl::OnGetSpdySessionInfo));
   web_ui_->RegisterMessageCallback(
       "getSpdyStatus",
       proxy_->CreateCallback(&IOThreadImpl::OnGetSpdyStatus));
   web_ui_->RegisterMessageCallback(
       "getSpdyAlternateProtocolMappings",
       proxy_->CreateCallback(
           &IOThreadImpl::OnGetSpdyAlternateProtocolMappings));
 #ifdef OS_WIN
@@ skipping 572 matching lines @@
   net::HttpNetworkSession* http_network_session =
       GetHttpNetworkSession(context_getter_->GetURLRequestContext());
 
   Value* socket_pool_info = NULL;
   if (http_network_session)
     socket_pool_info = http_network_session->SocketPoolInfoToValue();
 
   CallJavascriptFunction(L"g_browser.receivedSocketPoolInfo", socket_pool_info);
 }
 
+
+void NetInternalsMessageHandler::IOThreadImpl::OnFlushSocketPools(
+    const ListValue* list) {
+  net::HttpNetworkSession* http_network_session =
+      GetHttpNetworkSession(context_getter_->GetURLRequestContext());
+
+  if (http_network_session)
+    http_network_session->CloseAllConnections();
+}
+
+void NetInternalsMessageHandler::IOThreadImpl::OnCloseIdleSockets(
+    const ListValue* list) {
+  net::HttpNetworkSession* http_network_session =
+      GetHttpNetworkSession(context_getter_->GetURLRequestContext());
+
+  if (http_network_session)
+    http_network_session->CloseIdleConnections();
+}
+
 void NetInternalsMessageHandler::IOThreadImpl::OnGetSpdySessionInfo(
     const ListValue* list) {
   net::HttpNetworkSession* http_network_session =
       GetHttpNetworkSession(context_getter_->GetURLRequestContext());
 
   Value* spdy_info = NULL;
   if (http_network_session) {
     spdy_info = http_network_session->SpdySessionPoolInfoToValue();
   }
 
@@ skipping 217 matching lines @@
 ////////////////////////////////////////////////////////////////////////////////
 
 NetInternalsUI::NetInternalsUI(TabContents* contents) : WebUI(contents) {
   AddMessageHandler((new NetInternalsMessageHandler())->Attach(this));
 
   NetInternalsHTMLSource* html_source = new NetInternalsHTMLSource();
 
   // Set up the chrome://net-internals/ source.
   contents->profile()->GetChromeURLDataManager()->AddDataSource(html_source);
 }