From ed75535fc24217c51f900d42385309c8c8b36cc3 Mon Sep 17 00:00:00 2001 From: Ben Harris Date: Tue, 10 Jan 2023 00:19:02 +0000 Subject: [PATCH] Last-ditch maximum size limit for Map This makes sure that width * height <= INT_MAX, which it rather needs to be. Also a similar check in decode_params when defaulting the number of regions. --- map.c | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/map.c b/map.c index e55bf7b..553a683 100644 --- a/map.c +++ b/map.c @@ -14,6 +14,7 @@ #include #include #include +#include #include #include "puzzles.h" @@ -180,7 +181,9 @@ static void decode_params(game_params *params, char const *string) params->n = atoi(p); while (*p && (*p == '.' || isdigit((unsigned char)*p))) p++; } else { - params->n = params->w * params->h / 8; + if (params->h > 0 && params->w > 0 && + params->w <= INT_MAX / params->h) + params->n = params->w * params->h / 8; } if (*p == 'd') { int i; @@ -252,6 +255,8 @@ static const char *validate_params(const game_params *params, bool full) { if (params->w < 2 || params->h < 2) return "Width and height must be at least two"; + if (params->w > INT_MAX / params->h) + return "Width times height must not be unreasonably large"; if (params->n < 5) return "Must have at least five regions"; if (params->n > params->w * params->h)