1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/extensions/navigation_extension_enabler.h"
#include <memory>
#include "base/functional/bind.h"
#include "chrome/browser/extensions/extension_service.h"
#include "content/public/browser/navigation_details.h"
#include "content/public/browser/navigation_entry.h"
#include "extensions/browser/extension_prefs.h"
#include "extensions/browser/extension_system.h"
namespace extensions {
NavigationExtensionEnabler::NavigationExtensionEnabler(
content::WebContents* web_contents)
: content::WebContentsObserver(web_contents),
content::WebContentsUserData<NavigationExtensionEnabler>(*web_contents) {
extension_registry_observation_.Observe(
ExtensionRegistry::Get(web_contents->GetBrowserContext()));
}
NavigationExtensionEnabler::~NavigationExtensionEnabler() = default;
void NavigationExtensionEnabler::NavigationEntryCommitted(
const content::LoadCommittedDetails& load_details) {
PromptToEnableExtensionIfNecessary(load_details.entry->GetURL());
}
void NavigationExtensionEnabler::PromptToEnableExtensionIfNecessary(
const GURL& url) {
// Bail out if we're already running a prompt.
if (!in_progress_prompt_extension_id_.empty()) {
return;
}
// NOTE: We only consider chrome-extension:// urls, and deliberately don't
// consider hosted app urls. This is because it's really annoying to visit the
// site associated with a hosted app (like calendar.google.com or
// drive.google.com) and have it repeatedly prompt you to re-enable an item.
// Visiting a chrome-extension:// url is a much stronger signal, and, without
// the item enabled, we won't show anything.
// TODO(devlin): While true, I still wonder how useful this is. We should get
// metrics.
if (!url.SchemeIs(kExtensionScheme)) {
return;
}
const Extension* extension =
ExtensionRegistry::Get(web_contents()->GetBrowserContext())
->disabled_extensions()
.GetExtensionOrAppByURL(url);
if (!extension) {
return;
}
ExtensionPrefs* extension_prefs =
ExtensionPrefs::Get(web_contents()->GetBrowserContext());
// TODO(devlin): Why do we only consider extensions that escalate permissions?
// Maybe because it's the only one we have a good prompt for?
if (!extension_prefs->DidExtensionEscalatePermissions(extension->id())) {
return;
}
// Keep track of the extension id we're prompting for. These must be reset in
// OnInstallPromptDone.
in_progress_prompt_extension_id_ = extension->id();
extension_install_prompt_ =
std::make_unique<ExtensionInstallPrompt>(web_contents());
ExtensionInstallPrompt::PromptType type =
ExtensionInstallPrompt::GetReEnablePromptTypeForExtension(
web_contents()->GetBrowserContext(), extension);
extension_install_prompt_->ShowDialog(
base::BindRepeating(&NavigationExtensionEnabler::OnInstallPromptDone,
weak_factory_.GetWeakPtr()),
extension, nullptr,
std::make_unique<ExtensionInstallPrompt::Prompt>(type),
ExtensionInstallPrompt::GetDefaultShowDialogCallback());
}
void NavigationExtensionEnabler::OnInstallPromptDone(
ExtensionInstallPrompt::DoneCallbackPayload payload) {
// This dialog doesn't support the "withhold permissions" checkbox.
DCHECK_NE(payload.result,
ExtensionInstallPrompt::Result::ACCEPTED_WITH_WITHHELD_PERMISSIONS);
// The extension was already uninstalled.
if (in_progress_prompt_extension_id_.empty()) {
return;
}
ExtensionRegistry* extension_registry =
ExtensionRegistry::Get(web_contents()->GetBrowserContext());
const Extension* extension = extension_registry->GetExtensionById(
in_progress_prompt_extension_id_, ExtensionRegistry::EVERYTHING);
CHECK(extension);
if (payload.result == ExtensionInstallPrompt::Result::ACCEPTED) {
ExtensionRegistrar* extension_registrar =
ExtensionRegistrar::Get(web_contents()->GetBrowserContext());
// Grant permissions, re-enable the extension, and then reload the tab.
extension_registrar->GrantPermissionsAndEnableExtension(*extension);
web_contents()->GetController().Reload(content::ReloadType::NORMAL, true);
}
in_progress_prompt_extension_id_.clear();
extension_install_prompt_.reset();
}
void NavigationExtensionEnabler::OnExtensionUninstalled(
content::BrowserContext* browser_context,
const Extension* extension,
UninstallReason reason) {
if (in_progress_prompt_extension_id_.empty() ||
in_progress_prompt_extension_id_ != extension->id()) {
return;
}
in_progress_prompt_extension_id_ = std::string();
extension_install_prompt_.reset();
}
WEB_CONTENTS_USER_DATA_KEY_IMPL(NavigationExtensionEnabler);
} // namespace extensions
|
