OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/memory/shared_memory.h" | 5 #include "base/memory/shared_memory.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 #include <stdint.h> | 8 #include <stdint.h> |
9 | 9 |
10 #include <memory> | 10 #include <memory> |
11 | 11 |
12 #include "base/atomicops.h" | 12 #include "base/atomicops.h" |
13 #include "base/macros.h" | 13 #include "base/macros.h" |
14 #include "base/memory/shared_memory_handle.h" | 14 #include "base/memory/shared_memory_handle.h" |
15 #include "base/process/kill.h" | 15 #include "base/process/kill.h" |
16 #include "base/rand_util.h" | 16 #include "base/rand_util.h" |
17 #include "base/strings/string_number_conversions.h" | 17 #include "base/strings/string_number_conversions.h" |
18 #include "base/sys_info.h" | 18 #include "base/sys_info.h" |
19 #include "base/test/multiprocess_test.h" | 19 #include "base/test/multiprocess_test.h" |
20 #include "base/threading/platform_thread.h" | 20 #include "base/threading/platform_thread.h" |
21 #include "base/time/time.h" | 21 #include "base/time/time.h" |
| 22 #include "base/unguessable_token.h" |
22 #include "build/build_config.h" | 23 #include "build/build_config.h" |
23 #include "testing/gtest/include/gtest/gtest.h" | 24 #include "testing/gtest/include/gtest/gtest.h" |
24 #include "testing/multiprocess_func_list.h" | 25 #include "testing/multiprocess_func_list.h" |
25 | 26 |
26 #if defined(OS_POSIX) | 27 #if defined(OS_POSIX) |
27 #include <errno.h> | 28 #include <errno.h> |
28 #include <fcntl.h> | 29 #include <fcntl.h> |
29 #include <sys/mman.h> | 30 #include <sys/mman.h> |
30 #include <sys/stat.h> | 31 #include <sys/stat.h> |
31 #include <sys/types.h> | 32 #include <sys/types.h> |
(...skipping 294 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
326 #if defined(OS_MACOSX) && !defined(OS_IOS) | 327 #if defined(OS_MACOSX) && !defined(OS_IOS) |
327 // The Mach functionality is tested in shared_memory_mac_unittest.cc. | 328 // The Mach functionality is tested in shared_memory_mac_unittest.cc. |
328 options.type = SharedMemoryHandle::POSIX; | 329 options.type = SharedMemoryHandle::POSIX; |
329 #endif | 330 #endif |
330 ASSERT_TRUE(writable_shmem.Create(options)); | 331 ASSERT_TRUE(writable_shmem.Create(options)); |
331 ASSERT_TRUE(writable_shmem.Map(options.size)); | 332 ASSERT_TRUE(writable_shmem.Map(options.size)); |
332 memcpy(writable_shmem.memory(), contents.data(), contents.size()); | 333 memcpy(writable_shmem.memory(), contents.data(), contents.size()); |
333 EXPECT_TRUE(writable_shmem.Unmap()); | 334 EXPECT_TRUE(writable_shmem.Unmap()); |
334 | 335 |
335 SharedMemoryHandle readonly_handle = writable_shmem.GetReadOnlyHandle(); | 336 SharedMemoryHandle readonly_handle = writable_shmem.GetReadOnlyHandle(); |
| 337 EXPECT_EQ(writable_shmem.handle().GetGUID(), readonly_handle.GetGUID()); |
336 ASSERT_TRUE(readonly_handle.IsValid()); | 338 ASSERT_TRUE(readonly_handle.IsValid()); |
337 SharedMemory readonly_shmem(readonly_handle, /*readonly=*/true); | 339 SharedMemory readonly_shmem(readonly_handle, /*readonly=*/true); |
338 | 340 |
339 ASSERT_TRUE(readonly_shmem.Map(contents.size())); | 341 ASSERT_TRUE(readonly_shmem.Map(contents.size())); |
340 EXPECT_EQ(contents, | 342 EXPECT_EQ(contents, |
341 StringPiece(static_cast<const char*>(readonly_shmem.memory()), | 343 StringPiece(static_cast<const char*>(readonly_shmem.memory()), |
342 contents.size())); | 344 contents.size())); |
343 EXPECT_TRUE(readonly_shmem.Unmap()); | 345 EXPECT_TRUE(readonly_shmem.Unmap()); |
344 | 346 |
345 // Make sure the writable instance is still writable. | 347 // Make sure the writable instance is still writable. |
(...skipping 252 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
598 PAGE_READONLY | SEC_IMAGE, 0, 0, kTestSectionName)); | 600 PAGE_READONLY | SEC_IMAGE, 0, 0, kTestSectionName)); |
599 EXPECT_TRUE(section_handle.IsValid()); | 601 EXPECT_TRUE(section_handle.IsValid()); |
600 | 602 |
601 // Check direct opening by name, from handle and duplicated from handle. | 603 // Check direct opening by name, from handle and duplicated from handle. |
602 SharedMemory shared_memory_open; | 604 SharedMemory shared_memory_open; |
603 EXPECT_TRUE(shared_memory_open.Open(kTestSectionName, true)); | 605 EXPECT_TRUE(shared_memory_open.Open(kTestSectionName, true)); |
604 EXPECT_FALSE(shared_memory_open.Map(1)); | 606 EXPECT_FALSE(shared_memory_open.Map(1)); |
605 EXPECT_EQ(nullptr, shared_memory_open.memory()); | 607 EXPECT_EQ(nullptr, shared_memory_open.memory()); |
606 | 608 |
607 SharedMemory shared_memory_handle_local( | 609 SharedMemory shared_memory_handle_local( |
608 SharedMemoryHandle(section_handle.Take()), true); | 610 SharedMemoryHandle(section_handle.Take(), UnguessableToken::Create()), |
| 611 true); |
609 EXPECT_FALSE(shared_memory_handle_local.Map(1)); | 612 EXPECT_FALSE(shared_memory_handle_local.Map(1)); |
610 EXPECT_EQ(nullptr, shared_memory_handle_local.memory()); | 613 EXPECT_EQ(nullptr, shared_memory_handle_local.memory()); |
611 | 614 |
612 // Check that a handle without SECTION_QUERY also can't be mapped as it can't | 615 // Check that a handle without SECTION_QUERY also can't be mapped as it can't |
613 // be checked. | 616 // be checked. |
614 SharedMemory shared_memory_handle_dummy; | 617 SharedMemory shared_memory_handle_dummy; |
615 SharedMemoryCreateOptions options; | 618 SharedMemoryCreateOptions options; |
616 options.size = 0x1000; | 619 options.size = 0x1000; |
617 EXPECT_TRUE(shared_memory_handle_dummy.Create(options)); | 620 EXPECT_TRUE(shared_memory_handle_dummy.Create(options)); |
618 HANDLE handle_no_query; | 621 HANDLE handle_no_query; |
619 EXPECT_TRUE(::DuplicateHandle( | 622 EXPECT_TRUE(::DuplicateHandle( |
620 ::GetCurrentProcess(), shared_memory_handle_dummy.handle().GetHandle(), | 623 ::GetCurrentProcess(), shared_memory_handle_dummy.handle().GetHandle(), |
621 ::GetCurrentProcess(), &handle_no_query, FILE_MAP_READ, FALSE, 0)); | 624 ::GetCurrentProcess(), &handle_no_query, FILE_MAP_READ, FALSE, 0)); |
622 SharedMemory shared_memory_handle_no_query( | 625 SharedMemory shared_memory_handle_no_query( |
623 SharedMemoryHandle(handle_no_query), true); | 626 SharedMemoryHandle(handle_no_query, UnguessableToken::Create()), true); |
624 EXPECT_FALSE(shared_memory_handle_no_query.Map(1)); | 627 EXPECT_FALSE(shared_memory_handle_no_query.Map(1)); |
625 EXPECT_EQ(nullptr, shared_memory_handle_no_query.memory()); | 628 EXPECT_EQ(nullptr, shared_memory_handle_no_query.memory()); |
626 } | 629 } |
627 #endif // defined(OS_WIN) | 630 #endif // defined(OS_WIN) |
628 | 631 |
629 // iOS does not allow multiple processes. | 632 // iOS does not allow multiple processes. |
630 // Android ashmem does not support named shared memory. | 633 // Android ashmem does not support named shared memory. |
631 // Mac SharedMemory does not support named shared memory. crbug.com/345734 | 634 // Mac SharedMemory does not support named shared memory. crbug.com/345734 |
632 #if !defined(OS_IOS) && !defined(OS_ANDROID) && !defined(OS_MACOSX) | 635 #if !defined(OS_IOS) && !defined(OS_ANDROID) && !defined(OS_MACOSX) |
633 // On POSIX it is especially important we test shmem across processes, | 636 // On POSIX it is especially important we test shmem across processes, |
(...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
700 memory.Close(); | 703 memory.Close(); |
701 SharedMemoryProcessTest::CleanUp(); | 704 SharedMemoryProcessTest::CleanUp(); |
702 } | 705 } |
703 | 706 |
704 MULTIPROCESS_TEST_MAIN(SharedMemoryTestMain) { | 707 MULTIPROCESS_TEST_MAIN(SharedMemoryTestMain) { |
705 return SharedMemoryProcessTest::TaskTestMain(); | 708 return SharedMemoryProcessTest::TaskTestMain(); |
706 } | 709 } |
707 #endif // !defined(OS_IOS) && !defined(OS_ANDROID) && !defined(OS_MACOSX) | 710 #endif // !defined(OS_IOS) && !defined(OS_ANDROID) && !defined(OS_MACOSX) |
708 | 711 |
709 } // namespace base | 712 } // namespace base |
OLD | NEW |