| // 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 "content/public/browser/browser_thread.h" |
| #include "content/public/browser/resource_context.h" |
| #include "content/public/browser/resource_request_info.h" |
| #include "net/url_request/url_request.h" |
| #include "url/gurl.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)); |
| } |
| |
| InstantIOContext::~InstantIOContext() { |
| } |
| |
| // static |
| 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.get())); |
| } |
| |
| // static |
| void InstantIOContext::AddInstantProcessOnIO( |
| scoped_refptr<InstantIOContext> instant_io_context, |
| int process_id) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| instant_io_context->process_ids_.insert(process_id); |
| } |
| |
| // 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); |
| } |
| |
| // static |
| void InstantIOContext::ClearInstantProcessesOnIO( |
| scoped_refptr<InstantIOContext> instant_io_context) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| instant_io_context->process_ids_.clear(); |
| } |
| |
| // static |
| bool InstantIOContext::ShouldServiceRequest(const net::URLRequest* request) { |
| const content::ResourceRequestInfo* info = |
| content::ResourceRequestInfo::ForRequest(request); |
| if (!info) |
| return false; |
| |
| InstantIOContext* instant_io_context = GetDataForRequest(request); |
| if (!instant_io_context) |
| return false; |
| |
| int process_id = -1; |
| int render_frame_id = -1; |
| if (info->GetAssociatedRenderFrame(&process_id, &render_frame_id) && |
| instant_io_context->IsInstantProcess(process_id)) |
| return true; |
| return false; |
| } |
| |
| bool InstantIOContext::IsInstantProcess(int process_id) const { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| return process_ids_.find(process_id) != process_ids_.end(); |
| } |