diff mbox series

coroutines : Convert await_ready () expressions to bool [PR99047].

Message ID 3502A472-2F28-4818-8899-05ECDCD9C7AE@sandoe.co.uk
State New
Headers show
Series coroutines : Convert await_ready () expressions to bool [PR99047]. | expand

Commit Message

Iain Sandoe March 15, 2021, 12:29 a.m. UTC
Hi

The awaiter.await_ready() should be converted per [expr.await]/3

(3.6) await-ready is the expression e.await_ready(), contextually
      converted to bool.

The conversion had been omitted, fixed as below,

tested on x86_64-darwin, x86_64-linux-gnu, and on cppcoro and
folly/coroutines.

OK for master / 10.x?
thanks
Iain

gcc/cp/ChangeLog:

	* coroutines.cc (expand_one_await_expression): If the
	await_ready() expression is not a boolean then convert it
	as required.

gcc/testsuite/ChangeLog:

	* g++.dg/coroutines/pr99047.C: New test.
---
 gcc/cp/coroutines.cc                      |  8 ++-
 gcc/testsuite/g++.dg/coroutines/pr99047.C | 83 +++++++++++++++++++++++
 2 files changed, 90 insertions(+), 1 deletion(-)
 create mode 100644 gcc/testsuite/g++.dg/coroutines/pr99047.C

Comments

Nathan Sidwell March 15, 2021, 11:53 a.m. UTC | #1
On 3/14/21 8:29 PM, Iain Sandoe wrote:
> Hi
> 
> The awaiter.await_ready() should be converted per [expr.await]/3
> 
> (3.6) await-ready is the expression e.await_ready(), contextually
>        converted to bool.
> 
> The conversion had been omitted, fixed as below,
> 
> tested on x86_64-darwin, x86_64-linux-gnu, and on cppcoro and
> folly/coroutines.
> 
> OK for master / 10.x?
> thanks
> Iain
> 
> gcc/cp/ChangeLog:
> 
> 	* coroutines.cc (expand_one_await_expression): If the
> 	await_ready() expression is not a boolean then convert it
> 	as required.
> 

ok
diff mbox series

Patch

diff --git a/gcc/cp/coroutines.cc b/gcc/cp/coroutines.cc
index ea714da146b..29923a04fae 100644
--- a/gcc/cp/coroutines.cc
+++ b/gcc/cp/coroutines.cc
@@ -1586,7 +1586,13 @@  expand_one_await_expression (tree *stmt, tree *await_expr, void *d)
 
   /* Use the await_ready() call to test if we need to suspend.  */
   tree ready_cond = TREE_VEC_ELT (awaiter_calls, 0); /* await_ready().  */
-  ready_cond = build1_loc (loc, TRUTH_NOT_EXPR, boolean_type_node, ready_cond);
+  /* Convert to bool, if necessary.  */
+  if (TREE_CODE (TREE_TYPE (ready_cond)) != BOOLEAN_TYPE)
+    ready_cond = cp_convert (boolean_type_node, ready_cond,
+			     tf_warning_or_error);
+  /* Be aggressive in folding here, since there are a significant number of cases
+     where the ready condition is constant.  */
+  ready_cond = invert_truthvalue_loc (loc, ready_cond);
   ready_cond
     = build1_loc (loc, CLEANUP_POINT_EXPR, boolean_type_node, ready_cond);
 
diff --git a/gcc/testsuite/g++.dg/coroutines/pr99047.C b/gcc/testsuite/g++.dg/coroutines/pr99047.C
new file mode 100644
index 00000000000..124633a08e6
--- /dev/null
+++ b/gcc/testsuite/g++.dg/coroutines/pr99047.C
@@ -0,0 +1,83 @@ 
+#include <optional>
+#include <coroutine>
+
+template <typename T>
+struct [[nodiscard]] task {
+    struct promise_type  {
+        std::suspend_always initial_suspend() {
+            return {};
+        }
+        auto final_suspend() noexcept {
+            struct awaiter {
+#if 1
+                std::false_type await_ready() noexcept {
+                    return {};
+                }
+#else
+                bool await_ready() noexcept {
+                    return false;
+                }
+#endif
+                std::coroutine_handle<> await_suspend(std::coroutine_handle<>) noexcept {
+                    return next;
+                }
+                void await_resume() noexcept {
+                }
+                std::coroutine_handle<> next;
+            };
+            return awaiter{next};
+        }
+
+        void unhandled_exception() noexcept {
+            std::terminate();
+        }
+        auto get_return_object() {
+            return task(this);
+        }
+        auto coro() {
+            return std::coroutine_handle<promise_type>::from_promise(*this);
+        }
+        void return_value(T val) {
+            result.emplace(std::move(val));
+        }
+
+        std::coroutine_handle<> next;
+        std::optional<T> result;
+    };
+
+    task(task&& source) : p(std::exchange(source.p, nullptr)) {}
+    explicit task(promise_type* p) : p(p) {}
+    ~task() {
+        if (p)
+            p->coro().destroy();
+    }
+
+    bool await_ready() noexcept {
+        return p->coro().done();
+    }
+    std::coroutine_handle<> await_suspend(std::coroutine_handle<> next) noexcept {
+        p->next = next;
+        return p->coro();
+    }
+    const T& await_resume() const& noexcept {
+        return *p->result;
+    }
+
+    promise_type* p;
+};
+
+task<int> five() {
+    co_return 5;
+}
+
+task<int> six() {
+    co_return co_await five() + 1;
+}
+
+
+int main() {
+    auto task = six();
+    task.p->next = std::noop_coroutine();
+    task.p->coro().resume();
+    return *task.p->result;
+}