Cancel prerender when we discover a download starting from a page we are prer...

chrome/browser/renderer_host/download_throttling_resource_handler.cc

 // Copyright (c) 2011 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/renderer_host/download_throttling_resource_handler.h"
 
 #include "base/logging.h"
+#include "chrome/browser/cert_store.h"
 #include "chrome/browser/renderer_host/download_resource_handler.h"
+#include "chrome/browser/renderer_host/render_view_host.h"
+#include "chrome/browser/renderer_host/render_view_host_delegate.h"
 #include "chrome/browser/renderer_host/resource_dispatcher_host.h"
+#include "chrome/browser/renderer_host/resource_dispatcher_host_request_info.h"
+#include "chrome/browser/renderer_host/resource_request_details.h"
+#include "chrome/common/notification_service.h"
 #include "chrome/common/resource_response.h"
 #include "net/base/io_buffer.h"
 #include "net/base/mime_sniffer.h"
 
 DownloadThrottlingResourceHandler::DownloadThrottlingResourceHandler(
     ResourceDispatcherHost* host,
     net::URLRequest* request,
     const GURL& url,
     int render_process_host_id,
     int render_view_id,
     int request_id,
     bool in_complete)
     : host_(host),
       request_(request),
       url_(url),
       render_process_host_id_(render_process_host_id),
       render_view_id_(render_view_id),
       request_id_(request_id),
       tmp_buffer_length_(0),
       ignore_on_read_complete_(in_complete) {
   // Pause the request.
   host_->PauseRequest(render_process_host_id_, request_id_, true);
   host_->download_request_limiter()->CanDownloadOnIOThread(
       render_process_host_id_, render_view_id, this);
  }
 
 DownloadThrottlingResourceHandler::~DownloadThrottlingResourceHandler() {
 }
 
+ResourceRequestDetails*
+  DownloadThrottlingResourceHandler::GetResourceRequestDetails() {
+  int certID = 0;
+  if (request_->ssl_info().cert) {
+    ResourceDispatcherHostRequestInfo* info =
+        ResourceDispatcherHost::InfoForRequest(request_);
+
+    certID = CertStore::GetInstance()->StoreCert(request_->ssl_info().cert,

[cbentzel, 2011/02/09 12:11:14]

I have no idea what's going on here, but it seems like storing SSL certs for a
download notification is probably not correct.

Also, none of the consumers currently use any of the details - just the source.
Could you use a generic details?

[dominich, 2011/02/09 22:33:30]

Done.

+                                                 info->child_id());
+  }
+
+  return new ResourceRequestDetails(request_, certID);
+}
+
+template<NotificationType::Type T>
+void DownloadThrottlingResourceHandler::NotifyOnUI(
+    int process_id, int view_id,
+    ResourceRequestDetails* details) {
+  RenderViewHost* rvh = RenderViewHost::FromID(process_id, view_id);
+  if (rvh) {
+    RenderViewHostDelegate* rvhd = rvh->delegate();
+    NotificationService::current()->Notify(T,
+        Source<RenderViewHostDelegate>(rvhd),
+        Details<ResourceRequestDetails>(details));
+  }
+  delete details;
+}
+
 bool DownloadThrottlingResourceHandler::OnUploadProgress(int request_id,
                                                          uint64 position,
                                                          uint64 size) {
   if (download_handler_.get())
     return download_handler_->OnUploadProgress(request_id, position, size);
   return true;
 }
 
 bool DownloadThrottlingResourceHandler::OnRequestRedirected(
     int request_id,
     const GURL& url,
     ResourceResponse* response,
     bool* defer) {
   if (download_handler_.get()) {
     return download_handler_->OnRequestRedirected(
         request_id, url, response, defer);
   }
   url_ = url;
   return true;
 }
 
 bool DownloadThrottlingResourceHandler::OnResponseStarted(
     int request_id,
     ResourceResponse* response) {
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      NewRunnableFunction(
+        &DownloadThrottlingResourceHandler::NotifyOnUI
+          <NotificationType::DOWNLOAD_STARTED>,
+          render_process_host_id_, render_view_id_,
+          GetResourceRequestDetails()));
+
   if (download_handler_.get())
     return download_handler_->OnResponseStarted(request_id, response);
   response_ = response;
   return true;
 }
 
 bool DownloadThrottlingResourceHandler::OnWillStart(int request_id,
                                                     const GURL& url,
                                                     bool* defer) {
   if (download_handler_.get())
@@ skipping 61 matching lines @@
   NOTREACHED();
   return true;
 }
 
 void DownloadThrottlingResourceHandler::OnRequestClosed() {
   if (download_handler_.get())
     download_handler_->OnRequestClosed();
 }
 
 void DownloadThrottlingResourceHandler::CancelDownload() {
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      NewRunnableFunction(
+          &DownloadThrottlingResourceHandler::NotifyOnUI
+            <NotificationType::DOWNLOAD_CANCELLED>,

[cbentzel, 2011/02/09 12:11:14]

There are no consumers of DOWNLOAD_CANCELLED right now - should you skip adding
it in this CL?

[dominich, 2011/02/09 22:33:30]

I can. I thought that it would be good to have both in for completeness, but
I'll strip this out.

[dominich, 2011/02/09 22:33:30]

Done.

+          render_process_host_id_, render_view_id_,
+          GetResourceRequestDetails()));
+
   host_->CancelRequest(render_process_host_id_, request_id_, false);
 }
 
 void DownloadThrottlingResourceHandler::ContinueDownload() {
   DCHECK(!download_handler_.get());
   download_handler_ =
       new DownloadResourceHandler(host_,
                                   render_process_host_id_,
                                   render_view_id_,
                                   request_id_,
@@ skipping 22 matching lines @@
   int buf_size;
   if (download_handler_->OnWillRead(request_id_, &buffer, &buf_size,
                                     tmp_buffer_length_)) {
     CHECK(buf_size >= tmp_buffer_length_);
     memcpy(buffer->data(), tmp_buffer_->data(), tmp_buffer_length_);
     download_handler_->OnReadCompleted(request_id_, &tmp_buffer_length_);
   }
   tmp_buffer_length_ = 0;
   tmp_buffer_ = NULL;
 }