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: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 60 // CHECK: call i8* @_Znwm(i64 %[[SIZE]]) 61 62 // CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.frame() 63 // CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]]) 64 // CHECK: call void @_ZdlPv(i8* %[[MEM]]) 65 co_return; 66 } 67 68 struct promise_new_tag {}; 69 70 template<> 71 struct std::experimental::coroutine_traits<void, promise_new_tag> { 72 struct promise_type { 73 void *operator new(unsigned long); 74 void get_return_object() {} 75 suspend_always initial_suspend() { return {}; } 76 suspend_always final_suspend() { return {}; } 77 void return_void() {} 78 }; 79 }; 80 81 // CHECK-LABEL: f1( 82 extern "C" void f1(promise_new_tag ) { 83 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 84 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 85 // CHECK: call i8* @_ZNSt12experimental16coroutine_traitsIJv15promise_new_tagEE12promise_typenwEm(i64 %[[SIZE]]) 86 87 // CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.frame() 88 // CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]]) 89 // CHECK: call void @_ZdlPv(i8* %[[MEM]]) 90 co_return; 91 } 92 93 struct promise_delete_tag {}; 94 95 template<> 96 struct std::experimental::coroutine_traits<void, promise_delete_tag> { 97 struct promise_type { 98 void operator delete(void*); 99 void get_return_object() {} 100 suspend_always initial_suspend() { return {}; } 101 suspend_always final_suspend() { return {}; } 102 void return_void() {} 103 }; 104 }; 105 106 // CHECK-LABEL: f2( 107 extern "C" void f2(promise_delete_tag) { 108 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 109 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 110 // CHECK: call i8* @_Znwm(i64 %[[SIZE]]) 111 112 // CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.frame() 113 // CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]]) 114 // CHECK: call void @_ZNSt12experimental16coroutine_traitsIJv18promise_delete_tagEE12promise_typedlEPv(i8* %[[MEM]]) 115 co_return; 116 } 117 118 struct promise_sized_delete_tag {}; 119 120 template<> 121 struct std::experimental::coroutine_traits<void, promise_sized_delete_tag> { 122 struct promise_type { 123 void operator delete(void*, unsigned long); 124 void get_return_object() {} 125 suspend_always initial_suspend() { return {}; } 126 suspend_always final_suspend() { return {}; } 127 void return_void() {} 128 }; 129 }; 130 131 // CHECK-LABEL: f3( 132 extern "C" void f3(promise_sized_delete_tag) { 133 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 134 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 135 // CHECK: call i8* @_Znwm(i64 %[[SIZE]]) 136 137 // CHECK: %[[FRAME:.+]] = call i8* @llvm.coro.frame() 138 // CHECK: %[[MEM:.+]] = call i8* @llvm.coro.free(token %[[ID]], i8* %[[FRAME]]) 139 // CHECK: %[[SIZE2:.+]] = call i64 @llvm.coro.size.i64() 140 // CHECK: call void @_ZNSt12experimental16coroutine_traitsIJv24promise_sized_delete_tagEE12promise_typedlEPvm(i8* %[[MEM]], i64 %[[SIZE2]]) 141 co_return; 142 } 143 144 struct promise_on_alloc_failure_tag {}; 145 146 template<> 147 struct std::experimental::coroutine_traits<int, promise_on_alloc_failure_tag> { 148 struct promise_type { 149 int get_return_object() { return 0; } 150 suspend_always initial_suspend() { return {}; } 151 suspend_always final_suspend() { return {}; } 152 void return_void() {} 153 static int get_return_object_on_allocation_failure() { return -1; } 154 }; 155 }; 156 157 // CHECK-LABEL: f4( 158 extern "C" int f4(promise_on_alloc_failure_tag) { 159 // CHECK: %[[ID:.+]] = call token @llvm.coro.id(i32 16 160 // CHECK: %[[SIZE:.+]] = call i64 @llvm.coro.size.i64() 161 // CHECK: %[[MEM:.+]] = call i8* @_ZnwmRKSt9nothrow_t(i64 %[[SIZE]], %"struct.std::nothrow_t"* dereferenceable(1) @_ZStL7nothrow) 162 // CHECK: %[[OK:.+]] = icmp ne i8* %[[MEM]], null 163 // CHECK: br i1 %[[OK]], label %[[OKBB:.+]], label %[[ERRBB:.+]] 164 165 // CHECK: [[ERRBB]]: 166 // CHECK: %[[RETVAL:.+]] = call i32 @_ZNSt12experimental16coroutine_traitsIJi28promise_on_alloc_failure_tagEE12promise_type39get_return_object_on_allocation_failureEv( 167 // CHECK: ret i32 %[[RETVAL]] 168 co_return; 169 } 170