blob: 14f597513b054be0e774469cbea1177321746468 [file] [edit]
// Protocol Buffers - Google's data interchange format
// Copyright 2008 Google Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd
// A common header that is included across all protobuf headers. We do our best
// to avoid #defining any macros here; instead we generally put macros in
// port_def.inc and port_undef.inc so they are not visible from outside of
// protobuf.
#ifndef GOOGLE_PROTOBUF_PORT_H__
#define GOOGLE_PROTOBUF_PORT_H__
#include <atomic>
#include <cassert>
#include <cstddef>
#include <cstdint>
#include <new>
#include <string>
#include <type_traits>
#include <typeinfo>
#if defined(__ARM_FEATURE_CRC32)
#include <arm_acle.h>
#endif
#include "absl/base/optimization.h"
#include "absl/base/attributes.h"
#include "absl/base/config.h"
#include "absl/strings/string_view.h"
#include "absl/types/optional.h"
#if defined(ABSL_HAVE_ADDRESS_SANITIZER)
#include <sanitizer/asan_interface.h>
#endif
// must be last
#include "google/protobuf/port_def.inc"
namespace google {
namespace protobuf {
class MessageLite;
namespace internal {
PROTOBUF_EXPORT size_t StringSpaceUsedExcludingSelfLong(const std::string& str);
struct MessageTraitsImpl;
template <typename T>
PROTOBUF_ALWAYS_INLINE void StrongPointer(T* var) {
#if defined(__GNUC__)
asm("" : : "r"(var));
#else
auto volatile unused = var;
(void)&unused; // Use address to avoid an extra load of "unused".
#endif
}
#if defined(__x86_64__) && defined(__linux__) && !defined(__APPLE__) && \
!defined(__ANDROID__) && defined(__clang__) && __clang_major__ >= 19
// Optimized implementation for clang where we can generate a relocation without
// adding runtime instructions.
template <typename T, T ptr>
PROTOBUF_ALWAYS_INLINE void StrongPointer() {
// This injects a relocation in the code path without having to run code, but
// we can only do it with a newer clang.
asm(".reloc ., BFD_RELOC_NONE, %p0" ::"Ws"(ptr));
}
template <typename T, typename TraitsImpl = MessageTraitsImpl>
PROTOBUF_ALWAYS_INLINE void StrongReferenceToType() {
static constexpr auto ptr =
decltype(TraitsImpl::template value<T>)::StrongPointer();
// This is identical to the implementation of StrongPointer() above, but it
// has to be explicitly inlined here or else Clang 19 will raise an error in
// some configurations.
asm(".reloc ., BFD_RELOC_NONE, %p0" ::"Ws"(ptr));
}
#else // .reloc
// Portable fallback. It usually generates a single LEA instruction or
// equivalent.
template <typename T, T ptr>
PROTOBUF_ALWAYS_INLINE void StrongPointer() {
StrongPointer(ptr);
}
template <typename T, typename TraitsImpl = MessageTraitsImpl>
PROTOBUF_ALWAYS_INLINE void StrongReferenceToType() {
return StrongPointer(
decltype(TraitsImpl::template value<T>)::StrongPointer());
}
#endif // .reloc
// See comments on `AllocateAtLeast` for information on size returning new.
struct SizedPtr {
void* p;
size_t n;
};
// Debug hook allowing setting up test scenarios for AllocateAtLeast usage.
using AllocateAtLeastHookFn = SizedPtr (*)(size_t, void*);
// `AllocAtLeastHook` API
constexpr bool HaveAllocateAtLeastHook();
void SetAllocateAtLeastHook(AllocateAtLeastHookFn fn, void* context = nullptr);
#if !defined(NDEBUG) && defined(ABSL_HAVE_THREAD_LOCAL) && \
defined(__cpp_inline_variables)
// Hook data for current thread. These vars must not be accessed directly, use
// the 'HaveAllocateAtLeastHook()` and `SetAllocateAtLeastHook()` API instead.
inline thread_local AllocateAtLeastHookFn allocate_at_least_hook = nullptr;
inline thread_local void* allocate_at_least_hook_context = nullptr;
constexpr bool HaveAllocateAtLeastHook() { return true; }
inline void SetAllocateAtLeastHook(AllocateAtLeastHookFn fn, void* context) {
allocate_at_least_hook = fn;
allocate_at_least_hook_context = context;
}
#else // !NDEBUG && ABSL_HAVE_THREAD_LOCAL && __cpp_inline_variables
constexpr bool HaveAllocateAtLeastHook() { return false; }
inline void SetAllocateAtLeastHook(AllocateAtLeastHookFn fn, void* context) {}
#endif // !NDEBUG && ABSL_HAVE_THREAD_LOCAL && __cpp_inline_variables
// Allocates `size` bytes. This wrapper allows memory allocations to be
// optimized by the compiler since `operator new` is considered observable.
inline void* Allocate(size_t size) {
#if ABSL_HAVE_BUILTIN(__builtin_operator_new)
// Allows the compiler to merge or optimize away the allocation even if it
// would violate the observability guarantees of ::operator new.
return __builtin_operator_new(size);
#else
return ::operator new(size);
#endif
}
// Allocates at least `size` bytes. This function follows the c++ language
// proposal from D0901R10 (http://wg21.link/D0901R10) and will be implemented
// in terms of the new operator new semantics when available. The allocated
// memory should be released by a call to `SizedDelete` or `::operator delete`.
inline SizedPtr AllocateAtLeast(size_t size) {
#if !defined(NDEBUG) && defined(ABSL_HAVE_THREAD_LOCAL) && \
defined(__cpp_inline_variables)
if (allocate_at_least_hook != nullptr) {
return allocate_at_least_hook(size, allocate_at_least_hook_context);
}
#endif // !NDEBUG && ABSL_HAVE_THREAD_LOCAL && __cpp_inline_variables
return {Allocate(size), size};
}
inline void SizedDelete(void* p, size_t size) {
#if defined(__cpp_sized_deallocation)
::operator delete(p, size);
#else
// Avoid -Wunused-parameter
(void)size;
::operator delete(p);
#endif
}
inline void SizedArrayDelete(void* p, size_t size) {
#if defined(__cpp_sized_deallocation)
::operator delete[](p, size);
#else
// Avoid -Wunused-parameter
(void)size;
::operator delete[](p);
#endif
}
// Tag type used to invoke the constinit constructor overload of classes
// such as ArenaStringPtr and MapFieldBase. Such constructors are internal
// implementation details of the library.
struct ConstantInitialized {
explicit ConstantInitialized() = default;
};
// Tag type used to invoke the arena constructor overload of classes such
// as ExtensionSet and MapFieldLite in aggregate initialization. These
// classes typically don't have move/copy constructors, which rules out
// explicit initialization in pre-C++17.
struct ArenaInitialized {
explicit ArenaInitialized() = default;
};
template <typename To, typename From>
void AssertDownCast(From* from) {
static_assert(std::is_base_of<From, To>::value, "illegal DownCast");
// Check that this function is not used to downcast message types.
// For those we should use {Down,Dynamic}CastTo{Message,Generated}.
static_assert(!std::is_base_of_v<MessageLite, To>);
#if PROTOBUF_RTTI
// RTTI: debug mode only!
assert(from == nullptr || dynamic_cast<To*>(from) != nullptr);
#endif
}
template <typename To, typename From>
inline To DownCast(From* f) {
AssertDownCast<std::remove_pointer_t<To>>(f);
return static_cast<To>(f);
}
template <typename ToRef, typename From>
inline ToRef DownCast(From& f) {
AssertDownCast<std::remove_reference_t<ToRef>>(&f);
return static_cast<ToRef>(f);
}
// Looks up the name of `T` via RTTI, if RTTI is available.
template <typename T>
inline absl::optional<absl::string_view> RttiTypeName() {
#if PROTOBUF_RTTI
return typeid(T).name();
#else
return absl::nullopt;
#endif
}
// Helpers for identifying our supported types.
template <typename T>
struct is_supported_integral_type
: std::disjunction<std::is_same<T, int32_t>, std::is_same<T, uint32_t>,
std::is_same<T, int64_t>, std::is_same<T, uint64_t>,
std::is_same<T, bool>> {};
template <typename T>
struct is_supported_floating_point_type
: std::disjunction<std::is_same<T, float>, std::is_same<T, double>> {};
template <typename T>
struct is_supported_string_type
: std::disjunction<std::is_same<T, std::string>> {};
template <typename T>
struct is_supported_scalar_type
: std::disjunction<is_supported_integral_type<T>,
is_supported_floating_point_type<T>,
is_supported_string_type<T>> {};
template <typename T>
struct is_supported_message_type
: std::disjunction<std::is_base_of<MessageLite, T>> {
static constexpr auto force_complete_type = sizeof(T);
};
// To prevent sharing cache lines between threads
#ifdef __cpp_aligned_new
enum { kCacheAlignment = 64 };
#else
enum { kCacheAlignment = alignof(max_align_t) }; // do the best we can
#endif
// The maximum byte alignment we support.
enum { kMaxMessageAlignment = 8 };
inline constexpr bool EnableStableExperiments() {
#if defined(PROTOBUF_ENABLE_STABLE_EXPERIMENTS)
return true;
#else
return false;
#endif
}
inline constexpr bool EnableExperimentalMicroString() {
#if defined(PROTOBUF_ENABLE_EXPERIMENTAL_MICRO_STRING)
return true;
#endif
return EnableStableExperiments();
}
inline constexpr bool ForceInlineStringInProtoc() {
return EnableStableExperiments();
}
inline constexpr bool ForceEagerlyVerifiedLazyInProtoc() {
return EnableStableExperiments();
}
inline constexpr bool ForceSplitFieldsInProtoc() {
#if defined(PROTOBUF_FORCE_SPLIT)
return true;
#else
return false;
#endif
}
// Returns true if hasbits for repeated fields are enabled (b/391445226). This
// flag-gates the rollout of the feature, and if disabled will disable the
// feature. This will be removed once the feature is fully rolled out and
// verified.
inline constexpr bool EnableExperimentalHintHasBitsForRepeatedFields() {
return true;
}
// Returns true if debug hardening for clearing oneof message on arenas is
// enabled.
inline constexpr bool DebugHardenClearOneofMessageOnArena() {
#ifdef NDEBUG
return false;
#else
return true;
#endif
}
constexpr bool HasAnySanitizer() {
#if defined(ABSL_HAVE_ADDRESS_SANITIZER) || \
defined(ABSL_HAVE_MEMORY_SANITIZER) || defined(ABSL_HAVE_THREAD_SANITIZER)
return true;
#else
return false;
#endif
}
constexpr bool PerformDebugChecks() {
if (HasAnySanitizer()) return true;
#if defined(NDEBUG)
return false;
#else
return true;
#endif
}
// Force copy the default string to a string field so that non-optimized builds
// have harder-to-rely-on address stability.
constexpr bool DebugHardenForceCopyDefaultString() {
return false;
}
constexpr bool DebugHardenForceCopyInRelease() {
return false;
}
constexpr bool DebugHardenForceCopyInSwap() {
return false;
}
constexpr bool DebugHardenForceCopyInMove() {
return false;
}
constexpr bool DebugHardenForceAllocationOnConstruction() {
return false;
}
constexpr bool DebugHardenFuzzMessageSpaceUsedLong() {
return false;
}
inline constexpr bool DebugHardenCheckHasBitConsistency() {
#if !defined(NDEBUG) || defined(ABSL_HAVE_ADDRESS_SANITIZER) || \
defined(ABSL_HAVE_MEMORY_SANITIZER) || defined(ABSL_HAVE_THREAD_SANITIZER)
return true;
#endif
return false;
}
// Reads n bytes from p, if PerformDebugChecks() is true. This allows ASAN to
// detect if a range of memory is not valid when we expect it to be. The
// volatile keyword is necessary here to prevent the compiler from optimizing
// away the memory reads below.
inline void AssertBytesAreReadable(const volatile char* p, int n) {
if (PerformDebugChecks()) {
for (int i = 0; i < n; ++i) {
p[i];
}
}
}
// Returns true if pointers are 8B aligned, leaving least significant 3 bits
// available.
inline constexpr bool PtrIsAtLeast8BAligned() { return alignof(void*) >= 8; }
inline constexpr bool IsLazyParsingSupported() {
// We need 3 bits for pointer tagging in lazy parsing.
return PtrIsAtLeast8BAligned();
}
#if defined(ABSL_IS_LITTLE_ENDIAN)
constexpr bool IsLittleEndian() { return true; }
#elif defined(ABSL_IS_BIG_ENDIAN)
constexpr bool IsLittleEndian() { return false; }
#else
#error "Only little-endian and big-endian are supported"
#endif
constexpr bool IsBigEndian() { return !IsLittleEndian(); }
//----------------------- Cache-prefetching utilities --------------------------
struct PrefetchOpts {
// WARNING: The numeric values of `Locality` and `MemOp` are significant
// because they are directly consumed by `__builtin_prefetch()`:
// see https://gcc.gnu.org/onlinedocs/gcc/Other-Builtins.html.
// Indicates the cache locality to prefetch into.
enum Locality : int {
// Prefetch data into non-temporal cache structure and into a location close
// to the processor, minimizing cache pollution.
kNta = 0,
// Prefetch data into L3 cache, or an implementation-specific choice.
kLow = 1,
// Prefetch data into L3 and L2 cache.
kMedium = 2,
// Prefetch data into all levels of cache.
kHigh = 3,
};
// Indicates the intended memory access type to optimize prefetching for.
enum MemOp : int { kRead = 0, kWrite = 1 };
// Specifies the unit of `Amount` below.
enum Unit : int { kBytes, kLines, kObjects };
// The amount to prefetch, or the distance to prefetch from.
struct Amount {
#ifdef ABSL_REQUIRE_EXPLICIT_INIT
const size_t num ABSL_REQUIRE_EXPLICIT_INIT;
const Unit unit ABSL_REQUIRE_EXPLICIT_INIT;
#else
const size_t num = 1;
const Unit unit = kLines;
#endif
// Scales this amount to bytes. If `unit` is `kObjects`, `T` must be a valid
// pointed-to type. If it is not, an invalid zero amount is returned.
template <typename T>
constexpr Amount ToBytes() const {
switch (unit) {
case kBytes:
return *this;
case kLines:
return {num * ABSL_CACHELINE_SIZE, kBytes};
case kObjects:
if constexpr (!std::is_same_v<T, void>) {
return {num * sizeof(T), kBytes};
} else {
// Can't use `assert()` or `__builtin_trap()` here because they're
// not constexpr. Just return an invalid amount instead.
return {0, kBytes};
}
}
}
// Scales this amount to whole cache lines, rounding up. If `unit` is
// `kObjects`, `T` must be a valid pointed-to type. If it is not, an invalid
// zero amount is returned.
template <typename T>
constexpr Amount ToLines() const {
switch (unit) {
case kBytes:
return {
(num + ABSL_CACHELINE_SIZE - 1) / ABSL_CACHELINE_SIZE,
kLines,
};
case kLines:
return *this;
case kObjects:
if constexpr (!std::is_same_v<T, void>) {
return {
(num * sizeof(T) + ABSL_CACHELINE_SIZE - 1) /
ABSL_CACHELINE_SIZE,
kLines,
};
} else {
// Can't use `assert()` or `__builtin_trap()` here because they're
// not constexpr. Just return an invalid amount instead.
return {0, kBytes};
}
}
}
};
#ifdef ABSL_REQUIRE_EXPLICIT_INIT
const Amount num ABSL_REQUIRE_EXPLICIT_INIT;
#else
const Amount num = {1, kLines};
#endif
const Amount from = {0, kBytes};
const Locality locality = kHigh;
const MemOp mem_op = kRead;
};
// NOTE: Enable prefetching with Clang only: various problems with other
// compilers, especially old ones.
#if defined(__clang__) && ABSL_HAVE_BUILTIN(__builtin_prefetch)
namespace detail {
// Prefetches a single cache line. To form the address to prefetch, the base
// `ptr` is first offset by `kOpts.from.num` bytes and furthermore by `line`
// cache lines (note that `line` overrides `kOpts.num.num`).
template <const PrefetchOpts& kOpts>
PROTOBUF_ALWAYS_INLINE void PrefetchLine(const void* ptr, size_t line) {
static_assert(kOpts.from.unit == PrefetchOpts::kBytes);
const ptrdiff_t offset = kOpts.from.num + (line * ABSL_CACHELINE_SIZE);
// Pointer + offset overflows don't matter for prefetching, because the
// prefetch instruction is just a no-op for invalid addresses (although
// potentially incurring the cost of a TLB page-walk if there's no valid
// mapping for the page - but that should be rare in practice). Still, to
// formally avoid UB, we perform the arithmetic in uintptr_t space.
const void* prefetch_ptr =
reinterpret_cast<const void*>(reinterpret_cast<uintptr_t>(ptr) + offset);
__builtin_prefetch(prefetch_ptr, kOpts.mem_op, kOpts.locality);
}
} // namespace detail
// Prefetches a sequence of `kOpts.num.ToLines()` cache lines to the levels of
// cache specified by `kOpts.locality`, starting at `ptr` base pointer
// furthermore offset by `kOpts.from.ToBytes()` bytes, and optimized for
// `kOpts.mem_op` type of expected memory access.
//
// The `kOpts` template parameter must be a compile-time constant, which means
// either `inline constexpr` in the global scope or `static constexpr` in a
// function or class.
//
// When `kOpts.num.unit` or `kOpts.from.unit` is `kObjects`, the `T` template
// parameter must be explicitly specified and `sizeof(T)` must be valid and
// non-zero (i.e. T must be a non-void, complete type): it is used to scale
// `kOpts.num.num` and `kOpts.from.num` to bytes and lines, respectively.
//
// The `U` template parameter doesn't need to be explicitly specified: it is
// deduced from `ptr` and, if non-void and `T` is also non-void, checked for
// compatibility with `T` to prevent accidental mismatches between the actual
// pointed-to and declared prefetched types.
//
// WARNING: Do not default `T` to `U` or vice versa: that may hide subtle errors
// at call sites, e.g. when `ptr` points at the base class of the actual object.
//
// TODO: Simplify definition/usages after C++20 per the bug.
template <const PrefetchOpts& kOpts, typename T = void, typename U>
PROTOBUF_ALWAYS_INLINE void Prefetch(const U* ptr) {
// TODO: Add a check: prefetched amount <= some reasonable limit.
if constexpr (kOpts.num.unit == PrefetchOpts::kObjects ||
kOpts.from.unit == PrefetchOpts::kObjects) {
static_assert(sizeof(T) > 0, "Need explicit, non-void, complete T");
}
if constexpr (!std::is_void_v<T> && !std::is_void_v<U>) {
// Prevent accidental mistakes, but only when it's matters.
static_assert(std::is_convertible_v<T*, U*>, "Type mismatch");
}
static constexpr PrefetchOpts kScaledOpts = {
kOpts.num.ToLines<T>(),
kOpts.from.ToBytes<T>(),
kOpts.locality,
kOpts.mem_op,
};
// Unroll the loop iterations by blocks of 16 in optimized builds.
#pragma unroll 16
for (size_t line = 0; line < kScaledOpts.num.num; ++line) {
detail::PrefetchLine<kScaledOpts>(ptr, line);
}
}
// Legacy prefetch functions.
// TODO: Replace calls to these functions and remove them per the
// bug.
// Prefetch 5 64-byte cache line starting from 7 cache-lines ahead.
// Constants are somewhat arbitrary and pretty aggressive, but were
// chosen to give a better benchmark results. E.g. this is ~20%
// faster, single cache line prefetch is ~12% faster, increasing
// decreasing distance makes results 2-4% worse. Important note,
// prefetch doesn't require a valid address, so it is ok to prefetch
// past the end of message/valid memory. Only insert prefetch once per function.
PROTOBUF_ALWAYS_INLINE void Prefetch5LinesFrom7Lines(const void* ptr) {
static constexpr PrefetchOpts kOpts = {
/*num=*/{5, PrefetchOpts::kLines},
/*from=*/{7, PrefetchOpts::kLines},
/*locality=*/PrefetchOpts::kHigh,
};
Prefetch<kOpts>(ptr);
}
// Prefetch 5 64-byte cache lines starting from 1 cache-line ahead.
PROTOBUF_ALWAYS_INLINE void Prefetch5LinesFrom1Line(const void* ptr) {
static constexpr PrefetchOpts kOpts = {
/*num=*/{5, PrefetchOpts::kLines},
/*from=*/{1, PrefetchOpts::kLines},
/*locality=*/PrefetchOpts::kHigh,
};
Prefetch<kOpts>(ptr);
}
// This trampoline allows calling from codegen without needing a #include to
// absl. It simplifies IWYU and deps.
inline void PrefetchToLocalCache(const void* ptr) {
static constexpr PrefetchOpts kOpts = {
/*num=*/{1, PrefetchOpts::kLines},
/*from=*/{0, PrefetchOpts::kLines},
/*locality=*/PrefetchOpts::kHigh,
};
Prefetch<kOpts>(ptr);
}
#else // defined(__clang__) || ABSL_HAVE_BUILTIN(__builtin_prefetch)
template <const PrefetchOpts& kOpts, typename T, typename U>
PROTOBUF_ALWAYS_INLINE void Prefetch(const void*) {}
PROTOBUF_ALWAYS_INLINE void Prefetch5LinesFrom7Lines(const void* ptr) {}
PROTOBUF_ALWAYS_INLINE void Prefetch5LinesFrom1Line(const void* ptr) {}
inline void PrefetchToLocalCache(const void* ptr) {}
#endif // defined(__clang__) && ABSL_HAVE_BUILTIN(__builtin_prefetch)
#if defined(NDEBUG) && ABSL_HAVE_BUILTIN(__builtin_unreachable)
[[noreturn]] ABSL_ATTRIBUTE_COLD PROTOBUF_ALWAYS_INLINE void Unreachable() {
__builtin_unreachable();
}
#elif ABSL_HAVE_BUILTIN(__builtin_FILE) && ABSL_HAVE_BUILTIN(__builtin_LINE)
[[noreturn]] ABSL_ATTRIBUTE_COLD inline void Unreachable(
const char* file = __builtin_FILE(), int line = __builtin_LINE()) {
protobuf_assumption_failed("Unreachable", file, line);
}
#else
[[noreturn]] ABSL_ATTRIBUTE_COLD inline void Unreachable() {
protobuf_assumption_failed("Unreachable", "", 0);
}
#endif
constexpr bool HasMemoryPoisoning() {
#if defined(ABSL_HAVE_ADDRESS_SANITIZER)
return true;
#else
return false;
#endif
}
// Poison memory region when supported by sanitizer config.
inline void PoisonMemoryRegion([[maybe_unused]] const void* p,
[[maybe_unused]] size_t n) {
#if defined(ABSL_HAVE_ADDRESS_SANITIZER)
ASAN_POISON_MEMORY_REGION(p, n);
#else
// Nothing
#endif
}
inline void UnpoisonMemoryRegion([[maybe_unused]] const void* p,
[[maybe_unused]] size_t n) {
#if defined(ABSL_HAVE_ADDRESS_SANITIZER)
ASAN_UNPOISON_MEMORY_REGION(p, n);
#else
// Nothing
#endif
}
inline bool IsMemoryPoisoned([[maybe_unused]] const void* p) {
#if defined(ABSL_HAVE_ADDRESS_SANITIZER)
return __asan_address_is_poisoned(p);
#else
return false;
#endif
}
inline constexpr bool ShouldBatchSingularString() {
#ifdef PROTOBUF_INTERNAL_BATCH_SINGULAR_STRING
return true;
#else
return false;
#endif
}
inline constexpr bool ShouldBatchRepeatedString() {
#ifdef PROTOBUF_INTERNAL_BATCH_REPEATED_STRING
return true;
#else
return false;
#endif
}
inline constexpr bool ShouldBatchRepeatedNumeric() {
#ifdef PROTOBUF_INTERNAL_BATCH_REPEATED_NUMERIC
return true;
#else
return false;
#endif
}
inline constexpr bool UseBatchOffset() {
#ifdef PROTOBUF_INTERNAL_USE_BATCH_OFFSET
return true;
#else
return false;
#endif
}
#if defined(ABSL_HAVE_THREAD_SANITIZER)
// TODO: it would be preferable to use __tsan_external_read/
// __tsan_external_write, but they can cause dlopen issues.
template <typename T>
PROTOBUF_ALWAYS_INLINE void TSanRead(const T* impl) {
char protobuf_tsan_dummy = impl->_tsan_detect_race;
asm volatile("" : "+r"(protobuf_tsan_dummy));
}
// We currently use a dedicated member for TSan checking so the value of this
// member is not important. We can unconditionally write to it without affecting
// correctness of the rest of the class.
template <typename T>
PROTOBUF_ALWAYS_INLINE void TSanWrite(T* impl) {
impl->_tsan_detect_race = 0;
}
#else
PROTOBUF_ALWAYS_INLINE void TSanRead(const void*) {}
PROTOBUF_ALWAYS_INLINE void TSanWrite(const void*) {}
#endif
// Like C++20's std::type_identity_t, usually used to alter type deduction in
// templates.
template <typename T>
using type_identity_t = std::enable_if_t<true, T>;
template <typename T>
constexpr T* Launder(T* p) {
#if defined(__cpp_lib_launder) && __cpp_lib_launder >= 201606L
return std::launder(p);
#elif ABSL_HAVE_BUILTIN(__builtin_launder)
return __builtin_launder(p);
#else
return p;
#endif
}
#if defined(PROTOBUF_CUSTOM_VTABLE)
template <typename T>
constexpr bool EnableCustomNewFor() {
return true;
}
#elif ABSL_HAVE_BUILTIN(__is_bitwise_cloneable)
template <typename T>
constexpr bool EnableCustomNewFor() {
return __is_bitwise_cloneable(T);
}
#else
template <typename T>
constexpr bool EnableCustomNewFor() {
return false;
}
#endif
// Counter library for debugging internal protobuf logic.
// It allows instrumenting code that has different options (eg fast vs slow
// path) to get visibility into how much we are hitting each path.
// When compiled with -DPROTOBUF_INTERNAL_ENABLE_DEBUG_COUNTERS, the counters
// register an atexit handler to dump the table. Otherwise, they are a noop and
// have not runtime cost.
//
// Usage:
//
// if (do_fast) {
// PROTOBUF_DEBUG_COUNTER("Foo.Fast").Inc();
// ...
// } else {
// PROTOBUF_DEBUG_COUNTER("Foo.Slow").Inc();
// ...
// }
class PROTOBUF_EXPORT RealDebugCounter {
public:
explicit RealDebugCounter(absl::string_view name) { Register(name); }
// Lossy increment.
void Inc() { counter_.store(value() + 1, std::memory_order_relaxed); }
size_t value() const { return counter_.load(std::memory_order_relaxed); }
private:
void Register(absl::string_view name);
std::atomic<size_t> counter_{};
};
// When the feature is not enabled, the type is a noop.
class NoopDebugCounter {
public:
explicit constexpr NoopDebugCounter() = default;
constexpr void Inc() {}
};
// Pretty random large number that seems like a safe allocation on most systems.
inline constexpr size_t kSafeStringSize = 50000000;
// Default empty string object. Don't use this directly. Instead, call
// GetEmptyString() to get the reference. This empty string is aligned with a
// minimum alignment of 8 bytes to match the requirement of ArenaStringPtr.
// Take advantage of C++20 constexpr support in std::string.
class alignas(8) GlobalEmptyStringConstexpr {
public:
const std::string& get() const { return value_; }
// Nothing to init, or destroy.
std::string* Init() const { return nullptr; }
// Disable the optimization for MSVC and Xtensa.
// There are some builds where the default constructed string can't be used as
// `constinit` even though the constructor is `constexpr` and can be used
// during constant evaluation.
#if !defined(_MSC_VER) && !defined(__XTENSA__)
// Compilation fails on Xtensa: b/467129751
template <typename T = std::string, bool = (T(), true)>
static constexpr std::true_type HasConstexprDefaultConstructor(int) {
return {};
}
#endif
static constexpr std::false_type HasConstexprDefaultConstructor(char) {
return {};
}
private:
std::string value_;
};
class alignas(8) GlobalEmptyStringDynamicInit {
public:
const std::string& get() const {
return *reinterpret_cast<const std::string*>(internal::Launder(buffer_));
}
std::string* Init() {
return ::new (static_cast<void*>(buffer_)) std::string();
}
private:
alignas(std::string) char buffer_[sizeof(std::string)];
};
using GlobalEmptyString = std::conditional_t<
GlobalEmptyStringConstexpr::HasConstexprDefaultConstructor(0),
const GlobalEmptyStringConstexpr, GlobalEmptyStringDynamicInit>;
PROTOBUF_EXPORT extern GlobalEmptyString fixed_address_empty_string;
enum class BoundsCheckMode { kNoEnforcement, kReturnDefault, kAbort };
PROTOBUF_EXPORT constexpr BoundsCheckMode GetBoundsCheckMode() {
#if defined(PROTO2_OPENSOURCE) || \
defined(PROTOBUF_INTERNAL_BOUNDS_CHECK_MODE_ABORT)
return BoundsCheckMode::kAbort;
#elif defined(PROTOBUF_INTERNAL_BOUNDS_CHECK_MODE_RETURN_DEFAULT)
return BoundsCheckMode::kReturnDefault;
#else
return BoundsCheckMode::kNoEnforcement;
#endif
}
#if defined(__x86_64__) && defined(__SSE4_2__)
constexpr bool HasCrc32() { return true; }
inline uint32_t Crc32(uint32_t crc, uint64_t v) {
return __builtin_ia32_crc32di(crc, v);
}
#elif defined(__ARM_FEATURE_CRC32)
constexpr bool HasCrc32() { return true; }
inline uint32_t Crc32(uint32_t crc, uint64_t v) { return __crc32cd(crc, v); }
#else
constexpr bool HasCrc32() { return false; }
inline uint32_t Crc32(uint32_t, uint64_t) { return 0; }
#endif
} // namespace internal
} // namespace protobuf
} // namespace google
#include "google/protobuf/port_undef.inc"
#endif // GOOGLE_PROTOBUF_PORT_H__