[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 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 | |
Ryan Hamilton | a3ee93a7 | 2018-08-01 22:03:08 | [diff] [blame] | 5 | #include "net/quic/quic_address_mismatch.h" |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 6 | |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 7 | #include "testing/gtest/include/gtest/gtest.h" |
| 8 | |
| 9 | namespace net { |
| 10 | namespace test { |
| 11 | |
| 12 | // Test all cases of the GetAddressMismatch function. |
| 13 | TEST(QuicAddressMismatchTest, GetAddressMismatch) { |
martijn | cc5402d | 2016-02-16 19:08:58 | [diff] [blame] | 14 | IPAddress ip4_1; |
| 15 | IPAddress ip4_2; |
| 16 | IPAddress ip6_1; |
| 17 | IPAddress ip6_2; |
| 18 | IPAddress ip4_mapped_1; |
| 19 | IPAddress ip4_mapped_2; |
| 20 | ASSERT_TRUE(ip4_1.AssignFromIPLiteral("1.2.3.4")); |
| 21 | ASSERT_TRUE(ip4_2.AssignFromIPLiteral("5.6.7.8")); |
| 22 | ASSERT_TRUE(ip6_1.AssignFromIPLiteral("1234::1")); |
| 23 | ASSERT_TRUE(ip6_2.AssignFromIPLiteral("1234::2")); |
| 24 | ip4_mapped_1 = ConvertIPv4ToIPv4MappedIPv6(ip4_1); |
| 25 | ip4_mapped_2 = ConvertIPv4ToIPv4MappedIPv6(ip4_2); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 26 | ASSERT_NE(ip4_1, ip4_2); |
| 27 | ASSERT_NE(ip6_1, ip6_2); |
[email protected] | dc657464 | 2014-05-20 21:32:59 | [diff] [blame] | 28 | ASSERT_NE(ip4_mapped_1, ip4_mapped_2); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 29 | |
| 30 | EXPECT_EQ(-1, GetAddressMismatch(IPEndPoint(), IPEndPoint())); |
| 31 | EXPECT_EQ(-1, GetAddressMismatch(IPEndPoint(), IPEndPoint(ip4_1, 443))); |
| 32 | EXPECT_EQ(-1, GetAddressMismatch(IPEndPoint(ip4_1, 443), IPEndPoint())); |
| 33 | |
| 34 | EXPECT_EQ(QUIC_ADDRESS_AND_PORT_MATCH_V4_V4, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 35 | GetAddressMismatch(IPEndPoint(ip4_1, 443), IPEndPoint(ip4_1, 443))); |
[email protected] | dc657464 | 2014-05-20 21:32:59 | [diff] [blame] | 36 | EXPECT_EQ(QUIC_ADDRESS_AND_PORT_MATCH_V4_V4, |
| 37 | GetAddressMismatch(IPEndPoint(ip4_1, 443), |
| 38 | IPEndPoint(ip4_mapped_1, 443))); |
| 39 | EXPECT_EQ(QUIC_ADDRESS_AND_PORT_MATCH_V4_V4, |
| 40 | GetAddressMismatch(IPEndPoint(ip4_mapped_1, 443), |
| 41 | IPEndPoint(ip4_mapped_1, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 42 | EXPECT_EQ(QUIC_ADDRESS_AND_PORT_MATCH_V6_V6, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 43 | GetAddressMismatch(IPEndPoint(ip6_1, 443), IPEndPoint(ip6_1, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 44 | |
| 45 | EXPECT_EQ(QUIC_PORT_MISMATCH_V4_V4, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 46 | GetAddressMismatch(IPEndPoint(ip4_1, 80), IPEndPoint(ip4_1, 443))); |
| 47 | EXPECT_EQ( |
| 48 | QUIC_PORT_MISMATCH_V4_V4, |
| 49 | GetAddressMismatch(IPEndPoint(ip4_1, 80), IPEndPoint(ip4_mapped_1, 443))); |
[email protected] | dc657464 | 2014-05-20 21:32:59 | [diff] [blame] | 50 | EXPECT_EQ(QUIC_PORT_MISMATCH_V4_V4, |
| 51 | GetAddressMismatch(IPEndPoint(ip4_mapped_1, 80), |
| 52 | IPEndPoint(ip4_mapped_1, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 53 | EXPECT_EQ(QUIC_PORT_MISMATCH_V6_V6, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 54 | GetAddressMismatch(IPEndPoint(ip6_1, 80), IPEndPoint(ip6_1, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 55 | |
| 56 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V4_V4, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 57 | GetAddressMismatch(IPEndPoint(ip4_1, 443), IPEndPoint(ip4_2, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 58 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V4_V4, |
[email protected] | dc657464 | 2014-05-20 21:32:59 | [diff] [blame] | 59 | GetAddressMismatch(IPEndPoint(ip4_1, 443), |
| 60 | IPEndPoint(ip4_mapped_2, 443))); |
| 61 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V4_V4, |
| 62 | GetAddressMismatch(IPEndPoint(ip4_mapped_1, 443), |
| 63 | IPEndPoint(ip4_mapped_2, 443))); |
| 64 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V4_V4, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 65 | GetAddressMismatch(IPEndPoint(ip4_1, 80), IPEndPoint(ip4_2, 443))); |
| 66 | EXPECT_EQ( |
| 67 | QUIC_ADDRESS_MISMATCH_V4_V4, |
| 68 | GetAddressMismatch(IPEndPoint(ip4_1, 80), IPEndPoint(ip4_mapped_2, 443))); |
[email protected] | dc657464 | 2014-05-20 21:32:59 | [diff] [blame] | 69 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V4_V4, |
| 70 | GetAddressMismatch(IPEndPoint(ip4_mapped_1, 80), |
| 71 | IPEndPoint(ip4_mapped_2, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 72 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V6_V6, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 73 | GetAddressMismatch(IPEndPoint(ip6_1, 443), IPEndPoint(ip6_2, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 74 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V6_V6, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 75 | GetAddressMismatch(IPEndPoint(ip6_1, 80), IPEndPoint(ip6_2, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 76 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V4_V6, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 77 | GetAddressMismatch(IPEndPoint(ip4_1, 443), IPEndPoint(ip6_1, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 78 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V4_V6, |
[email protected] | dc657464 | 2014-05-20 21:32:59 | [diff] [blame] | 79 | GetAddressMismatch(IPEndPoint(ip4_mapped_1, 443), |
| 80 | IPEndPoint(ip6_1, 443))); |
| 81 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V4_V6, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 82 | GetAddressMismatch(IPEndPoint(ip4_1, 80), IPEndPoint(ip6_1, 443))); |
| 83 | EXPECT_EQ( |
| 84 | QUIC_ADDRESS_MISMATCH_V4_V6, |
| 85 | GetAddressMismatch(IPEndPoint(ip4_mapped_1, 80), IPEndPoint(ip6_1, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 86 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V6_V4, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 87 | GetAddressMismatch(IPEndPoint(ip6_1, 443), IPEndPoint(ip4_1, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 88 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V6_V4, |
[email protected] | dc657464 | 2014-05-20 21:32:59 | [diff] [blame] | 89 | GetAddressMismatch(IPEndPoint(ip6_1, 443), |
| 90 | IPEndPoint(ip4_mapped_1, 443))); |
| 91 | EXPECT_EQ(QUIC_ADDRESS_MISMATCH_V6_V4, |
rjshade | d5ced07 | 2015-12-18 19:26:02 | [diff] [blame] | 92 | GetAddressMismatch(IPEndPoint(ip6_1, 80), IPEndPoint(ip4_1, 443))); |
| 93 | EXPECT_EQ( |
| 94 | QUIC_ADDRESS_MISMATCH_V6_V4, |
| 95 | GetAddressMismatch(IPEndPoint(ip6_1, 80), IPEndPoint(ip4_mapped_1, 443))); |
[email protected] | 235dac08 | 2014-02-20 17:47:25 | [diff] [blame] | 96 | } |
| 97 | |
| 98 | } // namespace test |
| 99 | } // namespace net |