diff options
Diffstat (limited to 'tests/storage/test_chewing_table.cpp')
-rw-r--r-- | tests/storage/test_chewing_table.cpp | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tests/storage/test_chewing_table.cpp b/tests/storage/test_chewing_table.cpp index b65b20f..cae766d 100644 --- a/tests/storage/test_chewing_table.cpp +++ b/tests/storage/test_chewing_table.cpp @@ -38,9 +38,12 @@ int main(int argc, char * argv[]) { largetable.store(new_chunk); largetable.load(new_chunk); - char* linebuf = NULL; size_t size = 0; - while( getline(&linebuf, &size, stdin) ){ - linebuf[strlen(linebuf)-1] = '\0'; + char* linebuf = NULL; size_t size = 0; ssize_t read; + while ((read = getline(&linebuf, &size, stdin)) != -1) { + if ( '\n' == linebuf[strlen(linebuf) - 1] ) { + linebuf[strlen(linebuf) - 1] = '\0'; + } + if ( strcmp ( linebuf, "quit" ) == 0) break; |