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
|
// Copyright 2019 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/api/safe_browsing_private/safe_browsing_private_api.h"
#include <utility>
#include "base/strings/stringprintf.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/extensions/api/safe_browsing_private/safe_browsing_util.h"
#include "chrome/browser/extensions/extension_tab_util.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/safe_browsing/safe_browsing_navigation_observer_manager_factory.h"
#include "chrome/common/extensions/api/safe_browsing_private.h"
#include "components/safe_browsing/content/browser/safe_browsing_navigation_observer_manager.h"
#include "content/public/browser/web_contents.h"
#include "extensions/browser/extension_function.h"
#if BUILDFLAG(SAFE_BROWSING_AVAILABLE)
#include "chrome/browser/safe_browsing/safe_browsing_service.h"
#endif
using safe_browsing::SafeBrowsingNavigationObserverManager;
namespace extensions {
#if BUILDFLAG(SAFE_BROWSING_AVAILABLE)
namespace {
// The number of user gestures we trace back for the referrer chain.
const int kReferrerUserGestureLimit = 2;
} // namespace
#endif
////////////////////////////////////////////////////////////////////////////////
// SafeBrowsingPrivateGetReferrerChainFunction
SafeBrowsingPrivateGetReferrerChainFunction::
SafeBrowsingPrivateGetReferrerChainFunction() = default;
SafeBrowsingPrivateGetReferrerChainFunction::
~SafeBrowsingPrivateGetReferrerChainFunction() = default;
ExtensionFunction::ResponseAction
SafeBrowsingPrivateGetReferrerChainFunction::Run() {
std::optional<api::safe_browsing_private::GetReferrerChain::Params> params =
api::safe_browsing_private::GetReferrerChain::Params::Create(args());
EXTENSION_FUNCTION_VALIDATE(params);
content::WebContents* contents = nullptr;
if (!ExtensionTabUtil::GetTabById(params->tab_id, browser_context(),
include_incognito_information(),
&contents)) {
return RespondNow(Error(
base::StringPrintf("Could not find tab with id %d.", params->tab_id)));
}
#if BUILDFLAG(SAFE_BROWSING_AVAILABLE)
Profile* profile = Profile::FromBrowserContext(browser_context());
if (!SafeBrowsingNavigationObserverManager::IsEnabledAndReady(
profile->GetPrefs(), g_browser_process->safe_browsing_service()))
return RespondNow(NoArguments());
SafeBrowsingNavigationObserverManager* navigation_observer_manager =
safe_browsing::SafeBrowsingNavigationObserverManagerFactory::
GetForBrowserContext(profile);
safe_browsing::ReferrerChain referrer_chain;
SafeBrowsingNavigationObserverManager::AttributionResult result =
navigation_observer_manager->IdentifyReferrerChainByRenderFrameHost(
contents->GetPrimaryMainFrame(), kReferrerUserGestureLimit,
&referrer_chain);
// If the referrer chain is incomplete we'll append the most recent
// navigations to referrer chain for diagnostic purposes. This only happens if
// the user is not in incognito mode and has opted into extended reporting or
// Scout reporting. Otherwise, |CountOfRecentNavigationsToAppend| returns 0.
int recent_navigations_to_collect =
SafeBrowsingNavigationObserverManager::CountOfRecentNavigationsToAppend(
profile, profile->GetPrefs(), result);
if (recent_navigations_to_collect > 0) {
navigation_observer_manager->AppendRecentNavigations(
recent_navigations_to_collect, &referrer_chain);
}
std::vector<api::safe_browsing_private::ReferrerChainEntry> referrer_entries;
referrer_entries.reserve(referrer_chain.size());
for (const auto& entry : referrer_chain) {
referrer_entries.emplace_back(
safe_browsing_util::ReferrerToReferrerChainEntry(entry));
}
return RespondNow(ArgumentList(
api::safe_browsing_private::GetReferrerChain::Results::Create(
referrer_entries)));
#else
return RespondNow(NoArguments());
#endif
}
} // namespace extensions
|
