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

Side by Side Diff: base/mac/dispatch_source_mach_unittest.cc

Issue 1136953012: Move DispatchSourceMach from //sandbox/mac to //base/mac. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 7 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
« no previous file with comments | « base/mac/dispatch_source_mach.cc ('k') | sandbox/mac/BUILD.gn » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright 2014 The Chromium Authors. All rights reserved. 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 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 "sandbox/mac/dispatch_source_mach.h" 5 #include "base/mac/dispatch_source_mach.h"
6 6
7 #include <mach/mach.h> 7 #include <mach/mach.h>
8 8
9 #include "base/logging.h" 9 #include "base/logging.h"
10 #include "base/mac/scoped_mach_port.h" 10 #include "base/mac/scoped_mach_port.h"
11 #include "base/memory/scoped_ptr.h" 11 #include "base/memory/scoped_ptr.h"
12 #include "base/test/test_timeouts.h" 12 #include "base/test/test_timeouts.h"
13 #include "testing/gtest/include/gtest/gtest.h" 13 #include "testing/gtest/include/gtest/gtest.h"
14 14
15 namespace sandbox { 15 namespace base {
16 16
17 class DispatchSourceMachTest : public testing::Test { 17 class DispatchSourceMachTest : public testing::Test {
18 public: 18 public:
19 void SetUp() override { 19 void SetUp() override {
20 mach_port_t port = MACH_PORT_NULL; 20 mach_port_t port = MACH_PORT_NULL;
21 ASSERT_EQ(KERN_SUCCESS, mach_port_allocate(mach_task_self(), 21 ASSERT_EQ(KERN_SUCCESS, mach_port_allocate(mach_task_self(),
22 MACH_PORT_RIGHT_RECEIVE, &port)); 22 MACH_PORT_RIGHT_RECEIVE, &port));
23 receive_right_.reset(port); 23 receive_right_.reset(port);
24 24
25 ASSERT_EQ(KERN_SUCCESS, mach_port_insert_right(mach_task_self(), port, 25 ASSERT_EQ(KERN_SUCCESS, mach_port_insert_right(mach_task_self(), port,
(...skipping 11 matching lines...) Expand all
37 37
38 private: 38 private:
39 base::mac::ScopedMachReceiveRight receive_right_; 39 base::mac::ScopedMachReceiveRight receive_right_;
40 base::mac::ScopedMachSendRight send_right_; 40 base::mac::ScopedMachSendRight send_right_;
41 }; 41 };
42 42
43 TEST_F(DispatchSourceMachTest, ReceiveAfterResume) { 43 TEST_F(DispatchSourceMachTest, ReceiveAfterResume) {
44 dispatch_semaphore_t signal = dispatch_semaphore_create(0); 44 dispatch_semaphore_t signal = dispatch_semaphore_create(0);
45 45
46 bool __block did_receive = false; 46 bool __block did_receive = false;
47 DispatchSourceMach source("org.chromium.sandbox.test.ReceiveAfterResume", 47 DispatchSourceMach source("org.chromium.base.test.ReceiveAfterResume",
48 port(), ^{ 48 port(), ^{
49 mach_msg_empty_rcv_t msg = {{0}}; 49 mach_msg_empty_rcv_t msg = {{0}};
50 msg.header.msgh_size = sizeof(msg); 50 msg.header.msgh_size = sizeof(msg);
51 msg.header.msgh_local_port = port(); 51 msg.header.msgh_local_port = port();
52 mach_msg_receive(&msg.header); 52 mach_msg_receive(&msg.header);
53 did_receive = true; 53 did_receive = true;
54 54
55 dispatch_semaphore_signal(signal); 55 dispatch_semaphore_signal(signal);
56 }); 56 });
57 57
(...skipping 10 matching lines...) Expand all
68 WaitForSemaphore(signal); 68 WaitForSemaphore(signal);
69 69
70 EXPECT_TRUE(did_receive); 70 EXPECT_TRUE(did_receive);
71 } 71 }
72 72
73 TEST_F(DispatchSourceMachTest, NoMessagesAfterDestruction) { 73 TEST_F(DispatchSourceMachTest, NoMessagesAfterDestruction) {
74 scoped_ptr<int> count(new int(0)); 74 scoped_ptr<int> count(new int(0));
75 int* __block count_ptr = count.get(); 75 int* __block count_ptr = count.get();
76 76
77 scoped_ptr<DispatchSourceMach> source(new DispatchSourceMach( 77 scoped_ptr<DispatchSourceMach> source(new DispatchSourceMach(
78 "org.chromium.sandbox.test.NoMessagesAfterDestruction", 78 "org.chromium.base.test.NoMessagesAfterDestruction",
79 port(), ^{ 79 port(), ^{
80 mach_msg_empty_rcv_t msg = {{0}}; 80 mach_msg_empty_rcv_t msg = {{0}};
81 msg.header.msgh_size = sizeof(msg); 81 msg.header.msgh_size = sizeof(msg);
82 msg.header.msgh_local_port = port(); 82 msg.header.msgh_local_port = port();
83 mach_msg_receive(&msg.header); 83 mach_msg_receive(&msg.header);
84 LOG(INFO) << "Receieve " << *count_ptr; 84 LOG(INFO) << "Receieve " << *count_ptr;
85 ++(*count_ptr); 85 ++(*count_ptr);
86 })); 86 }));
87 source->Resume(); 87 source->Resume();
88 88
89 dispatch_queue_t queue = 89 dispatch_queue_t queue =
90 dispatch_queue_create("org.chromium.sandbox.test.MessageSend", NULL); 90 dispatch_queue_create("org.chromium.base.test.MessageSend", NULL);
91 dispatch_semaphore_t signal = dispatch_semaphore_create(0); 91 dispatch_semaphore_t signal = dispatch_semaphore_create(0);
92 for (int i = 0; i < 30; ++i) { 92 for (int i = 0; i < 30; ++i) {
93 dispatch_async(queue, ^{ 93 dispatch_async(queue, ^{
94 mach_msg_empty_send_t msg = {{0}}; 94 mach_msg_empty_send_t msg = {{0}};
95 msg.header.msgh_size = sizeof(msg); 95 msg.header.msgh_size = sizeof(msg);
96 msg.header.msgh_remote_port = port(); 96 msg.header.msgh_remote_port = port();
97 msg.header.msgh_bits = 97 msg.header.msgh_bits =
98 MACH_MSGH_BITS_REMOTE(MACH_MSG_TYPE_COPY_SEND); 98 MACH_MSGH_BITS_REMOTE(MACH_MSG_TYPE_COPY_SEND);
99 mach_msg_send(&msg.header); 99 mach_msg_send(&msg.header);
100 }); 100 });
101 101
102 // After sending five messages, shut down the source and taint the 102 // After sending five messages, shut down the source and taint the
103 // pointer the handler dereferences. The test will crash if |count_ptr| 103 // pointer the handler dereferences. The test will crash if |count_ptr|
104 // is being used after "free". 104 // is being used after "free".
105 if (i == 5) { 105 if (i == 5) {
106 scoped_ptr<DispatchSourceMach>* source_ptr = &source; 106 scoped_ptr<DispatchSourceMach>* source_ptr = &source;
107 dispatch_async(queue, ^{ 107 dispatch_async(queue, ^{
108 source_ptr->reset(); 108 source_ptr->reset();
109 count_ptr = reinterpret_cast<int*>(0xdeaddead); 109 count_ptr = reinterpret_cast<int*>(0xdeaddead);
110 dispatch_semaphore_signal(signal); 110 dispatch_semaphore_signal(signal);
111 }); 111 });
112 } 112 }
113 } 113 }
114 114
115 WaitForSemaphore(signal); 115 WaitForSemaphore(signal);
116 dispatch_release(queue); 116 dispatch_release(queue);
117 } 117 }
118 118
119 } // namespace sandbox 119 } // namespace base
OLDNEW
« no previous file with comments | « base/mac/dispatch_source_mach.cc ('k') | sandbox/mac/BUILD.gn » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698