| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 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 "net/quic/core/crypto/crypto_secret_boxer.h" | 5 #include "net/quic/core/crypto/crypto_secret_boxer.h" |
| 6 | 6 |
| 7 #include "net/quic/core/crypto/quic_random.h" | 7 #include "net/quic/core/crypto/quic_random.h" |
| 8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "net/quic/platform/api/quic_test.h" |
| 9 | 9 |
| 10 using std::string; | 10 using std::string; |
| 11 | 11 |
| 12 namespace net { | 12 namespace net { |
| 13 namespace test { | 13 namespace test { |
| 14 | 14 |
| 15 TEST(CryptoSecretBoxerTest, BoxAndUnbox) { | 15 class CryptoSecretBoxerTest : public QuicTest {}; |
| 16 |
| 17 TEST_F(CryptoSecretBoxerTest, BoxAndUnbox) { |
| 16 QuicStringPiece message("hello world"); | 18 QuicStringPiece message("hello world"); |
| 17 | 19 |
| 18 CryptoSecretBoxer boxer; | 20 CryptoSecretBoxer boxer; |
| 19 boxer.SetKeys({string(CryptoSecretBoxer::GetKeySize(), 0x11)}); | 21 boxer.SetKeys({string(CryptoSecretBoxer::GetKeySize(), 0x11)}); |
| 20 | 22 |
| 21 const string box = boxer.Box(QuicRandom::GetInstance(), message); | 23 const string box = boxer.Box(QuicRandom::GetInstance(), message); |
| 22 | 24 |
| 23 string storage; | 25 string storage; |
| 24 QuicStringPiece result; | 26 QuicStringPiece result; |
| 25 EXPECT_TRUE(boxer.Unbox(box, &storage, &result)); | 27 EXPECT_TRUE(boxer.Unbox(box, &storage, &result)); |
| (...skipping 14 matching lines...) Expand all Loading... |
| 40 const string boxed = encoder.Box(QuicRandom::GetInstance(), message); | 42 const string boxed = encoder.Box(QuicRandom::GetInstance(), message); |
| 41 string storage; | 43 string storage; |
| 42 QuicStringPiece result; | 44 QuicStringPiece result; |
| 43 bool ok = decoder.Unbox(boxed, &storage, &result); | 45 bool ok = decoder.Unbox(boxed, &storage, &result); |
| 44 if (ok) { | 46 if (ok) { |
| 45 EXPECT_EQ(result, message); | 47 EXPECT_EQ(result, message); |
| 46 } | 48 } |
| 47 return ok; | 49 return ok; |
| 48 } | 50 } |
| 49 | 51 |
| 50 TEST(CryptoSecretBoxerTest, MultipleKeys) { | 52 TEST_F(CryptoSecretBoxerTest, MultipleKeys) { |
| 51 string key_11(CryptoSecretBoxer::GetKeySize(), 0x11); | 53 string key_11(CryptoSecretBoxer::GetKeySize(), 0x11); |
| 52 string key_12(CryptoSecretBoxer::GetKeySize(), 0x12); | 54 string key_12(CryptoSecretBoxer::GetKeySize(), 0x12); |
| 53 | 55 |
| 54 CryptoSecretBoxer boxer_11, boxer_12, boxer; | 56 CryptoSecretBoxer boxer_11, boxer_12, boxer; |
| 55 boxer_11.SetKeys({key_11}); | 57 boxer_11.SetKeys({key_11}); |
| 56 boxer_12.SetKeys({key_12}); | 58 boxer_12.SetKeys({key_12}); |
| 57 boxer.SetKeys({key_12, key_11}); | 59 boxer.SetKeys({key_12, key_11}); |
| 58 | 60 |
| 59 // Neither single-key boxer can decode the other's tokens. | 61 // Neither single-key boxer can decode the other's tokens. |
| 60 EXPECT_FALSE(CanDecode(boxer_11, boxer_12)); | 62 EXPECT_FALSE(CanDecode(boxer_11, boxer_12)); |
| 61 EXPECT_FALSE(CanDecode(boxer_12, boxer_11)); | 63 EXPECT_FALSE(CanDecode(boxer_12, boxer_11)); |
| 62 | 64 |
| 63 // |boxer| encodes with the first key, which is key_12. | 65 // |boxer| encodes with the first key, which is key_12. |
| 64 EXPECT_TRUE(CanDecode(boxer_12, boxer)); | 66 EXPECT_TRUE(CanDecode(boxer_12, boxer)); |
| 65 EXPECT_FALSE(CanDecode(boxer_11, boxer)); | 67 EXPECT_FALSE(CanDecode(boxer_11, boxer)); |
| 66 | 68 |
| 67 // The boxer with both keys can decode tokens from either single-key boxer. | 69 // The boxer with both keys can decode tokens from either single-key boxer. |
| 68 EXPECT_TRUE(CanDecode(boxer, boxer_11)); | 70 EXPECT_TRUE(CanDecode(boxer, boxer_11)); |
| 69 EXPECT_TRUE(CanDecode(boxer, boxer_12)); | 71 EXPECT_TRUE(CanDecode(boxer, boxer_12)); |
| 70 | 72 |
| 71 // After we flush key_11 from |boxer|, it can no longer decode tokens from | 73 // After we flush key_11 from |boxer|, it can no longer decode tokens from |
| 72 // |boxer_11|. | 74 // |boxer_11|. |
| 73 boxer.SetKeys({key_12}); | 75 boxer.SetKeys({key_12}); |
| 74 EXPECT_FALSE(CanDecode(boxer, boxer_11)); | 76 EXPECT_FALSE(CanDecode(boxer, boxer_11)); |
| 75 } | 77 } |
| 76 | 78 |
| 77 } // namespace test | 79 } // namespace test |
| 78 } // namespace net | 80 } // namespace net |
| OLD | NEW |