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