1 // { dg-do run } 2 // { dg-output "main: returning(\n|\r\n|\r)" } 3 // { dg-output "Destroyed coro1(\n|\r\n|\r)" } 4 // { dg-output "Destroyed suspend_always_prt(\n|\r\n|\r)" } 5 // { dg-output "Destroyed Promise(\n|\r\n|\r)" } 6 7 // Check that we still get the right DTORs run when we let a suspended coro 8 // go out of scope. 9 10 #include "../coro.h" 11 12 struct coro1 { 13 struct promise_type; 14 using handle_type = coro::coroutine_handle<coro1::promise_type>; 15 handle_type handle; coro1coro116 coro1 () : handle(0) {} coro1coro117 coro1 (handle_type _handle) 18 : handle(_handle) { 19 PRINT("Created coro1 object from handle"); 20 } 21 coro1 (const coro1 &) = delete; // no copying coro1coro122 coro1 (coro1 &&s) : handle(s.handle) { 23 s.handle = nullptr; 24 PRINT("coro1 mv ctor "); 25 } 26 coro1 &operator = (coro1 &&s) { 27 handle = s.handle; 28 s.handle = nullptr; 29 PRINT("coro1 op= "); 30 return *this; 31 } ~coro1coro132 ~coro1() { 33 printf ("Destroyed coro1\n"); 34 if ( handle ) 35 handle.destroy(); 36 } 37 38 struct suspend_never_prt { await_readycoro1::suspend_never_prt39 bool await_ready() const noexcept { return true; } await_suspendcoro1::suspend_never_prt40 void await_suspend(handle_type) const noexcept { PRINT ("susp-never-susp");} await_resumecoro1::suspend_never_prt41 void await_resume() const noexcept { PRINT ("susp-never-resume");} ~suspend_never_prtcoro1::suspend_never_prt42 ~suspend_never_prt() {}; 43 }; 44 45 struct suspend_always_prt { await_readycoro1::suspend_always_prt46 bool await_ready() const noexcept { return false; } await_suspendcoro1::suspend_always_prt47 void await_suspend(handle_type) const noexcept { PRINT ("susp-always-susp");} await_resumecoro1::suspend_always_prt48 void await_resume() const noexcept { PRINT ("susp-always-resume");} ~suspend_always_prtcoro1::suspend_always_prt49 ~suspend_always_prt() { printf ("Destroyed suspend_always_prt\n"); } 50 }; 51 52 struct promise_type { promise_typecoro1::promise_type53 promise_type() { PRINT ("Created Promise"); } ~promise_typecoro1::promise_type54 ~promise_type() { printf ("Destroyed Promise\n"); } 55 get_return_objectcoro1::promise_type56 auto get_return_object () { 57 PRINT ("get_return_object: handle from promise"); 58 return handle_type::from_promise (*this); 59 } initial_suspendcoro1::promise_type60 auto initial_suspend () { 61 PRINT ("get initial_suspend (always)"); 62 return suspend_always_prt{}; 63 } final_suspendcoro1::promise_type64 auto final_suspend () noexcept { 65 PRINT ("get final_suspend (always)"); 66 return suspend_always_prt{}; 67 } return_voidcoro1::promise_type68 void return_void () { 69 PRINT ("return_void ()"); 70 } 71 unhandled_exceptioncoro1::promise_type72 void unhandled_exception() { PRINT ("** unhandled exception"); } 73 }; 74 }; 75 76 struct coro1 f()77f () noexcept 78 { 79 PRINT ("coro1: about to return"); 80 co_return; 81 } 82 main()83int main () 84 { 85 PRINT ("main: create coro1"); 86 struct coro1 x = f (); 87 PRINT ("main: got coro1 - resuming"); 88 if (x.handle.done()) 89 { 90 PRINT ("main: f() should be suspended, says it's done"); 91 abort(); 92 } 93 94 #if __has_builtin (__builtin_coro_suspended) 95 if (! __builtin_coro_suspended(handle)) 96 { 97 PRINT ("main: f() should be suspended, but says it isn't"); 98 abort(); 99 } 100 #endif 101 102 /* We are suspended... so let everything out of scope and therefore 103 destroy it. */ 104 105 puts ("main: returning"); 106 return 0; 107 } 108