1 /* $Header: /src/pub/tcsh/tw.color.c,v 1.11 2002/06/25 19:02:11 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("$Id: tw.color.c,v 1.11 2002/06/25 19:02:11 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 char *s; 45 int 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 }; 84 85 enum FileType { 86 VDir, VSym, VOrph, VPipe, VSock, VDoor, VBlock, VChr, VExe, 87 VFile, VNormal, VMiss, VLeft, VRight, VEnd 88 }; 89 90 #define nvariables (sizeof(variables)/sizeof(variables[0])) 91 92 typedef struct { 93 Str extension; /* file extension */ 94 Str color; /* color string */ 95 } Extension; 96 97 static Extension *extensions = NULL; 98 static int nextensions = 0; 99 100 static char *colors = NULL; 101 bool color_context_ls = FALSE; /* do colored ls */ 102 static bool color_context_lsmF = FALSE; /* do colored ls-F */ 103 104 static bool getstring __P((char **, const Char **, Str *, int)); 105 static void put_color __P((Str *)); 106 static void print_color __P((Char *, size_t, int)); 107 108 /* set_color_context(): 109 */ 110 void 111 set_color_context() 112 { 113 struct varent *vp = adrof(STRcolor); 114 115 if (vp == NULL || vp->vec == NULL) { 116 color_context_ls = FALSE; 117 color_context_lsmF = FALSE; 118 } else if (!vp->vec[0] || vp->vec[0][0] == '\0') { 119 color_context_ls = TRUE; 120 color_context_lsmF = TRUE; 121 } else { 122 size_t i; 123 124 color_context_ls = FALSE; 125 color_context_lsmF = FALSE; 126 for (i = 0; vp->vec[i]; i++) 127 if (Strcmp(vp->vec[i], STRls) == 0) 128 color_context_ls = TRUE; 129 else if (Strcmp(vp->vec[i], STRlsmF) == 0) 130 color_context_lsmF = TRUE; 131 } 132 } 133 134 135 /* getstring(): 136 */ 137 static bool 138 getstring(dp, sp, pd, f) 139 char **dp; /* dest buffer */ 140 const Char **sp; /* source buffer */ 141 Str *pd; /* pointer to dest buffer */ 142 int f; /* final character */ 143 { 144 const Char *s = *sp; 145 char *d = *dp; 146 int sc; 147 148 while (*s && (*s & CHAR) != f && (*s & CHAR) != ':') { 149 if ((*s & CHAR) == '\\' || (*s & CHAR) == '^') { 150 if ((sc = parseescape(&s)) == -1) 151 return 0; 152 else 153 *d++ = (char) sc; 154 } 155 else 156 *d++ = *s++ & CHAR; 157 } 158 159 pd->s = *dp; 160 pd->len = (int) (d - *dp); 161 *sp = s; 162 *dp = d; 163 return *s == f; 164 } 165 166 167 /* parseLS_COLORS(): 168 * Parse the LS_COLORS environment variable 169 */ 170 void 171 parseLS_COLORS(value) 172 Char *value; /* LS_COLOR variable's value */ 173 { 174 int i; 175 size_t len; 176 const Char *v; /* pointer in value */ 177 char *c; /* pointer in colors */ 178 Extension *e; /* pointer in extensions */ 179 jmp_buf_t osetexit; 180 181 (void) &e; 182 183 /* init */ 184 if (extensions) 185 xfree((ptr_t) extensions); 186 for (i = 0; i < nvariables; i++) 187 variables[i].color = variables[i].defaultcolor; 188 colors = NULL; 189 extensions = NULL; 190 nextensions = 0; 191 192 if (value == NULL) 193 return; 194 195 len = Strlen(value); 196 /* allocate memory */ 197 i = 1; 198 for (v = value; *v; v++) 199 if ((*v & CHAR) == ':') 200 i++; 201 extensions = (Extension *) xmalloc((size_t) (len + i * sizeof(Extension))); 202 colors = i * sizeof(Extension) + (char *)extensions; 203 nextensions = 0; 204 205 /* init pointers */ 206 v = value; 207 c = colors; 208 e = &extensions[0]; 209 210 /* Prevent from crashing if unknown parameters are given. */ 211 212 getexit(osetexit); 213 214 if (setexit() == 0) { 215 216 /* parse */ 217 while (*v) { 218 switch (*v & CHAR) { 219 case ':': 220 v++; 221 continue; 222 223 case '*': /* :*ext=color: */ 224 v++; 225 if (getstring(&c, &v, &e->extension, '=') && 226 0 < e->extension.len) { 227 v++; 228 getstring(&c, &v, &e->color, ':'); 229 e++; 230 continue; 231 } 232 break; 233 234 default: /* :vl=color: */ 235 if (v[0] && v[1] && (v[2] & CHAR) == '=') { 236 for (i = 0; i < nvariables; i++) 237 if (variables[i].variable[0] == (v[0] & CHAR) && 238 variables[i].variable[1] == (v[1] & CHAR)) 239 break; 240 if (i < nvariables) { 241 v += 3; 242 getstring(&c, &v, &variables[i].color, ':'); 243 continue; 244 } 245 else 246 stderror(ERR_BADCOLORVAR, v[0], v[1]); 247 } 248 break; 249 } 250 while (*v && (*v & CHAR) != ':') 251 v++; 252 } 253 } 254 255 resexit(osetexit); 256 257 nextensions = (int) (e - extensions); 258 } 259 260 261 /* put_color(): 262 */ 263 static void 264 put_color(color) 265 Str *color; 266 { 267 extern bool output_raw; /* PWP: in sh.print.c */ 268 size_t i; 269 char *c = color->s; 270 bool original_output_raw = output_raw; 271 272 output_raw = TRUE; 273 for (i = color->len; 0 < i; i--) 274 xputchar(*c++); 275 output_raw = original_output_raw; 276 } 277 278 279 /* print_color(): 280 */ 281 static void 282 print_color(fname, len, suffix) 283 Char *fname; 284 size_t len; 285 int suffix; 286 { 287 int i; 288 char *filename = short2str(fname); 289 char *last = filename + len; 290 Str *color = &variables[VFile].color; 291 292 switch (suffix) { 293 case '>': /* File is a symbolic link pointing to 294 * a directory */ 295 color = &variables[VDir].color; 296 break; 297 case '+': /* File is a hidden directory [aix] or 298 * context dependent [hpux] */ 299 case ':': /* File is network special [hpux] */ 300 break; 301 default: 302 for (i = 0; i < nvariables; i++) 303 if (variables[i].suffix != NOS && 304 variables[i].suffix == suffix) { 305 color = &variables[i].color; 306 break; 307 } 308 if (i == nvariables) { 309 for (i = 0; i < nextensions; i++) 310 if (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(filename, len, suffix) 330 Char *filename; 331 size_t len; 332 int suffix; 333 { 334 if (color_context_lsmF && 335 (haderr ? (didfds ? is2atty : isdiagatty) : 336 (didfds ? is1atty : isoutatty))) { 337 print_color(filename, len, suffix); 338 xprintf("%S", filename); 339 if (0 < variables[VEnd].color.len) 340 put_color(&variables[VEnd].color); 341 else { 342 put_color(&variables[VLeft].color); 343 put_color(&variables[VNormal].color); 344 put_color(&variables[VRight].color); 345 } 346 xputchar(suffix); 347 } 348 else 349 xprintf("%S%c", filename, suffix); 350 } 351 352 353 #endif /* COLOR_LS_F */ 354