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

Side by Side Diff: ipc/attachment_broker_unprivileged_win_unittest.cc

Issue 1256993003: ipc: Create AttachmentBrokerPrivileged and AttachmentBrokerUnprivileged. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@add_ipc_message_feature
Patch Set: Rebase. Created 5 years, 4 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
OLDNEW
1 // Copyright 2015 The Chromium Authors. All rights reserved. 1 // Copyright 2015 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 "build/build_config.h" 5 #include "build/build_config.h"
6 6
7 #include <windows.h> 7 #include <windows.h>
8 8
9 #include "base/memory/ref_counted.h" 9 #include "base/memory/ref_counted.h"
10 #include "base/memory/scoped_ptr.h" 10 #include "base/memory/scoped_ptr.h"
11 #include "base/process/process.h" 11 #include "base/process/process.h"
12 #include "ipc/attachment_broker_messages.h" 12 #include "ipc/attachment_broker_messages.h"
13 #include "ipc/attachment_broker_win.h" 13 #include "ipc/attachment_broker_unprivileged_win.h"
14 #include "ipc/handle_attachment_win.h" 14 #include "ipc/handle_attachment_win.h"
15 #include "testing/gtest/include/gtest/gtest.h" 15 #include "testing/gtest/include/gtest/gtest.h"
16 16
17 namespace IPC { 17 namespace IPC {
18 18
19 TEST(AttachmentBrokerWinTest, ReceiveValidMessage) { 19 TEST(AttachmentBrokerUnprivilegedWinTest, ReceiveValidMessage) {
20 HANDLE handle = LongToHandle(8); 20 HANDLE handle = LongToHandle(8);
21 base::ProcessId destination = base::Process::Current().Pid(); 21 base::ProcessId destination = base::Process::Current().Pid();
22 scoped_refptr<internal::HandleAttachmentWin> attachment( 22 scoped_refptr<internal::HandleAttachmentWin> attachment(
23 new internal::HandleAttachmentWin(handle)); 23 new internal::HandleAttachmentWin(handle));
24 AttachmentBrokerMsg_WinHandleHasBeenDuplicated msg( 24 AttachmentBrokerMsg_WinHandleHasBeenDuplicated msg(
25 attachment->GetWireFormat(destination)); 25 attachment->GetWireFormat(destination));
26 AttachmentBrokerWin attachment_broker; 26 AttachmentBrokerUnprivilegedWin attachment_broker;
27 EXPECT_TRUE(attachment_broker.OnMessageReceived(msg)); 27 EXPECT_TRUE(attachment_broker.OnMessageReceived(msg));
28 EXPECT_EQ(1u, attachment_broker.attachments_.size()); 28 EXPECT_EQ(1u, attachment_broker.attachments_.size());
29 29
30 scoped_refptr<BrokerableAttachment> received_attachment = 30 scoped_refptr<BrokerableAttachment> received_attachment =
31 attachment_broker.attachments_.front(); 31 attachment_broker.attachments_.front();
32 EXPECT_EQ(BrokerableAttachment::WIN_HANDLE, 32 EXPECT_EQ(BrokerableAttachment::WIN_HANDLE,
33 received_attachment->GetBrokerableType()); 33 received_attachment->GetBrokerableType());
34 internal::HandleAttachmentWin* received_handle_attachment = 34 internal::HandleAttachmentWin* received_handle_attachment =
35 static_cast<internal::HandleAttachmentWin*>(received_attachment.get()); 35 static_cast<internal::HandleAttachmentWin*>(received_attachment.get());
36 EXPECT_EQ(handle, received_handle_attachment->get_handle()); 36 EXPECT_EQ(handle, received_handle_attachment->get_handle());
37 } 37 }
38 38
39 TEST(AttachmentBrokerWinTest, ReceiveInvalidMessage) { 39 TEST(AttachmentBrokerUnprivilegedWinTest, ReceiveInvalidMessage) {
40 HANDLE handle = LongToHandle(8); 40 HANDLE handle = LongToHandle(8);
41 base::ProcessId destination = base::Process::Current().Pid() + 1; 41 base::ProcessId destination = base::Process::Current().Pid() + 1;
42 scoped_refptr<internal::HandleAttachmentWin> attachment( 42 scoped_refptr<internal::HandleAttachmentWin> attachment(
43 new internal::HandleAttachmentWin(handle)); 43 new internal::HandleAttachmentWin(handle));
44 AttachmentBrokerMsg_WinHandleHasBeenDuplicated msg( 44 AttachmentBrokerMsg_WinHandleHasBeenDuplicated msg(
45 attachment->GetWireFormat(destination)); 45 attachment->GetWireFormat(destination));
46 AttachmentBrokerWin attachment_broker; 46 AttachmentBrokerUnprivilegedWin attachment_broker;
47 EXPECT_TRUE(attachment_broker.OnMessageReceived(msg)); 47 EXPECT_TRUE(attachment_broker.OnMessageReceived(msg));
48 EXPECT_EQ(0u, attachment_broker.attachments_.size()); 48 EXPECT_EQ(0u, attachment_broker.attachments_.size());
49 } 49 }
50 50
51 } // namespace IPC 51 } // namespace IPC
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698