llvm/test/Transforms/InstCombine/deref-alloc-fns.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -passes=instcombine  -S < %s | FileCheck %s
   3 ; RUN: opt -mtriple=x86_64-unknown-linux-gnu < %s -passes=instcombine -S | FileCheck %s --check-prefixes=CHECK,GNU
   4
   5
   6 declare noalias ptr @malloc(i64) allockind("alloc,uninitialized") allocsize(0) "alloc-family"="malloc"
   7 declare noalias ptr @calloc(i64, i64) allockind("alloc,zeroed") allocsize(0,1) "alloc-family"="malloc"
   8 declare noalias ptr @realloc(ptr nocapture, i64) allockind("realloc") allocsize(1) "alloc-family"="malloc"
   9 declare noalias nonnull ptr @_Znam(i64) ; throwing version of 'new'
  10 declare noalias nonnull ptr @_Znwm(i64) ; throwing version of 'new'
  11 declare noalias ptr @strdup(ptr)
  12 declare noalias ptr @aligned_alloc(i64 allocalign, i64) allockind("alloc,uninitialized,aligned") allocsize(1) "alloc-family"="malloc"
  13 declare noalias align 16 ptr @memalign(i64 allocalign, i64) allocsize(1)
  14 ; new[](unsigned int, align_val_t)
  15 declare noalias ptr @_ZnamSt11align_val_t(i64 %size, i64 %align)
  16
  17 declare ptr @my_malloc(i64) allocsize(0)
  18 declare ptr @my_calloc(i64, i64) allocsize(0, 1)
  19
  20 @.str = private unnamed_addr constant [6 x i8] c"hello\00", align 1
  21
  22 define noalias ptr @malloc_nonconstant_size(i64 %n) {
  23 ; CHECK-LABEL: @malloc_nonconstant_size(
  24 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @malloc(i64 [[N:%.*]])
  25 ; CHECK-NEXT:    ret ptr [[CALL]]
  26 ;
  27   %call = tail call noalias ptr @malloc(i64 %n)
  28   ret ptr %call
  29 }
  30
  31 define noalias ptr @malloc_constant_size() {
  32 ; CHECK-LABEL: @malloc_constant_size(
  33 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(40) ptr @malloc(i64 40)
  34 ; CHECK-NEXT:    ret ptr [[CALL]]
  35 ;
  36   %call = tail call noalias ptr @malloc(i64 40)
  37   ret ptr %call
  38 }
  39
  40 define noalias ptr @aligned_alloc_constant_size() {
  41 ; CHECK-LABEL: @aligned_alloc_constant_size(
  42 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias align 32 dereferenceable_or_null(512) ptr @aligned_alloc(i64 32, i64 512)
  43 ; CHECK-NEXT:    ret ptr [[CALL]]
  44 ;
  45   %call = tail call noalias ptr @aligned_alloc(i64 32, i64 512)
  46   ret ptr %call
  47 }
  48
  49 define noalias ptr @aligned_alloc_unknown_size_nonzero(i1 %c) {
  50 ; CHECK-LABEL: @aligned_alloc_unknown_size_nonzero(
  51 ; CHECK-NEXT:    [[SIZE:%.*]] = select i1 [[C:%.*]], i64 64, i64 128
  52 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias align 32 ptr @aligned_alloc(i64 32, i64 [[SIZE]])
  53 ; CHECK-NEXT:    ret ptr [[CALL]]
  54 ;
  55   %size = select i1 %c, i64 64, i64 128
  56   %call = tail call noalias ptr @aligned_alloc(i64 32, i64 %size)
  57   ret ptr %call
  58 }
  59
  60 define noalias ptr @aligned_alloc_unknown_size_possibly_zero(i1 %c) {
  61 ; CHECK-LABEL: @aligned_alloc_unknown_size_possibly_zero(
  62 ; CHECK-NEXT:    [[SIZE:%.*]] = select i1 [[C:%.*]], i64 64, i64 0
  63 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias align 32 ptr @aligned_alloc(i64 32, i64 [[SIZE]])
  64 ; CHECK-NEXT:    ret ptr [[CALL]]
  65 ;
  66   %size = select i1 %c, i64 64, i64 0
  67   %call = tail call noalias ptr @aligned_alloc(i64 32, i64 %size)
  68   ret ptr %call
  69 }
  70
  71 define noalias ptr @aligned_alloc_unknown_align(i64 %align) {
  72 ; CHECK-LABEL: @aligned_alloc_unknown_align(
  73 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(128) ptr @aligned_alloc(i64 [[ALIGN:%.*]], i64 128)
  74 ; CHECK-NEXT:    ret ptr [[CALL]]
  75 ;
  76   %call = tail call noalias ptr @aligned_alloc(i64 %align, i64 128)
  77   ret ptr %call
  78 }
  79
  80 declare noalias ptr @foo(ptr, ptr, ptr)
  81
  82 define noalias ptr @aligned_alloc_dynamic_args(i64 %align, i64 %size) {
  83 ; CHECK-LABEL: @aligned_alloc_dynamic_args(
  84 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(1024) ptr @aligned_alloc(i64 [[ALIGN:%.*]], i64 1024)
  85 ; CHECK-NEXT:    [[CALL_1:%.*]] = tail call noalias dereferenceable_or_null(1024) ptr @aligned_alloc(i64 0, i64 1024)
  86 ; CHECK-NEXT:    [[CALL_2:%.*]] = tail call noalias align 32 ptr @aligned_alloc(i64 32, i64 [[SIZE:%.*]])
  87 ; CHECK-NEXT:    [[TMP1:%.*]] = call ptr @foo(ptr [[CALL]], ptr [[CALL_1]], ptr [[CALL_2]])
  88 ; CHECK-NEXT:    ret ptr [[CALL]]
  89 ;
  90   %call = tail call noalias ptr @aligned_alloc(i64 %align, i64 1024)
  91   %call_1 = tail call noalias ptr @aligned_alloc(i64 0, i64 1024)
  92   %call_2 = tail call noalias ptr @aligned_alloc(i64 32, i64 %size)
  93
  94   call ptr @foo(ptr %call, ptr %call_1, ptr %call_2)
  95   ret ptr %call
  96 }
  97
  98 define noalias ptr @memalign_constant_size() {
  99 ; GNU-LABEL: @memalign_constant_size(
 100 ; GNU-NEXT:    [[CALL:%.*]] = tail call noalias align 32 dereferenceable_or_null(512) ptr @memalign(i64 32, i64 512)
 101 ; GNU-NEXT:    ret ptr [[CALL]]
 102 ;
 103   %call = tail call noalias ptr @memalign(i64 32, i64 512)
 104   ret ptr %call
 105 }
 106
 107 define noalias ptr @memalign_unknown_size_nonzero(i1 %c) {
 108 ; GNU-LABEL: @memalign_unknown_size_nonzero(
 109 ; GNU-NEXT:    [[SIZE:%.*]] = select i1 [[C:%.*]], i64 64, i64 128
 110 ; GNU-NEXT:    [[CALL:%.*]] = tail call noalias align 32 ptr @memalign(i64 32, i64 [[SIZE]])
 111 ; GNU-NEXT:    ret ptr [[CALL]]
 112 ;
 113   %size = select i1 %c, i64 64, i64 128
 114   %call = tail call noalias ptr @memalign(i64 32, i64 %size)
 115   ret ptr %call
 116 }
 117
 118 define noalias ptr @memalign_unknown_size_possibly_zero(i1 %c) {
 119 ; GNU-LABEL: @memalign_unknown_size_possibly_zero(
 120 ; GNU-NEXT:    [[SIZE:%.*]] = select i1 [[C:%.*]], i64 64, i64 0
 121 ; GNU-NEXT:    [[CALL:%.*]] = tail call noalias align 32 ptr @memalign(i64 32, i64 [[SIZE]])
 122 ; GNU-NEXT:    ret ptr [[CALL]]
 123 ;
 124   %size = select i1 %c, i64 64, i64 0
 125   %call = tail call noalias ptr @memalign(i64 32, i64 %size)
 126   ret ptr %call
 127 }
 128
 129 define noalias ptr @memalign_unknown_align(i64 %align) {
 130 ; GNU-LABEL: @memalign_unknown_align(
 131 ; GNU-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(128) ptr @memalign(i64 [[ALIGN:%.*]], i64 128)
 132 ; GNU-NEXT:    ret ptr [[CALL]]
 133 ;
 134   %call = tail call noalias ptr @memalign(i64 %align, i64 128)
 135   ret ptr %call
 136 }
 137
 138 define noalias ptr @malloc_constant_size2() {
 139 ; CHECK-LABEL: @malloc_constant_size2(
 140 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(40) ptr @malloc(i64 40)
 141 ; CHECK-NEXT:    ret ptr [[CALL]]
 142 ;
 143   %call = tail call noalias dereferenceable_or_null(80) ptr @malloc(i64 40)
 144   ret ptr %call
 145 }
 146
 147 define noalias ptr @malloc_constant_size3() {
 148 ; CHECK-LABEL: @malloc_constant_size3(
 149 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable(80) dereferenceable_or_null(40) ptr @malloc(i64 40)
 150 ; CHECK-NEXT:    ret ptr [[CALL]]
 151 ;
 152   %call = tail call noalias dereferenceable(80) ptr @malloc(i64 40)
 153   ret ptr %call
 154 }
 155
 156 define noalias ptr @malloc_constant_zero_size() {
 157 ; CHECK-LABEL: @malloc_constant_zero_size(
 158 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @malloc(i64 0)
 159 ; CHECK-NEXT:    ret ptr [[CALL]]
 160 ;
 161   %call = tail call noalias ptr @malloc(i64 0)
 162   ret ptr %call
 163 }
 164
 165 define noalias ptr @realloc_nonconstant_size(ptr %p, i64 %n) {
 166 ; CHECK-LABEL: @realloc_nonconstant_size(
 167 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @realloc(ptr [[P:%.*]], i64 [[N:%.*]])
 168 ; CHECK-NEXT:    ret ptr [[CALL]]
 169 ;
 170   %call = tail call noalias ptr @realloc(ptr %p, i64 %n)
 171   ret ptr %call
 172 }
 173
 174 define noalias ptr @realloc_constant_zero_size(ptr %p) {
 175 ; CHECK-LABEL: @realloc_constant_zero_size(
 176 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @realloc(ptr [[P:%.*]], i64 0)
 177 ; CHECK-NEXT:    ret ptr [[CALL]]
 178 ;
 179   %call = tail call noalias ptr @realloc(ptr %p, i64 0)
 180   ret ptr %call
 181 }
 182
 183 define noalias ptr @realloc_constant_size(ptr %p) {
 184 ; CHECK-LABEL: @realloc_constant_size(
 185 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(40) ptr @realloc(ptr [[P:%.*]], i64 40)
 186 ; CHECK-NEXT:    ret ptr [[CALL]]
 187 ;
 188   %call = tail call noalias ptr @realloc(ptr %p, i64 40)
 189   ret ptr %call
 190 }
 191
 192 define noalias ptr @calloc_nonconstant_size(i64 %n) {
 193 ; CHECK-LABEL: @calloc_nonconstant_size(
 194 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 1, i64 [[N:%.*]])
 195 ; CHECK-NEXT:    ret ptr [[CALL]]
 196 ;
 197   %call = tail call noalias ptr @calloc(i64 1, i64 %n)
 198   ret ptr %call
 199 }
 200
 201 define noalias ptr @calloc_nonconstant_size2(i64 %n) {
 202 ; CHECK-LABEL: @calloc_nonconstant_size2(
 203 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 [[N:%.*]], i64 0)
 204 ; CHECK-NEXT:    ret ptr [[CALL]]
 205 ;
 206   %call = tail call noalias ptr @calloc(i64 %n, i64 0)
 207   ret ptr %call
 208 }
 209
 210 define noalias ptr @calloc_nonconstant_size3(i64 %n) {
 211 ; CHECK-LABEL: @calloc_nonconstant_size3(
 212 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 [[N:%.*]], i64 [[N]])
 213 ; CHECK-NEXT:    ret ptr [[CALL]]
 214 ;
 215   %call = tail call noalias ptr @calloc(i64 %n, i64 %n)
 216   ret ptr %call
 217 }
 218
 219 define noalias ptr @calloc_constant_zero_size() {
 220 ; CHECK-LABEL: @calloc_constant_zero_size(
 221 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 0, i64 0)
 222 ; CHECK-NEXT:    ret ptr [[CALL]]
 223 ;
 224   %call = tail call noalias ptr @calloc(i64 0, i64 0)
 225   ret ptr %call
 226 }
 227
 228 define noalias ptr @calloc_constant_zero_size2(i64 %n) {
 229 ; CHECK-LABEL: @calloc_constant_zero_size2(
 230 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 [[N:%.*]], i64 0)
 231 ; CHECK-NEXT:    ret ptr [[CALL]]
 232 ;
 233   %call = tail call noalias ptr @calloc(i64 %n, i64 0)
 234   ret ptr %call
 235 }
 236
 237
 238 define noalias ptr @calloc_constant_zero_size3(i64 %n) {
 239 ; CHECK-LABEL: @calloc_constant_zero_size3(
 240 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 0, i64 [[N:%.*]])
 241 ; CHECK-NEXT:    ret ptr [[CALL]]
 242 ;
 243   %call = tail call noalias ptr @calloc(i64 0, i64 %n)
 244   ret ptr %call
 245 }
 246
 247 define noalias ptr @calloc_constant_zero_size4(i64 %n) {
 248 ; CHECK-LABEL: @calloc_constant_zero_size4(
 249 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 0, i64 1)
 250 ; CHECK-NEXT:    ret ptr [[CALL]]
 251 ;
 252   %call = tail call noalias ptr @calloc(i64 0, i64 1)
 253   ret ptr %call
 254 }
 255
 256 define noalias ptr @calloc_constant_zero_size5(i64 %n) {
 257 ; CHECK-LABEL: @calloc_constant_zero_size5(
 258 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 1, i64 0)
 259 ; CHECK-NEXT:    ret ptr [[CALL]]
 260 ;
 261   %call = tail call noalias ptr @calloc(i64 1, i64 0)
 262   ret ptr %call
 263 }
 264
 265 define noalias ptr @calloc_constant_size() {
 266 ; CHECK-LABEL: @calloc_constant_size(
 267 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(128) ptr @calloc(i64 16, i64 8)
 268 ; CHECK-NEXT:    ret ptr [[CALL]]
 269 ;
 270   %call = tail call noalias ptr @calloc(i64 16, i64 8)
 271   ret ptr %call
 272 }
 273
 274 define noalias ptr @calloc_constant_size_overflow() {
 275 ; CHECK-LABEL: @calloc_constant_size_overflow(
 276 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @calloc(i64 2000000000000, i64 80000000000)
 277 ; CHECK-NEXT:    ret ptr [[CALL]]
 278 ;
 279   %call = tail call noalias ptr @calloc(i64 2000000000000, i64 80000000000)
 280   ret ptr %call
 281 }
 282
 283 define noalias ptr @op_new_nonconstant_size(i64 %n) {
 284 ; CHECK-LABEL: @op_new_nonconstant_size(
 285 ; CHECK-NEXT:    [[CALL:%.*]] = tail call ptr @_Znam(i64 [[N:%.*]])
 286 ; CHECK-NEXT:    ret ptr [[CALL]]
 287 ;
 288   %call = tail call ptr @_Znam(i64 %n)
 289   ret ptr %call
 290 }
 291
 292 define noalias ptr @op_new_constant_size() {
 293 ; CHECK-LABEL: @op_new_constant_size(
 294 ; CHECK-NEXT:    [[CALL:%.*]] = tail call dereferenceable(40) ptr @_Znam(i64 40)
 295 ; CHECK-NEXT:    ret ptr [[CALL]]
 296 ;
 297   %call = tail call ptr @_Znam(i64 40)
 298   ret ptr %call
 299 }
 300
 301 define noalias ptr @op_new_constant_size2() {
 302 ; CHECK-LABEL: @op_new_constant_size2(
 303 ; CHECK-NEXT:    [[CALL:%.*]] = tail call dereferenceable(40) ptr @_Znwm(i64 40)
 304 ; CHECK-NEXT:    ret ptr [[CALL]]
 305 ;
 306   %call = tail call ptr @_Znwm(i64 40)
 307   ret ptr %call
 308 }
 309
 310 define noalias ptr @op_new_constant_zero_size() {
 311 ; CHECK-LABEL: @op_new_constant_zero_size(
 312 ; CHECK-NEXT:    [[CALL:%.*]] = tail call ptr @_Znam(i64 0)
 313 ; CHECK-NEXT:    ret ptr [[CALL]]
 314 ;
 315   %call = tail call ptr @_Znam(i64 0)
 316   ret ptr %call
 317 }
 318
 319 define noalias ptr @strdup_constant_str() {
 320 ; CHECK-LABEL: @strdup_constant_str(
 321 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(6) ptr @strdup(ptr nonnull @.str)
 322 ; CHECK-NEXT:    ret ptr [[CALL]]
 323 ;
 324   %call = tail call noalias ptr @strdup(ptr @.str)
 325   ret ptr %call
 326 }
 327
 328 define noalias ptr @strdup_notconstant_str(ptr %str) {
 329 ; CHECK-LABEL: @strdup_notconstant_str(
 330 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias ptr @strdup(ptr [[STR:%.*]])
 331 ; CHECK-NEXT:    ret ptr [[CALL]]
 332 ;
 333   %call = tail call noalias ptr @strdup(ptr %str)
 334   ret ptr %call
 335 }
 336
 337 ; OSS-Fuzz #23214
 338 ; https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=23214
 339 define noalias ptr @ossfuzz_23214() {
 340 ; CHECK-LABEL: @ossfuzz_23214(
 341 ; CHECK-NEXT:  bb:
 342 ; CHECK-NEXT:    [[CALL:%.*]] = tail call noalias dereferenceable_or_null(512) ptr @aligned_alloc(i64 -9223372036854775808, i64 512)
 343 ; CHECK-NEXT:    ret ptr [[CALL]]
 344 ;
 345 bb:
 346   %and = and i64 -1, -9223372036854775808
 347   %call = tail call noalias ptr @aligned_alloc(i64 %and, i64 512)
 348   ret ptr %call
 349 }
 350
 351 define noalias ptr @op_new_align() {
 352 ; CHECK-LABEL: @op_new_align(
 353 ; CHECK-NEXT:    [[CALL:%.*]] = tail call align 32 dereferenceable_or_null(32) ptr @_ZnamSt11align_val_t(i64 32, i64 32)
 354 ; CHECK-NEXT:    ret ptr [[CALL]]
 355 ;
 356   %call = tail call ptr @_ZnamSt11align_val_t(i64 32, i64 32)
 357   ret ptr %call
 358 }
 359
 360 define ptr @my_malloc_constant_size() {
 361 ; CHECK-LABEL: @my_malloc_constant_size(
 362 ; CHECK-NEXT:    [[CALL:%.*]] = call dereferenceable_or_null(32) ptr @my_malloc(i64 32)
 363 ; CHECK-NEXT:    ret ptr [[CALL]]
 364 ;
 365   %call = call ptr @my_malloc(i64 32)
 366   ret ptr %call
 367 }
 368
 369 define ptr @my_calloc_constant_size() {
 370 ; CHECK-LABEL: @my_calloc_constant_size(
 371 ; CHECK-NEXT:    [[CALL:%.*]] = call dereferenceable_or_null(128) ptr @my_calloc(i64 32, i64 4)
 372 ; CHECK-NEXT:    ret ptr [[CALL]]
 373 ;
 374   %call = call ptr @my_calloc(i64 32, i64 4)
 375   ret ptr %call
 376 }