Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(476)

Side by Side Diff: net/quic/iovector_test.cc

Issue 662553002: Convert ARRAYSIZE_UNSAFE -> arraysize in net/. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 6 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « net/quic/crypto/crypto_utils_test.cc ('k') | net/quic/quic_socket_address_coder_test.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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/iovector.h" 5 #include "net/quic/iovector.h"
6 6
7 #include <string.h> 7 #include <string.h>
8 8
9 #include "base/logging.h" 9 #include "base/logging.h"
10 #include "testing/gtest/include/gtest/gtest.h" 10 #include "testing/gtest/include/gtest/gtest.h"
11 11
12 using std::string; 12 using std::string;
13 13
14 namespace net { 14 namespace net {
15 namespace test { 15 namespace test {
16 namespace { 16 namespace {
17 17
18 const char* const test_data[] = { 18 const char* const test_data[] = {
19 "test string 1, a medium size one.", 19 "test string 1, a medium size one.",
20 "test string2", 20 "test string2",
21 "test string 3, a looooooooooooong loooooooooooooooong string" 21 "test string 3, a looooooooooooong loooooooooooooooong string"
22 }; 22 };
23 23
24 TEST(IOVectorTest, CopyConstructor) { 24 TEST(IOVectorTest, CopyConstructor) {
25 IOVector iov1; 25 IOVector iov1;
26 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) { 26 for (size_t i = 0; i < arraysize(test_data); ++i) {
27 iov1.Append(const_cast<char*>(test_data[i]), strlen(test_data[i])); 27 iov1.Append(const_cast<char*>(test_data[i]), strlen(test_data[i]));
28 } 28 }
29 IOVector iov2 = iov1; 29 IOVector iov2 = iov1;
30 EXPECT_EQ(iov2.Size(), iov1.Size()); 30 EXPECT_EQ(iov2.Size(), iov1.Size());
31 for (size_t i = 0; i < iov2.Size(); ++i) { 31 for (size_t i = 0; i < iov2.Size(); ++i) {
32 EXPECT_TRUE(iov2.iovec()[i].iov_base == iov1.iovec()[i].iov_base); 32 EXPECT_TRUE(iov2.iovec()[i].iov_base == iov1.iovec()[i].iov_base);
33 EXPECT_EQ(iov2.iovec()[i].iov_len, iov1.iovec()[i].iov_len); 33 EXPECT_EQ(iov2.iovec()[i].iov_len, iov1.iovec()[i].iov_len);
34 } 34 }
35 EXPECT_EQ(iov2.TotalBufferSize(), iov1.TotalBufferSize()); 35 EXPECT_EQ(iov2.TotalBufferSize(), iov1.TotalBufferSize());
36 } 36 }
37 37
38 TEST(IOVectorTest, AssignmentOperator) { 38 TEST(IOVectorTest, AssignmentOperator) {
39 IOVector iov1; 39 IOVector iov1;
40 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) { 40 for (size_t i = 0; i < arraysize(test_data); ++i) {
41 iov1.Append(const_cast<char*>(test_data[i]), strlen(test_data[i])); 41 iov1.Append(const_cast<char*>(test_data[i]), strlen(test_data[i]));
42 } 42 }
43 IOVector iov2; 43 IOVector iov2;
44 iov2.Append(const_cast<char*>("ephemeral string"), 16); 44 iov2.Append(const_cast<char*>("ephemeral string"), 16);
45 // The following assignment results in a shallow copy; 45 // The following assignment results in a shallow copy;
46 // both IOVectors point to the same underlying data. 46 // both IOVectors point to the same underlying data.
47 iov2 = iov1; 47 iov2 = iov1;
48 EXPECT_EQ(iov2.Size(), iov1.Size()); 48 EXPECT_EQ(iov2.Size(), iov1.Size());
49 for (size_t i = 0; i < iov2.Size(); ++i) { 49 for (size_t i = 0; i < iov2.Size(); ++i) {
50 EXPECT_TRUE(iov2.iovec()[i].iov_base == iov1.iovec()[i].iov_base); 50 EXPECT_TRUE(iov2.iovec()[i].iov_base == iov1.iovec()[i].iov_base);
51 EXPECT_EQ(iov2.iovec()[i].iov_len, iov1.iovec()[i].iov_len); 51 EXPECT_EQ(iov2.iovec()[i].iov_len, iov1.iovec()[i].iov_len);
52 } 52 }
53 EXPECT_EQ(iov2.TotalBufferSize(), iov1.TotalBufferSize()); 53 EXPECT_EQ(iov2.TotalBufferSize(), iov1.TotalBufferSize());
54 } 54 }
55 55
56 TEST(IOVectorTest, Append) { 56 TEST(IOVectorTest, Append) {
57 IOVector iov; 57 IOVector iov;
58 int length = 0; 58 int length = 0;
59 const struct iovec* iov2 = iov.iovec(); 59 const struct iovec* iov2 = iov.iovec();
60 60
61 ASSERT_EQ(0u, iov.Size()); 61 ASSERT_EQ(0u, iov.Size());
62 ASSERT_TRUE(iov2 == nullptr); 62 ASSERT_TRUE(iov2 == nullptr);
63 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) { 63 for (size_t i = 0; i < arraysize(test_data); ++i) {
64 const int str_len = strlen(test_data[i]); 64 const int str_len = strlen(test_data[i]);
65 const int append_len = str_len / 2; 65 const int append_len = str_len / 2;
66 // This should append a new block 66 // This should append a new block
67 iov.Append(const_cast<char*>(test_data[i]), append_len); 67 iov.Append(const_cast<char*>(test_data[i]), append_len);
68 length += append_len; 68 length += append_len;
69 ASSERT_EQ(i + 1, static_cast<size_t>(iov.Size())); 69 ASSERT_EQ(i + 1, static_cast<size_t>(iov.Size()));
70 ASSERT_TRUE(iov.LastBlockEnd() == test_data[i] + append_len); 70 ASSERT_TRUE(iov.LastBlockEnd() == test_data[i] + append_len);
71 // This should just lengthen the existing block. 71 // This should just lengthen the existing block.
72 iov.Append(const_cast<char*>(test_data[i] + append_len), 72 iov.Append(const_cast<char*>(test_data[i] + append_len),
73 str_len - append_len); 73 str_len - append_len);
(...skipping 10 matching lines...) Expand all
84 } 84 }
85 } 85 }
86 86
87 TEST(IOVectorTest, AppendIovec) { 87 TEST(IOVectorTest, AppendIovec) {
88 IOVector iov; 88 IOVector iov;
89 const struct iovec test_iov[] = { 89 const struct iovec test_iov[] = {
90 {const_cast<char*>("foo"), 3}, 90 {const_cast<char*>("foo"), 3},
91 {const_cast<char*>("bar"), 3}, 91 {const_cast<char*>("bar"), 3},
92 {const_cast<char*>("buzzzz"), 6} 92 {const_cast<char*>("buzzzz"), 6}
93 }; 93 };
94 iov.AppendIovec(test_iov, ARRAYSIZE_UNSAFE(test_iov)); 94 iov.AppendIovec(test_iov, arraysize(test_iov));
95 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_iov); ++i) { 95 for (size_t i = 0; i < arraysize(test_iov); ++i) {
96 EXPECT_EQ(test_iov[i].iov_base, iov.iovec()[i].iov_base); 96 EXPECT_EQ(test_iov[i].iov_base, iov.iovec()[i].iov_base);
97 EXPECT_EQ(test_iov[i].iov_len, iov.iovec()[i].iov_len); 97 EXPECT_EQ(test_iov[i].iov_len, iov.iovec()[i].iov_len);
98 } 98 }
99 99
100 // Test AppendIovecAtMostBytes. 100 // Test AppendIovecAtMostBytes.
101 iov.Clear(); 101 iov.Clear();
102 // Stop in the middle of a block. 102 // Stop in the middle of a block.
103 EXPECT_EQ(5u, iov.AppendIovecAtMostBytes(test_iov, ARRAYSIZE_UNSAFE(test_iov), 103 EXPECT_EQ(5u, iov.AppendIovecAtMostBytes(test_iov, arraysize(test_iov), 5));
104 5));
105 EXPECT_EQ(5u, iov.TotalBufferSize()); 104 EXPECT_EQ(5u, iov.TotalBufferSize());
106 iov.Append(static_cast<char*>(test_iov[1].iov_base) + 2, 1); 105 iov.Append(static_cast<char*>(test_iov[1].iov_base) + 2, 1);
107 // Make sure the boundary case, where max_bytes == size of block also works. 106 // Make sure the boundary case, where max_bytes == size of block also works.
108 EXPECT_EQ(6u, iov.AppendIovecAtMostBytes(&test_iov[2], 1, 6)); 107 EXPECT_EQ(6u, iov.AppendIovecAtMostBytes(&test_iov[2], 1, 6));
109 ASSERT_LE(ARRAYSIZE_UNSAFE(test_iov), static_cast<size_t>(iov.Size())); 108 ASSERT_LE(arraysize(test_iov), static_cast<size_t>(iov.Size()));
110 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_iov); ++i) { 109 for (size_t i = 0; i < arraysize(test_iov); ++i) {
111 EXPECT_EQ(test_iov[i].iov_base, iov.iovec()[i].iov_base); 110 EXPECT_EQ(test_iov[i].iov_base, iov.iovec()[i].iov_base);
112 EXPECT_EQ(test_iov[i].iov_len, iov.iovec()[i].iov_len); 111 EXPECT_EQ(test_iov[i].iov_len, iov.iovec()[i].iov_len);
113 } 112 }
114 } 113 }
115 114
116 TEST(IOVectorTest, ConsumeHalfBlocks) { 115 TEST(IOVectorTest, ConsumeHalfBlocks) {
117 IOVector iov; 116 IOVector iov;
118 int length = 0; 117 int length = 0;
119 118
120 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) { 119 for (size_t i = 0; i < arraysize(test_data); ++i) {
121 const int str_len = strlen(test_data[i]); 120 const int str_len = strlen(test_data[i]);
122 iov.Append(const_cast<char*>(test_data[i]), str_len); 121 iov.Append(const_cast<char*>(test_data[i]), str_len);
123 length += str_len; 122 length += str_len;
124 } 123 }
125 const char* endp = iov.LastBlockEnd(); 124 const char* endp = iov.LastBlockEnd();
126 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) { 125 for (size_t i = 0; i < arraysize(test_data); ++i) {
127 const struct iovec* iov2 = iov.iovec(); 126 const struct iovec* iov2 = iov.iovec();
128 const size_t str_len = strlen(test_data[i]); 127 const size_t str_len = strlen(test_data[i]);
129 size_t tmp = str_len / 2; 128 size_t tmp = str_len / 2;
130 129
131 ASSERT_TRUE(iov2 != nullptr); 130 ASSERT_TRUE(iov2 != nullptr);
132 ASSERT_TRUE(iov2[0].iov_base == test_data[i]); 131 ASSERT_TRUE(iov2[0].iov_base == test_data[i]);
133 ASSERT_EQ(str_len, iov2[0].iov_len); 132 ASSERT_EQ(str_len, iov2[0].iov_len);
134 133
135 // Consume half of the first block. 134 // Consume half of the first block.
136 size_t consumed = iov.Consume(tmp); 135 size_t consumed = iov.Consume(tmp);
137 ASSERT_EQ(tmp, consumed); 136 ASSERT_EQ(tmp, consumed);
138 ASSERT_EQ(ARRAYSIZE_UNSAFE(test_data) - i, static_cast<size_t>(iov.Size())); 137 ASSERT_EQ(arraysize(test_data) - i, static_cast<size_t>(iov.Size()));
139 iov2 = iov.iovec(); 138 iov2 = iov.iovec();
140 ASSERT_TRUE(iov2 != nullptr); 139 ASSERT_TRUE(iov2 != nullptr);
141 ASSERT_TRUE(iov2[0].iov_base == test_data[i] + tmp); 140 ASSERT_TRUE(iov2[0].iov_base == test_data[i] + tmp);
142 ASSERT_EQ(iov2[0].iov_len, str_len - tmp); 141 ASSERT_EQ(iov2[0].iov_len, str_len - tmp);
143 142
144 // Consume the rest of the first block 143 // Consume the rest of the first block
145 consumed = iov.Consume(str_len - tmp); 144 consumed = iov.Consume(str_len - tmp);
146 ASSERT_EQ(str_len - tmp, consumed); 145 ASSERT_EQ(str_len - tmp, consumed);
147 ASSERT_EQ(ARRAYSIZE_UNSAFE(test_data) - i - 1, 146 ASSERT_EQ(arraysize(test_data) - i - 1, static_cast<size_t>(iov.Size()));
148 static_cast<size_t>(iov.Size()));
149 iov2 = iov.iovec(); 147 iov2 = iov.iovec();
150 if (iov.Size() > 0) { 148 if (iov.Size() > 0) {
151 ASSERT_TRUE(iov2 != nullptr); 149 ASSERT_TRUE(iov2 != nullptr);
152 ASSERT_TRUE(iov.LastBlockEnd() == endp); 150 ASSERT_TRUE(iov.LastBlockEnd() == endp);
153 } else { 151 } else {
154 ASSERT_TRUE(iov2 == nullptr); 152 ASSERT_TRUE(iov2 == nullptr);
155 ASSERT_TRUE(iov.LastBlockEnd() == nullptr); 153 ASSERT_TRUE(iov.LastBlockEnd() == nullptr);
156 } 154 }
157 } 155 }
158 } 156 }
159 157
160 TEST(IOVectorTest, ConsumeTwoAndHalfBlocks) { 158 TEST(IOVectorTest, ConsumeTwoAndHalfBlocks) {
161 IOVector iov; 159 IOVector iov;
162 int length = 0; 160 int length = 0;
163 161
164 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) { 162 for (size_t i = 0; i < arraysize(test_data); ++i) {
165 const int str_len = strlen(test_data[i]); 163 const int str_len = strlen(test_data[i]);
166 iov.Append(const_cast<char*>(test_data[i]), str_len); 164 iov.Append(const_cast<char*>(test_data[i]), str_len);
167 length += str_len; 165 length += str_len;
168 } 166 }
169 const size_t last_len = strlen(test_data[ARRAYSIZE_UNSAFE(test_data) - 1]); 167 const size_t last_len = strlen(test_data[arraysize(test_data) - 1]);
170 const size_t half_len = last_len / 2; 168 const size_t half_len = last_len / 2;
171 169
172 const char* endp = iov.LastBlockEnd(); 170 const char* endp = iov.LastBlockEnd();
173 size_t consumed = iov.Consume(length - half_len); 171 size_t consumed = iov.Consume(length - half_len);
174 ASSERT_EQ(length - half_len, consumed); 172 ASSERT_EQ(length - half_len, consumed);
175 const struct iovec* iov2 = iov.iovec(); 173 const struct iovec* iov2 = iov.iovec();
176 ASSERT_TRUE(iov2 != nullptr); 174 ASSERT_TRUE(iov2 != nullptr);
177 ASSERT_EQ(1u, iov.Size()); 175 ASSERT_EQ(1u, iov.Size());
178 ASSERT_TRUE(iov2[0].iov_base == 176 ASSERT_TRUE(iov2[0].iov_base ==
179 test_data[ARRAYSIZE_UNSAFE(test_data) - 1] + last_len - half_len); 177 test_data[arraysize(test_data) - 1] + last_len - half_len);
180 ASSERT_EQ(half_len, iov2[0].iov_len); 178 ASSERT_EQ(half_len, iov2[0].iov_len);
181 ASSERT_TRUE(iov.LastBlockEnd() == endp); 179 ASSERT_TRUE(iov.LastBlockEnd() == endp);
182 180
183 consumed = iov.Consume(half_len); 181 consumed = iov.Consume(half_len);
184 ASSERT_EQ(half_len, consumed); 182 ASSERT_EQ(half_len, consumed);
185 iov2 = iov.iovec(); 183 iov2 = iov.iovec();
186 ASSERT_EQ(0u, iov.Size()); 184 ASSERT_EQ(0u, iov.Size());
187 ASSERT_TRUE(iov2 == nullptr); 185 ASSERT_TRUE(iov2 == nullptr);
188 ASSERT_TRUE(iov.LastBlockEnd() == nullptr); 186 ASSERT_TRUE(iov.LastBlockEnd() == nullptr);
189 } 187 }
190 188
191 TEST(IOVectorTest, ConsumeTooMuch) { 189 TEST(IOVectorTest, ConsumeTooMuch) {
192 IOVector iov; 190 IOVector iov;
193 int length = 0; 191 int length = 0;
194 192
195 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) { 193 for (size_t i = 0; i < arraysize(test_data); ++i) {
196 const int str_len = strlen(test_data[i]); 194 const int str_len = strlen(test_data[i]);
197 iov.Append(const_cast<char*>(test_data[i]), str_len); 195 iov.Append(const_cast<char*>(test_data[i]), str_len);
198 length += str_len; 196 length += str_len;
199 } 197 }
200 198
201 int consumed = 0; 199 int consumed = 0;
202 consumed = iov.Consume(length); 200 consumed = iov.Consume(length);
203 // TODO(rtenneti): enable when chromium supports EXPECT_DFATAL. 201 // TODO(rtenneti): enable when chromium supports EXPECT_DFATAL.
204 /* 202 /*
205 EXPECT_DFATAL( 203 EXPECT_DFATAL(
206 {consumed = iov.Consume(length + 1);}, 204 {consumed = iov.Consume(length + 1);},
207 "Attempting to consume 1 non-existent bytes."); 205 "Attempting to consume 1 non-existent bytes.");
208 */ 206 */
209 ASSERT_EQ(length, consumed); 207 ASSERT_EQ(length, consumed);
210 const struct iovec* iov2 = iov.iovec(); 208 const struct iovec* iov2 = iov.iovec();
211 ASSERT_EQ(0u, iov.Size()); 209 ASSERT_EQ(0u, iov.Size());
212 ASSERT_TRUE(iov2 == nullptr); 210 ASSERT_TRUE(iov2 == nullptr);
213 ASSERT_TRUE(iov.LastBlockEnd() == nullptr); 211 ASSERT_TRUE(iov.LastBlockEnd() == nullptr);
214 } 212 }
215 213
216 TEST(IOVectorTest, Clear) { 214 TEST(IOVectorTest, Clear) {
217 IOVector iov; 215 IOVector iov;
218 int length = 0; 216 int length = 0;
219 217
220 for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) { 218 for (size_t i = 0; i < arraysize(test_data); ++i) {
221 const int str_len = strlen(test_data[i]); 219 const int str_len = strlen(test_data[i]);
222 iov.Append(const_cast<char*>(test_data[i]), str_len); 220 iov.Append(const_cast<char*>(test_data[i]), str_len);
223 length += str_len; 221 length += str_len;
224 } 222 }
225 const struct iovec* iov2 = iov.iovec(); 223 const struct iovec* iov2 = iov.iovec();
226 ASSERT_TRUE(iov2 != nullptr); 224 ASSERT_TRUE(iov2 != nullptr);
227 ASSERT_EQ(ARRAYSIZE_UNSAFE(test_data), static_cast<size_t>(iov.Size())); 225 ASSERT_EQ(arraysize(test_data), static_cast<size_t>(iov.Size()));
228 226
229 iov.Clear(); 227 iov.Clear();
230 iov2 = iov.iovec(); 228 iov2 = iov.iovec();
231 ASSERT_EQ(0u, iov.Size()); 229 ASSERT_EQ(0u, iov.Size());
232 ASSERT_TRUE(iov2 == nullptr); 230 ASSERT_TRUE(iov2 == nullptr);
233 } 231 }
234 232
235 TEST(IOVectorTest, Capacity) { 233 TEST(IOVectorTest, Capacity) {
236 IOVector iov; 234 IOVector iov;
237 // Note: IOVector merges adjacent Appends() into a single iov. 235 // Note: IOVector merges adjacent Appends() into a single iov.
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after
274 ASSERT_EQ(2u, iov2.Size()); 272 ASSERT_EQ(2u, iov2.Size());
275 EXPECT_EQ(&a[0], iov2.iovec()[0].iov_base); 273 EXPECT_EQ(&a[0], iov2.iovec()[0].iov_base);
276 EXPECT_EQ(1u, iov2.iovec()[0].iov_len); 274 EXPECT_EQ(1u, iov2.iovec()[0].iov_len);
277 EXPECT_EQ(&b[0], iov2.iovec()[1].iov_base); 275 EXPECT_EQ(&b[0], iov2.iovec()[1].iov_base);
278 EXPECT_EQ(1u, iov2.iovec()[1].iov_len); 276 EXPECT_EQ(1u, iov2.iovec()[1].iov_len);
279 } 277 }
280 278
281 } // namespace 279 } // namespace
282 } // namespace test 280 } // namespace test
283 } // namespace net 281 } // namespace net
OLDNEW
« no previous file with comments | « net/quic/crypto/crypto_utils_test.cc ('k') | net/quic/quic_socket_address_coder_test.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698