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 "mojo/edk/system/raw_channel.h" | 5 #include "mojo/edk/system/raw_channel.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 #include <stdio.h> | 8 #include <stdio.h> |
9 | 9 |
| 10 #include <memory> |
| 11 #include <utility> |
10 #include <vector> | 12 #include <vector> |
11 | 13 |
12 #include "base/bind.h" | 14 #include "base/bind.h" |
13 #include "base/location.h" | 15 #include "base/location.h" |
14 #include "base/logging.h" | 16 #include "base/logging.h" |
15 #include "base/memory/scoped_ptr.h" | |
16 #include "base/memory/scoped_vector.h" | 17 #include "base/memory/scoped_vector.h" |
17 #include "base/synchronization/waitable_event.h" | 18 #include "base/synchronization/waitable_event.h" |
18 #include "base/threading/simple_thread.h" | 19 #include "base/threading/simple_thread.h" |
19 #include "mojo/edk/embedder/platform_channel_pair.h" | 20 #include "mojo/edk/embedder/platform_channel_pair.h" |
20 #include "mojo/edk/embedder/platform_handle.h" | 21 #include "mojo/edk/embedder/platform_handle.h" |
21 #include "mojo/edk/embedder/scoped_platform_handle.h" | 22 #include "mojo/edk/embedder/scoped_platform_handle.h" |
22 #include "mojo/edk/system/message_in_transit.h" | 23 #include "mojo/edk/system/message_in_transit.h" |
23 #include "mojo/edk/system/mutex.h" | 24 #include "mojo/edk/system/mutex.h" |
24 #include "mojo/edk/system/test_utils.h" | 25 #include "mojo/edk/system/test_utils.h" |
25 #include "mojo/edk/system/transport_data.h" | 26 #include "mojo/edk/system/transport_data.h" |
26 #include "mojo/edk/test/scoped_test_dir.h" | 27 #include "mojo/edk/test/scoped_test_dir.h" |
27 #include "mojo/edk/test/test_io_thread.h" | 28 #include "mojo/edk/test/test_io_thread.h" |
28 #include "mojo/edk/test/test_utils.h" | 29 #include "mojo/edk/test/test_utils.h" |
29 #include "mojo/edk/util/make_unique.h" | 30 #include "mojo/edk/util/make_unique.h" |
30 #include "mojo/edk/util/scoped_file.h" | 31 #include "mojo/edk/util/scoped_file.h" |
31 #include "mojo/public/cpp/system/macros.h" | 32 #include "mojo/public/cpp/system/macros.h" |
32 #include "testing/gtest/include/gtest/gtest.h" | 33 #include "testing/gtest/include/gtest/gtest.h" |
33 | 34 |
34 namespace mojo { | 35 namespace mojo { |
35 namespace system { | 36 namespace system { |
36 namespace { | 37 namespace { |
37 | 38 |
38 scoped_ptr<MessageInTransit> MakeTestMessage(uint32_t num_bytes) { | 39 std::unique_ptr<MessageInTransit> MakeTestMessage(uint32_t num_bytes) { |
39 std::vector<unsigned char> bytes(num_bytes, 0); | 40 std::vector<unsigned char> bytes(num_bytes, 0); |
40 for (size_t i = 0; i < num_bytes; i++) | 41 for (size_t i = 0; i < num_bytes; i++) |
41 bytes[i] = static_cast<unsigned char>(i + num_bytes); | 42 bytes[i] = static_cast<unsigned char>(i + num_bytes); |
42 return make_scoped_ptr( | 43 return util::MakeUnique<MessageInTransit>( |
43 new MessageInTransit(MessageInTransit::Type::ENDPOINT_CLIENT, | 44 MessageInTransit::Type::ENDPOINT_CLIENT, |
44 MessageInTransit::Subtype::ENDPOINT_CLIENT_DATA, | 45 MessageInTransit::Subtype::ENDPOINT_CLIENT_DATA, num_bytes, |
45 num_bytes, bytes.empty() ? nullptr : &bytes[0])); | 46 bytes.empty() ? nullptr : &bytes[0]); |
46 } | 47 } |
47 | 48 |
48 bool CheckMessageData(const void* bytes, uint32_t num_bytes) { | 49 bool CheckMessageData(const void* bytes, uint32_t num_bytes) { |
49 const unsigned char* b = static_cast<const unsigned char*>(bytes); | 50 const unsigned char* b = static_cast<const unsigned char*>(bytes); |
50 for (uint32_t i = 0; i < num_bytes; i++) { | 51 for (uint32_t i = 0; i < num_bytes; i++) { |
51 if (b[i] != static_cast<unsigned char>(i + num_bytes)) | 52 if (b[i] != static_cast<unsigned char>(i + num_bytes)) |
52 return false; | 53 return false; |
53 } | 54 } |
54 return true; | 55 return true; |
55 } | 56 } |
56 | 57 |
57 void InitOnIOThread(RawChannel* raw_channel, RawChannel::Delegate* delegate) { | 58 void InitOnIOThread(RawChannel* raw_channel, RawChannel::Delegate* delegate) { |
58 raw_channel->Init(delegate); | 59 raw_channel->Init(delegate); |
59 } | 60 } |
60 | 61 |
61 bool WriteTestMessageToHandle(const embedder::PlatformHandle& handle, | 62 bool WriteTestMessageToHandle(const embedder::PlatformHandle& handle, |
62 uint32_t num_bytes) { | 63 uint32_t num_bytes) { |
63 scoped_ptr<MessageInTransit> message(MakeTestMessage(num_bytes)); | 64 std::unique_ptr<MessageInTransit> message(MakeTestMessage(num_bytes)); |
64 | 65 |
65 size_t write_size = 0; | 66 size_t write_size = 0; |
66 mojo::test::BlockingWrite(handle, message->main_buffer(), | 67 mojo::test::BlockingWrite(handle, message->main_buffer(), |
67 message->main_buffer_size(), &write_size); | 68 message->main_buffer_size(), &write_size); |
68 return write_size == message->main_buffer_size(); | 69 return write_size == message->main_buffer_size(); |
69 } | 70 } |
70 | 71 |
71 // ----------------------------------------------------------------------------- | 72 // ----------------------------------------------------------------------------- |
72 | 73 |
73 class RawChannelTest : public testing::Test { | 74 class RawChannelTest : public testing::Test { |
(...skipping 740 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
814 | 815 |
815 { | 816 { |
816 const char kHello[] = "hello"; | 817 const char kHello[] = "hello"; |
817 embedder::ScopedPlatformHandleVectorPtr platform_handles( | 818 embedder::ScopedPlatformHandleVectorPtr platform_handles( |
818 new embedder::PlatformHandleVector()); | 819 new embedder::PlatformHandleVector()); |
819 platform_handles->push_back( | 820 platform_handles->push_back( |
820 mojo::test::PlatformHandleFromFILE(fp1.Pass()).release()); | 821 mojo::test::PlatformHandleFromFILE(fp1.Pass()).release()); |
821 platform_handles->push_back( | 822 platform_handles->push_back( |
822 mojo::test::PlatformHandleFromFILE(fp2.Pass()).release()); | 823 mojo::test::PlatformHandleFromFILE(fp2.Pass()).release()); |
823 | 824 |
824 scoped_ptr<MessageInTransit> message( | 825 std::unique_ptr<MessageInTransit> message( |
825 new MessageInTransit(MessageInTransit::Type::ENDPOINT_CLIENT, | 826 new MessageInTransit(MessageInTransit::Type::ENDPOINT_CLIENT, |
826 MessageInTransit::Subtype::ENDPOINT_CLIENT_DATA, | 827 MessageInTransit::Subtype::ENDPOINT_CLIENT_DATA, |
827 sizeof(kHello), kHello)); | 828 sizeof(kHello), kHello)); |
828 message->SetTransportData(util::MakeUnique<TransportData>( | 829 message->SetTransportData(util::MakeUnique<TransportData>( |
829 platform_handles.Pass(), rc_write->GetSerializedPlatformHandleSize())); | 830 platform_handles.Pass(), rc_write->GetSerializedPlatformHandleSize())); |
830 EXPECT_TRUE(rc_write->WriteMessage(message.Pass())); | 831 EXPECT_TRUE(rc_write->WriteMessage(std::move(message))); |
831 } | 832 } |
832 | 833 |
833 read_delegate.Wait(); | 834 read_delegate.Wait(); |
834 | 835 |
835 io_thread()->PostTaskAndWait( | 836 io_thread()->PostTaskAndWait( |
836 FROM_HERE, | 837 FROM_HERE, |
837 base::Bind(&RawChannel::Shutdown, base::Unretained(rc_read.get()))); | 838 base::Bind(&RawChannel::Shutdown, base::Unretained(rc_read.get()))); |
838 io_thread()->PostTaskAndWait( | 839 io_thread()->PostTaskAndWait( |
839 FROM_HERE, | 840 FROM_HERE, |
840 base::Bind(&RawChannel::Shutdown, base::Unretained(rc_write.get()))); | 841 base::Bind(&RawChannel::Shutdown, base::Unretained(rc_write.get()))); |
841 } | 842 } |
842 | 843 |
843 } // namespace | 844 } // namespace |
844 } // namespace system | 845 } // namespace system |
845 } // namespace mojo | 846 } // namespace mojo |
OLD | NEW |