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()77 f () noexcept
78 {
79   PRINT ("coro1: about to return");
80   co_return;
81 }
82 
main()83 int 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