1 files changed, 7 insertions, 2 deletions

diff --git a/drivers/gpu/drm/radeon/atom.c b/drivers/gpu/drm/radeon/atom.c
index 5ce9bf51a8de..46a9c3772850 100644
--- a/drivers/gpu/drm/radeon/atom.c
+++ b/drivers/gpu/drm/radeon/atom.c
@@ -1238,6 +1238,8 @@ static int atom_iio_len[] = { 1, 2, 3, 3, 3, 3, 4, 4, 4, 3 };
 static void atom_index_iio(struct atom_context *ctx, int base)
 {
         ctx->iio = kzalloc(2 * 256, GFP_KERNEL);
+        if (!ctx->iio)
+                return;
         while (CU8(base) == ATOM_IIO_START) {
                 ctx->iio[CU8(base + 1)] = base + 2;
                 base += 2;
@@ -1287,6 +1289,10 @@ struct atom_context *atom_parse(struct card_info *card, void *bios)
         ctx->cmd_table = CU16(base + ATOM_ROM_CMD_PTR);
         ctx->data_table = CU16(base + ATOM_ROM_DATA_PTR);
         atom_index_iio(ctx, CU16(ctx->data_table + ATOM_DATA_IIO_PTR) + 4);
+        if (!ctx->iio) {
+                atom_destroy(ctx);
+                return NULL;
+        }
 
         str = CSTR(CU16(base + ATOM_ROM_MSG_PTR));
         while (*str && ((*str == '\n') || (*str == '\r')))
@@ -1335,8 +1341,7 @@ int atom_asic_init(struct atom_context *ctx)
 
 void atom_destroy(struct atom_context *ctx)
 {
-        if (ctx->iio)
-                kfree(ctx->iio);
+        kfree(ctx->iio);
         kfree(ctx);
 }