tw.color.c revision 145479
1145479Smp/* $Header: /src/pub/tcsh/tw.color.c,v 1.18 2005/03/03 16:40:53 kim 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 35145479SmpRCSID("$Id: tw.color.c,v 1.18 2005/03/03 16:40:53 kim Exp $") 3659243Sobrien 3759243Sobrien#include "tw.h" 3859243Sobrien#include "ed.h" 3959243Sobrien#include "tc.h" 4059243Sobrien 4159243Sobrien#ifdef COLOR_LS_F 4259243Sobrien 4359243Sobrientypedef struct { 44145479Smp const char *s; 4559243Sobrien int 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; 5959243Sobrien Str color; 6059243Sobrien 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) */ 8359243Sobrien}; 8459243Sobrien 8559243Sobrienenum FileType { 8683098Smp VDir, VSym, VOrph, VPipe, VSock, VDoor, VBlock, VChr, VExe, 8759243Sobrien VFile, VNormal, VMiss, VLeft, VRight, VEnd 8859243Sobrien}; 8959243Sobrien 9059243Sobrien#define nvariables (sizeof(variables)/sizeof(variables[0])) 9159243Sobrien 9259243Sobrientypedef struct { 9359243Sobrien Str extension; /* file extension */ 9459243Sobrien Str color; /* color string */ 9559243Sobrien} Extension; 9659243Sobrien 9759243Sobrienstatic Extension *extensions = NULL; 98145479Smpstatic size_t nextensions = 0; 9959243Sobrien 10059243Sobrienstatic char *colors = NULL; 101145479Smpint color_context_ls = FALSE; /* do colored ls */ 102145479Smpstatic int color_context_lsmF = FALSE; /* do colored ls-F */ 10359243Sobrien 104145479Smpstatic int getstring __P((char **, const Char **, Str *, int)); 10559243Sobrienstatic void put_color __P((Str *)); 106145479Smpstatic void print_color __P((Char *, size_t, Char)); 10759243Sobrien 10859243Sobrien/* set_color_context(): 10959243Sobrien */ 11059243Sobrienvoid 11159243Sobrienset_color_context() 11259243Sobrien{ 11359243Sobrien struct varent *vp = adrof(STRcolor); 11459243Sobrien 115100616Smp if (vp == NULL || vp->vec == NULL) { 11659243Sobrien color_context_ls = FALSE; 11759243Sobrien color_context_lsmF = FALSE; 118100616Smp } else if (!vp->vec[0] || vp->vec[0][0] == '\0') { 11959243Sobrien color_context_ls = TRUE; 12059243Sobrien color_context_lsmF = TRUE; 121100616Smp } else { 12259243Sobrien size_t i; 12359243Sobrien 12459243Sobrien color_context_ls = FALSE; 12559243Sobrien color_context_lsmF = FALSE; 12659243Sobrien for (i = 0; vp->vec[i]; i++) 12759243Sobrien if (Strcmp(vp->vec[i], STRls) == 0) 12859243Sobrien color_context_ls = TRUE; 12959243Sobrien else if (Strcmp(vp->vec[i], STRlsmF) == 0) 13059243Sobrien color_context_lsmF = TRUE; 13159243Sobrien } 13259243Sobrien} 13359243Sobrien 13459243Sobrien 13559243Sobrien/* getstring(): 13659243Sobrien */ 137145479Smpstatic int 13859243Sobriengetstring(dp, sp, pd, f) 13959243Sobrien char **dp; /* dest buffer */ 14059243Sobrien const Char **sp; /* source buffer */ 14159243Sobrien Str *pd; /* pointer to dest buffer */ 14259243Sobrien int f; /* final character */ 14359243Sobrien{ 14459243Sobrien const Char *s = *sp; 14559243Sobrien char *d = *dp; 146145479Smp eChar sc; 14759243Sobrien 148145479Smp while (*s && (*s & CHAR) != (Char)f && (*s & CHAR) != ':') { 14959243Sobrien if ((*s & CHAR) == '\\' || (*s & CHAR) == '^') { 150145479Smp if ((sc = parseescape(&s)) == CHAR_ERR) 15159243Sobrien return 0; 15259243Sobrien } 15359243Sobrien else 154145479Smp sc = *s++ & CHAR; 155145479Smp d += one_wctomb(d, sc); 15659243Sobrien } 15759243Sobrien 15859243Sobrien pd->s = *dp; 15959243Sobrien pd->len = (int) (d - *dp); 16059243Sobrien *sp = s; 16159243Sobrien *dp = d; 162145479Smp return *s == (Char)f; 16359243Sobrien} 16459243Sobrien 16559243Sobrien 16659243Sobrien/* parseLS_COLORS(): 16759243Sobrien * Parse the LS_COLORS environment variable 16859243Sobrien */ 16959243Sobrienvoid 17059243SobrienparseLS_COLORS(value) 17159243Sobrien Char *value; /* LS_COLOR variable's value */ 17259243Sobrien{ 173145479Smp size_t i, len; 17459243Sobrien const Char *v; /* pointer in value */ 17559243Sobrien char *c; /* pointer in colors */ 176145479Smp Extension *volatile e; /* pointer in extensions */ 17759415Sobrien jmp_buf_t osetexit; 17859243Sobrien 179100616Smp (void) &e; 180100616Smp 18159243Sobrien /* init */ 18259243Sobrien if (extensions) 18359243Sobrien xfree((ptr_t) extensions); 18459243Sobrien for (i = 0; i < nvariables; i++) 18559243Sobrien variables[i].color = variables[i].defaultcolor; 18659243Sobrien colors = NULL; 18759243Sobrien extensions = NULL; 18859243Sobrien nextensions = 0; 18959243Sobrien 19059243Sobrien if (value == NULL) 19159243Sobrien return; 19259243Sobrien 19359243Sobrien len = Strlen(value); 19459243Sobrien /* allocate memory */ 19559243Sobrien i = 1; 19659243Sobrien for (v = value; *v; v++) 19759243Sobrien if ((*v & CHAR) == ':') 19859243Sobrien i++; 19959243Sobrien extensions = (Extension *) xmalloc((size_t) (len + i * sizeof(Extension))); 20059243Sobrien colors = i * sizeof(Extension) + (char *)extensions; 20159243Sobrien nextensions = 0; 20259243Sobrien 20359243Sobrien /* init pointers */ 20459243Sobrien v = value; 20559243Sobrien c = colors; 20659243Sobrien e = &extensions[0]; 20759243Sobrien 20859415Sobrien /* Prevent from crashing if unknown parameters are given. */ 20959415Sobrien 21059415Sobrien getexit(osetexit); 21159415Sobrien 21259415Sobrien if (setexit() == 0) { 21359415Sobrien 21459243Sobrien /* parse */ 21559243Sobrien while (*v) { 21659243Sobrien switch (*v & CHAR) { 21759243Sobrien case ':': 21859243Sobrien v++; 21959243Sobrien continue; 22059243Sobrien 22159243Sobrien case '*': /* :*ext=color: */ 22259243Sobrien v++; 22359243Sobrien if (getstring(&c, &v, &e->extension, '=') && 22459243Sobrien 0 < e->extension.len) { 22559243Sobrien v++; 22659243Sobrien getstring(&c, &v, &e->color, ':'); 22759243Sobrien e++; 22859243Sobrien continue; 22959243Sobrien } 23059243Sobrien break; 23159243Sobrien 23259243Sobrien default: /* :vl=color: */ 23359243Sobrien if (v[0] && v[1] && (v[2] & CHAR) == '=') { 23459243Sobrien for (i = 0; i < nvariables; i++) 235145479Smp if ((Char)variables[i].variable[0] == (v[0] & CHAR) && 236145479Smp (Char)variables[i].variable[1] == (v[1] & CHAR)) 23759243Sobrien break; 23859243Sobrien if (i < nvariables) { 23959243Sobrien v += 3; 24059243Sobrien getstring(&c, &v, &variables[i].color, ':'); 24159243Sobrien continue; 24259243Sobrien } 24359243Sobrien else 24459243Sobrien stderror(ERR_BADCOLORVAR, v[0], v[1]); 24559243Sobrien } 24659243Sobrien break; 24759243Sobrien } 24859243Sobrien while (*v && (*v & CHAR) != ':') 24959243Sobrien v++; 25059243Sobrien } 25159415Sobrien } 25259243Sobrien 25359415Sobrien resexit(osetexit); 25459415Sobrien 25559243Sobrien nextensions = (int) (e - extensions); 25659243Sobrien} 25759243Sobrien 25859243Sobrien 25959243Sobrien/* put_color(): 26059243Sobrien */ 26159243Sobrienstatic void 26259243Sobrienput_color(color) 26359243Sobrien Str *color; 26459243Sobrien{ 26559243Sobrien size_t i; 266145479Smp const char *c = color->s; 267145479Smp int original_output_raw = output_raw; 26859243Sobrien 26959243Sobrien output_raw = TRUE; 27059243Sobrien for (i = color->len; 0 < i; i--) 27159243Sobrien xputchar(*c++); 27259243Sobrien output_raw = original_output_raw; 27359243Sobrien} 27459243Sobrien 27559243Sobrien 27659243Sobrien/* print_color(): 27759243Sobrien */ 27859243Sobrienstatic void 27959243Sobrienprint_color(fname, len, suffix) 28059243Sobrien Char *fname; 28159243Sobrien size_t len; 282145479Smp Char suffix; 28359243Sobrien{ 284145479Smp size_t i; 28559243Sobrien char *filename = short2str(fname); 28659243Sobrien char *last = filename + len; 28759243Sobrien Str *color = &variables[VFile].color; 28859243Sobrien 28959243Sobrien switch (suffix) { 29059243Sobrien case '>': /* File is a symbolic link pointing to 29159243Sobrien * a directory */ 29259243Sobrien color = &variables[VDir].color; 29359243Sobrien break; 29459243Sobrien case '+': /* File is a hidden directory [aix] or 29559243Sobrien * context dependent [hpux] */ 29659243Sobrien case ':': /* File is network special [hpux] */ 29759243Sobrien break; 29859243Sobrien default: 29959243Sobrien for (i = 0; i < nvariables; i++) 30059243Sobrien if (variables[i].suffix != NOS && 301145479Smp (Char)variables[i].suffix == suffix) { 30259243Sobrien color = &variables[i].color; 30359243Sobrien break; 30459243Sobrien } 30559243Sobrien if (i == nvariables) { 30659243Sobrien for (i = 0; i < nextensions; i++) 30759243Sobrien if (strncmp(last - extensions[i].extension.len, 30859243Sobrien extensions[i].extension.s, 30959243Sobrien 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 32659243Sobrienprint_with_color(filename, len, suffix) 32759243Sobrien Char *filename; 32859243Sobrien size_t len; 329145479Smp Char suffix; 33059243Sobrien{ 33159243Sobrien if (color_context_lsmF && 33259243Sobrien (haderr ? (didfds ? is2atty : isdiagatty) : 33359243Sobrien (didfds ? is1atty : isoutatty))) { 33459243Sobrien print_color(filename, len, suffix); 33559243Sobrien xprintf("%S", filename); 33659243Sobrien if (0 < variables[VEnd].color.len) 33759243Sobrien put_color(&variables[VEnd].color); 33859243Sobrien else { 33959243Sobrien put_color(&variables[VLeft].color); 34059243Sobrien put_color(&variables[VNormal].color); 34159243Sobrien put_color(&variables[VRight].color); 34259243Sobrien } 34359243Sobrien } 34459243Sobrien else 345145479Smp xprintf("%S", filename); 346145479Smp xputwchar(suffix); 34759243Sobrien} 34859243Sobrien 34959243Sobrien 35059243Sobrien#endif /* COLOR_LS_F */ 351