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
|
// 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.
//
// ThreatDOMDetails iterates over a document's frames and gathers
// interesting URLs such as those of scripts and frames. When done, it sends
// them to the ThreatDetails that requested them.
#ifndef COMPONENTS_SAFE_BROWSING_CONTENT_RENDERER_THREAT_DOM_DETAILS_H_
#define COMPONENTS_SAFE_BROWSING_CONTENT_RENDERER_THREAT_DOM_DETAILS_H_
#include <vector>
#include "base/compiler_specific.h"
#include "base/feature_list.h"
#include "components/safe_browsing/content/common/safe_browsing.mojom.h"
#include "content/public/renderer/render_frame_observer.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/receiver_set.h"
#include "services/service_manager/public/cpp/binder_registry.h"
namespace safe_browsing {
extern const char kTagAndAttributeParamName[];
// Represents the tag name of an HTML Element and its associated attributes.
// Used to determine which elements to collect. Populated from the param value
// of |kThreatDomDetailsTagAndAttributeFeature|.
class TagAndAttributesItem {
public:
TagAndAttributesItem();
TagAndAttributesItem(const std::string& tag_name_param,
const std::vector<std::string>& attributes_param);
TagAndAttributesItem(const TagAndAttributesItem& item);
~TagAndAttributesItem();
std::string tag_name;
std::vector<std::string> attributes;
};
// There is one ThreatDOMDetails per RenderFrame.
class ThreatDOMDetails : public content::RenderFrameObserver,
public mojom::ThreatReporter {
public:
// An upper limit on the number of nodes we collect. Not const for the test.
static uint32_t kMaxNodes;
// An upper limit on the number of attributes to collect per node. Not const
// for the test.
static uint32_t kMaxAttributes;
// An upper limit on the length of an attribute string.
static uint32_t kMaxAttributeStringLength;
static ThreatDOMDetails* Create(content::RenderFrame* render_frame,
service_manager::BinderRegistry* registry);
ThreatDOMDetails(const ThreatDOMDetails&) = delete;
ThreatDOMDetails& operator=(const ThreatDOMDetails&) = delete;
~ThreatDOMDetails() override;
// Begins extracting resource urls for the page currently loaded in
// this object's RenderFrame.
// Exposed for testing.
void ExtractResources(std::vector<mojom::ThreatDOMDetailsNodePtr>* resources);
std::vector<TagAndAttributesItem> GetTagAndAttributesListForTest() {
return tag_and_attributes_list_;
}
private:
// Creates a ThreatDOMDetails for the specified RenderFrame.
// The ThreatDOMDetails should be destroyed prior to destroying
// the RenderFrame.
ThreatDOMDetails(content::RenderFrame* render_frame,
service_manager::BinderRegistry* registry);
void OnDestruct() override;
// safe_browsing::mojom::ThreatReporter:
void GetThreatDOMDetails(GetThreatDOMDetailsCallback callback) override;
void OnThreatReporterReceiver(
mojo::PendingReceiver<mojom::ThreatReporter> receiver);
mojo::ReceiverSet<mojom::ThreatReporter> threat_reporter_receivers_;
// A list of tag names and associates attributes, used to determine which
// elements need to be collected.
std::vector<TagAndAttributesItem> tag_and_attributes_list_;
};
} // namespace safe_browsing
#endif // COMPONENTS_SAFE_BROWSING_CONTENT_RENDERER_THREAT_DOM_DETAILS_H_
|
