nvme: apple: Check memalign return value

memalign returns NULL if it fails.
This commit ensures that we handle this failure before
filling the buffer with 0s.

Signed-off-by: Francois Berder <fberder@outlook.fr>
Reviewed-by: Neil Armstrong <neil.armstrong@linaro.org>
Link: https://patch.msgid.link/BESP194MB280542535B098A33C8A815EEDA3A2@BESP194MB2805.EURP194.PROD.OUTLOOK.COM
Signed-off-by: Neil Armstrong <neil.armstrong@linaro.org>

diff --git a/drivers/nvme/nvme_apple.c b/drivers/nvme/nvme_apple.c
index 3e7d95c2b54667cf0228bd98df18525bd713f9c6..e674eda83445027f739361e0960f5b898728e663 100644 (file)
--- a/drivers/nvme/nvme_apple.c
+++ b/drivers/nvme/nvme_apple.c
@@ -88,6 +88,9 @@ static int apple_nvme_setup_queue(struct nvme_queue *nvmeq)
        }
 
        priv->tcbs[nvmeq->qid] = (void *)memalign(4096, ANS_NVMMU_TCB_SIZE);
+       if (!priv->tcbs[nvmeq->qid])
+               return -ENOMEM;
+
        memset((void *)priv->tcbs[nvmeq->qid], 0, ANS_NVMMU_TCB_SIZE);
 
        switch (nvmeq->qid) {