revng-revng/lib/Pipes/RootKind.cpp

/// \file RootKind.cpp
/// \brief the kind associated to non isolated root.

//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#include <optional>

#include "llvm/Support/Casting.h"

#include "revng/Pipeline/AllRegistries.h"
#include "revng/Pipeline/Kind.h"
#include "revng/Pipeline/Target.h"
#include "revng/Pipes/Kinds.h"
#include "revng/Pipes/ModelInvalidationEvent.h"
#include "revng/Pipes/RootKind.h"
#include "revng/Support/FunctionTags.h"
#include "revng/TupleTree/Visits.h"

using namespace pipeline;
using namespace ::revng::pipes;

std::optional<Target>
RootKind::symbolToTarget(const llvm::Function &Symbol) const {
  if (FunctionTags::Root.isTagOf(&Symbol)
      and not FunctionTags::IsolatedRoot.isTagOf(&Symbol))
    return Target({}, *this);

  return std::nullopt;
}

std::optional<Target>
IsolatedRootKind::symbolToTarget(const llvm::Function &Symbol) const {
  if (FunctionTags::IsolatedRoot.isTagOf(&Symbol))
    return Target({}, *this);

  return std::nullopt;
}

void RootKind::getInvalidations(TargetsList &ToRemove,
                                const InvalidationEventBase &Base) const {

  const auto *Event(llvm::dyn_cast<ModelInvalidationEvent>(&Base));
  if (not Event)
    return;

  const TupleTreePath ToCheck = *stringAsPath<model::Binary>("/ExtraCodeAddre"
                                                             "ss"
                                                             "es");

  bool RootChanged = llvm::any_of(Event->getDiff().Changes,
                                  [&ToCheck](const auto &Entry) {
                                    const auto &[Path, Old, New] = Entry;
                                    return ToCheck.isPrefixOf(Path);
                                  });

  if (RootChanged)
    ToRemove.emplace_back(*this);
}