allocation_fault_injection.h

/*
 * Copyright (C) Max Dymond, <cmeister2@gmail.com>, et al.
 *
 * SPDX-License-Identifier: curl
 */
 
#ifndef PROTO_FUZZER_ALLOCATION_FAULT_INJECTION_H_
#define PROTO_FUZZER_ALLOCATION_FAULT_INJECTION_H_
 
#include <cstddef>
#include <cstdint>
 
namespace proto_fuzzer::allocation_fault_injection {
 
namespace internal {

inline thread_local std::size_t failure_suppression_depth = 0;
inline thread_local std::uint64_t failure_suppression_generation = 0;

inline std::uint64_t BeginFailureSuppression() noexcept {
  ++failure_suppression_depth;
  return failure_suppression_generation;
}

inline void EndFailureSuppression(std::uint64_t generation) noexcept {
  if (generation == failure_suppression_generation && failure_suppression_depth != 0) {
    --failure_suppression_depth;
  }
}

inline bool FailuresSuppressed() noexcept { return failure_suppression_depth != 0; }

inline void ResetFailureSuppression() noexcept {
  failure_suppression_depth = 0;
  ++failure_suppression_generation;
}
 
}  // namespace internal

class ScopedFailureSuppression final {
 public:
  ScopedFailureSuppression() noexcept : generation_(internal::BeginFailureSuppression()) {}
  ~ScopedFailureSuppression() { internal::EndFailureSuppression(generation_); }
 
  ScopedFailureSuppression(const ScopedFailureSuppression&) = delete;
  ScopedFailureSuppression& operator=(const ScopedFailureSuppression&) = delete;
  ScopedFailureSuppression(ScopedFailureSuppression&&) = delete;
  ScopedFailureSuppression& operator=(ScopedFailureSuppression&&) = delete;
 
 private:
  std::uint64_t generation_;
};

struct Statistics {
  std::uint64_t allocation_attempts = 0;
  std::uint64_t injected_failures = 0;
  std::size_t configured_failures = 0;
  bool armed = false;
};
 
bool InitializeCurl() noexcept;
bool ReinitializeCurl() noexcept;
bool CurlInitializationSucceeded() noexcept;
bool ClaimCurlInitializationReference() noexcept;
void Configure(const std::uint32_t* ordinals, std::size_t count) noexcept;
void Arm() noexcept;
void ArmNextAllocationFailure() noexcept;
void Disarm() noexcept;
void Reset() noexcept;
Statistics GetStatistics() noexcept;
 
// These are the process-wide callbacks passed to curl_global_init_mem. They
// remain visible so the focused unit test can verify the allocator contracts
// without depending on a particular libcurl call's private allocation trace.
namespace internal {
 
void* Malloc(std::size_t size) noexcept;
void Free(void* pointer) noexcept;
void* Realloc(void* pointer, std::size_t size) noexcept;
char* Strdup(const char* string) noexcept;
void* Calloc(std::size_t count, std::size_t size) noexcept;
 
}  // namespace internal
}  // namespace proto_fuzzer::allocation_fault_injection
 
#endif  // PROTO_FUZZER_ALLOCATION_FAULT_INJECTION_H_