1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef CHROME_BROWSER_NET_NET_ERROR_TAB_HELPER_H_ 6 #define CHROME_BROWSER_NET_NET_ERROR_TAB_HELPER_H_ 7 8 #include "base/basictypes.h" 9 #include "base/bind.h" 10 #include "base/compiler_specific.h" 11 #include "base/memory/weak_ptr.h" 12 #include "base/prefs/pref_member.h" 13 #include "chrome/browser/net/dns_probe_service.h" 14 #include "chrome/common/net/net_error_info.h" 15 #include "content/public/browser/web_contents_observer.h" 16 #include "content/public/browser/web_contents_user_data.h" 17 18 namespace chrome_browser_net { 19 20 // A TabHelper that monitors loads for certain types of network errors and 21 // does interesting things with them. Currently, starts DNS probes using the 22 // DnsProbeService whenever a page fails to load with a DNS-related error. 23 class NetErrorTabHelper 24 : public content::WebContentsObserver, 25 public content::WebContentsUserData<NetErrorTabHelper> { 26 public: 27 enum TestingState { 28 TESTING_DEFAULT, 29 TESTING_FORCE_DISABLED, 30 TESTING_FORCE_ENABLED 31 }; 32 33 typedef base::Callback<void(chrome_common_net::DnsProbeStatus)> 34 DnsProbeStatusSnoopCallback; 35 36 virtual ~NetErrorTabHelper(); 37 38 static void set_state_for_testing(TestingState testing_state); 39 40 // Sets a callback that will be called immediately after the helper sends 41 // a NetErrorHelper IPC. (Used by the DNS probe browser test to know when to 42 // check the error page for updates, instead of polling.) set_dns_probe_status_snoop_callback_for_testing(const DnsProbeStatusSnoopCallback & dns_probe_status_snoop_callback)43 void set_dns_probe_status_snoop_callback_for_testing( 44 const DnsProbeStatusSnoopCallback& dns_probe_status_snoop_callback) { 45 dns_probe_status_snoop_callback_ = dns_probe_status_snoop_callback; 46 } 47 48 // content::WebContentsObserver implementation. 49 virtual void DidStartNavigationToPendingEntry( 50 const GURL& url, 51 content::NavigationController::ReloadType reload_type) OVERRIDE; 52 53 virtual void DidStartProvisionalLoadForFrame( 54 int64 frame_id, 55 int64 parent_frame_id, 56 bool is_main_frame, 57 const GURL& validated_url, 58 bool is_error_page, 59 bool is_iframe_srcdoc, 60 content::RenderViewHost* render_view_host) OVERRIDE; 61 62 virtual void DidCommitProvisionalLoadForFrame( 63 int64 frame_id, 64 const base::string16& frame_unique_name, 65 bool is_main_frame, 66 const GURL& url, 67 content::PageTransition transition_type, 68 content::RenderViewHost* render_view_host) OVERRIDE; 69 70 virtual void DidFailProvisionalLoad( 71 int64 frame_id, 72 const base::string16& frame_unique_name, 73 bool is_main_frame, 74 const GURL& validated_url, 75 int error_code, 76 const base::string16& error_description, 77 content::RenderViewHost* render_view_host) OVERRIDE; 78 79 protected: 80 // |contents| is the WebContents of the tab this NetErrorTabHelper is 81 // attached to. 82 explicit NetErrorTabHelper(content::WebContents* contents); 83 virtual void StartDnsProbe(); 84 virtual void SendInfo(); 85 void OnDnsProbeFinished(chrome_common_net::DnsProbeStatus result); 86 dns_probe_status()87 chrome_common_net::DnsProbeStatus dns_probe_status() const { 88 return dns_probe_status_; 89 } 90 91 private: 92 friend class content::WebContentsUserData<NetErrorTabHelper>; 93 94 void OnMainFrameDnsError(); 95 96 void InitializePref(content::WebContents* contents); 97 bool ProbesAllowed() const; 98 99 base::WeakPtrFactory<NetErrorTabHelper> weak_factory_; 100 101 // True if the last provisional load that started was for an error page. 102 bool is_error_page_; 103 104 // True if the helper has seen a main frame page load fail with a DNS error, 105 // but has not yet seen a new page commit successfully afterwards. 106 bool dns_error_active_; 107 108 // True if the helper has seen an error page commit while |dns_error_active_| 109 // is true. (This should never be true if |dns_error_active_| is false.) 110 bool dns_error_page_committed_; 111 112 // The status of a DNS probe that may or may not have started or finished. 113 // Since the renderer can change out from under the helper (in cross-process 114 // navigations), it re-sends the status whenever an error page commits. 115 chrome_common_net::DnsProbeStatus dns_probe_status_; 116 117 // Optional callback for browser test to snoop on outgoing NetErrorInfo IPCs. 118 DnsProbeStatusSnoopCallback dns_probe_status_snoop_callback_; 119 120 // "Use a web service to resolve navigation errors" preference is required 121 // to allow probes. 122 BooleanPrefMember resolve_errors_with_web_service_; 123 124 DISALLOW_COPY_AND_ASSIGN(NetErrorTabHelper); 125 }; 126 127 } // namespace chrome_browser_net 128 129 #endif // CHROME_BROWSER_NET_NET_ERROR_TAB_HELPER_H_ 130