tw.color.c revision 195609
1195609Smp/* $Header: /p/tcsh/cvsroot/tcsh/tw.color.c,v 1.25 2008/10/17 19:57:33 christos Exp $ */ 259243Sobrien/* 359243Sobrien * tw.color.c: builtin color ls-F 459243Sobrien */ 559243Sobrien/*- 659243Sobrien * Copyright (c) 1998 The Regents of the University of California. 759243Sobrien * All rights reserved. 859243Sobrien * 959243Sobrien * Redistribution and use in source and binary forms, with or without 1059243Sobrien * modification, are permitted provided that the following conditions 1159243Sobrien * are met: 1259243Sobrien * 1. Redistributions of source code must retain the above copyright 1359243Sobrien * notice, this list of conditions and the following disclaimer. 1459243Sobrien * 2. Redistributions in binary form must reproduce the above copyright 1559243Sobrien * notice, this list of conditions and the following disclaimer in the 1659243Sobrien * documentation and/or other materials provided with the distribution. 17100616Smp * 3. Neither the name of the University nor the names of its contributors 1859243Sobrien * may be used to endorse or promote products derived from this software 1959243Sobrien * without specific prior written permission. 2059243Sobrien * 2159243Sobrien * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 2259243Sobrien * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 2359243Sobrien * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2459243Sobrien * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 2559243Sobrien * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2659243Sobrien * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2759243Sobrien * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2859243Sobrien * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2959243Sobrien * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 3059243Sobrien * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 3159243Sobrien * SUCH DAMAGE. 3259243Sobrien */ 3359243Sobrien#include "sh.h" 3459243Sobrien 35195609SmpRCSID("$tcsh: tw.color.c,v 1.25 2008/10/17 19:57:33 christos Exp $") 3659243Sobrien 3759243Sobrien#include "tw.h" 3859243Sobrien#include "ed.h" 3959243Sobrien#include "tc.h" 4059243Sobrien 4159243Sobrien#ifdef COLOR_LS_F 4259243Sobrien 4359243Sobrientypedef struct { 44167465Smp const char *s; 45167465Smp size_t len; 4659243Sobrien} Str; 4759243Sobrien 4859243Sobrien 4959243Sobrien#define VAR(suffix,variable,defaultcolor) \ 5059243Sobrien{ \ 5159243Sobrien suffix, variable, { defaultcolor, sizeof(defaultcolor) - 1 }, \ 5259243Sobrien { defaultcolor, sizeof(defaultcolor) - 1 } \ 5359243Sobrien} 5459243Sobrien#define NOS '\0' /* no suffix */ 5559243Sobrien 5659243Sobrientypedef struct { 5759243Sobrien const char suffix; 5859243Sobrien const char *variable; 59167465Smp Str color; 60167465Smp Str defaultcolor; 6159243Sobrien} Variable; 6259243Sobrien 6359243Sobrienstatic Variable variables[] = { 6459243Sobrien VAR('/', "di", "01;34"), /* Directory */ 6559243Sobrien VAR('@', "ln", "01;36"), /* Symbolic link */ 6659243Sobrien VAR('&', "or", ""), /* Orphanned symbolic link (defaults to ln) */ 6759243Sobrien VAR('|', "pi", "33"), /* Named pipe (FIFO) */ 6859243Sobrien VAR('=', "so", "01;35"), /* Socket */ 6983098Smp VAR('>', "do", "01;35"), /* Door (solaris fast ipc mechanism) */ 7059243Sobrien VAR('#', "bd", "01;33"), /* Block device */ 7159243Sobrien VAR('%', "cd", "01;33"), /* Character device */ 7259243Sobrien VAR('*', "ex", "01;32"), /* Executable file */ 7359243Sobrien VAR(NOS, "fi", "0"), /* Regular file */ 7459243Sobrien VAR(NOS, "no", "0"), /* Normal (non-filename) text */ 7559243Sobrien VAR(NOS, "mi", ""), /* Missing file (defaults to fi) */ 7669408Sache#ifdef IS_ASCII 7769408Sache VAR(NOS, "lc", "\033["), /* Left code (ASCII) */ 7869408Sache#else 7959243Sobrien VAR(NOS, "lc", "\x27["), /* Left code (EBCDIC)*/ 8069408Sache#endif 8159243Sobrien VAR(NOS, "rc", "m"), /* Right code */ 8259243Sobrien VAR(NOS, "ec", ""), /* End code (replaces lc+no+rc) */ 83167465Smp VAR(NOS, "su", ""), /* Setuid file (u+s) */ 84167465Smp VAR(NOS, "sg", ""), /* Setgid file (g+s) */ 85167465Smp VAR(NOS, "tw", ""), /* Sticky and other writable dir (+t,o+w) */ 86167465Smp VAR(NOS, "ow", ""), /* Other writable dir (o+w) but not sticky */ 87167465Smp VAR(NOS, "st", ""), /* Sticky dir (+t) but not other writable */ 88195609Smp VAR(NOS, "rs", "0"), /* Reset to normal color */ 8959243Sobrien}; 9059243Sobrien 9159243Sobrienenum FileType { 9283098Smp VDir, VSym, VOrph, VPipe, VSock, VDoor, VBlock, VChr, VExe, 9359243Sobrien VFile, VNormal, VMiss, VLeft, VRight, VEnd 9459243Sobrien}; 9559243Sobrien 9659243Sobrien#define nvariables (sizeof(variables)/sizeof(variables[0])) 9759243Sobrien 9859243Sobrientypedef struct { 99167465Smp Str extension; /* file extension */ 100167465Smp Str color; /* color string */ 10159243Sobrien} Extension; 10259243Sobrien 10359243Sobrienstatic Extension *extensions = NULL; 104145479Smpstatic size_t nextensions = 0; 10559243Sobrien 10659243Sobrienstatic char *colors = NULL; 107145479Smpint color_context_ls = FALSE; /* do colored ls */ 108167465Smpstatic int color_context_lsmF = FALSE; /* do colored ls-F */ 10959243Sobrien 110167465Smpstatic int getstring (char **, const Char **, Str *, int); 111167465Smpstatic void put_color (const Str *); 112167465Smpstatic void print_color (const Char *, size_t, Char); 11359243Sobrien 11459243Sobrien/* set_color_context(): 11559243Sobrien */ 11659243Sobrienvoid 117167465Smpset_color_context(void) 11859243Sobrien{ 11959243Sobrien struct varent *vp = adrof(STRcolor); 12059243Sobrien 121100616Smp if (vp == NULL || vp->vec == NULL) { 12259243Sobrien color_context_ls = FALSE; 12359243Sobrien color_context_lsmF = FALSE; 124100616Smp } else if (!vp->vec[0] || vp->vec[0][0] == '\0') { 12559243Sobrien color_context_ls = TRUE; 12659243Sobrien color_context_lsmF = TRUE; 127100616Smp } else { 12859243Sobrien size_t i; 12959243Sobrien 13059243Sobrien color_context_ls = FALSE; 13159243Sobrien color_context_lsmF = FALSE; 13259243Sobrien for (i = 0; vp->vec[i]; i++) 13359243Sobrien if (Strcmp(vp->vec[i], STRls) == 0) 13459243Sobrien color_context_ls = TRUE; 13559243Sobrien else if (Strcmp(vp->vec[i], STRlsmF) == 0) 13659243Sobrien color_context_lsmF = TRUE; 13759243Sobrien } 13859243Sobrien} 13959243Sobrien 14059243Sobrien 14159243Sobrien/* getstring(): 14259243Sobrien */ 143167465Smpstatic int 144167465Smpgetstring(char **dp, const Char **sp, Str *pd, int f) 14559243Sobrien{ 14659243Sobrien const Char *s = *sp; 14759243Sobrien char *d = *dp; 148145479Smp eChar sc; 14959243Sobrien 150145479Smp while (*s && (*s & CHAR) != (Char)f && (*s & CHAR) != ':') { 15159243Sobrien if ((*s & CHAR) == '\\' || (*s & CHAR) == '^') { 152145479Smp if ((sc = parseescape(&s)) == CHAR_ERR) 15359243Sobrien return 0; 15459243Sobrien } 15559243Sobrien else 156145479Smp sc = *s++ & CHAR; 157145479Smp d += one_wctomb(d, sc); 15859243Sobrien } 15959243Sobrien 16059243Sobrien pd->s = *dp; 161167465Smp pd->len = d - *dp; 16259243Sobrien *sp = s; 16359243Sobrien *dp = d; 164145479Smp return *s == (Char)f; 16559243Sobrien} 16659243Sobrien 16759243Sobrien 16859243Sobrien/* parseLS_COLORS(): 16959243Sobrien * Parse the LS_COLORS environment variable 17059243Sobrien */ 17159243Sobrienvoid 172167465SmpparseLS_COLORS(const Char *value) 17359243Sobrien{ 174145479Smp size_t i, len; 175167465Smp const Char *v; /* pointer in value */ 17659243Sobrien char *c; /* pointer in colors */ 177145479Smp Extension *volatile e; /* pointer in extensions */ 17859415Sobrien jmp_buf_t osetexit; 179167465Smp size_t omark; 18059243Sobrien 181100616Smp (void) &e; 182100616Smp 18359243Sobrien /* init */ 184167465Smp xfree(extensions); 18559243Sobrien for (i = 0; i < nvariables; i++) 18659243Sobrien variables[i].color = variables[i].defaultcolor; 18759243Sobrien colors = NULL; 18859243Sobrien extensions = NULL; 18959243Sobrien nextensions = 0; 19059243Sobrien 19159243Sobrien if (value == NULL) 19259243Sobrien return; 19359243Sobrien 19459243Sobrien len = Strlen(value); 19559243Sobrien /* allocate memory */ 19659243Sobrien i = 1; 19759243Sobrien for (v = value; *v; v++) 19859243Sobrien if ((*v & CHAR) == ':') 19959243Sobrien i++; 200167465Smp extensions = xmalloc(len + i * sizeof(Extension)); 20159243Sobrien colors = i * sizeof(Extension) + (char *)extensions; 20259243Sobrien nextensions = 0; 20359243Sobrien 20459243Sobrien /* init pointers */ 20559243Sobrien v = value; 20659243Sobrien c = colors; 20759243Sobrien e = &extensions[0]; 20859243Sobrien 20959415Sobrien /* Prevent from crashing if unknown parameters are given. */ 21059415Sobrien 211167465Smp omark = cleanup_push_mark(); 21259415Sobrien getexit(osetexit); 21359415Sobrien 21459415Sobrien if (setexit() == 0) { 215167465Smp 21659243Sobrien /* parse */ 21759243Sobrien while (*v) { 21859243Sobrien switch (*v & CHAR) { 21959243Sobrien case ':': 22059243Sobrien v++; 22159243Sobrien continue; 22259243Sobrien 22359243Sobrien case '*': /* :*ext=color: */ 22459243Sobrien v++; 22559243Sobrien if (getstring(&c, &v, &e->extension, '=') && 22659243Sobrien 0 < e->extension.len) { 22759243Sobrien v++; 22859243Sobrien getstring(&c, &v, &e->color, ':'); 22959243Sobrien e++; 23059243Sobrien continue; 23159243Sobrien } 23259243Sobrien break; 23359243Sobrien 23459243Sobrien default: /* :vl=color: */ 23559243Sobrien if (v[0] && v[1] && (v[2] & CHAR) == '=') { 23659243Sobrien for (i = 0; i < nvariables; i++) 237145479Smp if ((Char)variables[i].variable[0] == (v[0] & CHAR) && 238145479Smp (Char)variables[i].variable[1] == (v[1] & CHAR)) 23959243Sobrien break; 24059243Sobrien if (i < nvariables) { 24159243Sobrien v += 3; 24259243Sobrien getstring(&c, &v, &variables[i].color, ':'); 24359243Sobrien continue; 24459243Sobrien } 24559243Sobrien else 24659243Sobrien stderror(ERR_BADCOLORVAR, v[0], v[1]); 24759243Sobrien } 24859243Sobrien break; 24959243Sobrien } 25059243Sobrien while (*v && (*v & CHAR) != ':') 25159243Sobrien v++; 25259243Sobrien } 25359415Sobrien } 25459243Sobrien 255167465Smp cleanup_pop_mark(omark); 25659415Sobrien resexit(osetexit); 25759415Sobrien 258167465Smp nextensions = e - extensions; 25959243Sobrien} 26059243Sobrien 26159243Sobrien/* put_color(): 26259243Sobrien */ 26359243Sobrienstatic void 264167465Smpput_color(const Str *color) 26559243Sobrien{ 26659243Sobrien size_t i; 267167465Smp const char *c = color->s; 268167465Smp int original_output_raw = output_raw; 26959243Sobrien 27059243Sobrien output_raw = TRUE; 271167465Smp cleanup_push(&original_output_raw, output_raw_restore); 27259243Sobrien for (i = color->len; 0 < i; i--) 27359243Sobrien xputchar(*c++); 274167465Smp cleanup_until(&original_output_raw); 27559243Sobrien} 27659243Sobrien 27759243Sobrien 27859243Sobrien/* print_color(): 27959243Sobrien */ 28059243Sobrienstatic void 281167465Smpprint_color(const Char *fname, size_t len, Char suffix) 28259243Sobrien{ 283145479Smp size_t i; 28459243Sobrien char *filename = short2str(fname); 28559243Sobrien char *last = filename + len; 286167465Smp Str *color = &variables[VFile].color; 28759243Sobrien 28859243Sobrien switch (suffix) { 28959243Sobrien case '>': /* File is a symbolic link pointing to 29059243Sobrien * a directory */ 291167465Smp color = &variables[VDir].color; 292167465Smp break; 29359243Sobrien case '+': /* File is a hidden directory [aix] or 29459243Sobrien * context dependent [hpux] */ 29559243Sobrien case ':': /* File is network special [hpux] */ 296167465Smp break; 29759243Sobrien default: 29859243Sobrien for (i = 0; i < nvariables; i++) 29959243Sobrien if (variables[i].suffix != NOS && 300145479Smp (Char)variables[i].suffix == suffix) { 30159243Sobrien color = &variables[i].color; 30259243Sobrien break; 30359243Sobrien } 30459243Sobrien if (i == nvariables) { 30559243Sobrien for (i = 0; i < nextensions; i++) 306167465Smp if (len >= extensions[i].extension.len 307167465Smp && strncmp(last - extensions[i].extension.len, 308167465Smp extensions[i].extension.s, 309167465Smp extensions[i].extension.len) == 0) { 31059243Sobrien color = &extensions[i].color; 31159243Sobrien break; 31259243Sobrien } 31359243Sobrien } 31459243Sobrien break; 31559243Sobrien } 31659243Sobrien 31759243Sobrien put_color(&variables[VLeft].color); 31859243Sobrien put_color(color); 31959243Sobrien put_color(&variables[VRight].color); 32059243Sobrien} 32159243Sobrien 32259243Sobrien 32359243Sobrien/* print_with_color(): 32459243Sobrien */ 32559243Sobrienvoid 326167465Smpprint_with_color(const Char *filename, size_t len, Char suffix) 32759243Sobrien{ 32859243Sobrien if (color_context_lsmF && 32959243Sobrien (haderr ? (didfds ? is2atty : isdiagatty) : 33059243Sobrien (didfds ? is1atty : isoutatty))) { 33159243Sobrien print_color(filename, len, suffix); 33259243Sobrien xprintf("%S", filename); 33359243Sobrien if (0 < variables[VEnd].color.len) 33459243Sobrien put_color(&variables[VEnd].color); 33559243Sobrien else { 33659243Sobrien put_color(&variables[VLeft].color); 33759243Sobrien put_color(&variables[VNormal].color); 33859243Sobrien put_color(&variables[VRight].color); 33959243Sobrien } 34059243Sobrien } 34159243Sobrien else 342145479Smp xprintf("%S", filename); 343145479Smp xputwchar(suffix); 34459243Sobrien} 34559243Sobrien 34659243Sobrien 34759243Sobrien#endif /* COLOR_LS_F */ 348