chrome/browser/extensions/api/web_navigation/frame_navigation_state.cc

   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 #include "chrome/browser/extensions/api/web_navigation/frame_navigation_state.h"
   6
   7 #include "base/logging.h"
   8 #include "chrome/common/url_constants.h"
   9 #include "content/public/browser/render_frame_host.h"
  10 #include "extensions/common/constants.h"
  11
  12 namespace extensions {
  13
  14 namespace {
  15
  16 // URL schemes for which we'll send events.
  17 const char* const kValidSchemes[] = {
  18     content::kChromeUIScheme,
  19     url::kHttpScheme,
  20     url::kHttpsScheme,
  21     url::kFileScheme,
  22     url::kFtpScheme,
  23     url::kJavaScriptScheme,
  24     url::kDataScheme,
  25     url::kFileSystemScheme,
  26 };
  27
  28 }  // namespace
  29
  30 FrameNavigationState::FrameState::FrameState() {}
  31
  32 // static
  33 bool FrameNavigationState::allow_extension_scheme_ = false;
  34
  35 FrameNavigationState::FrameNavigationState() {
  36 }
  37
  38 FrameNavigationState::~FrameNavigationState() {}
  39
  40 // static
  41 bool FrameNavigationState::IsValidUrl(const GURL& url) {
  42   for (unsigned i = 0; i < arraysize(kValidSchemes); ++i) {
  43     if (url.scheme() == kValidSchemes[i])
  44       return true;
  45   }
  46   // Allow about:blank and about:srcdoc.
  47   if (url.spec() == url::kAboutBlankURL ||
  48       url.spec() == content::kAboutSrcDocURL) {
  49     return true;
  50   }
  51   return allow_extension_scheme_ && url.scheme() == kExtensionScheme;
  52 }
  53
  54 bool FrameNavigationState::CanSendEvents(
  55     content::RenderFrameHost* frame_host) const {
  56   FrameHostToStateMap::const_iterator it =
  57       frame_host_state_map_.find(frame_host);
  58   if (it == frame_host_state_map_.end() || it->second.error_occurred) {
  59     return false;
  60   }
  61   return IsValidUrl(it->second.url);
  62 }
  63
  64 void FrameNavigationState::StartTrackingNavigation(
  65     content::RenderFrameHost* frame_host,
  66     const GURL& url,
  67     bool is_error_page,
  68     bool is_iframe_srcdoc) {
  69   FrameState& frame_state = frame_host_state_map_[frame_host];
  70   frame_state.error_occurred = is_error_page;
  71   frame_state.url = url;
  72   frame_state.is_iframe_srcdoc = is_iframe_srcdoc;
  73   DCHECK(!is_iframe_srcdoc || url == GURL(url::kAboutBlankURL));
  74   frame_state.is_navigating = true;
  75   frame_state.is_committed = false;
  76   frame_state.is_server_redirected = false;
  77   frame_state.is_parsing = true;
  78 }
  79
  80 void FrameNavigationState::FrameHostCreated(
  81     content::RenderFrameHost* frame_host) {
  82   frame_hosts_.insert(frame_host);
  83 }
  84
  85 void FrameNavigationState::FrameHostDeleted(
  86     content::RenderFrameHost* frame_host) {
  87   frame_host_state_map_.erase(frame_host);
  88   frame_hosts_.erase(frame_host);
  89 }
  90
  91 void FrameNavigationState::UpdateFrame(content::RenderFrameHost* frame_host,
  92                                        const GURL& url) {
  93   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
  94   if (it == frame_host_state_map_.end()) {
  95     NOTREACHED();
  96     return;
  97   }
  98   it->second.url = url;
  99 }
 100
 101 bool FrameNavigationState::IsValidFrame(
 102     content::RenderFrameHost* frame_host) const {
 103   return frame_host_state_map_.find(frame_host) != frame_host_state_map_.end();
 104 }
 105
 106 GURL FrameNavigationState::GetUrl(content::RenderFrameHost* frame_host) const {
 107   FrameHostToStateMap::const_iterator it =
 108       frame_host_state_map_.find(frame_host);
 109   if (it == frame_host_state_map_.end()) {
 110     NOTREACHED();
 111     return GURL();
 112   }
 113   if (it->second.is_iframe_srcdoc)
 114     return GURL(content::kAboutSrcDocURL);
 115   return it->second.url;
 116 }
 117
 118 void FrameNavigationState::SetErrorOccurredInFrame(
 119     content::RenderFrameHost* frame_host) {
 120   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 121   if (it == frame_host_state_map_.end()) {
 122     NOTREACHED();
 123     return;
 124   }
 125   it->second.error_occurred = true;
 126 }
 127
 128 bool FrameNavigationState::GetErrorOccurredInFrame(
 129     content::RenderFrameHost* frame_host) const {
 130   FrameHostToStateMap::const_iterator it =
 131       frame_host_state_map_.find(frame_host);
 132   DCHECK(it != frame_host_state_map_.end());
 133   return it == frame_host_state_map_.end() || it->second.error_occurred;
 134 }
 135
 136 void FrameNavigationState::SetNavigationCompleted(
 137     content::RenderFrameHost* frame_host) {
 138   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 139   if (it == frame_host_state_map_.end()) {
 140     NOTREACHED();
 141     return;
 142   }
 143   it->second.is_navigating = false;
 144 }
 145
 146 bool FrameNavigationState::GetNavigationCompleted(
 147     content::RenderFrameHost* frame_host) const {
 148   FrameHostToStateMap::const_iterator it =
 149       frame_host_state_map_.find(frame_host);
 150   DCHECK(it != frame_host_state_map_.end());
 151   return it == frame_host_state_map_.end() || !it->second.is_navigating;
 152 }
 153
 154 void FrameNavigationState::SetParsingFinished(
 155     content::RenderFrameHost* frame_host) {
 156   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 157   if (it == frame_host_state_map_.end()) {
 158     NOTREACHED();
 159     return;
 160   }
 161   it->second.is_parsing = false;
 162 }
 163
 164 bool FrameNavigationState::GetParsingFinished(
 165     content::RenderFrameHost* frame_host) const {
 166   FrameHostToStateMap::const_iterator it =
 167       frame_host_state_map_.find(frame_host);
 168   DCHECK(it != frame_host_state_map_.end());
 169   return it == frame_host_state_map_.end() || !it->second.is_parsing;
 170 }
 171
 172 void FrameNavigationState::SetNavigationCommitted(
 173     content::RenderFrameHost* frame_host) {
 174   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 175   if (it == frame_host_state_map_.end()) {
 176     NOTREACHED();
 177     return;
 178   }
 179   it->second.is_committed = true;
 180 }
 181
 182 bool FrameNavigationState::GetNavigationCommitted(
 183     content::RenderFrameHost* frame_host) const {
 184   FrameHostToStateMap::const_iterator it =
 185       frame_host_state_map_.find(frame_host);
 186   DCHECK(it != frame_host_state_map_.end());
 187   return it != frame_host_state_map_.end() && it->second.is_committed;
 188 }
 189
 190 void FrameNavigationState::SetIsServerRedirected(
 191     content::RenderFrameHost* frame_host) {
 192   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 193   if (it == frame_host_state_map_.end()) {
 194     NOTREACHED();
 195     return;
 196   }
 197   it->second.is_server_redirected = true;
 198 }
 199
 200 bool FrameNavigationState::GetIsServerRedirected(
 201     content::RenderFrameHost* frame_host) const {
 202   FrameHostToStateMap::const_iterator it =
 203       frame_host_state_map_.find(frame_host);
 204   DCHECK(it != frame_host_state_map_.end());
 205   return it != frame_host_state_map_.end() && it->second.is_server_redirected;
 206 }
 207
 208 }  // namespace extensions