gin/v8_platform_thread_isolated_allocator.cc

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "gin/v8_platform_thread_isolated_allocator.h"
#include "base/allocator/partition_allocator/partition_root.h"

#if BUILDFLAG(ENABLE_THREAD_ISOLATION)

#include <sys/mman.h>
#include <sys/syscall.h>

#include "base/allocator/partition_allocator/thread_isolation/pkey.h"
#include "gin/thread_isolation.h"

#if BUILDFLAG(ENABLE_PKEYS)
#else  // BUILDFLAG(ENABLE_PKEYS)
#error Not implemented for non-pkey thread isolation
#endif  // BUILDFLAG(ENABLE_PKEYS)

namespace gin {

ThreadIsolatedAllocator::ThreadIsolatedAllocator() = default;
ThreadIsolatedAllocator::~ThreadIsolatedAllocator() = default;

void ThreadIsolatedAllocator::Initialize(int pkey) {
  pkey_ = pkey;
  allocator_.init(partition_alloc::PartitionOptions{
      .aligned_alloc =
          partition_alloc::PartitionOptions::AlignedAlloc::kAllowed,
      .cookie = partition_alloc::PartitionOptions::Cookie::kAllowed,
      .thread_isolation = partition_alloc::ThreadIsolationOption(pkey_),
  });
}

void* ThreadIsolatedAllocator::Allocate(size_t size) {
  return allocator_.root()->AllocWithFlagsNoHooks(
      0, size, partition_alloc::PartitionPageSize());
}

void ThreadIsolatedAllocator::Free(void* object) {
  allocator_.root()->FreeNoHooks(object);
}

enum ThreadIsolatedAllocator::Type ThreadIsolatedAllocator::Type() const {
  return Type::kPkey;
}

int ThreadIsolatedAllocator::Pkey() const {
  return pkey_;
}

}  // namespace gin

#endif  // BUILDFLAG(ENABLE_THREAD_ISOLATION)