diff options
-rw-r--r-- | src/split.c | 11 | ||||
-rwxr-xr-x | tests/misc/split-lchunk | 4 |
2 files changed, 8 insertions, 7 deletions
diff --git a/src/split.c b/src/split.c index e09ebcec8..6c75080a7 100644 --- a/src/split.c +++ b/src/split.c @@ -599,6 +599,7 @@ lines_chunk_split (uintmax_t k, uintmax_t n, char *buf, size_t bufsize, off_t chunk_end = chunk_size - 1; off_t n_written = 0; bool new_file_flag = true; + bool chunk_truncated = false; if (k > 1) { @@ -620,6 +621,7 @@ lines_chunk_split (uintmax_t k, uintmax_t n, char *buf, size_t bufsize, error (EXIT_FAILURE, errno, "%s", infile); else if (n_read == 0) break; /* eof. */ + chunk_truncated = false; eob = buf + n_read; while (bp != eob) @@ -659,11 +661,7 @@ lines_chunk_split (uintmax_t k, uintmax_t n, char *buf, size_t bufsize, if (!next && bp == eob) { /* replenish buf, before going to next chunk. */ - - /* If we're going to stop reading, - then count the current chunk. */ - if (n_written >= file_size) - chunk_no++; + chunk_truncated = true; break; } chunk_no++; @@ -684,6 +682,9 @@ lines_chunk_split (uintmax_t k, uintmax_t n, char *buf, size_t bufsize, } } + if (chunk_truncated) + chunk_no++; + /* Ensure NUMBER files are created, which truncates any existing files or notifies any consumers on fifos. FIXME: Should we do this before EXIT_FAILURE? */ diff --git a/tests/misc/split-lchunk b/tests/misc/split-lchunk index 762138b97..8b4d157a1 100755 --- a/tests/misc/split-lchunk +++ b/tests/misc/split-lchunk @@ -74,8 +74,8 @@ for ELIDE_EMPTY in '' '-e'; do rm -f x* if test -z "$ELIDE_EMPTY"; then - split ---io-blksize=$IO_BLKSIZE $ELIDE_EMPTY -n l/2/$N in > chunk.k - stat x* >/dev/null 2>/dev/null && fail=1 + split ---io-blksize=$IO_BLKSIZE -n l/2/$N in > chunk.k + stat x* 2>/dev/null && fail=1 fi split ---io-blksize=$IO_BLKSIZE $ELIDE_EMPTY -n l/$N in |