1 // RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -fcoroutines-ts -std=c++14 \ 2 // RUN: -Wno-coroutine-missing-unhandled-exception -emit-llvm %s -o - -disable-llvm-passes \ 3 // RUN: | FileCheck %s 4 5 namespace std { 6 namespace experimental { 7 template <typename... T> 8 struct coroutine_traits; // expected-note {{declared here}} 9 10 template <class Promise = void> 11 struct coroutine_handle { 12 coroutine_handle() = default; 13 static coroutine_handle from_address(void *) { return {}; } 14 }; 15 16 template <> 17 struct coroutine_handle<void> { 18 static coroutine_handle from_address(void *) { return {}; } 19 coroutine_handle() = default; 20 template <class PromiseType> 21 coroutine_handle(coroutine_handle<PromiseType>) {} 22 }; 23 24 } // end namespace experimental 25 26 struct nothrow_t {}; 27 constexpr nothrow_t nothrow = {}; 28 29 } // end namespace std 30 31 // Required when get_return_object_on_allocation_failure() is defined by 32 // the promise. 33 using SizeT = decltype(sizeof(int)); 34 void* operator new(SizeT __sz, const std::nothrow_t&) noexcept; 35 void operator delete(void* __p, const std::nothrow_t&) noexcept; 36 37 38 struct suspend_always { 39 bool await_ready() { return false; } 40 void await_suspend(std::experimental::coroutine_handle<>) {} 41 void await_resume() {} 42 }; 43 44 struct global_new_delete_tag {}; 45 46 template<> 47 struct std::experimental::coroutine_traits<void, global_new_delete_tag> { 48 struct promise_type { 49 void get_return_object() {} 50 suspend_always initial_suspend() { return {}; } 51 suspend_always final_suspend() { return {}; } 52 void return_void() {} 53 }; 54 }; 55 56 // CHECK-LABEL: f0( 57 extern "C" void f0(global_new_delete_tag) { 58 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 59 // CHECK: %[[NeedAlloc:.+]] = call i1 @llvm.coro.alloc(token %[[ID]]) 60 // CHECK: br i1 %[[NeedAlloc]], label %[[AllocBB:.+]], label %[[InitBB:.+]] 61 62 // CHECK: [[AllocBB]]: 63 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 64 // CHECK: %[[MEM:.+]] = call i8* @_Znwm(i64 %[[SIZE]]) 65 // CHECK: br label %[[InitBB]] 66 67 // CHECK: [[InitBB]]: 68 // CHECK: %[[PHI:.+]] = phi i8* [ null, %{{.+}} ], [ %call, %[[AllocBB]] ] 69 // CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.begin(token %[[ID]], i8* %[[PHI]]) 70 71 // CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]]) 72 // CHECK: %[[NeedDealloc:.+]] = icmp ne i8* %[[MEM]], null 73 // CHECK: br i1 %[[NeedDealloc]], label %[[FreeBB:.+]], label %[[Afterwards:.+]] 74 75 // CHECK: [[FreeBB]]: 76 // CHECK: call void @_ZdlPv(i8* %[[MEM]]) 77 // CHECK: br label %[[Afterwards]] 78 79 // CHECK: [[Afterwards]]: 80 // CHECK: ret void 81 co_return; 82 } 83 84 struct promise_new_tag {}; 85 86 template<> 87 struct std::experimental::coroutine_traits<void, promise_new_tag> { 88 struct promise_type { 89 void *operator new(unsigned long); 90 void get_return_object() {} 91 suspend_always initial_suspend() { return {}; } 92 suspend_always final_suspend() { return {}; } 93 void return_void() {} 94 }; 95 }; 96 97 // CHECK-LABEL: f1( 98 extern "C" void f1(promise_new_tag ) { 99 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 100 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 101 // CHECK: call i8* @_ZNSt12experimental16coroutine_traitsIJv15promise_new_tagEE12promise_typenwEm(i64 %[[SIZE]]) 102 103 // CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.begin( 104 // CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]]) 105 // CHECK: call void @_ZdlPv(i8* %[[MEM]]) 106 co_return; 107 } 108 109 struct promise_delete_tag {}; 110 111 template<> 112 struct std::experimental::coroutine_traits<void, promise_delete_tag> { 113 struct promise_type { 114 void operator delete(void*); 115 void get_return_object() {} 116 suspend_always initial_suspend() { return {}; } 117 suspend_always final_suspend() { return {}; } 118 void return_void() {} 119 }; 120 }; 121 122 // CHECK-LABEL: f2( 123 extern "C" void f2(promise_delete_tag) { 124 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 125 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 126 // CHECK: call i8* @_Znwm(i64 %[[SIZE]]) 127 128 // CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.begin( 129 // CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]]) 130 // CHECK: call void @_ZNSt12experimental16coroutine_traitsIJv18promise_delete_tagEE12promise_typedlEPv(i8* %[[MEM]]) 131 co_return; 132 } 133 134 struct promise_sized_delete_tag {}; 135 136 template<> 137 struct std::experimental::coroutine_traits<void, promise_sized_delete_tag> { 138 struct promise_type { 139 void operator delete(void*, unsigned long); 140 void get_return_object() {} 141 suspend_always initial_suspend() { return {}; } 142 suspend_always final_suspend() { return {}; } 143 void return_void() {} 144 }; 145 }; 146 147 // CHECK-LABEL: f3( 148 extern "C" void f3(promise_sized_delete_tag) { 149 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 150 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 151 // CHECK: call i8* @_Znwm(i64 %[[SIZE]]) 152 153 // CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.begin( 154 // CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]]) 155 // CHECK: %[[SIZE2:.+]] = call i64 @llvm.coro.size.i64() 156 // CHECK: call void @_ZNSt12experimental16coroutine_traitsIJv24promise_sized_delete_tagEE12promise_typedlEPvm(i8* %[[MEM]], i64 %[[SIZE2]]) 157 co_return; 158 } 159 160 struct promise_on_alloc_failure_tag {}; 161 162 template<> 163 struct std::experimental::coroutine_traits<int, promise_on_alloc_failure_tag> { 164 struct promise_type { 165 int get_return_object() { return 0; } 166 suspend_always initial_suspend() { return {}; } 167 suspend_always final_suspend() { return {}; } 168 void return_void() {} 169 static int get_return_object_on_allocation_failure() { return -1; } 170 }; 171 }; 172 173 // CHECK-LABEL: f4( 174 extern "C" int f4(promise_on_alloc_failure_tag) { 175 // CHECK: %[[RetVal:.+]] = alloca i32 176 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 177 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 178 // CHECK: %[[MEM:.+]] = call i8* @_ZnwmRKSt9nothrow_t(i64 %[[SIZE]], %"struct.std::nothrow_t"* dereferenceable(1) @_ZStL7nothrow) 179 // CHECK: %[[OK:.+]] = icmp ne i8* %[[MEM]], null 180 // CHECK: br i1 %[[OK]], label %[[OKBB:.+]], label %[[ERRBB:.+]] 181 182 // CHECK: [[ERRBB]]: 183 // CHECK: %[[FailRet:.+]] = call i32 @_ZNSt12experimental16coroutine_traitsIJi28promise_on_alloc_failure_tagEE12promise_type39get_return_object_on_allocation_failureEv( 184 // CHECK: store i32 %[[FailRet]], i32* %[[RetVal]] 185 // CHECK: br label %[[RetBB:.+]] 186 187 // CHECK: [[OKBB]]: 188 // CHECK: %[[OkRet:.+]] = call i32 @_ZNSt12experimental16coroutine_traitsIJi28promise_on_alloc_failure_tagEE12promise_type17get_return_objectEv( 189 // CHECK: store i32 %[[OkRet]], i32* %[[RetVal]] 190 191 // CHECK: [[RetBB]]: 192 // CHECK: %[[LoadRet:.+]] = load i32, i32* %[[RetVal]], align 4 193 // CHECK: ret i32 %[[LoadRet]] 194 co_return; 195 } 196