Support color on certain platforms, even when started at boot (before TERM is set)
[asterisk/asterisk.git] / main / term.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * Mark Spencer <markster@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief Terminal Routines 
22  *
23  * \author Mark Spencer <markster@digium.com> 
24  */
25
26 #include "asterisk.h"
27
28 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
29
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include <sys/time.h>
34 #include <signal.h>
35 #include <errno.h>
36 #include <sys/types.h>
37 #include <sys/stat.h>
38 #include <fcntl.h>
39 #include <unistd.h>
40
41 #include "asterisk/term.h"
42 #include "asterisk/options.h"
43 #include "asterisk/lock.h"
44 #include "asterisk/utils.h"
45
46 static int vt100compat;
47
48 static char prepdata[80] = "";
49 static char enddata[80] = "";
50 static char quitdata[80] = "";
51
52 static const char *termpath[] = {
53         "/usr/share/terminfo",
54         "/usr/local/share/misc/terminfo",
55         "/usr/lib/terminfo",
56         NULL
57         };
58
59 /* Ripped off from Ross Ridge, but it's public domain code (libmytinfo) */
60 static short convshort(char *s)
61 {
62         register int a,b;
63
64         a = (int) s[0] & 0377;
65         b = (int) s[1] & 0377;
66
67         if (a == 0377 && b == 0377)
68                 return -1;
69         if (a == 0376 && b == 0377)
70                 return -2;
71
72         return a + b * 256;
73 }
74
75 int ast_term_init(void)
76 {
77         char *term = getenv("TERM");
78         char termfile[256] = "";
79         char buffer[512] = "";
80         int termfd = -1, parseokay = 0, i;
81
82         if (!term)
83 #ifdef linux
84                 term = "linux";
85 #elif defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__)
86                 term = "cons25";
87 #elif defined(SOLARIS)
88                 term = "sun-color";
89 #else
90                 return 0;
91 #endif
92         if (ast_opt_no_color)
93                 return 0;
94
95         for (i=0 ;; i++) {
96                 if (termpath[i] == NULL) {
97                         break;
98                 }
99                 snprintf(termfile, sizeof(termfile), "%s/%c/%s", termpath[i], *term, term);
100                 termfd = open(termfile, O_RDONLY);
101                 if (termfd > -1) {
102                         break;
103                 }
104         }
105         if (termfd > -1) {
106                 int actsize = read(termfd, buffer, sizeof(buffer) - 1);
107                 short sz_names = convshort(buffer + 2);
108                 short sz_bools = convshort(buffer + 4);
109                 short n_nums   = convshort(buffer + 6);
110
111                 /* if ((sz_names + sz_bools) & 1)
112                         sz_bools++; */
113
114                 if (sz_names + sz_bools + n_nums < actsize) {
115                         /* Offset 13 is defined in /usr/include/term.h, though we do not
116                          * include it here, as it conflicts with include/asterisk/term.h */
117                         short max_colors = convshort(buffer + 12 + sz_names + sz_bools + 13 * 2);
118                         if (max_colors > 0) {
119                                 vt100compat = 1;
120                         }
121                         parseokay = 1;
122                 }
123                 close(termfd);
124         }
125
126         if (!parseokay) {
127                 /* These comparisons should not be substrings nor case-insensitive, as
128                  * terminal types are very particular about how they treat suffixes and
129                  * capitalization.  For example, terminal type 'linux-m' does NOT
130                  * support color, while 'linux' does.  Not even all vt100* terminals
131                  * support color, either (e.g. 'vt100+fnkeys'). */
132                 if (!strcmp(term, "linux")) {
133                         vt100compat = 1;
134                 } else if (!strcmp(term, "xterm")) {
135                         vt100compat = 1;
136                 } else if (!strcmp(term, "xterm-color")) {
137                         vt100compat = 1;
138                 } else if (!strncmp(term, "Eterm", 5)) {
139                         /* Both entries which start with Eterm support color */
140                         vt100compat = 1;
141                 } else if (!strcmp(term, "vt100")) {
142                         vt100compat = 1;
143                 } else if (!strncmp(term, "crt", 3)) {
144                         /* Both crt terminals support color */
145                         vt100compat = 1;
146                 }
147         }
148
149         if (vt100compat) {
150                 /* Make commands show up in nice colors */
151                 snprintf(prepdata, sizeof(prepdata), "%c[%d;%d;%dm", ESC, ATTR_BRIGHT, COLOR_BROWN, COLOR_BLACK + 10);
152                 snprintf(enddata, sizeof(enddata), "%c[%d;%d;%dm", ESC, ATTR_RESET, COLOR_WHITE, COLOR_BLACK + 10);
153                 snprintf(quitdata, sizeof(quitdata), "%c[0m", ESC);
154         }
155         return 0;
156 }
157
158 char *term_color(char *outbuf, const char *inbuf, int fgcolor, int bgcolor, int maxout)
159 {
160         int attr=0;
161         char tmp[40];
162         if (!vt100compat) {
163                 ast_copy_string(outbuf, inbuf, maxout);
164                 return outbuf;
165         }
166         if (!fgcolor && !bgcolor) {
167                 ast_copy_string(outbuf, inbuf, maxout);
168                 return outbuf;
169         }
170         if ((fgcolor & 128) && (bgcolor & 128)) {
171                 /* Can't both be highlighted */
172                 ast_copy_string(outbuf, inbuf, maxout);
173                 return outbuf;
174         }
175         if (!bgcolor)
176                 bgcolor = COLOR_BLACK;
177
178         if (bgcolor) {
179                 bgcolor &= ~128;
180                 bgcolor += 10;
181         }
182         if (fgcolor & 128) {
183                 attr = ATTR_BRIGHT;
184                 fgcolor &= ~128;
185         }
186         if (fgcolor && bgcolor) {
187                 snprintf(tmp, sizeof(tmp), "%d;%d", fgcolor, bgcolor);
188         } else if (bgcolor) {
189                 snprintf(tmp, sizeof(tmp), "%d", bgcolor);
190         } else if (fgcolor) {
191                 snprintf(tmp, sizeof(tmp), "%d", fgcolor);
192         }
193         if (attr) {
194                 snprintf(outbuf, maxout, "%c[%d;%sm%s%c[0;%d;%dm", ESC, attr, tmp, inbuf, ESC, COLOR_WHITE, COLOR_BLACK + 10);
195         } else {
196                 snprintf(outbuf, maxout, "%c[%sm%s%c[0;%d;%dm", ESC, tmp, inbuf, ESC, COLOR_WHITE, COLOR_BLACK + 10);
197         }
198         return outbuf;
199 }
200
201 char *term_color_code(char *outbuf, int fgcolor, int bgcolor, int maxout)
202 {
203         int attr=0;
204         char tmp[40];
205         if ((!vt100compat) || (!fgcolor && !bgcolor)) {
206                 *outbuf = '\0';
207                 return outbuf;
208         }
209         if ((fgcolor & 128) && (bgcolor & 128)) {
210                 /* Can't both be highlighted */
211                 *outbuf = '\0';
212                 return outbuf;
213         }
214         if (!bgcolor)
215                 bgcolor = COLOR_BLACK;
216
217         if (bgcolor) {
218                 bgcolor &= ~128;
219                 bgcolor += 10;
220         }
221         if (fgcolor & 128) {
222                 attr = ATTR_BRIGHT;
223                 fgcolor &= ~128;
224         }
225         if (fgcolor && bgcolor) {
226                 snprintf(tmp, sizeof(tmp), "%d;%d", fgcolor, bgcolor);
227         } else if (bgcolor) {
228                 snprintf(tmp, sizeof(tmp), "%d", bgcolor);
229         } else if (fgcolor) {
230                 snprintf(tmp, sizeof(tmp), "%d", fgcolor);
231         }
232         if (attr) {
233                 snprintf(outbuf, maxout, "%c[%d;%sm", ESC, attr, tmp);
234         } else {
235                 snprintf(outbuf, maxout, "%c[%sm", ESC, tmp);
236         }
237         return outbuf;
238 }
239
240 char *term_strip(char *outbuf, char *inbuf, int maxout)
241 {
242         char *outbuf_ptr = outbuf, *inbuf_ptr = inbuf;
243
244         while (outbuf_ptr < outbuf + maxout) {
245                 switch (*inbuf_ptr) {
246                         case ESC:
247                                 while (*inbuf_ptr && (*inbuf_ptr != 'm'))
248                                         inbuf_ptr++;
249                                 break;
250                         default:
251                                 *outbuf_ptr = *inbuf_ptr;
252                                 outbuf_ptr++;
253                 }
254                 if (! *inbuf_ptr)
255                         break;
256                 inbuf_ptr++;
257         }
258         return outbuf;
259 }
260
261 char *term_prompt(char *outbuf, const char *inbuf, int maxout)
262 {
263         if (!vt100compat) {
264                 ast_copy_string(outbuf, inbuf, maxout);
265                 return outbuf;
266         }
267         snprintf(outbuf, maxout, "%c[%d;%d;%dm%c%c[%d;%d;%dm%s",
268                 ESC, ATTR_BRIGHT, COLOR_BLUE, COLOR_BLACK + 10,
269                 inbuf[0],
270                 ESC, 0, COLOR_WHITE, COLOR_BLACK + 10,
271                 inbuf + 1);
272         return outbuf;
273 }
274
275 /* filter escape sequences */
276 void term_filter_escapes(char *line)
277 {
278         int i;
279         int len = strlen(line);
280
281         for (i = 0; i < len; i++) {
282                 if (line[i] != ESC)
283                         continue;
284                 if ((i < (len - 2)) &&
285                     (line[i + 1] == 0x5B)) {
286                         switch (line[i + 2]) {
287                         case 0x30:
288                         case 0x31:
289                         case 0x33:
290                                 continue;
291                         }
292                 }
293                 /* replace ESC with a space */
294                 line[i] = ' ';
295         }
296 }
297
298 char *term_prep(void)
299 {
300         return prepdata;
301 }
302
303 char *term_end(void)
304 {
305         return enddata;
306 }
307
308 char *term_quit(void)
309 {
310         return quitdata;
311 }