Sanitizers.cpp

Go to the documentation of this file.

//===- Sanitizers.cpp - C Language Family Language Options ----------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
//  This file defines the classes from Sanitizers.h
//
//===----------------------------------------------------------------------===//
 
#include "clang/Basic/Sanitizers.h"
#include "llvm/ADT/Hashing.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringSwitch.h"
#include "llvm/Support/Format.h"
#include "llvm/Support/MathExtras.h"
#include "llvm/Support/raw_ostream.h"
#include <algorithm>
#include <optional>
 
using namespace clang;
 
static const double SanitizerMaskCutoffsEps = 0.000000001f;
 
void SanitizerMaskCutoffs::set(SanitizerMask K, double V) {
  if (V < SanitizerMaskCutoffsEps && Cutoffs.empty())
    return;
  for (unsigned int i = 0; i < SanitizerKind::SO_Count; ++i)
    if (K & SanitizerMask::bitPosToMask(i)) {
      Cutoffs.resize(SanitizerKind::SO_Count);
      Cutoffs[i] = V;
    }
}
 
std::optional<double> SanitizerMaskCutoffs::operator[](unsigned Kind) const {
  if (Cutoffs.empty() || Cutoffs[Kind] < SanitizerMaskCutoffsEps)
    return std::nullopt;
 
  return Cutoffs[Kind];
}
 
void SanitizerMaskCutoffs::clear(SanitizerMask K) { set(K, 0); }
 
// Once LLVM switches to C++17, the constexpr variables can be inline and we
// won't need this.
#define SANITIZER(NAME, ID) constexpr SanitizerMask SanitizerKind::ID;
#define SANITIZER_GROUP(NAME, ID, ALIAS)                                       \
  constexpr SanitizerMask SanitizerKind::ID;                                   \
  constexpr SanitizerMask SanitizerKind::ID##Group;
#include "clang/Basic/Sanitizers.def"
 
SanitizerMask clang::parseSanitizerValue(StringRef Value, bool AllowGroups) {
  SanitizerMask ParsedKind = llvm::StringSwitch<SanitizerMask>(Value)
#define SANITIZER(NAME, ID) .Case(NAME, SanitizerKind::ID)
#define SANITIZER_GROUP(NAME, ID, ALIAS)                                       \
  .Case(NAME, AllowGroups ? SanitizerKind::ID##Group : SanitizerMask())
#include "clang/Basic/Sanitizers.def"
    .Default(SanitizerMask());
  return ParsedKind;
}
 
bool clang::parseSanitizerWeightedValue(StringRef Value, bool AllowGroups,
                                        SanitizerMaskCutoffs &Cutoffs) {
  SanitizerMask ParsedKind = llvm::StringSwitch<SanitizerMask>(Value)
#define SANITIZER(NAME, ID) .StartsWith(NAME "=", SanitizerKind::ID)
#define SANITIZER_GROUP(NAME, ID, ALIAS)                                       \
  .StartsWith(NAME "=",                                                        \
              AllowGroups ? SanitizerKind::ID##Group : SanitizerMask())
#include "clang/Basic/Sanitizers.def"
                                 .Default(SanitizerMask());
 
  if (!ParsedKind)
    return false;
  auto [N, W] = Value.split('=');
  double A;
  if (W.getAsDouble(A))
    return false;
  A = std::clamp(A, 0.0, 1.0);
  // AllowGroups is already taken into account for ParsedKind,
  // hence we unconditionally expandSanitizerGroups.
  Cutoffs.set(expandSanitizerGroups(ParsedKind), A);
  return true;
}
 
void clang::serializeSanitizerSet(SanitizerSet Set,
                                  SmallVectorImpl<StringRef> &Values) {
#define SANITIZER(NAME, ID)                                                    \
  if (Set.has(SanitizerKind::ID))                                              \
    Values.push_back(NAME);
#include "clang/Basic/Sanitizers.def"
}
 
void clang::serializeSanitizerMaskCutoffs(
    const SanitizerMaskCutoffs &Cutoffs, SmallVectorImpl<std::string> &Values) {
#define SANITIZER(NAME, ID)                                                    \
  if (auto C = Cutoffs[SanitizerKind::SO_##ID]) {                              \
    std::string Str;                                                           \
    llvm::raw_string_ostream OS(Str);                                          \
    OS << NAME "=" << llvm::format("%.8f", *C);                                \
    Values.emplace_back(StringRef(Str).rtrim('0'));                            \
  }
#include "clang/Basic/Sanitizers.def"
}
 
SanitizerMask clang::expandSanitizerGroups(