1 #include <stdio.h>
2 #include <stdarg.h>
3 #include <stdlib.h>
4 #include <string.h>
5 #include <sys/socket.h>
6 #include <sys/types.h>
7 #include <linux/if.h>
8
9 #include "color.h"
10
11 enum color {
12 C_RED,
13 C_GREEN,
14 C_YELLOW,
15 C_BLUE,
16 C_MAGENTA,
17 C_CYAN,
18 C_WHITE,
19 C_BOLD_RED,
20 C_BOLD_GREEN,
21 C_BOLD_YELLOW,
22 C_BOLD_BLUE,
23 C_BOLD_MAGENTA,
24 C_BOLD_CYAN,
25 C_BOLD_WHITE,
26 C_CLEAR
27 };
28
29 static const char * const color_codes[] = {
30 "\e[31m",
31 "\e[32m",
32 "\e[33m",
33 "\e[34m",
34 "\e[35m",
35 "\e[36m",
36 "\e[37m",
37 "\e[1;31m",
38 "\e[1;32m",
39 "\e[1;33m",
40 "\e[1;34m",
41 "\e[1;35m",
42 "\e[1;36m",
43 "\e[1;37m",
44 "\e[0m",
45 NULL,
46 };
47
48 /* light background */
49 static enum color attr_colors_light[] = {
50 C_CYAN,
51 C_YELLOW,
52 C_MAGENTA,
53 C_BLUE,
54 C_GREEN,
55 C_RED,
56 C_CLEAR,
57 };
58
59 /* dark background */
60 static enum color attr_colors_dark[] = {
61 C_BOLD_CYAN,
62 C_BOLD_YELLOW,
63 C_BOLD_MAGENTA,
64 C_BOLD_BLUE,
65 C_BOLD_GREEN,
66 C_BOLD_RED,
67 C_CLEAR
68 };
69
70 static int is_dark_bg;
71 static int color_is_enabled;
72
enable_color(void)73 void enable_color(void)
74 {
75 color_is_enabled = 1;
76 set_color_palette();
77 }
78
set_color_palette(void)79 void set_color_palette(void)
80 {
81 char *p = getenv("COLORFGBG");
82
83 /*
84 * COLORFGBG environment variable usually contains either two or three
85 * values separated by semicolons; we want the last value in either case.
86 * If this value is 0-6 or 8, background is dark.
87 */
88 if (p && (p = strrchr(p, ';')) != NULL
89 && ((p[1] >= '0' && p[1] <= '6') || p[1] == '8')
90 && p[2] == '\0')
91 is_dark_bg = 1;
92 }
93
check_if_color_enabled(void)94 void check_if_color_enabled(void)
95 {
96 if (color_is_enabled) {
97 fprintf(stderr, "Option \"-json\" conflicts with \"-color\".\n");
98 exit(1);
99 }
100 }
101
color_fprintf(FILE * fp,enum color_attr attr,const char * fmt,...)102 int color_fprintf(FILE *fp, enum color_attr attr, const char *fmt, ...)
103 {
104 int ret = 0;
105 va_list args;
106
107 va_start(args, fmt);
108
109 if (!color_is_enabled || attr == COLOR_NONE) {
110 ret = vfprintf(fp, fmt, args);
111 goto end;
112 }
113
114 ret += fprintf(fp, "%s", color_codes[is_dark_bg ?
115 attr_colors_dark[attr] : attr_colors_light[attr]]);
116
117 ret += vfprintf(fp, fmt, args);
118 ret += fprintf(fp, "%s", color_codes[C_CLEAR]);
119
120 end:
121 va_end(args);
122 return ret;
123 }
124
ifa_family_color(__u8 ifa_family)125 enum color_attr ifa_family_color(__u8 ifa_family)
126 {
127 switch (ifa_family) {
128 case AF_INET:
129 return COLOR_INET;
130 case AF_INET6:
131 return COLOR_INET6;
132 default:
133 return COLOR_NONE;
134 }
135 }
136
oper_state_color(__u8 state)137 enum color_attr oper_state_color(__u8 state)
138 {
139 switch (state) {
140 case IF_OPER_UP:
141 return COLOR_OPERSTATE_UP;
142 case IF_OPER_DOWN:
143 return COLOR_OPERSTATE_DOWN;
144 default:
145 return COLOR_NONE;
146 }
147 }
148