oysteikt/mpd
0
0
Fork 0
Code Issues Pull Requests Releases Activity

text_input_stream: detect end-of-file

Browse Source 
Fixes endless loop when the last line of a text file was not
terminated (bug 3470).
This commit is contained in:
Max Kellermann
2012-03-19 19:51:19 +01:00
parent 8dcefaf2e3
commit d5be3cce9c
1 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/text_input_stream.c
View File

@@ -24,6 +24,7 @@
#include <glib.h> #include <glib.h>
#include <assert.h>
#include <string.h> #include <string.h>
struct text_input_stream { struct text_input_stream {
@@ -67,7 +68,12 @@ text_input_stream_read(struct text_input_stream *tis)
do { do {
dest = fifo_buffer_write(tis->buffer, &length); dest = fifo_buffer_write(tis->buffer, &length);
if (dest != NULL) { if (dest != NULL && length >= 2) {
/* reserve one byte for the null terminator if
the last line is not terminated by a
newline character */
--length;
nbytes = input_stream_lock_read(tis->is, dest, length, nbytes = input_stream_lock_read(tis->is, dest, length,
&error); &error);
if (nbytes > 0) if (nbytes > 0)
@@ -77,13 +83,22 @@ text_input_stream_read(struct text_input_stream *tis)
g_error_free(error); g_error_free(error);
return NULL; return NULL;
} }
} } else
nbytes = 0;
src = fifo_buffer_read(tis->buffer, &length); src = fifo_buffer_read(tis->buffer, &length);
if (src == NULL) if (src == NULL)
return NULL; return NULL;
p = memchr(src, '\n', length); p = memchr(src, '\n', length);
if (p == NULL && nbytes == 0) {
/* end of file (or line too long): terminate
the current line */
dest = fifo_buffer_write(tis->buffer, &nbytes);
assert(dest != NULL);
*(char *)dest = '\n';
fifo_buffer_append(tis->buffer, 1);
}
} while (p == NULL); } while (p == NULL);
length = p - src + 1; length = p - src + 1;