99 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 52; RUN: opt < %s -passes='inferattrs,alloc-token<mode=increment>' -S | FileCheck %s3 4target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"5 6declare ptr @malloc(i64)7declare ptr @calloc(i64, i64)8declare ptr @realloc(ptr, i64)9declare ptr @_Znwm(i64)10declare ptr @_Znam(i64)11declare void @free(ptr)12declare void @_ZdlPv(ptr)13declare i32 @foobar(i64)14 15; Test basic allocation call rewriting16define ptr @test_basic_rewriting() sanitize_alloc_token {17; CHECK-LABEL: define ptr @test_basic_rewriting(18; CHECK-SAME: ) #[[ATTR5:[0-9]+]] {19; CHECK-NEXT: [[ENTRY:.*:]]20; CHECK-NEXT: [[TMP0:%.*]] = call ptr @__alloc_token_malloc(i64 64, i64 0)21; CHECK-NEXT: [[TMP1:%.*]] = call ptr @__alloc_token_calloc(i64 8, i64 8, i64 1)22; CHECK-NEXT: [[TMP2:%.*]] = call ptr @__alloc_token_realloc(ptr [[TMP0]], i64 128, i64 2)23; CHECK-NEXT: ret ptr [[TMP2]]24;25entry:26 %ptr1 = call ptr @malloc(i64 64)27 %ptr2 = call ptr @calloc(i64 8, i64 8)28 %ptr3 = call ptr @realloc(ptr %ptr1, i64 128)29 ret ptr %ptr330}31 32; Test C++ operator rewriting33define ptr @test_cpp_operators() sanitize_alloc_token {34; CHECK-LABEL: define ptr @test_cpp_operators(35; CHECK-SAME: ) #[[ATTR5]] {36; CHECK-NEXT: [[ENTRY:.*:]]37; CHECK-NEXT: [[TMP0:%.*]] = call ptr @__alloc_token__Znwm(i64 32, i64 3)38; CHECK-NEXT: [[TMP1:%.*]] = call ptr @__alloc_token__Znam(i64 64, i64 4)39; CHECK-NEXT: ret ptr [[TMP0]]40;41entry:42 %ptr1 = call ptr @_Znwm(i64 32)43 %ptr2 = call ptr @_Znam(i64 64)44 ret ptr %ptr145}46 47; Functions without sanitize_alloc_token do not get instrumented48define ptr @without_attribute() {49; CHECK-LABEL: define ptr @without_attribute() {50; CHECK-NEXT: [[ENTRY:.*:]]51; CHECK-NEXT: [[PTR:%.*]] = call ptr @malloc(i64 16)52; CHECK-NEXT: ret ptr [[PTR]]53;54entry:55 %ptr = call ptr @malloc(i64 16)56 ret ptr %ptr57}58 59; Test that free/delete are untouched60define void @test_free_untouched(ptr %ptr) sanitize_alloc_token {61; CHECK-LABEL: define void @test_free_untouched(62; CHECK-SAME: ptr [[PTR:%.*]]) #[[ATTR5]] {63; CHECK-NEXT: [[ENTRY:.*:]]64; CHECK-NEXT: call void @free(ptr [[PTR]])65; CHECK-NEXT: call void @_ZdlPv(ptr [[PTR]])66; CHECK-NEXT: ret void67;68entry:69 call void @free(ptr %ptr)70 call void @_ZdlPv(ptr %ptr)71 ret void72}73 74; Non-allocation functions are untouched75define i32 @no_allocations(i32 %x) sanitize_alloc_token {76; CHECK-LABEL: define i32 @no_allocations(77; CHECK-SAME: i32 [[X:%.*]]) #[[ATTR5]] {78; CHECK-NEXT: [[ENTRY:.*:]]79; CHECK-NEXT: [[RESULT:%.*]] = call i32 @foobar(i64 42)80; CHECK-NEXT: ret i32 [[RESULT]]81;82entry:83 %result = call i32 @foobar(i64 42)84 ret i32 %result85}86 87; Test that tail calls are preserved88define ptr @test_tail_call_preserved() sanitize_alloc_token {89; CHECK-LABEL: define ptr @test_tail_call_preserved(90; CHECK-SAME: ) #[[ATTR5]] {91; CHECK-NEXT: [[ENTRY:.*:]]92; CHECK-NEXT: [[TMP0:%.*]] = tail call ptr @__alloc_token_malloc(i64 42, i64 5)93; CHECK-NEXT: ret ptr [[TMP0]]94;95entry:96 %result = tail call ptr @malloc(i64 42)97 ret ptr %result98}99