mailbox: mediatek: Add check for possible failure of kzalloc

[sfrench/cifs-2.6.git] / drivers / mailbox / mtk-cmdq-mailbox.c

diff --git a/drivers/mailbox/mtk-cmdq-mailbox.c b/drivers/mailbox/mtk-cmdq-mailbox.c
index aec46d5d3506181a7dc34bee920d80645c2c95f0..f7cc29c00302a493ee4136fe41273775b706c35a 100644 (file)
--- a/drivers/mailbox/mtk-cmdq-mailbox.c
+++ b/drivers/mailbox/mtk-cmdq-mailbox.c
@@ -363,6 +363,9 @@ static int cmdq_mbox_send_data(struct mbox_chan *chan, void *data)
        WARN_ON(cmdq->suspended);
 
        task = kzalloc(sizeof(*task), GFP_ATOMIC);
+       if (!task)
+               return -ENOMEM;
+
        task->cmdq = cmdq;
        INIT_LIST_HEAD(&task->list_entry);
        task->pa_base = pkt->pa_base;