OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "components/nacl/loader/nacl_validation_db.h" | 5 #include "components/nacl/loader/nacl_validation_db.h" |
6 #include "components/nacl/loader/nacl_validation_query.h" | 6 #include "components/nacl/loader/nacl_validation_query.h" |
7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
8 | 8 |
9 // This test makes sure that validation signature generation is performed | 9 // This test makes sure that validation signature generation is performed |
10 // correctly. In effect, this means that we are checking all of the data | 10 // correctly. In effect, this means that we are checking all of the data |
(...skipping 21 matching lines...) Expand all Loading... |
32 "1234567890123456789012345678901234567890123456789012345678901234567890"; | 32 "1234567890123456789012345678901234567890123456789012345678901234567890"; |
33 | 33 |
34 class MockValidationDB : public NaClValidationDB { | 34 class MockValidationDB : public NaClValidationDB { |
35 public: | 35 public: |
36 MockValidationDB() | 36 MockValidationDB() |
37 : did_query_(false), | 37 : did_query_(false), |
38 did_set_(false), | 38 did_set_(false), |
39 status_(true) { | 39 status_(true) { |
40 } | 40 } |
41 | 41 |
42 virtual bool QueryKnownToValidate(const std::string& signature) OVERRIDE { | 42 virtual bool QueryKnownToValidate(const std::string& signature) override { |
43 // The typecast is needed to work around gtest trying to take the address | 43 // The typecast is needed to work around gtest trying to take the address |
44 // of a constant. | 44 // of a constant. |
45 EXPECT_EQ((int) NaClValidationQuery::kDigestLength, | 45 EXPECT_EQ((int) NaClValidationQuery::kDigestLength, |
46 (int) signature.length()); | 46 (int) signature.length()); |
47 EXPECT_FALSE(did_query_); | 47 EXPECT_FALSE(did_query_); |
48 EXPECT_FALSE(did_set_); | 48 EXPECT_FALSE(did_set_); |
49 did_query_ = true; | 49 did_query_ = true; |
50 memcpy(query_signature_, signature.data(), | 50 memcpy(query_signature_, signature.data(), |
51 NaClValidationQuery::kDigestLength); | 51 NaClValidationQuery::kDigestLength); |
52 return status_; | 52 return status_; |
53 } | 53 } |
54 | 54 |
55 virtual void SetKnownToValidate(const std::string& signature) OVERRIDE { | 55 virtual void SetKnownToValidate(const std::string& signature) override { |
56 // The typecast is needed to work around gtest trying to take the address | 56 // The typecast is needed to work around gtest trying to take the address |
57 // of a constant. | 57 // of a constant. |
58 ASSERT_EQ((int) NaClValidationQuery::kDigestLength, | 58 ASSERT_EQ((int) NaClValidationQuery::kDigestLength, |
59 (int) signature.length()); | 59 (int) signature.length()); |
60 EXPECT_TRUE(did_query_); | 60 EXPECT_TRUE(did_query_); |
61 EXPECT_FALSE(did_set_); | 61 EXPECT_FALSE(did_set_); |
62 did_set_ = true; | 62 did_set_ = true; |
63 memcpy(set_signature_, signature.data(), | 63 memcpy(set_signature_, signature.data(), |
64 NaClValidationQuery::kDigestLength); | 64 NaClValidationQuery::kDigestLength); |
65 // Signatures should be the same. | 65 // Signatures should be the same. |
66 EXPECT_EQ(0, memcmp(query_signature_, set_signature_, | 66 EXPECT_EQ(0, memcmp(query_signature_, set_signature_, |
67 NaClValidationQuery::kDigestLength)); | 67 NaClValidationQuery::kDigestLength)); |
68 } | 68 } |
69 | 69 |
70 virtual bool ResolveFileToken(struct NaClFileToken* file_token, int32* fd, | 70 virtual bool ResolveFileToken(struct NaClFileToken* file_token, int32* fd, |
71 std::string* path) OVERRIDE { | 71 std::string* path) override { |
72 *fd = -1; | 72 *fd = -1; |
73 *path = ""; | 73 *path = ""; |
74 return false; | 74 return false; |
75 } | 75 } |
76 | 76 |
77 bool did_query_; | 77 bool did_query_; |
78 bool did_set_; | 78 bool did_set_; |
79 bool status_; | 79 bool status_; |
80 | 80 |
81 uint8 query_signature_[NaClValidationQuery::kDigestLength]; | 81 uint8 query_signature_[NaClValidationQuery::kDigestLength]; |
(...skipping 192 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
274 query1->query->AddData(kShortData, sizeof(kShortData)); | 274 query1->query->AddData(kShortData, sizeof(kShortData)); |
275 query1->query->QueryKnownToValidate(); | 275 query1->query->QueryKnownToValidate(); |
276 | 276 |
277 query2->query->AddData(kShortData, sizeof(kShortData)); | 277 query2->query->AddData(kShortData, sizeof(kShortData)); |
278 query2->query->QueryKnownToValidate(); | 278 query2->query->QueryKnownToValidate(); |
279 | 279 |
280 AssertQueryDifferent(); | 280 AssertQueryDifferent(); |
281 } | 281 } |
282 | 282 |
283 } | 283 } |
OLD | NEW |