80486DX2-66/gists
1
0
Fork 0
mirror of https://gitlab.com/80486DX2-66/gists synced 2024-12-27 05:40:17 +05:30
Code

asprintf.c: fix use of malloc

Browse Source 
1. Pass `size_t` variable to `malloc`
2. Multiply the count of bytes to allocate by `sizeof(char)`
This commit is contained in:
Intel A80486DX2-66 2024-01-28 13:49:18 +03:00
parent cde36888a3
commit b10cdd7215
Signed by: 80486DX2-66
GPG Key ID: 83631EF27054609B
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
c-programming/asprintf.c
View File

@ -11,19 +11,20 @@ ssize_t asprintf(char** strp, char* format, ...) {
va_list args; va_list args;
va_start(args, format); va_start(args, format);
ssize_t size = (ssize_t) vsnprintf(NULL, 0, format, args); ssize_t result = (ssize_t) vsnprintf(NULL, 0, format, args);
if (size < 0) { if (result < 0) {
va_end(args); va_end(args);
return -1; return -1;
} }
*strp = malloc(++size); size_t size = (size_t) result + 1;
*strp = malloc(size * sizeof(char));
if (*strp == NULL) { if (*strp == NULL) {
va_end(args); va_end(args);
return -1; return -1;
} }
ssize_t result = (ssize_t) vsnprintf(*strp, size, format, args); result = (ssize_t) vsnprintf(*strp, size, format, args);
va_end(args); va_end(args);
if (result < 0) { if (result < 0) {