third_party/perfetto/src/profiling/memory/scoped_spinlock.cc

/*
 * Copyright (C) 2019 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "src/profiling/memory/scoped_spinlock.h"

#include <unistd.h>

#include <atomic>

#include "perfetto/ext/base/utils.h"

namespace {
constexpr bool IsPowerOfTwo(size_t v) {
  return (v != 0 && ((v & (v - 1)) == 0));
}
// Wait for ~1s before timing out (+- spurious wakeups from the sleeps).
constexpr unsigned kSleepAttempts = 1000;
constexpr unsigned kLockAttemptsPerSleep = 1024;
constexpr unsigned kSleepDurationUs = 1000;

static_assert(IsPowerOfTwo(kLockAttemptsPerSleep),
              "lock attempts of power of 2 produce faster code.");
}  // namespace

namespace perfetto {
namespace profiling {

void PoisonSpinlock(Spinlock* lock) {
  lock->poisoned.store(true, std::memory_order_relaxed);
}

void ScopedSpinlock::LockSlow(Mode mode) {
  size_t sleeps = 0;
  // We need to start with attempt = 1, otherwise
  // attempt % kLockAttemptsPerSleep is zero for the first iteration.
  for (size_t attempt = 1; mode == Mode::Blocking ||
                           attempt < kLockAttemptsPerSleep * kSleepAttempts;
       attempt++) {
    if (!lock_->locked.load(std::memory_order_relaxed) &&
        PERFETTO_LIKELY(
            !lock_->locked.exchange(true, std::memory_order_acquire))) {
      locked_ = true;
      break;
    }
    if (attempt % kLockAttemptsPerSleep == 0) {
      usleep(kSleepDurationUs);
      sleeps++;
    }
  }
  blocked_us_ = kSleepDurationUs * sleeps;
}

}  // namespace profiling
}  // namespace perfetto