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
17namespace clang {
18
19XRayInstrMask 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
34void 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