src/android_webview/browser/aw_request_interceptor.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 "android_webview/browser/aw_request_interceptor.h"
   6
   7 #include "android_webview/browser/aw_contents_io_thread_client.h"
   8 #include "android_webview/browser/intercepted_request_data.h"
   9 #include "base/android/jni_string.h"
  10 #include "base/memory/scoped_ptr.h"
  11 #include "content/public/browser/browser_thread.h"
  12 #include "content/public/browser/render_view_host.h"
  13 #include "content/public/browser/resource_request_info.h"
  14 #include "net/url_request/url_request.h"
  15 #include "net/url_request/url_request_context.h"
  16 #include "net/url_request/url_request_context_getter.h"
  17 #include "net/url_request/url_request_job.h"
  18
  19 using content::BrowserThread;
  20 using content::RenderViewHost;
  21 using content::ResourceRequestInfo;
  22
  23 namespace android_webview {
  24
  25 namespace {
  26
  27 const void* kURLRequestUserDataKey = &kURLRequestUserDataKey;
  28
  29 class URLRequestUserData : public base::SupportsUserData::Data {
  30  public:
  31     URLRequestUserData(
  32         scoped_ptr<InterceptedRequestData> intercepted_request_data)
  33         : intercepted_request_data_(intercepted_request_data.Pass()) {
  34     }
  35
  36     static URLRequestUserData* Get(net::URLRequest* request) {
  37       return reinterpret_cast<URLRequestUserData*>(
  38           request->GetUserData(kURLRequestUserDataKey));
  39     }
  40
  41     const InterceptedRequestData* intercepted_request_data() const {
  42       return intercepted_request_data_.get();
  43     }
  44
  45  private:
  46   scoped_ptr<InterceptedRequestData> intercepted_request_data_;
  47 };
  48
  49 } // namespace
  50
  51 AwRequestInterceptor::AwRequestInterceptor() {
  52 }
  53
  54 AwRequestInterceptor::~AwRequestInterceptor() {
  55 }
  56
  57 scoped_ptr<InterceptedRequestData>
  58 AwRequestInterceptor::QueryForInterceptedRequestData(
  59     const GURL& location,
  60     net::URLRequest* request) const {
  61   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
  62   int render_process_id, render_view_id;
  63   if (!ResourceRequestInfo::GetRenderViewForRequest(
  64       request, &render_process_id, &render_view_id))
  65     return scoped_ptr<InterceptedRequestData>();
  66
  67   scoped_ptr<AwContentsIoThreadClient> io_thread_client =
  68     AwContentsIoThreadClient::FromID(render_process_id, render_view_id);
  69
  70   if (!io_thread_client.get())
  71     return scoped_ptr<InterceptedRequestData>();
  72
  73   return io_thread_client->ShouldInterceptRequest(location, request).Pass();
  74 }
  75
  76 net::URLRequestJob* AwRequestInterceptor::MaybeCreateJob(
  77     net::URLRequest* request,
  78     net::NetworkDelegate* network_delegate) const {
  79   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
  80
  81   // See if we've already found out the intercepted_request_data for this
  82   // request.
  83   // This is done not only for efficiency reasons, but also for correctness
  84   // as it is possible for the Interceptor chain to be invoked more than once
  85   // (in which case we don't want to query the embedder multiple times).
  86   URLRequestUserData* user_data = URLRequestUserData::Get(request);
  87
  88   if (!user_data) {
  89     // To ensure we only query the embedder once, we rely on the fact that the
  90     // user_data object will be created and attached to the URLRequest after a
  91     // call to QueryForInterceptedRequestData is made (regardless of whether
  92     // the result of that call is a valid InterceptedRequestData* pointer or
  93     // NULL.
  94     user_data = new URLRequestUserData(
  95         QueryForInterceptedRequestData(request->url(), request));
  96     request->SetUserData(kURLRequestUserDataKey, user_data);
  97   }
  98
  99   const InterceptedRequestData* intercepted_request_data =
 100       user_data->intercepted_request_data();
 101
 102   if (!intercepted_request_data)
 103     return NULL;
 104   return intercepted_request_data->CreateJobFor(request, network_delegate);
 105 }
 106
 107 } // namespace android_webview