1 /*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1989, 1993, 1994
5 * The Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32 #ifndef lint
33 static const char copyright[] =
34 "@(#) Copyright (c) 1989, 1993, 1994\n\
35 The Regents of the University of California. All rights reserved.\n";
36 #endif
37
38 #if 0
39 #ifndef lint
40 static char sccsid[] = "@(#)column.c 8.4 (Berkeley) 5/4/95";
41 #endif
42 #endif
43
44 #include <sys/cdefs.h>
45 __FBSDID("$FreeBSD$");
46
47 #include <sys/types.h>
48 #include <sys/ioctl.h>
49 #include <sys/param.h>
50
51 #include <err.h>
52 #include <limits.h>
53 #include <locale.h>
54 #include <stdio.h>
55 #include <stdlib.h>
56 #include <string.h>
57 #include <unistd.h>
58 #include <wchar.h>
59 #include <wctype.h>
60
61 #define TAB 8
62
63 static void c_columnate(void);
64 static void input(FILE *);
65 static void maketbl(void);
66 static void print(void);
67 static void r_columnate(void);
68 static void usage(void);
69 static int width(const wchar_t *);
70
71 static int termwidth = 80; /* default terminal width */
72
73 static int entries; /* number of records */
74 static int eval; /* exit value */
75 static int maxlength; /* longest record */
76 static wchar_t **list; /* array of pointers to records */
77 static const wchar_t *separator = L"\t "; /* field separator for table option */
78
79 int
main(int argc,char ** argv)80 main(int argc, char **argv)
81 {
82 struct winsize win;
83 FILE *fp;
84 int ch, tflag, xflag;
85 char *p;
86 const char *src;
87 wchar_t *newsep;
88 size_t seplen;
89
90 setlocale(LC_ALL, "");
91
92 if (ioctl(1, TIOCGWINSZ, &win) == -1 || !win.ws_col) {
93 if ((p = getenv("COLUMNS")))
94 termwidth = atoi(p);
95 } else
96 termwidth = win.ws_col;
97
98 tflag = xflag = 0;
99 while ((ch = getopt(argc, argv, "c:s:tx")) != -1)
100 switch(ch) {
101 case 'c':
102 termwidth = atoi(optarg);
103 break;
104 case 's':
105 src = optarg;
106 seplen = mbsrtowcs(NULL, &src, 0, NULL);
107 if (seplen == (size_t)-1)
108 err(1, "bad separator");
109 newsep = malloc((seplen + 1) * sizeof(wchar_t));
110 if (newsep == NULL)
111 err(1, NULL);
112 mbsrtowcs(newsep, &src, seplen + 1, NULL);
113 separator = newsep;
114 break;
115 case 't':
116 tflag = 1;
117 break;
118 case 'x':
119 xflag = 1;
120 break;
121 case '?':
122 default:
123 usage();
124 }
125 argc -= optind;
126 argv += optind;
127
128 if (!*argv)
129 input(stdin);
130 else for (; *argv; ++argv)
131 if ((fp = fopen(*argv, "r"))) {
132 input(fp);
133 (void)fclose(fp);
134 } else {
135 warn("%s", *argv);
136 eval = 1;
137 }
138
139 if (!entries)
140 exit(eval);
141
142 maxlength = roundup(maxlength + 1, TAB);
143 if (tflag)
144 maketbl();
145 else if (maxlength >= termwidth)
146 print();
147 else if (xflag)
148 c_columnate();
149 else
150 r_columnate();
151 exit(eval);
152 }
153
154 static void
c_columnate(void)155 c_columnate(void)
156 {
157 int chcnt, col, cnt, endcol, numcols;
158 wchar_t **lp;
159
160 numcols = termwidth / maxlength;
161 endcol = maxlength;
162 for (chcnt = col = 0, lp = list;; ++lp) {
163 wprintf(L"%ls", *lp);
164 chcnt += width(*lp);
165 if (!--entries)
166 break;
167 if (++col == numcols) {
168 chcnt = col = 0;
169 endcol = maxlength;
170 putwchar('\n');
171 } else {
172 while ((cnt = roundup(chcnt + 1, TAB)) <= endcol) {
173 (void)putwchar('\t');
174 chcnt = cnt;
175 }
176 endcol += maxlength;
177 }
178 }
179 if (chcnt)
180 putwchar('\n');
181 }
182
183 static void
r_columnate(void)184 r_columnate(void)
185 {
186 int base, chcnt, cnt, col, endcol, numcols, numrows, row;
187
188 numcols = termwidth / maxlength;
189 numrows = entries / numcols;
190 if (entries % numcols)
191 ++numrows;
192
193 for (row = 0; row < numrows; ++row) {
194 endcol = maxlength;
195 for (base = row, chcnt = col = 0; col < numcols; ++col) {
196 wprintf(L"%ls", list[base]);
197 chcnt += width(list[base]);
198 if ((base += numrows) >= entries)
199 break;
200 while ((cnt = roundup(chcnt + 1, TAB)) <= endcol) {
201 (void)putwchar('\t');
202 chcnt = cnt;
203 }
204 endcol += maxlength;
205 }
206 putwchar('\n');
207 }
208 }
209
210 static void
print(void)211 print(void)
212 {
213 int cnt;
214 wchar_t **lp;
215
216 for (cnt = entries, lp = list; cnt--; ++lp)
217 (void)wprintf(L"%ls\n", *lp);
218 }
219
220 typedef struct _tbl {
221 wchar_t **list;
222 int cols, *len;
223 } TBL;
224 #define DEFCOLS 25
225
226 static void
maketbl(void)227 maketbl(void)
228 {
229 TBL *t;
230 int coloff, cnt;
231 wchar_t *p, **lp;
232 int *lens, maxcols;
233 TBL *tbl;
234 wchar_t **cols;
235 wchar_t *last;
236
237 if ((t = tbl = calloc(entries, sizeof(TBL))) == NULL)
238 err(1, NULL);
239 if ((cols = calloc((maxcols = DEFCOLS), sizeof(*cols))) == NULL)
240 err(1, NULL);
241 if ((lens = calloc(maxcols, sizeof(int))) == NULL)
242 err(1, NULL);
243 for (cnt = 0, lp = list; cnt < entries; ++cnt, ++lp, ++t) {
244 for (coloff = 0, p = *lp;
245 (cols[coloff] = wcstok(p, separator, &last));
246 p = NULL)
247 if (++coloff == maxcols) {
248 if (!(cols = realloc(cols, ((u_int)maxcols +
249 DEFCOLS) * sizeof(wchar_t *))) ||
250 !(lens = realloc(lens,
251 ((u_int)maxcols + DEFCOLS) * sizeof(int))))
252 err(1, NULL);
253 memset((char *)lens + maxcols * sizeof(int),
254 0, DEFCOLS * sizeof(int));
255 maxcols += DEFCOLS;
256 }
257 if ((t->list = calloc(coloff, sizeof(*t->list))) == NULL)
258 err(1, NULL);
259 if ((t->len = calloc(coloff, sizeof(int))) == NULL)
260 err(1, NULL);
261 for (t->cols = coloff; --coloff >= 0;) {
262 t->list[coloff] = cols[coloff];
263 t->len[coloff] = width(cols[coloff]);
264 if (t->len[coloff] > lens[coloff])
265 lens[coloff] = t->len[coloff];
266 }
267 }
268 for (cnt = 0, t = tbl; cnt < entries; ++cnt, ++t) {
269 for (coloff = 0; coloff < t->cols - 1; ++coloff)
270 (void)wprintf(L"%ls%*ls", t->list[coloff],
271 lens[coloff] - t->len[coloff] + 2, L" ");
272 (void)wprintf(L"%ls\n", t->list[coloff]);
273 free(t->list);
274 free(t->len);
275 }
276 free(lens);
277 free(cols);
278 free(tbl);
279 }
280
281 #define DEFNUM 1000
282 #define MAXLINELEN (LINE_MAX + 1)
283
284 static void
input(FILE * fp)285 input(FILE *fp)
286 {
287 static int maxentry;
288 int len;
289 wchar_t *p, buf[MAXLINELEN];
290
291 if (!list)
292 if ((list = calloc((maxentry = DEFNUM), sizeof(*list))) ==
293 NULL)
294 err(1, NULL);
295 while (fgetws(buf, MAXLINELEN, fp)) {
296 for (p = buf; *p && iswspace(*p); ++p);
297 if (!*p)
298 continue;
299 if (!(p = wcschr(p, L'\n'))) {
300 warnx("line too long");
301 eval = 1;
302 continue;
303 }
304 *p = L'\0';
305 len = width(buf);
306 if (maxlength < len)
307 maxlength = len;
308 if (entries == maxentry) {
309 maxentry += DEFNUM;
310 if (!(list = realloc(list,
311 (u_int)maxentry * sizeof(*list))))
312 err(1, NULL);
313 }
314 list[entries] = malloc((wcslen(buf) + 1) * sizeof(wchar_t));
315 if (list[entries] == NULL)
316 err(1, NULL);
317 wcscpy(list[entries], buf);
318 entries++;
319 }
320 }
321
322 /* Like wcswidth(), but ignores non-printing characters. */
323 static int
width(const wchar_t * wcs)324 width(const wchar_t *wcs)
325 {
326 int w, cw;
327
328 for (w = 0; *wcs != L'\0'; wcs++)
329 if ((cw = wcwidth(*wcs)) > 0)
330 w += cw;
331 return (w);
332 }
333
334 static void
usage(void)335 usage(void)
336 {
337
338 (void)fprintf(stderr,
339 "usage: column [-tx] [-c columns] [-s sep] [file ...]\n");
340 exit(1);
341 }
342