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