Staging: hv: return correct error values in Connection.c
authorBill Pemberton <wfp5p@virginia.edu>
Wed, 5 May 2010 19:27:32 +0000 (15:27 -0400)
committerGreg Kroah-Hartman <gregkh@suse.de>
Tue, 11 May 2010 18:36:13 +0000 (11:36 -0700)
Also check the kzalloc call return value.

Signed-off-by: Bill Pemberton <wfp5p@virginia.edu>
Cc: Hank Janssen <hjanssen@microsoft.com>
Cc: Haiyang Zhang <haiyangz@microsoft.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@suse.de>
drivers/staging/hv/Connection.c

index dbf00560e0ac765340405b6421df1b4917d3e25f..1e4111412ab6e639d971c518320ce97b83b9a351 100644 (file)
@@ -93,7 +93,7 @@ int VmbusConnect(void)
                          sizeof(struct vmbus_channel_initiate_contact),
                          GFP_KERNEL);
        if (msgInfo == NULL) {
-               ret = -1;
+               ret = -ENOMEM;
                goto Cleanup;
        }
 
@@ -195,6 +195,8 @@ int VmbusDisconnect(void)
                return -1;
 
        msg = kzalloc(sizeof(struct vmbus_channel_message_header), GFP_KERNEL);
+       if (!msg)
+               return -ENOMEM;
 
        msg->MessageType = ChannelMessageUnload;