lib/Analysis/MemoryLocation.cpp

   1 //===- MemoryLocation.cpp - Memory location descriptions -------------------==//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "llvm/Analysis/MemoryLocation.h"
  11 #include "llvm/IR/BasicBlock.h"
  12 #include "llvm/IR/DataLayout.h"
  13 #include "llvm/IR/Instructions.h"
  14 #include "llvm/IR/IntrinsicInst.h"
  15 #include "llvm/IR/LLVMContext.h"
  16 #include "llvm/IR/Module.h"
  17 #include "llvm/IR/Type.h"
  18 using namespace llvm;
  19
  20 MemoryLocation MemoryLocation::get(const LoadInst *LI) {
  21   AAMDNodes AATags;
  22   LI->getAAMetadata(AATags);
  23   const auto &DL = LI->getModule()->getDataLayout();
  24
  25   return MemoryLocation(LI->getPointerOperand(),
  26                         DL.getTypeStoreSize(LI->getType()), AATags);
  27 }
  28
  29 MemoryLocation MemoryLocation::get(const StoreInst *SI) {
  30   AAMDNodes AATags;
  31   SI->getAAMetadata(AATags);
  32   const auto &DL = SI->getModule()->getDataLayout();
  33
  34   return MemoryLocation(SI->getPointerOperand(),
  35                         DL.getTypeStoreSize(SI->getValueOperand()->getType()),
  36                         AATags);
  37 }
  38
  39 MemoryLocation MemoryLocation::get(const VAArgInst *VI) {
  40   AAMDNodes AATags;
  41   VI->getAAMetadata(AATags);
  42
  43   return MemoryLocation(VI->getPointerOperand(), UnknownSize, AATags);
  44 }
  45
  46 MemoryLocation MemoryLocation::get(const AtomicCmpXchgInst *CXI) {
  47   AAMDNodes AATags;
  48   CXI->getAAMetadata(AATags);
  49   const auto &DL = CXI->getModule()->getDataLayout();
  50
  51   return MemoryLocation(
  52       CXI->getPointerOperand(),
  53       DL.getTypeStoreSize(CXI->getCompareOperand()->getType()), AATags);
  54 }
  55
  56 MemoryLocation MemoryLocation::get(const AtomicRMWInst *RMWI) {
  57   AAMDNodes AATags;
  58   RMWI->getAAMetadata(AATags);
  59   const auto &DL = RMWI->getModule()->getDataLayout();
  60
  61   return MemoryLocation(RMWI->getPointerOperand(),
  62                         DL.getTypeStoreSize(RMWI->getValOperand()->getType()),
  63                         AATags);
  64 }
  65
  66 MemoryLocation MemoryLocation::getForSource(const MemTransferInst *MTI) {
  67   uint64_t Size = UnknownSize;
  68   if (ConstantInt *C = dyn_cast<ConstantInt>(MTI->getLength()))
  69     Size = C->getValue().getZExtValue();
  70
  71   // memcpy/memmove can have AA tags. For memcpy, they apply
  72   // to both the source and the destination.
  73   AAMDNodes AATags;
  74   MTI->getAAMetadata(AATags);
  75
  76   return MemoryLocation(MTI->getRawSource(), Size, AATags);
  77 }
  78
  79 MemoryLocation MemoryLocation::getForDest(const MemIntrinsic *MTI) {
  80   uint64_t Size = UnknownSize;
  81   if (ConstantInt *C = dyn_cast<ConstantInt>(MTI->getLength()))
  82     Size = C->getValue().getZExtValue();
  83
  84   // memcpy/memmove can have AA tags. For memcpy, they apply
  85   // to both the source and the destination.
  86   AAMDNodes AATags;
  87   MTI->getAAMetadata(AATags);
  88
  89   return MemoryLocation(MTI->getRawDest(), Size, AATags);
  90 }