Most visited thumbnails and favicons need id-based urls

chrome/browser/instant/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/instant/instant_io_context.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 {
 
 // Retrieves the Instant data from the |context|'s named user-data.
 InstantIOContext* GetDataForResourceContext(
     content::ResourceContext* context) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   return base::UserDataAdapter<InstantIOContext>::Get(
       context, InstantIOContext::kInstantIOContextKeyName);
 }
 
+InstantIOContext* GetDataForRequest(const net::URLRequest* request) {
+  const content::ResourceRequestInfo* info =
+      content::ResourceRequestInfo::ForRequest(request);
+  if (!info)
+    return NULL;
+
+  return GetDataForResourceContext(info->GetContext());
+}
+
 }  // namespace
 
 const char InstantIOContext::kInstantIOContextKeyName[] = "instant_io_context";
 
 InstantIOContext::InstantIOContext() {
   // The InstantIOContext is created on the UI thread but is accessed
   // on the IO thread.
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 }
 
@@ skipping 24 matching lines @@
 }
 
 // static
 void InstantIOContext::ClearInstantProcessesOnIO(
     scoped_refptr<InstantIOContext> instant_io_context) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   instant_io_context->process_ids_.clear();
 }
 
 // static
+void InstantIOContext::AddRestrictedIDOnIO(
+    scoped_refptr<InstantIOContext> instant_io_context,
+    uint64 restricted_id, const GURL& url) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+  instant_io_context->id_to_url_map_[restricted_id] = url;
+}
+
+// static
 bool InstantIOContext::ShouldServiceRequest(const net::URLRequest* request) {
   const content::ResourceRequestInfo* info =
       content::ResourceRequestInfo::ForRequest(request);
   if (!info)
     return false;
 
-  InstantIOContext* instant_io_context =
-      GetDataForResourceContext(info->GetContext());
+  InstantIOContext* instant_io_context = GetDataForRequest(request);
   if (!instant_io_context)
     return false;
 
   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
+bool InstantIOContext::GetURLForRestrictedId(const net::URLRequest* request,
+                                             uint64 restricted_id, GURL* url) {
+  InstantIOContext* instant_io_context = GetDataForRequest(request);
+  if (!instant_io_context) {
+    *url = GURL();
+    return false;
+  }
+
+  return instant_io_context->GetURLForRestrictedId(restricted_id, url);
+}
+
 bool InstantIOContext::IsInstantProcess(int process_id) const {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   return process_ids_.count(process_id) != 0;
 }
+
+bool InstantIOContext::GetURLForRestrictedId(uint64 restricted_id, GURL* url) {
+  std::map<uint64, GURL>::iterator it = id_to_url_map_.find(restricted_id);
+  if (it != id_to_url_map_.end()) {
+    *url = it->second;
+    return true;
+  }
+  *url = GURL();
+  return false;
+}