| 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/quic_socket_address_coder.h" | 5 #include "net/quic/quic_socket_address_coder.h" |
| 6 | 6 |
| 7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.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 TEST(QuicSocketAddressCoderTest, EncodeIPv4) { |
| 15 IPAddressNumber ip; | 15 IPAddressNumber ip; |
| 16 ASSERT_TRUE(ParseIPLiteralToNumber("4.31.198.44", &ip)); | 16 ASSERT_TRUE(ParseIPLiteralToNumber("4.31.198.44", &ip)); |
| 17 QuicSocketAddressCoder coder(IPEndPoint(ip, 0x1234)); | 17 QuicSocketAddressCoder coder(IPEndPoint(ip, 0x1234)); |
| 18 string serialized = coder.Encode(); | 18 string serialized = coder.Encode(); |
| 19 string expected("\x02\x00\x04\x1f\xc6\x2c\x34\x12", 8); | 19 string expected("\x02\x00\x04\x1f\xc6\x2c\x34\x12", 8); |
| 20 EXPECT_EQ(expected, serialized); | 20 EXPECT_EQ(expected, serialized); |
| 21 } | 21 } |
| 22 | 22 |
| 23 TEST(QuicSocketAddressCoderTest, EncodeIPv6) { | 23 TEST(QuicSocketAddressCoderTest, EncodeIPv6) { |
| 24 IPAddressNumber ip; | 24 IPAddressNumber ip; |
| 25 ASSERT_TRUE(ParseIPLiteralToNumber("2001:700:300:1800::f", &ip)); | 25 ASSERT_TRUE(ParseIPLiteralToNumber("2001:700:300:1800::f", &ip)); |
| 26 QuicSocketAddressCoder coder(IPEndPoint(ip, 0x5678)); | 26 QuicSocketAddressCoder coder(IPEndPoint(ip, 0x5678)); |
| 27 string serialized = coder.Encode(); | 27 string serialized = coder.Encode(); |
| 28 string expected("\x0a\x00" | 28 string expected( |
| 29 "\x20\x01\x07\x00\x03\x00\x18\x00" | 29 "\x0a\x00" |
| 30 "\x00\x00\x00\x00\x00\x00\x00\x0f" | 30 "\x20\x01\x07\x00\x03\x00\x18\x00" |
| 31 "\x78\x56", 20); | 31 "\x00\x00\x00\x00\x00\x00\x00\x0f" |
| 32 "\x78\x56", |
| 33 20); |
| 32 EXPECT_EQ(expected, serialized); | 34 EXPECT_EQ(expected, serialized); |
| 33 } | 35 } |
| 34 | 36 |
| 35 TEST(QuicSocketAddressCoderTest, DecodeIPv4) { | 37 TEST(QuicSocketAddressCoderTest, DecodeIPv4) { |
| 36 string serialized("\x02\x00\x04\x1f\xc6\x2c\x34\x12", 8); | 38 string serialized("\x02\x00\x04\x1f\xc6\x2c\x34\x12", 8); |
| 37 QuicSocketAddressCoder coder; | 39 QuicSocketAddressCoder coder; |
| 38 ASSERT_TRUE(coder.Decode(serialized.data(), serialized.length())); | 40 ASSERT_TRUE(coder.Decode(serialized.data(), serialized.length())); |
| 39 EXPECT_EQ(AF_INET, ConvertAddressFamily(GetAddressFamily(coder.ip()))); | 41 EXPECT_EQ(AF_INET, ConvertAddressFamily(GetAddressFamily(coder.ip()))); |
| 40 string expected_addr("\x04\x1f\xc6\x2c", 4); | 42 string expected_addr("\x04\x1f\xc6\x2c", 4); |
| 41 EXPECT_EQ(expected_addr, IPAddressToPackedString(coder.ip())); | 43 EXPECT_EQ(expected_addr, IPAddressToPackedString(coder.ip())); |
| 42 EXPECT_EQ(0x1234, coder.port()); | 44 EXPECT_EQ(0x1234, coder.port()); |
| 43 } | 45 } |
| 44 | 46 |
| 45 TEST(QuicSocketAddressCoderTest, DecodeIPv6) { | 47 TEST(QuicSocketAddressCoderTest, DecodeIPv6) { |
| 46 string serialized("\x0a\x00" | 48 string serialized( |
| 47 "\x20\x01\x07\x00\x03\x00\x18\x00" | 49 "\x0a\x00" |
| 48 "\x00\x00\x00\x00\x00\x00\x00\x0f" | 50 "\x20\x01\x07\x00\x03\x00\x18\x00" |
| 49 "\x78\x56", 20); | 51 "\x00\x00\x00\x00\x00\x00\x00\x0f" |
| 52 "\x78\x56", |
| 53 20); |
| 50 QuicSocketAddressCoder coder; | 54 QuicSocketAddressCoder coder; |
| 51 ASSERT_TRUE(coder.Decode(serialized.data(), serialized.length())); | 55 ASSERT_TRUE(coder.Decode(serialized.data(), serialized.length())); |
| 52 EXPECT_EQ(AF_INET6, ConvertAddressFamily(GetAddressFamily(coder.ip()))); | 56 EXPECT_EQ(AF_INET6, ConvertAddressFamily(GetAddressFamily(coder.ip()))); |
| 53 string expected_addr("\x20\x01\x07\x00\x03\x00\x18\x00" | 57 string expected_addr( |
| 54 "\x00\x00\x00\x00\x00\x00\x00\x0f", 16); | 58 "\x20\x01\x07\x00\x03\x00\x18\x00" |
| 59 "\x00\x00\x00\x00\x00\x00\x00\x0f", |
| 60 16); |
| 55 EXPECT_EQ(expected_addr, IPAddressToPackedString(coder.ip())); | 61 EXPECT_EQ(expected_addr, IPAddressToPackedString(coder.ip())); |
| 56 EXPECT_EQ(0x5678, coder.port()); | 62 EXPECT_EQ(0x5678, coder.port()); |
| 57 } | 63 } |
| 58 | 64 |
| 59 TEST(QuicSocketAddressCoderTest, DecodeBad) { | 65 TEST(QuicSocketAddressCoderTest, DecodeBad) { |
| 60 string serialized("\x0a\x00" | 66 string serialized( |
| 61 "\x20\x01\x07\x00\x03\x00\x18\x00" | 67 "\x0a\x00" |
| 62 "\x00\x00\x00\x00\x00\x00\x00\x0f" | 68 "\x20\x01\x07\x00\x03\x00\x18\x00" |
| 63 "\x78\x56", 20); | 69 "\x00\x00\x00\x00\x00\x00\x00\x0f" |
| 70 "\x78\x56", |
| 71 20); |
| 64 QuicSocketAddressCoder coder; | 72 QuicSocketAddressCoder coder; |
| 65 EXPECT_TRUE(coder.Decode(serialized.data(), serialized.length())); | 73 EXPECT_TRUE(coder.Decode(serialized.data(), serialized.length())); |
| 66 // Append junk. | 74 // Append junk. |
| 67 serialized.push_back('\0'); | 75 serialized.push_back('\0'); |
| 68 EXPECT_FALSE(coder.Decode(serialized.data(), serialized.length())); | 76 EXPECT_FALSE(coder.Decode(serialized.data(), serialized.length())); |
| 69 // Undo. | 77 // Undo. |
| 70 serialized.resize(20); | 78 serialized.resize(20); |
| 71 EXPECT_TRUE(coder.Decode(serialized.data(), serialized.length())); | 79 EXPECT_TRUE(coder.Decode(serialized.data(), serialized.length())); |
| 72 | 80 |
| 73 // Set an unknown address family. | 81 // Set an unknown address family. |
| (...skipping 11 matching lines...) Expand all Loading... |
| 85 EXPECT_FALSE(coder.Decode(serialized.data(), serialized.length())); | 93 EXPECT_FALSE(coder.Decode(serialized.data(), serialized.length())); |
| 86 } | 94 } |
| 87 EXPECT_TRUE(serialized.empty()); | 95 EXPECT_TRUE(serialized.empty()); |
| 88 } | 96 } |
| 89 | 97 |
| 90 TEST(QuicSocketAddressCoderTest, EncodeAndDecode) { | 98 TEST(QuicSocketAddressCoderTest, EncodeAndDecode) { |
| 91 struct { | 99 struct { |
| 92 const char* ip_literal; | 100 const char* ip_literal; |
| 93 uint16 port; | 101 uint16 port; |
| 94 } test_case[] = { | 102 } test_case[] = { |
| 95 { "93.184.216.119", 0x1234 }, | 103 {"93.184.216.119", 0x1234}, |
| 96 { "199.204.44.194", 80 }, | 104 {"199.204.44.194", 80}, |
| 97 { "149.20.4.69", 443 }, | 105 {"149.20.4.69", 443}, |
| 98 { "127.0.0.1", 8080 }, | 106 {"127.0.0.1", 8080}, |
| 99 { "2001:700:300:1800::", 0x5678 }, | 107 {"2001:700:300:1800::", 0x5678}, |
| 100 { "::1", 65534 }, | 108 {"::1", 65534}, |
| 101 }; | 109 }; |
| 102 | 110 |
| 103 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_case); i++) { | 111 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_case); i++) { |
| 104 IPAddressNumber ip; | 112 IPAddressNumber ip; |
| 105 ASSERT_TRUE(ParseIPLiteralToNumber(test_case[i].ip_literal, &ip)); | 113 ASSERT_TRUE(ParseIPLiteralToNumber(test_case[i].ip_literal, &ip)); |
| 106 QuicSocketAddressCoder encoder(IPEndPoint(ip, test_case[i].port)); | 114 QuicSocketAddressCoder encoder(IPEndPoint(ip, test_case[i].port)); |
| 107 string serialized = encoder.Encode(); | 115 string serialized = encoder.Encode(); |
| 108 | 116 |
| 109 QuicSocketAddressCoder decoder; | 117 QuicSocketAddressCoder decoder; |
| 110 ASSERT_TRUE(decoder.Decode(serialized.data(), serialized.length())); | 118 ASSERT_TRUE(decoder.Decode(serialized.data(), serialized.length())); |
| 111 EXPECT_EQ(encoder.ip(), decoder.ip()); | 119 EXPECT_EQ(encoder.ip(), decoder.ip()); |
| 112 EXPECT_EQ(encoder.port(), decoder.port()); | 120 EXPECT_EQ(encoder.port(), decoder.port()); |
| 113 } | 121 } |
| 114 } | 122 } |
| 115 | 123 |
| 116 } // namespace test | 124 } // namespace test |
| 117 } // namespace net | 125 } // namespace net |
| OLD | NEW |