[PATCH] D53342: [SimplifyLibCalls][WIP] Mark known arguments with nonnull

[Dávid Bolvanský via Phabricator via llvm-commits]

xbolva00 updated this revision to Diff 169895.

https://reviews.llvm.org/D53342

Files:
  lib/Transforms/Utils/SimplifyLibCalls.cpp


Index: lib/Transforms/Utils/SimplifyLibCalls.cpp
===================================================================
--- lib/Transforms/Utils/SimplifyLibCalls.cpp
+++ lib/Transforms/Utils/SimplifyLibCalls.cpp
@@ -178,11 +178,25 @@
   return true;
 }
 
+template <typename T>
+static bool isPositiveSize(T *I, unsigned ArgNo) {
+  Value *Arg = I->getArgOperand(ArgNo);
+  return isa<ConstantInt>(Arg) && !cast<ConstantInt>(Arg)->isZero();
+}
+
+static CallInst *setNonNullParam(CallInst *CI, unsigned ArgNo) {
+  if (!CI->paramHasAttr(ArgNo, Attribute::NonNull))
+    CI->addParamAttr(ArgNo, Attribute::NonNull);
+  return nullptr;
+}
+
 //===----------------------------------------------------------------------===//
 // String and Memory Library Call Optimizations
 //===----------------------------------------------------------------------===//
 
 Value *LibCallSimplifier::optimizeStrCat(CallInst *CI, IRBuilder<> &B) {
+  setNonNullParam(CI, 0);
+  setNonNullParam(CI, 1);
   // Extract some information from the instruction
   Value *Dst = CI->getArgOperand(0);
   Value *Src = CI->getArgOperand(1);
@@ -221,6 +235,10 @@
 }
 
 Value *LibCallSimplifier::optimizeStrNCat(CallInst *CI, IRBuilder<> &B) {
+  if (isPositiveSize(CI, 2)) {
+    setNonNullParam(CI, 0);
+    setNonNullParam(CI, 1);
+  }
   // Extract some information from the instruction.
   Value *Dst = CI->getArgOperand(0);
   Value *Src = CI->getArgOperand(1);


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D53342.169895.patch
Type: text/x-patch
Size: 1432 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20181016/5c580ad9/attachment.bin>