Robert Shearman : widl: Add a function to generate the Ndr type calls.

[Alexandre Julliard]

Module: wine
Branch: refs/heads/master
Commit: 04632f6ab716c4089f6d3e399d0a516f5c209107
URL:    http://source.winehq.org/git/?p=wine.git;a=commit;h=04632f6ab716c4089f6d3e39...

Author: Robert Shearman <rob(a)codeweavers.com>
Date:   Fri Mar 31 12:47:35 2006 +0100

widl: Add a function to generate the Ndr type calls.

- Add a function to generate the Ndr type calls.
- Fix a typo where the aligned buffer was added to itself, rather than
assigned.

---

 tools/widl/typegen.c |   69 ++++++++++++++++++++++++--------------------------
 1 files changed, 33 insertions(+), 36 deletions(-)

diff --git a/tools/widl/typegen.c b/tools/widl/typegen.c
index d31110a..a284557 100644
--- a/tools/widl/typegen.c
+++ b/tools/widl/typegen.c
@@ -1396,20 +1396,41 @@ unsigned int get_required_buffer_size(co
     return get_required_buffer_size_type(var->type, var->ptr_level, var->array, var->name, alignment);
 }
 
-static inline const char *function_from_phase(enum remoting_phase phase)
+static void print_phase_function(FILE *file, int indent, const char *type,
+                                 enum remoting_phase phase,
+                                 const char *varname, unsigned int type_offset)
 {
+    const char *function;
     switch (phase)
     {
     case PHASE_BUFFERSIZE:
-        return "BufferSize";
+        function = "BufferSize";
+        break;
     case PHASE_MARSHAL:
-        return "Marshall";
+        function = "Marshall";
+        break;
     case PHASE_UNMARSHAL:
-        return "Unmarshall";
+        function = "Unmarshall";
+        break;
     case PHASE_FREE:
-        return "Free";
+        function = "Free";
+        break;
+    default:
+        assert(0);
+        return;
     }
-    return NULL;
+
+    print_file(file, indent, "Ndr%s%s(\n", type, function);
+    indent++;
+    print_file(file, indent, "(PMIDL_STUB_MESSAGE)&_StubMsg,\n");
+    print_file(file, indent, "%s%s,\n",
+               (phase == PHASE_UNMARSHAL) ? "(unsigned char **)&" : "(unsigned char *)",
+               varname);
+    print_file(file, indent, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]%s\n",
+               type_offset, (phase == PHASE_UNMARSHAL) ? "," : ");");
+    if (phase == PHASE_UNMARSHAL)
+        print_file(file, indent, "0);\n");
+    indent--;
 }
 
 /* returns whether the MaxCount, Offset or ActualCount members need to be
@@ -1470,9 +1491,7 @@ void write_remoting_arguments(FILE *file
         if (is_string_type(var->attrs, var->ptr_level, var->array))
         {
             if (var->array && var->array->is_const)
-                print_file(file, indent,
-                           "NdrNonConformantString%s(&_StubMsg, (unsigned char *)%s, &__MIDL_TypeFormatString.Format[%d]);\n",
-                           function_from_phase(phase), var->name, *type_offset);
+                print_phase_function(file, indent, "NonConformantString", phase, var->name, *type_offset);
             else
             {
                 if (size_is && is_size_needed_for_phase(phase))
@@ -1494,16 +1513,7 @@ void write_remoting_arguments(FILE *file
                 }
                 else
                 {
-                    print_file(file, indent, "NdrConformantString%s(\n", function_from_phase(phase));
-                    indent++;
-                    print_file(file, indent, "(PMIDL_STUB_MESSAGE)&_StubMsg,\n");
-                    print_file(file, indent, "(unsigned char *)%s,\n", var->name);
-                    print_file(file, indent, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]%s\n",
-                               *type_offset + (has_size ? 4 : 2),
-                               (phase == PHASE_MARSHAL) ? ");" : ",");
-                    if (phase == PHASE_UNMARSHAL)
-                        print_file(file, indent, "(unsigned char)0);\n");
-                    indent--;
+                    print_phase_function(file, indent, "ConformantString", phase, var->name, *type_offset);
                 }
             }
         }
@@ -1554,10 +1564,7 @@ void write_remoting_arguments(FILE *file
                 }
             }
 
-            print_file(file, indent,
-                       "Ndr%s%s(&_StubMsg, (unsigned char *)%s, &__MIDL_TypeFormatString.Format[%d]);\n",
-                       array_type, function_from_phase(phase), var->name,
-                       *type_offset);
+            print_phase_function(file, indent, array_type, phase, var->name, *type_offset);
         }
         else if (var->ptr_level == 0 && is_base_type(type->type))
         {
@@ -1606,7 +1613,7 @@ void write_remoting_arguments(FILE *file
 
             if (phase == PHASE_MARSHAL || phase == PHASE_UNMARSHAL)
             {
-                print_file(file, indent, "_StubMsg.Buffer += (unsigned char *)(((long)_StubMsg.Buffer + %u) & ~0x%x);\n",
+                print_file(file, indent, "_StubMsg.Buffer = (unsigned char *)(((long)_StubMsg.Buffer + %u) & ~0x%x);\n",
                            alignment - 1, alignment - 1);
 
                 if (phase == PHASE_MARSHAL)
@@ -1658,9 +1665,7 @@ void write_remoting_arguments(FILE *file
                 ndrtype = NULL;
             }
 
-            print_file(file, indent,
-                "Ndr%s%s(&_StubMsg, (unsigned char *)%s, &__MIDL_TypeFormatString.Format[%d]);\n",
-                ndrtype, function_from_phase(phase), var->name, *type_offset);
+            print_phase_function(file, indent, ndrtype, phase, var->name, *type_offset);
         }
         else
         {
@@ -1734,15 +1739,7 @@ void write_remoting_arguments(FILE *file
             }
             else
             {
-                print_file(file, indent, "NdrPointer%s(\n", function_from_phase(phase));
-                indent++;
-                print_file(file, indent, "(PMIDL_STUB_MESSAGE)&_StubMsg,\n");
-                print_file(file, indent, "(unsigned char *)%s,\n", var->name);
-                print_file(file, indent, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]%s\n",
-                           *type_offset, (phase == PHASE_MARSHAL) ? ");" : ",");
-                if (phase == PHASE_UNMARSHAL)
-                    print_file(file, indent, "(unsigned char *)0);\n");
-                indent--;
+                print_phase_function(file, indent, "Pointer", phase, var->name, *type_offset);
             }
         }
         fprintf(file, "\n");