usage.c: remove reference to busybox.h
[oweals/busybox.git] / networking / netstat.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini netstat implementation(s) for busybox
4  * based in part on the netstat implementation from net-tools.
5  *
6  * Copyright (C) 2002 by Bart Visscher <magick@linux-fan.com>
7  *
8  * 2002-04-20
9  * IPV6 support added by Bart Visscher <magick@linux-fan.com>
10  *
11  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
12  */
13
14 #include "libbb.h"
15 #include "inet_common.h"
16
17 enum {
18         OPT_extended = 0x4,
19         OPT_showroute = 0x100,
20         OPT_widedisplay = 0x200 * ENABLE_FEATURE_NETSTAT_WIDE,
21 };
22 # define NETSTAT_OPTS "laentuwxr"USE_FEATURE_NETSTAT_WIDE("W")
23
24 #define NETSTAT_CONNECTED 0x01
25 #define NETSTAT_LISTENING 0x02
26 #define NETSTAT_NUMERIC   0x04
27 /* Must match getopt32 option string */
28 #define NETSTAT_TCP       0x10
29 #define NETSTAT_UDP       0x20
30 #define NETSTAT_RAW       0x40
31 #define NETSTAT_UNIX      0x80
32 #define NETSTAT_ALLPROTO  (NETSTAT_TCP|NETSTAT_UDP|NETSTAT_RAW|NETSTAT_UNIX)
33
34 static smallint flags = NETSTAT_CONNECTED | NETSTAT_ALLPROTO;
35
36 enum {
37         TCP_ESTABLISHED = 1,
38         TCP_SYN_SENT,
39         TCP_SYN_RECV,
40         TCP_FIN_WAIT1,
41         TCP_FIN_WAIT2,
42         TCP_TIME_WAIT,
43         TCP_CLOSE,
44         TCP_CLOSE_WAIT,
45         TCP_LAST_ACK,
46         TCP_LISTEN,
47         TCP_CLOSING /* now a valid state */
48 };
49
50 static const char * const tcp_state[] =
51 {
52         "",
53         "ESTABLISHED",
54         "SYN_SENT",
55         "SYN_RECV",
56         "FIN_WAIT1",
57         "FIN_WAIT2",
58         "TIME_WAIT",
59         "CLOSE",
60         "CLOSE_WAIT",
61         "LAST_ACK",
62         "LISTEN",
63         "CLOSING"
64 };
65
66 typedef enum {
67         SS_FREE = 0,     /* not allocated                */
68         SS_UNCONNECTED,  /* unconnected to any socket    */
69         SS_CONNECTING,   /* in process of connecting     */
70         SS_CONNECTED,    /* connected to socket          */
71         SS_DISCONNECTING /* in process of disconnecting  */
72 } socket_state;
73
74 #define SO_ACCEPTCON (1<<16)    /* performed a listen           */
75 #define SO_WAITDATA  (1<<17)    /* wait data to read            */
76 #define SO_NOSPACE   (1<<18)    /* no space to write            */
77
78 /* Standard printout size */
79 #define PRINT_IP_MAX_SIZE           23
80 #define PRINT_NET_CONN              "%s   %6ld %6ld %-23s %-23s %-12s\n"
81 #define PRINT_NET_CONN_HEADER       "\nProto Recv-Q Send-Q %-23s %-23s State\n"
82
83 /* When there are IPv6 connections the IPv6 addresses will be
84  * truncated to none-recognition. The '-W' option makes the
85  * address columns wide enough to accomodate for longest possible
86  * IPv6 addresses, i.e. addresses of the form
87  * xxxx:xxxx:xxxx:xxxx:xxxx:xxxx:ddd.ddd.ddd.ddd
88  */
89 #define PRINT_IP_MAX_SIZE_WIDE      51  /* INET6_ADDRSTRLEN + 5 for the port number */
90 #define PRINT_NET_CONN_WIDE         "%s   %6ld %6ld %-51s %-51s %-12s\n"
91 #define PRINT_NET_CONN_HEADER_WIDE  "\nProto Recv-Q Send-Q %-51s %-51s State\n"
92
93 static const char *net_conn_line = PRINT_NET_CONN;
94
95
96 #if ENABLE_FEATURE_IPV6
97 static void build_ipv6_addr(char* local_addr, struct sockaddr_in6* localaddr)
98 {
99         char addr6[INET6_ADDRSTRLEN];
100         struct in6_addr in6;
101                 
102         sscanf(local_addr, "%08X%08X%08X%08X",
103                    &in6.s6_addr32[0], &in6.s6_addr32[1],
104                    &in6.s6_addr32[2], &in6.s6_addr32[3]);
105         inet_ntop(AF_INET6, &in6, addr6, sizeof(addr6));
106         inet_pton(AF_INET6, addr6, (struct sockaddr *) &localaddr->sin6_addr);
107                 
108         localaddr->sin6_family = AF_INET6;
109 }
110 #endif
111
112 #if ENABLE_FEATURE_IPV6
113 static void build_ipv4_addr(char* local_addr, struct sockaddr_in6* localaddr)
114 #else
115 static void build_ipv4_addr(char* local_addr, struct sockaddr_in* localaddr)
116 #endif
117 {
118         sscanf(local_addr, "%X",
119                    &((struct sockaddr_in *) localaddr)->sin_addr.s_addr);
120         ((struct sockaddr *) localaddr)->sa_family = AF_INET;
121 }
122
123 static const char *get_sname(int port, const char *proto, int num)
124 {
125         /* hummm, we return static buffer here!! */
126         const char *str = itoa(ntohs(port));
127         if (!num) {
128                 struct servent *se = getservbyport(port, proto);
129                 if (se)
130                         str = se->s_name;
131         }
132         if (!port) {
133                 str = "*";
134         }
135         return str;
136 }
137
138 static void snprint_ip_port(char *ip_port, int size, struct sockaddr *addr, int port, const char *proto, int numeric)
139 {
140         const char *port_name;
141         int max_len;
142         int port_name_len;
143
144 // TODO: replace by xmalloc_sockaddr2host?
145 #if ENABLE_FEATURE_IPV6
146         if (addr->sa_family == AF_INET6) {
147                 INET6_rresolve(ip_port, size, (struct sockaddr_in6 *)addr,
148                         (numeric & NETSTAT_NUMERIC) ? 0x0fff : 0);
149         } else
150 #endif
151         {
152                 INET_rresolve(ip_port, size, (struct sockaddr_in *)addr,
153                         0x4000 | ((numeric & NETSTAT_NUMERIC) ? 0x0fff : 0),
154                         0xffffffff);
155         }
156         port_name = get_sname(htons(port), proto, numeric);
157         
158         max_len = (option_mask32 & OPT_widedisplay)
159                         ? (PRINT_IP_MAX_SIZE_WIDE - 1)
160                         : (PRINT_IP_MAX_SIZE - 1);
161         port_name_len = strlen(port_name);
162         if ((strlen(ip_port) + port_name_len) > max_len)
163                 ip_port[max_len - port_name_len] = '\0';
164         ip_port += strlen(ip_port);
165         *ip_port++ = ':';
166         strcpy(ip_port, port_name);
167 }
168
169 static void tcp_do_one(int lnr, const char *line)
170 {
171         char local_addr[64], rem_addr[64];
172         char more[512];
173         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
174 #if ENABLE_FEATURE_IPV6
175         struct sockaddr_in6 localaddr, remaddr;
176 #else
177         struct sockaddr_in localaddr, remaddr;
178 #endif
179         unsigned long rxq, txq, time_len, retr, inode;
180
181         if (lnr == 0)
182                 return;
183
184         more[0] = '\0';
185         num = sscanf(line,
186                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
187                         &d, local_addr, &local_port,
188                         rem_addr, &rem_port, &state,
189                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
190
191         if (strlen(local_addr) > 8) {
192 #if ENABLE_FEATURE_IPV6
193                 build_ipv6_addr(local_addr, &localaddr);
194                 build_ipv6_addr(rem_addr, &remaddr);
195 #endif
196         } else {
197                 build_ipv4_addr(local_addr, &localaddr);
198                 build_ipv4_addr(rem_addr, &remaddr);
199         }
200
201         if (num < 10) {
202                 bb_error_msg("warning, got bogus tcp line");
203                 return;
204         }
205
206         if ((rem_port && (flags & NETSTAT_CONNECTED))
207          || (!rem_port && (flags & NETSTAT_LISTENING))
208         ) {
209                 snprint_ip_port(local_addr, sizeof(local_addr),
210                                 (struct sockaddr *) &localaddr, local_port,
211                                 "tcp", flags & NETSTAT_NUMERIC);
212                 snprint_ip_port(rem_addr, sizeof(rem_addr),
213                                 (struct sockaddr *) &remaddr, rem_port,
214                                 "tcp", flags & NETSTAT_NUMERIC);
215                 printf(net_conn_line,
216                         "tcp", rxq, txq, local_addr, rem_addr, tcp_state[state]);
217         }
218 }
219
220 static void udp_do_one(int lnr, const char *line)
221 {
222         char local_addr[64], rem_addr[64];
223         const char *state_str;
224         char more[512];
225         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
226 #if ENABLE_FEATURE_IPV6
227         struct sockaddr_in6 localaddr, remaddr;
228 #else
229         struct sockaddr_in localaddr, remaddr;
230 #endif
231         unsigned long rxq, txq, time_len, retr, inode;
232
233         if (lnr == 0)
234                 return;
235
236         more[0] = '\0';
237         num = sscanf(line,
238                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
239                         &d, local_addr, &local_port,
240                         rem_addr, &rem_port, &state,
241                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
242
243         if (strlen(local_addr) > 8) {
244 #if ENABLE_FEATURE_IPV6
245                 /* Demangle what the kernel gives us */
246                 build_ipv6_addr(local_addr, &localaddr);
247                 build_ipv6_addr(rem_addr, &remaddr);
248 #endif
249         } else {
250                 build_ipv4_addr(local_addr, &localaddr);
251                 build_ipv4_addr(rem_addr, &remaddr);
252         }
253
254         if (num < 10) {
255                 bb_error_msg("warning, got bogus udp line");
256                 return;
257         }
258         switch (state) {
259                 case TCP_ESTABLISHED:
260                         state_str = "ESTABLISHED";
261                         break;
262                 case TCP_CLOSE:
263                         state_str = "";
264                         break;
265                 default:
266                         state_str = "UNKNOWN";
267                         break;
268         }
269
270 #if ENABLE_FEATURE_IPV6
271 # define notnull(A) ( \
272         ( (A.sin6_family == AF_INET6)                               \
273           && (A.sin6_addr.s6_addr32[0] | A.sin6_addr.s6_addr32[1] | \
274               A.sin6_addr.s6_addr32[2] | A.sin6_addr.s6_addr32[3])  \
275         ) || (                                                      \
276           (A.sin6_family == AF_INET)                                \
277           && ((struct sockaddr_in*)&A)->sin_addr.s_addr             \
278         )                                                           \
279 )
280 #else
281 # define notnull(A) (A.sin_addr.s_addr)
282 #endif
283         {
284                 int have_remaddr = notnull(remaddr);
285                 if ((have_remaddr && (flags & NETSTAT_CONNECTED))
286                  || (!have_remaddr && (flags & NETSTAT_LISTENING))
287                 ) {
288                         snprint_ip_port(local_addr, sizeof(local_addr),
289                                 (struct sockaddr *) &localaddr, local_port,
290                                 "udp", flags & NETSTAT_NUMERIC);
291                         snprint_ip_port(rem_addr, sizeof(rem_addr),
292                                 (struct sockaddr *) &remaddr, rem_port,
293                                 "udp", flags & NETSTAT_NUMERIC);
294                         printf(net_conn_line,
295                                 "udp", rxq, txq, local_addr, rem_addr, state_str);      
296                 }
297         }
298 }
299
300 static void raw_do_one(int lnr, const char *line)
301 {
302         char local_addr[64], rem_addr[64];
303         char more[512];
304         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
305 #if ENABLE_FEATURE_IPV6
306         struct sockaddr_in6 localaddr, remaddr;
307 #else
308         struct sockaddr_in localaddr, remaddr;
309 #endif
310         unsigned long rxq, txq, time_len, retr, inode;
311
312         if (lnr == 0)
313                 return;
314
315         more[0] = '\0';
316         num = sscanf(line,
317                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
318                         &d, local_addr, &local_port,
319                         rem_addr, &rem_port, &state,
320                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
321
322         if (strlen(local_addr) > 8) {
323 #if ENABLE_FEATURE_IPV6
324                 build_ipv6_addr(local_addr, &localaddr);
325                 build_ipv6_addr(rem_addr, &remaddr);
326 #endif
327         } else {
328                 build_ipv4_addr(local_addr, &localaddr);
329                 build_ipv4_addr(rem_addr, &remaddr);
330         }
331
332         if (num < 10) {
333                 bb_error_msg("warning, got bogus raw line");
334                 return;
335         }
336
337         {
338                 int have_remaddr = notnull(remaddr);
339                 if ((have_remaddr && (flags & NETSTAT_CONNECTED))
340                  || (!have_remaddr && (flags & NETSTAT_LISTENING))
341                 ) {
342                         snprint_ip_port(local_addr, sizeof(local_addr),
343                                 (struct sockaddr *) &localaddr, local_port,
344                                 "raw", flags & NETSTAT_NUMERIC);
345                         snprint_ip_port(rem_addr, sizeof(rem_addr),
346                                 (struct sockaddr *) &remaddr, rem_port,
347                                 "raw", flags & NETSTAT_NUMERIC);
348                         printf(net_conn_line,
349                                 "raw", rxq, txq, local_addr, rem_addr, itoa(state));
350                 }
351         }
352 }
353
354 static void unix_do_one(int nr, const char *line)
355 {
356         static smallint has_inode = 0;
357
358         char path[PATH_MAX], ss_flags[32];
359         const char *ss_proto, *ss_state, *ss_type;
360         int num, state, type, inode;
361         void *d;
362         unsigned long refcnt, proto, unix_flags;
363
364         if (nr == 0) {
365                 if (strstr(line, "Inode"))
366                         has_inode = 1;
367                 return;
368         }
369         path[0] = '\0';
370         num = sscanf(line, "%p: %lX %lX %lX %X %X %d %s",
371                         &d, &refcnt, &proto, &unix_flags, &type, &state, &inode, path);
372         if (num < 6) {
373                 bb_error_msg("warning, got bogus unix line");
374                 return;
375         }
376         if (!has_inode)
377                 sprintf(path, "%d", inode);
378
379         if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) != (NETSTAT_LISTENING|NETSTAT_CONNECTED)) {
380                 if ((state == SS_UNCONNECTED) && (unix_flags & SO_ACCEPTCON)) {
381                         if (!(flags & NETSTAT_LISTENING))
382                                 return;
383                 } else {
384                         if (!(flags & NETSTAT_CONNECTED))
385                                 return;
386                 }
387         }
388
389         switch (proto) {
390                 case 0:
391                         ss_proto = "unix";
392                         break;
393                 default:
394                         ss_proto = "??";
395         }
396
397         switch (type) {
398                 case SOCK_STREAM:
399                         ss_type = "STREAM";
400                         break;
401                 case SOCK_DGRAM:
402                         ss_type = "DGRAM";
403                         break;
404                 case SOCK_RAW:
405                         ss_type = "RAW";
406                         break;
407                 case SOCK_RDM:
408                         ss_type = "RDM";
409                         break;
410                 case SOCK_SEQPACKET:
411                         ss_type = "SEQPACKET";
412                         break;
413                 default:
414                         ss_type = "UNKNOWN";
415         }
416
417         switch (state) {
418                 case SS_FREE:
419                         ss_state = "FREE";
420                         break;
421                 case SS_UNCONNECTED:
422                         /*
423                          * Unconnected sockets may be listening
424                          * for something.
425                          */
426                         if (unix_flags & SO_ACCEPTCON) {
427                                 ss_state = "LISTENING";
428                         } else {
429                                 ss_state = "";
430                         }
431                         break;
432                 case SS_CONNECTING:
433                         ss_state = "CONNECTING";
434                         break;
435                 case SS_CONNECTED:
436                         ss_state = "CONNECTED";
437                         break;
438                 case SS_DISCONNECTING:
439                         ss_state = "DISCONNECTING";
440                         break;
441                 default:
442                         ss_state = "UNKNOWN";
443         }
444
445         strcpy(ss_flags, "[ ");
446         if (unix_flags & SO_ACCEPTCON)
447                 strcat(ss_flags, "ACC ");
448         if (unix_flags & SO_WAITDATA)
449                 strcat(ss_flags, "W ");
450         if (unix_flags & SO_NOSPACE)
451                 strcat(ss_flags, "N ");
452         strcat(ss_flags, "]");
453
454         printf("%-5s %-6ld %-11s %-10s %-13s ",
455                    ss_proto, refcnt, ss_flags, ss_type, ss_state);
456         if (has_inode)
457                 printf("%-6d ", inode);
458         else
459                 printf("-      ");
460         puts(path);
461 }
462
463 #define _PATH_PROCNET_UDP "/proc/net/udp"
464 #define _PATH_PROCNET_UDP6 "/proc/net/udp6"
465 #define _PATH_PROCNET_TCP "/proc/net/tcp"
466 #define _PATH_PROCNET_TCP6 "/proc/net/tcp6"
467 #define _PATH_PROCNET_RAW "/proc/net/raw"
468 #define _PATH_PROCNET_RAW6 "/proc/net/raw6"
469 #define _PATH_PROCNET_UNIX "/proc/net/unix"
470
471 static void do_info(const char *file, const char *name, void (*proc)(int, const char *))
472 {
473         int lnr = 0;
474         FILE *procinfo;
475
476         procinfo = fopen(file, "r");
477         if (procinfo == NULL) {
478                 if (errno != ENOENT) {
479                         bb_perror_msg("%s", file);
480                 } else {
481                         bb_error_msg("no support for '%s' on this system", name);
482                 }
483                 return;
484         }
485         do {
486                 char *buffer = xmalloc_fgets(procinfo);
487                 if (buffer) {
488                         (proc)(lnr++, buffer);
489                         free(buffer);
490                 }
491         } while (!feof(procinfo));
492         fclose(procinfo);
493 }
494
495 /*
496  * Our main function.
497  */
498
499 int netstat_main(int argc, char **argv);
500 int netstat_main(int argc, char **argv)
501 {
502         const char *net_conn_line_header = PRINT_NET_CONN_HEADER;
503         unsigned opt;
504 #if ENABLE_FEATURE_IPV6
505         smallint inet = 1;
506         smallint inet6 = 1;
507 #else
508         enum { inet = 1, inet6 = 0 };
509 #endif
510
511         /* Option string must match NETSTAT_xxx constants */
512         opt = getopt32(argc, argv, NETSTAT_OPTS);
513         if (opt & 0x1) { // -l
514                 flags &= ~NETSTAT_CONNECTED;
515                 flags |= NETSTAT_LISTENING;
516         }
517         if (opt & 0x2) flags |= NETSTAT_LISTENING | NETSTAT_CONNECTED; // -a
518         //if (opt & 0x4) // -e
519         if (opt & 0x8) flags |= NETSTAT_NUMERIC; // -n
520         //if (opt & 0x10) // -t: NETSTAT_TCP
521         //if (opt & 0x20) // -u: NETSTAT_UDP
522         //if (opt & 0x40) // -w: NETSTAT_RAW
523         //if (opt & 0x80) // -x: NETSTAT_UNIX
524         if (opt & OPT_showroute) { // -r
525 #if ENABLE_ROUTE
526                 bb_displayroutes(flags & NETSTAT_NUMERIC, !(opt & OPT_extended));
527                 return 0;
528 #else
529                 bb_show_usage();
530 #endif
531         }
532
533         if (opt & OPT_widedisplay) { // -W
534                 net_conn_line = PRINT_NET_CONN_WIDE;
535                 net_conn_line_header = PRINT_NET_CONN_HEADER_WIDE;
536         }
537
538         opt &= NETSTAT_ALLPROTO;
539         if (opt) {
540                 flags &= ~NETSTAT_ALLPROTO;
541                 flags |= opt;
542         }
543         if (flags & (NETSTAT_TCP|NETSTAT_UDP|NETSTAT_RAW)) {
544                 printf("Active Internet connections "); /* xxx */
545
546                 if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) == (NETSTAT_LISTENING|NETSTAT_CONNECTED))
547                         printf("(servers and established)");
548                 else if (flags & NETSTAT_LISTENING)
549                         printf("(only servers)");
550                 else
551                         printf("(w/o servers)");
552                 printf(net_conn_line_header, "Local Address", "Foreign Address");
553         }
554         if (inet && flags & NETSTAT_TCP)
555                 do_info(_PATH_PROCNET_TCP, "AF INET (tcp)", tcp_do_one);
556 #if ENABLE_FEATURE_IPV6
557         if (inet6 && flags & NETSTAT_TCP)
558                 do_info(_PATH_PROCNET_TCP6, "AF INET6 (tcp)", tcp_do_one);
559 #endif
560         if (inet && flags & NETSTAT_UDP)
561                 do_info(_PATH_PROCNET_UDP, "AF INET (udp)", udp_do_one);
562 #if ENABLE_FEATURE_IPV6
563         if (inet6 && flags & NETSTAT_UDP)
564                 do_info(_PATH_PROCNET_UDP6, "AF INET6 (udp)", udp_do_one);
565 #endif
566         if (inet && flags & NETSTAT_RAW)
567                 do_info(_PATH_PROCNET_RAW, "AF INET (raw)", raw_do_one);
568 #if ENABLE_FEATURE_IPV6
569         if (inet6 && flags & NETSTAT_RAW)
570                 do_info(_PATH_PROCNET_RAW6, "AF INET6 (raw)", raw_do_one);
571 #endif
572         if (flags & NETSTAT_UNIX) {
573                 printf("Active UNIX domain sockets ");
574                 if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) == (NETSTAT_LISTENING|NETSTAT_CONNECTED))
575                         printf("(servers and established)");
576                 else if (flags & NETSTAT_LISTENING)
577                         printf("(only servers)");
578                 else
579                         printf("(w/o servers)");
580                 printf("\nProto RefCnt Flags       Type       State         I-Node Path\n");
581                 do_info(_PATH_PROCNET_UNIX, "AF UNIX", unix_do_one);
582         }
583         return 0;
584 }