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 "mojo/system/platform_handle_dispatcher.h" 6 7 #include <stdio.h> 8 9 #include "base/files/file_path.h" 10 #include "base/files/file_util.h" 11 #include "base/files/scoped_file.h" 12 #include "base/files/scoped_temp_dir.h" 13 #include "base/memory/ref_counted.h" 14 #include "mojo/common/test/test_utils.h" 15 #include "testing/gtest/include/gtest/gtest.h" 16 17 namespace mojo { 18 namespace system { 19 namespace { 20 21 TEST(PlatformHandleDispatcherTest, Basic) { 22 base::ScopedTempDir temp_dir; 23 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 24 25 static const char kHelloWorld[] = "hello world"; 26 27 base::FilePath unused; 28 base::ScopedFILE fp( 29 CreateAndOpenTemporaryFileInDir(temp_dir.path(), &unused)); 30 ASSERT_TRUE(fp); 31 EXPECT_EQ(sizeof(kHelloWorld), 32 fwrite(kHelloWorld, 1, sizeof(kHelloWorld), fp.get())); 33 34 embedder::ScopedPlatformHandle h( 35 mojo::test::PlatformHandleFromFILE(fp.Pass())); 36 EXPECT_FALSE(fp); 37 ASSERT_TRUE(h.is_valid()); 38 39 scoped_refptr<PlatformHandleDispatcher> dispatcher( 40 new PlatformHandleDispatcher(h.Pass())); 41 EXPECT_FALSE(h.is_valid()); 42 EXPECT_EQ(Dispatcher::kTypePlatformHandle, dispatcher->GetType()); 43 44 h = dispatcher->PassPlatformHandle().Pass(); 45 EXPECT_TRUE(h.is_valid()); 46 47 fp = mojo::test::FILEFromPlatformHandle(h.Pass(), "rb").Pass(); 48 EXPECT_FALSE(h.is_valid()); 49 EXPECT_TRUE(fp); 50 51 rewind(fp.get()); 52 char read_buffer[1000] = {}; 53 EXPECT_EQ(sizeof(kHelloWorld), 54 fread(read_buffer, 1, sizeof(read_buffer), fp.get())); 55 EXPECT_STREQ(kHelloWorld, read_buffer); 56 57 // Try getting the handle again. (It should fail cleanly.) 58 h = dispatcher->PassPlatformHandle().Pass(); 59 EXPECT_FALSE(h.is_valid()); 60 61 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); 62 } 63 64 TEST(PlatformHandleDispatcherTest, CreateEquivalentDispatcherAndClose) { 65 base::ScopedTempDir temp_dir; 66 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 67 68 static const char kFooBar[] = "foo bar"; 69 70 base::FilePath unused; 71 base::ScopedFILE fp( 72 CreateAndOpenTemporaryFileInDir(temp_dir.path(), &unused)); 73 EXPECT_EQ(sizeof(kFooBar), fwrite(kFooBar, 1, sizeof(kFooBar), fp.get())); 74 75 scoped_refptr<PlatformHandleDispatcher> dispatcher( 76 new PlatformHandleDispatcher( 77 mojo::test::PlatformHandleFromFILE(fp.Pass()))); 78 79 DispatcherTransport transport( 80 test::DispatcherTryStartTransport(dispatcher.get())); 81 EXPECT_TRUE(transport.is_valid()); 82 EXPECT_EQ(Dispatcher::kTypePlatformHandle, transport.GetType()); 83 EXPECT_FALSE(transport.IsBusy()); 84 85 scoped_refptr<Dispatcher> generic_dispatcher = 86 transport.CreateEquivalentDispatcherAndClose(); 87 ASSERT_TRUE(generic_dispatcher.get()); 88 89 transport.End(); 90 EXPECT_TRUE(dispatcher->HasOneRef()); 91 dispatcher = nullptr; 92 93 ASSERT_EQ(Dispatcher::kTypePlatformHandle, generic_dispatcher->GetType()); 94 dispatcher = static_cast<PlatformHandleDispatcher*>(generic_dispatcher.get()); 95 96 fp = mojo::test::FILEFromPlatformHandle(dispatcher->PassPlatformHandle(), 97 "rb").Pass(); 98 EXPECT_TRUE(fp); 99 100 rewind(fp.get()); 101 char read_buffer[1000] = {}; 102 EXPECT_EQ(sizeof(kFooBar), 103 fread(read_buffer, 1, sizeof(read_buffer), fp.get())); 104 EXPECT_STREQ(kFooBar, read_buffer); 105 106 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); 107 } 108 109 } // namespace 110 } // namespace system 111 } // namespace mojo 112