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