OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 <algorithm> | 5 #include <algorithm> |
6 #include <vector> | 6 #include <vector> |
7 | 7 |
8 #include "base/basictypes.h" | 8 #include "base/basictypes.h" |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
(...skipping 18 matching lines...) Expand all Loading... |
29 | 29 |
30 const bool kEntropyFlag[] = { | 30 const bool kEntropyFlag[] = { |
31 false, | 31 false, |
32 true, | 32 true, |
33 true, | 33 true, |
34 false, | 34 false, |
35 true, | 35 true, |
36 true, | 36 true, |
37 }; | 37 }; |
38 | 38 |
39 const bool kTestFecPacketEntropy = false; | |
40 | |
41 } // namespace | 39 } // namespace |
42 | 40 |
43 class QuicFecGroupTest : public ::testing::Test { | 41 class QuicFecGroupTest : public ::testing::Test { |
44 protected: | 42 protected: |
45 void RunTest(size_t num_packets, size_t lost_packet, bool out_of_order) { | 43 void RunTest(size_t num_packets, size_t lost_packet, bool out_of_order) { |
46 size_t max_len = strlen(kData[0]); | 44 size_t max_len = strlen(kData[0]); |
47 scoped_ptr<char[]> redundancy(new char[max_len]); | 45 scoped_ptr<char[]> redundancy(new char[max_len]); |
48 bool entropy_redundancy = false; | |
49 for (size_t packet = 0; packet < num_packets; ++packet) { | 46 for (size_t packet = 0; packet < num_packets; ++packet) { |
50 for (size_t i = 0; i < max_len; i++) { | 47 for (size_t i = 0; i < max_len; i++) { |
51 if (packet == 0) { | 48 if (packet == 0) { |
52 // Initialize to the first packet. | 49 // Initialize to the first packet. |
53 redundancy[i] = kData[0][i]; | 50 redundancy[i] = kData[0][i]; |
54 continue; | 51 continue; |
55 } | 52 } |
56 // XOR in the remaining packets. | 53 // XOR in the remaining packets. |
57 uint8 byte = i > strlen(kData[packet]) ? 0x00 : kData[packet][i]; | 54 uint8 byte = i > strlen(kData[packet]) ? 0x00 : kData[packet][i]; |
58 redundancy[i] = redundancy[i] ^ byte; | 55 redundancy[i] = redundancy[i] ^ byte; |
59 } | 56 } |
60 entropy_redundancy = (entropy_redundancy != kEntropyFlag[packet]); | |
61 } | 57 } |
62 | 58 |
63 QuicFecGroup group; | 59 QuicFecGroup group; |
64 | 60 |
65 // If we're out of order, send the FEC packet in the position of the | 61 // If we're out of order, send the FEC packet in the position of the |
66 // lost packet. Otherwise send all (non-missing) packets, then FEC. | 62 // lost packet. Otherwise send all (non-missing) packets, then FEC. |
67 if (out_of_order) { | 63 if (out_of_order) { |
68 // Update the FEC state for each non-lost packet. | 64 // Update the FEC state for each non-lost packet. |
69 for (size_t packet = 0; packet < num_packets; packet++) { | 65 for (size_t packet = 0; packet < num_packets; packet++) { |
70 if (packet == lost_packet) { | 66 if (packet == lost_packet) { |
71 ASSERT_FALSE(group.IsFinished()); | 67 ASSERT_FALSE(group.IsFinished()); |
72 QuicFecData fec; | 68 QuicFecData fec; |
73 fec.fec_group = 0; | 69 fec.fec_group = 0; |
74 fec.redundancy = StringPiece(redundancy.get(), strlen(kData[0])); | 70 fec.redundancy = StringPiece(redundancy.get(), strlen(kData[0])); |
75 ASSERT_TRUE(group.UpdateFec(num_packets, entropy_redundancy, fec)); | 71 ASSERT_TRUE(group.UpdateFec(num_packets, fec)); |
76 } else { | 72 } else { |
77 QuicPacketHeader header; | 73 QuicPacketHeader header; |
78 header.packet_sequence_number = packet; | 74 header.packet_sequence_number = packet; |
79 header.entropy_flag = kEntropyFlag[packet]; | 75 header.entropy_flag = kEntropyFlag[packet]; |
80 ASSERT_TRUE(group.Update(header, kData[packet])); | 76 ASSERT_TRUE(group.Update(header, kData[packet])); |
81 } | 77 } |
82 ASSERT_TRUE(group.CanRevive() == (packet == num_packets - 1)); | 78 ASSERT_TRUE(group.CanRevive() == (packet == num_packets - 1)); |
83 } | 79 } |
84 } else { | 80 } else { |
85 // Update the FEC state for each non-lost packet. | 81 // Update the FEC state for each non-lost packet. |
86 for (size_t packet = 0; packet < num_packets; packet++) { | 82 for (size_t packet = 0; packet < num_packets; packet++) { |
87 if (packet == lost_packet) { | 83 if (packet == lost_packet) { |
88 continue; | 84 continue; |
89 } | 85 } |
90 | 86 |
91 QuicPacketHeader header; | 87 QuicPacketHeader header; |
92 header.packet_sequence_number = packet; | 88 header.packet_sequence_number = packet; |
93 header.entropy_flag = kEntropyFlag[packet]; | 89 header.entropy_flag = kEntropyFlag[packet]; |
94 ASSERT_TRUE(group.Update(header, kData[packet])); | 90 ASSERT_TRUE(group.Update(header, kData[packet])); |
95 ASSERT_FALSE(group.CanRevive()); | 91 ASSERT_FALSE(group.CanRevive()); |
96 } | 92 } |
97 | 93 |
98 ASSERT_FALSE(group.IsFinished()); | 94 ASSERT_FALSE(group.IsFinished()); |
99 // Attempt to revive the missing packet. | 95 // Attempt to revive the missing packet. |
100 QuicFecData fec; | 96 QuicFecData fec; |
101 fec.fec_group = 0; | 97 fec.fec_group = 0; |
102 fec.redundancy = StringPiece(redundancy.get(), strlen(kData[0])); | 98 fec.redundancy = StringPiece(redundancy.get(), strlen(kData[0])); |
103 | 99 |
104 ASSERT_TRUE(group.UpdateFec(num_packets, entropy_redundancy, fec)); | 100 ASSERT_TRUE(group.UpdateFec(num_packets, fec)); |
105 } | 101 } |
106 QuicPacketHeader header; | 102 QuicPacketHeader header; |
107 char recovered[kMaxPacketSize]; | 103 char recovered[kMaxPacketSize]; |
108 ASSERT_TRUE(group.CanRevive()); | 104 ASSERT_TRUE(group.CanRevive()); |
109 size_t len = group.Revive(&header, recovered, arraysize(recovered)); | 105 size_t len = group.Revive(&header, recovered, arraysize(recovered)); |
110 ASSERT_NE(0u, len) | 106 ASSERT_NE(0u, len) |
111 << "Failed to revive packet " << lost_packet << " out of " | 107 << "Failed to revive packet " << lost_packet << " out of " |
112 << num_packets; | 108 << num_packets; |
113 EXPECT_EQ(lost_packet, header.packet_sequence_number) | 109 EXPECT_EQ(lost_packet, header.packet_sequence_number) |
114 << "Failed to revive packet " << lost_packet << " out of " | 110 << "Failed to revive packet " << lost_packet << " out of " |
115 << num_packets; | 111 << num_packets; |
116 EXPECT_EQ(kEntropyFlag[lost_packet], header.entropy_flag); | 112 // Revived packets have an unknown entropy. |
| 113 EXPECT_FALSE(header.entropy_flag); |
117 ASSERT_GE(len, strlen(kData[lost_packet])) << "Incorrect length"; | 114 ASSERT_GE(len, strlen(kData[lost_packet])) << "Incorrect length"; |
118 for (size_t i = 0; i < strlen(kData[lost_packet]); i++) { | 115 for (size_t i = 0; i < strlen(kData[lost_packet]); i++) { |
119 EXPECT_EQ(kData[lost_packet][i], recovered[i]); | 116 EXPECT_EQ(kData[lost_packet][i], recovered[i]); |
120 } | 117 } |
121 ASSERT_TRUE(group.IsFinished()); | 118 ASSERT_TRUE(group.IsFinished()); |
122 } | 119 } |
123 }; | 120 }; |
124 | 121 |
125 TEST_F(QuicFecGroupTest, UpdateAndRevive) { | 122 TEST_F(QuicFecGroupTest, UpdateAndRevive) { |
126 RunTest(2, 0, false); | 123 RunTest(2, 0, false); |
(...skipping 24 matching lines...) Expand all Loading... |
151 | 148 |
152 QuicPacketHeader header; | 149 QuicPacketHeader header; |
153 header.packet_sequence_number = 3; | 150 header.packet_sequence_number = 3; |
154 group.Update(header, data1); | 151 group.Update(header, data1); |
155 | 152 |
156 QuicFecData fec; | 153 QuicFecData fec; |
157 fec.fec_group = 1; | 154 fec.fec_group = 1; |
158 fec.redundancy = redundancy; | 155 fec.redundancy = redundancy; |
159 | 156 |
160 header.packet_sequence_number = 2; | 157 header.packet_sequence_number = 2; |
161 ASSERT_FALSE(group.UpdateFec(2, kTestFecPacketEntropy, fec)); | 158 ASSERT_FALSE(group.UpdateFec(2, fec)); |
162 } | 159 } |
163 | 160 |
164 TEST_F(QuicFecGroupTest, ProtectsPacketsBefore) { | 161 TEST_F(QuicFecGroupTest, ProtectsPacketsBefore) { |
165 QuicPacketHeader header; | 162 QuicPacketHeader header; |
166 header.packet_sequence_number = 3; | 163 header.packet_sequence_number = 3; |
167 | 164 |
168 QuicFecGroup group; | 165 QuicFecGroup group; |
169 ASSERT_TRUE(group.Update(header, kData[0])); | 166 ASSERT_TRUE(group.Update(header, kData[0])); |
170 | 167 |
171 EXPECT_FALSE(group.ProtectsPacketsBefore(1)); | 168 EXPECT_FALSE(group.ProtectsPacketsBefore(1)); |
(...skipping 28 matching lines...) Expand all Loading... |
200 EXPECT_TRUE(group.ProtectsPacketsBefore(9)); | 197 EXPECT_TRUE(group.ProtectsPacketsBefore(9)); |
201 EXPECT_TRUE(group.ProtectsPacketsBefore(50)); | 198 EXPECT_TRUE(group.ProtectsPacketsBefore(50)); |
202 } | 199 } |
203 | 200 |
204 TEST_F(QuicFecGroupTest, ProtectsPacketsBeforeWithFecData) { | 201 TEST_F(QuicFecGroupTest, ProtectsPacketsBeforeWithFecData) { |
205 QuicFecData fec; | 202 QuicFecData fec; |
206 fec.fec_group = 2; | 203 fec.fec_group = 2; |
207 fec.redundancy = kData[0]; | 204 fec.redundancy = kData[0]; |
208 | 205 |
209 QuicFecGroup group; | 206 QuicFecGroup group; |
210 ASSERT_TRUE(group.UpdateFec(3, kTestFecPacketEntropy, fec)); | 207 ASSERT_TRUE(group.UpdateFec(3, fec)); |
211 | 208 |
212 EXPECT_FALSE(group.ProtectsPacketsBefore(1)); | 209 EXPECT_FALSE(group.ProtectsPacketsBefore(1)); |
213 EXPECT_FALSE(group.ProtectsPacketsBefore(2)); | 210 EXPECT_FALSE(group.ProtectsPacketsBefore(2)); |
214 EXPECT_TRUE(group.ProtectsPacketsBefore(3)); | 211 EXPECT_TRUE(group.ProtectsPacketsBefore(3)); |
215 EXPECT_TRUE(group.ProtectsPacketsBefore(4)); | 212 EXPECT_TRUE(group.ProtectsPacketsBefore(4)); |
216 EXPECT_TRUE(group.ProtectsPacketsBefore(5)); | 213 EXPECT_TRUE(group.ProtectsPacketsBefore(5)); |
217 EXPECT_TRUE(group.ProtectsPacketsBefore(50)); | 214 EXPECT_TRUE(group.ProtectsPacketsBefore(50)); |
218 } | 215 } |
219 | 216 |
220 } // namespace net | 217 } // namespace net |
OLD | NEW |