1 // { dg-do run }
2 #include "../coro.h"
3
4 #include <stdexcept>
5
6 int frame_live = 0;
7 int promise_live = 0;
8 int task_live = 0;
9
10 struct Task
11 {
12 struct promise_type;
13 using handle = std::coroutine_handle<promise_type>;
14
15 struct promise_type
16 {
promise_typeTask::promise_type17 promise_type () { promise_live++; PRINT ("promise_type ()"); }
~promise_typeTask::promise_type18 ~promise_type () { promise_live--; PRINT ("~promise_type ()"); }
newTask::promise_type19 void* operator new(size_t sz) {
20 PRINT("operator new()");
21 frame_live++;
22 return ::operator new(sz);
23 }
deleteTask::promise_type24 void operator delete(void* p, size_t sz) {
25 PRINT("operator delete");
26 frame_live--;
27 return ::operator delete(p, sz);
28 }
29
get_return_objectTask::promise_type30 Task get_return_object() { return handle::from_promise(*this); }
initial_suspendTask::promise_type31 auto initial_suspend() noexcept { return std::suspend_always{}; }
final_suspendTask::promise_type32 auto final_suspend() noexcept { return std::suspend_always{}; }
return_voidTask::promise_type33 void return_void() noexcept {}
34
yield_valueTask::promise_type35 auto yield_value(int x) noexcept
36 {
37 PRINTF ("yield_value(%d)\n", x);
38 return std::suspend_always{};
39 }
40
unhandled_exceptionTask::promise_type41 void unhandled_exception()
42 {
43 PRINT ("unhandled_exception()");
44 throw;
45 }
46 };
47
TaskTask48 Task(handle h) : coro(h) { task_live++; PRINT ("Task(handle h)"); }
~TaskTask49 ~Task() { task_live--; PRINT ("~Task()"); if (coro) coro.destroy(); }
50
51 handle coro;
52 };
53
myco()54 Task myco()
55 {
56 co_yield 42;
57 throw std::out_of_range("TEST EXCEPTION");
58 }
59
main()60 int main()
61 {
62 {
63 Task task = myco();
64 PRINT ("START");
65 try {
66 PRINTF ("done #0 = %d\n", task.coro.done());
67 if (task.coro.done())
68 abort();
69 task.coro.resume(); // will yield 42
70 PRINTF ("done #1 = %d\n", task.coro.done());
71 if (task.coro.done())
72 abort();
73 task.coro.resume(); // will throw exception
74 PRINT ("should not be reached");
75 abort ();
76 }
77 catch (const std::exception&) {
78 PRINTF ("done exc = %d\n", task.coro.done());
79 if (!task.coro.done())
80 abort();
81 }
82 if (!task.coro.done())
83 abort();
84 } // should cause cause the destroy () to run.
85 if (task_live || promise_live || frame_live)
86 {
87 PRINTF ("task_live = %d, promise_live = %d, frame_live = %d\n",
88 task_live, promise_live, frame_live);
89 abort ();
90 }
91 }
92