Fix potential memory leak

This commit is contained in:
Alexander 'z33ky' Hirsch 2015-08-20 02:24:47 +02:00
parent 22675b0111
commit f85d0740a8

View file

@ -17,18 +17,22 @@ char *read_line(FILE *file) {
continue;
}
if (length == size) {
string = realloc(string, size *= 2);
if (!string) {
char *new_string = realloc(string, size *= 2);
if (!new_string) {
free(string);
return NULL;
}
string = new_string;
}
string[length++] = c;
}
if (length + 1 == size) {
string = realloc(string, length + 1);
if (!string) {
char *new_string = realloc(string, length + 1);
if (!new_string) {
free(string);
return NULL;
}
string = new_string;
}
string[length] = '\0';
return string;