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