Fixing iframe jank in the local omnibox popup.

chrome/browser/search/instant_io_context.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 "chrome/browser/search/instant_io_context.h"
 
+#include "base/string_util.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/resource_context.h"
 #include "content/public/browser/resource_request_info.h"
 #include "googleurl/src/gurl.h"
 #include "net/url_request/url_request.h"
 
 using content::BrowserThread;
 
 namespace {
 
@@ skipping 32 matching lines @@
 void InstantIOContext::SetUserDataOnIO(
     content::ResourceContext* resource_context,
     scoped_refptr<InstantIOContext> instant_io_context) {
   resource_context->SetUserData(
       InstantIOContext::kInstantIOContextKeyName,
       new base::UserDataAdapter<InstantIOContext>(instant_io_context));
 }
 
 // static
 void InstantIOContext::AddInstantProcessOnIO(
-    scoped_refptr<InstantIOContext> instant_io_context, int process_id) {
+    scoped_refptr<InstantIOContext> instant_io_context,
+    int process_id,
+    const GURL& site_url) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   instant_io_context->process_ids_.insert(process_id);
+  instant_io_context->process_sites_[process_id] = site_url.spec();
 }
 
 // static
 void InstantIOContext::RemoveInstantProcessOnIO(
     scoped_refptr<InstantIOContext> instant_io_context, int process_id) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   instant_io_context->process_ids_.erase(process_id);
+  instant_io_context->process_sites_.erase(process_id);
 }
 
 // static
 void InstantIOContext::ClearInstantProcessesOnIO(
     scoped_refptr<InstantIOContext> instant_io_context) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   instant_io_context->process_ids_.clear();
+  instant_io_context->process_sites_.clear();
 }
 
 // static
 void InstantIOContext::AddMostVisitedItemsOnIO(
     scoped_refptr<InstantIOContext> instant_io_context,
     std::vector<InstantMostVisitedItemIDPair> items) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   instant_io_context->most_visited_item_cache_.AddItemsWithRestrictedID(items);
 }
 
@@ skipping 11 matching lines @@
 
   int process_id = -1;
   int render_view_id = -1;
   if (info->GetAssociatedRenderView(&process_id, &render_view_id) &&
       instant_io_context->IsInstantProcess(process_id))
     return true;
   return false;
 }
 
 // static
+std::string InstantIOContext::GetExpectedOrigin(
+    const net::URLRequest* request) {
+  const content::ResourceRequestInfo* info =
+      content::ResourceRequestInfo::ForRequest(request);
+  if (!info)
+    return std::string();
+
+  InstantIOContext* instant_io_context = GetDataForRequest(request);
+  if (!instant_io_context)
+    return std::string();
+
+  int process_id = -1;
+  int render_view_id = -1;
+  if (info->GetAssociatedRenderView(&process_id, &render_view_id)) {
+    GURL site(instant_io_context->GetSiteForInstantProcess(process_id));
+    std::string origin = site.GetOrigin().spec();
+    // Origin should not include a trailing slash. That is part of the path.
+    TrimString(origin, "/", &origin);
+    return origin;
+  }
+  return std::string();
+}
+
+// static
 bool InstantIOContext::GetURLForMostVisitedItemID(
     const net::URLRequest* request,
     InstantRestrictedID most_visited_item_id,
     GURL* url) {
   InstantIOContext* instant_io_context = GetDataForRequest(request);
   if (!instant_io_context) {
     *url = GURL();
     return false;
   }
 
   return instant_io_context->GetURLForMostVisitedItemID(most_visited_item_id,
                                                         url);
 }
 
 bool InstantIOContext::IsInstantProcess(int process_id) const {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
-  return process_ids_.count(process_id) != 0;
+  return process_ids_.find(process_id) != process_ids_.end();
+}
+
+// Gets the site URL associated with the given Instant process if any.
+std::string InstantIOContext::GetSiteForInstantProcess(int process_id) const {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+  std::map<int, std::string>::const_iterator site =
+      process_sites_.find(process_id);
+  if (site != process_sites_.end())
+    return site->second;
+  return std::string();
 }
 
 bool InstantIOContext::GetURLForMostVisitedItemID(
     InstantRestrictedID most_visited_item_id,
     GURL* url) const {
   InstantMostVisitedItem item;
   if (most_visited_item_cache_.GetItemWithRestrictedID(most_visited_item_id,
                                                        &item)) {
     *url = item.url;
     return true;
   }
 
   *url = GURL();
   return false;
 }