Robert Shearman : rpcrt4: Fix conformant varying structs to comply to the DCE/RPC spec.

[Alexandre Julliard]

Module: wine
Branch: refs/heads/master
Commit: 124acc9b67fddfb149d43e2ac4379d44f0ba9f58
URL:    http://source.winehq.org/git/?p=wine.git;a=commit;h=124acc9b67fddfb149d43e2a...

Author: Robert Shearman <rob(a)codeweavers.com>
Date:   Mon May 15 16:56:55 2006 +0100

rpcrt4: Fix conformant varying structs to comply to the DCE/RPC spec.

---

 dlls/rpcrt4/ndr_marshall.c |   43 ++++++++++++++++++++++++++++---------------
 1 files changed, 28 insertions(+), 15 deletions(-)

diff --git a/dlls/rpcrt4/ndr_marshall.c b/dlls/rpcrt4/ndr_marshall.c
index 4b7f6e0..4f5d336 100644
--- a/dlls/rpcrt4/ndr_marshall.c
+++ b/dlls/rpcrt4/ndr_marshall.c
@@ -2894,15 +2894,21 @@ unsigned char *  WINAPI NdrConformantVar
     }
 
     WriteConformance(pStubMsg);
-    WriteVariance(pStubMsg);
 
     ALIGN_POINTER(pStubMsg->Buffer, pCVStructFormat->alignment + 1);
 
     TRACE("memory_size = %d\n", pCVStructFormat->memory_size);
 
+    /* write constant sized part */
     pStubMsg->BufferMark = pStubMsg->Buffer;
-    memcpy(pStubMsg->Buffer, pMemory, pCVStructFormat->memory_size + pStubMsg->MaxCount * esize);
-    pStubMsg->Buffer += pCVStructFormat->memory_size + pStubMsg->MaxCount * esize;
+    memcpy(pStubMsg->Buffer, pMemory, pCVStructFormat->memory_size);
+    pStubMsg->Buffer += pCVStructFormat->memory_size;
+
+    WriteVariance(pStubMsg);
+
+    /* write array part */
+    memcpy(pStubMsg->Buffer, pMemory + pCVStructFormat->memory_size, pStubMsg->ActualCount * esize);
+    pStubMsg->Buffer += pStubMsg->ActualCount * esize;
 
     EmbeddedPointerMarshall(pStubMsg, pMemory, pFormat);
 
@@ -2942,7 +2948,6 @@ unsigned char *  WINAPI NdrConformantVar
     case RPC_FC_CVARRAY:
         esize = *(const WORD*)(pCVArrayFormat+2);
         pCVArrayFormat = ReadConformance(pStubMsg, pCVArrayFormat + 4);
-        pCVArrayFormat = ReadVariance(pStubMsg, pCVArrayFormat);
         break;
     case RPC_FC_C_CSTRING:
         esize = sizeof(char);
@@ -2975,10 +2980,17 @@ unsigned char *  WINAPI NdrConformantVar
         *ppMemory = NdrAllocate(pStubMsg, size);
     }
 
-    /* now copy the data */
+    /* copy the constant data */
     pStubMsg->BufferMark = pStubMsg->Buffer;
-    memcpy(*ppMemory, pStubMsg->Buffer, pCVStructFormat->memory_size + pStubMsg->MaxCount * esize);
-    pStubMsg->Buffer += pCVStructFormat->memory_size + pStubMsg->MaxCount * esize;
+    memcpy(*ppMemory, pStubMsg->Buffer, pCVStructFormat->memory_size);
+    pStubMsg->Buffer += pCVStructFormat->memory_size;
+
+    pCVArrayFormat = ReadVariance(pStubMsg, pCVArrayFormat);
+
+    /* copy the array data */
+    memcpy(*ppMemory + pCVStructFormat->memory_size, pStubMsg->Buffer,
+           pStubMsg->ActualCount * esize);
+    pStubMsg->Buffer += pStubMsg->ActualCount * esize;
 
     if (cvarray_type == RPC_FC_C_CSTRING)
         TRACE("string=%s\n", debugstr_a((char *)(*ppMemory + pCVStructFormat->memory_size)));
@@ -3050,13 +3062,14 @@ void WINAPI NdrConformantVaryingStructBu
     }
 
     SizeConformance(pStubMsg);
-    SizeVariance(pStubMsg);
 
     ALIGN_LENGTH(pStubMsg->BufferLength, pCVStructFormat->alignment + 1);
 
     TRACE("memory_size = %d\n", pCVStructFormat->memory_size);
 
-    pStubMsg->BufferLength += pCVStructFormat->memory_size + esize * pStubMsg->MaxCount;
+    pStubMsg->BufferLength += pCVStructFormat->memory_size;
+    SizeVariance(pStubMsg);
+    pStubMsg->BufferLength += esize * pStubMsg->MaxCount;
 
     EmbeddedPointerBufferSize(pStubMsg, pMemory, pFormat);
 }
@@ -3071,7 +3084,6 @@ unsigned long WINAPI NdrConformantVaryin
     PFORMAT_STRING pCVArrayFormat;
     ULONG esize;
     unsigned char cvarray_type;
-    ULONG size;
 
     TRACE("(%p, %p)\n", pStubMsg, pFormat);
 
@@ -3091,7 +3103,6 @@ unsigned long WINAPI NdrConformantVaryin
     case RPC_FC_CVARRAY:
         esize = *(const WORD*)(pCVArrayFormat+2);
         pCVArrayFormat = ReadConformance(pStubMsg, pCVArrayFormat + 4);
-        pCVArrayFormat = ReadVariance(pStubMsg, pCVArrayFormat);
         break;
     case RPC_FC_C_CSTRING:
         esize = sizeof(char);
@@ -3117,13 +3128,15 @@ unsigned long WINAPI NdrConformantVaryin
 
     TRACE("memory_size = %d\n", pCVStructFormat->memory_size);
 
-    size = pCVStructFormat->memory_size + pStubMsg->MaxCount * esize;
-    pStubMsg->Buffer += size;
-    pStubMsg->MemorySize += size;
+    pStubMsg->Buffer += pCVStructFormat->memory_size;
+    pCVArrayFormat = ReadVariance(pStubMsg, pCVArrayFormat);
+    pStubMsg->Buffer += pCVStructFormat->memory_size + pStubMsg->ActualCount * esize;
+
+    pStubMsg->MemorySize += pCVStructFormat->memory_size + pStubMsg->MaxCount * esize;
 
     EmbeddedPointerMemorySize(pStubMsg, pFormat);
 
-    return size;
+    return pCVStructFormat->memory_size + pStubMsg->MaxCount * esize;
 }
 
 /***********************************************************************