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 |