Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.bin/column/column.c	Lines:	116	149	77.9 %
Date:	2017-11-13	Branches:	86	121	71.1 %


/*	$OpenBSD: column.c,v 1.25 2016/09/04 20:33:36 martijn Exp $	*/
/*	$NetBSD: column.c,v 1.4 1995/09/02 05:53:03 jtc Exp $	*/

/*
 * Copyright (c) 1989, 1993, 1994
 *	The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/types.h>
#include <sys/ioctl.h>

#include <ctype.h>
#include <err.h>
#include <limits.h>
#include <locale.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <wchar.h>
#include <wctype.h>

void  c_columnate(void);
void *ereallocarray(void *, size_t, size_t);
void *ecalloc(size_t, size_t);
void  input(FILE *);
void  maketbl(void);
void  print(void);
void  r_columnate(void);
__dead void usage(void);

struct field {
	char *content;
	int width;
};

int termwidth;			/* default terminal width */
int entries;			/* number of records */
int eval;			/* exit value */
int *maxwidths;			/* longest record per column */
struct field **table;		/* one array of pointers per line */
wchar_t *separator = L"\t ";	/* field separator for table option */

int
main(int argc, char *argv[])
{
	struct winsize win;
	FILE *fp;
	int ch, tflag, xflag;
	char *p;
	const char *errstr;

	setlocale(LC_CTYPE, "");

	termwidth = 0;
	if ((p = getenv("COLUMNS")) != NULL)
		termwidth = strtonum(p, 1, INT_MAX, NULL);
	if (termwidth == 0 && ioctl(STDOUT_FILENO, TIOCGWINSZ, &win) == 0 &&
	    win.ws_col > 0)
		termwidth = win.ws_col;
	if (termwidth == 0)
		termwidth = 80;

	if (pledge("stdio rpath flock cpath wpath", NULL) == -1)
		err(1, "pledge");

	tflag = xflag = 0;
	while ((ch = getopt(argc, argv, "c:s:tx")) != -1) {
		switch(ch) {
		case 'c':
			termwidth = strtonum(optarg, 1, INT_MAX, &errstr);
			if (errstr != NULL)
				errx(1, "%s: %s", errstr, optarg);
			break;
		case 's':
			if ((separator = reallocarray(NULL, strlen(optarg) + 1,
			    sizeof(*separator))) == NULL)
				err(1, NULL);
			if (mbstowcs(separator, optarg, strlen(optarg) + 1) ==
			    (size_t) -1)
				err(1, "sep");
			break;
		case 't':
			tflag = 1;
			break;
		case 'x':
			xflag = 1;
			break;
		default:
			usage();
		}
	}

	if (!tflag)
		separator = L"";
	argv += optind;

	if (*argv == NULL) {
		input(stdin);
	} else {
		for (; *argv; ++argv) {
			if ((fp = fopen(*argv, "r"))) {
				input(fp);
				(void)fclose(fp);
			} else {
				warn("%s", *argv);
				eval = 1;
			}
		}
	}

	if (pledge("stdio flock rpath cpath wpath", NULL) == -1)
		err(1, "pledge");

	if (!entries)
		return eval;

	if (tflag)
		maketbl();
	else if (*maxwidths >= termwidth)
		print();
	else if (xflag)
		c_columnate();
	else
		r_columnate();
	return eval;
}

#define	INCR_NEXTTAB(x)	(x = (x + 8) & ~7)
void
c_columnate(void)
{
	int col, numcols;
	struct field **row;

	INCR_NEXTTAB(*maxwidths);
	if ((numcols = termwidth / *maxwidths) == 0)
		numcols = 1;
	for (col = 0, row = table;; ++row) {
		fputs((*row)->content, stdout);
		if (!--entries)
			break;
		if (++col == numcols) {
			col = 0;
			putchar('\n');
		} else {
			while (INCR_NEXTTAB((*row)->width) <= *maxwidths)
				putchar('\t');
		}
	}
	putchar('\n');
}

void
r_columnate(void)
{
	int base, col, numcols, numrows, row;

	INCR_NEXTTAB(*maxwidths);
	if ((numcols = termwidth / *maxwidths) == 0)
		numcols = 1;
	numrows = entries / numcols;
	if (entries % numcols)
		++numrows;

	for (base = row = 0; row < numrows; base = ++row) {
		for (col = 0; col < numcols; ++col, base += numrows) {
			fputs(table[base]->content, stdout);
			if (base + numrows >= entries)
				break;
			while (INCR_NEXTTAB(table[base]->width) <= *maxwidths)
				putchar('\t');
		}
		putchar('\n');
	}
}

void
print(void)
{
	int row;

	for (row = 0; row < entries; row++)
		puts(table[row]->content);
}


void
maketbl(void)
{
	struct field **row;
	int col;

	for (row = table; entries--; ++row) {
		for (col = 0; (*row)[col + 1].content != NULL; ++col)
			printf("%s%*s  ", (*row)[col].content,
			    maxwidths[col] - (*row)[col].width, "");
		puts((*row)[col].content);
	}
}

#define	DEFNUM		1000
#define	DEFCOLS		25

void
input(FILE *fp)
{
	static int maxentry = 0;
	static int maxcols = 0;
	static struct field *cols = NULL;
	int col, width, twidth;
	size_t blen;
	ssize_t llen;
	char *p, *s, *buf = NULL;
	wchar_t wc;
	int wlen;

	while ((llen = getline(&buf, &blen, fp)) > -1) {
		if (buf[llen - 1] == '\n')
			buf[llen - 1] = '\0';

		p = buf;
		for (col = 0;; col++) {

			/* Skip lines containing nothing but whitespace. */

			for (s = p; (wlen = mbtowc(&wc, s, MB_CUR_MAX)) > 0;
			     s += wlen)
				if (!iswspace(wc))
					break;
			if (*s == '\0')
				break;

			/* Skip leading, multiple, and trailing separators. */

			while ((wlen = mbtowc(&wc, p, MB_CUR_MAX)) > 0 &&
			    wcschr(separator, wc) != NULL)
				p += wlen;
			if (*p == '\0')
				break;

			/*
			 * Found a non-empty field.
			 * Remember the start and measure the width.
			 */

			s = p;
			width = 0;
			while (*p != '\0') {
				if ((wlen = mbtowc(&wc, p, MB_CUR_MAX)) == -1) {
					width++;
					p++;
					continue;
				}
				if (wcschr(separator, wc) != NULL)
					break;
				if (*p == '\t')
					INCR_NEXTTAB(width);
				else  {
					width += (twidth = wcwidth(wc)) == -1 ?
					    1 : twidth;
				}
				p += wlen;
			}

			if (col + 1 >= maxcols) {
				if (maxcols > INT_MAX - DEFCOLS)
					err(1, "too many columns");
				maxcols += DEFCOLS;
				cols = ereallocarray(cols, maxcols,
				    sizeof(*cols));
				maxwidths = ereallocarray(maxwidths, maxcols,
				    sizeof(*maxwidths));
				memset(maxwidths + col, 0,
				    DEFCOLS * sizeof(*maxwidths));
			}

			/*
			 * Remember the width of the field,
			 * NUL-terminate and remeber the content,
			 * and advance beyond the separator, if any.
			 */

			cols[col].width = width;
			if (maxwidths[col] < width)
				maxwidths[col] = width;
			if (*p != '\0') {
				*p = '\0';
				p += wlen;
			}
			if ((cols[col].content = strdup(s)) == NULL)
				err(1, NULL);
		}
		if (col == 0)
			continue;

		/* Found a non-empty line; remember it. */

		if (entries == maxentry) {
			if (maxentry > INT_MAX - DEFNUM)
				errx(1, "too many input lines");
			maxentry += DEFNUM;
			table = ereallocarray(table, maxentry, sizeof(*table));
		}
		table[entries] = ereallocarray(NULL, col + 1,
		    sizeof(*(table[entries])));
		table[entries][col].content = NULL;
		while (col--)
			table[entries][col] = cols[col];
		entries++;
	}
}

void *
ereallocarray(void *ptr, size_t nmemb, size_t size)
{
	if ((ptr = reallocarray(ptr, nmemb, size)) == NULL)
		err(1, NULL);
	return ptr;
}

void *
ecalloc(size_t nmemb, size_t size)
{
	void *ptr;

	if ((ptr = calloc(nmemb, size)) == NULL)
		err(1, NULL);
	return ptr;
}

__dead void
usage(void)
{
	(void)fprintf(stderr,
	    "usage: column [-tx] [-c columns] [-s sep] [file ...]\n");
	exit(1);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: column.c,v 1.25 2016/09/04 20:33:36 martijn Exp $ */
2			/* $NetBSD: column.c,v 1.4 1995/09/02 05:53:03 jtc Exp $ */
3
4			/*
5			* Copyright (c) 1989, 1993, 1994
6			* The Regents of the University of California. All rights reserved.
7			*
8			* Redistribution and use in source and binary forms, with or without
9			* modification, are permitted provided that the following conditions
10			* are met:
11			* 1. Redistributions of source code must retain the above copyright
12			* notice, this list of conditions and the following disclaimer.
13			* 2. Redistributions in binary form must reproduce the above copyright
14			* notice, this list of conditions and the following disclaimer in the
15			* documentation and/or other materials provided with the distribution.
16			* 3. Neither the name of the University nor the names of its contributors
17			* may be used to endorse or promote products derived from this software
18			* without specific prior written permission.
19			*
20			* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21			* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22			* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23			* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24			* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25			* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26			* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27			* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28			* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29			* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30			* SUCH DAMAGE.
31			*/
32
33			#include <sys/types.h>
34			#include <sys/ioctl.h>
35
36			#include <ctype.h>
37			#include <err.h>
38			#include <limits.h>
39			#include <locale.h>
40			#include <stdio.h>
41			#include <stdlib.h>
42			#include <string.h>
43			#include <unistd.h>
44			#include <wchar.h>
45			#include <wctype.h>
46
47			void c_columnate(void);
48			void ereallocarray(void , size_t, size_t);
49			void *ecalloc(size_t, size_t);
50			void input(FILE *);
51			void maketbl(void);
52			void print(void);
53			void r_columnate(void);
54			__dead void usage(void);
55
56			struct field {
57			char *content;
58			int width;
59			};
60
61			int termwidth; /* default terminal width */
62			int entries; /* number of records */
63			int eval; /* exit value */
64			int maxwidths; / longest record per column */
65			struct field *table; / one array of pointers per line */
66			wchar_t separator = L"\t "; / field separator for table option */
67
68			int
69			main(int argc, char *argv[])
70			{
71		150	struct winsize win;
72			FILE *fp;
73			int ch, tflag, xflag;
74			char *p;
75		75	const char *errstr;
76
77		75	setlocale(LC_CTYPE, "");
78
79		75	termwidth = 0;
80	✗✓	75	if ((p = getenv("COLUMNS")) != NULL)
81			termwidth = strtonum(p, 1, INT_MAX, NULL);
82	✓✗✗✓ ✗✗	150	if (termwidth == 0 && ioctl(STDOUT_FILENO, TIOCGWINSZ, &win) == 0 &&
83			win.ws_col > 0)
84			termwidth = win.ws_col;
85	✓✗	75	if (termwidth == 0)
86		75	termwidth = 80;
87
88	✗✓	75	if (pledge("stdio rpath flock cpath wpath", NULL) == -1)
89			err(1, "pledge");
90
91			tflag = xflag = 0;
92	✓✓	360	while ((ch = getopt(argc, argv, "c:s:tx")) != -1) {
93	✓✓✓✓ ✗	105	switch(ch) {
94			case 'c':
95		60	termwidth = strtonum(optarg, 1, INT_MAX, &errstr);
96	✗✓	60	if (errstr != NULL)
97			errx(1, "%s: %s", errstr, optarg);
98			break;
99			case 's':
100	✗✓	10	if ((separator = reallocarray(NULL, strlen(optarg) + 1,
101		5	sizeof(*separator))) == NULL)
102			err(1, NULL);
103	✗✓	5	if (mbstowcs(separator, optarg, strlen(optarg) + 1) ==
104			(size_t) -1)
105			err(1, "sep");
106			break;
107			case 't':
108			tflag = 1;
109		15	break;
110			case 'x':
111			xflag = 1;
112		25	break;
113			default:
114			usage();
115			}
116			}
117
118	✓✓	75	if (!tflag)
119		60	separator = L"";
120		75	argv += optind;
121
122	✓✗	75	if (*argv == NULL) {
123		75	input(stdin);
124		75	} else {
125			for (; *argv; ++argv) {
126			if ((fp = fopen(*argv, "r"))) {
127			input(fp);
128			(void)fclose(fp);
129			} else {
130			warn("%s", *argv);
131			eval = 1;
132			}
133			}
134			}
135
136	✗✓	75	if (pledge("stdio flock rpath cpath wpath", NULL) == -1)
137			err(1, "pledge");
138
139	✗✓	75	if (!entries)
140			return eval;
141
142	✓✓	75	if (tflag)
143		15	maketbl();
144	✗✓	60	else if (*maxwidths >= termwidth)
145			print();
146	✓✓	60	else if (xflag)
147		25	c_columnate();
148			else
149		35	r_columnate();
150		75	return eval;
151		75	}
152
153			#define INCR_NEXTTAB(x) (x = (x + 8) & ~7)
154			void
155			c_columnate(void)
156			{
157			int col, numcols;
158			struct field **row;
159
160		50	INCR_NEXTTAB(*maxwidths);
161		25	if ((numcols = termwidth / *maxwidths) == 0)
162			numcols = 1;
163		135	for (col = 0, row = table;; ++row) {
164		135	fputs((*row)->content, stdout);
165	✓✓	135	if (!--entries)
166			break;
167	✓✓	110	if (++col == numcols) {
168			col = 0;
169	✓✗	100	putchar('\n');
170			} else {
171	✓✓	250	while (INCR_NEXTTAB((row)->width) <= maxwidths)
172	✓✗	130	putchar('\t');
173			}
174			}
175	✓✗	50	putchar('\n');
176		25	}
177
178			void
179			r_columnate(void)
180			{
181			int base, col, numcols, numrows, row;
182
183		70	INCR_NEXTTAB(*maxwidths);
184		35	if ((numcols = termwidth / *maxwidths) == 0)
185			numcols = 1;
186		35	numrows = entries / numcols;
187	✓✓	35	if (entries % numcols)
188		10	++numrows;
189
190	✓✓	290	for (base = row = 0; row < numrows; base = ++row) {
191	✓✗	340	for (col = 0; col < numcols; ++col, base += numrows) {
192		170	fputs(table[base]->content, stdout);
193	✓✓	170	if (base + numrows >= entries)
194			break;
195	✓✓	250	while (INCR_NEXTTAB(table[base]->width) <= *maxwidths)
196	✓✗	130	putchar('\t');
197			}
198	✓✗	220	putchar('\n');
199			}
200		35	}
201
202			void
203			print(void)
204			{
205			int row;
206
207			for (row = 0; row < entries; row++)
208			puts(table[row]->content);
209			}
210
211
212			void
213			maketbl(void)
214			{
215			struct field **row;
216			int col;
217
218	✓✓	115	for (row = table; entries--; ++row) {
219	✓✓	170	for (col = 0; (*row)[col + 1].content != NULL; ++col)
220		50	printf("%s%s ", (row)[col].content,
221		50	maxwidths[col] - (*row)[col].width, "");
222		35	puts((*row)[col].content);
223			}
224		15	}
225
226			#define DEFNUM 1000
227			#define DEFCOLS 25
228
229			void
230			input(FILE *fp)
231			{
232			static int maxentry = 0;
233			static int maxcols = 0;
234			static struct field *cols = NULL;
235			int col, width, twidth;
236		150	size_t blen;
237			ssize_t llen;
238		75	char p, s, *buf = NULL;
239		75	wchar_t wc;
240			int wlen;
241
242	✓✓	890	while ((llen = getline(&buf, &blen, fp)) > -1) {
243	✓✓	400	if (buf[llen - 1] == '\n')
244		375	buf[llen - 1] = '\0';
245
246		400	p = buf;
247		790	for (col = 0;; col++) {
248
249			/* Skip lines containing nothing but whitespace. */
250
251	✓✓	1870	for (s = p; (wlen = mbtowc(&wc, s, MB_CUR_MAX)) > 0;
252		145	s += wlen)
253	✓✓	540	if (!iswspace(wc))
254			break;
255	✓✓	790	if (*s == '\0')
256			break;
257
258			/* Skip leading, multiple, and trailing separators. */
259
260	✓✓✓✓	1270	while ((wlen = mbtowc(&wc, p, MB_CUR_MAX)) > 0 &&
261		420	wcschr(separator, wc) != NULL)
262		30	p += wlen;
263	✓✓	395	if (*p == '\0')
264			break;
265
266			/*
267			* Found a non-empty field.
268			* Remember the start and measure the width.
269			*/
270
271			s = p;
272			width = 0;
273	✓✓	3270	while (*p != '\0') {
274	✗✓	1300	if ((wlen = mbtowc(&wc, p, MB_CUR_MAX)) == -1) {
275			width++;
276			p++;
277			continue;
278			}
279	✓✓	1300	if (wcschr(separator, wc) != NULL)
280			break;
281	✗✓	1245	if (*p == '\t')
282			INCR_NEXTTAB(width);
283			else {
284		1245	width += (twidth = wcwidth(wc)) == -1 ?
285			1 : twidth;
286			}
287		1245	p += wlen;
288			}
289
290	✓✓	390	if (col + 1 >= maxcols) {
291	✗✓	75	if (maxcols > INT_MAX - DEFCOLS)
292			err(1, "too many columns");
293		75	maxcols += DEFCOLS;
294		75	cols = ereallocarray(cols, maxcols,
295			sizeof(*cols));
296		75	maxwidths = ereallocarray(maxwidths, maxcols,
297			sizeof(*maxwidths));
298		75	memset(maxwidths + col, 0,
299			DEFCOLS * sizeof(*maxwidths));
300		75	}
301
302			/*
303			* Remember the width of the field,
304			* NUL-terminate and remeber the content,
305			* and advance beyond the separator, if any.
306			*/
307
308		390	cols[col].width = width;
309	✓✓	390	if (maxwidths[col] < width)
310		170	maxwidths[col] = width;
311	✓✓	390	if (*p != '\0') {
312		55	*p = '\0';
313		55	p += wlen;
314		55	}
315	✗✓	390	if ((cols[col].content = strdup(s)) == NULL)
316			err(1, NULL);
317			}
318	✓✓	400	if (col == 0)
319		60	continue;
320
321			/* Found a non-empty line; remember it. */
322
323	✓✓	340	if (entries == maxentry) {
324	✗✓	75	if (maxentry > INT_MAX - DEFNUM)
325			errx(1, "too many input lines");
326		75	maxentry += DEFNUM;
327		75	table = ereallocarray(table, maxentry, sizeof(*table));
328		75	}
329		340	table[entries] = ereallocarray(NULL, col + 1,
330			sizeof(*(table[entries])));
331		340	table[entries][col].content = NULL;
332	✓✓	1460	while (col--)
333		390	table[entries][col] = cols[col];
334		340	entries++;
335			}
336		75	}
337
338			void *
339			ereallocarray(void *ptr, size_t nmemb, size_t size)
340			{
341	✗✓	1130	if ((ptr = reallocarray(ptr, nmemb, size)) == NULL)
342			err(1, NULL);
343		565	return ptr;
344			}
345
346			void *
347			ecalloc(size_t nmemb, size_t size)
348			{
349			void *ptr;
350
351			if ((ptr = calloc(nmemb, size)) == NULL)
352			err(1, NULL);
353			return ptr;
354			}
355
356			__dead void
357			usage(void)
358			{
359			(void)fprintf(stderr,
360			"usage: column [-tx] [-c columns] [-s sep] [file ...]\n");
361			exit(1);
362			}