blob: b4860a212161d5253debbfb86d924e1e665ec610 (
plain)
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
|
// Copyright (c) 2010 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.
//
// Implementation of the MalwareDetails class.
#include "chrome/browser/safe_browsing/malware_details.h"
#include "chrome/browser/safe_browsing/safe_browsing_service.h"
#include "chrome/browser/tab_contents/navigation_entry.h"
#include "chrome/browser/tab_contents/tab_contents.h"
#include "chrome/browser/safe_browsing/report.pb.h"
// Create a MalwareDetails for the given tab. Runs in the UI thread.
MalwareDetails::MalwareDetails(
TabContents* tab_contents,
const SafeBrowsingService::UnsafeResource resource)
: tab_contents_(tab_contents),
resource_(resource) {
StartCollection();
}
MalwareDetails::~MalwareDetails() {}
bool MalwareDetails::IsPublicUrl(const GURL& url) const {
return url.SchemeIs("http"); // TODO(panayiotis): also skip internal urls.
}
void MalwareDetails::AddNode(const std::string& url,
const std::string& parent) {
if (!IsPublicUrl(GURL(url)))
return;
linked_ptr<safe_browsing::ClientMalwareReportRequest::Resource> resource(
new safe_browsing::ClientMalwareReportRequest::Resource());
resource->set_url(url);
if (!parent.empty() && IsPublicUrl(GURL(parent)))
resource->set_parent(parent);
urls_[url] = resource;
}
void MalwareDetails::StartCollection() {
DVLOG(1) << "Starting to compute malware details.";
report_.reset(new safe_browsing::ClientMalwareReportRequest());
if (IsPublicUrl(resource_.url)) {
report_->set_malware_url(resource_.url.spec());
}
GURL page_url = tab_contents_->GetURL();
if (IsPublicUrl(page_url)) {
report_->set_page_url(page_url.spec());
}
GURL referrer_url;
NavigationEntry* nav_entry = tab_contents_->controller().GetActiveEntry();
if (nav_entry) {
referrer_url = nav_entry->referrer();
if (IsPublicUrl(referrer_url)) {
report_->set_referrer_url(referrer_url.spec());
}
}
// Add the nodes, starting from the page url.
AddNode(page_url.spec(), "");
// Add the resource_url and its original url, if non-empty and different.
if (!resource_.original_url.spec().empty() &&
resource_.url != resource_.original_url) {
// Add original_url, as the parent of resource_url.
AddNode(resource_.original_url.spec(), "");
AddNode(resource_.url.spec(), resource_.original_url.spec());
} else {
AddNode(resource_.url.spec(), "");
}
// Add the referrer url.
if (nav_entry && !referrer_url.spec().empty()) {
AddNode(referrer_url.spec(), "");
}
// Add all the urls in our |urls_| map to the |report_| protobuf.
for (ResourceMap::const_iterator it = urls_.begin();
it != urls_.end(); it++) {
safe_browsing::ClientMalwareReportRequest::Resource* pb_resource =
report_->add_nodes();
pb_resource->CopyFrom(*(it->second));
}
}
// Called from the SB Service on the IO thread.
const std::string* MalwareDetails::GetSerializedReport() {
scoped_ptr<std::string> request_data(new std::string());
if (!report_->SerializeToString(request_data.get())) {
DLOG(ERROR) << "Unable to serialize the malware report.";
}
return request_data.release();
}
|
