diff --git a/src/pool.c b/src/pool.c index a23641145..146f118b4 100644 --- a/src/pool.c +++ b/src/pool.c @@ -194,6 +194,9 @@ char *git_pool_strndup(git_pool *pool, const char *str, size_t n) assert(pool && str && pool->item_size == sizeof(char)); + if ((uint32_t)(n + 1) < n) + return NULL; + if ((ptr = git_pool_malloc(pool, (uint32_t)(n + 1))) != NULL) { memcpy(ptr, str, n); ptr[n] = '\0'; diff --git a/tests/core/pool.c b/tests/core/pool.c index 3073c4a45..7a8b2dea6 100644 --- a/tests/core/pool.c +++ b/tests/core/pool.c @@ -139,7 +139,11 @@ void test_core_pool__strndup_limit(void) git_pool p; cl_git_pass(git_pool_init(&p, 1, 100)); - cl_assert(git_pool_strndup(&p, "foo", -1) == NULL); + /* ensure 64 bit doesn't overflow */ + cl_assert(git_pool_strndup(&p, "foo", (size_t)-1) == NULL); + + /* ensure 32 bit doesn't overflow */ + cl_assert(git_pool_strndup(&p, "bar", 0xfffffffful + 32) == NULL); git_pool_clear(&p); }