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

base / check_deref.h [blame]

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

#include "base/check.h"
#include "base/compiler_specific.h"

#if CHECK_WILL_STREAM()
#include "base/logging.h"
#endif  // CHECK_WILL_STREAM()

#ifndef BASE_CHECK_DEREF_H_
#define BASE_CHECK_DEREF_H_

namespace logging {

// Returns a reference to pointee of `ptr` if `ptr` is not null, or dies if
// `ptr` is null.
//
// It is useful in initializers and direct assignments, where a direct `CHECK`
// call can't be used:
//
//   MyType& type_ref = CHECK_DEREF(MethodReturningAPointer());
//
// If your raw pointer is stored in a wrapped type like `unique_ptr` or
// `raw_ptr`, you should use their `.get()` methods to get the raw pointer
// before calling `CHECK_DEREF()`:
//
//   MyType& type_ref = CHECK_DEREF(your_wrapped_pointer.get());
//
#define CHECK_DEREF(ptr) \
  ::logging::CheckDeref(ptr, #ptr " != nullptr", __FILE__, __LINE__)

template <typename T>
[[nodiscard]] T& CheckDeref(T* ptr,
                            const char* message,
                            const char* file,
                            int line) {
  // Note: we can't just call `CHECK_NE(ptr, nullptr)` here, as that would
  // cause the error to be reported from this header, and we want the error
  // to be reported at the file and line of the caller.
  if (ptr == nullptr) [[unlikely]] {
#if !CHECK_WILL_STREAM()
    CheckFailure();
#else
    // `LogMessage` will die with a fatal error in its destructor.
    LogMessage(file, line, message);
#endif  // !CHECK_WILL_STREAM()
  }
  return *ptr;
}

}  // namespace logging

#endif  // BASE_CHECK_DEREF_H_