Explicit Call Graph

include/phasar/ControlFlow/CallGraph.h

@@ -0,0 +1,304 @@
+/******************************************************************************
+ * Copyright (c) 2023 Fabian Schiebel.
+ * All rights reserved. This program and the accompanying materials are made
+ * available under the terms of LICENSE.txt.
+ *
+ * Contributors:
+ *     Fabian Schiebel and others
+ *****************************************************************************/
+
+#ifndef PHASAR_PHASARLLVM_CONTROLFLOW_CALLGRAPH_H
+#define PHASAR_PHASARLLVM_CONTROLFLOW_CALLGRAPH_H
+
+#include "phasar/ControlFlow/CallGraphBase.h"
+#include "phasar/Utils/ByRef.h"
+#include "phasar/Utils/Logger.h"
+#include "phasar/Utils/StableVector.h"
+#include "phasar/Utils/Utilities.h"
+
+#include "llvm/ADT/ArrayRef.h"
+#include "llvm/ADT/DenseMap.h"
+#include "llvm/ADT/STLExtras.h"
+
+#include "nlohmann/json.hpp"
+
+#include <functional>
+#include <utility>
+#include <vector>
+
+namespace psr {
+template <typename N, typename F> class CallGraphBuilder;
+template <typename N, typename F> class CallGraph;
+
+template <typename N, typename F> struct CGTraits<CallGraph<N, F>> {
+  using n_t = N;
+  using f_t = F;
+};
+
+/// An explicit graph-representation of a call-graph. Only represents the data,
+/// not the call-graph analysis that creates it.
+///
+/// This type is immutable. To incrementally build it from your call-graph
+/// analysis, use the CallGraphBuilder
+template <typename N, typename F>
+class CallGraph : public CallGraphBase<CallGraph<N, F>> {
+  using base_t = CallGraphBase<CallGraph<N, F>>;
+  friend base_t;
+  friend class CallGraphBuilder<N, F>;
+
+public:
+  using typename base_t::f_t;
+  using typename base_t::n_t;
+  using FunctionVertexTy = llvm::SmallVector<n_t>;
+  using InstructionVertexTy = llvm::SmallVector<f_t>;
+
+  /// Creates a new, empty call-graph
+  CallGraph() noexcept = default;
+
+  /// Deserializes a previously computed call-graph
+  template <typename FunctionGetter, typename InstructionGetter>
+  [[nodiscard]] static CallGraph
+  deserialize(const nlohmann::json &PrecomputedCG,
+              FunctionGetter GetFunctionFromName,
+              InstructionGetter GetInstructionFromId);
+
+  /// A range of all functions that are vertices in the call-graph. The number
+  /// of vertex functions can be retrieved by getNumVertexFunctions().
+  [[nodiscard]] auto getAllVertexFunctions() const noexcept {
+    return llvm::make_first_range(CallersOf);
+  }
+
+  /// A range of all call-sites that are vertices in the call-graph. The number
+  /// of vertex-callsites can be retrived by getNumVertexCallSites().
+  [[nodiscard]] auto getAllVertexCallSites() const noexcept {
+    return llvm::make_first_range(CalleesAt);
+  }
+
+  [[nodiscard]] size_t getNumVertexFunctions() const noexcept {
+    return CallersOf.size();
+  }
+  [[nodiscard]] size_t getNumVertexCallSites() const noexcept {
+    return CalleesAt.size();
+  }
+
+  /// The number of functions within this call-graph
+  [[nodiscard]] size_t size() const noexcept { return getNumVertexFunctions(); }
+
+  [[nodiscard]] bool empty() const noexcept { return CallersOf.empty(); }
+
+  /// Creates a JSON representation of this call-graph suitable for presistent
+  /// storage.
+  /// Use the ctor taking a json object for deserialization
+  template <typename FunctionIdGetter, typename InstIdGetter>
+  [[nodiscard]] nlohmann::json getAsJson(FunctionIdGetter GetFunctionId,
+                                         InstIdGetter GetInstructionId) const {
+    nlohmann::json J;
+
+    for (const auto &[Fun, Callers] : CallersOf) {
+      auto &JCallers = J[std::invoke(GetFunctionId, Fun)];
+
+      for (const auto &CS : *Callers) {
+        JCallers.push_back(std::invoke(GetInstructionId, CS));
+      }
+    }
+
+    return J;
+  }
+
+  template <typename FunctionLabelGetter, typename InstParentGetter,
+            typename InstLabelGetter>
+  void printAsDot(llvm::raw_ostream &OS, FunctionLabelGetter GetFunctionLabel,
+                  InstParentGetter GetFunctionFromInst,
+                  InstLabelGetter GetInstLabel) const {
+    OS << "digraph CallGraph{\n";
+    scope_exit CloseBrace = [&OS] { OS << "}\n"; };
+
+    llvm::DenseMap<f_t, size_t> Fun2Id;
+    Fun2Id.reserve(CallersOf.size());
+
+    size_t CurrId = 0;
+    for (const auto &Fun : getAllVertexFunctions()) {
+      OS << CurrId << "[label=\"";
+      OS.write_escaped(std::invoke(GetFunctionLabel, Fun)) << "\"];\n";
+      Fun2Id[Fun] = CurrId++;
+    }
+
+    for (const auto &[CS, Callees] : CalleesAt) {
+      const auto &Fun = std::invoke(GetFunctionFromInst, CS);
+
+      for (const auto &Succ : *Callees) {
+        OS << Fun2Id.lookup(Fun) << "->" << Fun2Id.lookup(Succ) << "[label=\"";
+        OS.write_escaped(std::invoke(GetInstLabel, CS)) << "\"];\n";
+      }
+    }
+  }
+
+private:
+  [[nodiscard]] llvm::ArrayRef<f_t>
+  getCalleesOfCallAtImpl(ByConstRef<n_t> Inst) const noexcept {
+    if (const auto *CalleesPtr = CalleesAt.lookup(Inst)) {
+      return *CalleesPtr;
+    }
+    return {};
+  }
+
+  [[nodiscard]] llvm::ArrayRef<n_t>
+  getCallersOfImpl(ByConstRef<f_t> Fun) const noexcept {
+    if (const auto *CallersPtr = CallersOf.lookup(Fun)) {
+      return *CallersPtr;
+    }
+    return {};
+  }
+
+  // ---
+
+  StableVector<InstructionVertexTy> InstVertexOwner;
+  std::vector<FunctionVertexTy> FunVertexOwner;
+
+  llvm::DenseMap<N, InstructionVertexTy *> CalleesAt{};
+  llvm::DenseMap<F, FunctionVertexTy *> CallersOf{};
+};
+
+/// A mutable wrapper over a CallGraph. Use this to build a call-graph from
+/// within your call-graph ananlysis.
+template <typename N, typename F> class CallGraphBuilder {
+public:
+  using n_t = typename CallGraph<N, F>::n_t;
+  using f_t = typename CallGraph<N, F>::f_t;
+  using FunctionVertexTy = typename CallGraph<n_t, f_t>::FunctionVertexTy;
+  using InstructionVertexTy = typename CallGraph<n_t, f_t>::InstructionVertexTy;
+
+  void reserve(size_t MaxNumFunctions) {
+    CG.FunVertexOwner.reserve(MaxNumFunctions);
+    CG.CalleesAt.reserve(MaxNumFunctions);
+    CG.CallersOf.reserve(MaxNumFunctions);
+  }
+
+  /// Registeres a new function in the call-graph. Returns a list of all
+  /// call-sites that are known so far to potentially call this function.
+  /// Do not manually add elements to this vector -- use addCallEdge instead.
+  [[nodiscard]] FunctionVertexTy *addFunctionVertex(f_t Fun) {
+    auto [It, Inserted] = CG.CallersOf.try_emplace(std::move(Fun), nullptr);
+    if (Inserted) {
+      auto Cap = CG.FunVertexOwner.capacity();
+      assert(CG.FunVertexOwner.size() < Cap &&
+             "Trying to add more than MaxNumFunctions Function Vertices");
+      It->second = &CG.FunVertexOwner.emplace_back();
+    }
+    return It->second;
+  }
+
+  /// Registeres a new call-site in the call-graph. Returns a list of all
+  /// callee functions that are known so far to potentially be called by this
+  /// function.
+  /// Do not manually add elements to this vector -- use addCallEdge instead.
+  [[nodiscard]] InstructionVertexTy *addInstructionVertex(n_t Inst) {
+    auto [It, Inserted] = CG.CalleesAt.try_emplace(std::move(Inst), nullptr);
+    if (Inserted) {
+      It->second = &CG.InstVertexOwner.emplace_back();
+    }
+    return It->second;
+  }
+
+  /// Tries to lookup the InstructionVertex for the given call-site. Returns
+  /// nullptr on failure.
+  [[nodiscard]] InstructionVertexTy *
+  getInstVertexOrNull(ByConstRef<n_t> Inst) const noexcept {
+    return CG.CalleesAt.lookup(Inst);
+  }
+
+  /// Adds a new directional edge to the call-graph indicating that CS may call
+  /// Callee
+  void addCallEdge(n_t CS, f_t Callee) {
+    auto IVtx = addInstructionVertex(CS);
+    auto FVtx = addFunctionVertex(Callee);
+    addCallEdge(std::move(CS), IVtx, std::move(Callee), FVtx);
+  }
+
+  /// Same as addCallEdge(n_t, f_t), but uses an already known
+  /// InstructionVertexTy to save a lookup
+  void addCallEdge(n_t CS, InstructionVertexTy *Callees, f_t Callee) {
+    auto *Callers = addFunctionVertex(Callee);
+    addCallEdge(std::move(CS), Callees, std::move(Callee), Callers);
+  }
+
+  /// Same as addCallEdge(n_t, f_t), but uses an already known
+  /// FunctionVertexTy to save a lookup
+  void addCallEdge(n_t CS, f_t Callee, FunctionVertexTy *Callers) {
+    auto *Callees = addInstructionVertex(CS);
+    addCallEdge(std::move(CS), Callees, std::move(Callee), Callers);
+  }
+
+  /// Moves the completely built call-graph out of this builder for further
+  /// use. Do not use the builder after it anymore.
+  [[nodiscard]] CallGraph<n_t, f_t> consumeCallGraph() noexcept {
+    return std::move(CG);
+  }
+
+  /// Returns a view on the current (partial) call-graph that has already been
+  /// constructed
+  [[nodiscard]] const CallGraph<n_t, f_t> &viewCallGraph() const noexcept {
+    return CG;
+  }
+
+private:
+  void addCallEdge(n_t CS, InstructionVertexTy *Callees, f_t Callee,
+                   FunctionVertexTy *Callers) {
+    Callees->push_back(std::move(Callee));
+    Callers->push_back(std::move(CS));
+  }
+
+  CallGraph<n_t, f_t> CG{};
+};
+
+template <typename N, typename F>
+template <typename FunctionGetter, typename InstructionGetter>
+[[nodiscard]] CallGraph<N, F>
+CallGraph<N, F>::deserialize(const nlohmann::json &PrecomputedCG,
+                             FunctionGetter GetFunctionFromName,
+                             InstructionGetter GetInstructionFromId) {
+  if (!PrecomputedCG.is_object()) {
+    PHASAR_LOG_LEVEL_CAT(ERROR, "CallGraph", "Invalid Json. Expected object");
+    return {};
+  }
+
+  CallGraphBuilder<N, F> CGBuilder;
+  CGBuilder.reserve(PrecomputedCG.size());
+
+  for (const auto &[FunName, CallerIDs] : PrecomputedCG.items()) {
+    const auto &Fun = std::invoke(GetFunctionFromName, FunName);
+    if (!Fun) {
+      PHASAR_LOG_LEVEL_CAT(WARNING, "CallGraph",
+                           "Invalid function name: " << FunName);
+      continue;
+    }
+
+    auto *CEdges = CGBuilder.addFunctionVertex(Fun);
+    CEdges->reserve(CallerIDs.size());
+
+    for (const auto &JId : CallerIDs) {
+      auto Id = JId.get<size_t>();
+      const auto &CS = std::invoke(GetInstructionFromId, Id);
+      if (!CS) {
+        PHASAR_LOG_LEVEL_CAT(WARNING, "CallGraph",
+                             "Invalid CAll-Instruction Id: " << Id);
+      }
+
+      CGBuilder.addCallEdge(CS, Fun);
+    }
+  }
+  return CGBuilder.consumeCallGraph();
+}
+} // namespace psr
+
+namespace llvm {
+class Function;
+class Instruction;
+} // namespace llvm
+
+extern template class psr::CallGraph<const llvm::Instruction *,
+                                     const llvm::Function *>;
+extern template class psr::CallGraphBuilder<const llvm::Instruction *,
+                                            const llvm::Function *>;
+
+#endif // PHASAR_PHASARLLVM_CONTROLFLOW_CALLGRAPH_H

include/phasar/ControlFlow/CallGraphBase.h

@@ -0,0 +1,59 @@
+/******************************************************************************
+ * Copyright (c) 2023 Fabian Schiebel.
+ * All rights reserved. This program and the accompanying materials are made
+ * available under the terms of LICENSE.txt.
+ *
+ * Contributors:
+ *     Fabian Schiebel and others
+ *****************************************************************************/
+
+#ifndef PHASAR_PHASARLLVM_CONTROLFLOW_CALLGRAPHBASE_H
+#define PHASAR_PHASARLLVM_CONTROLFLOW_CALLGRAPHBASE_H
+
+#include "phasar/Utils/ByRef.h"
+#include "phasar/Utils/TypeTraits.h"
+
+#include "nlohmann/json.hpp"
+
+namespace psr {
+template <typename T> struct CGTraits {
+  // using n_t
+  // using f_t
+};
+
+/// Base class of all CallGraph implementations within phasar (currently only
+/// CallGraph<N, F>).
+/// Only represents the data, not how to create it.
+template <typename Derived> class CallGraphBase {
+public:
+  using n_t = typename CGTraits<Derived>::n_t;
+  using f_t = typename CGTraits<Derived>::f_t;
+
+  /// Returns an iterable range of all possible callee candidates at the given
+  /// call-site induced by the used call-graph.
+  ///
+  /// NOTE: This function is typically called in a hot part of the analysis and
+  /// should therefore be very fast
+  [[nodiscard]] decltype(auto) getCalleesOfCallAt(ByConstRef<n_t> Inst) const
+      noexcept(noexcept(self().getCalleesOfCallAtImpl(Inst))) {
+    static_assert(
+        is_iterable_over_v<decltype(self().getCalleesOfCallAtImpl(Inst)), f_t>);
+    return self().getCalleesOfCallAtImpl(Inst);
+  }
+
+  /// Returns an iterable range of all possible call-site candidates that may
+  /// call the given function induced by the used call-graph.
+  [[nodiscard]] decltype(auto) getCallersOf(ByConstRef<f_t> Fun) const {
+    static_assert(
+        is_iterable_over_v<decltype(self().getCallersOfImpl(Fun)), n_t>);
+    return self().getCallersOfImpl(Fun);
+  }
+
+private:
+  const Derived &self() const noexcept {
+    return static_cast<const Derived &>(*this);
+  }
+};
+} // namespace psr
+
+#endif // PHASAR_PHASARLLVM_CONTROLFLOW_CALLGRAPHBASE_H