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  * 2008-07-10
  12  * optional '-p' flag support ported from net-tools by G. Somlo <somlo@cmu.edu>
  13  *
  14  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  15  */
  16
  17 #include "libbb.h"
  18 #include "inet_common.h"
  19
  20 #define NETSTAT_OPTS "laentuwx" \
  21         USE_ROUTE(               "r") \
  22         USE_FEATURE_NETSTAT_WIDE("W") \
  23         USE_FEATURE_NETSTAT_PRG( "p")
  24
  25 enum {
  26         OPTBIT_KEEP_OLD = 7,
  27         USE_ROUTE(               OPTBIT_ROUTE,)
  28         USE_FEATURE_NETSTAT_WIDE(OPTBIT_WIDE ,)
  29         USE_FEATURE_NETSTAT_PRG( OPTBIT_PRG  ,)
  30         OPT_sock_listen = 1 << 0, // l
  31         OPT_sock_all    = 1 << 1, // a
  32         OPT_extended    = 1 << 2, // e
  33         OPT_noresolve   = 1 << 3, // n
  34         OPT_sock_tcp    = 1 << 4, // t
  35         OPT_sock_udp    = 1 << 5, // u
  36         OPT_sock_raw    = 1 << 6, // w
  37         OPT_sock_unix   = 1 << 7, // x
  38         OPT_route       = USE_ROUTE(               (1 << OPTBIT_ROUTE)) + 0, // r
  39         OPT_wide        = USE_FEATURE_NETSTAT_WIDE((1 << OPTBIT_WIDE )) + 0, // W
  40         OPT_prg         = USE_FEATURE_NETSTAT_PRG( (1 << OPTBIT_PRG  )) + 0, // p
  41 };
  42
  43 #define NETSTAT_CONNECTED 0x01
  44 #define NETSTAT_LISTENING 0x02
  45 #define NETSTAT_NUMERIC   0x04
  46 /* Must match getopt32 option string */
  47 #define NETSTAT_TCP       0x10
  48 #define NETSTAT_UDP       0x20
  49 #define NETSTAT_RAW       0x40
  50 #define NETSTAT_UNIX      0x80
  51 #define NETSTAT_ALLPROTO  (NETSTAT_TCP|NETSTAT_UDP|NETSTAT_RAW|NETSTAT_UNIX)
  52
  53
  54 enum {
  55         TCP_ESTABLISHED = 1,
  56         TCP_SYN_SENT,
  57         TCP_SYN_RECV,
  58         TCP_FIN_WAIT1,
  59         TCP_FIN_WAIT2,
  60         TCP_TIME_WAIT,
  61         TCP_CLOSE,
  62         TCP_CLOSE_WAIT,
  63         TCP_LAST_ACK,
  64         TCP_LISTEN,
  65         TCP_CLOSING, /* now a valid state */
  66 };
  67
  68 static const char *const tcp_state[] = {
  69         "",
  70         "ESTABLISHED",
  71         "SYN_SENT",
  72         "SYN_RECV",
  73         "FIN_WAIT1",
  74         "FIN_WAIT2",
  75         "TIME_WAIT",
  76         "CLOSE",
  77         "CLOSE_WAIT",
  78         "LAST_ACK",
  79         "LISTEN",
  80         "CLOSING"
  81 };
  82
  83 typedef enum {
  84         SS_FREE = 0,     /* not allocated                */
  85         SS_UNCONNECTED,  /* unconnected to any socket    */
  86         SS_CONNECTING,   /* in process of connecting     */
  87         SS_CONNECTED,    /* connected to socket          */
  88         SS_DISCONNECTING /* in process of disconnecting  */
  89 } socket_state;
  90
  91 #define SO_ACCEPTCON (1<<16)    /* performed a listen           */
  92 #define SO_WAITDATA  (1<<17)    /* wait data to read            */
  93 #define SO_NOSPACE   (1<<18)    /* no space to write            */
  94
  95 /* Standard printout size */
  96 #define PRINT_IP_MAX_SIZE           23
  97 #define PRINT_NET_CONN              "%s   %6ld %6ld %-23s %-23s %-12s"
  98 #define PRINT_NET_CONN_HEADER       "\nProto Recv-Q Send-Q %-23s %-23s State       "
  99
 100 /* When there are IPv6 connections the IPv6 addresses will be
 101  * truncated to none-recognition. The '-W' option makes the
 102  * address columns wide enough to accomodate for longest possible
 103  * IPv6 addresses, i.e. addresses of the form
 104  * xxxx:xxxx:xxxx:xxxx:xxxx:xxxx:ddd.ddd.ddd.ddd
 105  */
 106 #define PRINT_IP_MAX_SIZE_WIDE      51  /* INET6_ADDRSTRLEN + 5 for the port number */
 107 #define PRINT_NET_CONN_WIDE         "%s   %6ld %6ld %-51s %-51s %-12s"
 108 #define PRINT_NET_CONN_HEADER_WIDE  "\nProto Recv-Q Send-Q %-51s %-51s State       "
 109
 110
 111 #define PROGNAME_WIDTH     20
 112 #define PROGNAME_WIDTH_STR "20"
 113 /* PROGNAME_WIDTH chars: 12345678901234567890 */
 114 #define PROGNAME_BANNER "PID/Program name    "
 115
 116 struct prg_node {
 117         struct prg_node *next;
 118         long inode;
 119         char name[PROGNAME_WIDTH];
 120 };
 121
 122 #define PRG_HASH_SIZE 211
 123
 124
 125 struct globals {
 126         const char *net_conn_line;
 127         smallint flags;
 128 #if ENABLE_FEATURE_NETSTAT_PRG
 129         smallint prg_cache_loaded;
 130         struct prg_node *prg_hash[PRG_HASH_SIZE];
 131 #endif
 132 };
 133 #define G (*ptr_to_globals)
 134 #define flags            (G.flags           )
 135 #define net_conn_line    (G.net_conn_line   )
 136 #define prg_hash         (G.prg_hash        )
 137 #define prg_cache_loaded (G.prg_cache_loaded)
 138 #define INIT_G() do { \
 139         SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
 140         flags = NETSTAT_CONNECTED | NETSTAT_ALLPROTO; \
 141         net_conn_line = PRINT_NET_CONN; \
 142 } while (0)
 143
 144
 145 #if ENABLE_FEATURE_NETSTAT_PRG
 146
 147 /* Deliberately truncating long to unsigned *int* */
 148 #define PRG_HASHIT(x) ((unsigned)(x) % PRG_HASH_SIZE)
 149
 150 #define print_progname_banner() do { \
 151         if (option_mask32 & OPT_prg) printf(PROGNAME_BANNER); \
 152 } while (0)
 153
 154 static void prg_cache_add(long inode, char *name)
 155 {
 156         unsigned hi = PRG_HASHIT(inode);
 157         struct prg_node **pnp, *pn;
 158
 159         prg_cache_loaded = 2;
 160         for (pnp = prg_hash + hi; (pn = *pnp) != NULL; pnp = &pn->next) {
 161                 if (pn->inode == inode) {
 162                         /* Some warning should be appropriate here
 163                            as we got multiple processes for one i-node */
 164                         return;
 165                 }
 166         }
 167         *pnp = xzalloc(sizeof(struct prg_node));
 168         pn = *pnp;
 169         pn->inode = inode;
 170         safe_strncpy(pn->name, name, PROGNAME_WIDTH);
 171 }
 172
 173 static const char *prg_cache_get(long inode)
 174 {
 175         unsigned hi = PRG_HASHIT(inode);
 176         struct prg_node *pn;
 177
 178         for (pn = prg_hash[hi]; pn; pn = pn->next)
 179                 if (pn->inode == inode)
 180                         return pn->name;
 181         return "-";
 182 }
 183
 184 #if ENABLE_FEATURE_CLEAN_UP
 185 static void prg_cache_clear(void)
 186 {
 187         struct prg_node **pnp, *pn;
 188
 189         for (pnp = prg_hash; pnp < prg_hash + PRG_HASH_SIZE; pnp++) {
 190                 while ((pn = *pnp) != NULL) {
 191                         *pnp = pn->next;
 192                         free(pn);
 193                 }
 194         }
 195 }
 196 #else
 197 #define prg_cache_clear() ((void)0)
 198 #endif
 199
 200 static long extract_socket_inode(const char *lname) {
 201
 202         long inode = -1;
 203
 204         if (strncmp(lname, "socket:[", sizeof("socket:[")-1) == 0) {
 205                 /* "socket:[12345]", extract the "12345" as inode */
 206                 inode = bb_strtol(lname + sizeof("socket:[")-1, (char**)&lname, 0);
 207                 if (*lname != ']')
 208                         inode = -1;
 209         } else if (strncmp(lname, "[0000]:", sizeof("[0000]:")-1) == 0) {
 210                 /* "[0000]:12345", extract the "12345" as inode */
 211                 inode = bb_strtol(lname + sizeof("[0000]:")-1, NULL, 0);
 212                 if (errno) /* not NUL terminated? */
 213                         inode = -1;
 214         }
 215
 216 #if 0 /* bb_strtol returns all-ones bit pattern on ERANGE anyway */
 217         if (errno == ERANGE)
 218                 inode = -1;
 219 #endif
 220         return inode;
 221 }
 222
 223 static int FAST_FUNC file_act(const char *fileName,
 224                 struct stat *statbuf UNUSED_PARAM,
 225                 void *userData,
 226                 int depth UNUSED_PARAM)
 227 {
 228         char *linkname;
 229         long inode;
 230
 231         linkname = xmalloc_readlink(fileName);
 232         if (linkname != NULL) {
 233                 inode = extract_socket_inode(linkname);
 234                 free(linkname);
 235                 if (inode >= 0)
 236                         prg_cache_add(inode, (char *)userData);
 237         }
 238         return TRUE;
 239 }
 240
 241 static int FAST_FUNC dir_act(const char *fileName,
 242                 struct stat *statbuf UNUSED_PARAM,
 243                 void *userData UNUSED_PARAM,
 244                 int depth)
 245 {
 246         const char *shortName;
 247         char *p, *q;
 248         char cmdline_buf[512];
 249         int i;
 250
 251         if (depth == 0) /* "/proc" itself */
 252                 return TRUE; /* continue looking one level below /proc */
 253
 254         shortName = fileName + sizeof("/proc/")-1; /* point after "/proc/" */
 255         if (!isdigit(shortName[0])) /* skip /proc entries whic aren't processes */
 256                 return SKIP;
 257
 258         p = concat_path_file(fileName, "cmdline"); /* "/proc/PID/cmdline" */
 259         i = open_read_close(p, cmdline_buf, sizeof(cmdline_buf) - 1);
 260         free(p);
 261         if (i < 0)
 262                 return FALSE;
 263         cmdline_buf[i] = '\0';
 264         q = concat_path_file(shortName, bb_basename(cmdline_buf)); /* "PID/argv0" */
 265
 266         /* go through all files in /proc/PID/fd */
 267         p = concat_path_file(fileName, "fd");
 268         i = recursive_action(p, ACTION_RECURSE | ACTION_QUIET,
 269                                 file_act, NULL, (void *)q, 0);
 270
 271         free(p);
 272         free(q);
 273
 274         if (!i)
 275                 return FALSE;   /* signal permissions error to caller */
 276
 277         return SKIP;            /* caller should not recurse further into this dir. */
 278 }
 279
 280 static void prg_cache_load(void)
 281 {
 282         int load_ok;
 283
 284         prg_cache_loaded = 1;
 285         load_ok = recursive_action("/proc", ACTION_RECURSE | ACTION_QUIET,
 286                                 NULL, dir_act, NULL, 0);
 287         if (load_ok)
 288                 return;
 289
 290         if (prg_cache_loaded == 1)
 291                 bb_error_msg("can't scan /proc - are you root?");
 292         else
 293                 bb_error_msg("showing only processes with your user ID");
 294 }
 295
 296 #else
 297
 298 #define prg_cache_clear()       ((void)0)
 299 #define print_progname_banner() ((void)0)
 300
 301 #endif //ENABLE_FEATURE_NETSTAT_PRG
 302
 303
 304 #if ENABLE_FEATURE_IPV6
 305 static void build_ipv6_addr(char* local_addr, struct sockaddr_in6* localaddr)
 306 {
 307         char addr6[INET6_ADDRSTRLEN];
 308         struct in6_addr in6;
 309
 310         sscanf(local_addr, "%08X%08X%08X%08X",
 311                    &in6.s6_addr32[0], &in6.s6_addr32[1],
 312                    &in6.s6_addr32[2], &in6.s6_addr32[3]);
 313         inet_ntop(AF_INET6, &in6, addr6, sizeof(addr6));
 314         inet_pton(AF_INET6, addr6, (struct sockaddr *) &localaddr->sin6_addr);
 315
 316         localaddr->sin6_family = AF_INET6;
 317 }
 318 #endif
 319
 320 #if ENABLE_FEATURE_IPV6
 321 static void build_ipv4_addr(char* local_addr, struct sockaddr_in6* localaddr)
 322 #else
 323 static void build_ipv4_addr(char* local_addr, struct sockaddr_in* localaddr)
 324 #endif
 325 {
 326         sscanf(local_addr, "%X",
 327                    &((struct sockaddr_in *) localaddr)->sin_addr.s_addr);
 328         ((struct sockaddr *) localaddr)->sa_family = AF_INET;
 329 }
 330
 331 static const char *get_sname(int port, const char *proto, int numeric)
 332 {
 333         if (!port)
 334                 return "*";
 335         if (!numeric) {
 336                 struct servent *se = getservbyport(port, proto);
 337                 if (se)
 338                         return se->s_name;
 339         }
 340         /* hummm, we may return static buffer here!! */
 341         return itoa(ntohs(port));
 342 }
 343
 344 static char *ip_port_str(struct sockaddr *addr, int port, const char *proto, int numeric)
 345 {
 346         enum { salen = USE_FEATURE_IPV6(sizeof(struct sockaddr_in6)) SKIP_FEATURE_IPV6(sizeof(struct sockaddr_in)) };
 347         char *host, *host_port;
 348
 349         /* Code which used "*" for INADDR_ANY is removed: it's ambiguous in IPv6,
 350          * while "0.0.0.0" is not. */
 351
 352         host = numeric ? xmalloc_sockaddr2dotted_noport(addr)
 353                        : xmalloc_sockaddr2host_noport(addr);
 354
 355         host_port = xasprintf("%s:%s", host, get_sname(htons(port), proto, numeric));
 356         free(host);
 357         return host_port;
 358 }
 359
 360 static int tcp_do_one(int lnr, char *line)
 361 {
 362         char local_addr[64], rem_addr[64];
 363         char more[512];
 364         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
 365 #if ENABLE_FEATURE_IPV6
 366         struct sockaddr_in6 localaddr, remaddr;
 367 #else
 368         struct sockaddr_in localaddr, remaddr;
 369 #endif
 370         unsigned long rxq, txq, time_len, retr, inode;
 371
 372         if (lnr == 0)
 373                 return 0;
 374
 375         more[0] = '\0';
 376         num = sscanf(line,
 377                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
 378                         &d, local_addr, &local_port,
 379                         rem_addr, &rem_port, &state,
 380                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
 381
 382         if (num < 10) {
 383                 return 1; /* error */
 384         }
 385
 386         if (strlen(local_addr) > 8) {
 387 #if ENABLE_FEATURE_IPV6
 388                 build_ipv6_addr(local_addr, &localaddr);
 389                 build_ipv6_addr(rem_addr, &remaddr);
 390 #endif
 391         } else {
 392                 build_ipv4_addr(local_addr, &localaddr);
 393                 build_ipv4_addr(rem_addr, &remaddr);
 394         }
 395
 396         if ((rem_port && (flags & NETSTAT_CONNECTED))
 397          || (!rem_port && (flags & NETSTAT_LISTENING))
 398         ) {
 399                 char *l = ip_port_str(
 400                                 (struct sockaddr *) &localaddr, local_port,
 401                                 "tcp", flags & NETSTAT_NUMERIC);
 402                 char *r = ip_port_str(
 403                                 (struct sockaddr *) &remaddr, rem_port,
 404                                 "tcp", flags & NETSTAT_NUMERIC);
 405                 printf(net_conn_line,
 406                         "tcp", rxq, txq, l, r, tcp_state[state]);
 407 #if ENABLE_FEATURE_NETSTAT_PRG
 408                 if (option_mask32 & OPT_prg)
 409                         printf("%."PROGNAME_WIDTH_STR"s", prg_cache_get(inode));
 410 #endif
 411                 bb_putchar('\n');
 412                 free(l);
 413                 free(r);
 414         }
 415         return 0;
 416 }
 417
 418 static int udp_do_one(int lnr, char *line)
 419 {
 420         char local_addr[64], rem_addr[64];
 421         const char *state_str;
 422         char more[512];
 423         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
 424 #if ENABLE_FEATURE_IPV6
 425         struct sockaddr_in6 localaddr, remaddr;
 426 #else
 427         struct sockaddr_in localaddr, remaddr;
 428 #endif
 429         unsigned long rxq, txq, time_len, retr, inode;
 430
 431         if (lnr == 0)
 432                 return 0;
 433
 434         more[0] = '\0';
 435         num = sscanf(line,
 436                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
 437                         &d, local_addr, &local_port,
 438                         rem_addr, &rem_port, &state,
 439                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
 440
 441         if (strlen(local_addr) > 8) {
 442 #if ENABLE_FEATURE_IPV6
 443                 /* Demangle what the kernel gives us */
 444                 build_ipv6_addr(local_addr, &localaddr);
 445                 build_ipv6_addr(rem_addr, &remaddr);
 446 #endif
 447         } else {
 448                 build_ipv4_addr(local_addr, &localaddr);
 449                 build_ipv4_addr(rem_addr, &remaddr);
 450         }
 451
 452         if (num < 10) {
 453                 return 1; /* error */
 454         }
 455         switch (state) {
 456                 case TCP_ESTABLISHED:
 457                         state_str = "ESTABLISHED";
 458                         break;
 459                 case TCP_CLOSE:
 460                         state_str = "";
 461                         break;
 462                 default:
 463                         state_str = "UNKNOWN";
 464                         break;
 465         }
 466
 467 #if ENABLE_FEATURE_IPV6
 468 # define notnull(A) ( \
 469         ( (A.sin6_family == AF_INET6)                               \
 470           && (A.sin6_addr.s6_addr32[0] | A.sin6_addr.s6_addr32[1] | \
 471               A.sin6_addr.s6_addr32[2] | A.sin6_addr.s6_addr32[3])  \
 472         ) || (                                                      \
 473           (A.sin6_family == AF_INET)                                \
 474           && ((struct sockaddr_in*)&A)->sin_addr.s_addr             \
 475         )                                                           \
 476 )
 477 #else
 478 # define notnull(A) (A.sin_addr.s_addr)
 479 #endif
 480         {
 481                 int have_remaddr = notnull(remaddr);
 482                 if ((have_remaddr && (flags & NETSTAT_CONNECTED))
 483                  || (!have_remaddr && (flags & NETSTAT_LISTENING))
 484                 ) {
 485                         char *l = ip_port_str(
 486                                 (struct sockaddr *) &localaddr, local_port,
 487                                 "udp", flags & NETSTAT_NUMERIC);
 488                         char *r = ip_port_str(
 489                                 (struct sockaddr *) &remaddr, rem_port,
 490                                 "udp", flags & NETSTAT_NUMERIC);
 491                         printf(net_conn_line,
 492                                 "udp", rxq, txq, l, r, state_str);
 493 #if ENABLE_FEATURE_NETSTAT_PRG
 494                         if (option_mask32 & OPT_prg)
 495                                 printf("%."PROGNAME_WIDTH_STR"s", prg_cache_get(inode));
 496 #endif
 497                         bb_putchar('\n');
 498                         free(l);
 499                         free(r);
 500                 }
 501         }
 502         return 0;
 503 }
 504
 505 static int raw_do_one(int lnr, char *line)
 506 {
 507         char local_addr[64], rem_addr[64];
 508         char more[512];
 509         int num, local_port, rem_port, d, state, timer_run, uid, timeout;
 510 #if ENABLE_FEATURE_IPV6
 511         struct sockaddr_in6 localaddr, remaddr;
 512 #else
 513         struct sockaddr_in localaddr, remaddr;
 514 #endif
 515         unsigned long rxq, txq, time_len, retr, inode;
 516
 517         if (lnr == 0)
 518                 return 0;
 519
 520         more[0] = '\0';
 521         num = sscanf(line,
 522                         "%d: %64[0-9A-Fa-f]:%X %64[0-9A-Fa-f]:%X %X %lX:%lX %X:%lX %lX %d %d %ld %512s\n",
 523                         &d, local_addr, &local_port,
 524                         rem_addr, &rem_port, &state,
 525                         &txq, &rxq, &timer_run, &time_len, &retr, &uid, &timeout, &inode, more);
 526
 527         if (strlen(local_addr) > 8) {
 528 #if ENABLE_FEATURE_IPV6
 529                 build_ipv6_addr(local_addr, &localaddr);
 530                 build_ipv6_addr(rem_addr, &remaddr);
 531 #endif
 532         } else {
 533                 build_ipv4_addr(local_addr, &localaddr);
 534                 build_ipv4_addr(rem_addr, &remaddr);
 535         }
 536
 537         if (num < 10) {
 538                 return 1; /* error */
 539         }
 540
 541         {
 542                 int have_remaddr = notnull(remaddr);
 543                 if ((have_remaddr && (flags & NETSTAT_CONNECTED))
 544                  || (!have_remaddr && (flags & NETSTAT_LISTENING))
 545                 ) {
 546                         char *l = ip_port_str(
 547                                 (struct sockaddr *) &localaddr, local_port,
 548                                 "raw", flags & NETSTAT_NUMERIC);
 549                         char *r = ip_port_str(
 550                                 (struct sockaddr *) &remaddr, rem_port,
 551                                 "raw", flags & NETSTAT_NUMERIC);
 552                         printf(net_conn_line,
 553                                 "raw", rxq, txq, l, r, itoa(state));
 554 #if ENABLE_FEATURE_NETSTAT_PRG
 555                         if (option_mask32 & OPT_prg)
 556                                 printf("%-"PROGNAME_WIDTH_STR"s", prg_cache_get(inode));
 557 #endif
 558                         bb_putchar('\n');
 559                         free(l);
 560                         free(r);
 561                 }
 562         }
 563         return 0;
 564 }
 565
 566 static int unix_do_one(int nr, char *line)
 567 {
 568         unsigned long refcnt, proto, unix_flags;
 569         unsigned long inode;
 570         int type, state;
 571         int num, path_ofs;
 572         void *d;
 573         const char *ss_proto, *ss_state, *ss_type;
 574         char ss_flags[32];
 575
 576         if (nr == 0)
 577                 return 0; /* skip header */
 578
 579         /* 2.6.15 may report lines like "... @/tmp/fam-user-^@^@^@^@^@^@^@..."
 580          * Other users report long lines filled by NUL bytes.
 581          * (those ^@ are NUL bytes too). We see them as empty lines. */
 582         if (!line[0])
 583                 return 0;
 584
 585         path_ofs = 0; /* paranoia */
 586         num = sscanf(line, "%p: %lX %lX %lX %X %X %lu %n",
 587                         &d, &refcnt, &proto, &unix_flags, &type, &state, &inode, &path_ofs);
 588         if (num < 7) {
 589                 return 1; /* error */
 590         }
 591         if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) != (NETSTAT_LISTENING|NETSTAT_CONNECTED)) {
 592                 if ((state == SS_UNCONNECTED) && (unix_flags & SO_ACCEPTCON)) {
 593                         if (!(flags & NETSTAT_LISTENING))
 594                                 return 0;
 595                 } else {
 596                         if (!(flags & NETSTAT_CONNECTED))
 597                                 return 0;
 598                 }
 599         }
 600
 601         switch (proto) {
 602                 case 0:
 603                         ss_proto = "unix";
 604                         break;
 605                 default:
 606                         ss_proto = "??";
 607         }
 608
 609         switch (type) {
 610                 case SOCK_STREAM:
 611                         ss_type = "STREAM";
 612                         break;
 613                 case SOCK_DGRAM:
 614                         ss_type = "DGRAM";
 615                         break;
 616                 case SOCK_RAW:
 617                         ss_type = "RAW";
 618                         break;
 619                 case SOCK_RDM:
 620                         ss_type = "RDM";
 621                         break;
 622                 case SOCK_SEQPACKET:
 623                         ss_type = "SEQPACKET";
 624                         break;
 625                 default:
 626                         ss_type = "UNKNOWN";
 627         }
 628
 629         switch (state) {
 630                 case SS_FREE:
 631                         ss_state = "FREE";
 632                         break;
 633                 case SS_UNCONNECTED:
 634                         /*
 635                          * Unconnected sockets may be listening
 636                          * for something.
 637                          */
 638                         if (unix_flags & SO_ACCEPTCON) {
 639                                 ss_state = "LISTENING";
 640                         } else {
 641                                 ss_state = "";
 642                         }
 643                         break;
 644                 case SS_CONNECTING:
 645                         ss_state = "CONNECTING";
 646                         break;
 647                 case SS_CONNECTED:
 648                         ss_state = "CONNECTED";
 649                         break;
 650                 case SS_DISCONNECTING:
 651                         ss_state = "DISCONNECTING";
 652                         break;
 653                 default:
 654                         ss_state = "UNKNOWN";
 655         }
 656
 657         strcpy(ss_flags, "[ ");
 658         if (unix_flags & SO_ACCEPTCON)
 659                 strcat(ss_flags, "ACC ");
 660         if (unix_flags & SO_WAITDATA)
 661                 strcat(ss_flags, "W ");
 662         if (unix_flags & SO_NOSPACE)
 663                 strcat(ss_flags, "N ");
 664         strcat(ss_flags, "]");
 665
 666         printf("%-5s %-6ld %-11s %-10s %-13s %6lu ",
 667                 ss_proto, refcnt, ss_flags, ss_type, ss_state, inode
 668                 );
 669
 670 #if ENABLE_FEATURE_NETSTAT_PRG
 671         if (option_mask32 & OPT_prg)
 672                 printf("%-"PROGNAME_WIDTH_STR"s", prg_cache_get(inode));
 673 #endif
 674
 675         /* TODO: currently we stop at first NUL byte. Is it a problem? */
 676         line += path_ofs;
 677         *strchrnul(line, '\n') = '\0';
 678         while (*line)
 679                 fputc_printable(*line++, stdout);
 680         bb_putchar('\n');
 681         return 0;
 682 }
 683
 684 #define _PATH_PROCNET_UDP "/proc/net/udp"
 685 #define _PATH_PROCNET_UDP6 "/proc/net/udp6"
 686 #define _PATH_PROCNET_TCP "/proc/net/tcp"
 687 #define _PATH_PROCNET_TCP6 "/proc/net/tcp6"
 688 #define _PATH_PROCNET_RAW "/proc/net/raw"
 689 #define _PATH_PROCNET_RAW6 "/proc/net/raw6"
 690 #define _PATH_PROCNET_UNIX "/proc/net/unix"
 691
 692 static void do_info(const char *file, const char *name, int (*proc)(int, char *))
 693 {
 694         int lnr;
 695         FILE *procinfo;
 696         char *buffer;
 697
 698         procinfo = fopen_for_read(file);
 699         if (procinfo == NULL) {
 700                 if (errno != ENOENT) {
 701                         bb_simple_perror_msg(file);
 702                 } else {
 703                         bb_error_msg("no kernel support for %s", name);
 704                 }
 705                 return;
 706         }
 707         lnr = 0;
 708         /* Why? because xmalloc_fgets_str doesn't stop on NULs */
 709         while ((buffer = xmalloc_fgets_str(procinfo, "\n")) != NULL) {
 710                 if (proc(lnr++, buffer))
 711                         bb_error_msg("%s: bogus data on line %d", file, lnr);
 712                 free(buffer);
 713         }
 714         fclose(procinfo);
 715 }
 716
 717 int netstat_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 718 int netstat_main(int argc UNUSED_PARAM, char **argv)
 719 {
 720         const char *net_conn_line_header = PRINT_NET_CONN_HEADER;
 721         unsigned opt;
 722 #if ENABLE_FEATURE_IPV6
 723         smallint inet = 1;
 724         smallint inet6 = 1;
 725 #else
 726         enum { inet = 1, inet6 = 0 };
 727 #endif
 728
 729         INIT_G();
 730
 731         /* Option string must match NETSTAT_xxx constants */
 732         opt = getopt32(argv, NETSTAT_OPTS);
 733         if (opt & 0x1) { // -l
 734                 flags &= ~NETSTAT_CONNECTED;
 735                 flags |= NETSTAT_LISTENING;
 736         }
 737         if (opt & 0x2) flags |= NETSTAT_LISTENING | NETSTAT_CONNECTED; // -a
 738         //if (opt & 0x4) // -e
 739         if (opt & 0x8) flags |= NETSTAT_NUMERIC; // -n
 740         //if (opt & 0x10) // -t: NETSTAT_TCP
 741         //if (opt & 0x20) // -u: NETSTAT_UDP
 742         //if (opt & 0x40) // -w: NETSTAT_RAW
 743         //if (opt & 0x80) // -x: NETSTAT_UNIX
 744         if (opt & OPT_route) { // -r
 745 #if ENABLE_ROUTE
 746                 bb_displayroutes(flags & NETSTAT_NUMERIC, !(opt & OPT_extended));
 747                 return 0;
 748 #else
 749                 bb_show_usage();
 750 #endif
 751         }
 752         if (opt & OPT_wide) { // -W
 753                 net_conn_line = PRINT_NET_CONN_WIDE;
 754                 net_conn_line_header = PRINT_NET_CONN_HEADER_WIDE;
 755         }
 756 #if ENABLE_FEATURE_NETSTAT_PRG
 757         if (opt & OPT_prg) { // -p
 758                 prg_cache_load();
 759         }
 760 #endif
 761
 762         opt &= NETSTAT_ALLPROTO;
 763         if (opt) {
 764                 flags &= ~NETSTAT_ALLPROTO;
 765                 flags |= opt;
 766         }
 767         if (flags & (NETSTAT_TCP|NETSTAT_UDP|NETSTAT_RAW)) {
 768                 printf("Active Internet connections "); /* xxx */
 769
 770                 if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) == (NETSTAT_LISTENING|NETSTAT_CONNECTED))
 771                         printf("(servers and established)");
 772                 else if (flags & NETSTAT_LISTENING)
 773                         printf("(only servers)");
 774                 else
 775                         printf("(w/o servers)");
 776                 printf(net_conn_line_header, "Local Address", "Foreign Address");
 777                 print_progname_banner();
 778                 bb_putchar('\n');
 779         }
 780         if (inet && (flags & NETSTAT_TCP))
 781                 do_info(_PATH_PROCNET_TCP, "AF INET (tcp)", tcp_do_one);
 782 #if ENABLE_FEATURE_IPV6
 783         if (inet6 && (flags & NETSTAT_TCP))
 784                 do_info(_PATH_PROCNET_TCP6, "AF INET6 (tcp)", tcp_do_one);
 785 #endif
 786         if (inet && (flags & NETSTAT_UDP))
 787                 do_info(_PATH_PROCNET_UDP, "AF INET (udp)", udp_do_one);
 788 #if ENABLE_FEATURE_IPV6
 789         if (inet6 && (flags & NETSTAT_UDP))
 790                 do_info(_PATH_PROCNET_UDP6, "AF INET6 (udp)", udp_do_one);
 791 #endif
 792         if (inet && (flags & NETSTAT_RAW))
 793                 do_info(_PATH_PROCNET_RAW, "AF INET (raw)", raw_do_one);
 794 #if ENABLE_FEATURE_IPV6
 795         if (inet6 && (flags & NETSTAT_RAW))
 796                 do_info(_PATH_PROCNET_RAW6, "AF INET6 (raw)", raw_do_one);
 797 #endif
 798         if (flags & NETSTAT_UNIX) {
 799                 printf("Active UNIX domain sockets ");
 800                 if ((flags & (NETSTAT_LISTENING|NETSTAT_CONNECTED)) == (NETSTAT_LISTENING|NETSTAT_CONNECTED))
 801                         printf("(servers and established)");
 802                 else if (flags & NETSTAT_LISTENING)
 803                         printf("(only servers)");
 804                 else
 805                         printf("(w/o servers)");
 806                 printf("\nProto RefCnt Flags       Type       State         I-Node ");
 807                 print_progname_banner();
 808                 printf("Path\n");
 809                 do_info(_PATH_PROCNET_UNIX, "AF UNIX", unix_do_one);
 810         }
 811         prg_cache_clear();
 812         return 0;
 813 }