buffer: add buffer_grow

1 files changed, 2 insertions, 2 deletions

diff --git a/register.c b/register.c
index 2e630b0..69409c1 100644
--- a/register.c
+++ b/register.c
@@ -61,7 +61,7 @@ bool register_put_range(Vis *vis, Register *reg, Text *txt, Filerange *range) {
 	case REGISTER_NORMAL:
 	{
 		size_t len = text_range_size(range);
-		if (!buffer_reserve(&reg->buf, len+1))
+		if (len == SIZE_MAX || !buffer_reserve(&reg->buf, len+1))
 			return false;
 		reg->buf.len = text_bytes_get(txt, range->start, len, reg->buf.data);
 		return buffer_append(&reg->buf, "\0", 1);
@@ -91,7 +91,7 @@ bool register_append_range(Register *reg, Text *txt, Filerange *range) {
 	case REGISTER_NORMAL:
 	{
 		size_t len = text_range_size(range);
-		if (!buffer_reserve(&reg->buf, reg->buf.len + len + 1))
+		if (!buffer_grow(&reg->buf, len+1))
 			return false;
 		if (reg->buf.len > 0 && reg->buf.data[reg->buf.len-1] == '\0')
 			reg->buf.len--;