| 1 | // Verify that coroutine promise and allocated memory are freed up on exception. |
|---|
| 2 | // RUN: %clang_cc1 -std=c++1z -fcoroutines-ts -triple=x86_64-unknown-linux-gnu -emit-llvm -o - %s -fexceptions -fcxx-exceptions -disable-llvm-passes | FileCheck %s |
|---|
| 3 | |
|---|
| 4 | namespace std::experimental { |
|---|
| 5 | template <typename... T> struct coroutine_traits; |
|---|
| 6 | |
|---|
| 7 | template <class Promise = void> struct coroutine_handle { |
|---|
| 8 | coroutine_handle() = default; |
|---|
| 9 | static coroutine_handle from_address(void *) noexcept; |
|---|
| 10 | }; |
|---|
| 11 | template <> struct coroutine_handle<void> { |
|---|
| 12 | static coroutine_handle from_address(void *) noexcept; |
|---|
| 13 | coroutine_handle() = default; |
|---|
| 14 | template <class PromiseType> |
|---|
| 15 | coroutine_handle(coroutine_handle<PromiseType>) noexcept; |
|---|
| 16 | }; |
|---|
| 17 | } |
|---|
| 18 | |
|---|
| 19 | struct suspend_always { |
|---|
| 20 | bool await_ready() noexcept; |
|---|
| 21 | void await_suspend(std::experimental::coroutine_handle<>) noexcept; |
|---|
| 22 | void await_resume() noexcept; |
|---|
| 23 | }; |
|---|
| 24 | |
|---|
| 25 | template <> struct std::experimental::coroutine_traits<void> { |
|---|
| 26 | struct promise_type { |
|---|
| 27 | void get_return_object() noexcept; |
|---|
| 28 | suspend_always initial_suspend() noexcept; |
|---|
| 29 | suspend_always final_suspend() noexcept; |
|---|
| 30 | void return_void() noexcept; |
|---|
| 31 | promise_type(); |
|---|
| 32 | ~promise_type(); |
|---|
| 33 | void unhandled_exception() noexcept; |
|---|
| 34 | }; |
|---|
| 35 | }; |
|---|
| 36 | |
|---|
| 37 | struct Cleanup { ~Cleanup(); }; |
|---|
| 38 | void may_throw(); |
|---|
| 39 | |
|---|
| 40 | // CHECK-LABEL: define void @_Z1fv( |
|---|
| 41 | void f() { |
|---|
| 42 | // CHECK: call i8* @_Znwm(i64 |
|---|
| 43 | |
|---|
| 44 | // If promise constructor throws, check that we free the memory. |
|---|
| 45 | |
|---|
| 46 | // CHECK: invoke void @_ZNSt12experimental16coroutine_traitsIJvEE12promise_typeC1Ev( |
|---|
| 47 | // CHECK-NEXT: to label %{{.+}} unwind label %[[DeallocPad:.+]] |
|---|
| 48 | |
|---|
| 49 | // CHECK: [[DeallocPad]]: |
|---|
| 50 | // CHECK-NEXT: landingpad |
|---|
| 51 | // CHECK-NEXT: cleanup |
|---|
| 52 | // CHECK: br label %[[Dealloc:.+]] |
|---|
| 53 | |
|---|
| 54 | Cleanup cleanup; |
|---|
| 55 | may_throw(); |
|---|
| 56 | |
|---|
| 57 | // if may_throw throws, check that we destroy the promise and free the memory. |
|---|
| 58 | |
|---|
| 59 | // CHECK: invoke void @_Z9may_throwv( |
|---|
| 60 | // CHECK-NEXT: to label %{{.+}} unwind label %[[CatchPad:.+]] |
|---|
| 61 | |
|---|
| 62 | // CHECK: [[CatchPad]]: |
|---|
| 63 | // CHECK-NEXT: landingpad |
|---|
| 64 | // CHECK-NEXT: catch i8* null |
|---|
| 65 | // CHECK: call void @_ZN7CleanupD1Ev( |
|---|
| 66 | // CHECK: br label %[[Catch:.+]] |
|---|
| 67 | |
|---|
| 68 | // CHECK: [[Catch]]: |
|---|
| 69 | // CHECK: call i8* @__cxa_begin_catch( |
|---|
| 70 | // CHECK: call void @_ZNSt12experimental16coroutine_traitsIJvEE12promise_type19unhandled_exceptionEv( |
|---|
| 71 | // CHECK: invoke void @__cxa_end_catch() |
|---|
| 72 | // CHECK-NEXT: to label %[[Cont:.+]] unwind |
|---|
| 73 | |
|---|
| 74 | // CHECK: [[Cont]]: |
|---|
| 75 | // CHECK-NEXT: br label %[[Cont2:.+]] |
|---|
| 76 | // CHECK: [[Cont2]]: |
|---|
| 77 | // CHECK-NEXT: br label %[[Cleanup:.+]] |
|---|
| 78 | |
|---|
| 79 | // CHECK: [[Cleanup]]: |
|---|
| 80 | // CHECK: call void @_ZNSt12experimental16coroutine_traitsIJvEE12promise_typeD1Ev( |
|---|
| 81 | // CHECK: %[[Mem0:.+]] = call i8* @llvm.coro.free( |
|---|
| 82 | // CHECK: call void @_ZdlPv(i8* %[[Mem0]] |
|---|
| 83 | |
|---|
| 84 | // CHECK: [[Dealloc]]: |
|---|
| 85 | // CHECK: %[[Mem:.+]] = call i8* @llvm.coro.free( |
|---|
| 86 | // CHECK: call void @_ZdlPv(i8* %[[Mem]]) |
|---|
| 87 | |
|---|
| 88 | co_return; |
|---|
| 89 | } |
|---|
| 90 | |
|---|
| 91 | // CHECK-LABEL: define void @_Z1gv( |
|---|
| 92 | void g() { |
|---|
| 93 | for (;;) |
|---|
| 94 | co_await suspend_always{}; |
|---|
| 95 | // Since this is the endless loop there should be no fallthrough handler (call to 'return_void'). |
|---|
| 96 | // CHECK-NOT: call void @_ZNSt12experimental16coroutine_traitsIJvEE12promise_type11return_voidEv |
|---|
| 97 | } |
|---|
| 98 | |
|---|