llvm-project

1
//===--- XRayInstr.cpp ------------------------------------------*- C++ -*-===//
2
//
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
// See https://llvm.org/LICENSE.txt for license information.
5
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
//
7
//===----------------------------------------------------------------------===//
8
//
9
// This is part of XRay, a function call instrumentation system.
10
//
11
//===----------------------------------------------------------------------===//
12

13
#include "clang/Basic/XRayInstr.h"
14
#include "llvm/ADT/SmallVector.h"
15
#include "llvm/ADT/StringSwitch.h"
16

17
namespace clang {
18

19
XRayInstrMask parseXRayInstrValue(StringRef Value) {
20
  XRayInstrMask ParsedKind =
21
      llvm::StringSwitch<XRayInstrMask>(Value)
22
          .Case("all", XRayInstrKind::All)
23
          .Case("custom", XRayInstrKind::Custom)
24
          .Case("function",
25
                XRayInstrKind::FunctionEntry | XRayInstrKind::FunctionExit)
26
          .Case("function-entry", XRayInstrKind::FunctionEntry)
27
          .Case("function-exit", XRayInstrKind::FunctionExit)
28
          .Case("typed", XRayInstrKind::Typed)
29
          .Case("none", XRayInstrKind::None)
30
          .Default(XRayInstrKind::None);
31
  return ParsedKind;
32
}
33

34
void serializeXRayInstrValue(XRayInstrSet Set,
35
                             SmallVectorImpl<StringRef> &Values) {
36
  if (Set.Mask == XRayInstrKind::All) {
37
    Values.push_back("all");
38
    return;
39
  }
40

41
  if (Set.Mask == XRayInstrKind::None) {
42
    Values.push_back("none");
43
    return;
44
  }
45

46
  if (Set.has(XRayInstrKind::Custom))
47
    Values.push_back("custom");
48

49
  if (Set.has(XRayInstrKind::Typed))
50
    Values.push_back("typed");
51

52
  if (Set.has(XRayInstrKind::FunctionEntry) &&
53
      Set.has(XRayInstrKind::FunctionExit))
54
    Values.push_back("function");
55
  else if (Set.has(XRayInstrKind::FunctionEntry))
56
    Values.push_back("function-entry");
57
  else if (Set.has(XRayInstrKind::FunctionExit))
58
    Values.push_back("function-exit");
59
}
60
} // namespace clang
61