Fixed ICE-6602: optionals marshaling bug

5 files changed, 231 insertions, 95 deletions

diff --git a/cpp/src/slice2cs/Gen.cpp b/cpp/src/slice2cs/Gen.cpp
index 81f83b2202e..afcb39346cc 100644
--- a/cpp/src/slice2cs/Gen.cpp
+++ b/cpp/src/slice2cs/Gen.cpp
@@ -1456,7 +1456,7 @@ Slice::CsVisitor::writeDispatchAndMarshalling(const ClassDefPtr& p, bool stream)
     }
     _out << eb;
 
-    if(allClassMembers.size() != 0)
+    if(classMembers.size() != 0)
     {
         _out << sp;
         if(!p->isInterface())
@@ -1479,14 +1479,14 @@ Slice::CsVisitor::writeDispatchAndMarshalling(const ClassDefPtr& p, bool stream)
             _out << sp << nl << "internal Patcher__(string type, Ice.Object instance";
         }
 
-        if(allClassMembers.size() > 1)
+        if(classMembers.size() > 1)
         {
             _out << ", int member";
         }
         _out << ") : base(type)";
         _out << sb;
         _out << nl << "_instance = (" << name << ")instance;";
-        if(allClassMembers.size() > 1)
+        if(classMembers.size() > 1)
         {
             _out << nl << "_member = member;";
         }
@@ -1496,15 +1496,19 @@ Slice::CsVisitor::writeDispatchAndMarshalling(const ClassDefPtr& p, bool stream)
         _out << sb;
         _out << nl << "try";
         _out << sb;
-        if(allClassMembers.size() > 1)
+        if(classMembers.size() > 1)
         {
             _out << nl << "switch(_member)";
             _out << sb;
         }
         int memberCount = 0;
-        for(DataMemberList::const_iterator d = allClassMembers.begin(); d != allClassMembers.end(); ++d)
+        for(DataMemberList::const_iterator d = classMembers.begin(); d != classMembers.end(); ++d)
         {
-            if(allClassMembers.size() > 1)
+            if((*d)->optional())
+            {
+                continue;
+            }
+            if(classMembers.size() > 1)
             {
                 _out.dec();
                 _out << nl << "case " << memberCount << ":";
@@ -1512,36 +1516,69 @@ Slice::CsVisitor::writeDispatchAndMarshalling(const ClassDefPtr& p, bool stream)
             }
             string memberName = fixId((*d)->name(), DotNet::ICloneable, true);
             string memberType = typeToString((*d)->type(), (*d)->optional());
-            if((*d)->optional())
+            
+            if(ClassDeclPtr::dynamicCast((*d)->type()))
             {
-                if(ClassDeclPtr::dynamicCast((*d)->type()))
-                {
-                    _out << nl << "_instance." << memberName << " = new " << memberType << "(("
-                         << typeToString((*d)->type()) << ")v);";
-                }
-                else
-                {
-                    _out << nl << "_instance." << memberName << " = new " << memberType << "(v);";
-                }
+                _out << nl << "_instance." << memberName << " = (" << memberType << ")v;";
             }
             else
             {
-                if(ClassDeclPtr::dynamicCast((*d)->type()))
+                _out << nl << "_instance." << memberName << " = v;";
+            }
+            
+            if(classMembers.size() > 1)
+            {
+                _out << nl << "break;";
+            }
+            memberCount++;
+        }
+        
+        for(DataMemberList::const_iterator d = optionalMembers.begin(); d != optionalMembers.end(); ++d)
+        {
+            TypePtr paramType = (*d)->type();
+            BuiltinPtr builtin = BuiltinPtr::dynamicCast(paramType);
+            if((builtin && builtin->kind() == Builtin::KindObject) || ClassDeclPtr::dynamicCast(paramType))
+            {
+                if(classMembers.size() > 1)
                 {
-                    _out << nl << "_instance." << memberName << " = (" << memberType << ")v;";
+                    _out.dec();
+                    _out << nl << "case " << memberCount << ":";
+                    _out.inc();
+                }
+                string memberName = fixId((*d)->name(), DotNet::ICloneable, true);
+                string memberType = typeToString((*d)->type(), (*d)->optional());
+                if((*d)->optional())
+                {
+                    if(ClassDeclPtr::dynamicCast((*d)->type()))
+                    {
+                        _out << nl << "_instance." << memberName << " = new " << memberType << "(("
+                            << typeToString((*d)->type()) << ")v);";
+                    }
+                    else
+                    {
+                        _out << nl << "_instance." << memberName << " = new " << memberType << "(v);";
+                    }
                 }
                 else
                 {
-                    _out << nl << "_instance." << memberName << " = v;";
+                    if(ClassDeclPtr::dynamicCast((*d)->type()))
+                    {
+                        _out << nl << "_instance." << memberName << " = (" << memberType << ")v;";
+                    }
+                    else
+                    {
+                        _out << nl << "_instance." << memberName << " = v;";
+                    }
                 }
+                if(classMembers.size() > 1)
+                {
+                    _out << nl << "break;";
+                }
+                memberCount++;
             }
-            if(allClassMembers.size() > 1)
-            {
-                _out << nl << "break;";
-            }
-            memberCount++;
         }
-        if(allClassMembers.size() > 1)
+        
+        if(classMembers.size() > 1)
         {
             _out << eb;
         }
@@ -1553,7 +1590,7 @@ Slice::CsVisitor::writeDispatchAndMarshalling(const ClassDefPtr& p, bool stream)
         _out << eb;
 
         _out << sp << nl << "private " << name << " _instance;";
-        if(allClassMembers.size() > 1)
+        if(classMembers.size() > 1)
         {
             _out << nl << "private int _member;";
         }
@@ -1570,7 +1607,7 @@ Slice::CsVisitor::writeDispatchAndMarshalling(const ClassDefPtr& p, bool stream)
     _out << sb;
     _out << nl << "is__.startReadSlice();";
     int classMemberCount = static_cast<int>(allClassMembers.size() - classMembers.size());
-    const bool needCustomPatcher = classMembers.size() > 1 || allClassMembers.size() > 1;
+    const bool needCustomPatcher = classMembers.size() > 1;
     for(DataMemberList::const_iterator d = members.begin(); d != members.end(); ++d)
     {
         if(!(*d)->optional())
@@ -3516,6 +3553,9 @@ Slice::Gen::TypesVisitor::visitExceptionEnd(const ExceptionPtr& p)
 
     DataMemberList allDataMembers = p->allDataMembers();
     DataMemberList dataMembers = p->dataMembers();
+    DataMemberList allClassMembers = p->allClassDataMembers();
+    DataMemberList classMembers = p->classDataMembers();
+    DataMemberList optionalMembers = p->orderedOptionalDataMembers();
 
     vector<string> allParamDecl;
     for(DataMemberList::const_iterator q = allDataMembers.begin(); q != allDataMembers.end(); ++q)
@@ -3823,8 +3863,8 @@ Slice::Gen::TypesVisitor::visitExceptionEnd(const ExceptionPtr& p)
         }
         _out << eb;
 
-        DataMemberList allClassMembers = p->allClassDataMembers();
-        if(allClassMembers.size() != 0)
+        
+        if(classMembers.size() != 0)
         {
             _out << sp;
             emitGeneratedCodeAttribute();
@@ -3836,14 +3876,14 @@ Slice::Gen::TypesVisitor::visitExceptionEnd(const ExceptionPtr& p)
             _out << "class Patcher__ : IceInternal.Patcher";
             _out << sb;
             _out << sp << nl << "internal Patcher__(string type, Ice.Exception instance";
-            if(allClassMembers.size() > 1)
+            if(classMembers.size() > 1)
             {
                 _out << ", int member";
             }
             _out << ") : base(type)";
             _out << sb;
             _out << nl << "_instance = (" << name << ")instance;";
-            if(allClassMembers.size() > 1)
+            if(classMembers.size() > 1)
             {
                 _out << nl << "_member = member;";
             }
@@ -3853,15 +3893,19 @@ Slice::Gen::TypesVisitor::visitExceptionEnd(const ExceptionPtr& p)
             _out << sb;
             _out << nl << "try";
             _out << sb;
-            if(allClassMembers.size() > 1)
+            if(classMembers.size() > 1)
             {
                 _out << nl << "switch(_member)";
                 _out << sb;
             }
             int memberCount = 0;
-            for(DataMemberList::const_iterator q = allClassMembers.begin(); q != allClassMembers.end(); ++q)
+            for(DataMemberList::const_iterator q = classMembers.begin(); q != classMembers.end(); ++q)
             {
-                if(allClassMembers.size() > 1)
+                if((*q)->optional())
+                {
+                    continue;
+                }
+                if(classMembers.size() > 1)
                 {
                     _out.dec();
                     _out << nl << "case " << memberCount << ":";
@@ -3892,13 +3936,60 @@ Slice::Gen::TypesVisitor::visitExceptionEnd(const ExceptionPtr& p)
                         _out << nl << "_instance." << memberName << " = v;";
                     }
                 }
-                if(allClassMembers.size() > 1)
+                if(classMembers.size() > 1)
                 {
                     _out << nl << "break;";
                 }
                 memberCount++;
             }
-            if(allClassMembers.size() > 1)
+            
+            for(DataMemberList::const_iterator q = optionalMembers.begin(); q != optionalMembers.end(); ++q)
+            {
+
+                TypePtr paramType = (*q)->type();
+                BuiltinPtr builtin = BuiltinPtr::dynamicCast(paramType);
+                if((builtin && builtin->kind() == Builtin::KindObject) || ClassDeclPtr::dynamicCast(paramType))
+                {
+                    if(classMembers.size() > 1)
+                    {
+                        _out.dec();
+                        _out << nl << "case " << memberCount << ":";
+                        _out.inc();
+                    }
+                    string memberName = fixId((*q)->name(), DotNet::Exception);
+                    string memberType = typeToString((*q)->type(), (*q)->optional());
+                    if((*q)->optional())
+                    {
+                        if(ClassDeclPtr::dynamicCast((*q)->type()))
+                        {
+                            _out << nl << "_instance." << memberName << " = new " << memberType << "(("
+                                << typeToString((*q)->type()) << ")v);";
+                        }
+                        else
+                        {
+                            _out << nl << "_instance." << memberName << " = new " << memberType << "(v);";
+                        }
+                    }
+                    else
+                    {
+                        if(ClassDeclPtr::dynamicCast((*q)->type()))
+                        {
+                            _out << nl << "_instance." << memberName << " = (" << memberType << ")v;";
+                        }
+                        else
+                        {
+                            _out << nl << "_instance." << memberName << " = v;";
+                        }
+                    }
+                    if(classMembers.size() > 1)
+                    {
+                        _out << nl << "break;";
+                    }
+                    memberCount++;
+                }
+            }
+            
+            if(classMembers.size() > 1)
             {
                 _out << eb;
             }
@@ -3910,7 +4001,7 @@ Slice::Gen::TypesVisitor::visitExceptionEnd(const ExceptionPtr& p)
             _out << eb;
 
             _out << sp << nl << "private " << name << " _instance;";
-            if(allClassMembers.size() > 1)
+            if(classMembers.size() > 1)
             {
                 _out << nl << "private int _member;";
             }
@@ -3922,9 +4013,9 @@ Slice::Gen::TypesVisitor::visitExceptionEnd(const ExceptionPtr& p)
         _out << nl << "protected override void readImpl__(IceInternal.BasicStream is__)";
         _out << sb;
         _out << nl << "is__.startReadSlice();";
-        DataMemberList classMembers = p->classDataMembers();
+        
         int classMemberCount = static_cast<int>(allClassMembers.size() - classMembers.size());
-        const bool needCustomPatcher = classMembers.size() > 1 || allClassMembers.size() > 1;
+        const bool needCustomPatcher = classMembers.size() > 1;
         for(DataMemberList::const_iterator q = dataMembers.begin(); q != dataMembers.end(); ++q)
         {
             writeUnmarshalDataMember(*q, fixId((*q)->name(), DotNet::Exception), needCustomPatcher, classMemberCount);
diff --git a/cpp/src/slice2java/Gen.cpp b/cpp/src/slice2java/Gen.cpp
index 0cfbdc57c37..469a8751f5e 100644
--- a/cpp/src/slice2java/Gen.cpp
+++ b/cpp/src/slice2java/Gen.cpp
@@ -855,7 +855,8 @@ Slice::JavaVisitor::writeStreamUnmarshalDataMember(Output& out, const string& pa
 }
 
 void
-Slice::JavaVisitor::writePatcher(Output& out, const string& package, const DataMemberList& classMembers, bool stream)
+Slice::JavaVisitor::writePatcher(Output& out, const string& package, const DataMemberList& classMembers,
+                                 const DataMemberList& optionalMembers, bool stream)
 {
     out << sp << nl << "private class Patcher implements IceInternal.Patcher";
     if(stream)
@@ -881,6 +882,11 @@ Slice::JavaVisitor::writePatcher(Output& out, const string& package, const DataM
     int memberCount = 0;
     for(DataMemberList::const_iterator d = classMembers.begin(); d != classMembers.end(); ++d)
     {
+        if((*d)->optional())
+        {
+            continue;
+        }
+        
         BuiltinPtr b = BuiltinPtr::dynamicCast((*d)->type());
         if(b)
         {
@@ -902,8 +908,60 @@ Slice::JavaVisitor::writePatcher(Output& out, const string& package, const DataM
             }
         }
 
-        if((*d)->optional())
+        string memberName = fixKwd((*d)->name());
+        if(b)
+        {
+            out << nl << memberName << " = v;";
+        }
+        else
+        {
+            string memberType = typeToString((*d)->type(), TypeModeMember, package);
+            out << nl << "if(v == null || v instanceof " << memberType << ")";
+            out << sb;
+            out << nl << memberName << " = (" << memberType << ")v;";
+            out << eb;
+            out << nl << "else";
+            out << sb;
+            out << nl << "IceInternal.Ex.throwUOE(type(), v);";
+            out << eb;
+        }
+
+        if(classMembers.size() > 1)
+        {
+            out << nl << "break;";
+        }
+
+        memberCount++;
+    }
+    
+    for(DataMemberList::const_iterator d = optionalMembers.begin(); d != optionalMembers.end(); ++d)
+    {        
+        BuiltinPtr b = BuiltinPtr::dynamicCast((*d)->type());
+        if(b && b->kind() != Builtin::KindObject)
         {
+            continue;
+        }
+        
+        TypePtr paramType = (*d)->type();
+        BuiltinPtr builtin = BuiltinPtr::dynamicCast(paramType);
+        if((builtin && builtin->kind() == Builtin::KindObject) || ClassDeclPtr::dynamicCast(paramType))
+        {
+
+            if(classMembers.size() > 1)
+            {
+                out.dec();
+                out << nl << "case " << memberCount << ":";
+                out.inc();
+                if(b)
+                {
+                    out << nl << "__typeId = Ice.ObjectImpl.ice_staticId();";
+                }
+                else
+                {
+                    out << nl << "__typeId = \"" << (*d)->type()->typeId() << "\";";
+                }
+            }
+            
             string capName = (*d)->name();
             capName[0] = toupper(static_cast<unsigned char>(capName[0]));
 
@@ -923,35 +981,16 @@ Slice::JavaVisitor::writePatcher(Output& out, const string& package, const DataM
                 out << nl << "IceInternal.Ex.throwUOE(type(), v);";
                 out << eb;
             }
-        }
-        else
-        {
-            string memberName = fixKwd((*d)->name());
-            if(b)
+
+            if(classMembers.size() > 1)
             {
-                out << nl << memberName << " = v;";
+                out << nl << "break;";
             }
-            else
-            {
-                string memberType = typeToString((*d)->type(), TypeModeMember, package);
-                out << nl << "if(v == null || v instanceof " << memberType << ")";
-                out << sb;
-                out << nl << memberName << " = (" << memberType << ")v;";
-                out << eb;
-                out << nl << "else";
-                out << sb;
-                out << nl << "IceInternal.Ex.throwUOE(type(), v);";
-                out << eb;
-            }
-        }
 
-        if(classMembers.size() > 1)
-        {
-            out << nl << "break;";
+            memberCount++;
         }
-
-        memberCount++;
     }
+    
     if(classMembers.size() > 1)
     {
         out << eb;
@@ -1709,18 +1748,20 @@ Slice::JavaVisitor::writeDispatchAndMarshalling(Output& out, const ClassDefPtr&
     }
     out << eb;
 
+    DataMemberList classMembers = p->classDataMembers();
     DataMemberList allClassMembers = p->allClassDataMembers();
-    if(allClassMembers.size() != 0)
+    
+    if(classMembers.size() != 0)
     {
-        writePatcher(out, package, allClassMembers, stream);
+        writePatcher(out, package, classMembers, optionalMembers, stream);
     }
 
     out << sp << nl << "protected void __readImpl(IceInternal.BasicStream __is)";
     out << sb;
     out << nl << "__is.startReadSlice();";
-    DataMemberList classMembers = p->classDataMembers();
+    
     int classMemberCount = static_cast<int>(allClassMembers.size() - classMembers.size());
-    const bool needCustomPatcher = classMembers.size() > 1 || allClassMembers.size() > 1;
+    const bool needCustomPatcher = classMembers.size() > 1;
     iter = 0;
     for(DataMemberList::const_iterator d = members.begin(); d != members.end(); ++d)
     {
@@ -3697,18 +3738,19 @@ Slice::Gen::TypesVisitor::visitExceptionEnd(const ExceptionPtr& p)
         }
         out << eb;
 
+        DataMemberList classMembers = p->classDataMembers();
         DataMemberList allClassMembers = p->allClassDataMembers();
-        if(allClassMembers.size() != 0)
+    
+        if(classMembers.size() != 0)
         {
-            writePatcher(out, package, allClassMembers, _stream);
+            writePatcher(out, package, classMembers, optionalMembers, _stream);
         }
         out << sp << nl << "protected void" << nl << "__readImpl(IceInternal.BasicStream __is)";
         out << sb;
         out << nl << "__is.startReadSlice();";
         iter = 0;
-        DataMemberList classMembers = p->classDataMembers();
         int classMemberCount = static_cast<int>(allClassMembers.size() - classMembers.size());
-        const bool needCustomPatcher = classMembers.size() > 1 || allClassMembers.size() > 1;
+        const bool needCustomPatcher = classMembers.size() > 1;
         for(DataMemberList::const_iterator d = members.begin(); d != members.end(); ++d)
         {
             if(!(*d)->optional())
@@ -4046,7 +4088,7 @@ Slice::Gen::TypesVisitor::visitStructEnd(const StructPtr& p)
 
         if(classMembers.size() != 0)
         {
-            writePatcher(out, package, classMembers, _stream);
+            writePatcher(out, package, classMembers, DataMemberList(), _stream);
         }
 
         out << sp << nl << "public void" << nl << "__read(IceInternal.BasicStream __is)";
diff --git a/cpp/src/slice2java/Gen.h b/cpp/src/slice2java/Gen.h
index e4764171406..e4def10d75e 100644
--- a/cpp/src/slice2java/Gen.h
+++ b/cpp/src/slice2java/Gen.h
@@ -86,7 +86,7 @@ protected:
     //
     // Generate a patcher class.
     //
-    void writePatcher(::IceUtilInternal::Output&, const std::string&, const DataMemberList&, bool);
+    void writePatcher(::IceUtilInternal::Output&, const std::string&, const DataMemberList&, const DataMemberList&, bool);
 
     //
     // Generate dispatch and marshalling methods for a class or interface.
diff --git a/cpp/src/slice2js/Gen.cpp b/cpp/src/slice2js/Gen.cpp
index 0f14a9be27e..a7fdc7913b2 100644
--- a/cpp/src/slice2js/Gen.cpp
+++ b/cpp/src/slice2js/Gen.cpp
@@ -170,23 +170,24 @@ Slice::JsVisitor::writeMarshalUnmarshalParams(const ParamDeclList& params, const
 }
 
 void
-Slice::JsVisitor::writeMarshalDataMembers(const DataMemberList& dataMembers)
+Slice::JsVisitor::writeMarshalDataMembers(const DataMemberList& dataMembers, const DataMemberList& optionalMembers)
 {
     for(DataMemberList::const_iterator q = dataMembers.begin(); q != dataMembers.end(); ++q)
     {
-        if((*q)->optional())
-        {
-            writeOptionalMarshalUnmarshalCode(_out, (*q)->type(), "this." + fixId((*q)->name()), (*q)->tag(), true);
-        }
-        else
+        if(!(*q)->optional())
         {
             writeMarshalUnmarshalCode(_out, (*q)->type(), "this." + fixId((*q)->name()), true);
         }
     }
+    
+    for(DataMemberList::const_iterator q = optionalMembers.begin(); q != optionalMembers.end(); ++q)
+    {
+        writeOptionalMarshalUnmarshalCode(_out, (*q)->type(), "this." + fixId((*q)->name()), (*q)->tag(), true);
+    }
 }
 
 void
-Slice::JsVisitor::writeUnmarshalDataMembers(const DataMemberList& dataMembers)
+Slice::JsVisitor::writeUnmarshalDataMembers(const DataMemberList& dataMembers, const DataMemberList& optionalMembers)
 {
     for(DataMemberList::const_iterator q = dataMembers.begin(); q != dataMembers.end(); ++q)
     {
@@ -199,15 +200,16 @@ Slice::JsVisitor::writeUnmarshalDataMembers(const DataMemberList& dataMembers)
 
     for(DataMemberList::const_iterator q = dataMembers.begin(); q != dataMembers.end(); ++q)
     {
-        if((*q)->optional())
-        {
-            writeOptionalMarshalUnmarshalCode(_out, (*q)->type(), "this." + fixId((*q)->name()), (*q)->tag(), false);
-        }
-        else
+        if(!(*q)->optional())
         {
             writeMarshalUnmarshalCode(_out, (*q)->type(), "this." + fixId((*q)->name()), false);
         }
     }
+    
+    for(DataMemberList::const_iterator q = optionalMembers.begin(); q != optionalMembers.end(); ++q)
+    {
+        writeOptionalMarshalUnmarshalCode(_out, (*q)->type(), "this." + fixId((*q)->name()), (*q)->tag(), false);
+    }
 }
 
 void
@@ -1220,11 +1222,11 @@ Slice::Gen::TypesVisitor::visitClassDefStart(const ClassDefPtr& p)
         {
             _out << nl << "function(__os)";
             _out << sb;
-            writeMarshalDataMembers(dataMembers);
+            writeMarshalDataMembers(dataMembers, optionalMembers);
             _out << eb << ",";
             _out << nl << "function(__is)";
             _out << sb;
-            writeUnmarshalDataMembers(dataMembers);
+            writeUnmarshalDataMembers(dataMembers, optionalMembers);
             _out << eb << ",";
             _out << nl;
         }
@@ -1543,6 +1545,7 @@ Slice::Gen::TypesVisitor::visitExceptionStart(const ExceptionPtr& p)
 
     const DataMemberList allDataMembers = p->allDataMembers();
     const DataMemberList dataMembers = p->dataMembers();
+    const DataMemberList optionalMembers = p->orderedOptionalDataMembers();
 
     vector<string> allParamNames;
     for(DataMemberList::const_iterator q = allDataMembers.begin(); q != allDataMembers.end(); ++q)
@@ -1584,11 +1587,11 @@ Slice::Gen::TypesVisitor::visitExceptionStart(const ExceptionPtr& p)
         {
             _out << nl << "function(__os)";
             _out << sb;
-            writeMarshalDataMembers(dataMembers);
+            writeMarshalDataMembers(dataMembers, optionalMembers);
             _out << eb << ",";
             _out << nl << "function(__is)";
             _out << sb;
-            writeUnmarshalDataMembers(dataMembers);
+            writeUnmarshalDataMembers(dataMembers, optionalMembers);
             _out << eb;
         }
         else
@@ -1653,11 +1656,11 @@ Slice::Gen::TypesVisitor::visitStructStart(const StructPtr& p)
         _out << ",";
         _out << nl << "function(__os)";
         _out << sb;
-        writeMarshalDataMembers(dataMembers);
+        writeMarshalDataMembers(dataMembers, DataMemberList());
         _out << eb << ",";
         _out << nl << "function(__is)";
         _out << sb;
-        writeUnmarshalDataMembers(dataMembers);
+        writeUnmarshalDataMembers(dataMembers, DataMemberList());
         _out << eb << "," << nl << p->minWireSize() << ", " << nl << (p->isVariableLength() ? "false" : "true");
         _out.dec();
         _out << ");";
diff --git a/cpp/src/slice2js/Gen.h b/cpp/src/slice2js/Gen.h
index 3c9781471c7..4d17cf75a90 100644
--- a/cpp/src/slice2js/Gen.h
+++ b/cpp/src/slice2js/Gen.h
@@ -26,8 +26,8 @@ protected:
 
     void writeMarshalUnmarshalParams(const ParamDeclList&, const OperationPtr&, bool);
     void writePostUnmarshalParams(const ParamDeclList&, const OperationPtr&);
-    void writeMarshalDataMembers(const DataMemberList&);
-    void writeUnmarshalDataMembers(const DataMemberList&);
+    void writeMarshalDataMembers(const DataMemberList&, const DataMemberList&);
+    void writeUnmarshalDataMembers(const DataMemberList&, const DataMemberList&);
     void writeInitDataMembers(const DataMemberList&, const std::string&);
 
     virtual std::vector<std::string> getParams(const OperationPtr&);