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 <string> | 5 #include <string> |
6 | 6 |
7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
9 #include "base/pickle.h" | 9 #include "base/pickle.h" |
10 #include "base/string16.h" | 10 #include "base/string16.h" |
(...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
94 VerifyResult(pickle2); | 94 VerifyResult(pickle2); |
95 | 95 |
96 // test operator= | 96 // test operator= |
97 Pickle pickle3; | 97 Pickle pickle3; |
98 pickle3 = pickle; | 98 pickle3 = pickle; |
99 VerifyResult(pickle3); | 99 VerifyResult(pickle3); |
100 } | 100 } |
101 | 101 |
102 // Tests that we can handle really small buffers. | 102 // Tests that we can handle really small buffers. |
103 TEST(PickleTest, SmallBuffer) { | 103 TEST(PickleTest, SmallBuffer) { |
104 scoped_array<char> buffer(new char[1]); | 104 scoped_ptr<char[]> buffer(new char[1]); |
105 | 105 |
106 // We should not touch the buffer. | 106 // We should not touch the buffer. |
107 Pickle pickle(buffer.get(), 1); | 107 Pickle pickle(buffer.get(), 1); |
108 | 108 |
109 PickleIterator iter(pickle); | 109 PickleIterator iter(pickle); |
110 int data; | 110 int data; |
111 EXPECT_FALSE(pickle.ReadInt(&iter, &data)); | 111 EXPECT_FALSE(pickle.ReadInt(&iter, &data)); |
112 } | 112 } |
113 | 113 |
114 // Tests that we can handle improper headers. | 114 // Tests that we can handle improper headers. |
(...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
178 const char* start = reinterpret_cast<const char*>(pickle.data()); | 178 const char* start = reinterpret_cast<const char*>(pickle.data()); |
179 const char* end = start + pickle.size(); | 179 const char* end = start + pickle.size(); |
180 | 180 |
181 EXPECT_TRUE(end == Pickle::FindNext(pickle.header_size_, start, end)); | 181 EXPECT_TRUE(end == Pickle::FindNext(pickle.header_size_, start, end)); |
182 EXPECT_TRUE(NULL == Pickle::FindNext(pickle.header_size_, start, end - 1)); | 182 EXPECT_TRUE(NULL == Pickle::FindNext(pickle.header_size_, start, end - 1)); |
183 EXPECT_TRUE(end == Pickle::FindNext(pickle.header_size_, start, end + 1)); | 183 EXPECT_TRUE(end == Pickle::FindNext(pickle.header_size_, start, end + 1)); |
184 } | 184 } |
185 | 185 |
186 TEST(PickleTest, FindNextWithIncompleteHeader) { | 186 TEST(PickleTest, FindNextWithIncompleteHeader) { |
187 size_t header_size = sizeof(Pickle::Header); | 187 size_t header_size = sizeof(Pickle::Header); |
188 scoped_array<char> buffer(new char[header_size - 1]); | 188 scoped_ptr<char[]> buffer(new char[header_size - 1]); |
189 memset(buffer.get(), 0x1, header_size - 1); | 189 memset(buffer.get(), 0x1, header_size - 1); |
190 | 190 |
191 const char* start = buffer.get(); | 191 const char* start = buffer.get(); |
192 const char* end = start + header_size - 1; | 192 const char* end = start + header_size - 1; |
193 | 193 |
194 EXPECT_TRUE(NULL == Pickle::FindNext(header_size, start, end)); | 194 EXPECT_TRUE(NULL == Pickle::FindNext(header_size, start, end)); |
195 } | 195 } |
196 | 196 |
197 TEST(PickleTest, GetReadPointerAndAdvance) { | 197 TEST(PickleTest, GetReadPointerAndAdvance) { |
198 Pickle pickle; | 198 Pickle pickle; |
199 | 199 |
200 PickleIterator iter(pickle); | 200 PickleIterator iter(pickle); |
201 EXPECT_FALSE(iter.GetReadPointerAndAdvance(1)); | 201 EXPECT_FALSE(iter.GetReadPointerAndAdvance(1)); |
202 | 202 |
203 EXPECT_TRUE(pickle.WriteInt(1)); | 203 EXPECT_TRUE(pickle.WriteInt(1)); |
204 EXPECT_TRUE(pickle.WriteInt(2)); | 204 EXPECT_TRUE(pickle.WriteInt(2)); |
205 int bytes = sizeof(int) * 2; | 205 int bytes = sizeof(int) * 2; |
206 | 206 |
207 EXPECT_TRUE(PickleIterator(pickle).GetReadPointerAndAdvance(0)); | 207 EXPECT_TRUE(PickleIterator(pickle).GetReadPointerAndAdvance(0)); |
208 EXPECT_TRUE(PickleIterator(pickle).GetReadPointerAndAdvance(1)); | 208 EXPECT_TRUE(PickleIterator(pickle).GetReadPointerAndAdvance(1)); |
209 EXPECT_FALSE(PickleIterator(pickle).GetReadPointerAndAdvance(-1)); | 209 EXPECT_FALSE(PickleIterator(pickle).GetReadPointerAndAdvance(-1)); |
210 EXPECT_TRUE(PickleIterator(pickle).GetReadPointerAndAdvance(bytes)); | 210 EXPECT_TRUE(PickleIterator(pickle).GetReadPointerAndAdvance(bytes)); |
211 EXPECT_FALSE(PickleIterator(pickle).GetReadPointerAndAdvance(bytes + 1)); | 211 EXPECT_FALSE(PickleIterator(pickle).GetReadPointerAndAdvance(bytes + 1)); |
212 EXPECT_FALSE(PickleIterator(pickle).GetReadPointerAndAdvance(INT_MAX)); | 212 EXPECT_FALSE(PickleIterator(pickle).GetReadPointerAndAdvance(INT_MAX)); |
213 EXPECT_FALSE(PickleIterator(pickle).GetReadPointerAndAdvance(INT_MIN)); | 213 EXPECT_FALSE(PickleIterator(pickle).GetReadPointerAndAdvance(INT_MIN)); |
214 } | 214 } |
215 | 215 |
216 TEST(PickleTest, Resize) { | 216 TEST(PickleTest, Resize) { |
217 size_t unit = Pickle::kPayloadUnit; | 217 size_t unit = Pickle::kPayloadUnit; |
218 scoped_array<char> data(new char[unit]); | 218 scoped_ptr<char[]> data(new char[unit]); |
219 char* data_ptr = data.get(); | 219 char* data_ptr = data.get(); |
220 for (size_t i = 0; i < unit; i++) | 220 for (size_t i = 0; i < unit; i++) |
221 data_ptr[i] = 'G'; | 221 data_ptr[i] = 'G'; |
222 | 222 |
223 // construct a message that will be exactly the size of one payload unit, | 223 // construct a message that will be exactly the size of one payload unit, |
224 // note that any data will have a 4-byte header indicating the size | 224 // note that any data will have a 4-byte header indicating the size |
225 const size_t payload_size_after_header = unit - sizeof(uint32); | 225 const size_t payload_size_after_header = unit - sizeof(uint32); |
226 Pickle pickle; | 226 Pickle pickle; |
227 pickle.WriteData(data_ptr, | 227 pickle.WriteData(data_ptr, |
228 static_cast<int>(payload_size_after_header - sizeof(uint32))); | 228 static_cast<int>(payload_size_after_header - sizeof(uint32))); |
(...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
333 EXPECT_TRUE(pickle.WriteBytes(&data, sizeof(data))); | 333 EXPECT_TRUE(pickle.WriteBytes(&data, sizeof(data))); |
334 | 334 |
335 PickleIterator iter(pickle); | 335 PickleIterator iter(pickle); |
336 const char* outdata_char = NULL; | 336 const char* outdata_char = NULL; |
337 EXPECT_TRUE(pickle.ReadBytes(&iter, &outdata_char, sizeof(data))); | 337 EXPECT_TRUE(pickle.ReadBytes(&iter, &outdata_char, sizeof(data))); |
338 | 338 |
339 int outdata; | 339 int outdata; |
340 memcpy(&outdata, outdata_char, sizeof(outdata)); | 340 memcpy(&outdata, outdata_char, sizeof(outdata)); |
341 EXPECT_EQ(data, outdata); | 341 EXPECT_EQ(data, outdata); |
342 } | 342 } |
OLD | NEW |