Index: components/safe_browsing_db/v4_protocol_manager_util.cc |
diff --git a/components/safe_browsing_db/v4_protocol_manager_util.cc b/components/safe_browsing_db/v4_protocol_manager_util.cc |
index c9f5349d3c5bdfa7a32f3ebf0f0a07714edb49bd..e5ee76e547f79b7339ffff8a26ad863eb068fd1e 100644 |
--- a/components/safe_browsing_db/v4_protocol_manager_util.cc |
+++ b/components/safe_browsing_db/v4_protocol_manager_util.cc |
@@ -62,9 +62,9 @@ std::string Escape(const std::string& url) { |
} // namespace |
std::ostream& operator<<(std::ostream& os, const ListIdentifier& id) { |
- os << "{hash: " << id.hash() << "; platform_type: " << id.platform_type |
- << "; threat_entry_type: " << id.threat_entry_type |
- << "; threat_type: " << id.threat_type << "}"; |
+ os << "{hash: " << id.hash() << "; platform_type: " << id.platform_type() |
+ << "; threat_entry_type: " << id.threat_entry_type() |
+ << "; threat_type: " << id.threat_type() << "}"; |
return os; |
} |
@@ -86,16 +86,18 @@ return LINUX_PLATFORM; |
} |
const ListIdentifier GetChromeUrlApiId() { |
- return ListIdentifier(CHROME_PLATFORM, URL, API_ABUSE); |
+ return ListIdentifier(CHROME_PLATFORM, URL, API_ABUSE, |
+ SB_THREAT_TYPE_API_ABUSE); |
} |
const ListIdentifier GetUrlMalwareId() { |
- return ListIdentifier(GetCurrentPlatformType(), URL, MALWARE_THREAT); |
+ return ListIdentifier(GetCurrentPlatformType(), URL, MALWARE_THREAT, |
+ SB_THREAT_TYPE_URL_MALWARE); |
} |
const ListIdentifier GetUrlSocEngId() { |
return ListIdentifier(GetCurrentPlatformType(), URL, |
- SOCIAL_ENGINEERING_PUBLIC); |
+ SOCIAL_ENGINEERING_PUBLIC, SB_THREAT_TYPE_URL_PHISHING); |
} |
// The Safe Browsing V4 server URL prefix. |
@@ -123,9 +125,9 @@ size_t StoreAndHashPrefix::hash() const { |
} |
bool ListIdentifier::operator==(const ListIdentifier& other) const { |
- return platform_type == other.platform_type && |
- threat_entry_type == other.threat_entry_type && |
- threat_type == other.threat_type; |
+ return platform_type_ == other.platform_type_ && |
+ threat_entry_type_ == other.threat_entry_type_ && |
+ threat_type_ == other.threat_type_; |
} |
bool ListIdentifier::operator!=(const ListIdentifier& other) const { |
@@ -133,9 +135,9 @@ bool ListIdentifier::operator!=(const ListIdentifier& other) const { |
} |
size_t ListIdentifier::hash() const { |
- std::size_t first = std::hash<unsigned int>()(platform_type); |
- std::size_t second = std::hash<unsigned int>()(threat_entry_type); |
- std::size_t third = std::hash<unsigned int>()(threat_type); |
+ std::size_t first = std::hash<unsigned int>()(platform_type_); |
+ std::size_t second = std::hash<unsigned int>()(threat_entry_type_); |
+ std::size_t third = std::hash<unsigned int>()(threat_type_); |
std::size_t interim = base::HashInts(first, second); |
return base::HashInts(interim, third); |
@@ -145,14 +147,23 @@ ListIdentifier::ListIdentifier() {} |
ListIdentifier::ListIdentifier(PlatformType platform_type, |
ThreatEntryType threat_entry_type, |
- ThreatType threat_type) |
- : platform_type(platform_type), |
- threat_entry_type(threat_entry_type), |
- threat_type(threat_type) { |
+ ThreatType threat_type, |
+ SBThreatType sb_threat_type) |
+ : platform_type_(platform_type), |
+ threat_entry_type_(threat_entry_type), |
+ threat_type_(threat_type), |
+ sb_threat_type_(sb_threat_type) { |
DCHECK(PlatformType_IsValid(platform_type)); |
DCHECK(ThreatEntryType_IsValid(threat_entry_type)); |
DCHECK(ThreatType_IsValid(threat_type)); |
} |
+ListIdentifier::ListIdentifier(PlatformType platform_type, |
+ ThreatEntryType threat_entry_type, |
+ ThreatType threat_type) |
+ : ListIdentifier(platform_type, |
+ threat_entry_type, |
+ threat_type, |
+ SB_THREAT_TYPE_SAFE) {} |
ListIdentifier::ListIdentifier(const ListUpdateResponse& response) |
: ListIdentifier(response.platform_type(), |