[llvm] 39b02d4 - [instcombine] Support and test __builtin_object_size interaction with __strdup and __strndup

[via llvm-commits]

Author: serge-sans-paille
Date: 2022-03-21T11:30:51+01:00
New Revision: 39b02d49cc5ed89a4eaf766d03411d53c6579c47

URL: https://github.com/llvm/llvm-project/commit/39b02d49cc5ed89a4eaf766d03411d53c6579c47
DIFF: https://github.com/llvm/llvm-project/commit/39b02d49cc5ed89a4eaf766d03411d53c6579c47.diff

LOG: [instcombine] Support and test __builtin_object_size interaction with __strdup and __strndup

Differential Revision: https://reviews.llvm.org/D122005

Added: 
    llvm/test/Transforms/InstCombine/builtin-object-size-strdup-family.ll

Modified: 
    llvm/lib/Analysis/MemoryBuiltins.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Analysis/MemoryBuiltins.cpp b/llvm/lib/Analysis/MemoryBuiltins.cpp
index 34f7fc78619c5..05e38ebcfac59 100644
--- a/llvm/lib/Analysis/MemoryBuiltins.cpp
+++ b/llvm/lib/Analysis/MemoryBuiltins.cpp
@@ -149,7 +149,9 @@ static const std::pair<LibFunc, AllocFnsTy> AllocationFnData[] = {
     {LibFunc_vec_realloc,                       {ReallocLike,      2,  1, -1, -1, MallocFamily::VecMalloc}},
     {LibFunc_reallocf,                          {ReallocLike,      2,  1, -1, -1, MallocFamily::Malloc}},
     {LibFunc_strdup,                            {StrDupLike,       1, -1, -1, -1, MallocFamily::Malloc}},
+    {LibFunc_dunder_strdup,                     {StrDupLike,       1, -1, -1, -1, MallocFamily::Malloc}},
     {LibFunc_strndup,                           {StrDupLike,       2,  1, -1, -1, MallocFamily::Malloc}},
+    {LibFunc_dunder_strndup,                    {StrDupLike,       2,  1, -1, -1, MallocFamily::Malloc}},
     {LibFunc___kmpc_alloc_shared,               {MallocLike,       1,  0, -1, -1, MallocFamily::KmpcAllocShared}},
     // TODO: Handle "int posix_memalign(void **, size_t, size_t)"
 };

diff  --git a/llvm/test/Transforms/InstCombine/builtin-object-size-strdup-family.ll b/llvm/test/Transforms/InstCombine/builtin-object-size-strdup-family.ll
new file mode 100644
index 0000000000000..4d21693c6d5f2
--- /dev/null
+++ b/llvm/test/Transforms/InstCombine/builtin-object-size-strdup-family.ll
@@ -0,0 +1,51 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -instcombine -S < %s | FileCheck %s
+
+
+target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-unknown-linux-gnu"
+
+declare dso_local noalias noundef i8* @malloc(i64 noundef) local_unnamed_addr
+declare i64 @llvm.objectsize.i64.p0i8(i8*, i1 immarg, i1 immarg, i1 immarg)
+declare noalias i8* @strdup(i8*);
+declare noalias i8* @__strdup(i8*);
+declare noalias i8* @strndup(i8*, i64);
+declare noalias i8* @__strndup(i8*, i64);
+
+ at str = dso_local constant [11 x i8] c"toulbroc'h\00"
+
+define dso_local i64 @check_strdup(i32 noundef %n) local_unnamed_addr {
+; CHECK-LABEL: @check_strdup(
+; CHECK-NEXT:    ret i64 11
+;
+  %ptr = call noalias i8* @strdup(i8* noundef getelementptr inbounds ([11 x i8], [11 x i8]* @str, i64 0, i64 0))
+  %size = call i64 @llvm.objectsize.i64.p0i8(i8* %ptr, i1 false, i1 true, i1 false)
+  ret i64 %size
+}
+
+define dso_local i64 @check_dunder_strdup(i32 noundef %n) local_unnamed_addr {
+; CHECK-LABEL: @check_dunder_strdup(
+; CHECK-NEXT:    ret i64 11
+;
+  %ptr = call noalias i8* @__strdup(i8* noundef getelementptr inbounds ([11 x i8], [11 x i8]* @str, i64 0, i64 0))
+  %size = call i64 @llvm.objectsize.i64.p0i8(i8* %ptr, i1 false, i1 true, i1 false)
+  ret i64 %size
+}
+
+define dso_local i64 @check_strndup(i32 noundef %n) local_unnamed_addr {
+; CHECK-LABEL: @check_strndup(
+; CHECK-NEXT:    ret i64 5
+;
+  %ptr = call noalias i8* @strndup(i8* noundef getelementptr inbounds ([11 x i8], [11 x i8]* @str, i64 0, i64 0), i64 4)
+  %size = call i64 @llvm.objectsize.i64.p0i8(i8* %ptr, i1 false, i1 true, i1 false)
+  ret i64 %size
+}
+
+define dso_local i64 @check_dunder_strndup(i32 noundef %n) local_unnamed_addr {
+; CHECK-LABEL: @check_dunder_strndup(
+; CHECK-NEXT:    ret i64 5
+;
+  %ptr = call noalias i8* @__strndup(i8* noundef getelementptr inbounds ([11 x i8], [11 x i8]* @str, i64 0, i64 0), i64 4)
+  %size = call i64 @llvm.objectsize.i64.p0i8(i8* %ptr, i1 false, i1 true, i1 false)
+  ret i64 %size
+}