pes: new_block() fails if undefined color is referenced

Fixes crash when the file's color table is too short.

Signed-off-by: Max Kellermann <max.kellermann@gmail.com>

1 files changed, 6 insertions, 1 deletions

diff --git a/pes.c b/pes.c
index 1e6b0b3..07b2e0d 100644
--- a/pes.c
+++ b/pes.c
@@ -161,10 +161,15 @@ static struct pes_block *new_block(struct pes *pes)
 	struct pes_block *block = calloc(1, sizeof(*block));
 
 	if (block) {
+		block->color = my_colors[pes->nr_colors++];
+		if (!block->color) {
+			free(block);
+			return NULL;
+		}
+
 		struct pes_block **pp = pes->last ? &pes->last->next : &pes->blocks;
 		*pp = block;
 		pes->last = block;
-		block->color = my_colors[pes->nr_colors++];
 	}
 	return block;
 }