public inbox for linux-kernel@vger.kernel.org
 help / color / mirror / Atom feed
* [PATCH] usb: storage: add error handling for kcalloc
@ 2018-06-11  8:52 Zhouyang Jia
  2018-06-12 14:31 ` Alan Stern
  2018-06-14 13:29 ` [PATCH v2] " Zhouyang Jia
  0 siblings, 2 replies; 7+ messages in thread
From: Zhouyang Jia @ 2018-06-11  8:52 UTC (permalink / raw)
  Cc: Zhouyang Jia, Alan Stern, Greg Kroah-Hartman, linux-usb,
	usb-storage, linux-kernel

When kcalloc fails, the lack of error-handling code may
cause unexpected results.

This patch adds error-handling code after calling kcalloc.

Signed-off-by: Zhouyang Jia <jiazhouyang09@gmail.com>
---
 drivers/usb/storage/alauda.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/drivers/usb/storage/alauda.c b/drivers/usb/storage/alauda.c
index 900591d..c56355c 100644
--- a/drivers/usb/storage/alauda.c
+++ b/drivers/usb/storage/alauda.c
@@ -437,6 +437,11 @@ static int alauda_init_media(struct us_data *us)
 		+ MEDIA_INFO(us).blockshift + MEDIA_INFO(us).pageshift);
 	MEDIA_INFO(us).pba_to_lba = kcalloc(num_zones, sizeof(u16*), GFP_NOIO);
 	MEDIA_INFO(us).lba_to_pba = kcalloc(num_zones, sizeof(u16*), GFP_NOIO);
+	if ((MEDIA_INFO(us).pba_to_lba == NULL)
+		|| (MEDIA_INFO(us).lba_to_pba == NULL)) {
+		pr_warn("alauda_init_media: memory allocation failed\n");
+		return USB_STOR_TRANSPORT_ERROR;
+	}
 
 	if (alauda_reset_media(us) != USB_STOR_XFER_GOOD)
 		return USB_STOR_TRANSPORT_ERROR;
-- 
2.7.4

^ permalink raw reply related	[flat|nested] 7+ messages in thread

end of thread, other threads:[~2018-06-28 11:01 UTC | newest]

Thread overview: 7+ messages (download: mbox.gz follow: Atom feed
-- links below jump to the message on this page --
2018-06-11  8:52 [PATCH] usb: storage: add error handling for kcalloc Zhouyang Jia
2018-06-12 14:31 ` Alan Stern
2018-06-14 13:29 ` [PATCH v2] " Zhouyang Jia
2018-06-14 14:58   ` Alan Stern
2018-06-25 12:33   ` Greg Kroah-Hartman
2018-06-25 15:22     ` Alan Stern
2018-06-28 10:52       ` Greg Kroah-Hartman

This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox