test/CodeGen/X86/implicit-null-check.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -verify-machineinstrs -O3 -mtriple=x86_64-apple-macosx -enable-implicit-null-checks < %s | FileCheck %s
   3
   4 define i32 @imp_null_check_load(i32* %x) {
   5 ; CHECK-LABEL: imp_null_check_load:
   6 ; CHECK:       ## %bb.0: ## %entry
   7 ; CHECK-NEXT:  Ltmp0:
   8 ; CHECK-NEXT:    movl (%rdi), %eax ## on-fault: LBB0_1
   9 ; CHECK-NEXT:  ## %bb.2: ## %not_null
  10 ; CHECK-NEXT:    retq
  11 ; CHECK-NEXT:  LBB0_1: ## %is_null
  12 ; CHECK-NEXT:    movl $42, %eax
  13 ; CHECK-NEXT:    retq
  14
  15  entry:
  16   %c = icmp eq i32* %x, null
  17   br i1 %c, label %is_null, label %not_null, !make.implicit !0
  18
  19  is_null:
  20   ret i32 42
  21
  22  not_null:
  23   %t = load i32, i32* %x
  24   ret i32 %t
  25 }
  26
  27 ; TODO: can make implicit
  28 define i32 @imp_null_check_unordered_load(i32* %x) {
  29 ; CHECK-LABEL: imp_null_check_unordered_load:
  30 ; CHECK:       ## %bb.0: ## %entry
  31 ; CHECK-NEXT:  Ltmp1:
  32 ; CHECK-NEXT:    movl (%rdi), %eax ## on-fault: LBB1_1
  33 ; CHECK-NEXT:  ## %bb.2: ## %not_null
  34 ; CHECK-NEXT:    retq
  35 ; CHECK-NEXT:  LBB1_1: ## %is_null
  36 ; CHECK-NEXT:    movl $42, %eax
  37 ; CHECK-NEXT:    retq
  38
  39  entry:
  40   %c = icmp eq i32* %x, null
  41   br i1 %c, label %is_null, label %not_null, !make.implicit !0
  42
  43  is_null:
  44   ret i32 42
  45
  46  not_null:
  47   %t = load atomic i32, i32* %x unordered, align 4
  48   ret i32 %t
  49 }
  50
  51 ;; Probably could be implicit, but we're conservative for now
  52 define i32 @imp_null_check_seq_cst_load(i32* %x) {
  53 ; CHECK-LABEL: imp_null_check_seq_cst_load:
  54 ; CHECK:       ## %bb.0: ## %entry
  55 ; CHECK-NEXT:    testq %rdi, %rdi
  56 ; CHECK-NEXT:    je LBB2_1
  57 ; CHECK-NEXT:  ## %bb.2: ## %not_null
  58 ; CHECK-NEXT:    movl (%rdi), %eax
  59 ; CHECK-NEXT:    retq
  60 ; CHECK-NEXT:  LBB2_1: ## %is_null
  61 ; CHECK-NEXT:    movl $42, %eax
  62 ; CHECK-NEXT:    retq
  63
  64  entry:
  65   %c = icmp eq i32* %x, null
  66   br i1 %c, label %is_null, label %not_null, !make.implicit !0
  67
  68  is_null:
  69   ret i32 42
  70
  71  not_null:
  72   %t = load atomic i32, i32* %x seq_cst, align 4
  73   ret i32 %t
  74 }
  75
  76 ;; Might be memory mapped IO, so can't rely on fault behavior
  77 define i32 @imp_null_check_volatile_load(i32* %x) {
  78 ; CHECK-LABEL: imp_null_check_volatile_load:
  79 ; CHECK:       ## %bb.0: ## %entry
  80 ; CHECK-NEXT:    testq %rdi, %rdi
  81 ; CHECK-NEXT:    je LBB3_1
  82 ; CHECK-NEXT:  ## %bb.2: ## %not_null
  83 ; CHECK-NEXT:    movl (%rdi), %eax
  84 ; CHECK-NEXT:    retq
  85 ; CHECK-NEXT:  LBB3_1: ## %is_null
  86 ; CHECK-NEXT:    movl $42, %eax
  87 ; CHECK-NEXT:    retq
  88
  89  entry:
  90   %c = icmp eq i32* %x, null
  91   br i1 %c, label %is_null, label %not_null, !make.implicit !0
  92
  93  is_null:
  94   ret i32 42
  95
  96  not_null:
  97   %t = load volatile i32, i32* %x, align 4
  98   ret i32 %t
  99 }
 100
 101
 102 define i8 @imp_null_check_load_i8(i8* %x) {
 103 ; CHECK-LABEL: imp_null_check_load_i8:
 104 ; CHECK:       ## %bb.0: ## %entry
 105 ; CHECK-NEXT:  Ltmp2:
 106 ; CHECK-NEXT:    movb (%rdi), %al ## on-fault: LBB4_1
 107 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 108 ; CHECK-NEXT:    retq
 109 ; CHECK-NEXT:  LBB4_1: ## %is_null
 110 ; CHECK-NEXT:    movb $42, %al
 111 ; CHECK-NEXT:    retq
 112
 113  entry:
 114   %c = icmp eq i8* %x, null
 115   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 116
 117  is_null:
 118   ret i8 42
 119
 120  not_null:
 121   %t = load i8, i8* %x
 122   ret i8 %t
 123 }
 124
 125 define i256 @imp_null_check_load_i256(i256* %x) {
 126 ; CHECK-LABEL: imp_null_check_load_i256:
 127 ; CHECK:       ## %bb.0: ## %entry
 128 ; CHECK-NEXT:    movq %rdi, %rax
 129 ; CHECK-NEXT:  Ltmp3:
 130 ; CHECK-NEXT:    movq (%rsi), %rcx ## on-fault: LBB5_1
 131 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 132 ; CHECK-NEXT:    movq 8(%rsi), %rdx
 133 ; CHECK-NEXT:    movq 16(%rsi), %rdi
 134 ; CHECK-NEXT:    movq 24(%rsi), %rsi
 135 ; CHECK-NEXT:    movq %rsi, 24(%rax)
 136 ; CHECK-NEXT:    movq %rdi, 16(%rax)
 137 ; CHECK-NEXT:    movq %rdx, 8(%rax)
 138 ; CHECK-NEXT:    movq %rcx, (%rax)
 139 ; CHECK-NEXT:    retq
 140 ; CHECK-NEXT:  LBB5_1: ## %is_null
 141 ; CHECK-NEXT:    movq $0, 24(%rax)
 142 ; CHECK-NEXT:    movq $0, 16(%rax)
 143 ; CHECK-NEXT:    movq $0, 8(%rax)
 144 ; CHECK-NEXT:    movq $42, (%rax)
 145 ; CHECK-NEXT:    retq
 146
 147  entry:
 148   %c = icmp eq i256* %x, null
 149   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 150
 151  is_null:
 152   ret i256 42
 153
 154  not_null:
 155   %t = load i256, i256* %x
 156   ret i256 %t
 157 }
 158
 159
 160
 161 define i32 @imp_null_check_gep_load(i32* %x) {
 162 ; CHECK-LABEL: imp_null_check_gep_load:
 163 ; CHECK:       ## %bb.0: ## %entry
 164 ; CHECK-NEXT:  Ltmp4:
 165 ; CHECK-NEXT:    movl 128(%rdi), %eax ## on-fault: LBB6_1
 166 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 167 ; CHECK-NEXT:    retq
 168 ; CHECK-NEXT:  LBB6_1: ## %is_null
 169 ; CHECK-NEXT:    movl $42, %eax
 170 ; CHECK-NEXT:    retq
 171
 172  entry:
 173   %c = icmp eq i32* %x, null
 174   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 175
 176  is_null:
 177   ret i32 42
 178
 179  not_null:
 180   %x.gep = getelementptr i32, i32* %x, i32 32
 181   %t = load i32, i32* %x.gep
 182   ret i32 %t
 183 }
 184
 185 define i32 @imp_null_check_add_result(i32* %x, i32 %p) {
 186 ; CHECK-LABEL: imp_null_check_add_result:
 187 ; CHECK:       ## %bb.0: ## %entry
 188 ; CHECK-NEXT:  Ltmp5:
 189 ; CHECK-NEXT:    addl (%rdi), %esi ## on-fault: LBB7_1
 190 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 191 ; CHECK-NEXT:    movl %esi, %eax
 192 ; CHECK-NEXT:    retq
 193 ; CHECK-NEXT:  LBB7_1: ## %is_null
 194 ; CHECK-NEXT:    movl $42, %eax
 195 ; CHECK-NEXT:    retq
 196
 197  entry:
 198   %c = icmp eq i32* %x, null
 199   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 200
 201  is_null:
 202   ret i32 42
 203
 204  not_null:
 205   %t = load i32, i32* %x
 206   %p1 = add i32 %t, %p
 207   ret i32 %p1
 208 }
 209
 210 define i32 @imp_null_check_sub_result(i32* %x, i32 %p) {
 211 ; CHECK-LABEL: imp_null_check_sub_result:
 212 ; CHECK:       ## %bb.0: ## %entry
 213 ; CHECK-NEXT:  Ltmp6:
 214 ; CHECK-NEXT:    movl (%rdi), %eax ## on-fault: LBB8_1
 215 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 216 ; CHECK-NEXT:    subl %esi, %eax
 217 ; CHECK-NEXT:    retq
 218 ; CHECK-NEXT:  LBB8_1: ## %is_null
 219 ; CHECK-NEXT:    movl $42, %eax
 220 ; CHECK-NEXT:    retq
 221
 222  entry:
 223   %c = icmp eq i32* %x, null
 224   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 225
 226  is_null:
 227   ret i32 42
 228
 229  not_null:
 230   %t = load i32, i32* %x
 231   %p1 = sub i32 %t, %p
 232   ret i32 %p1
 233 }
 234
 235 define i32 @imp_null_check_mul_result(i32* %x, i32 %p) {
 236 ; CHECK-LABEL: imp_null_check_mul_result:
 237 ; CHECK:       ## %bb.0: ## %entry
 238 ; CHECK-NEXT:  Ltmp7:
 239 ; CHECK-NEXT:    imull (%rdi), %esi ## on-fault: LBB9_1
 240 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 241 ; CHECK-NEXT:    movl %esi, %eax
 242 ; CHECK-NEXT:    retq
 243 ; CHECK-NEXT:  LBB9_1: ## %is_null
 244 ; CHECK-NEXT:    movl $42, %eax
 245 ; CHECK-NEXT:    retq
 246
 247  entry:
 248   %c = icmp eq i32* %x, null
 249   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 250
 251  is_null:
 252   ret i32 42
 253
 254  not_null:
 255   %t = load i32, i32* %x
 256   %p1 = mul i32 %t, %p
 257   ret i32 %p1
 258 }
 259
 260 define i32 @imp_null_check_udiv_result(i32* %x, i32 %p) {
 261 ; CHECK-LABEL: imp_null_check_udiv_result:
 262 ; CHECK:       ## %bb.0: ## %entry
 263 ; CHECK-NEXT:  Ltmp8:
 264 ; CHECK-NEXT:    movl (%rdi), %eax ## on-fault: LBB10_1
 265 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 266 ; CHECK-NEXT:    xorl %edx, %edx
 267 ; CHECK-NEXT:    divl %esi
 268 ; CHECK-NEXT:    retq
 269 ; CHECK-NEXT:  LBB10_1: ## %is_null
 270 ; CHECK-NEXT:    movl $42, %eax
 271 ; CHECK-NEXT:    retq
 272
 273  entry:
 274   %c = icmp eq i32* %x, null
 275   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 276
 277  is_null:
 278   ret i32 42
 279
 280  not_null:
 281   %t = load i32, i32* %x
 282   %p1 = udiv i32 %t, %p
 283   ret i32 %p1
 284 }
 285
 286 define i32 @imp_null_check_shl_result(i32* %x, i32 %p) {
 287 ; CHECK-LABEL: imp_null_check_shl_result:
 288 ; CHECK:       ## %bb.0: ## %entry
 289 ; CHECK-NEXT:  Ltmp9:
 290 ; CHECK-NEXT:    movl (%rdi), %eax ## on-fault: LBB11_1
 291 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 292 ; CHECK-NEXT:    movl %esi, %ecx
 293 ; CHECK-NEXT:    shll %cl, %eax
 294 ; CHECK-NEXT:    retq
 295 ; CHECK-NEXT:  LBB11_1: ## %is_null
 296 ; CHECK-NEXT:    movl $42, %eax
 297 ; CHECK-NEXT:    retq
 298
 299  entry:
 300   %c = icmp eq i32* %x, null
 301   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 302
 303  is_null:
 304   ret i32 42
 305
 306  not_null:
 307   %t = load i32, i32* %x
 308   %p1 = shl i32 %t, %p
 309   ret i32 %p1
 310 }
 311
 312 define i32 @imp_null_check_lshr_result(i32* %x, i32 %p) {
 313 ; CHECK-LABEL: imp_null_check_lshr_result:
 314 ; CHECK:       ## %bb.0: ## %entry
 315 ; CHECK-NEXT:  Ltmp10:
 316 ; CHECK-NEXT:    movl (%rdi), %eax ## on-fault: LBB12_1
 317 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 318 ; CHECK-NEXT:    movl %esi, %ecx
 319 ; CHECK-NEXT:    shrl %cl, %eax
 320 ; CHECK-NEXT:    retq
 321 ; CHECK-NEXT:  LBB12_1: ## %is_null
 322 ; CHECK-NEXT:    movl $42, %eax
 323 ; CHECK-NEXT:    retq
 324
 325  entry:
 326   %c = icmp eq i32* %x, null
 327   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 328
 329  is_null:
 330   ret i32 42
 331
 332  not_null:
 333   %t = load i32, i32* %x
 334   %p1 = lshr i32 %t, %p
 335   ret i32 %p1
 336 }
 337
 338
 339
 340
 341 define i32 @imp_null_check_hoist_over_unrelated_load(i32* %x, i32* %y, i32* %z) {
 342 ; CHECK-LABEL: imp_null_check_hoist_over_unrelated_load:
 343 ; CHECK:       ## %bb.0: ## %entry
 344 ; CHECK-NEXT:  Ltmp11:
 345 ; CHECK-NEXT:    movl (%rdi), %eax ## on-fault: LBB13_1
 346 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 347 ; CHECK-NEXT:    movl (%rsi), %ecx
 348 ; CHECK-NEXT:    movl %ecx, (%rdx)
 349 ; CHECK-NEXT:    retq
 350 ; CHECK-NEXT:  LBB13_1: ## %is_null
 351 ; CHECK-NEXT:    movl $42, %eax
 352 ; CHECK-NEXT:    retq
 353
 354  entry:
 355   %c = icmp eq i32* %x, null
 356   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 357
 358  is_null:
 359   ret i32 42
 360
 361  not_null:
 362   %t0 = load i32, i32* %y
 363   %t1 = load i32, i32* %x
 364   store i32 %t0, i32* %z
 365   ret i32 %t1
 366 }
 367
 368 define i32 @imp_null_check_via_mem_comparision(i32* %x, i32 %val) {
 369 ; CHECK-LABEL: imp_null_check_via_mem_comparision:
 370 ; CHECK:       ## %bb.0: ## %entry
 371 ; CHECK-NEXT:  Ltmp12:
 372 ; CHECK-NEXT:    cmpl %esi, 4(%rdi) ## on-fault: LBB14_3
 373 ; CHECK-NEXT:  ## %bb.1: ## %not_null
 374 ; CHECK-NEXT:    jge LBB14_2
 375 ; CHECK-NEXT:  ## %bb.4: ## %ret_100
 376 ; CHECK-NEXT:    movl $100, %eax
 377 ; CHECK-NEXT:    retq
 378 ; CHECK-NEXT:  LBB14_3: ## %is_null
 379 ; CHECK-NEXT:    movl $42, %eax
 380 ; CHECK-NEXT:    retq
 381 ; CHECK-NEXT:  LBB14_2: ## %ret_200
 382 ; CHECK-NEXT:    movl $200, %eax
 383 ; CHECK-NEXT:    retq
 384
 385  entry:
 386   %c = icmp eq i32* %x, null
 387   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 388
 389  is_null:
 390   ret i32 42
 391
 392  not_null:
 393   %x.loc = getelementptr i32, i32* %x, i32 1
 394   %t = load i32, i32* %x.loc
 395   %m = icmp slt i32 %t, %val
 396   br i1 %m, label %ret_100, label %ret_200
 397
 398  ret_100:
 399   ret i32 100
 400
 401  ret_200:
 402   ret i32 200
 403 }
 404
 405 define i32 @imp_null_check_gep_load_with_use_dep(i32* %x, i32 %a) {
 406 ; CHECK-LABEL: imp_null_check_gep_load_with_use_dep:
 407 ; CHECK:       ## %bb.0: ## %entry
 408 ; CHECK-NEXT:    ## kill: def $esi killed $esi def $rsi
 409 ; CHECK-NEXT:  Ltmp13:
 410 ; CHECK-NEXT:    movl (%rdi), %eax ## on-fault: LBB15_1
 411 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 412 ; CHECK-NEXT:    addl %edi, %esi
 413 ; CHECK-NEXT:    leal 4(%rax,%rsi), %eax
 414 ; CHECK-NEXT:    retq
 415 ; CHECK-NEXT:  LBB15_1: ## %is_null
 416 ; CHECK-NEXT:    movl $42, %eax
 417 ; CHECK-NEXT:    retq
 418
 419  entry:
 420   %c = icmp eq i32* %x, null
 421   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 422
 423  is_null:
 424   ret i32 42
 425
 426  not_null:
 427   %x.loc = getelementptr i32, i32* %x, i32 1
 428   %y = ptrtoint i32* %x.loc to i32
 429   %b = add i32 %a, %y
 430   %t = load i32, i32* %x
 431   %z = add i32 %t, %b
 432   ret i32 %z
 433 }
 434
 435 ;; TODO: We could handle this case as we can lift the fence into the
 436 ;; previous block before the conditional without changing behavior.
 437 define i32 @imp_null_check_load_fence1(i32* %x) {
 438 ; CHECK-LABEL: imp_null_check_load_fence1:
 439 ; CHECK:       ## %bb.0: ## %entry
 440 ; CHECK-NEXT:    testq %rdi, %rdi
 441 ; CHECK-NEXT:    je LBB16_1
 442 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 443 ; CHECK-NEXT:    ##MEMBARRIER
 444 ; CHECK-NEXT:    movl (%rdi), %eax
 445 ; CHECK-NEXT:    retq
 446 ; CHECK-NEXT:  LBB16_1: ## %is_null
 447 ; CHECK-NEXT:    movl $42, %eax
 448 ; CHECK-NEXT:    retq
 449
 450 entry:
 451   %c = icmp eq i32* %x, null
 452   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 453
 454 is_null:
 455   ret i32 42
 456
 457 not_null:
 458   fence acquire
 459   %t = load i32, i32* %x
 460   ret i32 %t
 461 }
 462
 463 ;; TODO: We could handle this case as we can lift the fence into the
 464 ;; previous block before the conditional without changing behavior.
 465 define i32 @imp_null_check_load_fence2(i32* %x) {
 466 ; CHECK-LABEL: imp_null_check_load_fence2:
 467 ; CHECK:       ## %bb.0: ## %entry
 468 ; CHECK-NEXT:    testq %rdi, %rdi
 469 ; CHECK-NEXT:    je LBB17_1
 470 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 471 ; CHECK-NEXT:    mfence
 472 ; CHECK-NEXT:    movl (%rdi), %eax
 473 ; CHECK-NEXT:    retq
 474 ; CHECK-NEXT:  LBB17_1: ## %is_null
 475 ; CHECK-NEXT:    movl $42, %eax
 476 ; CHECK-NEXT:    retq
 477
 478 entry:
 479   %c = icmp eq i32* %x, null
 480   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 481
 482 is_null:
 483   ret i32 42
 484
 485 not_null:
 486   fence seq_cst
 487   %t = load i32, i32* %x
 488   ret i32 %t
 489 }
 490
 491 define void @imp_null_check_store(i32* %x) {
 492 ; CHECK-LABEL: imp_null_check_store:
 493 ; CHECK:       ## %bb.0: ## %entry
 494 ; CHECK-NEXT:  Ltmp14:
 495 ; CHECK-NEXT:    movl $1, (%rdi) ## on-fault: LBB18_1
 496 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 497 ; CHECK-NEXT:    retq
 498 ; CHECK-NEXT:  LBB18_1: ## %is_null
 499 ; CHECK-NEXT:    retq
 500
 501  entry:
 502   %c = icmp eq i32* %x, null
 503   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 504
 505  is_null:
 506   ret void
 507
 508  not_null:
 509   store i32 1, i32* %x
 510   ret void
 511 }
 512
 513 ;; TODO: can be implicit
 514 define void @imp_null_check_unordered_store(i32* %x) {
 515 ; CHECK-LABEL: imp_null_check_unordered_store:
 516 ; CHECK:       ## %bb.0: ## %entry
 517 ; CHECK-NEXT:  Ltmp15:
 518 ; CHECK-NEXT:    movl $1, (%rdi) ## on-fault: LBB19_1
 519 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 520 ; CHECK-NEXT:    retq
 521 ; CHECK-NEXT:  LBB19_1: ## %is_null
 522 ; CHECK-NEXT:    retq
 523
 524  entry:
 525   %c = icmp eq i32* %x, null
 526   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 527
 528  is_null:
 529   ret void
 530
 531  not_null:
 532   store atomic i32 1, i32* %x unordered, align 4
 533   ret void
 534 }
 535
 536 define i32 @imp_null_check_neg_gep_load(i32* %x) {
 537 ; CHECK-LABEL: imp_null_check_neg_gep_load:
 538 ; CHECK:       ## %bb.0: ## %entry
 539 ; CHECK-NEXT:  Ltmp16:
 540 ; CHECK-NEXT:    movl -128(%rdi), %eax ## on-fault: LBB20_1
 541 ; CHECK-NEXT:  ## %bb.2: ## %not_null
 542 ; CHECK-NEXT:    retq
 543 ; CHECK-NEXT:  LBB20_1: ## %is_null
 544 ; CHECK-NEXT:    movl $42, %eax
 545 ; CHECK-NEXT:    retq
 546
 547  entry:
 548   %c = icmp eq i32* %x, null
 549   br i1 %c, label %is_null, label %not_null, !make.implicit !0
 550
 551  is_null:
 552   ret i32 42
 553
 554  not_null:
 555   %x.gep = getelementptr i32, i32* %x, i32 -32
 556   %t = load i32, i32* %x.gep
 557   ret i32 %t
 558 }
 559
 560 !0 = !{}