Commit b1f5f54e authored by Vasiliy Kulikov's avatar Vasiliy Kulikov Committed by Greg Kroah-Hartman
Browse files

staging: keucr: check kmalloc() return value 



kmalloc() may fail, if so return error code.

Signed-off-by: default avatarVasiliy Kulikov <segooon@gmail.com>
Signed-off-by: default avatarGreg Kroah-Hartman <gregkh@suse.de>
parent 83ef41f0

drivers/staging/keucr/init.c

+2 −0
Original line number Diff line number Diff line
@@ -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

drivers/staging/keucr/msscsi.c

+4 −0
Original line number Diff line number Diff line
@@ -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);

drivers/staging/keucr/smscsi.c

+4 −0
Original line number Diff line number Diff line
@@ -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);