mirror of
https://github.com/red0124/ssp.git
synced 2025-01-23 04:55:20 +01:00
Update ssp.hpp
This commit is contained in:
parent
09e628020d
commit
c0ee100f99
8
ssp.hpp
8
ssp.hpp
@ -657,14 +657,15 @@ ssize_t get_line(char** lineptr, size_t* n, FILE* fp) {
|
|||||||
char buff[get_line_initial_buffer_size];
|
char buff[get_line_initial_buffer_size];
|
||||||
|
|
||||||
if (*lineptr == nullptr || *n < sizeof(buff)) {
|
if (*lineptr == nullptr || *n < sizeof(buff)) {
|
||||||
*n = sizeof(buff);
|
size_t new_n = sizeof(buff);
|
||||||
auto new_lineptr = static_cast<char*>(realloc(*lineptr, *n));
|
auto new_lineptr = static_cast<char*>(realloc(*lineptr, new_n));
|
||||||
if (new_lineptr == nullptr) {
|
if (new_lineptr == nullptr) {
|
||||||
errno = ENOMEM;
|
errno = ENOMEM;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
*lineptr = new_lineptr;
|
*lineptr = new_lineptr;
|
||||||
|
*n = new_n;
|
||||||
}
|
}
|
||||||
|
|
||||||
(*lineptr)[0] = '\0';
|
(*lineptr)[0] = '\0';
|
||||||
@ -674,7 +675,7 @@ ssize_t get_line(char** lineptr, size_t* n, FILE* fp) {
|
|||||||
size_t buff_used = strlen(buff);
|
size_t buff_used = strlen(buff);
|
||||||
|
|
||||||
if (*n < buff_used + line_used) {
|
if (*n < buff_used + line_used) {
|
||||||
*n *= 2;
|
size_t new_n = *n * 2;
|
||||||
|
|
||||||
auto new_lineptr = static_cast<char*>(realloc(*lineptr, *n));
|
auto new_lineptr = static_cast<char*>(realloc(*lineptr, *n));
|
||||||
if (new_lineptr == nullptr) {
|
if (new_lineptr == nullptr) {
|
||||||
@ -683,6 +684,7 @@ ssize_t get_line(char** lineptr, size_t* n, FILE* fp) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
*lineptr = new_lineptr;
|
*lineptr = new_lineptr;
|
||||||
|
*n = new_n;
|
||||||
}
|
}
|
||||||
|
|
||||||
memcpy(*lineptr + line_used, buff, buff_used);
|
memcpy(*lineptr + line_used, buff, buff_used);
|
||||||
|
Loading…
Reference in New Issue
Block a user