Don't prompt to save malicious downloads on exit

chrome/browser/download/download_service.cc

 // Copyright (c) 2012 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/download/download_service.h"
 
 #include "base/callback.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/download/chrome_download_manager_delegate.h"
 #include "chrome/browser/download/download_history.h"
@@ skipping 76 matching lines @@
 bool DownloadService::HasCreatedDownloadManager() {
   return download_manager_created_;
 }
 
 int DownloadService::DownloadCount() const {
   if (!download_manager_created_)
     return 0;
   return BrowserContext::GetDownloadManager(profile_)->InProgressCount();
 }
 
+int DownloadService::NonDangerousDownloadCount() const {
+  if (!download_manager_created_)
+    return 0;
+  return BrowserContext::GetDownloadManager(profile_)->
+      NonDangerousInProgressCount();
+}
+
 // static
 int DownloadService::DownloadCountAllProfiles() {
   std::vector<Profile*> profiles(
       g_browser_process->profile_manager()->GetLoadedProfiles());
 
   int count = 0;
   for (std::vector<Profile*>::iterator it = profiles.begin();
        it < profiles.end(); ++it) {
     count += DownloadServiceFactory::GetForBrowserContext(*it)->DownloadCount();
     if ((*it)->HasOffTheRecordProfile())
       count += DownloadServiceFactory::GetForBrowserContext(
           (*it)->GetOffTheRecordProfile())->DownloadCount();
   }
 
   return count;
 }
 
+// static
+int DownloadService::NonDangerousDownloadCountAllProfiles() {
+  std::vector<Profile*> profiles(
+      g_browser_process->profile_manager()->GetLoadedProfiles());
+
+  int count = 0;
+  for (std::vector<Profile*>::iterator it = profiles.begin();
+       it < profiles.end(); ++it) {
+    count += DownloadServiceFactory::GetForBrowserContext(*it)->
+        NonDangerousDownloadCount();
+    if ((*it)->HasOffTheRecordProfile())
+      count += DownloadServiceFactory::GetForBrowserContext(
+          (*it)->GetOffTheRecordProfile())->NonDangerousDownloadCount();
+  }
+
+  return count;
+}
+
 void DownloadService::SetDownloadManagerDelegateForTesting(
     ChromeDownloadManagerDelegate* new_delegate) {
   // Set the new delegate first so that if BrowserContext::GetDownloadManager()
   // causes a new download manager to be created, we won't create a redundant
   // ChromeDownloadManagerDelegate().
   manager_delegate_ = new_delegate;
   // Guarantee everything is properly initialized.
   DownloadManager* dm = BrowserContext::GetDownloadManager(profile_);
   if (dm->GetDelegate() != new_delegate) {
     dm->SetDelegate(new_delegate);
@@ skipping 18 matching lines @@
     // code) when the DownloadManager is shutting down. So we shut it down
     // manually earlier. See http://crbug.com/131692
     BrowserContext::GetDownloadManager(profile_)->Shutdown();
   }
 #if !defined(OS_ANDROID)
   extension_event_router_.reset();
 #endif
   manager_delegate_ = NULL;
   download_history_.reset();
 }