| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 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/quic_socket_address_coder.h" | 5 #include "net/quic/core/quic_socket_address_coder.h" |
| 6 | 6 |
| 7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "net/quic/platform/api/quic_test.h" |
| 8 | 8 |
| 9 using std::string; | 9 using std::string; |
| 10 | 10 |
| 11 namespace net { | 11 namespace net { |
| 12 namespace test { | 12 namespace test { |
| 13 | 13 |
| 14 TEST(QuicSocketAddressCoderTest, EncodeIPv4) { | 14 class QuicSocketAddressCoderTest : public QuicTest {}; |
| 15 |
| 16 TEST_F(QuicSocketAddressCoderTest, EncodeIPv4) { |
| 15 QuicIpAddress ip; | 17 QuicIpAddress ip; |
| 16 ip.FromString("4.31.198.44"); | 18 ip.FromString("4.31.198.44"); |
| 17 QuicSocketAddressCoder coder(QuicSocketAddress(ip, 0x1234)); | 19 QuicSocketAddressCoder coder(QuicSocketAddress(ip, 0x1234)); |
| 18 string serialized = coder.Encode(); | 20 string serialized = coder.Encode(); |
| 19 string expected("\x02\x00\x04\x1f\xc6\x2c\x34\x12", 8); | 21 string expected("\x02\x00\x04\x1f\xc6\x2c\x34\x12", 8); |
| 20 EXPECT_EQ(expected, serialized); | 22 EXPECT_EQ(expected, serialized); |
| 21 } | 23 } |
| 22 | 24 |
| 23 TEST(QuicSocketAddressCoderTest, EncodeIPv6) { | 25 TEST_F(QuicSocketAddressCoderTest, EncodeIPv6) { |
| 24 QuicIpAddress ip; | 26 QuicIpAddress ip; |
| 25 ip.FromString("2001:700:300:1800::f"); | 27 ip.FromString("2001:700:300:1800::f"); |
| 26 QuicSocketAddressCoder coder(QuicSocketAddress(ip, 0x5678)); | 28 QuicSocketAddressCoder coder(QuicSocketAddress(ip, 0x5678)); |
| 27 string serialized = coder.Encode(); | 29 string serialized = coder.Encode(); |
| 28 string expected( | 30 string expected( |
| 29 "\x0a\x00" | 31 "\x0a\x00" |
| 30 "\x20\x01\x07\x00\x03\x00\x18\x00" | 32 "\x20\x01\x07\x00\x03\x00\x18\x00" |
| 31 "\x00\x00\x00\x00\x00\x00\x00\x0f" | 33 "\x00\x00\x00\x00\x00\x00\x00\x0f" |
| 32 "\x78\x56", | 34 "\x78\x56", |
| 33 20); | 35 20); |
| 34 EXPECT_EQ(expected, serialized); | 36 EXPECT_EQ(expected, serialized); |
| 35 } | 37 } |
| 36 | 38 |
| 37 TEST(QuicSocketAddressCoderTest, DecodeIPv4) { | 39 TEST_F(QuicSocketAddressCoderTest, DecodeIPv4) { |
| 38 string serialized("\x02\x00\x04\x1f\xc6\x2c\x34\x12", 8); | 40 string serialized("\x02\x00\x04\x1f\xc6\x2c\x34\x12", 8); |
| 39 QuicSocketAddressCoder coder; | 41 QuicSocketAddressCoder coder; |
| 40 ASSERT_TRUE(coder.Decode(serialized.data(), serialized.length())); | 42 ASSERT_TRUE(coder.Decode(serialized.data(), serialized.length())); |
| 41 EXPECT_EQ(IpAddressFamily::IP_V4, coder.ip().address_family()); | 43 EXPECT_EQ(IpAddressFamily::IP_V4, coder.ip().address_family()); |
| 42 string expected_addr("\x04\x1f\xc6\x2c", 4); | 44 string expected_addr("\x04\x1f\xc6\x2c", 4); |
| 43 EXPECT_EQ(expected_addr, coder.ip().ToPackedString()); | 45 EXPECT_EQ(expected_addr, coder.ip().ToPackedString()); |
| 44 EXPECT_EQ(0x1234, coder.port()); | 46 EXPECT_EQ(0x1234, coder.port()); |
| 45 } | 47 } |
| 46 | 48 |
| 47 TEST(QuicSocketAddressCoderTest, DecodeIPv6) { | 49 TEST_F(QuicSocketAddressCoderTest, DecodeIPv6) { |
| 48 string serialized( | 50 string serialized( |
| 49 "\x0a\x00" | 51 "\x0a\x00" |
| 50 "\x20\x01\x07\x00\x03\x00\x18\x00" | 52 "\x20\x01\x07\x00\x03\x00\x18\x00" |
| 51 "\x00\x00\x00\x00\x00\x00\x00\x0f" | 53 "\x00\x00\x00\x00\x00\x00\x00\x0f" |
| 52 "\x78\x56", | 54 "\x78\x56", |
| 53 20); | 55 20); |
| 54 QuicSocketAddressCoder coder; | 56 QuicSocketAddressCoder coder; |
| 55 ASSERT_TRUE(coder.Decode(serialized.data(), serialized.length())); | 57 ASSERT_TRUE(coder.Decode(serialized.data(), serialized.length())); |
| 56 EXPECT_EQ(IpAddressFamily::IP_V6, coder.ip().address_family()); | 58 EXPECT_EQ(IpAddressFamily::IP_V6, coder.ip().address_family()); |
| 57 string expected_addr( | 59 string expected_addr( |
| 58 "\x20\x01\x07\x00\x03\x00\x18\x00" | 60 "\x20\x01\x07\x00\x03\x00\x18\x00" |
| 59 "\x00\x00\x00\x00\x00\x00\x00\x0f", | 61 "\x00\x00\x00\x00\x00\x00\x00\x0f", |
| 60 16); | 62 16); |
| 61 EXPECT_EQ(expected_addr, coder.ip().ToPackedString()); | 63 EXPECT_EQ(expected_addr, coder.ip().ToPackedString()); |
| 62 EXPECT_EQ(0x5678, coder.port()); | 64 EXPECT_EQ(0x5678, coder.port()); |
| 63 } | 65 } |
| 64 | 66 |
| 65 TEST(QuicSocketAddressCoderTest, DecodeBad) { | 67 TEST_F(QuicSocketAddressCoderTest, DecodeBad) { |
| 66 string serialized( | 68 string serialized( |
| 67 "\x0a\x00" | 69 "\x0a\x00" |
| 68 "\x20\x01\x07\x00\x03\x00\x18\x00" | 70 "\x20\x01\x07\x00\x03\x00\x18\x00" |
| 69 "\x00\x00\x00\x00\x00\x00\x00\x0f" | 71 "\x00\x00\x00\x00\x00\x00\x00\x0f" |
| 70 "\x78\x56", | 72 "\x78\x56", |
| 71 20); | 73 20); |
| 72 QuicSocketAddressCoder coder; | 74 QuicSocketAddressCoder coder; |
| 73 EXPECT_TRUE(coder.Decode(serialized.data(), serialized.length())); | 75 EXPECT_TRUE(coder.Decode(serialized.data(), serialized.length())); |
| 74 // Append junk. | 76 // Append junk. |
| 75 serialized.push_back('\0'); | 77 serialized.push_back('\0'); |
| (...skipping 12 matching lines...) Expand all Loading... |
| 88 // Truncate. | 90 // Truncate. |
| 89 size_t len = serialized.length(); | 91 size_t len = serialized.length(); |
| 90 for (size_t i = 0; i < len; i++) { | 92 for (size_t i = 0; i < len; i++) { |
| 91 ASSERT_FALSE(serialized.empty()); | 93 ASSERT_FALSE(serialized.empty()); |
| 92 serialized.erase(serialized.length() - 1); | 94 serialized.erase(serialized.length() - 1); |
| 93 EXPECT_FALSE(coder.Decode(serialized.data(), serialized.length())); | 95 EXPECT_FALSE(coder.Decode(serialized.data(), serialized.length())); |
| 94 } | 96 } |
| 95 EXPECT_TRUE(serialized.empty()); | 97 EXPECT_TRUE(serialized.empty()); |
| 96 } | 98 } |
| 97 | 99 |
| 98 TEST(QuicSocketAddressCoderTest, EncodeAndDecode) { | 100 TEST_F(QuicSocketAddressCoderTest, EncodeAndDecode) { |
| 99 struct { | 101 struct { |
| 100 const char* ip_literal; | 102 const char* ip_literal; |
| 101 uint16_t port; | 103 uint16_t port; |
| 102 } test_case[] = { | 104 } test_case[] = { |
| 103 {"93.184.216.119", 0x1234}, | 105 {"93.184.216.119", 0x1234}, |
| 104 {"199.204.44.194", 80}, | 106 {"199.204.44.194", 80}, |
| 105 {"149.20.4.69", 443}, | 107 {"149.20.4.69", 443}, |
| 106 {"127.0.0.1", 8080}, | 108 {"127.0.0.1", 8080}, |
| 107 {"2001:700:300:1800::", 0x5678}, | 109 {"2001:700:300:1800::", 0x5678}, |
| 108 {"::1", 65534}, | 110 {"::1", 65534}, |
| 109 }; | 111 }; |
| 110 | 112 |
| 111 for (size_t i = 0; i < arraysize(test_case); i++) { | 113 for (size_t i = 0; i < arraysize(test_case); i++) { |
| 112 QuicIpAddress ip; | 114 QuicIpAddress ip; |
| 113 ASSERT_TRUE(ip.FromString(test_case[i].ip_literal)); | 115 ASSERT_TRUE(ip.FromString(test_case[i].ip_literal)); |
| 114 QuicSocketAddressCoder encoder(QuicSocketAddress(ip, test_case[i].port)); | 116 QuicSocketAddressCoder encoder(QuicSocketAddress(ip, test_case[i].port)); |
| 115 string serialized = encoder.Encode(); | 117 string serialized = encoder.Encode(); |
| 116 | 118 |
| 117 QuicSocketAddressCoder decoder; | 119 QuicSocketAddressCoder decoder; |
| 118 ASSERT_TRUE(decoder.Decode(serialized.data(), serialized.length())); | 120 ASSERT_TRUE(decoder.Decode(serialized.data(), serialized.length())); |
| 119 EXPECT_EQ(encoder.ip(), decoder.ip()); | 121 EXPECT_EQ(encoder.ip(), decoder.ip()); |
| 120 EXPECT_EQ(encoder.port(), decoder.port()); | 122 EXPECT_EQ(encoder.port(), decoder.port()); |
| 121 } | 123 } |
| 122 } | 124 } |
| 123 | 125 |
| 124 } // namespace test | 126 } // namespace test |
| 125 } // namespace net | 127 } // namespace net |
| OLD | NEW |