blob: bb08939cdd3efd227911f859a1ebe4b069e2516e [file] [log] [blame]
// Copyright 2015 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_commands.h"
#include <stdint.h>
#include "base/base64.h"
#include "base/files/file_util.h"
#include "base/macros.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/utf_string_conversions.h"
#include "base/task_scheduler/post_task.h"
#include "base/threading/sequenced_worker_pool.h"
#include "build/build_config.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/download/download_crx_util.h"
#include "chrome/browser/download/download_item_model.h"
#include "chrome/browser/download/download_prefs.h"
#include "chrome/browser/image_decoder.h"
#include "chrome/browser/profiles/profile_manager.h"
#include "chrome/browser/safe_browsing/download_protection/download_protection_service.h"
#include "chrome/browser/safe_browsing/safe_browsing_service.h"
#include "chrome/browser/ui/browser_finder.h"
#include "chrome/browser/ui/scoped_tabbed_browser_displayer.h"
#include "chrome/common/safe_browsing/file_type_policies.h"
#include "chrome/common/url_constants.h"
#include "chrome/grit/theme_resources.h"
#include "components/google/core/browser/google_util.h"
#include "components/safe_browsing/proto/csd.pb.h"
#include "net/base/url_util.h"
#include "ui/base/clipboard/scoped_clipboard_writer.h"
#include "ui/base/resource/resource_bundle.h"
#if defined(OS_WIN)
#include "chrome/browser/download/download_target_determiner.h"
#include "chrome/browser/ui/pdf/adobe_reader_info_win.h"
#endif
#if defined(OS_CHROMEOS)
#include "chrome/browser/chromeos/note_taking_helper.h"
#endif // defined(OS_CHROMEOS)
namespace {
// Maximum size (compressed) of image to be copied to the clipboard. If the
// image exceeds this size, the image is not copied.
const int64_t kMaxImageClipboardSize = 20 * 1024 * 1024; // 20 MB
class ImageClipboardCopyManager : public ImageDecoder::ImageRequest {
public:
static void Start(const base::FilePath& file_path,
base::SequencedTaskRunner* task_runner) {
new ImageClipboardCopyManager(file_path, task_runner);
}
private:
ImageClipboardCopyManager(const base::FilePath& file_path,
base::SequencedTaskRunner* task_runner)
: file_path_(file_path) {
// Constructor must be called in the UI thread.
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
task_runner->PostTask(
FROM_HERE, base::BindOnce(&ImageClipboardCopyManager::StartDecoding,
base::Unretained(this)));
}
void StartDecoding() {
base::ThreadRestrictions::AssertIOAllowed();
// Re-check the filesize since the file may be modified after downloaded.
int64_t filesize;
if (!GetFileSize(file_path_, &filesize) ||
filesize > kMaxImageClipboardSize) {
OnFailedBeforeDecoding();
return;
}
std::string data;
bool ret = base::ReadFileToString(file_path_, &data);
if (!ret || data.empty()) {
OnFailedBeforeDecoding();
return;
}
// Note: An image over 128MB (uncompressed) may fail, due to the limitation
// of IPC message size.
ImageDecoder::Start(this, data);
}
void OnImageDecoded(const SkBitmap& decoded_image) override {
// This method is called on the same thread as constructor (the UI thread).
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
ui::ScopedClipboardWriter scw(ui::CLIPBOARD_TYPE_COPY_PASTE);
scw.Reset();
if (!decoded_image.empty() && !decoded_image.isNull())
scw.WriteImage(decoded_image);
delete this;
}
void OnDecodeImageFailed() override {
// This method is called on the same thread as constructor (the UI thread).
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
delete this;
}
void OnFailedBeforeDecoding() {
// We don't need to cancel the job, since it shouldn't be started here.
task_runner()->DeleteSoon(FROM_HERE, this);
}
const base::FilePath file_path_;
DISALLOW_IMPLICIT_CONSTRUCTORS(ImageClipboardCopyManager);
};
} // anonymous namespace
DownloadCommands::DownloadCommands(content::DownloadItem* download_item)
: download_item_(download_item) {
DCHECK(download_item);
}
DownloadCommands::~DownloadCommands() = default;
int DownloadCommands::GetCommandIconId(Command command) const {
switch (command) {
case PAUSE:
return IDR_DOWNLOAD_NOTIFICATION_MENU_PAUSE;
case RESUME:
return IDR_DOWNLOAD_NOTIFICATION_MENU_DOWNLOAD;
case SHOW_IN_FOLDER:
return IDR_DOWNLOAD_NOTIFICATION_MENU_FOLDER;
case KEEP:
return IDR_DOWNLOAD_NOTIFICATION_MENU_DOWNLOAD;
case DISCARD:
return IDR_DOWNLOAD_NOTIFICATION_MENU_DELETE;
case CANCEL:
return IDR_DOWNLOAD_NOTIFICATION_MENU_CANCEL;
case LEARN_MORE_SCANNING:
return IDR_NOTIFICATION_WELCOME_LEARN_MORE;
case COPY_TO_CLIPBOARD:
return IDR_DOWNLOAD_NOTIFICATION_MENU_COPY_TO_CLIPBOARD;
case ANNOTATE:
return IDR_DOWNLOAD_NOTIFICATION_MENU_ANNOTATE;
case OPEN_WHEN_COMPLETE:
case ALWAYS_OPEN_TYPE:
case PLATFORM_OPEN:
case LEARN_MORE_INTERRUPTED:
return -1;
}
NOTREACHED();
return -1;
}
GURL DownloadCommands::GetLearnMoreURLForInterruptedDownload() const {
GURL learn_more_url(chrome::kDownloadInterruptedLearnMoreURL);
learn_more_url = google_util::AppendGoogleLocaleParam(
learn_more_url, g_browser_process->GetApplicationLocale());
return net::AppendQueryParameter(
learn_more_url, "ctx",
base::IntToString(static_cast<int>(download_item_->GetLastReason())));
}
gfx::Image DownloadCommands::GetCommandIcon(Command command) {
ResourceBundle& bundle = ResourceBundle::GetSharedInstance();
return bundle.GetImageNamed(GetCommandIconId(command));
}
bool DownloadCommands::IsCommandEnabled(Command command) const {
switch (command) {
case SHOW_IN_FOLDER:
return download_item_->CanShowInFolder();
case OPEN_WHEN_COMPLETE:
case PLATFORM_OPEN:
return download_item_->CanOpenDownload() &&
!download_crx_util::IsExtensionDownload(*download_item_);
case ALWAYS_OPEN_TYPE:
// For temporary downloads, the target filename might be a temporary
// filename. Don't base an "Always open" decision based on it. Also
// exclude extensions.
return download_item_->CanOpenDownload() &&
safe_browsing::FileTypePolicies::GetInstance()
->IsAllowedToOpenAutomatically(
download_item_->GetTargetFilePath()) &&
!download_crx_util::IsExtensionDownload(*download_item_);
case CANCEL:
return !download_item_->IsDone();
case PAUSE:
return !download_item_->IsDone() && !download_item_->IsPaused() &&
!download_item_->IsSavePackageDownload() &&
download_item_->GetState() == content::DownloadItem::IN_PROGRESS;
case RESUME:
return download_item_->CanResume() &&
(download_item_->IsPaused() ||
download_item_->GetState() != content::DownloadItem::IN_PROGRESS);
case COPY_TO_CLIPBOARD:
return (download_item_->GetState() == content::DownloadItem::COMPLETE &&
download_item_->GetReceivedBytes() <= kMaxImageClipboardSize);
case ANNOTATE:
return download_item_->GetState() == content::DownloadItem::COMPLETE;
case DISCARD:
case KEEP:
case LEARN_MORE_SCANNING:
case LEARN_MORE_INTERRUPTED:
return true;
}
NOTREACHED();
return false;
}
bool DownloadCommands::IsCommandChecked(Command command) const {
switch (command) {
case OPEN_WHEN_COMPLETE:
return download_item_->GetOpenWhenComplete() ||
download_crx_util::IsExtensionDownload(*download_item_);
case ALWAYS_OPEN_TYPE:
#if defined(OS_WIN) || defined(OS_LINUX) || defined(OS_MACOSX)
if (CanOpenPdfInSystemViewer()) {
DownloadPrefs* prefs = DownloadPrefs::FromBrowserContext(
download_item_->GetBrowserContext());
return prefs->ShouldOpenPdfInSystemReader();
}
#endif
return download_item_->ShouldOpenFileBasedOnExtension();
case PAUSE:
case RESUME:
return download_item_->IsPaused();
case SHOW_IN_FOLDER:
case PLATFORM_OPEN:
case CANCEL:
case DISCARD:
case KEEP:
case LEARN_MORE_SCANNING:
case LEARN_MORE_INTERRUPTED:
case COPY_TO_CLIPBOARD:
case ANNOTATE:
return false;
}
return false;
}
bool DownloadCommands::IsCommandVisible(Command command) const {
if (command == PLATFORM_OPEN)
return (DownloadItemModel(download_item_).ShouldPreferOpeningInBrowser());
return true;
}
void DownloadCommands::ExecuteCommand(Command command) {
switch (command) {
case SHOW_IN_FOLDER:
download_item_->ShowDownloadInShell();
break;
case OPEN_WHEN_COMPLETE:
download_item_->OpenDownload();
break;
case ALWAYS_OPEN_TYPE: {
bool is_checked = IsCommandChecked(ALWAYS_OPEN_TYPE);
DownloadPrefs* prefs = DownloadPrefs::FromBrowserContext(
download_item_->GetBrowserContext());
#if defined(OS_WIN) || defined(OS_LINUX) || defined(OS_MACOSX)
if (CanOpenPdfInSystemViewer()) {
prefs->SetShouldOpenPdfInSystemReader(!is_checked);
DownloadItemModel(download_item_)
.SetShouldPreferOpeningInBrowser(is_checked);
break;
}
#endif
base::FilePath path = download_item_->GetTargetFilePath();
if (is_checked)
prefs->DisableAutoOpenBasedOnExtension(path);
else
prefs->EnableAutoOpenBasedOnExtension(path);
break;
}
case PLATFORM_OPEN:
DownloadItemModel(download_item_).OpenUsingPlatformHandler();
break;
case CANCEL:
download_item_->Cancel(true /* Cancelled by user */);
break;
case DISCARD:
download_item_->Remove();
break;
case KEEP:
// Only sends uncommon download accept report if :
// 1. FULL_SAFE_BROWSING is enabled, and
// 2. Download verdict is uncommon, and
// 3. Download URL is not empty, and
// 4. User is not in incognito mode.
#if defined(FULL_SAFE_BROWSING)
if (download_item_->GetDangerType() ==
content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT &&
!download_item_->GetURL().is_empty() &&
!download_item_->GetBrowserContext()->IsOffTheRecord()) {
safe_browsing::SafeBrowsingService* sb_service =
g_browser_process->safe_browsing_service();
// Compiles the uncommon download warning report.
safe_browsing::ClientSafeBrowsingReportRequest report;
report.set_type(safe_browsing::ClientSafeBrowsingReportRequest::
DANGEROUS_DOWNLOAD_WARNING);
report.set_download_verdict(
safe_browsing::ClientDownloadResponse::UNCOMMON);
report.set_url(download_item_->GetURL().spec());
report.set_did_proceed(true);
std::string token =
safe_browsing::DownloadProtectionService::GetDownloadPingToken(
download_item_);
if (!token.empty())
report.set_token(token);
std::string serialized_report;
if (report.SerializeToString(&serialized_report)) {
sb_service->SendSerializedDownloadReport(serialized_report);
} else {
DCHECK(false)
<< "Unable to serialize the uncommon download warning report.";
}
}
#endif
download_item_->ValidateDangerousDownload();
break;
case LEARN_MORE_SCANNING: {
#if defined(FULL_SAFE_BROWSING)
using safe_browsing::DownloadProtectionService;
safe_browsing::SafeBrowsingService* sb_service =
g_browser_process->safe_browsing_service();
DownloadProtectionService* protection_service =
(sb_service ? sb_service->download_protection_service() : nullptr);
if (protection_service)
protection_service->ShowDetailsForDownload(*download_item_,
GetBrowser());
#else
// Should only be getting invoked if we are using safe browsing.
NOTREACHED();
#endif
break;
}
case LEARN_MORE_INTERRUPTED:
GetBrowser()->OpenURL(content::OpenURLParams(
GetLearnMoreURLForInterruptedDownload(), content::Referrer(),
WindowOpenDisposition::NEW_FOREGROUND_TAB, ui::PAGE_TRANSITION_LINK,
false));
break;
case PAUSE:
download_item_->Pause();
break;
case RESUME:
download_item_->Resume();
break;
case COPY_TO_CLIPBOARD:
CopyFileAsImageToClipboard();
break;
case ANNOTATE:
#if defined(OS_CHROMEOS)
if (DownloadItemModel(download_item_).HasSupportedImageMimeType()) {
chromeos::NoteTakingHelper::Get()->LaunchAppForNewNote(
Profile::FromBrowserContext(download_item_->GetBrowserContext()),
download_item_->GetTargetFilePath());
}
#endif // defined(OS_CHROMEOS)
break;
}
}
Browser* DownloadCommands::GetBrowser() const {
Profile* profile =
Profile::FromBrowserContext(download_item_->GetBrowserContext());
chrome::ScopedTabbedBrowserDisplayer browser_displayer(profile);
DCHECK(browser_displayer.browser());
return browser_displayer.browser();
}
#if defined(OS_WIN) || defined(OS_MACOSX) || defined(OS_LINUX)
bool DownloadCommands::IsDownloadPdf() const {
base::FilePath path = download_item_->GetTargetFilePath();
return path.MatchesExtension(FILE_PATH_LITERAL(".pdf"));
}
#endif
bool DownloadCommands::CanOpenPdfInSystemViewer() const {
#if defined(OS_WIN)
bool is_adobe_pdf_reader_up_to_date = false;
if (IsDownloadPdf() && IsAdobeReaderDefaultPDFViewer()) {
is_adobe_pdf_reader_up_to_date =
DownloadTargetDeterminer::IsAdobeReaderUpToDate();
}
return IsDownloadPdf() &&
(IsAdobeReaderDefaultPDFViewer() ? is_adobe_pdf_reader_up_to_date
: true);
#elif defined(OS_MACOSX) || defined(OS_LINUX)
return IsDownloadPdf();
#endif
}
void DownloadCommands::CopyFileAsImageToClipboard() {
if (download_item_->GetState() != content::DownloadItem::COMPLETE ||
download_item_->GetReceivedBytes() > kMaxImageClipboardSize) {
return;
}
if (!DownloadItemModel(download_item_).HasSupportedImageMimeType())
return;
base::FilePath file_path = download_item_->GetFullPath();
if (!task_runner_) {
task_runner_ = base::CreateSequencedTaskRunnerWithTraits(
{base::MayBlock(), base::TaskPriority::BACKGROUND,
base::TaskShutdownBehavior::SKIP_ON_SHUTDOWN});
}
ImageClipboardCopyManager::Start(file_path, task_runner_.get());
}