OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/crypto/null_decrypter.h" | 5 #include "net/quic/crypto/null_decrypter.h" |
6 #include "net/quic/test_tools/quic_test_utils.h" | 6 #include "net/quic/test_tools/quic_test_utils.h" |
7 | 7 |
8 using base::StringPiece; | 8 using base::StringPiece; |
9 | 9 |
10 namespace net { | 10 namespace net { |
11 namespace test { | 11 namespace test { |
12 | 12 |
13 class NullDecrypterTest : public ::testing::TestWithParam<bool> { | 13 class NullDecrypterTest : public ::testing::TestWithParam<bool> { |
14 }; | 14 }; |
15 | 15 |
16 TEST_F(NullDecrypterTest, Decrypt) { | 16 TEST_F(NullDecrypterTest, Decrypt) { |
17 unsigned char expected[] = { | 17 unsigned char expected[] = { |
18 // fnv hash | 18 // fnv hash |
19 0xa0, 0x6f, 0x44, 0x8a, | 19 0xa0, 0x6f, 0x44, 0x8a, |
20 0x44, 0xf8, 0x18, 0x3b, | 20 0x44, 0xf8, 0x18, 0x3b, |
21 0x47, 0x91, 0xb2, 0x13, | 21 0x47, 0x91, 0xb2, 0x13, |
22 // payload | 22 // payload |
23 'g', 'o', 'o', 'd', | 23 'g', 'o', 'o', 'd', |
24 'b', 'y', 'e', '!', | 24 'b', 'y', 'e', '!', |
25 }; | 25 }; |
26 const char* data = reinterpret_cast<const char*>(expected); | 26 const char* data = reinterpret_cast<const char*>(expected); |
27 size_t len = arraysize(expected); | 27 size_t len = arraysize(expected); |
28 NullDecrypter decrypter; | 28 NullDecrypter decrypter; |
29 scoped_ptr<QuicData> decrypted( | 29 char buffer[256]; |
30 decrypter.DecryptPacket(0, "hello world!", StringPiece(data, len))); | 30 size_t length = 0; |
31 ASSERT_TRUE(decrypted.get()); | 31 ASSERT_TRUE(decrypter.DecryptPacket(0, "hello world!", StringPiece(data, len), |
32 EXPECT_EQ("goodbye!", decrypted->AsStringPiece()); | 32 buffer, &length, 256)); |
| 33 EXPECT_LT(0u, length); |
| 34 EXPECT_EQ("goodbye!", StringPiece(buffer, length)); |
33 } | 35 } |
34 | 36 |
35 TEST_F(NullDecrypterTest, BadHash) { | 37 TEST_F(NullDecrypterTest, BadHash) { |
36 unsigned char expected[] = { | 38 unsigned char expected[] = { |
37 // fnv hash | 39 // fnv hash |
38 0x46, 0x11, 0xea, 0x5f, | 40 0x46, 0x11, 0xea, 0x5f, |
39 0xcf, 0x1d, 0x66, 0x5b, | 41 0xcf, 0x1d, 0x66, 0x5b, |
40 0xba, 0xf0, 0xbc, 0xfd, | 42 0xba, 0xf0, 0xbc, 0xfd, |
41 // payload | 43 // payload |
42 'g', 'o', 'o', 'd', | 44 'g', 'o', 'o', 'd', |
43 'b', 'y', 'e', '!', | 45 'b', 'y', 'e', '!', |
44 }; | 46 }; |
45 const char* data = reinterpret_cast<const char*>(expected); | 47 const char* data = reinterpret_cast<const char*>(expected); |
46 size_t len = arraysize(expected); | 48 size_t len = arraysize(expected); |
47 NullDecrypter decrypter; | 49 NullDecrypter decrypter; |
48 scoped_ptr<QuicData> decrypted( | 50 char buffer[256]; |
49 decrypter.DecryptPacket(0, "hello world!", StringPiece(data, len))); | 51 size_t length = 0; |
50 ASSERT_FALSE(decrypted.get()); | 52 ASSERT_FALSE(decrypter.DecryptPacket( |
| 53 0, "hello world!", StringPiece(data, len), buffer, &length, 256)); |
51 } | 54 } |
52 | 55 |
53 TEST_F(NullDecrypterTest, ShortInput) { | 56 TEST_F(NullDecrypterTest, ShortInput) { |
54 unsigned char expected[] = { | 57 unsigned char expected[] = { |
55 // fnv hash (truncated) | 58 // fnv hash (truncated) |
56 0x46, 0x11, 0xea, 0x5f, | 59 0x46, 0x11, 0xea, 0x5f, |
57 0xcf, 0x1d, 0x66, 0x5b, | 60 0xcf, 0x1d, 0x66, 0x5b, |
58 0xba, 0xf0, 0xbc, | 61 0xba, 0xf0, 0xbc, |
59 }; | 62 }; |
60 const char* data = reinterpret_cast<const char*>(expected); | 63 const char* data = reinterpret_cast<const char*>(expected); |
61 size_t len = arraysize(expected); | 64 size_t len = arraysize(expected); |
62 NullDecrypter decrypter; | 65 NullDecrypter decrypter; |
63 scoped_ptr<QuicData> decrypted( | 66 char buffer[256]; |
64 decrypter.DecryptPacket(0, "hello world!", StringPiece(data, len))); | 67 size_t length = 0; |
65 ASSERT_FALSE(decrypted.get()); | 68 ASSERT_FALSE(decrypter.DecryptPacket( |
| 69 0, "hello world!", StringPiece(data, len), buffer, &length, 256)); |
66 } | 70 } |
67 | 71 |
68 } // namespace test | 72 } // namespace test |
69 } // namespace net | 73 } // namespace net |
OLD | NEW |