Mojo C++ bindings: support passing associated interface pointers/requests in method parameter lists…

mojo/public/tools/bindings/generators/cpp_templates/struct_macros.tmpl

 {#  TODO(yzshen): Make these templates more readable. #}
 
 {#  Computes the serialized size for the specified struct.
     |struct| is the struct definition.
     |input_field_pattern| should be a pattern that contains one string
     placeholder, for example, "input->%s", "p_%s". The placeholder will be
     substituted with struct field names to refer to the input fields.
     This macro is expanded to compute seriailized size for both:
     - user-defined structs: the input is an instance of the corresponding struct
       wrapper class.
@@ skipping 72 matching lines @@
   MOJO_INTERNAL_DLOG_SERIALIZATION_WARNING(
 {%-       if kind|is_interface_kind %}
       !{{output}}->{{name}}.handle.is_valid(),
 {%-       else %}
       !{{output}}->{{name}}.is_valid(),
 {%-       endif %}
       mojo::internal::VALIDATION_ERROR_UNEXPECTED_INVALID_HANDLE,
       "invalid {{name}} in {{struct_display_name}}");
 {%-     endif %}
 {%-   elif kind|is_associated_kind %}
-  // TODO(yzshen): add serialization logic for associated kinds. 
+  // TODO(yzshen): add some DCHECKs for non-nullable, is_local, correct router, etc.
+{%-     if kind|is_associated_interface_kind %}
+  mojo::internal::AssociatedInterfacePtrInfoToData({{input_field}}.Pass(), &{{output}}->{{name}});
+{%-     else %}
+  {{output}}->{{name}} = mojo::internal::AssociatedInterfaceRequestHelper::PassHandle(&{{input_field}}).release();
+{%-     endif %}
 {%-   else %}
   {{output}}->{{name}} = {{input_field}};
 {%-   endif %}
 {%- endfor %}
 {%- endmacro -%}
 
 {#  Deserializes the specified struct.
     |struct| is the struct definition.
     |input| is the name of the input struct instance.
     |output_field_pattern| should be a pattern that contains one string
     placeholder, for example, "result->%s", "p_%s". The placeholder will be
     substituted with struct field names to refer to the output fields.
     This macro is expanded to do deserialization for both:
     - user-defined structs: the output is an instance of the corresponding
       struct wrapper class.
     - method parameters/response parameters: the output is a list of
       arguments. #}
-{%- macro deserialize(struct, input, output_field_pattern) -%}
+    |context| is the name of the serialization context.
+{%- macro deserialize(struct, input, output_field_pattern, context) -%}
   do {
     // NOTE: The memory backing |{{input}}| may has be smaller than
     // |sizeof(*{{input}})| if the message comes from an older version.
 {#-   Before deserialize fields introduced at a certain version, we need to add
       a version check, which makes sure we skip further deserialization if
       |input| is from an earlier version. |last_checked_version| records the
       last version that we have added such version check. #}
 {%-   set last_checked_version = 0 %}
 {%-   for pf in struct.packed.packed_fields_in_ordinal_order %}
 {%-     set output_field = output_field_pattern|format(pf.field.name) %}
 {%-     set name = pf.field.name %}
 {%-     set kind = pf.field.kind %}
 {%-     if pf.min_version > last_checked_version %}
 {%-       set last_checked_version = pf.min_version %}
     if ({{input}}->header_.version < {{pf.min_version}})
       break;
 {%-     endif %}
 {%-     if kind|is_object_kind %}
 {%-       if kind|is_union_kind %}
-    Deserialize_(&{{input}}->{{name}}, &{{output_field}});
+    Deserialize_(&{{input}}->{{name}}, &{{output_field}}, {{context}});
 {%-       else %}
-    Deserialize_({{input}}->{{name}}.ptr, &{{output_field}});
+    Deserialize_({{input}}->{{name}}.ptr, &{{output_field}}, {{context}});
 {%-       endif %}
 {%-     elif kind|is_interface_kind %}
     mojo::internal::InterfaceDataToPointer(&{{input}}->{{name}}, &{{output_field}});
 {%-     elif kind|is_interface_request_kind %}
     {{output_field}}.Bind(mojo::MakeScopedHandle(mojo::internal::FetchAndReset(&{{input}}->{{name}})));
 {%-     elif kind|is_any_handle_kind %}
     {{output_field}}.reset(mojo::internal::FetchAndReset(&{{input}}->{{name}}));
-{%-   elif kind|is_associated_kind %}
-    // TODO(yzshen): add deserialization logic for associated kinds.
+{%-   elif kind|is_associated_interface_kind %}
+    mojo::internal::AssociatedInterfaceDataToPtrInfo(&{{input}}->{{name}}, &{{output_field}}, ({{context}})->router.get());
+{%-   elif kind|is_associated_interface_request_kind %}
+    mojo::internal::AssociatedInterfaceRequestHelper::SetHandle(
+        &{{output_field}},
+        ({{context}})->router->CreateLocalEndpointHandle(mojo::internal::FetchAndReset(&{{input}}->{{name}})));
 {%-     elif kind|is_enum_kind %}
     {{output_field}} = static_cast<{{kind|cpp_wrapper_type}}>({{input}}->{{name}});
 {%-     else %}
     {{output_field}} = {{input}}->{{name}};
 {%-     endif %}
 {%-   endfor %}
   } while (false);
 {%- endmacro %}