off-by-one bugfixes

This commit is contained in:
Willem Toorop 2015-12-23 17:38:36 +01:00
parent fbae577a54
commit 29b033c14c
1 changed files with 4 additions and 3 deletions

View File

@ -382,8 +382,9 @@ getdns_rr_dict2str(
if (r != GETDNS_RETURN_GOOD && r != GETDNS_RETURN_NEED_MORE_SPACE) if (r != GETDNS_RETURN_GOOD && r != GETDNS_RETURN_NEED_MORE_SPACE)
return r; return r;
if (!(buf = malloc(buf_len + 1))) buf_len += 1;
if (!(buf = malloc(buf_len)))
return GETDNS_RETURN_MEMORY_ERROR; return GETDNS_RETURN_MEMORY_ERROR;
if (!r) if (!r)
@ -460,7 +461,7 @@ getdns_rr_dict2str_scan(
r = GETDNS_RETURN_NEED_MORE_SPACE; r = GETDNS_RETURN_NEED_MORE_SPACE;
} else { } else {
*str_len = sz; *str_len = sz;
*str = 0; **str = 0;
} }
if (buf != buf_spc) if (buf != buf_spc)
GETDNS_FREE(rr_dict->mf, buf); GETDNS_FREE(rr_dict->mf, buf);