paste: fix output when file lengths differ

If the files being pasted had different numbers of lines the
output was incorrect.

Rewrite the loop over all lines to allow for this.  Add tests for
such conditions.

function                                             old     new   delta
paste_main                                           458     526     +68

Signed-off-by: Ron Yorston <rmy@pobox.com>
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>

1 files changed, 24 insertions, 12 deletions

diff --git a/coreutils/paste.c b/coreutils/paste.c
index 3e5f20158..363340e3d 100644
--- a/coreutils/paste.c
+++ b/coreutils/paste.c
@@ -34,27 +34,38 @@
 
 static void paste_files(FILE** files, int file_cnt, char* delims, int del_cnt)
 {
-        char *line;
+        char **line = xmalloc(file_cnt * sizeof(char *));
         char delim;
         int active_files = file_cnt;
         int i;
 
         while (active_files > 0) {
                 int del_idx = 0;
+                int got_line = FALSE;
 
                 for (i = 0; i < file_cnt; ++i) {
-                        if (files[i] == NULL)
-                                continue;
-
-                        line = xmalloc_fgetline(files[i]);
-                        if (!line) {
-                                fclose_if_not_stdin(files[i]);
-                                files[i] = NULL;
-                                --active_files;
-                                continue;
+                        if (files[i]) {
+                                line[i] = xmalloc_fgetline(files[i]);
+                                if (!line[i]) {
+                                        fclose_if_not_stdin(files[i]);
+                                        files[i] = NULL;
+                                        --active_files;
+                                } else {
+                                        got_line = TRUE;
+                                }
+                        } else {
+                                line[i] = NULL;
+                        }
+                }
+
+                if (!got_line)
+                        break;
+
+                for (i = 0; i < file_cnt; ++i) {
+                        if (line[i]) {
+                                fputs_stdout(line[i]);
+                                free(line[i]);
                         }
-                        fputs_stdout(line);
-                        free(line);
                         delim = '\n';
                         if (i != file_cnt - 1) {
                                 delim = delims[del_idx++];
@@ -65,6 +76,7 @@ static void paste_files(FILE** files, int file_cnt, char* delims, int del_cnt)
                                 fputc(delim, stdout);
                 }
         }
+        free(line);
 }
 
 static void paste_files_separate(FILE** files, char* delims, int del_cnt)