diff options
| author | Vasiliy Kulikov <segooon@gmail.com> | 2010-09-19 11:36:23 +0400 |
|---|---|---|
| committer | Greg Kroah-Hartman <gregkh@suse.de> | 2010-09-20 17:03:41 -0700 |
| commit | b1f5f54e91e30ecd5ac670edc8bb2c0ee5990d0f (patch) | |
| tree | 553414d531108b28cdbd0a7dc2ad2e3e4c0d5d9f | |
| parent | 83ef41f0858b48cce4f8ca23f674ec55d50bf47d (diff) | |
| download | linux-b1f5f54e91e30ecd5ac670edc8bb2c0ee5990d0f.tar.bz2 | |
staging: keucr: check kmalloc() return value
kmalloc() may fail, if so return error code.
Signed-off-by: Vasiliy Kulikov <segooon@gmail.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@suse.de>
| -rw-r--r-- | drivers/staging/keucr/init.c | 2 | ||||
| -rw-r--r-- | drivers/staging/keucr/msscsi.c | 4 | ||||
| -rw-r--r-- | drivers/staging/keucr/smscsi.c | 4 |
3 files changed, 10 insertions, 0 deletions
diff --git a/drivers/staging/keucr/init.c b/drivers/staging/keucr/init.c index 2ae129bfb608..1934805844f2 100644 --- a/drivers/staging/keucr/init.c +++ b/drivers/staging/keucr/init.c @@ -300,6 +300,8 @@ int ENE_LoadBinCode(struct us_data *us, BYTE flag) return USB_STOR_TRANSPORT_GOOD; buf = kmalloc(0x800, GFP_KERNEL); + if (buf == NULL) + return USB_STOR_TRANSPORT_ERROR; switch ( flag ) { // For SD diff --git a/drivers/staging/keucr/msscsi.c b/drivers/staging/keucr/msscsi.c index b908a2383966..ad0c5c629935 100644 --- a/drivers/staging/keucr/msscsi.c +++ b/drivers/staging/keucr/msscsi.c @@ -168,6 +168,8 @@ int MS_SCSI_Read(struct us_data *us, struct scsi_cmnd *srb) DWORD blkno; buf = kmalloc(blenByte, GFP_KERNEL); + if (buf == NULL) + return USB_STOR_TRANSPORT_ERROR; result = ENE_LoadBinCode(us, MS_RW_PATTERN); if (result != USB_STOR_XFER_GOOD) @@ -271,6 +273,8 @@ int MS_SCSI_Write(struct us_data *us, struct scsi_cmnd *srb) WORD len, oldphy, newphy; buf = kmalloc(blenByte, GFP_KERNEL); + if (buf == NULL) + return USB_STOR_TRANSPORT_ERROR; usb_stor_set_xfer_buf(us, buf, blenByte, srb, FROM_XFER_BUF); result = ENE_LoadBinCode(us, MS_RW_PATTERN); diff --git a/drivers/staging/keucr/smscsi.c b/drivers/staging/keucr/smscsi.c index 43e32c6ded90..62116869b38a 100644 --- a/drivers/staging/keucr/smscsi.c +++ b/drivers/staging/keucr/smscsi.c @@ -145,6 +145,8 @@ int SM_SCSI_Read(struct us_data *us, struct scsi_cmnd *srb) return USB_STOR_TRANSPORT_ERROR; buf = kmalloc(blenByte, GFP_KERNEL); + if (buf == NULL) + return USB_STOR_TRANSPORT_ERROR; result = Media_D_ReadSector(us, bn, blen, buf); usb_stor_set_xfer_buf(us, buf, blenByte, srb, TO_XFER_BUF); kfree(buf); @@ -175,6 +177,8 @@ int SM_SCSI_Write(struct us_data *us, struct scsi_cmnd *srb) return USB_STOR_TRANSPORT_ERROR; buf = kmalloc(blenByte, GFP_KERNEL); + if (buf == NULL) + return USB_STOR_TRANSPORT_ERROR; usb_stor_set_xfer_buf(us, buf, blenByte, srb, FROM_XFER_BUF); result = Media_D_CopySector(us, bn, blen, buf); kfree(buf); |