4 files changed, 232 insertions, 6 deletions

diff --git a/src/libs/dutil/WixToolset.DUtil/inc/buffutil.h b/src/libs/dutil/WixToolset.DUtil/inc/buffutil.h
index 322209e6..116b67bd 100644
--- a/src/libs/dutil/WixToolset.DUtil/inc/buffutil.h
+++ b/src/libs/dutil/WixToolset.DUtil/inc/buffutil.h
@@ -9,9 +9,35 @@ extern "C" {
 
 // macro definitions
 
-#define ReleaseBuffer ReleaseMem
-#define ReleaseNullBuffer ReleaseNullMem
-#define BuffFree MemFree
+#define ReleaseBuffer(b) BuffFree(b)
+#define ReleaseNullBuffer(b) BuffFree(b)
+#define BuffFree(b) if (b.pbData) { MemFree(b.pbData); b.pbData = NULL; } b.cbData = 0
+
+
+// structs
+
+// A buffer that owns its data and must be freed with BuffFree().
+typedef struct _BUFF_BUFFER
+{
+    LPBYTE pbData;
+    SIZE_T cbData;
+} BUFF_BUFFER;
+
+// A read-only buffer with internal pointer that can be advanced for multiple reads.
+typedef struct _BUFF_READER
+{
+    LPCBYTE pbData;
+    SIZE_T cbData;
+
+    SIZE_T iBuffer;
+} BUFF_READER;
+
+// A write buffer that does not own its data.
+typedef struct _BUFF_WRITER
+{
+    LPBYTE *ppbData;
+    SIZE_T *pcbData;
+} BUFF_WRITER;
 
 
 // function declarations
@@ -53,6 +79,37 @@ HRESULT BuffReadStream(
     __deref_inout_bcount(*pcbStream) BYTE** ppbStream,
     __out SIZE_T* pcbStream
     );
+HRESULT BuffSkipExtraData(
+    __in SIZE_T cbExpectedSize,
+    __in SIZE_T cbActualSize,
+    __inout SIZE_T* piBuffer
+    );
+
+HRESULT BuffReaderReadNumber(
+    __in BUFF_READER* pReader,
+    __out DWORD* pdw
+    );
+HRESULT BuffReaderReadNumber64(
+    __in BUFF_READER* pReader,
+    __out DWORD64* pdw64
+    );
+HRESULT BuffReaderReadPointer(
+    __in BUFF_READER* pReader,
+    __out DWORD_PTR* pdw
+);
+HRESULT BuffReaderReadString(
+    __in BUFF_READER* pReader,
+    __deref_out_z LPWSTR* pscz
+    );
+HRESULT BuffReaderReadStringAnsi(
+    __in BUFF_READER* pReader,
+    __deref_out_z LPSTR* pscz
+    );
+HRESULT BuffReaderReadStream(
+    __in BUFF_READER* pReader,
+    __deref_inout_bcount(*pcbStream) BYTE** ppbStream,
+    __out SIZE_T* pcbStream
+    );
 
 HRESULT BuffWriteNumber(
     __deref_inout_bcount(*piBuffer) BYTE** ppbBuffer,
@@ -86,6 +143,58 @@ HRESULT BuffWriteStream(
     __in SIZE_T cbStream
     );
 
+HRESULT BuffWriteNumberToBuffer(
+    __in BUFF_BUFFER* pBuffer,
+    __in DWORD dw
+    );
+HRESULT BuffWriteNumber64ToBuffer(
+    __in BUFF_BUFFER* pBuffer,
+    __in DWORD64 dw64
+    );
+HRESULT BuffWritePointerToBuffer(
+    __in BUFF_BUFFER* pBuffer,
+    __in DWORD_PTR dw
+    );
+HRESULT BuffWriteStringToBuffer(
+    __in BUFF_BUFFER* pBuffer,
+    __in_z_opt LPCWSTR scz
+    );
+HRESULT BuffWriteStringAnsiToBuffer(
+    __in BUFF_BUFFER* pBuffer,
+    __in_z_opt LPCSTR scz
+    );
+HRESULT BuffWriteStreamToBuffer(
+    __in BUFF_BUFFER* pBuffer,
+    __in_bcount(cbStream) const BYTE* pbStream,
+    __in SIZE_T cbStream
+    );
+
+HRESULT BuffWriterWriteNumber(
+    __in BUFF_WRITER* pWriter,
+    __in DWORD dw
+    );
+HRESULT BuffWriterWriteNumber64(
+    __in BUFF_WRITER* pWriter,
+    __in DWORD64 dw64
+    );
+HRESULT BuffWriterWritePointer(
+    __in BUFF_WRITER* pWriter,
+    __in DWORD_PTR dw
+    );
+HRESULT BuffWriterWriteString(
+    __in BUFF_WRITER* pWriter,
+    __in_z_opt LPCWSTR scz
+    );
+HRESULT BuffWriterWriteStringAnsi(
+    __in BUFF_WRITER* pWriter,
+    __in_z_opt LPCSTR scz
+    );
+HRESULT BuffWriterWriteStream(
+    __in BUFF_WRITER* pWriter,
+    __in_bcount(cbStream) const BYTE* pbStream,
+    __in SIZE_T cbStream
+    );
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/src/libs/dutil/WixToolset.DUtil/inc/dlutil.h b/src/libs/dutil/WixToolset.DUtil/inc/dlutil.h
index 3e95103a..912c2732 100644
--- a/src/libs/dutil/WixToolset.DUtil/inc/dlutil.h
+++ b/src/libs/dutil/WixToolset.DUtil/inc/dlutil.h
@@ -27,6 +27,7 @@ typedef struct _DOWNLOAD_SOURCE
     LPWSTR sczUrl;
     LPWSTR sczUser;
     LPWSTR sczPassword;
+    LPWSTR sczAuthorizationHeader;
 } DOWNLOAD_SOURCE;
 
 typedef struct _DOWNLOAD_CACHE_CALLBACK
diff --git a/src/libs/dutil/WixToolset.DUtil/inc/dutil.h b/src/libs/dutil/WixToolset.DUtil/inc/dutil.h
index 4a172f46..7d8c7d81 100644
--- a/src/libs/dutil/WixToolset.DUtil/inc/dutil.h
+++ b/src/libs/dutil/WixToolset.DUtil/inc/dutil.h
@@ -41,6 +41,13 @@ HRESULT DAPI DutilInitialize(
 *******************************************************************/
 void DAPI DutilUninitialize();
 
+/*******************************************************************
+ DutilSizetToDword - safely convert SIZE_T to DWORD.
+
+ Returns
+   E_INVALIDARG - If SIZE_T value is greater than DWORD_MAX.
+********************************************************************/
+HRESULT DAPI DutilSizetToDword(SIZE_T sizet, DWORD* pdw);
 
 /********************************************************************
  DutilSuppressTraceErrorSource - tells dutil to skip calling
diff --git a/src/libs/dutil/WixToolset.DUtil/inc/pipeutil.h b/src/libs/dutil/WixToolset.DUtil/inc/pipeutil.h
index d16d768c..e92462d2 100644
--- a/src/libs/dutil/WixToolset.DUtil/inc/pipeutil.h
+++ b/src/libs/dutil/WixToolset.DUtil/inc/pipeutil.h
@@ -6,7 +6,9 @@
 extern "C" {
 #endif
 
-#define ReleasePipeHandle(h) if (h != INVALID_HANDLE_VALUE) { ::CloseHandle(h); }
+// macro definitions
+
+#define ReleasePipeHandle(h) if (h != INVALID_HANDLE_VALUE) { ::CloseHandle(h); h = INVALID_HANDLE_VALUE; }
 #define ReleasePipeMessage(pMsg) if (pMsg) { PipeFreeMessage(pMsg); }
 
 
@@ -27,6 +29,23 @@ typedef struct _PIPE_MESSAGE
     LPVOID pvData;
 } PIPE_MESSAGE;
 
+typedef struct _PIPE_RPC_HANDLE
+{
+    HANDLE hPipe;
+    CRITICAL_SECTION cs;
+
+    BOOL fInitialized;
+    BOOL fOwnHandle;
+} PIPE_RPC_HANDLE;
+
+typedef struct _PIPE_RPC_RESULT
+{
+    HRESULT hr;
+
+    DWORD cbData;
+    LPBYTE pbData;
+} PIPE_RPC_RESULT;
+
 
 // functions
 
@@ -72,17 +91,87 @@ DAPI_(HRESULT) PipeReadMessage(
 );
 
 /*******************************************************************
- PipeWriteMessage - writes a message to the pipe.
+ PipeRpcInitiailize - initializes a RPC pipe handle from a pipe handle.
 
 *******************************************************************/
-DAPI_(HRESULT) PipeWriteMessage(
+DAPI_(void) PipeRpcInitialize(
+    __in PIPE_RPC_HANDLE* phRpcPipe,
     __in HANDLE hPipe,
+    __in BOOL fTakeHandleOwnership
+);
+
+/*******************************************************************
+ PipeRpcInitialized - checks if a RPC pipe handle is initialized.
+
+*******************************************************************/
+DAPI_(BOOL) PipeRpcInitialized(
+    __in PIPE_RPC_HANDLE* phRpcPipe
+);
+
+/*******************************************************************
+ PipeRpcUninitiailize - uninitializes a RPC pipe handle.
+
+*******************************************************************/
+DAPI_(void) PipeRpcUninitiailize(
+    __in PIPE_RPC_HANDLE* phRpcPipe
+);
+
+/*******************************************************************
+ PipeRpcReadMessage - reads a message from the pipe. Free with
+    PipeFreeMessage().
+
+*******************************************************************/
+DAPI_(HRESULT) PipeRpcReadMessage(
+    __in PIPE_RPC_HANDLE* phRpcPipe,
+    __in PIPE_MESSAGE* pMsg
+);
+
+/*******************************************************************
+ PipeRpcRequest - sends message and reads a response over the pipe.
+    Free with PipeFreeRpcResult().
+
+*******************************************************************/
+DAPI_(HRESULT) PipeRpcRequest(
+    __in PIPE_RPC_HANDLE* phRpcPipe,
+    __in DWORD dwMessageType,
+    __in_bcount(cbArgs) LPVOID pbArgs,
+    __in SIZE_T cbArgs,
+    __in PIPE_RPC_RESULT* pResult
+);
+
+/*******************************************************************
+ PipeRpcResponse - sends response over the pipe.
+
+*******************************************************************/
+DAPI_(HRESULT) PipeRpcResponse(
+    __in PIPE_RPC_HANDLE* phPipe,
+    __in DWORD dwMessageType,
+    __in HRESULT hrResult,
+    __in_bcount(cbResult) LPVOID pvResult,
+    __in SIZE_T cbResult
+    );
+
+/*******************************************************************
+ PipeRpcWriteMessage - writes a message to the pipe.
+
+*******************************************************************/
+DAPI_(HRESULT) PipeRpcWriteMessage(
+    __in PIPE_RPC_HANDLE* phPipe,
     __in DWORD dwMessageType,
     __in_bcount_opt(cbData) LPVOID pvData,
     __in SIZE_T cbData
 );
 
 /*******************************************************************
+ PipeWriteDisconnect - writes a message to the pipe indicating the
+    client should disconnect.
+
+*******************************************************************/
+DAPI_(HRESULT) PipeWriteDisconnect(
+    __in HANDLE hPipe
+    );
+
+/*******************************************************************
  PipeFreeMessage - frees any memory allocated in PipeReadMessage.
 
 *******************************************************************/
@@ -91,14 +180,34 @@ DAPI_(void) PipeFreeMessage(
 );
 
 /*******************************************************************
+ PipeFreeRpcResult - frees any memory allocated in PipeRpcRequest.
+
+*******************************************************************/
+DAPI_(void) PipeFreeRpcResult(
+    __in PIPE_RPC_RESULT* pResult
+);
+
+/*******************************************************************
  PipeServerWaitForClientConnect - Called from the server process to
     wait for a client to connect back to the provided pipe.
 
 *******************************************************************/
 DAPI_(HRESULT) PipeServerWaitForClientConnect(
+    __in HANDLE hClientProcess,
     __in HANDLE hPipe
 );
 
+/*******************************************************************
+ PipeWriteMessage - writes a message to the pipe.
+
+*******************************************************************/
+DAPI_(HRESULT) PipeWriteMessage(
+    __in HANDLE hPipe,
+    __in DWORD dwMessageType,
+    __in_bcount_opt(cbData) LPVOID pvData,
+    __in SIZE_T cbData
+);
+
 #ifdef __cplusplus
 }
 #endif