xref: /freebsd/bin/ls/util.c (revision 7bd6fde3951af84ef3b68e4d1eadc1840c2fc1b3)
1 /*-
2  * Copyright (c) 1989, 1993, 1994
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software contributed to Berkeley by
6  * Michael Fischbein.
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  * 4. 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 #if 0
34 #ifndef lint
35 static char sccsid[] = "@(#)util.c	8.3 (Berkeley) 4/2/94";
36 #endif /* not lint */
37 #endif
38 #include <sys/cdefs.h>
39 __FBSDID("$FreeBSD$");
40 
41 #include <sys/types.h>
42 #include <sys/stat.h>
43 
44 #include <ctype.h>
45 #include <err.h>
46 #include <fts.h>
47 #include <limits.h>
48 #include <stdio.h>
49 #include <stdlib.h>
50 #include <string.h>
51 #include <wchar.h>
52 #include <wctype.h>
53 
54 #include "ls.h"
55 #include "extern.h"
56 
57 int
58 prn_normal(const char *s)
59 {
60 	mbstate_t mbs;
61 	wchar_t wc;
62 	int i, n;
63 	size_t clen;
64 
65 	memset(&mbs, 0, sizeof(mbs));
66 	n = 0;
67 	while ((clen = mbrtowc(&wc, s, MB_LEN_MAX, &mbs)) != 0) {
68 		if (clen == (size_t)-2) {
69 			n += printf("%s", s);
70 			break;
71 		}
72 		if (clen == (size_t)-1) {
73 			memset(&mbs, 0, sizeof(mbs));
74 			putchar((unsigned char)*s);
75 			s++;
76 			n++;
77 			continue;
78 		}
79 		for (i = 0; i < (int)clen; i++)
80 			putchar((unsigned char)s[i]);
81 		s += clen;
82 		if (iswprint(wc))
83 			n += wcwidth(wc);
84 	}
85 	return (n);
86 }
87 
88 int
89 prn_printable(const char *s)
90 {
91 	mbstate_t mbs;
92 	wchar_t wc;
93 	int i, n;
94 	size_t clen;
95 
96 	memset(&mbs, 0, sizeof(mbs));
97 	n = 0;
98 	while ((clen = mbrtowc(&wc, s, MB_LEN_MAX, &mbs)) != 0) {
99 		if (clen == (size_t)-1) {
100 			putchar('?');
101 			s++;
102 			n++;
103 			memset(&mbs, 0, sizeof(mbs));
104 			continue;
105 		}
106 		if (clen == (size_t)-2) {
107 			putchar('?');
108 			n++;
109 			break;
110 		}
111 		if (!iswprint(wc)) {
112 			putchar('?');
113 			s += clen;
114 			n++;
115 			continue;
116 		}
117 		for (i = 0; i < (int)clen; i++)
118 			putchar((unsigned char)s[i]);
119 		s += clen;
120 		n += wcwidth(wc);
121 	}
122 	return (n);
123 }
124 
125 /*
126  * The fts system makes it difficult to replace fts_name with a different-
127  * sized string, so we just calculate the real length here and do the
128  * conversion in prn_octal()
129  *
130  * XXX when using f_octal_escape (-b) rather than f_octal (-B), the
131  * length computed by len_octal may be too big. I just can't be buggered
132  * to fix this as an efficient fix would involve a lookup table. Same goes
133  * for the rather inelegant code in prn_octal.
134  *
135  *                                              DES 1998/04/23
136  */
137 
138 size_t
139 len_octal(const char *s, int len)
140 {
141 	mbstate_t mbs;
142 	wchar_t wc;
143 	size_t clen, r;
144 
145 	memset(&mbs, 0, sizeof(mbs));
146 	r = 0;
147 	while (len != 0 && (clen = mbrtowc(&wc, s, len, &mbs)) != 0) {
148 		if (clen == (size_t)-1) {
149 			r += 4;
150 			s++;
151 			len--;
152 			memset(&mbs, 0, sizeof(mbs));
153 			continue;
154 		}
155 		if (clen == (size_t)-2) {
156 			r += 4 * len;
157 			break;
158 		}
159 		if (iswprint(wc))
160 			r++;
161 		else
162 			r += 4 * clen;
163 		s += clen;
164 	}
165 	return (r);
166 }
167 
168 int
169 prn_octal(const char *s)
170 {
171 	static const char esc[] = "\\\\\"\"\aa\bb\ff\nn\rr\tt\vv";
172 	const char *p;
173 	mbstate_t mbs;
174 	wchar_t wc;
175 	size_t clen;
176 	unsigned char ch;
177 	int goodchar, i, len, prtlen;
178 
179 	memset(&mbs, 0, sizeof(mbs));
180 	len = 0;
181 	while ((clen = mbrtowc(&wc, s, MB_LEN_MAX, &mbs)) != 0) {
182 		goodchar = clen != (size_t)-1 && clen != (size_t)-2;
183 		if (goodchar && iswprint(wc) && wc != L'\"' && wc != L'\\') {
184 			for (i = 0; i < (int)clen; i++)
185 				putchar((unsigned char)s[i]);
186 			len += wcwidth(wc);
187 		} else if (goodchar && f_octal_escape && wc >= 0 &&
188 		    wc <= (wchar_t)UCHAR_MAX &&
189 		    (p = strchr(esc, (char)wc)) != NULL) {
190 			putchar('\\');
191 			putchar(p[1]);
192 			len += 2;
193 		} else {
194 			if (goodchar)
195 				prtlen = clen;
196 			else if (clen == (size_t)-1)
197 				prtlen = 1;
198 			else
199 				prtlen = strlen(s);
200 			for (i = 0; i < prtlen; i++) {
201 				ch = (unsigned char)s[i];
202 				putchar('\\');
203 		                putchar('0' + (ch >> 6));
204 		                putchar('0' + ((ch >> 3) & 7));
205 		                putchar('0' + (ch & 7));
206 				len += 4;
207 			}
208 		}
209 		if (clen == (size_t)-2)
210 			break;
211 		if (clen == (size_t)-1) {
212 			memset(&mbs, 0, sizeof(mbs));
213 			s++;
214 		} else
215 			s += clen;
216 	}
217 	return (len);
218 }
219 
220 void
221 usage(void)
222 {
223 	(void)fprintf(stderr,
224 #ifdef COLORLS
225 	"usage: ls [-ABCFGHILPRSTUWZabcdfghiklmnopqrstuwx1]"
226 #else
227 	"usage: ls [-ABCFHILPRSTUWZabcdfghiklmnopqrstuwx1]"
228 #endif
229 		      " [file ...]\n");
230 	exit(1);
231 }
232