staging: tidspbridge: check return code of kzalloc

[net-next-2.6.git] / drivers / staging / tidspbridge / rmgr / dbdcd.c

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. */