From: Bill Pemberton <[email protected]>

Signed-off-by: Bill Pemberton <[email protected]>
Cc: Hank Janssen <[email protected]>
Cc: Haiyang Zhang <[email protected]>
Signed-off-by: Greg Kroah-Hartman <[email protected]>
---
 drivers/staging/hv/Channel.c |    7 ++++---
 1 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/drivers/staging/hv/Channel.c b/drivers/staging/hv/Channel.c
index bd1a336..0a9ca33 100644
--- a/drivers/staging/hv/Channel.c
+++ b/drivers/staging/hv/Channel.c
@@ -793,7 +793,8 @@ int VmbusChannelSendPacketPageBuffer(struct vmbus_channel 
*Channel,
 
        DPRINT_ENTER(VMBUS);
 
-       ASSERT(PageCount <= MAX_PAGE_BUFFER_COUNT);
+       if (PageCount > MAX_PAGE_BUFFER_COUNT)
+               return -EINVAL;
 
        DumpVmbusChannel(Channel);
 
@@ -864,8 +865,8 @@ int VmbusChannelSendPacketMultiPageBuffer(struct 
vmbus_channel *Channel,
        DPRINT_DBG(VMBUS, "data buffer - offset %u len %u pfn count %u",
                   MultiPageBuffer->Offset, MultiPageBuffer->Length, PfnCount);
 
-       ASSERT(PfnCount > 0);
-       ASSERT(PfnCount <= MAX_MULTIPAGE_BUFFER_COUNT);
+       if ((PfnCount < 0) || (PfnCount > MAX_MULTIPAGE_BUFFER_COUNT))
+               return -EINVAL;
 
        /*
         * Adjust the size down since VMBUS_CHANNEL_PACKET_MULITPAGE_BUFFER is
-- 
1.7.0.3

_______________________________________________
devel mailing list
[email protected]
http://driverdev.linuxdriverproject.org/mailman/listinfo/devel

Reply via email to