diff options
-rw-r--r-- | src/vm/diagnosticsprotocol.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/vm/diagnosticsprotocol.h b/src/vm/diagnosticsprotocol.h index a565030447..657e669033 100644 --- a/src/vm/diagnosticsprotocol.h +++ b/src/vm/diagnosticsprotocol.h @@ -203,6 +203,8 @@ namespace DiagnosticsIpc } CONTRACTL_END; + m_Header = header; + return FlattenImpl<T>(payload); }; @@ -218,6 +220,8 @@ namespace DiagnosticsIpc } CONTRACTL_END; + m_Header = header; + return FlattenImpl<T>(payload); }; @@ -375,6 +379,11 @@ namespace DiagnosticsIpc return false; } + if (m_Header.Size < sizeof(IpcHeader)) + { + return false; + } + m_Size = m_Header.Size; // Then read out payload to buffer |