Prevent hang when caller does not pass large enough buffer.

This commit is contained in:
Jim Norman 2006-09-29 22:22:26 +00:00
parent aa244fe2ab
commit c62ce742de

View File

@ -2268,22 +2268,21 @@ int32_t ipc_ReadBinaryKey
}
else
{
//buffer allocated by application is not sufficient to hold the data.
//buffer allocated by application is not sufficient to hold the data.
*bytesRequired = dataLen;
{
// Cleanup the channel by reading the remaining and return error.
int n;
n = dataLen + MSG_DWORD_LEN;
n = dataLen;
while(n)
{
int bytes = IPC_READ(ssHandle->platHandle, gpReplyBuf, MIN_REPLY_BUF_LEN);
if( bytes > 0)
n -= bytes;
int bytesRead = IPC_READ(ssHandle->platHandle, gpReplyBuf, n);
if( bytesRead > 0)
n -= bytesRead;
else
break;
}
// Read the sscs return code also.
// IPC_READ(ssHandle->platHandle, (Byte *) &sockReturn, MSG_DWORD_LEN);
retCode = NSSCS_E_ENUM_BUFF_TOO_SHORT;
break;
}