1: /*
2: * tui.c:
3: *
4: * Based on ui.c from the original iftop sources.
5: *
6: * This user interface does not make use of curses. Instead, it prints its
7: * output to STDOUT. This output is activated by providing the '-t' flag.
8: *
9: */
10:
11: #include "config.h"
12:
13: #include <string.h>
14: #include <stdio.h>
15: #include <signal.h>
16: #include <stdlib.h>
17: #include <unistd.h>
18:
19: #if defined(HAVE_TERMIOS_H)
20: #include <termios.h>
21: #elif defined(HAVE_SGTTY_H) && defined(TIOCSETP)
22: #include <sys/ioctl.h>
23: #include <sgtty.h>
24: #elif defined(HAVE_TERMIO_H)
25: #include <sys/ioctl.h>
26: #include <termio.h>
27: #else
28: #include <stdlib.h>
29: #endif
30:
31: #include "sorted_list.h"
32: #include "options.h"
33: #include "ui_common.h"
34:
35: /* Width of the host column in the output */
36: #define PRINT_WIDTH 40
37:
38:
39: /*
40: * UI print function
41: */
42: void tui_print() {
43: sorted_list_node* nn = NULL;
44: char host1[HOSTNAME_LENGTH], host2[HOSTNAME_LENGTH];
45: char buf0_10[10], buf1_10[10], buf2_10[10];
46: int j;
47: static char *label;
48: static char *labellong;
49: int l = 0;
50:
51: if (!label) {
52: xfree(label);
53: label = (char *)calloc(PRINT_WIDTH + 1, 1);
54: }
55:
56: if (!labellong) {
57: xfree(labellong);
58: labellong = (char *)calloc(PRINT_WIDTH + 1 + 9, 1);
59: }
60:
61: if (options.paused ) {
62: return;
63: }
64:
65: /* Headings */
66: snprintf(label, PRINT_WIDTH, "%-*s", PRINT_WIDTH, "Host name (port/service if enabled)");
67: printf("%s %s %10s %10s %10s %10s\n", " #", label, "last 2s", "last 10s", "last 40s", "cumulative");
68:
69: /* Divider line */
70: for (j = 0; j < PRINT_WIDTH + 52; j++) {
71: printf("-");
72: }
73: printf("\n");
74:
75: /* Traverse the list of all connections */
76: while((nn = sorted_list_next_item(&screen_list, nn)) != NULL && l < options.num_lines) {
77: /* Increment the line counter */
78: l++;
79:
80: /* Get the connection information */
81: host_pair_line* screen_line = (host_pair_line*)nn->data;
82:
83: /* Assemble host information */
84: sprint_host(host1, screen_line->ap.af, &(screen_line->ap.src6), screen_line->ap.src_port, screen_line->ap.protocol, PRINT_WIDTH, options.aggregate_src);
85: sprint_host(host2, screen_line->ap.af, &(screen_line->ap.dst6), screen_line->ap.dst_port, screen_line->ap.protocol, PRINT_WIDTH, options.aggregate_dest);
86:
87: /* Send rate per connection */
88: printf("%4d %s%s", l, host1, " =>");
89: for(j = 0; j < HISTORY_DIVISIONS; j++) {
90: readable_size(screen_line->sent[j], buf0_10, 10, 1024, options.bandwidth_in_bytes);
91: printf(" %10s", buf0_10);
92: }
93: /* Cumulative sent data per connection */
94: readable_size(screen_line->total_sent, buf0_10, 10, 1024, 1);
95: printf(" %10s\n", buf0_10);
96:
97: /* Receive rate per connection */
98: printf(" %s%s", host2, " <=");
99: for(j = 0; j < HISTORY_DIVISIONS; j++) {
100: readable_size(screen_line->recv[j], buf0_10, 10, 1024, options.bandwidth_in_bytes);
101: printf(" %10s", buf0_10);
102: }
103: /* Cumulative received data per connection */
104: readable_size(screen_line->total_recv, buf0_10, 10, 1024, 1);
105: printf(" %10s\n", buf0_10);
106: }
107:
108: /* Divider line */
109: for (j = 0; j < PRINT_WIDTH + 52; j++) {
110: printf("-");
111: }
112: printf("\n");
113:
114: /* Rate totals */
115: snprintf(labellong, PRINT_WIDTH + 9, "%-*s", PRINT_WIDTH + 9, "Total send rate:");
116: printf("%s ", labellong);
117: for(j = 0; j < HISTORY_DIVISIONS; j++) {
118: readable_size((((host_pair_line *)&totals)->sent[j]) , buf0_10, 10, 1024, options.bandwidth_in_bytes);
119: printf("%10s%c", buf0_10, j == HISTORY_DIVISIONS - 1 ? '\n' : ' ');
120: }
121:
122: snprintf(labellong, PRINT_WIDTH + 9, "%-*s", PRINT_WIDTH + 9, "Total receive rate:");
123: printf("%s ", labellong);
124: for(j = 0; j < HISTORY_DIVISIONS; j++) {
125: readable_size((((host_pair_line *)&totals)->recv[j]) , buf0_10, 10, 1024, options.bandwidth_in_bytes);
126: printf("%10s%c", buf0_10, j == HISTORY_DIVISIONS - 1 ? '\n' : ' ');
127: }
128:
129: snprintf(labellong, PRINT_WIDTH + 9, "%-*s", PRINT_WIDTH + 9, "Total send and receive rate:");
130: printf("%s ", labellong);
131: for(j = 0; j < HISTORY_DIVISIONS; j++) {
132: readable_size((((host_pair_line *)&totals)->sent[j] + ((host_pair_line *)&totals)->recv[j]) , buf0_10, 10, 1024, options.bandwidth_in_bytes);
133: printf("%10s%c", buf0_10, j == HISTORY_DIVISIONS - 1 ? '\n' : ' ');
134: }
135:
136: /* Divider line */
137: for (j = 0; j < PRINT_WIDTH + 52; j++) {
138: printf("-");
139: }
140: printf("\n");
141:
142: /* Peak traffic */
143: snprintf(labellong, PRINT_WIDTH + 9, "%-*s", PRINT_WIDTH + 9, "Peak rate (sent/received/total):");
144: readable_size(peaksent / RESOLUTION, buf0_10, 10, 1024, options.bandwidth_in_bytes);
145: readable_size(peakrecv / RESOLUTION, buf1_10, 10, 1024, options.bandwidth_in_bytes);
146: readable_size(peaktotal / RESOLUTION, buf2_10, 10, 1024, options.bandwidth_in_bytes);
147: printf("%s %10s %10s %10s\n", labellong, buf0_10, buf1_10, buf2_10);
148:
149: /* Cumulative totals */
150: snprintf(labellong, PRINT_WIDTH + 9, "%-*s", PRINT_WIDTH + 9, "Cumulative (sent/received/total):");
151: readable_size(history_totals.total_sent, buf0_10, 10, 1024, 1);
152: readable_size(history_totals.total_recv, buf1_10, 10, 1024, 1);
153: readable_size(history_totals.total_recv + history_totals.total_sent, buf2_10, 10, 1024, 1);
154: printf("%s %10s %10s %10s\n", labellong, buf0_10, buf1_10, buf2_10);
155:
156: /* Double divider line */
157: for (j = 0; j < PRINT_WIDTH + 52; j++) {
158: printf("=");
159: }
160: printf("\n\n");
161: }
162:
163:
164: /*
165: * Text interface data structure initializations.
166: */
167: void tui_init() {
168: screen_list_init();
169: screen_hash = addr_hash_create();
170: service_hash = serv_hash_create();
171: serv_hash_initialise(service_hash);
172:
173: printf("Listening on %s\n", options.interface);
174: }
175:
176:
177: /*
178: * Tick function indicating screen refresh
179: */
180: void tui_tick(int print) {
181: if (print) {
182: tui_print();
183: }
184: }
185:
186:
187: /*
188: * Main UI loop. Code any interactive character inputs here.
189: */
190: void tui_loop() {
191: int i;
192: extern sig_atomic_t foad;
193:
194: #if defined(HAVE_TERMIOS_H)
195: struct termios new_termios, old_termios;
196:
197: tcgetattr(STDIN_FILENO, &old_termios);
198: new_termios = old_termios;
199: new_termios.c_lflag &= ~(ICANON|ECHO);
200: new_termios.c_cc[VMIN] = 1;
201: new_termios.c_cc[VTIME] = 0;
202: tcsetattr(STDIN_FILENO, TCSANOW, &new_termios);
203: #elif defined(HAVE_SGTTY_H) && defined(TIOCSETP)
204: struct sgttyb new_tty, old_tty;
205:
206: ioctl(STDIN_FILENO, TIOCGETP, &old_tty);
207: new_tty = old_tty;
208: new_tty.sg_flags &= ~(ICANON|ECHO);
209: ioctl(STDIN_FILENO, TIOCSETP, &new_tty);
210: #elif defined(HAVE_TERMIO_H)
211: struct termio new_termio, old_termio;
212:
213: ioctl(0, TCGETA, &old_termio);
214: new_termio = old_termio;
215: new_termio.c_lflag &= ~(ICANON|ECHO);
216: new_termio.c_cc[VMIN] = 1;
217: new_termio.c_cc[VTIME] = 0;
218: ioctl(0, TCSETA, &new_termio);
219: #else
220: system("/bin/stty cbreak -echo >/dev/null 2>&1");
221: #endif
222:
223: while ((i = getchar()) != 'q' && foad == 0) {
224: switch (i) {
225: case 'u':
226: tick(1);
227: break;
228: case 'n':
229: options.dnsresolution ^= 1;
230: printf("DNS resolution is %s.\n\n", options.dnsresolution ? "ON" : "OFF");
231: tick(1);
232: break;
233: case 'N':
234: options.portresolution ^= 1;
235: printf("Port resolution is %s.\n\n", options.portresolution ? "ON" : "OFF");
236: tick(1);
237: break;
238: case 's':
239: options.aggregate_src ^= 1;
240: printf("%s source host\n\n", options.aggregate_src ? "Hide" : "Show");
241: tick(1);
242: break;
243: case 'd':
244: options.aggregate_dest ^= 1;
245: printf("%s destination host\n\n", options.aggregate_dest ? "Hide" : "Show");
246: tick(1);
247: break;
248: case 'S':
249: if (options.showports == OPTION_PORTS_OFF) {
250: options.showports = OPTION_PORTS_SRC;
251: }
252: else if (options.showports == OPTION_PORTS_DEST) {
253: options.showports = OPTION_PORTS_ON;
254: }
255: else if(options.showports == OPTION_PORTS_ON) {
256: options.showports = OPTION_PORTS_DEST;
257: }
258: else {
259: options.showports = OPTION_PORTS_OFF;
260: }
261: printf("Showing ports:%s%s%s%s.\n\n",
262: options.showports == OPTION_PORTS_SRC ? " src" : "",
263: options.showports == OPTION_PORTS_DEST ? " dest" : "",
264: options.showports == OPTION_PORTS_ON ? " both" : "",
265: options.showports == OPTION_PORTS_OFF ? " none" : "");
266: tick(1);
267: break;
268: case 'D':
269: if (options.showports == OPTION_PORTS_OFF) {
270: options.showports = OPTION_PORTS_DEST;
271: }
272: else if (options.showports == OPTION_PORTS_SRC) {
273: options.showports = OPTION_PORTS_ON;
274: }
275: else if(options.showports == OPTION_PORTS_ON) {
276: options.showports = OPTION_PORTS_SRC;
277: }
278: else {
279: options.showports = OPTION_PORTS_OFF;
280: }
281: printf("Showing ports:%s%s%s%s.\n\n",
282: options.showports == OPTION_PORTS_SRC ? " src" : "",
283: options.showports == OPTION_PORTS_DEST ? " dest" : "",
284: options.showports == OPTION_PORTS_ON ? " both" : "",
285: options.showports == OPTION_PORTS_OFF ? " none" : "");
286: tick(1);
287: break;
288: case 'p':
289: options.showports =
290: (options.showports == OPTION_PORTS_OFF) ?
291: OPTION_PORTS_ON :
292: OPTION_PORTS_OFF;
293: printf("Showing ports:%s%s%s%s.\n\n",
294: options.showports == OPTION_PORTS_SRC ? " src" : "",
295: options.showports == OPTION_PORTS_DEST ? " dest" : "",
296: options.showports == OPTION_PORTS_ON ? " both" : "",
297: options.showports == OPTION_PORTS_OFF ? " none" : "");
298: tick(1);
299: break;
300: case 'P':
301: options.paused ^= 1;
302: if (options.paused) {
303: printf("Pausing... press 'P' again to continue.\n");
304: }
305: else {
306: printf("Continuing.\n\n");
307: tick(1);
308: }
309: break;
310: case 'o':
311: options.freezeorder ^= 1;
312: printf("Order %s.\n\n", options.freezeorder ? "frozen" : "unfrozen");
313: tick(1);
314: break;
315: case '1':
316: options.sort = OPTION_SORT_DIV1;
317: printf("Sorting by column 1.\n\n");
318: tick(1);
319: break;
320: case '2':
321: options.sort = OPTION_SORT_DIV2;
322: printf("Sorting by column 2.\n\n");
323: tick(1);
324: break;
325: case '3':
326: options.sort = OPTION_SORT_DIV3;
327: printf("Sorting by column 3.\n\n");
328: tick(1);
329: break;
330: case '<':
331: options.sort = OPTION_SORT_SRC;
332: printf("Sorting by column source.\n\n");
333: tick(1);
334: break;
335: case '>':
336: options.sort = OPTION_SORT_DEST;
337: printf("Sorting by column destination.\n\n");
338: tick(1);
339: break;
340: default:
341: break;
342: }
343: }
344:
345: #if defined(HAVE_TERMIOS_H)
346: tcsetattr(STDIN_FILENO, TCSANOW, &old_termios);
347: #elif defined(HAVE_SGTTY_H) && defined(TIOCSETP)
348: ioctl(0, TIOCSETP, &old_tty);
349: #elif defined(HAVE_TERMIO_H)
350: ioctl(0, TCSETA, &old_termio);
351: #else
352: system("/bin/stty -cbreak echo >/dev/null 2>&1");
353: #endif
354: }
355:
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to tui.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / iftop