1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72


gpu / command_buffer / common / shm_count.cc [blame]

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

#include "gpu/command_buffer/common/shm_count.h"

#include "base/memory/shared_memory_mapping.h"
#include "base/memory/unsafe_shared_memory_region.h"

namespace gpu {

ShmCountBase::ShmCountBase() = default;
ShmCountBase::ShmCountBase(ShmCountBase&& other) = default;
ShmCountBase::~ShmCountBase() = default;

void ShmCountBase::Initialize(base::UnsafeSharedMemoryRegion region) {
  region_ = std::move(region);
  mapping_ = region_.Map();
}

bool ShmCountBase::IsInitialized() const {
  return region().IsValid();
}

volatile ShmCountBase::AtomicType* ShmCountBase::AsAtomic() {
  return reinterpret_cast<volatile AtomicType*>(mapping_.memory());
}

GpuProcessShmCount::GpuProcessShmCount() = default;
GpuProcessShmCount::GpuProcessShmCount(GpuProcessShmCount&& other) = default;

GpuProcessShmCount::GpuProcessShmCount(base::UnsafeSharedMemoryRegion region) {
  // In cases where we are running without a GpuProcessHost, we may not
  // have a valid handle. In this case, just return.
  if (!region.IsValid()) {
    return;
  }

  Initialize(std::move(region));
}

void GpuProcessShmCount::Increment() {
  // In cases where we are running without a GpuProcessHost, we may not
  // initialize the GpuProcessShmCount. In this case, just return.
  if (!IsInitialized()) {
    return;
  }

  AsAtomic()->fetch_add(1, std::memory_order_release);
}

void GpuProcessShmCount::Decrement() {
  // In cases where we are running without a GpuProcessHost, we may not
  // initialize the GpuProcessShmCount. In this case, just return.
  if (!IsInitialized()) {
    return;
  }

  CountType old_value = AsAtomic()->fetch_add(-1, std::memory_order_release);
  CHECK_GE(old_value, 1);
}

GpuProcessHostShmCount::GpuProcessHostShmCount() {
  Initialize(base::UnsafeSharedMemoryRegion::Create(sizeof(AtomicType)));
}

GpuProcessShmCount::CountType GpuProcessHostShmCount::GetCount() {
  DCHECK(IsInitialized());
  return AsAtomic()->load(std::memory_order_acquire);
}

}  // namespace gpu