1// Copyright (c) 2013 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "content/common/host_shared_bitmap_manager.h"
6#include "testing/gtest/include/gtest/gtest.h"
7
8namespace content {
9namespace {
10
11class HostSharedBitmapManagerTest : public testing::Test {
12 protected:
13  virtual void SetUp() { manager_.reset(new HostSharedBitmapManager()); }
14  scoped_ptr<HostSharedBitmapManager> manager_;
15};
16
17TEST_F(HostSharedBitmapManagerTest, TestCreate) {
18  gfx::Size bitmap_size(1, 1);
19  size_t size_in_bytes;
20  EXPECT_TRUE(cc::SharedBitmap::SizeInBytes(bitmap_size, &size_in_bytes));
21  scoped_ptr<base::SharedMemory> bitmap(new base::SharedMemory());
22  bitmap->CreateAndMapAnonymous(size_in_bytes);
23  memset(bitmap->memory(), 0xff, size_in_bytes);
24  cc::SharedBitmapId id = cc::SharedBitmap::GenerateId();
25
26  base::SharedMemoryHandle handle;
27  bitmap->ShareToProcess(base::GetCurrentProcessHandle(), &handle);
28  manager_->ChildAllocatedSharedBitmap(
29      size_in_bytes, handle, base::GetCurrentProcessHandle(), id);
30
31  scoped_ptr<cc::SharedBitmap> large_bitmap;
32  large_bitmap = manager_->GetSharedBitmapFromId(gfx::Size(1024, 1024), id);
33  EXPECT_TRUE(large_bitmap.get() == NULL);
34
35  scoped_ptr<cc::SharedBitmap> very_large_bitmap;
36  very_large_bitmap =
37      manager_->GetSharedBitmapFromId(gfx::Size(1, (1 << 30) | 1), id);
38  EXPECT_TRUE(very_large_bitmap.get() == NULL);
39
40  scoped_ptr<cc::SharedBitmap> negative_size_bitmap;
41  negative_size_bitmap =
42      manager_->GetSharedBitmapFromId(gfx::Size(-1, 1024), id);
43  EXPECT_TRUE(negative_size_bitmap.get() == NULL);
44
45  cc::SharedBitmapId id2 = cc::SharedBitmap::GenerateId();
46  scoped_ptr<cc::SharedBitmap> invalid_bitmap;
47  invalid_bitmap = manager_->GetSharedBitmapFromId(bitmap_size, id2);
48  EXPECT_TRUE(invalid_bitmap.get() == NULL);
49
50  scoped_ptr<cc::SharedBitmap> shared_bitmap;
51  shared_bitmap = manager_->GetSharedBitmapFromId(bitmap_size, id);
52  ASSERT_TRUE(shared_bitmap.get() != NULL);
53  EXPECT_EQ(memcmp(shared_bitmap->pixels(), bitmap->memory(), 4), 0);
54
55  scoped_ptr<cc::SharedBitmap> large_bitmap2;
56  large_bitmap2 = manager_->GetSharedBitmapFromId(gfx::Size(1024, 1024), id);
57  EXPECT_TRUE(large_bitmap2.get() == NULL);
58
59  scoped_ptr<cc::SharedBitmap> shared_bitmap2;
60  shared_bitmap2 = manager_->GetSharedBitmapFromId(bitmap_size, id);
61  EXPECT_TRUE(shared_bitmap2->pixels() == shared_bitmap->pixels());
62  shared_bitmap2.reset();
63  EXPECT_EQ(memcmp(shared_bitmap->pixels(), bitmap->memory(), size_in_bytes),
64            0);
65
66  manager_->ChildDeletedSharedBitmap(id);
67
68  memset(bitmap->memory(), 0, size_in_bytes);
69
70  EXPECT_EQ(memcmp(shared_bitmap->pixels(), bitmap->memory(), size_in_bytes),
71            0);
72  bitmap.reset();
73  shared_bitmap.reset();
74}
75
76TEST_F(HostSharedBitmapManagerTest, TestCreateForChild) {
77  gfx::Size bitmap_size(1, 1);
78  size_t size_in_bytes;
79  EXPECT_TRUE(cc::SharedBitmap::SizeInBytes(bitmap_size, &size_in_bytes));
80  cc::SharedBitmapId id = cc::SharedBitmap::GenerateId();
81  base::SharedMemoryHandle handle;
82  manager_->AllocateSharedBitmapForChild(
83      base::GetCurrentProcessHandle(), size_in_bytes, id, &handle);
84
85  EXPECT_TRUE(base::SharedMemory::IsHandleValid(handle));
86  scoped_ptr<base::SharedMemory> bitmap(new base::SharedMemory(handle, false));
87  EXPECT_TRUE(bitmap->Map(size_in_bytes));
88  memset(bitmap->memory(), 0xff, size_in_bytes);
89
90  scoped_ptr<cc::SharedBitmap> shared_bitmap;
91  shared_bitmap = manager_->GetSharedBitmapFromId(bitmap_size, id);
92  EXPECT_TRUE(shared_bitmap);
93  EXPECT_TRUE(
94      memcmp(bitmap->memory(), shared_bitmap->pixels(), size_in_bytes) == 0);
95}
96
97TEST_F(HostSharedBitmapManagerTest, RemoveProcess) {
98  gfx::Size bitmap_size(1, 1);
99  size_t size_in_bytes;
100  EXPECT_TRUE(cc::SharedBitmap::SizeInBytes(bitmap_size, &size_in_bytes));
101  scoped_ptr<base::SharedMemory> bitmap(new base::SharedMemory());
102  bitmap->CreateAndMapAnonymous(size_in_bytes);
103  memset(bitmap->memory(), 0xff, size_in_bytes);
104  cc::SharedBitmapId id = cc::SharedBitmap::GenerateId();
105
106  base::SharedMemoryHandle handle;
107  bitmap->ShareToProcess(base::GetCurrentProcessHandle(), &handle);
108  manager_->ChildAllocatedSharedBitmap(
109      size_in_bytes, handle, base::GetCurrentProcessHandle(), id);
110
111  manager_->ProcessRemoved(base::kNullProcessHandle);
112
113  scoped_ptr<cc::SharedBitmap> shared_bitmap;
114  shared_bitmap = manager_->GetSharedBitmapFromId(bitmap_size, id);
115  ASSERT_TRUE(shared_bitmap.get() != NULL);
116
117  manager_->ProcessRemoved(base::GetCurrentProcessHandle());
118
119  scoped_ptr<cc::SharedBitmap> shared_bitmap2;
120  shared_bitmap2 = manager_->GetSharedBitmapFromId(bitmap_size, id);
121  EXPECT_TRUE(shared_bitmap2.get() == NULL);
122  EXPECT_EQ(memcmp(shared_bitmap->pixels(), bitmap->memory(), size_in_bytes),
123            0);
124
125  shared_bitmap.reset();
126
127  // Should no-op.
128  manager_->ChildDeletedSharedBitmap(id);
129}
130
131TEST_F(HostSharedBitmapManagerTest, AddDuplicate) {
132  gfx::Size bitmap_size(1, 1);
133  size_t size_in_bytes;
134  EXPECT_TRUE(cc::SharedBitmap::SizeInBytes(bitmap_size, &size_in_bytes));
135  scoped_ptr<base::SharedMemory> bitmap(new base::SharedMemory());
136  bitmap->CreateAndMapAnonymous(size_in_bytes);
137  memset(bitmap->memory(), 0xff, size_in_bytes);
138  cc::SharedBitmapId id = cc::SharedBitmap::GenerateId();
139
140  base::SharedMemoryHandle handle;
141  bitmap->ShareToProcess(base::GetCurrentProcessHandle(), &handle);
142  manager_->ChildAllocatedSharedBitmap(
143      size_in_bytes, handle, base::GetCurrentProcessHandle(), id);
144
145  scoped_ptr<base::SharedMemory> bitmap2(new base::SharedMemory());
146  bitmap2->CreateAndMapAnonymous(size_in_bytes);
147  memset(bitmap2->memory(), 0x00, size_in_bytes);
148
149  manager_->ChildAllocatedSharedBitmap(
150      size_in_bytes, bitmap2->handle(), base::GetCurrentProcessHandle(), id);
151
152  scoped_ptr<cc::SharedBitmap> shared_bitmap;
153  shared_bitmap = manager_->GetSharedBitmapFromId(bitmap_size, id);
154  ASSERT_TRUE(shared_bitmap.get() != NULL);
155  EXPECT_EQ(memcmp(shared_bitmap->pixels(), bitmap->memory(), size_in_bytes),
156            0);
157}
158
159}  // namespace
160}  // namespace content
161