1 /*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1991, 1993
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 #include <sys/types.h>
33 #include <err.h>
34 #include <errno.h>
35 #include <limits.h>
36 #include <locale.h>
37 #include <stdio.h>
38 #include <stdlib.h>
39 #include <string.h>
40 #include <unistd.h>
41 #include <wchar.h>
42
43 #include <capsicum_helpers.h>
44
45 #define TAB 8
46
47 void check(FILE *);
48 static void usage(void) __dead2;
49
50 int
main(int argc,char * argv[])51 main(int argc, char *argv[])
52 {
53 u_long column, start, stop;
54 int ch, width;
55 char *p;
56
57 setlocale(LC_ALL, "");
58
59 caph_cache_catpages();
60 if (caph_limit_stdio() < 0 || caph_enter() < 0)
61 err(EXIT_FAILURE, "capsicum");
62
63 while ((ch = getopt(argc, argv, "")) != -1)
64 switch(ch) {
65 case '?':
66 default:
67 usage();
68 }
69 argc -= optind;
70 argv += optind;
71
72 start = stop = 0;
73 switch(argc) {
74 case 2:
75 stop = strtol(argv[1], &p, 10);
76 if (stop <= 0 || *p)
77 errx(EXIT_FAILURE, "illegal column -- %s", argv[1]);
78 /* FALLTHROUGH */
79 case 1:
80 start = strtol(argv[0], &p, 10);
81 if (start <= 0 || *p)
82 errx(EXIT_FAILURE, "illegal column -- %s", argv[0]);
83 break;
84 case 0:
85 break;
86 default:
87 usage();
88 }
89
90 if (stop && start > stop)
91 errx(EXIT_FAILURE, "illegal start and stop columns");
92
93 for (column = 0;;) {
94 switch (ch = getwchar()) {
95 case WEOF:
96 check(stdin);
97 break;
98 case '\b':
99 if (column)
100 --column;
101 break;
102 case '\n':
103 column = 0;
104 break;
105 case '\t':
106 column = (column + TAB) & ~(TAB - 1);
107 break;
108 default:
109 if ((width = wcwidth(ch)) > 0)
110 column += width;
111 break;
112 }
113
114 if ((!start || column < start || (stop && column > stop)) &&
115 putwchar(ch) == WEOF)
116 check(stdout);
117 }
118 }
119
120 void
check(FILE * stream)121 check(FILE *stream)
122 {
123 if (feof(stream))
124 exit(EXIT_SUCCESS);
125 if (ferror(stream))
126 err(EXIT_FAILURE, "%s", stream == stdin ? "stdin" : "stdout");
127 }
128
129 void
usage(void)130 usage(void)
131 {
132 (void)fprintf(stderr, "usage: colrm [start [stop]]\n");
133 exit(EXIT_FAILURE);
134 }
135