OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/safe_browsing/binary_integrity_incident_handlers.h" | |
6 | |
7 #include "base/memory/scoped_ptr.h" | |
8 #include "chrome/common/safe_browsing/csd.pb.h" | |
9 #include "testing/gtest/include/gtest/gtest.h" | |
10 | |
11 namespace safe_browsing { | |
12 | |
13 namespace { | |
14 | |
15 scoped_ptr<ClientIncidentReport_IncidentData> MakeIncident() { | |
16 scoped_ptr<ClientIncidentReport_IncidentData> incident( | |
17 new ClientIncidentReport_IncidentData); | |
18 | |
19 incident->mutable_binary_integrity()->set_file_basename("foo"); | |
20 | |
21 // Set the signature. | |
22 incident->mutable_binary_integrity()->mutable_signature()->set_trusted(true); | |
23 ClientDownloadRequest_CertificateChain* certificate_chain = | |
24 incident->mutable_binary_integrity()->mutable_signature() | |
25 ->add_certificate_chain(); | |
26 | |
27 // Fill the certificate chain with 2 elements. | |
28 const unsigned char certificates[][5] = { | |
29 {42, 255, 100, 53, 2}, | |
30 {64, 33, 51, 91, 210}, | |
31 }; | |
32 for (size_t i = 0; i < arraysize(certificates); ++i) { | |
33 ClientDownloadRequest_CertificateChain_Element* element = | |
34 certificate_chain->add_element(); | |
35 element->set_certificate(certificates[i], arraysize(certificates[i])); | |
36 } | |
37 | |
38 return incident.Pass(); | |
39 } | |
40 | |
41 } // namespace | |
42 | |
43 TEST(BinaryIntegrityIncidentHandlersTest, GetKeyIsFile) { | |
44 ClientIncidentReport_IncidentData incident; | |
45 | |
46 incident.mutable_binary_integrity()->set_file_basename("foo"); | |
47 ASSERT_EQ(std::string("foo"), GetBinaryIntegrityIncidentKey(incident)); | |
48 } | |
49 | |
50 TEST(BinaryIntegrityIncidentHandlersTest, SameIncidentSameDigest) { | |
51 scoped_ptr<ClientIncidentReport_IncidentData> incident(MakeIncident()); | |
52 | |
53 uint32_t digest = GetBinaryIntegrityIncidentDigest(*incident); | |
54 ASSERT_EQ(digest, GetBinaryIntegrityIncidentDigest(*MakeIncident())); | |
55 } | |
56 | |
57 TEST(BinaryIntegrityIncidentHandlersTest, DifferentIncidentDifferentDigest) { | |
58 scoped_ptr<ClientIncidentReport_IncidentData> incident(MakeIncident()); | |
59 | |
60 uint32_t digest = GetBinaryIntegrityIncidentDigest(*incident); | |
61 | |
62 scoped_ptr<ClientIncidentReport_IncidentData> incident2(MakeIncident()); | |
63 incident2->mutable_binary_integrity()->set_file_basename("bar"); | |
64 | |
65 ASSERT_NE(digest, GetBinaryIntegrityIncidentDigest(*incident2)); | |
66 } | |
67 | |
68 } // namespace safe_browsing | |
OLD | NEW |