1 /* $NetBSD: vis.c,v 1.22 2013/02/20 17:04:45 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 1989, 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/cdefs.h> 33 #ifndef lint 34 __COPYRIGHT("@(#) Copyright (c) 1989, 1993\ 35 The Regents of the University of California. All rights reserved."); 36 #endif /* not lint */ 37 38 #ifndef lint 39 #if 0 40 static char sccsid[] = "@(#)vis.c 8.1 (Berkeley) 6/6/93"; 41 #endif 42 __RCSID("$NetBSD: vis.c,v 1.22 2013/02/20 17:04:45 christos Exp $"); 43 #endif /* not lint */ 44 45 #include <stdio.h> 46 #include <string.h> 47 #include <stdlib.h> 48 #include <string.h> 49 #include <errno.h> 50 #include <wchar.h> 51 #include <limits.h> 52 #include <unistd.h> 53 #include <err.h> 54 #include <vis.h> 55 56 #include "extern.h" 57 58 static int eflags, fold, foldwidth = 80, none, markeol; 59 #ifdef DEBUG 60 int debug; 61 #endif 62 static const char *extra = ""; 63 64 static void process(FILE *); 65 66 int 67 main(int argc, char *argv[]) 68 { 69 FILE *fp; 70 int ch; 71 int rval; 72 73 while ((ch = getopt(argc, argv, "bcde:F:fhlmnostw")) != -1) 74 switch((char)ch) { 75 case 'b': 76 eflags |= VIS_NOSLASH; 77 break; 78 case 'c': 79 eflags |= VIS_CSTYLE; 80 break; 81 #ifdef DEBUG 82 case 'd': 83 debug++; 84 break; 85 #endif 86 case 'e': 87 extra = optarg; 88 break; 89 case 'F': 90 if ((foldwidth = atoi(optarg)) < 5) { 91 errx(1, "can't fold lines to less than 5 cols"); 92 /* NOTREACHED */ 93 } 94 markeol++; 95 break; 96 case 'f': 97 fold++; /* fold output lines to 80 cols */ 98 break; /* using hidden newline */ 99 case 'h': 100 eflags |= VIS_HTTPSTYLE; 101 break; 102 case 'l': 103 markeol++; /* mark end of line with \$ */ 104 break; 105 case 'm': 106 eflags |= VIS_MIMESTYLE; 107 if (foldwidth == 80) 108 foldwidth = 76; 109 break; 110 case 'n': 111 none++; 112 break; 113 case 'o': 114 eflags |= VIS_OCTAL; 115 break; 116 case 's': 117 eflags |= VIS_SAFE; 118 break; 119 case 't': 120 eflags |= VIS_TAB; 121 break; 122 case 'w': 123 eflags |= VIS_WHITE; 124 break; 125 case '?': 126 default: 127 (void)fprintf(stderr, 128 "Usage: %s [-bcfhlmnostw] [-e extra]" 129 " [-F foldwidth] [file ...]\n", getprogname()); 130 return 1; 131 } 132 133 if ((eflags & (VIS_HTTPSTYLE|VIS_MIMESTYLE)) == 134 (VIS_HTTPSTYLE|VIS_MIMESTYLE)) 135 errx(1, "Can't specify -m and -h at the same time"); 136 137 argc -= optind; 138 argv += optind; 139 140 rval = 0; 141 142 if (*argv) 143 while (*argv) { 144 if ((fp = fopen(*argv, "r")) != NULL) { 145 process(fp); 146 (void)fclose(fp); 147 } else { 148 warn("%s", *argv); 149 rval = 1; 150 } 151 argv++; 152 } 153 else 154 process(stdin); 155 return rval; 156 } 157 158 static void 159 process(FILE *fp) 160 { 161 static int col = 0; 162 static char nul[] = "\0"; 163 char *cp = nul + 1; /* so *(cp-1) starts out != '\n' */ 164 wint_t c, c1, rachar; 165 char mbibuff[2 * MB_LEN_MAX + 1]; /* max space for 2 wchars */ 166 char buff[4 * MB_LEN_MAX + 1]; /* max encoding length for one char */ 167 int mbilen, cerr = 0, raerr = 0; 168 169 #if defined(__minix) 170 /* triggers a 'may be used uninitialized', when compiled with gcc, 171 * asserts off, and -Os. */ 172 rachar = 0; 173 #endif /* defined(__minix) */ 174 /* 175 * The input stream is considered to be multibyte characters. 176 * The input loop will read this data inputing one character, 177 * possibly multiple bytes, at a time and converting each to 178 * a wide character wchar_t. 179 * 180 * The vis(3) functions, however, require single either bytes 181 * or a multibyte string as their arguments. So we convert 182 * our input wchar_t and the following look-ahead wchar_t to 183 * a multibyte string for processing by vis(3). 184 */ 185 186 /* Read one multibyte character, store as wchar_t */ 187 c = getwc(fp); 188 if (c == WEOF && errno == EILSEQ) { 189 /* Error in multibyte data. Read one byte. */ 190 c = (wint_t)getc(fp); 191 cerr = 1; 192 } 193 while (c != WEOF) { 194 /* Clear multibyte input buffer. */ 195 memset(mbibuff, 0, sizeof(mbibuff)); 196 /* Read-ahead next multibyte character. */ 197 if (!cerr) 198 rachar = getwc(fp); 199 if (cerr || (rachar == WEOF && errno == EILSEQ)) { 200 /* Error in multibyte data. Read one byte. */ 201 rachar = (wint_t)getc(fp); 202 raerr = 1; 203 } 204 if (none) { 205 /* Handle -n flag. */ 206 cp = buff; 207 *cp++ = c; 208 if (c == '\\') 209 *cp++ = '\\'; 210 *cp = '\0'; 211 } else if (markeol && c == '\n') { 212 /* Handle -l flag. */ 213 cp = buff; 214 if ((eflags & VIS_NOSLASH) == 0) 215 *cp++ = '\\'; 216 *cp++ = '$'; 217 *cp++ = '\n'; 218 *cp = '\0'; 219 } else { 220 /* 221 * Convert character using vis(3) library. 222 * At this point we will process one character. 223 * But we must pass the vis(3) library this 224 * character plus the next one because the next 225 * one is used as a look-ahead to decide how to 226 * encode this one under certain circumstances. 227 * 228 * Since our characters may be multibyte, e.g., 229 * in the UTF-8 locale, we cannot use vis() and 230 * svis() which require byte input, so we must 231 * create a multibyte string and use strvisx(). 232 */ 233 /* Treat EOF as a NUL char. */ 234 c1 = rachar; 235 if (c1 == WEOF) 236 c1 = L'\0'; 237 /* 238 * If we hit a multibyte conversion error above, 239 * insert byte directly into string buff because 240 * wctomb() will fail. Else convert wchar_t to 241 * multibyte using wctomb(). 242 */ 243 if (cerr) { 244 *mbibuff = (char)c; 245 mbilen = 1; 246 } else 247 mbilen = wctomb(mbibuff, c); 248 /* Same for look-ahead character. */ 249 if (raerr) 250 mbibuff[mbilen] = (char)c1; 251 else 252 wctomb(mbibuff + mbilen, c1); 253 /* Perform encoding on just first character. */ 254 (void) strsenvisx(buff, 4 * MB_LEN_MAX, mbibuff, 255 1, eflags, extra, &cerr); 256 } 257 258 cp = buff; 259 if (fold) { 260 #ifdef DEBUG 261 if (debug) 262 (void)printf("<%02d,", col); 263 #endif 264 col = foldit(cp, col, foldwidth, eflags); 265 #ifdef DEBUG 266 if (debug) 267 (void)printf("%02d>", col); 268 #endif 269 } 270 do { 271 (void)putchar(*cp); 272 } while (*++cp); 273 c = rachar; 274 cerr = raerr; 275 } 276 /* 277 * terminate partial line with a hidden newline 278 */ 279 if (fold && *(cp - 1) != '\n') 280 (void)printf(eflags & VIS_MIMESTYLE ? "=\n" : "\\\n"); 281 } 282