Method findRelatedArg

lib/propanalysis/ArgFlow.cpp:339–378

void ArgFlow::setArgDir(unsigned ArgDir) { Direction = ArgDir; }

Argument *ArgFlow::findRelatedArg(Value &V, std::set<Value *> &Visit) {
  if (Visit.find(&V) != Visit.end() || isa<Constant>(&V) || isa<CmpInst>(&V) ||
      isa<GetElementPtrInst>(&V)) {
    return nullptr;
  } else if (auto *I = dyn_cast<Instruction>(&V)) {
    if (I->isTerminator()) {
      return nullptr;
    }
  }
  Visit.insert(&V);

  if (auto *A = dyn_cast<Argument>(&V))
    return A;

  // traverse Operands and Users
  if (auto *I = dyn_cast<Instruction>(&V)) {
    if (isUnionFieldAccess(I) || isa<CallBase>(I))
      return nullptr;

    for (auto *O : I->operand_values()) {
      if (isa<Constant>(O))
        continue;

      if (auto *RelatedArg = findRelatedArg(*O, Visit))
        return RelatedArg;
    }
  }

  // FIXME: it is workaround to reverse order of users
  std::list<User *> users;
  for (auto *U : V.users())
    users.push_front(U);

  for (auto *U : users) {
    if (auto *RelatedArg = findRelatedArg(*U, Visit))
      return RelatedArg;
  }

  return nullptr;
}

void ArgFlow::setAllocSize() { IsAllocSize = true; }