1 /* $Header: /p/tcsh/cvsroot/tcsh/tw.color.c,v 1.27 2010/08/19 05:52:19 christos Exp $ */ 2 /* 3 * tw.color.c: builtin color ls-F 4 */ 5 /*- 6 * Copyright (c) 1998 The Regents of the University of California. 7 * All rights reserved. 8 * 9 * Redistribution and use in source and binary forms, with or without 10 * modification, are permitted provided that the following conditions 11 * are met: 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 2. Redistributions in binary form must reproduce the above copyright 15 * notice, this list of conditions and the following disclaimer in the 16 * documentation and/or other materials provided with the distribution. 17 * 3. Neither the name of the University nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 #include "sh.h" 34 35 RCSID("$tcsh: tw.color.c,v 1.27 2010/08/19 05:52:19 christos Exp $") 36 37 #include "tw.h" 38 #include "ed.h" 39 #include "tc.h" 40 41 #ifdef COLOR_LS_F 42 43 typedef struct { 44 const char *s; 45 size_t len; 46 } Str; 47 48 49 #define VAR(suffix,variable,defaultcolor) \ 50 { \ 51 suffix, variable, { defaultcolor, sizeof(defaultcolor) - 1 }, \ 52 { defaultcolor, sizeof(defaultcolor) - 1 } \ 53 } 54 #define NOS '\0' /* no suffix */ 55 56 typedef struct { 57 const char suffix; 58 const char *variable; 59 Str color; 60 Str defaultcolor; 61 } Variable; 62 63 static Variable variables[] = { 64 VAR('/', "di", "01;34"), /* Directory */ 65 VAR('@', "ln", "01;36"), /* Symbolic link */ 66 VAR('&', "or", ""), /* Orphanned symbolic link (defaults to ln) */ 67 VAR('|', "pi", "33"), /* Named pipe (FIFO) */ 68 VAR('=', "so", "01;35"), /* Socket */ 69 VAR('>', "do", "01;35"), /* Door (solaris fast ipc mechanism) */ 70 VAR('#', "bd", "01;33"), /* Block device */ 71 VAR('%', "cd", "01;33"), /* Character device */ 72 VAR('*', "ex", "01;32"), /* Executable file */ 73 VAR(NOS, "fi", "0"), /* Regular file */ 74 VAR(NOS, "no", "0"), /* Normal (non-filename) text */ 75 VAR(NOS, "mi", ""), /* Missing file (defaults to fi) */ 76 #ifdef IS_ASCII 77 VAR(NOS, "lc", "\033["), /* Left code (ASCII) */ 78 #else 79 VAR(NOS, "lc", "\x27["), /* Left code (EBCDIC)*/ 80 #endif 81 VAR(NOS, "rc", "m"), /* Right code */ 82 VAR(NOS, "ec", ""), /* End code (replaces lc+no+rc) */ 83 VAR(NOS, "su", ""), /* Setuid file (u+s) */ 84 VAR(NOS, "sg", ""), /* Setgid file (g+s) */ 85 VAR(NOS, "tw", ""), /* Sticky and other writable dir (+t,o+w) */ 86 VAR(NOS, "ow", ""), /* Other writable dir (o+w) but not sticky */ 87 VAR(NOS, "st", ""), /* Sticky dir (+t) but not other writable */ 88 VAR(NOS, "rs", "0"), /* Reset to normal color */ 89 VAR(NOS, "hl", "44;37"), /* Reg file extra hard links, obsolete? */ 90 VAR(NOS, "mh", "44;37"), /* Reg file extra hard links */ 91 VAR(NOS, "ca", "30;41"), /* File with capability */ 92 }; 93 94 enum FileType { 95 VDir, VSym, VOrph, VPipe, VSock, VDoor, VBlock, VChr, VExe, 96 VFile, VNormal, VMiss, VLeft, VRight, VEnd 97 }; 98 99 #define nvariables (sizeof(variables)/sizeof(variables[0])) 100 101 typedef struct { 102 Str extension; /* file extension */ 103 Str color; /* color string */ 104 } Extension; 105 106 static Extension *extensions = NULL; 107 static size_t nextensions = 0; 108 109 static char *colors = NULL; 110 int color_context_ls = FALSE; /* do colored ls */ 111 static int color_context_lsmF = FALSE; /* do colored ls-F */ 112 113 static int getstring (char **, const Char **, Str *, int); 114 static void put_color (const Str *); 115 static void print_color (const Char *, size_t, Char); 116 117 /* set_color_context(): 118 */ 119 void 120 set_color_context(void) 121 { 122 struct varent *vp = adrof(STRcolor); 123 124 if (vp == NULL || vp->vec == NULL) { 125 color_context_ls = FALSE; 126 color_context_lsmF = FALSE; 127 } else if (!vp->vec[0] || vp->vec[0][0] == '\0') { 128 color_context_ls = TRUE; 129 color_context_lsmF = TRUE; 130 } else { 131 size_t i; 132 133 color_context_ls = FALSE; 134 color_context_lsmF = FALSE; 135 for (i = 0; vp->vec[i]; i++) 136 if (Strcmp(vp->vec[i], STRls) == 0) 137 color_context_ls = TRUE; 138 else if (Strcmp(vp->vec[i], STRlsmF) == 0) 139 color_context_lsmF = TRUE; 140 } 141 } 142 143 144 /* getstring(): 145 */ 146 static int 147 getstring(char **dp, const Char **sp, Str *pd, int f) 148 { 149 const Char *s = *sp; 150 char *d = *dp; 151 eChar sc; 152 153 while (*s && (*s & CHAR) != (Char)f && (*s & CHAR) != ':') { 154 if ((*s & CHAR) == '\\' || (*s & CHAR) == '^') { 155 if ((sc = parseescape(&s)) == CHAR_ERR) 156 return 0; 157 } 158 else 159 sc = *s++ & CHAR; 160 d += one_wctomb(d, sc); 161 } 162 163 pd->s = *dp; 164 pd->len = d - *dp; 165 *sp = s; 166 *dp = d; 167 return *s == (Char)f; 168 } 169 170 171 /* parseLS_COLORS(): 172 * Parse the LS_COLORS environment variable 173 */ 174 void 175 parseLS_COLORS(const Char *value) 176 { 177 size_t i, len; 178 const Char *v; /* pointer in value */ 179 char *c; /* pointer in colors */ 180 Extension *volatile e; /* pointer in extensions */ 181 jmp_buf_t osetexit; 182 size_t omark; 183 184 (void) &e; 185 186 /* init */ 187 xfree(extensions); 188 for (i = 0; i < nvariables; i++) 189 variables[i].color = variables[i].defaultcolor; 190 colors = NULL; 191 extensions = NULL; 192 nextensions = 0; 193 194 if (value == NULL) 195 return; 196 197 len = Strlen(value); 198 /* allocate memory */ 199 i = 1; 200 for (v = value; *v; v++) 201 if ((*v & CHAR) == ':') 202 i++; 203 extensions = xmalloc(len + i * sizeof(Extension)); 204 colors = i * sizeof(Extension) + (char *)extensions; 205 nextensions = 0; 206 207 /* init pointers */ 208 v = value; 209 c = colors; 210 e = &extensions[0]; 211 212 /* Prevent from crashing if unknown parameters are given. */ 213 214 omark = cleanup_push_mark(); 215 getexit(osetexit); 216 217 if (setexit() == 0) { 218 219 /* parse */ 220 while (*v) { 221 switch (*v & CHAR) { 222 case ':': 223 v++; 224 continue; 225 226 case '*': /* :*ext=color: */ 227 v++; 228 if (getstring(&c, &v, &e->extension, '=') && 229 0 < e->extension.len) { 230 v++; 231 getstring(&c, &v, &e->color, ':'); 232 e++; 233 continue; 234 } 235 break; 236 237 default: /* :vl=color: */ 238 if (v[0] && v[1] && (v[2] & CHAR) == '=') { 239 for (i = 0; i < nvariables; i++) 240 if ((Char)variables[i].variable[0] == (v[0] & CHAR) && 241 (Char)variables[i].variable[1] == (v[1] & CHAR)) 242 break; 243 if (i < nvariables) { 244 v += 3; 245 getstring(&c, &v, &variables[i].color, ':'); 246 continue; 247 } 248 else 249 stderror(ERR_BADCOLORVAR, v[0], v[1]); 250 } 251 break; 252 } 253 while (*v && (*v & CHAR) != ':') 254 v++; 255 } 256 } 257 258 cleanup_pop_mark(omark); 259 resexit(osetexit); 260 261 nextensions = e - extensions; 262 } 263 264 /* put_color(): 265 */ 266 static void 267 put_color(const Str *color) 268 { 269 size_t i; 270 const char *c = color->s; 271 int original_output_raw = output_raw; 272 273 output_raw = TRUE; 274 cleanup_push(&original_output_raw, output_raw_restore); 275 for (i = color->len; 0 < i; i--) 276 xputchar(*c++); 277 cleanup_until(&original_output_raw); 278 } 279 280 281 /* print_color(): 282 */ 283 static void 284 print_color(const Char *fname, size_t len, Char suffix) 285 { 286 size_t i; 287 char *filename = short2str(fname); 288 char *last = filename + len; 289 Str *color = &variables[VFile].color; 290 291 switch (suffix) { 292 case '>': /* File is a symbolic link pointing to 293 * a directory */ 294 color = &variables[VDir].color; 295 break; 296 case '+': /* File is a hidden directory [aix] or 297 * context dependent [hpux] */ 298 case ':': /* File is network special [hpux] */ 299 break; 300 default: 301 for (i = 0; i < nvariables; i++) 302 if (variables[i].suffix != NOS && 303 (Char)variables[i].suffix == suffix) { 304 color = &variables[i].color; 305 break; 306 } 307 if (i == nvariables) { 308 for (i = 0; i < nextensions; i++) 309 if (len >= extensions[i].extension.len 310 && strncmp(last - extensions[i].extension.len, 311 extensions[i].extension.s, 312 extensions[i].extension.len) == 0) { 313 color = &extensions[i].color; 314 break; 315 } 316 } 317 break; 318 } 319 320 put_color(&variables[VLeft].color); 321 put_color(color); 322 put_color(&variables[VRight].color); 323 } 324 325 326 /* print_with_color(): 327 */ 328 void 329 print_with_color(const Char *filename, size_t len, Char suffix) 330 { 331 if (color_context_lsmF && 332 (haderr ? (didfds ? is2atty : isdiagatty) : 333 (didfds ? is1atty : isoutatty))) { 334 print_color(filename, len, suffix); 335 xprintf("%S", filename); 336 if (0 < variables[VEnd].color.len) 337 put_color(&variables[VEnd].color); 338 else { 339 put_color(&variables[VLeft].color); 340 put_color(&variables[VNormal].color); 341 put_color(&variables[VRight].color); 342 } 343 } 344 else 345 xprintf("%S", filename); 346 xputwchar(suffix); 347 } 348 349 350 #endif /* COLOR_LS_F */ 351