OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2014 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/browser/renderer_host/clipboard_message_filter.h" | |
6 | |
7 #include <string.h> | |
8 | |
9 #include "base/memory/ref_counted.h" | |
10 #include "base/process/process_handle.h" | |
11 #include "base/run_loop.h" | |
12 #include "content/public/test/test_browser_thread_bundle.h" | |
13 #include "testing/gtest/include/gtest/gtest.h" | |
14 #include "third_party/skia/include/core/SkBitmap.h" | |
15 #include "ui/base/clipboard/clipboard.h" | |
16 #include "ui/gfx/size.h" | |
17 | |
18 #if defined(USE_X11) | |
19 #include "ui/events/platform/platform_event_source.h" | |
20 #endif | |
21 | |
22 namespace content { | |
23 | |
24 class ClipboardMessageFilterTest : public ::testing::Test { | |
25 protected: | |
26 ClipboardMessageFilterTest() | |
27 : filter_(new ClipboardMessageFilter), | |
28 #if defined(USE_X11) | |
29 event_source_(ui::PlatformEventSource::CreateDefault()), | |
30 #endif | |
31 clipboard_(ui::Clipboard::GetForCurrentThread()) { | |
32 filter_->set_peer_pid_for_testing(base::GetCurrentProcId()); | |
33 } | |
34 | |
35 virtual ~ClipboardMessageFilterTest() override { | |
36 ui::Clipboard::DestroyClipboardForCurrentThread(); | |
37 } | |
38 | |
39 scoped_ptr<base::SharedMemory> CreateAndMapReadOnlySharedMemory(size_t size) { | |
40 scoped_ptr<base::SharedMemory> m = CreateReadOnlySharedMemory(size); | |
41 if (!m->Map(size)) | |
42 return nullptr; | |
43 return m.Pass(); | |
44 } | |
45 | |
46 scoped_ptr<base::SharedMemory> CreateReadOnlySharedMemory(size_t size) { | |
47 auto m = make_scoped_ptr(new base::SharedMemory()); | |
Will Harris
2014/10/06 22:49:24
feels weird to see auto
dcheng
2014/10/06 22:51:44
Hmm. I guess. Basically, scoped_ptr<base::SharedMe
jamesr
2014/10/06 23:00:27
Comparing the two:
auto m = make_scoped_ptr(new b
dcheng
2014/10/06 23:34:41
Done.
| |
48 base::SharedMemoryCreateOptions options; | |
49 options.size = size; | |
50 options.share_read_only = true; | |
51 if (!m->Create(options)) | |
52 return nullptr; | |
53 return m.Pass(); | |
54 } | |
55 | |
56 void CallWriteImage(const gfx::Size& size, | |
57 base::SharedMemory* shared_memory) { | |
58 base::SharedMemoryHandle handle; | |
59 ASSERT_TRUE(shared_memory->GiveReadOnlyToProcess( | |
60 base::GetCurrentProcessHandle(), &handle)); | |
61 CallWriteImageDirectly(size, handle); | |
62 } | |
63 | |
64 // Prefer to use CallWriteImage() in tests. | |
65 void CallWriteImageDirectly(const gfx::Size& size, | |
66 base::SharedMemoryHandle handle) { | |
67 filter_->OnWriteImage(ui::CLIPBOARD_TYPE_COPY_PASTE, size, handle); | |
68 } | |
69 | |
70 void CallCommitWrite() { | |
71 filter_->OnCommitWrite(ui::CLIPBOARD_TYPE_COPY_PASTE); | |
72 base::RunLoop().RunUntilIdle(); | |
73 } | |
74 | |
75 ui::Clipboard* clipboard() { return clipboard_; } | |
76 | |
77 private: | |
78 const TestBrowserThreadBundle thread_bundle_; | |
79 const scoped_refptr<ClipboardMessageFilter> filter_; | |
80 #if defined(USE_X11) | |
81 const scoped_ptr<ui::PlatformEventSource> event_source_; | |
82 #endif | |
83 ui::Clipboard* const clipboard_; | |
84 }; | |
85 | |
86 // Test that it actually works. | |
87 TEST_F(ClipboardMessageFilterTest, SimpleImage) { | |
88 static const uint32_t bitmap_data[] = { | |
89 0x33333333, 0xdddddddd, 0xeeeeeeee, 0x00000000, | |
90 0x88888888, 0x66666666, 0x55555555, 0xbbbbbbbb, | |
91 0x44444444, 0xaaaaaaaa, 0x99999999, 0x77777777, | |
92 0xffffffff, 0x11111111, 0x22222222, 0xcccccccc, | |
93 }; | |
94 | |
95 scoped_ptr<base::SharedMemory> shared_memory = | |
96 CreateAndMapReadOnlySharedMemory(sizeof(bitmap_data)); | |
97 memcpy(shared_memory->memory(), bitmap_data, sizeof(bitmap_data)); | |
98 | |
99 CallWriteImage(gfx::Size(4, 4), shared_memory.get()); | |
100 uint64_t sequence_number = | |
101 clipboard()->GetSequenceNumber(ui::CLIPBOARD_TYPE_COPY_PASTE); | |
102 CallCommitWrite(); | |
103 | |
104 EXPECT_NE(sequence_number, | |
105 clipboard()->GetSequenceNumber(ui::CLIPBOARD_TYPE_COPY_PASTE)); | |
106 EXPECT_FALSE(clipboard()->IsFormatAvailable( | |
107 ui::Clipboard::GetPlainTextFormatType(), ui::CLIPBOARD_TYPE_COPY_PASTE)); | |
108 EXPECT_TRUE(clipboard()->IsFormatAvailable( | |
109 ui::Clipboard::GetBitmapFormatType(), ui::CLIPBOARD_TYPE_COPY_PASTE)); | |
110 | |
111 SkBitmap actual = clipboard()->ReadImage(ui::CLIPBOARD_TYPE_COPY_PASTE); | |
112 SkAutoLockPixels locked(actual); | |
113 EXPECT_EQ(sizeof(bitmap_data), actual.getSize()); | |
114 EXPECT_EQ(0, | |
115 memcmp(bitmap_data, actual.getAddr32(0, 0), sizeof(bitmap_data))); | |
116 } | |
117 | |
118 // Test with a size that would overflow a naive 32-bit row bytes calculation. | |
119 TEST_F(ClipboardMessageFilterTest, ImageSizeOverflows32BitRowBytes) { | |
120 scoped_ptr<base::SharedMemory> shared_memory = | |
121 CreateReadOnlySharedMemory(0x20000000); | |
122 | |
123 CallWriteImage(gfx::Size(0x20000000, 1), shared_memory.get()); | |
124 uint64_t sequence_number = | |
125 clipboard()->GetSequenceNumber(ui::CLIPBOARD_TYPE_COPY_PASTE); | |
126 CallCommitWrite(); | |
127 | |
128 EXPECT_EQ(sequence_number, | |
129 clipboard()->GetSequenceNumber(ui::CLIPBOARD_TYPE_COPY_PASTE)); | |
130 } | |
131 | |
132 TEST_F(ClipboardMessageFilterTest, InvalidSharedMemoryHandle) { | |
133 CallWriteImageDirectly(gfx::Size(5, 5), base::SharedMemory::NULLHandle()); | |
134 uint64_t sequence_number = | |
135 clipboard()->GetSequenceNumber(ui::CLIPBOARD_TYPE_COPY_PASTE); | |
136 CallCommitWrite(); | |
137 | |
138 EXPECT_EQ(sequence_number, | |
139 clipboard()->GetSequenceNumber(ui::CLIPBOARD_TYPE_COPY_PASTE)); | |
140 } | |
141 | |
142 } // namespace content | |
OLD | NEW |