OLD | NEW |
1 // Copyright (c) 2016 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2016 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/quic_one_block_arena.h" | 5 #include "net/quic/quic_one_block_arena.h" |
6 | 6 |
7 #include "net/quic/interval_set.h" | 7 #include "net/quic/interval_set.h" |
8 #include "net/quic/quic_flags.h" | 8 #include "net/quic/quic_flags.h" |
9 #include "net/quic/quic_utils.h" | 9 #include "net/quic/quic_utils.h" |
10 #include "net/test/gtest_util.h" | 10 #include "net/test/gtest_util.h" |
11 #include "testing/gmock/include/gmock/gmock.h" | 11 #include "testing/gmock/include/gmock/gmock.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
13 | 13 |
14 namespace net { | 14 namespace net { |
15 namespace { | 15 namespace { |
16 | 16 |
17 static const uint32_t kMaxAlign = 8; | 17 static const uint32_t kMaxAlign = 8; |
18 | 18 |
19 struct TestObject { | 19 struct TestObject { |
20 uint32_t value; | 20 uint32_t value; |
21 }; | 21 }; |
22 | 22 |
23 TEST(QuicOneBlockArenaTest, AllocateSuccess) { | 23 TEST(QuicOneBlockArenaTest, AllocateSuccess) { |
24 FLAGS_quic_enable_arena_allocation = true; | |
25 QuicOneBlockArena<1024> arena; | 24 QuicOneBlockArena<1024> arena; |
26 QuicArenaScopedPtr<TestObject> ptr = arena.New<TestObject>(); | 25 QuicArenaScopedPtr<TestObject> ptr = arena.New<TestObject>(); |
27 EXPECT_TRUE(ptr.is_from_arena()); | 26 EXPECT_TRUE(ptr.is_from_arena()); |
28 } | 27 } |
29 | 28 |
30 TEST(QuicOneBlockArenaTest, Disabled) { | |
31 FLAGS_quic_enable_arena_allocation = false; | |
32 QuicOneBlockArena<1024> arena; | |
33 QuicArenaScopedPtr<TestObject> ptr = arena.New<TestObject>(); | |
34 EXPECT_FALSE(ptr.is_from_arena()); | |
35 } | |
36 | |
37 TEST(QuicOneBlockArenaTest, Exhaust) { | 29 TEST(QuicOneBlockArenaTest, Exhaust) { |
38 FLAGS_quic_enable_arena_allocation = true; | |
39 QuicOneBlockArena<1024> arena; | 30 QuicOneBlockArena<1024> arena; |
40 for (size_t i = 0; i < 1024 / kMaxAlign; ++i) { | 31 for (size_t i = 0; i < 1024 / kMaxAlign; ++i) { |
41 QuicArenaScopedPtr<TestObject> ptr = arena.New<TestObject>(); | 32 QuicArenaScopedPtr<TestObject> ptr = arena.New<TestObject>(); |
42 EXPECT_TRUE(ptr.is_from_arena()); | 33 EXPECT_TRUE(ptr.is_from_arena()); |
43 } | 34 } |
44 QuicArenaScopedPtr<TestObject> ptr; | 35 QuicArenaScopedPtr<TestObject> ptr; |
45 EXPECT_DFATAL(ptr = arena.New<TestObject>(), | 36 EXPECT_DFATAL(ptr = arena.New<TestObject>(), |
46 "Ran out of space in QuicOneBlockArena"); | 37 "Ran out of space in QuicOneBlockArena"); |
47 EXPECT_FALSE(ptr.is_from_arena()); | 38 EXPECT_FALSE(ptr.is_from_arena()); |
48 } | 39 } |
49 | 40 |
50 TEST(QuicOneBlockArenaTest, NoOverlaps) { | 41 TEST(QuicOneBlockArenaTest, NoOverlaps) { |
51 FLAGS_quic_enable_arena_allocation = true; | |
52 QuicOneBlockArena<1024> arena; | 42 QuicOneBlockArena<1024> arena; |
53 std::vector<QuicArenaScopedPtr<TestObject>> objects; | 43 std::vector<QuicArenaScopedPtr<TestObject>> objects; |
54 IntervalSet<uintptr_t> used; | 44 IntervalSet<uintptr_t> used; |
55 for (size_t i = 0; i < 1024 / kMaxAlign; ++i) { | 45 for (size_t i = 0; i < 1024 / kMaxAlign; ++i) { |
56 QuicArenaScopedPtr<TestObject> ptr = arena.New<TestObject>(); | 46 QuicArenaScopedPtr<TestObject> ptr = arena.New<TestObject>(); |
57 EXPECT_TRUE(ptr.is_from_arena()); | 47 EXPECT_TRUE(ptr.is_from_arena()); |
58 | 48 |
59 uintptr_t begin = reinterpret_cast<uintptr_t>(ptr.get()); | 49 uintptr_t begin = reinterpret_cast<uintptr_t>(ptr.get()); |
60 uintptr_t end = begin + sizeof(TestObject); | 50 uintptr_t end = begin + sizeof(TestObject); |
61 EXPECT_FALSE(used.Contains(begin)); | 51 EXPECT_FALSE(used.Contains(begin)); |
62 EXPECT_FALSE(used.Contains(end - 1)); | 52 EXPECT_FALSE(used.Contains(end - 1)); |
63 used.Add(begin, end); | 53 used.Add(begin, end); |
64 } | 54 } |
65 } | 55 } |
66 | 56 |
67 } // namespace | 57 } // namespace |
68 } // namespace net | 58 } // namespace net |
OLD | NEW |