bolt/Passes/StackReachingUses.h

//===--- Passes/StackReachingUses.h ---------------------------------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_TOOLS_LLVM_BOLT_PASSES_STACKREACHINGUSES_H
#define LLVM_TOOLS_LLVM_BOLT_PASSES_STACKREACHINGUSES_H

#include "DataflowAnalysis.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/Timer.h"

namespace opts {
extern llvm::cl::opt<bool> TimeOpts;
}

namespace llvm {
namespace bolt {

class FrameAnalysis;
struct FrameIndexEntry;

class StackReachingUses
    : public InstrsDataflowAnalysis<StackReachingUses, /*Backward=*/true> {
  friend class DataflowAnalysis<StackReachingUses, BitVector, true>;

public:
  StackReachingUses(const FrameAnalysis &FA, const BinaryContext &BC,
                    BinaryFunction &BF)
      : InstrsDataflowAnalysis(BC, BF), FA(FA) {}
  virtual ~StackReachingUses() {}

  /// Return true if the stack position written by the store in \p StoreFIE was
  /// later consumed by a load to a different register (not the same one used in
  /// the store). Useful for identifying loads/stores of callee-saved regs.
  bool isLoadedInDifferentReg(const FrameIndexEntry &StoreFIE,
                              ExprIterator Candidates) const;

  /// Answer whether the stack position written by the store represented in
  /// \p StoreFIE is loaded from or consumed in any way. The set of all
  /// relevant expressions reaching this store should be in \p Candidates.
  /// If \p IncludelocalAccesses is false, we only consider wheter there is
  /// a callee that consumes this stack position.
  bool isStoreUsed(const FrameIndexEntry &StoreFIE, ExprIterator Candidates,
                   bool IncludeLocalAccesses = true) const;

  void run() {
    NamedRegionTimer T1("SRU", "Dataflow", opts::TimeOpts);
    InstrsDataflowAnalysis<StackReachingUses, true>::run();
  }

protected:
  // Reference to the result of stack frame analysis
  const FrameAnalysis &FA;

  void preflight();

  BitVector getStartingStateAtBB(const BinaryBasicBlock &BB) {
    return BitVector(NumInstrs, false);
  }

  BitVector getStartingStateAtPoint(const MCInst &Point) {
    return BitVector(NumInstrs, false);
  }

  void doConfluence(BitVector &StateOut, const BitVector &StateIn) {
    StateOut |= StateIn;
  }

  // Define the function computing the kill set -- whether expression Y, a
  // tracked expression, will be considered to be dead after executing X.
  bool doesXKillsY(const MCInst *X, const MCInst *Y);
  BitVector computeNext(const MCInst &Point, const BitVector &Cur);

  StringRef getAnnotationName() const { return StringRef("StackReachingUses"); }
};

} // end namespace bolt
} // end namespace llvm

#endif