| 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 |