src/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() : main_frame_host_(NULL) {
  36 }
  37
  38 FrameNavigationState::~FrameNavigationState() {}
  39
  40 bool FrameNavigationState::CanSendEvents(
  41     content::RenderFrameHost* frame_host) const {
  42   FrameHostToStateMap::const_iterator it =
  43       frame_host_state_map_.find(frame_host);
  44   if (it == frame_host_state_map_.end() || it->second.error_occurred) {
  45     return false;
  46   }
  47   return IsValidUrl(it->second.url);
  48 }
  49
  50 bool FrameNavigationState::IsValidUrl(const GURL& url) const {
  51   for (unsigned i = 0; i < arraysize(kValidSchemes); ++i) {
  52     if (url.scheme() == kValidSchemes[i])
  53       return true;
  54   }
  55   // Allow about:blank and about:srcdoc.
  56   if (url.spec() == url::kAboutBlankURL ||
  57       url.spec() == content::kAboutSrcDocURL) {
  58     return true;
  59   }
  60   return allow_extension_scheme_ && url.scheme() == kExtensionScheme;
  61 }
  62
  63 void FrameNavigationState::TrackFrame(content::RenderFrameHost* frame_host,
  64                                       const GURL& url,
  65                                       bool is_error_page,
  66                                       bool is_iframe_srcdoc) {
  67   FrameState& frame_state = frame_host_state_map_[frame_host];
  68   frame_state.error_occurred = is_error_page;
  69   frame_state.url = url;
  70   frame_state.is_iframe_srcdoc = is_iframe_srcdoc;
  71   DCHECK(!is_iframe_srcdoc || url == GURL(url::kAboutBlankURL));
  72   frame_state.is_navigating = true;
  73   frame_state.is_committed = false;
  74   frame_state.is_server_redirected = false;
  75   frame_state.is_parsing = true;
  76   frame_hosts_.insert(frame_host);
  77 }
  78
  79 void FrameNavigationState::FrameDetached(content::RenderFrameHost* frame_host) {
  80   if (frame_host == main_frame_host_)
  81     main_frame_host_ = NULL;
  82   frame_host_state_map_.erase(frame_host);
  83   frame_hosts_.erase(frame_host);
  84 }
  85
  86 void FrameNavigationState::StopTrackingFramesInRVH(
  87     content::RenderViewHost* render_view_host,
  88     content::RenderFrameHost* frame_host_to_skip) {
  89   for (std::set<content::RenderFrameHost*>::iterator it = frame_hosts_.begin();
  90        it != frame_hosts_.end();) {
  91     if ((*it)->GetRenderViewHost() != render_view_host ||
  92         *it == frame_host_to_skip) {
  93       ++it;
  94       continue;
  95     }
  96     if (*it == main_frame_host_)
  97       main_frame_host_ = NULL;
  98     frame_host_state_map_.erase(*it);
  99     frame_hosts_.erase(it++);
 100   }
 101 }
 102
 103 void FrameNavigationState::UpdateFrame(content::RenderFrameHost* frame_host,
 104                                        const GURL& url) {
 105   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 106   if (it == frame_host_state_map_.end()) {
 107     NOTREACHED();
 108     return;
 109   }
 110   it->second.url = url;
 111 }
 112
 113 bool FrameNavigationState::IsValidFrame(
 114     content::RenderFrameHost* frame_host) const {
 115   return frame_host_state_map_.find(frame_host) != frame_host_state_map_.end();
 116 }
 117
 118 GURL FrameNavigationState::GetUrl(content::RenderFrameHost* frame_host) const {
 119   FrameHostToStateMap::const_iterator it =
 120       frame_host_state_map_.find(frame_host);
 121   if (it == frame_host_state_map_.end()) {
 122     NOTREACHED();
 123     return GURL();
 124   }
 125   if (it->second.is_iframe_srcdoc)
 126     return GURL(content::kAboutSrcDocURL);
 127   return it->second.url;
 128 }
 129
 130 content::RenderFrameHost* FrameNavigationState::GetLastCommittedMainFrameHost()
 131     const {
 132   return main_frame_host_;
 133 }
 134
 135 void FrameNavigationState::SetErrorOccurredInFrame(
 136     content::RenderFrameHost* frame_host) {
 137   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 138   if (it == frame_host_state_map_.end()) {
 139     NOTREACHED();
 140     return;
 141   }
 142   it->second.error_occurred = true;
 143 }
 144
 145 bool FrameNavigationState::GetErrorOccurredInFrame(
 146     content::RenderFrameHost* frame_host) const {
 147   FrameHostToStateMap::const_iterator it =
 148       frame_host_state_map_.find(frame_host);
 149   DCHECK(it != frame_host_state_map_.end());
 150   return it == frame_host_state_map_.end() || it->second.error_occurred;
 151 }
 152
 153 void FrameNavigationState::SetNavigationCompleted(
 154     content::RenderFrameHost* frame_host) {
 155   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 156   if (it == frame_host_state_map_.end()) {
 157     NOTREACHED();
 158     return;
 159   }
 160   it->second.is_navigating = false;
 161 }
 162
 163 bool FrameNavigationState::GetNavigationCompleted(
 164     content::RenderFrameHost* frame_host) const {
 165   FrameHostToStateMap::const_iterator it =
 166       frame_host_state_map_.find(frame_host);
 167   DCHECK(it != frame_host_state_map_.end());
 168   return it == frame_host_state_map_.end() || !it->second.is_navigating;
 169 }
 170
 171 void FrameNavigationState::SetParsingFinished(
 172     content::RenderFrameHost* frame_host) {
 173   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 174   if (it == frame_host_state_map_.end()) {
 175     NOTREACHED();
 176     return;
 177   }
 178   it->second.is_parsing = false;
 179 }
 180
 181 bool FrameNavigationState::GetParsingFinished(
 182     content::RenderFrameHost* frame_host) const {
 183   FrameHostToStateMap::const_iterator it =
 184       frame_host_state_map_.find(frame_host);
 185   DCHECK(it != frame_host_state_map_.end());
 186   return it == frame_host_state_map_.end() || !it->second.is_parsing;
 187 }
 188
 189 void FrameNavigationState::SetNavigationCommitted(
 190     content::RenderFrameHost* frame_host) {
 191   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 192   if (it == frame_host_state_map_.end()) {
 193     NOTREACHED();
 194     return;
 195   }
 196   it->second.is_committed = true;
 197   if (!frame_host->GetParent())
 198     main_frame_host_ = frame_host;
 199 }
 200
 201 bool FrameNavigationState::GetNavigationCommitted(
 202     content::RenderFrameHost* frame_host) const {
 203   FrameHostToStateMap::const_iterator it =
 204       frame_host_state_map_.find(frame_host);
 205   DCHECK(it != frame_host_state_map_.end());
 206   return it != frame_host_state_map_.end() && it->second.is_committed;
 207 }
 208
 209 void FrameNavigationState::SetIsServerRedirected(
 210     content::RenderFrameHost* frame_host) {
 211   FrameHostToStateMap::iterator it = frame_host_state_map_.find(frame_host);
 212   if (it == frame_host_state_map_.end()) {
 213     NOTREACHED();
 214     return;
 215   }
 216   it->second.is_server_redirected = true;
 217 }
 218
 219 bool FrameNavigationState::GetIsServerRedirected(
 220     content::RenderFrameHost* frame_host) const {
 221   FrameHostToStateMap::const_iterator it =
 222       frame_host_state_map_.find(frame_host);
 223   DCHECK(it != frame_host_state_map_.end());
 224   return it != frame_host_state_map_.end() && it->second.is_server_redirected;
 225 }
 226
 227 }  // namespace extensions