Nathan Sidwell <nat...@acm.org> wrote:

> On 4/25/20 11:08 AM, Iain Sandoe wrote:

>> +  tree arg = DECL_ARGUMENTS (fndecl);
>> +  bool lambda_p = LAMBDA_TYPE_P (DECL_CONTEXT (fndecl));
> 
> I think LAMBDA_FUNCTION_P (fndecl) expresses intent better.
done in both places.

>> +
>>        parm.this_ptr = is_this_parameter (arg);
>> +      if (lambda_p)
>> +        {
>> +          parm.lambda_cobj = parm.this_ptr
>> +                             || (DECL_NAME (arg) == closure_identifier);
> 
> i'm confused by the need for || here.  Why isn't just the DECL_NAME test 
> needed?  The parser appears to give the object parameter that name.  (If you 
> do need the parm.this_ptr piece, it suggests somewhere else outside of coro 
> is not being consistent, but ICBW.)

It seems that, when a lambda is part of a class, the closure object gets the 
name ‘this’.
When a lambda is defined outside a class context, the closure object is named 
__closure.

I added a comment as to why we make the two checks (also, why we can’t just use
is_this_parameter() as the test elsewhere)

>> +        vec_safe_push (args, arg);
>> +      else if (0 && parm_i->this_ptr)
> 
> ^^ looks like now-disabled experimental code?
well caught, now enabled as intended.

thanks for the review is the revised below OK (testing now)

thanks
iain

===

We changed the argument passed to the promise parameter preview
to match a reference to *this.  However to be consistent with the
other ports, we do need to match the reference transformation in
the traits lookup and the promise allocator lookup.

gcc/cp/ChangeLog:

2020-04-27  Iain Sandoe  <i...@sandoe.co.uk>

        PR c++/94760
        * coroutines.cc (instantiate_coro_traits): Pass a reference to
        object type rather than a pointer type for 'this', for method
        coroutines.
        (struct param_info): Add a field to hold that the parm is a lambda
        closure pointer.
        (morph_fn_to_coro): Check for lambda closure pointers in the
        args.  Use a reference to *this when building the args list for the
        promise allocator lookup.

gcc/testsuite/ChangeLog:

2020-04-27  Iain Sandoe  <i...@sandoe.co.uk>

        PR c++/94760
        * g++.dg/coroutines/pr94760-mismatched-traits-and-promise-prev.C: New 
test.
---
 gcc/cp/coroutines.cc                          | 54 +++++++++++++++++--
 ...9xxxx-mismatched-traits-and-promise-prev.C | 29 ++++++++++
 2 files changed, 79 insertions(+), 4 deletions(-)
 create mode 100644 
gcc/testsuite/g++.dg/coroutines/pr9xxxx-mismatched-traits-and-promise-prev.C

diff --git a/gcc/cp/coroutines.cc b/gcc/cp/coroutines.cc
index 0a5a0c9b1d2..cfaa018c551 100644
--- a/gcc/cp/coroutines.cc
+++ b/gcc/cp/coroutines.cc
@@ -296,14 +296,25 @@ instantiate_coro_traits (tree fndecl, location_t kw)
      type.  */
 
   tree functyp = TREE_TYPE (fndecl);
+  tree arg = DECL_ARGUMENTS (fndecl);
+  bool lambda_p = LAMBDA_FUNCTION_P (fndecl);
   tree arg_node = TYPE_ARG_TYPES (functyp);
   tree argtypes = make_tree_vec (list_length (arg_node)-1);
   unsigned p = 0;
 
   while (arg_node != NULL_TREE && !VOID_TYPE_P (TREE_VALUE (arg_node)))
     {
-      TREE_VEC_ELT (argtypes, p++) = TREE_VALUE (arg_node);
+      if (is_this_parameter (arg) && !lambda_p)
+       {
+         /* We pass a reference to *this to the param preview.  */
+         tree ct = TREE_TYPE (TREE_TYPE (arg));
+         TREE_VEC_ELT (argtypes, p++) = cp_build_reference_type (ct, false);
+       }
+      else
+       TREE_VEC_ELT (argtypes, p++) = TREE_VALUE (arg_node);
+
       arg_node = TREE_CHAIN (arg_node);
+      arg = DECL_CHAIN (arg);
     }
 
   tree argtypepack = cxx_make_type (TYPE_ARGUMENT_PACK);
@@ -1766,6 +1777,7 @@ struct param_info
   bool pt_ref;       /* Was a pointer to object.  */
   bool trivial_dtor; /* The frame type has a trivial DTOR.  */
   bool this_ptr;     /* Is 'this' */
+  bool lambda_cobj;  /* Lambda capture object */
 };
 
 struct local_var_info
@@ -3652,6 +3664,7 @@ morph_fn_to_coro (tree orig, tree *resumer, tree 
*destroyer)
          The second two entries start out empty - and only get populated
          when we see uses.  */
       param_uses = new hash_map<tree, param_info>;
+      bool lambda_p = LAMBDA_FUNCTION_P (orig);
 
       unsigned no_name_parm = 0;
       for (tree arg = DECL_ARGUMENTS (orig); arg != NULL;
@@ -3692,7 +3705,19 @@ morph_fn_to_coro (tree orig, tree *resumer, tree 
*destroyer)
            }
          else
            parm.frame_type = actual_type;
+
+         /* The closure object pointer is called 'this' when a lambda is
+            part of a class, and __closure when it is not.  */
          parm.this_ptr = is_this_parameter (arg);
+         if (lambda_p)
+           {
+             parm.lambda_cobj = parm.this_ptr
+                                || (DECL_NAME (arg) == closure_identifier);
+             parm.this_ptr = false;
+           }
+         else
+           parm.lambda_cobj = false;
+
          parm.trivial_dtor = TYPE_HAS_TRIVIAL_DESTRUCTOR (parm.frame_type);
          char *buf;
          if (DECL_NAME (arg))
@@ -3838,9 +3863,28 @@ morph_fn_to_coro (tree orig, tree *resumer, tree 
*destroyer)
        those of the original function.  */
       vec<tree, va_gc> *args = make_tree_vector ();
       vec_safe_push (args, resizeable); /* Space needed.  */
+
       for (tree arg = DECL_ARGUMENTS (orig); arg != NULL;
           arg = DECL_CHAIN (arg))
-       vec_safe_push (args, arg);
+       {
+         param_info *parm_i = param_uses->get (arg);
+         gcc_checking_assert (parm_i);
+         if (parm_i->lambda_cobj)
+           vec_safe_push (args, arg);
+         else if (parm_i->this_ptr)
+           {
+             /* We pass a reference to *this to the allocator lookup.  */
+             tree tt = TREE_TYPE (TREE_TYPE (arg));
+             tree this_ref = build1 (INDIRECT_REF, tt, arg);
+             tt = cp_build_reference_type (tt, false);
+             this_ref = convert_to_reference (tt, this_ref, CONV_STATIC,
+                                              LOOKUP_NORMAL , NULL_TREE,
+                                              tf_warning_or_error);
+             vec_safe_push (args, this_ref);
+           }
+         else
+           vec_safe_push (args, arg);
+       }
 
       /* We might need to check that the provided function is nothrow.  */
       tree func;
@@ -4036,8 +4080,10 @@ morph_fn_to_coro (tree orig, tree *resumer, tree 
*destroyer)
                                              false, tf_warning_or_error);
 
          /* Add this to the promise CTOR arguments list, accounting for
-            refs and this ptr.  */
-         if (parm.this_ptr)
+            refs and special handling for method this ptr.  */
+         if (parm.lambda_cobj)
+           vec_safe_push (promise_args, arg);
+         else if (parm.this_ptr)
            {
              /* We pass a reference to *this to the param preview.  */
              tree tt = TREE_TYPE (arg);
diff --git 
a/gcc/testsuite/g++.dg/coroutines/pr9xxxx-mismatched-traits-and-promise-prev.C 
b/gcc/testsuite/g++.dg/coroutines/pr9xxxx-mismatched-traits-and-promise-prev.C
new file mode 100644
index 00000000000..235b5e757be
--- /dev/null
+++ 
b/gcc/testsuite/g++.dg/coroutines/pr9xxxx-mismatched-traits-and-promise-prev.C
@@ -0,0 +1,29 @@
+// { dg-do compile  { target c++17 } }
+
+#include "coro.h"
+
+// Test that we get matching types to traits and promise param
+// preview.
+
+// A separate issue from allowing non-class return types.
+struct Fake {} ;
+
+template<typename R, typename CallOp, typename ...T>
+struct std::coroutine_traits<R, CallOp, T...> {
+    struct promise_type {
+        promise_type (CallOp op, T ...args) {}
+        Fake get_return_object() { return {}; }
+        std::suspend_always initial_suspend() { return {}; }
+        std::suspend_never final_suspend() { return {}; }
+        void return_void() {}
+        void unhandled_exception() {}
+    };
+};
+
+
+struct Foo
+{
+  Fake operator() (int a) {
+    co_return;
+  }
+};
-- 
2.24.1


Reply via email to