Home | History | Annotate | Line # | Download | only in paste
paste.c revision 1.9
      1  1.9       dsl /*	$NetBSD: paste.c,v 1.9 2006/03/31 18:59:52 dsl Exp $	*/
      2  1.3       tls 
      3  1.1       cgd /*
      4  1.3       tls  * Copyright (c) 1989, 1993
      5  1.3       tls  *	The Regents of the University of California.  All rights reserved.
      6  1.1       cgd  *
      7  1.1       cgd  * This code is derived from software contributed to Berkeley by
      8  1.1       cgd  * Adam S. Moskowitz of Menlo Consulting.
      9  1.1       cgd  *
     10  1.1       cgd  * Redistribution and use in source and binary forms, with or without
     11  1.1       cgd  * modification, are permitted provided that the following conditions
     12  1.1       cgd  * are met:
     13  1.1       cgd  * 1. Redistributions of source code must retain the above copyright
     14  1.1       cgd  *    notice, this list of conditions and the following disclaimer.
     15  1.1       cgd  * 2. Redistributions in binary form must reproduce the above copyright
     16  1.1       cgd  *    notice, this list of conditions and the following disclaimer in the
     17  1.1       cgd  *    documentation and/or other materials provided with the distribution.
     18  1.7       agc  * 3. Neither the name of the University nor the names of its contributors
     19  1.1       cgd  *    may be used to endorse or promote products derived from this software
     20  1.1       cgd  *    without specific prior written permission.
     21  1.1       cgd  *
     22  1.1       cgd  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     23  1.1       cgd  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     24  1.1       cgd  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     25  1.1       cgd  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     26  1.1       cgd  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     27  1.1       cgd  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     28  1.1       cgd  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     29  1.1       cgd  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     30  1.1       cgd  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     31  1.1       cgd  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     32  1.1       cgd  * SUCH DAMAGE.
     33  1.1       cgd  */
     34  1.1       cgd 
     35  1.4     lukem #include <sys/cdefs.h>
     36  1.1       cgd #ifndef lint
     37  1.8       dsl __COPYRIGHT("@(#) Copyright (c) 1989, 1993\n"
     38  1.8       dsl 	"The Regents of the University of California.  All rights reserved.\n");
     39  1.1       cgd #endif /* not lint */
     40  1.1       cgd 
     41  1.1       cgd #ifndef lint
     42  1.3       tls /*static char sccsid[] = "from: @(#)paste.c	8.1 (Berkeley) 6/6/93";*/
     43  1.9       dsl __RCSID("$NetBSD: paste.c,v 1.9 2006/03/31 18:59:52 dsl Exp $");
     44  1.1       cgd #endif /* not lint */
     45  1.1       cgd 
     46  1.1       cgd #include <sys/types.h>
     47  1.4     lukem #include <err.h>
     48  1.1       cgd #include <errno.h>
     49  1.1       cgd #include <limits.h>
     50  1.1       cgd #include <stdio.h>
     51  1.4     lukem #include <stdlib.h>
     52  1.6  wsanchez #include <unistd.h>
     53  1.1       cgd #include <string.h>
     54  1.5     perry #include <unistd.h>
     55  1.1       cgd 
     56  1.9       dsl void	parallel(int, char **);
     57  1.8       dsl void	sequential(char **);
     58  1.8       dsl int	tr(char *);
     59  1.8       dsl void	usage(void);
     60  1.4     lukem 
     61  1.9       dsl char dflt_delim[] = "\t";
     62  1.9       dsl char *delim = dflt_delim;
     63  1.9       dsl int delimcnt = 1;
     64  1.1       cgd 
     65  1.4     lukem int
     66  1.8       dsl main(int argc, char **argv)
     67  1.1       cgd {
     68  1.1       cgd 	int ch, seq;
     69  1.1       cgd 
     70  1.1       cgd 	seq = 0;
     71  1.9       dsl 	while ((ch = getopt(argc, argv, "d:s")) != -1) {
     72  1.8       dsl 		switch (ch) {
     73  1.1       cgd 		case 'd':
     74  1.9       dsl 			delim = strdup(optarg);
     75  1.9       dsl 			delimcnt = tr(delim);
     76  1.1       cgd 			break;
     77  1.1       cgd 		case 's':
     78  1.1       cgd 			seq = 1;
     79  1.1       cgd 			break;
     80  1.1       cgd 		case '?':
     81  1.1       cgd 		default:
     82  1.1       cgd 			usage();
     83  1.1       cgd 		}
     84  1.9       dsl 	}
     85  1.1       cgd 	argc -= optind;
     86  1.1       cgd 	argv += optind;
     87  1.1       cgd 
     88  1.1       cgd 	if (seq)
     89  1.1       cgd 		sequential(argv);
     90  1.1       cgd 	else
     91  1.9       dsl 		parallel(argc, argv);
     92  1.1       cgd 	exit(0);
     93  1.1       cgd }
     94  1.1       cgd 
     95  1.4     lukem void
     96  1.9       dsl parallel(int argc, char **argv)
     97  1.1       cgd {
     98  1.9       dsl 	char ch, *dp, *line;
     99  1.9       dsl 	FILE **fpp, *fp;
    100  1.9       dsl 	size_t line_len;
    101  1.9       dsl 	int cnt, output;
    102  1.9       dsl 
    103  1.9       dsl 	fpp = calloc(argc, sizeof *fpp);
    104  1.9       dsl 	if (fpp == NULL)
    105  1.9       dsl 		err(1, "calloc");
    106  1.9       dsl 
    107  1.9       dsl 	for (cnt = 0; cnt < argc; cnt++) {
    108  1.9       dsl 		if (strcmp(argv[cnt], "-") == 0)
    109  1.9       dsl 			fpp[cnt] = stdin;
    110  1.9       dsl 		else if (!(fpp[cnt] = fopen(argv[cnt], "r")))
    111  1.9       dsl 			err(1, "%s", argv[cnt]);
    112  1.1       cgd 	}
    113  1.1       cgd 
    114  1.9       dsl 	for (;;) {
    115  1.9       dsl 		/* Start with the NUL at the end of 'delim' ... */
    116  1.9       dsl 		dp = delim + delimcnt;
    117  1.9       dsl 		output = 0;
    118  1.9       dsl 		for (cnt = 0; cnt < argc; cnt++) {
    119  1.9       dsl 			fp = fpp[cnt];
    120  1.9       dsl 			if (fp == NULL)
    121  1.9       dsl 				continue;
    122  1.9       dsl 			line = fgetln(fp, &line_len);
    123  1.9       dsl 			if (line == NULL) {
    124  1.9       dsl 				/* Assume EOF */
    125  1.9       dsl 				if (fp != stdin)
    126  1.9       dsl 					fclose(fp);
    127  1.9       dsl 				fpp[cnt] = NULL;
    128  1.1       cgd 				continue;
    129  1.1       cgd 			}
    130  1.9       dsl 			/* Output enough separators to catch up */
    131  1.9       dsl 			do {
    132  1.9       dsl 				ch = *dp++;
    133  1.9       dsl 				if (ch)
    134  1.1       cgd 					putchar(ch);
    135  1.9       dsl 				if (dp >= delim + delimcnt)
    136  1.9       dsl 					dp = delim;
    137  1.9       dsl 			} while (++output <= cnt);
    138  1.9       dsl 			/* Remove any trailing newline - check for last line */
    139  1.9       dsl 			if (line[line_len - 1] == '\n')
    140  1.9       dsl 				line_len--;
    141  1.9       dsl 			printf("%.*s", line_len, line);
    142  1.9       dsl 		}
    143  1.9       dsl 
    144  1.9       dsl 		if (!output)
    145  1.9       dsl 			break;
    146  1.9       dsl 
    147  1.9       dsl 		/* Add separators to end of line */
    148  1.9       dsl 		while (++output <= cnt) {
    149  1.9       dsl 			ch = *dp++;
    150  1.9       dsl 			if (ch)
    151  1.1       cgd 				putchar(ch);
    152  1.9       dsl 			if (dp >= delim + delimcnt)
    153  1.9       dsl 				dp = delim;
    154  1.1       cgd 		}
    155  1.9       dsl 		putchar('\n');
    156  1.1       cgd 	}
    157  1.9       dsl 
    158  1.9       dsl 	free(fpp);
    159  1.1       cgd }
    160  1.1       cgd 
    161  1.4     lukem void
    162  1.8       dsl sequential(char **argv)
    163  1.1       cgd {
    164  1.4     lukem 	FILE *fp;
    165  1.4     lukem 	int cnt;
    166  1.4     lukem 	char ch, *p, *dp;
    167  1.1       cgd 	char buf[_POSIX2_LINE_MAX + 1];
    168  1.1       cgd 
    169  1.4     lukem 	for (; (p = *argv) != NULL; ++argv) {
    170  1.1       cgd 		if (p[0] == '-' && !p[1])
    171  1.1       cgd 			fp = stdin;
    172  1.1       cgd 		else if (!(fp = fopen(p, "r"))) {
    173  1.4     lukem 			warn("%s", p);
    174  1.1       cgd 			continue;
    175  1.1       cgd 		}
    176  1.1       cgd 		if (fgets(buf, sizeof(buf), fp)) {
    177  1.1       cgd 			for (cnt = 0, dp = delim;;) {
    178  1.4     lukem 				if (!(p = strchr(buf, '\n')))
    179  1.4     lukem 					err(1, "%s: input line too long.",
    180  1.1       cgd 					    *argv);
    181  1.1       cgd 				*p = '\0';
    182  1.1       cgd 				(void)printf("%s", buf);
    183  1.1       cgd 				if (!fgets(buf, sizeof(buf), fp))
    184  1.1       cgd 					break;
    185  1.4     lukem 				if ((ch = *dp++) != 0)
    186  1.1       cgd 					putchar(ch);
    187  1.1       cgd 				if (++cnt == delimcnt) {
    188  1.1       cgd 					dp = delim;
    189  1.1       cgd 					cnt = 0;
    190  1.1       cgd 				}
    191  1.1       cgd 			}
    192  1.1       cgd 			putchar('\n');
    193  1.1       cgd 		}
    194  1.1       cgd 		if (fp != stdin)
    195  1.1       cgd 			(void)fclose(fp);
    196  1.1       cgd 	}
    197  1.1       cgd }
    198  1.1       cgd 
    199  1.4     lukem int
    200  1.8       dsl tr(char *arg)
    201  1.1       cgd {
    202  1.4     lukem 	int cnt;
    203  1.4     lukem 	char ch, *p;
    204  1.1       cgd 
    205  1.1       cgd 	for (p = arg, cnt = 0; (ch = *p++); ++arg, ++cnt)
    206  1.1       cgd 		if (ch == '\\')
    207  1.1       cgd 			switch(ch = *p++) {
    208  1.1       cgd 			case 'n':
    209  1.1       cgd 				*arg = '\n';
    210  1.1       cgd 				break;
    211  1.1       cgd 			case 't':
    212  1.1       cgd 				*arg = '\t';
    213  1.1       cgd 				break;
    214  1.1       cgd 			case '0':
    215  1.1       cgd 				*arg = '\0';
    216  1.1       cgd 				break;
    217  1.1       cgd 			default:
    218  1.1       cgd 				*arg = ch;
    219  1.1       cgd 				break;
    220  1.1       cgd 		} else
    221  1.1       cgd 			*arg = ch;
    222  1.1       cgd 
    223  1.4     lukem 	if (!cnt)
    224  1.4     lukem 		errx(1, "no delimiters specified.");
    225  1.1       cgd 	return(cnt);
    226  1.1       cgd }
    227  1.1       cgd 
    228  1.4     lukem void
    229  1.1       cgd usage()
    230  1.1       cgd {
    231  1.1       cgd 	(void)fprintf(stderr, "paste: [-s] [-d delimiters] file ...\n");
    232  1.1       cgd 	exit(1);
    233  1.1       cgd }
    234