staging: tidspbridge: check return code of kzalloc

Function kzalloc may fail. Check for it.

Signed-off-by: Vasiliy Kulikov <segooon@gmail.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@suse.de>

diff --git a/drivers/staging/tidspbridge/rmgr/dbdcd.c b/drivers/staging/tidspbridge/rmgr/dbdcd.c
index f71e8606f953082b8598d70404e4e38b11fcfa95..3581a55ed4dda6e552638ef8d79d3b0f9025696f 100644 (file)
--- a/drivers/staging/tidspbridge/rmgr/dbdcd.c
+++ b/drivers/staging/tidspbridge/rmgr/dbdcd.c
@@ -487,6 +487,10 @@ int dcd_get_object_def(struct dcd_manager *hdcd_mgr,
 
        /* Allocate zeroed buffer. */
        psz_coff_buf = kzalloc(ul_len + 4, GFP_KERNEL);
+       if (psz_coff_buf == NULL) {
+               status = -ENOMEM;
+               goto func_end;
+       }
 #ifdef _DB_TIOMAP
        if (strstr(dcd_key->path, "iva") == NULL) {
                /* Locate section by objectID and read its content. */
@@ -571,6 +575,10 @@ int dcd_get_objects(struct dcd_manager *hdcd_mgr,
 
        /* Allocate zeroed buffer. */
        psz_coff_buf = kzalloc(ul_len + 4, GFP_KERNEL);
+       if (psz_coff_buf == NULL) {
+               status = -ENOMEM;
+               goto func_cont;
+       }
 #ifdef _DB_TIOMAP
        if (strstr(sz_coff_path, "iva") == NULL) {
                /* Locate section by objectID and read its content. */