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 numeric)
 124 {
 125         if (!port)
 126                 return "*";
 127         if (!numeric) {
 128                 struct servent *se = getservbyport(port, proto);
 129                 if (se)
 130                         return se->s_name;
 131         }
 132         /* hummm, we may return static buffer here!! */
 133         return itoa(ntohs(port));
 134 }
 135
 136 static char *ip_port_str(struct sockaddr *addr, int port, const char *proto, int numeric)
 137 {
 138         enum { salen = USE_FEATURE_IPV6(sizeof(struct sockaddr_in6)) SKIP_FEATURE_IPV6(sizeof(struct sockaddr_in)) };
 139         char *host, *host_port;
 140
 141         /* Code which used "*" for INADDR_ANY is removed: it's ambiguous in IPv6,
 142          * while "0.0.0.0" is not. */
 143
 144         host = numeric ? xmalloc_sockaddr2dotted_noport(addr, salen)
 145                        : xmalloc_sockaddr2host_noport(addr, salen);
 146
 147         host_port = xasprintf("%s:%s", host, get_sname(htons(port), proto, numeric));
 148         free(host);
 149         return host_port;
 150 }
 151
 152 static void tcp_do_one(int lnr, const char *line)
 153 {
 154         char local_addr[64], rem_addr[64];
 155         char more[512];
 156         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
 157 #if ENABLE_FEATURE_IPV6
 158         struct sockaddr_in6 localaddr, remaddr;
 159 #else
 160         struct sockaddr_in localaddr, remaddr;
 161 #endif
 162         unsigned long rxq, txq, time_len, retr, inode;
 163
 164         if (lnr == 0)
 165                 return;
 166
 167         more[0] = '\0';
 168         num = sscanf(line,
 169                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
 170                         &d, local_addr, &local_port,
 171                         rem_addr, &rem_port, &state,
 172                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
 173
 174         if (strlen(local_addr) > 8) {
 175 #if ENABLE_FEATURE_IPV6
 176                 build_ipv6_addr(local_addr, &localaddr);
 177                 build_ipv6_addr(rem_addr, &remaddr);
 178 #endif
 179         } else {
 180                 build_ipv4_addr(local_addr, &localaddr);
 181                 build_ipv4_addr(rem_addr, &remaddr);
 182         }
 183
 184         if (num < 10) {
 185                 bb_error_msg("warning, got bogus tcp line");
 186                 return;
 187         }
 188
 189         if ((rem_port && (flags & NETSTAT_CONNECTED))
 190          || (!rem_port && (flags & NETSTAT_LISTENING))
 191         ) {
 192                 char *l = ip_port_str(
 193                                 (struct sockaddr *) &localaddr, local_port,
 194                                 "tcp", flags & NETSTAT_NUMERIC);
 195                 char *r = ip_port_str(
 196                                 (struct sockaddr *) &remaddr, rem_port,
 197                                 "tcp", flags & NETSTAT_NUMERIC);
 198                 printf(net_conn_line,
 199                         "tcp", rxq, txq, l, r, tcp_state[state]);
 200                 free(l);
 201                 free(r);
 202         }
 203 }
 204
 205 static void udp_do_one(int lnr, const char *line)
 206 {
 207         char local_addr[64], rem_addr[64];
 208         const char *state_str;
 209         char more[512];
 210         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
 211 #if ENABLE_FEATURE_IPV6
 212         struct sockaddr_in6 localaddr, remaddr;
 213 #else
 214         struct sockaddr_in localaddr, remaddr;
 215 #endif
 216         unsigned long rxq, txq, time_len, retr, inode;
 217
 218         if (lnr == 0)
 219                 return;
 220
 221         more[0] = '\0';
 222         num = sscanf(line,
 223                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
 224                         &d, local_addr, &local_port,
 225                         rem_addr, &rem_port, &state,
 226                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
 227
 228         if (strlen(local_addr) > 8) {
 229 #if ENABLE_FEATURE_IPV6
 230                 /* Demangle what the kernel gives us */
 231                 build_ipv6_addr(local_addr, &localaddr);
 232                 build_ipv6_addr(rem_addr, &remaddr);
 233 #endif
 234         } else {
 235                 build_ipv4_addr(local_addr, &localaddr);
 236                 build_ipv4_addr(rem_addr, &remaddr);
 237         }
 238
 239         if (num < 10) {
 240                 bb_error_msg("warning, got bogus udp line");
 241                 return;
 242         }
 243         switch (state) {
 244                 case TCP_ESTABLISHED:
 245                         state_str = "ESTABLISHED";
 246                         break;
 247                 case TCP_CLOSE:
 248                         state_str = "";
 249                         break;
 250                 default:
 251                         state_str = "UNKNOWN";
 252                         break;
 253         }
 254
 255 #if ENABLE_FEATURE_IPV6
 256 # define notnull(A) ( \
 257         ( (A.sin6_family == AF_INET6)                               \
 258           && (A.sin6_addr.s6_addr32[0] | A.sin6_addr.s6_addr32[1] | \
 259               A.sin6_addr.s6_addr32[2] | A.sin6_addr.s6_addr32[3])  \
 260         ) || (                                                      \
 261           (A.sin6_family == AF_INET)                                \
 262           && ((struct sockaddr_in*)&A)->sin_addr.s_addr             \
 263         )                                                           \
 264 )
 265 #else
 266 # define notnull(A) (A.sin_addr.s_addr)
 267 #endif
 268         {
 269                 int have_remaddr = notnull(remaddr);
 270                 if ((have_remaddr && (flags & NETSTAT_CONNECTED))
 271                  || (!have_remaddr && (flags & NETSTAT_LISTENING))
 272                 ) {
 273                         char *l = ip_port_str(
 274                                 (struct sockaddr *) &localaddr, local_port,
 275                                 "udp", flags & NETSTAT_NUMERIC);
 276                         char *r = ip_port_str(
 277                                 (struct sockaddr *) &remaddr, rem_port,
 278                                 "udp", flags & NETSTAT_NUMERIC);
 279                         printf(net_conn_line,
 280                                 "udp", rxq, txq, l, r, state_str);
 281                         free(l);
 282                         free(r);
 283                 }
 284         }
 285 }
 286
 287 static void raw_do_one(int lnr, const char *line)
 288 {
 289         char local_addr[64], rem_addr[64];
 290         char more[512];
 291         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
 292 #if ENABLE_FEATURE_IPV6
 293         struct sockaddr_in6 localaddr, remaddr;
 294 #else
 295         struct sockaddr_in localaddr, remaddr;
 296 #endif
 297         unsigned long rxq, txq, time_len, retr, inode;
 298
 299         if (lnr == 0)
 300                 return;
 301
 302         more[0] = '\0';
 303         num = sscanf(line,
 304                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
 305                         &d, local_addr, &local_port,
 306                         rem_addr, &rem_port, &state,
 307                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
 308
 309         if (strlen(local_addr) > 8) {
 310 #if ENABLE_FEATURE_IPV6
 311                 build_ipv6_addr(local_addr, &localaddr);
 312                 build_ipv6_addr(rem_addr, &remaddr);
 313 #endif
 314         } else {
 315                 build_ipv4_addr(local_addr, &localaddr);
 316                 build_ipv4_addr(rem_addr, &remaddr);
 317         }
 318
 319         if (num < 10) {
 320                 bb_error_msg("warning, got bogus raw line");
 321                 return;
 322         }
 323
 324         {
 325                 int have_remaddr = notnull(remaddr);
 326                 if ((have_remaddr && (flags & NETSTAT_CONNECTED))
 327                  || (!have_remaddr && (flags & NETSTAT_LISTENING))
 328                 ) {
 329                         char *l = ip_port_str(
 330                                 (struct sockaddr *) &localaddr, local_port,
 331                                 "raw", flags & NETSTAT_NUMERIC);
 332                         char *r = ip_port_str(
 333                                 (struct sockaddr *) &remaddr, rem_port,
 334                                 "raw", flags & NETSTAT_NUMERIC);
 335                         printf(net_conn_line,
 336                                 "raw", rxq, txq, l, r, itoa(state));
 337                         free(l);
 338                         free(r);
 339                 }
 340         }
 341 }
 342
 343 static void unix_do_one(int nr, const char *line)
 344 {
 345         static smallint has_inode = 0;
 346
 347         char path[PATH_MAX], ss_flags[32];
 348         const char *ss_proto, *ss_state, *ss_type;
 349         int num, state, type, inode;
 350         void *d;
 351         unsigned long refcnt, proto, unix_flags;
 352
 353         if (nr == 0) {
 354                 if (strstr(line, "Inode"))
 355                         has_inode = 1;
 356                 return;
 357         }
 358         path[0] = '\0';
 359         num = sscanf(line, "%p: %lX %lX %lX %X %X %d %s",
 360                         &d, &refcnt, &proto, &unix_flags, &type, &state, &inode, path);
 361         if (num < 6) {
 362                 bb_error_msg("warning, got bogus unix line");
 363                 return;
 364         }
 365         if (!has_inode)
 366                 sprintf(path, "%d", inode);
 367
 368         if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) != (NETSTAT_LISTENING|NETSTAT_CONNECTED)) {
 369                 if ((state == SS_UNCONNECTED) && (unix_flags & SO_ACCEPTCON)) {
 370                         if (!(flags & NETSTAT_LISTENING))
 371                                 return;
 372                 } else {
 373                         if (!(flags & NETSTAT_CONNECTED))
 374                                 return;
 375                 }
 376         }
 377
 378         switch (proto) {
 379                 case 0:
 380                         ss_proto = "unix";
 381                         break;
 382                 default:
 383                         ss_proto = "??";
 384         }
 385
 386         switch (type) {
 387                 case SOCK_STREAM:
 388                         ss_type = "STREAM";
 389                         break;
 390                 case SOCK_DGRAM:
 391                         ss_type = "DGRAM";
 392                         break;
 393                 case SOCK_RAW:
 394                         ss_type = "RAW";
 395                         break;
 396                 case SOCK_RDM:
 397                         ss_type = "RDM";
 398                         break;
 399                 case SOCK_SEQPACKET:
 400                         ss_type = "SEQPACKET";
 401                         break;
 402                 default:
 403                         ss_type = "UNKNOWN";
 404         }
 405
 406         switch (state) {
 407                 case SS_FREE:
 408                         ss_state = "FREE";
 409                         break;
 410                 case SS_UNCONNECTED:
 411                         /*
 412                          * Unconnected sockets may be listening
 413                          * for something.
 414                          */
 415                         if (unix_flags & SO_ACCEPTCON) {
 416                                 ss_state = "LISTENING";
 417                         } else {
 418                                 ss_state = "";
 419                         }
 420                         break;
 421                 case SS_CONNECTING:
 422                         ss_state = "CONNECTING";
 423                         break;
 424                 case SS_CONNECTED:
 425                         ss_state = "CONNECTED";
 426                         break;
 427                 case SS_DISCONNECTING:
 428                         ss_state = "DISCONNECTING";
 429                         break;
 430                 default:
 431                         ss_state = "UNKNOWN";
 432         }
 433
 434         strcpy(ss_flags, "[ ");
 435         if (unix_flags & SO_ACCEPTCON)
 436                 strcat(ss_flags, "ACC ");
 437         if (unix_flags & SO_WAITDATA)
 438                 strcat(ss_flags, "W ");
 439         if (unix_flags & SO_NOSPACE)
 440                 strcat(ss_flags, "N ");
 441         strcat(ss_flags, "]");
 442
 443         printf("%-5s %-6ld %-11s %-10s %-13s ",
 444                    ss_proto, refcnt, ss_flags, ss_type, ss_state);
 445         if (has_inode)
 446                 printf("%-6d ", inode);
 447         else
 448                 printf("-      ");
 449         puts(path);
 450 }
 451
 452 #define _PATH_PROCNET_UDP "/proc/net/udp"
 453 #define _PATH_PROCNET_UDP6 "/proc/net/udp6"
 454 #define _PATH_PROCNET_TCP "/proc/net/tcp"
 455 #define _PATH_PROCNET_TCP6 "/proc/net/tcp6"
 456 #define _PATH_PROCNET_RAW "/proc/net/raw"
 457 #define _PATH_PROCNET_RAW6 "/proc/net/raw6"
 458 #define _PATH_PROCNET_UNIX "/proc/net/unix"
 459
 460 static void do_info(const char *file, const char *name, void (*proc)(int, const char *))
 461 {
 462         int lnr = 0;
 463         FILE *procinfo;
 464
 465         procinfo = fopen(file, "r");
 466         if (procinfo == NULL) {
 467                 if (errno != ENOENT) {
 468                         bb_perror_msg("%s", file);
 469                 } else {
 470                         bb_error_msg("no support for '%s' on this system", name);
 471                 }
 472                 return;
 473         }
 474         do {
 475                 char *buffer = xmalloc_fgets(procinfo);
 476                 if (buffer) {
 477                         (proc)(lnr++, buffer);
 478                         free(buffer);
 479                 }
 480         } while (!feof(procinfo));
 481         fclose(procinfo);
 482 }
 483
 484 /*
 485  * Our main function.
 486  */
 487
 488 int netstat_main(int argc, char **argv);
 489 int netstat_main(int argc, char **argv)
 490 {
 491         const char *net_conn_line_header = PRINT_NET_CONN_HEADER;
 492         unsigned opt;
 493 #if ENABLE_FEATURE_IPV6
 494         smallint inet = 1;
 495         smallint inet6 = 1;
 496 #else
 497         enum { inet = 1, inet6 = 0 };
 498 #endif
 499
 500         /* Option string must match NETSTAT_xxx constants */
 501         opt = getopt32(argc, argv, NETSTAT_OPTS);
 502         if (opt & 0x1) { // -l
 503                 flags &= ~NETSTAT_CONNECTED;
 504                 flags |= NETSTAT_LISTENING;
 505         }
 506         if (opt & 0x2) flags |= NETSTAT_LISTENING | NETSTAT_CONNECTED; // -a
 507         //if (opt & 0x4) // -e
 508         if (opt & 0x8) flags |= NETSTAT_NUMERIC; // -n
 509         //if (opt & 0x10) // -t: NETSTAT_TCP
 510         //if (opt & 0x20) // -u: NETSTAT_UDP
 511         //if (opt & 0x40) // -w: NETSTAT_RAW
 512         //if (opt & 0x80) // -x: NETSTAT_UNIX
 513         if (opt & OPT_showroute) { // -r
 514 #if ENABLE_ROUTE
 515                 bb_displayroutes(flags & NETSTAT_NUMERIC, !(opt & OPT_extended));
 516                 return 0;
 517 #else
 518                 bb_show_usage();
 519 #endif
 520         }
 521
 522         if (opt & OPT_widedisplay) { // -W
 523                 net_conn_line = PRINT_NET_CONN_WIDE;
 524                 net_conn_line_header = PRINT_NET_CONN_HEADER_WIDE;
 525         }
 526
 527         opt &= NETSTAT_ALLPROTO;
 528         if (opt) {
 529                 flags &= ~NETSTAT_ALLPROTO;
 530                 flags |= opt;
 531         }
 532         if (flags & (NETSTAT_TCP|NETSTAT_UDP|NETSTAT_RAW)) {
 533                 printf("Active Internet connections "); /* xxx */
 534
 535                 if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) == (NETSTAT_LISTENING|NETSTAT_CONNECTED))
 536                         printf("(servers and established)");
 537                 else if (flags & NETSTAT_LISTENING)
 538                         printf("(only servers)");
 539                 else
 540                         printf("(w/o servers)");
 541                 printf(net_conn_line_header, "Local Address", "Foreign Address");
 542         }
 543         if (inet && flags & NETSTAT_TCP)
 544                 do_info(_PATH_PROCNET_TCP, "AF INET (tcp)", tcp_do_one);
 545 #if ENABLE_FEATURE_IPV6
 546         if (inet6 && flags & NETSTAT_TCP)
 547                 do_info(_PATH_PROCNET_TCP6, "AF INET6 (tcp)", tcp_do_one);
 548 #endif
 549         if (inet && flags & NETSTAT_UDP)
 550                 do_info(_PATH_PROCNET_UDP, "AF INET (udp)", udp_do_one);
 551 #if ENABLE_FEATURE_IPV6
 552         if (inet6 && flags & NETSTAT_UDP)
 553                 do_info(_PATH_PROCNET_UDP6, "AF INET6 (udp)", udp_do_one);
 554 #endif
 555         if (inet && flags & NETSTAT_RAW)
 556                 do_info(_PATH_PROCNET_RAW, "AF INET (raw)", raw_do_one);
 557 #if ENABLE_FEATURE_IPV6
 558         if (inet6 && flags & NETSTAT_RAW)
 559                 do_info(_PATH_PROCNET_RAW6, "AF INET6 (raw)", raw_do_one);
 560 #endif
 561         if (flags & NETSTAT_UNIX) {
 562                 printf("Active UNIX domain sockets ");
 563                 if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) == (NETSTAT_LISTENING|NETSTAT_CONNECTED))
 564                         printf("(servers and established)");
 565                 else if (flags & NETSTAT_LISTENING)
 566                         printf("(only servers)");
 567                 else
 568                         printf("(w/o servers)");
 569                 printf("\nProto RefCnt Flags       Type       State         I-Node Path\n");
 570                 do_info(_PATH_PROCNET_UNIX, "AF UNIX", unix_do_one);
 571         }
 572         return 0;
 573 }