shared_memory_hooks_unittest.cc 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. // Copyright 2019 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. #include "base/memory/shared_memory_hooks.h"
  5. #include "testing/gtest/include/gtest/gtest.h"
  6. #include "third_party/abseil-cpp/absl/types/optional.h"
  7. namespace base {
  8. class SharedMemoryHooksTest : public ::testing::Test {
  9. protected:
  10. void TearDown() override { SetCreateHooks(nullptr, nullptr, nullptr); }
  11. void SetCreateHooks(
  12. ReadOnlySharedMemoryRegion::CreateFunction* read_only_hook,
  13. UnsafeSharedMemoryRegion::CreateFunction* unsafe_hook,
  14. WritableSharedMemoryRegion::CreateFunction* writable_hook) {
  15. SharedMemoryHooks::SetCreateHooks(read_only_hook, unsafe_hook,
  16. writable_hook);
  17. }
  18. };
  19. absl::optional<size_t> requested_read_only_shmem_size;
  20. absl::optional<size_t> requested_unsafe_shmem_size;
  21. absl::optional<size_t> requested_writable_shmem_size;
  22. MappedReadOnlyRegion ReadOnlyShmemCreateHook(size_t size, SharedMemoryMapper* mapper) {
  23. requested_read_only_shmem_size = size;
  24. return {};
  25. }
  26. UnsafeSharedMemoryRegion UnsafeShmemCreateHook(size_t size) {
  27. requested_unsafe_shmem_size = size;
  28. return {};
  29. }
  30. WritableSharedMemoryRegion WritableShmemCreateHook(size_t size) {
  31. requested_writable_shmem_size = size;
  32. return {};
  33. }
  34. TEST_F(SharedMemoryHooksTest, Basic) {
  35. {
  36. auto region = ReadOnlySharedMemoryRegion::Create(3);
  37. EXPECT_TRUE(region.IsValid());
  38. EXPECT_FALSE(requested_read_only_shmem_size.has_value());
  39. }
  40. {
  41. auto region = UnsafeSharedMemoryRegion::Create(25);
  42. EXPECT_TRUE(region.IsValid());
  43. EXPECT_FALSE(requested_unsafe_shmem_size.has_value());
  44. }
  45. {
  46. auto region = WritableSharedMemoryRegion::Create(777);
  47. EXPECT_TRUE(region.IsValid());
  48. EXPECT_FALSE(requested_writable_shmem_size.has_value());
  49. }
  50. SetCreateHooks(&ReadOnlyShmemCreateHook, &UnsafeShmemCreateHook,
  51. &WritableShmemCreateHook);
  52. {
  53. auto region = ReadOnlySharedMemoryRegion::Create(3);
  54. EXPECT_FALSE(region.IsValid());
  55. EXPECT_EQ(3u, *requested_read_only_shmem_size);
  56. }
  57. {
  58. auto region = UnsafeSharedMemoryRegion::Create(25);
  59. EXPECT_FALSE(region.IsValid());
  60. EXPECT_EQ(25u, *requested_unsafe_shmem_size);
  61. }
  62. {
  63. auto region = WritableSharedMemoryRegion::Create(777);
  64. EXPECT_FALSE(region.IsValid());
  65. EXPECT_EQ(777u, *requested_writable_shmem_size);
  66. }
  67. }
  68. } // namespace base