| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/port_suggester.h" | 5 #include "net/quic/port_suggester.h" |
| 6 | 6 |
| 7 #include <set> | 7 #include <set> |
| 8 | 8 |
| 9 #include "base/basictypes.h" | 9 #include "base/basictypes.h" |
| 10 #include "net/base/host_port_pair.h" | 10 #include "net/base/host_port_pair.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
| 12 | 12 |
| 13 namespace net { | 13 namespace net { |
| 14 namespace test { | 14 namespace test { |
| 15 | 15 |
| 16 class PortSuggesterTest : public ::testing::Test { | 16 class PortSuggesterTest : public ::testing::Test { |
| 17 protected: | 17 protected: |
| 18 PortSuggesterTest() | 18 PortSuggesterTest() |
| 19 : entropy_(1345689), | 19 : entropy_(1345689), |
| 20 min_ephemeral_port_(1025), | 20 min_ephemeral_port_(1025), |
| 21 max_ephemeral_port_(65535) { | 21 max_ephemeral_port_(65535) {} |
| 22 } | |
| 23 | 22 |
| 24 uint64 entropy_; | 23 uint64 entropy_; |
| 25 int min_ephemeral_port_; | 24 int min_ephemeral_port_; |
| 26 int max_ephemeral_port_; | 25 int max_ephemeral_port_; |
| 27 }; | 26 }; |
| 28 | 27 |
| 29 TEST_F(PortSuggesterTest, SmallRangeTest) { | 28 TEST_F(PortSuggesterTest, SmallRangeTest) { |
| 30 // When the range is small (one wide), we always get that as our answer. | 29 // When the range is small (one wide), we always get that as our answer. |
| 31 scoped_refptr<PortSuggester> port_suggester = | 30 scoped_refptr<PortSuggester> port_suggester = |
| 32 new PortSuggester(HostPortPair("www.example.com", 443), entropy_); | 31 new PortSuggester(HostPortPair("www.example.com", 443), entropy_); |
| 33 // Test this for a few different (small) ranges. | 32 // Test this for a few different (small) ranges. |
| 34 for (int port = 2000; port < 2010; ++port) { | 33 for (int port = 2000; port < 2010; ++port) { |
| 35 // Use |port| for both |min| and |max| delimiting the suggestion range. | 34 // Use |port| for both |min| and |max| delimiting the suggestion range. |
| 36 EXPECT_EQ(port, port_suggester->SuggestPort(port, port)); | 35 EXPECT_EQ(port, port_suggester->SuggestPort(port, port)); |
| 37 EXPECT_EQ(port, port_suggester->previous_suggestion()); | 36 EXPECT_EQ(port, port_suggester->previous_suggestion()); |
| 38 } | 37 } |
| 39 } | 38 } |
| 40 | 39 |
| 41 TEST_F(PortSuggesterTest, SuggestAllPorts) { | 40 TEST_F(PortSuggesterTest, SuggestAllPorts) { |
| 42 // We should eventually fill out any range, but we'll just ensure that we | 41 // We should eventually fill out any range, but we'll just ensure that we |
| 43 // fill out a small range of ports. | 42 // fill out a small range of ports. |
| 44 scoped_refptr<PortSuggester> port_suggester = | 43 scoped_refptr<PortSuggester> port_suggester = |
| 45 new PortSuggester(HostPortPair("www.example.com", 443), entropy_); | 44 new PortSuggester(HostPortPair("www.example.com", 443), entropy_); |
| 46 std::set<int> ports; | 45 std::set<int> ports; |
| 47 const uint32 port_range = 20; | 46 const uint32 port_range = 20; |
| 48 const int insertion_limit = 200; // We should be done by then. | 47 const int insertion_limit = 200; // We should be done by then. |
| 49 for (int i = 0; i < insertion_limit; ++i) { | 48 for (int i = 0; i < insertion_limit; ++i) { |
| 50 ports.insert(port_suggester->SuggestPort(min_ephemeral_port_, | 49 ports.insert(port_suggester->SuggestPort( |
| 51 min_ephemeral_port_ + port_range - 1)); | 50 min_ephemeral_port_, min_ephemeral_port_ + port_range - 1)); |
| 52 if (ports.size() == port_range) { | 51 if (ports.size() == port_range) { |
| 53 break; | 52 break; |
| 54 } | 53 } |
| 55 } | 54 } |
| 56 EXPECT_EQ(port_range, ports.size()); | 55 EXPECT_EQ(port_range, ports.size()); |
| 57 } | 56 } |
| 58 | 57 |
| 59 TEST_F(PortSuggesterTest, AvoidDuplication) { | 58 TEST_F(PortSuggesterTest, AvoidDuplication) { |
| 60 // When the range is large, duplicates are rare, but we'll ask for a few | 59 // When the range is large, duplicates are rare, but we'll ask for a few |
| 61 // suggestions and make sure they are unique. | 60 // suggestions and make sure they are unique. |
| 62 scoped_refptr<PortSuggester> port_suggester = | 61 scoped_refptr<PortSuggester> port_suggester = |
| 63 new PortSuggester(HostPortPair("www.example.com", 80), entropy_); | 62 new PortSuggester(HostPortPair("www.example.com", 80), entropy_); |
| 64 std::set<int> ports; | 63 std::set<int> ports; |
| 65 const size_t port_count = 200; | 64 const size_t port_count = 200; |
| 66 for (size_t i = 0; i < port_count; ++i) { | 65 for (size_t i = 0; i < port_count; ++i) { |
| 67 ports.insert(port_suggester->SuggestPort(min_ephemeral_port_, | 66 ports.insert( |
| 68 max_ephemeral_port_)); | 67 port_suggester->SuggestPort(min_ephemeral_port_, max_ephemeral_port_)); |
| 69 } | 68 } |
| 70 EXPECT_EQ(port_suggester->call_count(), port_count); | 69 EXPECT_EQ(port_suggester->call_count(), port_count); |
| 71 EXPECT_EQ(port_count, ports.size()); | 70 EXPECT_EQ(port_count, ports.size()); |
| 72 } | 71 } |
| 73 | 72 |
| 74 TEST_F(PortSuggesterTest, ConsistentPorts) { | 73 TEST_F(PortSuggesterTest, ConsistentPorts) { |
| 75 // For given hostname, port, and entropy, we should always get the same | 74 // For given hostname, port, and entropy, we should always get the same |
| 76 // suggestions. | 75 // suggestions. |
| 77 scoped_refptr<PortSuggester> port_suggester1 = | 76 scoped_refptr<PortSuggester> port_suggester1 = |
| 78 new PortSuggester(HostPortPair("www.example.com", 443), entropy_); | 77 new PortSuggester(HostPortPair("www.example.com", 443), entropy_); |
| 79 scoped_refptr<PortSuggester> port_suggester2 = | 78 scoped_refptr<PortSuggester> port_suggester2 = |
| 80 new PortSuggester(HostPortPair("www.example.com", 443), entropy_); | 79 new PortSuggester(HostPortPair("www.example.com", 443), entropy_); |
| 81 for (int test_count = 20; test_count > 0; --test_count) { | 80 for (int test_count = 20; test_count > 0; --test_count) { |
| 82 EXPECT_EQ(port_suggester1->SuggestPort(min_ephemeral_port_, | 81 EXPECT_EQ( |
| 83 min_ephemeral_port_), | 82 port_suggester1->SuggestPort(min_ephemeral_port_, min_ephemeral_port_), |
| 84 port_suggester2->SuggestPort(min_ephemeral_port_, | 83 port_suggester2->SuggestPort(min_ephemeral_port_, min_ephemeral_port_)); |
| 85 min_ephemeral_port_)); | |
| 86 } | 84 } |
| 87 } | 85 } |
| 88 | 86 |
| 89 TEST_F(PortSuggesterTest, DifferentHostPortEntropy) { | 87 TEST_F(PortSuggesterTest, DifferentHostPortEntropy) { |
| 90 // When we have different hosts, port, or entropy, we probably won't collide. | 88 // When we have different hosts, port, or entropy, we probably won't collide. |
| 91 scoped_refptr<PortSuggester> port_suggester[] = { | 89 scoped_refptr<PortSuggester> port_suggester[] = { |
| 92 new PortSuggester(HostPortPair("www.example.com", 80), entropy_), | 90 new PortSuggester(HostPortPair("www.example.com", 80), entropy_), |
| 93 new PortSuggester(HostPortPair("www.example.ORG", 80), entropy_), | 91 new PortSuggester(HostPortPair("www.example.ORG", 80), entropy_), |
| 94 new PortSuggester(HostPortPair("www.example.com", 443), entropy_), | 92 new PortSuggester(HostPortPair("www.example.com", 443), entropy_), |
| 95 new PortSuggester(HostPortPair("www.example.com", 80), entropy_ + 123456), | 93 new PortSuggester(HostPortPair("www.example.com", 80), entropy_ + 123456), |
| 96 }; | 94 }; |
| 97 | 95 |
| 98 std::set<int> ports; | 96 std::set<int> ports; |
| 99 const int port_count = 40; | 97 const int port_count = 40; |
| 100 size_t insertion_count = 0; | 98 size_t insertion_count = 0; |
| 101 for (size_t j = 0; j < arraysize(port_suggester); ++j) { | 99 for (size_t j = 0; j < arraysize(port_suggester); ++j) { |
| 102 for (int i = 0; i < port_count; ++i) { | 100 for (int i = 0; i < port_count; ++i) { |
| 103 ports.insert(port_suggester[j]->SuggestPort(min_ephemeral_port_, | 101 ports.insert(port_suggester[j]->SuggestPort(min_ephemeral_port_, |
| 104 max_ephemeral_port_)); | 102 max_ephemeral_port_)); |
| 105 ++insertion_count; | 103 ++insertion_count; |
| 106 } | 104 } |
| 107 } | 105 } |
| 108 EXPECT_EQ(insertion_count, ports.size()); | 106 EXPECT_EQ(insertion_count, ports.size()); |
| 109 } | 107 } |
| 110 | 108 |
| 111 } // namespace test | 109 } // namespace test |
| 112 } // namespace net | 110 } // namespace net |
| OLD | NEW |