| OLD | NEW |
| 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 Loading... |
| 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 Loading... |
| 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 |
| OLD | NEW |