| // { dg-do run } |
| |
| #include "../coro.h" |
| |
| struct pt_b |
| { |
| constexpr std::suspend_never initial_suspend() const noexcept { return {}; } |
| constexpr std::suspend_never final_suspend() const noexcept { return {}; } |
| constexpr void return_void () {}; |
| }; |
| |
| int called_ueh_op = 0; |
| struct ueh |
| { |
| void operator ()() noexcept { |
| PRINT("call to operator UEH"); |
| called_ueh_op++; |
| } |
| }; |
| |
| struct pt_c : pt_b |
| { |
| using handle_t = std::coroutine_handle<pt_c>; |
| auto get_return_object() noexcept { return handle_t::from_promise(*this); } |
| ueh unhandled_exception; |
| }; |
| |
| int lambda_ueh = 0; |
| |
| struct pt_d : pt_b |
| { |
| using handle_t = std::coroutine_handle<pt_d>; |
| auto get_return_object() noexcept { return handle_t::from_promise(*this); } |
| static constexpr auto unhandled_exception |
| = [] () noexcept { PRINT("call to lambda UEH"); lambda_ueh++; }; |
| }; |
| |
| template <> |
| struct std::coroutine_traits<pt_c::handle_t> |
| { using promise_type = pt_c; }; |
| |
| static pt_c::handle_t |
| foo () |
| { |
| throw ("foo"); |
| co_return; |
| } |
| |
| template <> |
| struct std::coroutine_traits<pt_d::handle_t> |
| { using promise_type = pt_d; }; |
| |
| static pt_d::handle_t |
| bar () |
| { |
| throw ("bar"); |
| co_return; |
| } |
| |
| int main () |
| { |
| foo (); |
| bar (); |
| if (called_ueh_op != 1 || lambda_ueh != 1) |
| { |
| PRINT ("Failed to call one of the UEH cases"); |
| abort (); |
| } |
| } |