safecode/include/safecode/RegisterRuntimeInitializer.h

//===- include/InsertChecks/RegisterRuntimeInitializer.h --------*- C++ -*----//
// 
//                     The LLVM Compiler Infrastructure
//
// This file was developed by the LLVM research group and is distributed under
// the University of Illinois Open Source License. See LICENSE.TXT for details.
// 
//===----------------------------------------------------------------------===//
//
// Pass to register runtime initialization calls into user-space programs.
//
//===----------------------------------------------------------------------===//

#ifndef _REGISTER_RUNTIME_INITIALIZER_H_
#define _REGISTER_RUNTIME_INITIALIZER_H_

#include "llvm/Pass.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/DataLayout.h"

namespace llvm {

/// Base class of all passes which register variables into pools.
class RegisterRuntimeInitializer : public llvm::ModulePass {
private:
  const char * logfilename;

public:
  static char ID;
  RegisterRuntimeInitializer(const char * name = "") : llvm::ModulePass(ID) {
    logfilename = name;
  }
  virtual bool runOnModule(llvm::Module & M);
  virtual void getAnalysisUsage(llvm::AnalysisUsage &AU) const {
    AU.setPreservesAll();
  }

private:
  /// Construct the initializer
  void constructInitializer(llvm::Module & M);

  /// Insert the initializer into llvm.global_ctors
  void insertInitializerIntoGlobalCtorList(llvm::Module & M);

  /// Insert code to configure the log filename
  void setLogFileName (llvm::Module & M);
};

}

#endif