Home | History | Annotate | Download | only in misc
      1 /*
      2  * ss.c		"sockstat", socket statistics
      3  *
      4  *		This program is free software; you can redistribute it and/or
      5  *		modify it under the terms of the GNU General Public License
      6  *		as published by the Free Software Foundation; either version
      7  *		2 of the License, or (at your option) any later version.
      8  *
      9  * Authors:	Alexey Kuznetsov, <kuznet (at) ms2.inr.ac.ru>
     10  */
     11 
     12 #include <stdio.h>
     13 #include <stdlib.h>
     14 #include <unistd.h>
     15 #include <syslog.h>
     16 #include <fcntl.h>
     17 #include <sys/ioctl.h>
     18 #include <sys/socket.h>
     19 #include <sys/uio.h>
     20 #include <netinet/in.h>
     21 #include <string.h>
     22 #include <errno.h>
     23 #include <netdb.h>
     24 #include <arpa/inet.h>
     25 #include <resolv.h>
     26 #include <dirent.h>
     27 #include <fnmatch.h>
     28 #include <getopt.h>
     29 
     30 #include "utils.h"
     31 #include "rt_names.h"
     32 #include "ll_map.h"
     33 #include "libnetlink.h"
     34 #include "SNAPSHOT.h"
     35 
     36 #include <netinet/tcp.h>
     37 #include <linux/sock_diag.h>
     38 #include <linux/inet_diag.h>
     39 #include <linux/unix_diag.h>
     40 
     41 int resolve_hosts = 0;
     42 int resolve_services = 1;
     43 int preferred_family = AF_UNSPEC;
     44 int show_options = 0;
     45 int show_details = 0;
     46 int show_users = 0;
     47 int show_mem = 0;
     48 int show_tcpinfo = 0;
     49 
     50 int netid_width;
     51 int state_width;
     52 int addrp_width;
     53 int addr_width;
     54 int serv_width;
     55 int screen_width;
     56 
     57 static const char *TCP_PROTO = "tcp";
     58 static const char *UDP_PROTO = "udp";
     59 static const char *RAW_PROTO = "raw";
     60 static const char *dg_proto = NULL;
     61 
     62 enum
     63 {
     64 	TCP_DB,
     65 	DCCP_DB,
     66 	UDP_DB,
     67 	RAW_DB,
     68 	UNIX_DG_DB,
     69 	UNIX_ST_DB,
     70 	PACKET_DG_DB,
     71 	PACKET_R_DB,
     72 	NETLINK_DB,
     73 	MAX_DB
     74 };
     75 
     76 #define PACKET_DBM ((1<<PACKET_DG_DB)|(1<<PACKET_R_DB))
     77 #define UNIX_DBM ((1<<UNIX_DG_DB)|(1<<UNIX_ST_DB))
     78 #define ALL_DB ((1<<MAX_DB)-1)
     79 
     80 enum {
     81 	SS_UNKNOWN,
     82 	SS_ESTABLISHED,
     83 	SS_SYN_SENT,
     84 	SS_SYN_RECV,
     85 	SS_FIN_WAIT1,
     86 	SS_FIN_WAIT2,
     87 	SS_TIME_WAIT,
     88 	SS_CLOSE,
     89 	SS_CLOSE_WAIT,
     90 	SS_LAST_ACK,
     91 	SS_LISTEN,
     92 	SS_CLOSING,
     93 	SS_MAX
     94 };
     95 
     96 #define SS_ALL ((1<<SS_MAX)-1)
     97 
     98 #include "ssfilter.h"
     99 
    100 struct filter
    101 {
    102 	int dbs;
    103 	int states;
    104 	int families;
    105 	struct ssfilter *f;
    106 };
    107 
    108 struct filter default_filter = {
    109 	.dbs	=  (1<<TCP_DB),
    110 	.states = SS_ALL & ~((1<<SS_LISTEN)|(1<<SS_CLOSE)|(1<<SS_TIME_WAIT)|(1<<SS_SYN_RECV)),
    111 	.families= (1<<AF_INET)|(1<<AF_INET6),
    112 };
    113 
    114 struct filter current_filter;
    115 
    116 static FILE *generic_proc_open(const char *env, const char *name)
    117 {
    118 	const char *p = getenv(env);
    119 	char store[128];
    120 
    121 	if (!p) {
    122 		p = getenv("PROC_ROOT") ? : "/proc";
    123 		snprintf(store, sizeof(store)-1, "%s/%s", p, name);
    124 		p = store;
    125 	}
    126 
    127 	return fopen(p, "r");
    128 }
    129 
    130 static FILE *net_tcp_open(void)
    131 {
    132 	return generic_proc_open("PROC_NET_TCP", "net/tcp");
    133 }
    134 
    135 static FILE *net_tcp6_open(void)
    136 {
    137 	return generic_proc_open("PROC_NET_TCP6", "net/tcp6");
    138 }
    139 
    140 static FILE *net_udp_open(void)
    141 {
    142 	return generic_proc_open("PROC_NET_UDP", "net/udp");
    143 }
    144 
    145 static FILE *net_udp6_open(void)
    146 {
    147 	return generic_proc_open("PROC_NET_UDP6", "net/udp6");
    148 }
    149 
    150 static FILE *net_raw_open(void)
    151 {
    152 	return generic_proc_open("PROC_NET_RAW", "net/raw");
    153 }
    154 
    155 static FILE *net_raw6_open(void)
    156 {
    157 	return generic_proc_open("PROC_NET_RAW6", "net/raw6");
    158 }
    159 
    160 static FILE *net_unix_open(void)
    161 {
    162 	return generic_proc_open("PROC_NET_UNIX", "net/unix");
    163 }
    164 
    165 static FILE *net_packet_open(void)
    166 {
    167 	return generic_proc_open("PROC_NET_PACKET", "net/packet");
    168 }
    169 
    170 static FILE *net_netlink_open(void)
    171 {
    172 	return generic_proc_open("PROC_NET_NETLINK", "net/netlink");
    173 }
    174 
    175 static FILE *slabinfo_open(void)
    176 {
    177 	return generic_proc_open("PROC_SLABINFO", "slabinfo");
    178 }
    179 
    180 static FILE *net_sockstat_open(void)
    181 {
    182 	return generic_proc_open("PROC_NET_SOCKSTAT", "net/sockstat");
    183 }
    184 
    185 static FILE *net_sockstat6_open(void)
    186 {
    187 	return generic_proc_open("PROC_NET_SOCKSTAT6", "net/sockstat6");
    188 }
    189 
    190 static FILE *net_snmp_open(void)
    191 {
    192 	return generic_proc_open("PROC_NET_SNMP", "net/snmp");
    193 }
    194 
    195 static FILE *ephemeral_ports_open(void)
    196 {
    197 	return generic_proc_open("PROC_IP_LOCAL_PORT_RANGE", "sys/net/ipv4/ip_local_port_range");
    198 }
    199 
    200 struct user_ent {
    201 	struct user_ent	*next;
    202 	unsigned int	ino;
    203 	int		pid;
    204 	int		fd;
    205 	char		process[0];
    206 };
    207 
    208 #define USER_ENT_HASH_SIZE	256
    209 struct user_ent *user_ent_hash[USER_ENT_HASH_SIZE];
    210 
    211 static int user_ent_hashfn(unsigned int ino)
    212 {
    213 	int val = (ino >> 24) ^ (ino >> 16) ^ (ino >> 8) ^ ino;
    214 
    215 	return val & (USER_ENT_HASH_SIZE - 1);
    216 }
    217 
    218 static void user_ent_add(unsigned int ino, const char *process, int pid, int fd)
    219 {
    220 	struct user_ent *p, **pp;
    221 	int str_len;
    222 
    223 	str_len = strlen(process) + 1;
    224 	p = malloc(sizeof(struct user_ent) + str_len);
    225 	if (!p)
    226 		abort();
    227 	p->next = NULL;
    228 	p->ino = ino;
    229 	p->pid = pid;
    230 	p->fd = fd;
    231 	strcpy(p->process, process);
    232 
    233 	pp = &user_ent_hash[user_ent_hashfn(ino)];
    234 	p->next = *pp;
    235 	*pp = p;
    236 }
    237 
    238 static void user_ent_hash_build(void)
    239 {
    240 	const char *root = getenv("PROC_ROOT") ? : "/proc/";
    241 	struct dirent *d;
    242 	char name[1024];
    243 	int nameoff;
    244 	DIR *dir;
    245 
    246 	strcpy(name, root);
    247 	if (strlen(name) == 0 || name[strlen(name)-1] != '/')
    248 		strcat(name, "/");
    249 
    250 	nameoff = strlen(name);
    251 
    252 	dir = opendir(name);
    253 	if (!dir)
    254 		return;
    255 
    256 	while ((d = readdir(dir)) != NULL) {
    257 		struct dirent *d1;
    258 		char process[16];
    259 		int pid, pos;
    260 		DIR *dir1;
    261 		char crap;
    262 
    263 		if (sscanf(d->d_name, "%d%c", &pid, &crap) != 1)
    264 			continue;
    265 
    266 		sprintf(name + nameoff, "%d/fd/", pid);
    267 		pos = strlen(name);
    268 		if ((dir1 = opendir(name)) == NULL)
    269 			continue;
    270 
    271 		process[0] = '\0';
    272 
    273 		while ((d1 = readdir(dir1)) != NULL) {
    274 			const char *pattern = "socket:[";
    275 			unsigned int ino;
    276 			char lnk[64];
    277 			int fd;
    278 			ssize_t link_len;
    279 
    280 			if (sscanf(d1->d_name, "%d%c", &fd, &crap) != 1)
    281 				continue;
    282 
    283 			sprintf(name+pos, "%d", fd);
    284 
    285 			link_len = readlink(name, lnk, sizeof(lnk)-1);
    286 			if (link_len == -1)
    287 				continue;
    288 			lnk[link_len] = '\0';
    289 
    290 			if (strncmp(lnk, pattern, strlen(pattern)))
    291 				continue;
    292 
    293 			sscanf(lnk, "socket:[%u]", &ino);
    294 
    295 			if (process[0] == '\0') {
    296 				char tmp[1024];
    297 				FILE *fp;
    298 
    299 				snprintf(tmp, sizeof(tmp), "%s/%d/stat", root, pid);
    300 				if ((fp = fopen(tmp, "r")) != NULL) {
    301 					fscanf(fp, "%*d (%[^)])", process);
    302 					fclose(fp);
    303 				}
    304 			}
    305 
    306 			user_ent_add(ino, process, pid, fd);
    307 		}
    308 		closedir(dir1);
    309 	}
    310 	closedir(dir);
    311 }
    312 
    313 int find_users(unsigned ino, char *buf, int buflen)
    314 {
    315 	struct user_ent *p;
    316 	int cnt = 0;
    317 	char *ptr;
    318 
    319 	if (!ino)
    320 		return 0;
    321 
    322 	p = user_ent_hash[user_ent_hashfn(ino)];
    323 	ptr = buf;
    324 	while (p) {
    325 		if (p->ino != ino)
    326 			goto next;
    327 
    328 		if (ptr - buf >= buflen - 1)
    329 			break;
    330 
    331 		snprintf(ptr, buflen - (ptr - buf),
    332 			 "(\"%s\",%d,%d),",
    333 			 p->process, p->pid, p->fd);
    334 		ptr += strlen(ptr);
    335 		cnt++;
    336 
    337 	next:
    338 		p = p->next;
    339 	}
    340 
    341 	if (ptr != buf)
    342 		ptr[-1] = '\0';
    343 
    344 	return cnt;
    345 }
    346 
    347 /* Get stats from slab */
    348 
    349 struct slabstat
    350 {
    351 	int socks;
    352 	int tcp_ports;
    353 	int tcp_tws;
    354 	int tcp_syns;
    355 	int skbs;
    356 };
    357 
    358 struct slabstat slabstat;
    359 
    360 static const char *slabstat_ids[] =
    361 {
    362 	"sock",
    363 	"tcp_bind_bucket",
    364 	"tcp_tw_bucket",
    365 	"tcp_open_request",
    366 	"skbuff_head_cache",
    367 };
    368 
    369 int get_slabstat(struct slabstat *s)
    370 {
    371 	char buf[256];
    372 	FILE *fp;
    373 	int cnt;
    374 
    375 	memset(s, 0, sizeof(*s));
    376 
    377 	fp = slabinfo_open();
    378 	if (!fp)
    379 		return -1;
    380 
    381 	cnt = sizeof(*s)/sizeof(int);
    382 
    383 	fgets(buf, sizeof(buf), fp);
    384 	while(fgets(buf, sizeof(buf), fp) != NULL) {
    385 		int i;
    386 		for (i=0; i<sizeof(slabstat_ids)/sizeof(slabstat_ids[0]); i++) {
    387 			if (memcmp(buf, slabstat_ids[i], strlen(slabstat_ids[i])) == 0) {
    388 				sscanf(buf, "%*s%d", ((int *)s) + i);
    389 				cnt--;
    390 				break;
    391 			}
    392 		}
    393 		if (cnt <= 0)
    394 			break;
    395 	}
    396 
    397 	fclose(fp);
    398 	return 0;
    399 }
    400 
    401 static const char *sstate_name[] = {
    402 	"UNKNOWN",
    403 	[TCP_ESTABLISHED] = "ESTAB",
    404 	[TCP_SYN_SENT] = "SYN-SENT",
    405 	[TCP_SYN_RECV] = "SYN-RECV",
    406 	[TCP_FIN_WAIT1] = "FIN-WAIT-1",
    407 	[TCP_FIN_WAIT2] = "FIN-WAIT-2",
    408 	[TCP_TIME_WAIT] = "TIME-WAIT",
    409 	[TCP_CLOSE] = "UNCONN",
    410 	[TCP_CLOSE_WAIT] = "CLOSE-WAIT",
    411 	[TCP_LAST_ACK] = "LAST-ACK",
    412 	[TCP_LISTEN] = 	"LISTEN",
    413 	[TCP_CLOSING] = "CLOSING",
    414 };
    415 
    416 static const char *sstate_namel[] = {
    417 	"UNKNOWN",
    418 	[TCP_ESTABLISHED] = "established",
    419 	[TCP_SYN_SENT] = "syn-sent",
    420 	[TCP_SYN_RECV] = "syn-recv",
    421 	[TCP_FIN_WAIT1] = "fin-wait-1",
    422 	[TCP_FIN_WAIT2] = "fin-wait-2",
    423 	[TCP_TIME_WAIT] = "time-wait",
    424 	[TCP_CLOSE] = "unconnected",
    425 	[TCP_CLOSE_WAIT] = "close-wait",
    426 	[TCP_LAST_ACK] = "last-ack",
    427 	[TCP_LISTEN] = 	"listening",
    428 	[TCP_CLOSING] = "closing",
    429 };
    430 
    431 struct tcpstat
    432 {
    433 	inet_prefix	local;
    434 	inet_prefix	remote;
    435 	int		lport;
    436 	int		rport;
    437 	int		state;
    438 	int		rq, wq;
    439 	int		timer;
    440 	int		timeout;
    441 	int		retrs;
    442 	unsigned	ino;
    443 	int		probes;
    444 	unsigned	uid;
    445 	int		refcnt;
    446 	unsigned long long sk;
    447 	int		rto, ato, qack, cwnd, ssthresh;
    448 };
    449 
    450 static const char *tmr_name[] = {
    451 	"off",
    452 	"on",
    453 	"keepalive",
    454 	"timewait",
    455 	"persist",
    456 	"unknown"
    457 };
    458 
    459 const char *print_ms_timer(int timeout)
    460 {
    461 	static char buf[64];
    462 	int secs, msecs, minutes;
    463 	if (timeout < 0)
    464 		timeout = 0;
    465 	secs = timeout/1000;
    466 	minutes = secs/60;
    467 	secs = secs%60;
    468 	msecs = timeout%1000;
    469 	buf[0] = 0;
    470 	if (minutes) {
    471 		msecs = 0;
    472 		snprintf(buf, sizeof(buf)-16, "%dmin", minutes);
    473 		if (minutes > 9)
    474 			secs = 0;
    475 	}
    476 	if (secs) {
    477 		if (secs > 9)
    478 			msecs = 0;
    479 		sprintf(buf+strlen(buf), "%d%s", secs, msecs ? "." : "sec");
    480 	}
    481 	if (msecs)
    482 		sprintf(buf+strlen(buf), "%03dms", msecs);
    483 	return buf;
    484 }
    485 
    486 const char *print_hz_timer(int timeout)
    487 {
    488 	int hz = get_user_hz();
    489 	return print_ms_timer(((timeout*1000) + hz-1)/hz);
    490 }
    491 
    492 struct scache
    493 {
    494 	struct scache *next;
    495 	int port;
    496 	char *name;
    497 	const char *proto;
    498 };
    499 
    500 struct scache *rlist;
    501 
    502 void init_service_resolver(void)
    503 {
    504 	char buf[128];
    505 	FILE *fp = popen("/usr/sbin/rpcinfo -p 2>/dev/null", "r");
    506 	if (fp) {
    507 		fgets(buf, sizeof(buf), fp);
    508 		while (fgets(buf, sizeof(buf), fp) != NULL) {
    509 			unsigned int progn, port;
    510 			char proto[128], prog[128];
    511 			if (sscanf(buf, "%u %*d %s %u %s", &progn, proto,
    512 				   &port, prog+4) == 4) {
    513 				struct scache *c = malloc(sizeof(*c));
    514 				if (c) {
    515 					c->port = port;
    516 					memcpy(prog, "rpc.", 4);
    517 					c->name = strdup(prog);
    518 					if (strcmp(proto, TCP_PROTO) == 0)
    519 						c->proto = TCP_PROTO;
    520 					else if (strcmp(proto, UDP_PROTO) == 0)
    521 						c->proto = UDP_PROTO;
    522 					else
    523 						c->proto = NULL;
    524 					c->next = rlist;
    525 					rlist = c;
    526 				}
    527 			}
    528 		}
    529 		pclose(fp);
    530 	}
    531 }
    532 
    533 static int ip_local_port_min, ip_local_port_max;
    534 
    535 /* Even do not try default linux ephemeral port ranges:
    536  * default /etc/services contains so much of useless crap
    537  * wouldbe "allocated" to this area that resolution
    538  * is really harmful. I shrug each time when seeing
    539  * "socks" or "cfinger" in dumps.
    540  */
    541 static int is_ephemeral(int port)
    542 {
    543 	if (!ip_local_port_min) {
    544 		FILE *f = ephemeral_ports_open();
    545 		if (f) {
    546 			fscanf(f, "%d %d",
    547 			       &ip_local_port_min, &ip_local_port_max);
    548 			fclose(f);
    549 		} else {
    550 			ip_local_port_min = 1024;
    551 			ip_local_port_max = 4999;
    552 		}
    553 	}
    554 
    555 	return (port >= ip_local_port_min && port<= ip_local_port_max);
    556 }
    557 
    558 
    559 const char *__resolve_service(int port)
    560 {
    561 	struct scache *c;
    562 
    563 	for (c = rlist; c; c = c->next) {
    564 		if (c->port == port && c->proto == dg_proto)
    565 			return c->name;
    566 	}
    567 
    568 	if (!is_ephemeral(port)) {
    569 		static int notfirst;
    570 		struct servent *se;
    571 		if (!notfirst) {
    572 			setservent(1);
    573 			notfirst = 1;
    574 		}
    575 		se = getservbyport(htons(port), dg_proto);
    576 		if (se)
    577 			return se->s_name;
    578 	}
    579 
    580 	return NULL;
    581 }
    582 
    583 
    584 const char *resolve_service(int port)
    585 {
    586 	static char buf[128];
    587 	static struct scache cache[256];
    588 
    589 	if (port == 0) {
    590 		buf[0] = '*';
    591 		buf[1] = 0;
    592 		return buf;
    593 	}
    594 
    595 	if (resolve_services) {
    596 		if (dg_proto == RAW_PROTO) {
    597 			return inet_proto_n2a(port, buf, sizeof(buf));
    598 		} else {
    599 			struct scache *c;
    600 			const char *res;
    601 			int hash = (port^(((unsigned long)dg_proto)>>2))&255;
    602 
    603 			for (c = &cache[hash]; c; c = c->next) {
    604 				if (c->port == port &&
    605 				    c->proto == dg_proto) {
    606 					if (c->name)
    607 						return c->name;
    608 					goto do_numeric;
    609 				}
    610 			}
    611 
    612 			if ((res = __resolve_service(port)) != NULL) {
    613 				if ((c = malloc(sizeof(*c))) == NULL)
    614 					goto do_numeric;
    615 			} else {
    616 				c = &cache[hash];
    617 				if (c->name)
    618 					free(c->name);
    619 			}
    620 			c->port = port;
    621 			c->name = NULL;
    622 			c->proto = dg_proto;
    623 			if (res) {
    624 				c->name = strdup(res);
    625 				c->next = cache[hash].next;
    626 				cache[hash].next = c;
    627 			}
    628 			if (c->name)
    629 				return c->name;
    630 		}
    631 	}
    632 
    633 	do_numeric:
    634 	sprintf(buf, "%u", port);
    635 	return buf;
    636 }
    637 
    638 void formatted_print(const inet_prefix *a, int port)
    639 {
    640 	char buf[1024];
    641 	const char *ap = buf;
    642 	int est_len;
    643 
    644 	est_len = addr_width;
    645 
    646 	if (a->family == AF_INET) {
    647 		if (a->data[0] == 0) {
    648 			buf[0] = '*';
    649 			buf[1] = 0;
    650 		} else {
    651 			ap = format_host(AF_INET, 4, a->data, buf, sizeof(buf));
    652 		}
    653 	} else {
    654 		ap = format_host(a->family, 16, a->data, buf, sizeof(buf));
    655 		est_len = strlen(ap);
    656 		if (est_len <= addr_width)
    657 			est_len = addr_width;
    658 		else
    659 			est_len = addr_width + ((est_len-addr_width+3)/4)*4;
    660 	}
    661 	printf("%*s:%-*s ", est_len, ap, serv_width, resolve_service(port));
    662 }
    663 
    664 struct aafilter
    665 {
    666 	inet_prefix	addr;
    667 	int		port;
    668 	struct aafilter *next;
    669 };
    670 
    671 int inet2_addr_match(const inet_prefix *a, const inet_prefix *p, int plen)
    672 {
    673 	if (!inet_addr_match(a, p, plen))
    674 		return 0;
    675 
    676 	/* Cursed "v4 mapped" addresses: v4 mapped socket matches
    677 	 * pure IPv4 rule, but v4-mapped rule selects only v4-mapped
    678 	 * sockets. Fair? */
    679 	if (p->family == AF_INET && a->family == AF_INET6) {
    680 		if (a->data[0] == 0 && a->data[1] == 0 &&
    681 		    a->data[2] == htonl(0xffff)) {
    682 			inet_prefix tmp = *a;
    683 			tmp.data[0] = a->data[3];
    684 			return inet_addr_match(&tmp, p, plen);
    685 		}
    686 	}
    687 	return 1;
    688 }
    689 
    690 int unix_match(const inet_prefix *a, const inet_prefix *p)
    691 {
    692 	char *addr, *pattern;
    693 	memcpy(&addr, a->data, sizeof(addr));
    694 	memcpy(&pattern, p->data, sizeof(pattern));
    695 	if (pattern == NULL)
    696 		return 1;
    697 	if (addr == NULL)
    698 		addr = "";
    699 	return !fnmatch(pattern, addr, 0);
    700 }
    701 
    702 int run_ssfilter(struct ssfilter *f, struct tcpstat *s)
    703 {
    704 	switch (f->type) {
    705 		case SSF_S_AUTO:
    706 	{
    707                 static int low, high=65535;
    708 
    709 		if (s->local.family == AF_UNIX) {
    710 			char *p;
    711 			memcpy(&p, s->local.data, sizeof(p));
    712 			return p == NULL || (p[0] == '@' && strlen(p) == 6 &&
    713 					     strspn(p+1, "0123456789abcdef") == 5);
    714 		}
    715 		if (s->local.family == AF_PACKET)
    716 			return s->lport == 0 && s->local.data == 0;
    717 		if (s->local.family == AF_NETLINK)
    718 			return s->lport < 0;
    719 
    720                 if (!low) {
    721 			FILE *fp = ephemeral_ports_open();
    722 			if (fp) {
    723 				fscanf(fp, "%d%d", &low, &high);
    724 				fclose(fp);
    725 			}
    726 		}
    727 		return s->lport >= low && s->lport <= high;
    728 	}
    729 		case SSF_DCOND:
    730 	{
    731 		struct aafilter *a = (void*)f->pred;
    732 		if (a->addr.family == AF_UNIX)
    733 			return unix_match(&s->remote, &a->addr);
    734 		if (a->port != -1 && a->port != s->rport)
    735 			return 0;
    736 		if (a->addr.bitlen) {
    737 			do {
    738 				if (!inet2_addr_match(&s->remote, &a->addr, a->addr.bitlen))
    739 					return 1;
    740 			} while ((a = a->next) != NULL);
    741 			return 0;
    742 		}
    743 		return 1;
    744 	}
    745 		case SSF_SCOND:
    746 	{
    747 		struct aafilter *a = (void*)f->pred;
    748 		if (a->addr.family == AF_UNIX)
    749 			return unix_match(&s->local, &a->addr);
    750 		if (a->port != -1 && a->port != s->lport)
    751 			return 0;
    752 		if (a->addr.bitlen) {
    753 			do {
    754 				if (!inet2_addr_match(&s->local, &a->addr, a->addr.bitlen))
    755 					return 1;
    756 			} while ((a = a->next) != NULL);
    757 			return 0;
    758 		}
    759 		return 1;
    760 	}
    761 		case SSF_D_GE:
    762 	{
    763 		struct aafilter *a = (void*)f->pred;
    764 		return s->rport >= a->port;
    765 	}
    766 		case SSF_D_LE:
    767 	{
    768 		struct aafilter *a = (void*)f->pred;
    769 		return s->rport <= a->port;
    770 	}
    771 		case SSF_S_GE:
    772 	{
    773 		struct aafilter *a = (void*)f->pred;
    774 		return s->lport >= a->port;
    775 	}
    776 		case SSF_S_LE:
    777 	{
    778 		struct aafilter *a = (void*)f->pred;
    779 		return s->lport <= a->port;
    780 	}
    781 
    782 		/* Yup. It is recursion. Sorry. */
    783 		case SSF_AND:
    784 		return run_ssfilter(f->pred, s) && run_ssfilter(f->post, s);
    785 		case SSF_OR:
    786 		return run_ssfilter(f->pred, s) || run_ssfilter(f->post, s);
    787 		case SSF_NOT:
    788 		return !run_ssfilter(f->pred, s);
    789 		default:
    790 		abort();
    791 	}
    792 }
    793 
    794 /* Relocate external jumps by reloc. */
    795 static void ssfilter_patch(char *a, int len, int reloc)
    796 {
    797 	while (len > 0) {
    798 		struct inet_diag_bc_op *op = (struct inet_diag_bc_op*)a;
    799 		if (op->no == len+4)
    800 			op->no += reloc;
    801 		len -= op->yes;
    802 		a += op->yes;
    803 	}
    804 	if (len < 0)
    805 		abort();
    806 }
    807 
    808 static int ssfilter_bytecompile(struct ssfilter *f, char **bytecode)
    809 {
    810 	switch (f->type) {
    811 		case SSF_S_AUTO:
    812 	{
    813 		if (!(*bytecode=malloc(4))) abort();
    814 		((struct inet_diag_bc_op*)*bytecode)[0] = (struct inet_diag_bc_op){ INET_DIAG_BC_AUTO, 4, 8 };
    815 		return 4;
    816 	}
    817 		case SSF_DCOND:
    818 		case SSF_SCOND:
    819 	{
    820 		struct aafilter *a = (void*)f->pred;
    821 		struct aafilter *b;
    822 		char *ptr;
    823 		int  code = (f->type == SSF_DCOND ? INET_DIAG_BC_D_COND : INET_DIAG_BC_S_COND);
    824 		int len = 0;
    825 
    826 		for (b=a; b; b=b->next) {
    827 			len += 4 + sizeof(struct inet_diag_hostcond);
    828 			if (a->addr.family == AF_INET6)
    829 				len += 16;
    830 			else
    831 				len += 4;
    832 			if (b->next)
    833 				len += 4;
    834 		}
    835 		if (!(ptr = malloc(len))) abort();
    836 		*bytecode = ptr;
    837 		for (b=a; b; b=b->next) {
    838 			struct inet_diag_bc_op *op = (struct inet_diag_bc_op *)ptr;
    839 			int alen = (a->addr.family == AF_INET6 ? 16 : 4);
    840 			int oplen = alen + 4 + sizeof(struct inet_diag_hostcond);
    841 			struct inet_diag_hostcond *cond = (struct inet_diag_hostcond*)(ptr+4);
    842 
    843 			*op = (struct inet_diag_bc_op){ code, oplen, oplen+4 };
    844 			cond->family = a->addr.family;
    845 			cond->port = a->port;
    846 			cond->prefix_len = a->addr.bitlen;
    847 			memcpy(cond->addr, a->addr.data, alen);
    848 			ptr += oplen;
    849 			if (b->next) {
    850 				op = (struct inet_diag_bc_op *)ptr;
    851 				*op = (struct inet_diag_bc_op){ INET_DIAG_BC_JMP, 4, len - (ptr-*bytecode)};
    852 				ptr += 4;
    853 			}
    854 		}
    855 		return ptr - *bytecode;
    856 	}
    857 		case SSF_D_GE:
    858 	{
    859 		struct aafilter *x = (void*)f->pred;
    860 		if (!(*bytecode=malloc(8))) abort();
    861 		((struct inet_diag_bc_op*)*bytecode)[0] = (struct inet_diag_bc_op){ INET_DIAG_BC_D_GE, 8, 12 };
    862 		((struct inet_diag_bc_op*)*bytecode)[1] = (struct inet_diag_bc_op){ 0, 0, x->port };
    863 		return 8;
    864 	}
    865 		case SSF_D_LE:
    866 	{
    867 		struct aafilter *x = (void*)f->pred;
    868 		if (!(*bytecode=malloc(8))) abort();
    869 		((struct inet_diag_bc_op*)*bytecode)[0] = (struct inet_diag_bc_op){ INET_DIAG_BC_D_LE, 8, 12 };
    870 		((struct inet_diag_bc_op*)*bytecode)[1] = (struct inet_diag_bc_op){ 0, 0, x->port };
    871 		return 8;
    872 	}
    873 		case SSF_S_GE:
    874 	{
    875 		struct aafilter *x = (void*)f->pred;
    876 		if (!(*bytecode=malloc(8))) abort();
    877 		((struct inet_diag_bc_op*)*bytecode)[0] = (struct inet_diag_bc_op){ INET_DIAG_BC_S_GE, 8, 12 };
    878 		((struct inet_diag_bc_op*)*bytecode)[1] = (struct inet_diag_bc_op){ 0, 0, x->port };
    879 		return 8;
    880 	}
    881 		case SSF_S_LE:
    882 	{
    883 		struct aafilter *x = (void*)f->pred;
    884 		if (!(*bytecode=malloc(8))) abort();
    885 		((struct inet_diag_bc_op*)*bytecode)[0] = (struct inet_diag_bc_op){ INET_DIAG_BC_S_LE, 8, 12 };
    886 		((struct inet_diag_bc_op*)*bytecode)[1] = (struct inet_diag_bc_op){ 0, 0, x->port };
    887 		return 8;
    888 	}
    889 
    890 		case SSF_AND:
    891 	{
    892 		char *a1, *a2, *a, l1, l2;
    893 		l1 = ssfilter_bytecompile(f->pred, &a1);
    894 		l2 = ssfilter_bytecompile(f->post, &a2);
    895 		if (!(a = malloc(l1+l2))) abort();
    896 		memcpy(a, a1, l1);
    897 		memcpy(a+l1, a2, l2);
    898 		free(a1); free(a2);
    899 		ssfilter_patch(a, l1, l2);
    900 		*bytecode = a;
    901 		return l1+l2;
    902 	}
    903 		case SSF_OR:
    904 	{
    905 		char *a1, *a2, *a, l1, l2;
    906 		l1 = ssfilter_bytecompile(f->pred, &a1);
    907 		l2 = ssfilter_bytecompile(f->post, &a2);
    908 		if (!(a = malloc(l1+l2+4))) abort();
    909 		memcpy(a, a1, l1);
    910 		memcpy(a+l1+4, a2, l2);
    911 		free(a1); free(a2);
    912 		*(struct inet_diag_bc_op*)(a+l1) = (struct inet_diag_bc_op){ INET_DIAG_BC_JMP, 4, l2+4 };
    913 		*bytecode = a;
    914 		return l1+l2+4;
    915 	}
    916 		case SSF_NOT:
    917 	{
    918 		char *a1, *a, l1;
    919 		l1 = ssfilter_bytecompile(f->pred, &a1);
    920 		if (!(a = malloc(l1+4))) abort();
    921 		memcpy(a, a1, l1);
    922 		free(a1);
    923 		*(struct inet_diag_bc_op*)(a+l1) = (struct inet_diag_bc_op){ INET_DIAG_BC_JMP, 4, 8 };
    924 		*bytecode = a;
    925 		return l1+4;
    926 	}
    927 		default:
    928 		abort();
    929 	}
    930 }
    931 
    932 static int remember_he(struct aafilter *a, struct hostent *he)
    933 {
    934 	char **ptr = he->h_addr_list;
    935 	int cnt = 0;
    936 	int len;
    937 
    938 	if (he->h_addrtype == AF_INET)
    939 		len = 4;
    940 	else if (he->h_addrtype == AF_INET6)
    941 		len = 16;
    942 	else
    943 		return 0;
    944 
    945 	while (*ptr) {
    946 		struct aafilter *b = a;
    947 		if (a->addr.bitlen) {
    948 			if ((b = malloc(sizeof(*b))) == NULL)
    949 				return cnt;
    950 			*b = *a;
    951 			b->next = a->next;
    952 			a->next = b;
    953 		}
    954 		memcpy(b->addr.data, *ptr, len);
    955 		b->addr.bytelen = len;
    956 		b->addr.bitlen = len*8;
    957 		b->addr.family = he->h_addrtype;
    958 		ptr++;
    959 		cnt++;
    960 	}
    961 	return cnt;
    962 }
    963 
    964 static int get_dns_host(struct aafilter *a, const char *addr, int fam)
    965 {
    966 	static int notfirst;
    967 	int cnt = 0;
    968 	struct hostent *he;
    969 
    970 	a->addr.bitlen = 0;
    971 	if (!notfirst) {
    972 		sethostent(1);
    973 		notfirst = 1;
    974 	}
    975 	he = gethostbyname2(addr, fam == AF_UNSPEC ? AF_INET : fam);
    976 	if (he)
    977 		cnt = remember_he(a, he);
    978 	if (fam == AF_UNSPEC) {
    979 		he = gethostbyname2(addr, AF_INET6);
    980 		if (he)
    981 			cnt += remember_he(a, he);
    982 	}
    983 	return !cnt;
    984 }
    985 
    986 static int xll_initted = 0;
    987 
    988 static void xll_init(void)
    989 {
    990 	struct rtnl_handle rth;
    991 	rtnl_open(&rth, 0);
    992 	ll_init_map(&rth);
    993 	rtnl_close(&rth);
    994 	xll_initted = 1;
    995 }
    996 
    997 static const char *xll_index_to_name(int index)
    998 {
    999 	if (!xll_initted)
   1000 		xll_init();
   1001 	return ll_index_to_name(index);
   1002 }
   1003 
   1004 static int xll_name_to_index(const char *dev)
   1005 {
   1006 	if (!xll_initted)
   1007 		xll_init();
   1008 	return ll_name_to_index(dev);
   1009 }
   1010 
   1011 void *parse_hostcond(char *addr)
   1012 {
   1013 	char *port = NULL;
   1014 	struct aafilter a;
   1015 	struct aafilter *res;
   1016 	int fam = preferred_family;
   1017 
   1018 	memset(&a, 0, sizeof(a));
   1019 	a.port = -1;
   1020 
   1021 	if (fam == AF_UNIX || strncmp(addr, "unix:", 5) == 0) {
   1022 		char *p;
   1023 		a.addr.family = AF_UNIX;
   1024 		if (strncmp(addr, "unix:", 5) == 0)
   1025 			addr+=5;
   1026 		p = strdup(addr);
   1027 		a.addr.bitlen = 8*strlen(p);
   1028 		memcpy(a.addr.data, &p, sizeof(p));
   1029 		goto out;
   1030 	}
   1031 
   1032 	if (fam == AF_PACKET || strncmp(addr, "link:", 5) == 0) {
   1033 		a.addr.family = AF_PACKET;
   1034 		a.addr.bitlen = 0;
   1035 		if (strncmp(addr, "link:", 5) == 0)
   1036 			addr+=5;
   1037 		port = strchr(addr, ':');
   1038 		if (port) {
   1039 			*port = 0;
   1040 			if (port[1] && strcmp(port+1, "*")) {
   1041 				if (get_integer(&a.port, port+1, 0)) {
   1042 					if ((a.port = xll_name_to_index(port+1)) <= 0)
   1043 						return NULL;
   1044 				}
   1045 			}
   1046 		}
   1047 		if (addr[0] && strcmp(addr, "*")) {
   1048 			unsigned short tmp;
   1049 			a.addr.bitlen = 32;
   1050 			if (ll_proto_a2n(&tmp, addr))
   1051 				return NULL;
   1052 			a.addr.data[0] = ntohs(tmp);
   1053 		}
   1054 		goto out;
   1055 	}
   1056 
   1057 	if (fam == AF_NETLINK || strncmp(addr, "netlink:", 8) == 0) {
   1058 		a.addr.family = AF_NETLINK;
   1059 		a.addr.bitlen = 0;
   1060 		if (strncmp(addr, "netlink:", 8) == 0)
   1061 			addr+=8;
   1062 		port = strchr(addr, ':');
   1063 		if (port) {
   1064 			*port = 0;
   1065 			if (port[1] && strcmp(port+1, "*")) {
   1066 				if (get_integer(&a.port, port+1, 0)) {
   1067 					if (strcmp(port+1, "kernel") == 0)
   1068 						a.port = 0;
   1069 					else
   1070 						return NULL;
   1071 				}
   1072 			}
   1073 		}
   1074 		if (addr[0] && strcmp(addr, "*")) {
   1075 			a.addr.bitlen = 32;
   1076 			if (get_u32(a.addr.data, addr, 0)) {
   1077 				if (strcmp(addr, "rtnl") == 0)
   1078 					a.addr.data[0] = 0;
   1079 				else if (strcmp(addr, "fw") == 0)
   1080 					a.addr.data[0] = 3;
   1081 				else if (strcmp(addr, "tcpdiag") == 0)
   1082 					a.addr.data[0] = 4;
   1083 				else
   1084 					return NULL;
   1085 			}
   1086 		}
   1087 		goto out;
   1088 	}
   1089 
   1090 	if (strncmp(addr, "inet:", 5) == 0) {
   1091 		addr += 5;
   1092 		fam = AF_INET;
   1093 	} else if (strncmp(addr, "inet6:", 6) == 0) {
   1094 		addr += 6;
   1095 		fam = AF_INET6;
   1096 	}
   1097 
   1098 	/* URL-like literal [] */
   1099 	if (addr[0] == '[') {
   1100 		addr++;
   1101 		if ((port = strchr(addr, ']')) == NULL)
   1102 			return NULL;
   1103 		*port++ = 0;
   1104 	} else if (addr[0] == '*') {
   1105 		port = addr+1;
   1106 	} else {
   1107 		port = strrchr(strchr(addr, '/') ? : addr, ':');
   1108 	}
   1109 	if (port && *port) {
   1110 		if (*port != ':')
   1111 			return NULL;
   1112 		*port++ = 0;
   1113 		if (*port && *port != '*') {
   1114 			if (get_integer(&a.port, port, 0)) {
   1115 				struct servent *se1 = NULL;
   1116 				struct servent *se2 = NULL;
   1117 				if (current_filter.dbs&(1<<UDP_DB))
   1118 					se1 = getservbyname(port, UDP_PROTO);
   1119 				if (current_filter.dbs&(1<<TCP_DB))
   1120 					se2 = getservbyname(port, TCP_PROTO);
   1121 				if (se1 && se2 && se1->s_port != se2->s_port) {
   1122 					fprintf(stderr, "Error: ambiguous port \"%s\".\n", port);
   1123 					return NULL;
   1124 				}
   1125 				if (!se1)
   1126 					se1 = se2;
   1127 				if (se1) {
   1128 					a.port = ntohs(se1->s_port);
   1129 				} else {
   1130 					struct scache *s;
   1131 					for (s = rlist; s; s = s->next) {
   1132 						if ((s->proto == UDP_PROTO &&
   1133 						     (current_filter.dbs&(1<<UDP_DB))) ||
   1134 						    (s->proto == TCP_PROTO &&
   1135 						     (current_filter.dbs&(1<<TCP_DB)))) {
   1136 							if (s->name && strcmp(s->name, port) == 0) {
   1137 								if (a.port > 0 && a.port != s->port) {
   1138 									fprintf(stderr, "Error: ambiguous port \"%s\".\n", port);
   1139 									return NULL;
   1140 								}
   1141 								a.port = s->port;
   1142 							}
   1143 						}
   1144 					}
   1145 					if (a.port <= 0) {
   1146 						fprintf(stderr, "Error: \"%s\" does not look like a port.\n", port);
   1147 						return NULL;
   1148 					}
   1149 				}
   1150 			}
   1151 		}
   1152 	}
   1153 	if (addr && *addr && *addr != '*') {
   1154 		if (get_prefix_1(&a.addr, addr, fam)) {
   1155 			if (get_dns_host(&a, addr, fam)) {
   1156 				fprintf(stderr, "Error: an inet prefix is expected rather than \"%s\".\n", addr);
   1157 				return NULL;
   1158 			}
   1159 		}
   1160 	}
   1161 
   1162 	out:
   1163 	res = malloc(sizeof(*res));
   1164 	if (res)
   1165 		memcpy(res, &a, sizeof(a));
   1166 	return res;
   1167 }
   1168 
   1169 static int tcp_show_line(char *line, const struct filter *f, int family)
   1170 {
   1171 	struct tcpstat s;
   1172 	char *loc, *rem, *data;
   1173 	char opt[256];
   1174 	int n;
   1175 	char *p;
   1176 
   1177 	if ((p = strchr(line, ':')) == NULL)
   1178 		return -1;
   1179 	loc = p+2;
   1180 
   1181 	if ((p = strchr(loc, ':')) == NULL)
   1182 		return -1;
   1183 	p[5] = 0;
   1184 	rem = p+6;
   1185 
   1186 	if ((p = strchr(rem, ':')) == NULL)
   1187 		return -1;
   1188 	p[5] = 0;
   1189 	data = p+6;
   1190 
   1191 	do {
   1192 		int state = (data[1] >= 'A') ? (data[1] - 'A' + 10) : (data[1] - '0');
   1193 
   1194 		if (!(f->states & (1<<state)))
   1195 			return 0;
   1196 	} while (0);
   1197 
   1198 	s.local.family = s.remote.family = family;
   1199 	if (family == AF_INET) {
   1200 		sscanf(loc, "%x:%x", s.local.data, (unsigned*)&s.lport);
   1201 		sscanf(rem, "%x:%x", s.remote.data, (unsigned*)&s.rport);
   1202 		s.local.bytelen = s.remote.bytelen = 4;
   1203 	} else {
   1204 		sscanf(loc, "%08x%08x%08x%08x:%x",
   1205 		       s.local.data,
   1206 		       s.local.data+1,
   1207 		       s.local.data+2,
   1208 		       s.local.data+3,
   1209 		       &s.lport);
   1210 		sscanf(rem, "%08x%08x%08x%08x:%x",
   1211 		       s.remote.data,
   1212 		       s.remote.data+1,
   1213 		       s.remote.data+2,
   1214 		       s.remote.data+3,
   1215 		       &s.rport);
   1216 		s.local.bytelen = s.remote.bytelen = 16;
   1217 	}
   1218 
   1219 	if (f->f && run_ssfilter(f->f, &s) == 0)
   1220 		return 0;
   1221 
   1222 	opt[0] = 0;
   1223 	n = sscanf(data, "%x %x:%x %x:%x %x %d %d %u %d %llx %d %d %d %d %d %[^\n]\n",
   1224 		   &s.state, &s.wq, &s.rq,
   1225 		   &s.timer, &s.timeout, &s.retrs, &s.uid, &s.probes, &s.ino,
   1226 		   &s.refcnt, &s.sk, &s.rto, &s.ato, &s.qack,
   1227 		   &s.cwnd, &s.ssthresh, opt);
   1228 
   1229 	if (n < 17)
   1230 		opt[0] = 0;
   1231 
   1232 	if (n < 12) {
   1233 		s.rto = 0;
   1234 		s.cwnd = 2;
   1235 		s.ssthresh = -1;
   1236 		s.ato = s.qack = 0;
   1237 	}
   1238 
   1239 	if (netid_width)
   1240 		printf("%-*s ", netid_width, "tcp");
   1241 	if (state_width)
   1242 		printf("%-*s ", state_width, sstate_name[s.state]);
   1243 
   1244 	printf("%-6d %-6d ", s.rq, s.wq);
   1245 
   1246 	formatted_print(&s.local, s.lport);
   1247 	formatted_print(&s.remote, s.rport);
   1248 
   1249 	if (show_options) {
   1250 		if (s.timer) {
   1251 			if (s.timer > 4)
   1252 				s.timer = 5;
   1253 			printf(" timer:(%s,%s,%d)",
   1254 			       tmr_name[s.timer],
   1255 			       print_hz_timer(s.timeout),
   1256 			       s.timer != 1 ? s.probes : s.retrs);
   1257 		}
   1258 	}
   1259 	if (show_tcpinfo) {
   1260 		int hz = get_user_hz();
   1261 		if (s.rto && s.rto != 3*hz)
   1262 			printf(" rto:%g", (double)s.rto/hz);
   1263 		if (s.ato)
   1264 			printf(" ato:%g", (double)s.ato/hz);
   1265 		if (s.cwnd != 2)
   1266 			printf(" cwnd:%d", s.cwnd);
   1267 		if (s.ssthresh != -1)
   1268 			printf(" ssthresh:%d", s.ssthresh);
   1269 		if (s.qack/2)
   1270 			printf(" qack:%d", s.qack/2);
   1271 		if (s.qack&1)
   1272 			printf(" bidir");
   1273 	}
   1274 	if (show_users) {
   1275 		char ubuf[4096];
   1276 		if (find_users(s.ino, ubuf, sizeof(ubuf)) > 0)
   1277 			printf(" users:(%s)", ubuf);
   1278 	}
   1279 	if (show_details) {
   1280 		if (s.uid)
   1281 			printf(" uid:%u", (unsigned)s.uid);
   1282 		printf(" ino:%u", s.ino);
   1283 		printf(" sk:%llx", s.sk);
   1284 		if (opt[0])
   1285 			printf(" opt:\"%s\"", opt);
   1286 	}
   1287 	printf("\n");
   1288 
   1289 	return 0;
   1290 }
   1291 
   1292 static int generic_record_read(FILE *fp,
   1293 			       int (*worker)(char*, const struct filter *, int),
   1294 			       const struct filter *f, int fam)
   1295 {
   1296 	char line[256];
   1297 
   1298 	/* skip header */
   1299 	if (fgets(line, sizeof(line), fp) == NULL)
   1300 		goto outerr;
   1301 
   1302 	while (fgets(line, sizeof(line), fp) != NULL) {
   1303 		int n = strlen(line);
   1304 		if (n == 0 || line[n-1] != '\n') {
   1305 			errno = -EINVAL;
   1306 			return -1;
   1307 		}
   1308 		line[n-1] = 0;
   1309 
   1310 		if (worker(line, f, fam) < 0)
   1311 			return 0;
   1312 	}
   1313 outerr:
   1314 
   1315 	return ferror(fp) ? -1 : 0;
   1316 }
   1317 
   1318 static char *sprint_bw(char *buf, double bw)
   1319 {
   1320 	if (bw > 1000000.)
   1321 		sprintf(buf,"%.1fM", bw / 1000000.);
   1322 	else if (bw > 1000.)
   1323 		sprintf(buf,"%.1fK", bw / 1000.);
   1324 	else
   1325 		sprintf(buf, "%g", bw);
   1326 
   1327 	return buf;
   1328 }
   1329 
   1330 static void tcp_show_info(const struct nlmsghdr *nlh, struct inet_diag_msg *r)
   1331 {
   1332 	struct rtattr * tb[INET_DIAG_MAX+1];
   1333 	char b1[64];
   1334 	double rtt = 0;
   1335 
   1336 	parse_rtattr(tb, INET_DIAG_MAX, (struct rtattr*)(r+1),
   1337 		     nlh->nlmsg_len - NLMSG_LENGTH(sizeof(*r)));
   1338 
   1339 	if (tb[INET_DIAG_SKMEMINFO]) {
   1340 		const __u32 *skmeminfo =  RTA_DATA(tb[INET_DIAG_SKMEMINFO]);
   1341 		printf(" skmem:(r%u,rb%u,t%u,tb%u,f%u,w%u,o%u)",
   1342 			skmeminfo[SK_MEMINFO_RMEM_ALLOC],
   1343 			skmeminfo[SK_MEMINFO_RCVBUF],
   1344 			skmeminfo[SK_MEMINFO_WMEM_ALLOC],
   1345 			skmeminfo[SK_MEMINFO_SNDBUF],
   1346 			skmeminfo[SK_MEMINFO_FWD_ALLOC],
   1347 			skmeminfo[SK_MEMINFO_WMEM_QUEUED],
   1348 			skmeminfo[SK_MEMINFO_OPTMEM]);
   1349 	}else if (tb[INET_DIAG_MEMINFO]) {
   1350 		const struct inet_diag_meminfo *minfo
   1351 			= RTA_DATA(tb[INET_DIAG_MEMINFO]);
   1352 		printf(" mem:(r%u,w%u,f%u,t%u)",
   1353 		       minfo->idiag_rmem,
   1354 		       minfo->idiag_wmem,
   1355 		       minfo->idiag_fmem,
   1356 		       minfo->idiag_tmem);
   1357 	}
   1358 
   1359 	if (tb[INET_DIAG_INFO]) {
   1360 		struct tcp_info *info;
   1361 		int len = RTA_PAYLOAD(tb[INET_DIAG_INFO]);
   1362 
   1363 		/* workaround for older kernels with less fields */
   1364 		if (len < sizeof(*info)) {
   1365 			info = alloca(sizeof(*info));
   1366 			memset(info, 0, sizeof(*info));
   1367 			memcpy(info, RTA_DATA(tb[INET_DIAG_INFO]), len);
   1368 		} else
   1369 			info = RTA_DATA(tb[INET_DIAG_INFO]);
   1370 
   1371 		if (show_options) {
   1372 			if (info->tcpi_options & TCPI_OPT_TIMESTAMPS)
   1373 				printf(" ts");
   1374 			if (info->tcpi_options & TCPI_OPT_SACK)
   1375 				printf(" sack");
   1376 			if (info->tcpi_options & TCPI_OPT_ECN)
   1377 				printf(" ecn");
   1378 			if (info->tcpi_options & TCPI_OPT_ECN_SEEN)
   1379 				printf(" ecnseen");
   1380 		}
   1381 
   1382 		if (tb[INET_DIAG_CONG])
   1383 			printf(" %s", rta_getattr_str(tb[INET_DIAG_CONG]));
   1384 
   1385 		if (info->tcpi_options & TCPI_OPT_WSCALE)
   1386 			printf(" wscale:%d,%d", info->tcpi_snd_wscale,
   1387 			       info->tcpi_rcv_wscale);
   1388 		if (info->tcpi_rto && info->tcpi_rto != 3000000)
   1389 			printf(" rto:%g", (double)info->tcpi_rto/1000);
   1390 		if (info->tcpi_rtt)
   1391 			printf(" rtt:%g/%g", (double)info->tcpi_rtt/1000,
   1392 			       (double)info->tcpi_rttvar/1000);
   1393 		if (info->tcpi_ato)
   1394 			printf(" ato:%g", (double)info->tcpi_ato/1000);
   1395 		if (info->tcpi_snd_cwnd != 2)
   1396 			printf(" cwnd:%d", info->tcpi_snd_cwnd);
   1397 		if (info->tcpi_snd_ssthresh < 0xFFFF)
   1398 			printf(" ssthresh:%d", info->tcpi_snd_ssthresh);
   1399 
   1400 		rtt = (double) info->tcpi_rtt;
   1401 		if (tb[INET_DIAG_VEGASINFO]) {
   1402 			const struct tcpvegas_info *vinfo
   1403 				= RTA_DATA(tb[INET_DIAG_VEGASINFO]);
   1404 
   1405 			if (vinfo->tcpv_enabled &&
   1406 			    vinfo->tcpv_rtt && vinfo->tcpv_rtt != 0x7fffffff)
   1407 				rtt =  vinfo->tcpv_rtt;
   1408 		}
   1409 
   1410 		if (rtt > 0 && info->tcpi_snd_mss && info->tcpi_snd_cwnd) {
   1411 			printf(" send %sbps",
   1412 			       sprint_bw(b1, (double) info->tcpi_snd_cwnd *
   1413 					 (double) info->tcpi_snd_mss * 8000000.
   1414 					 / rtt));
   1415 		}
   1416 
   1417 		if (info->tcpi_rcv_rtt)
   1418 			printf(" rcv_rtt:%g", (double) info->tcpi_rcv_rtt/1000);
   1419 		if (info->tcpi_rcv_space)
   1420 			printf(" rcv_space:%d", info->tcpi_rcv_space);
   1421 
   1422 	}
   1423 }
   1424 
   1425 static int tcp_show_sock(struct nlmsghdr *nlh, struct filter *f)
   1426 {
   1427 	struct inet_diag_msg *r = NLMSG_DATA(nlh);
   1428 	struct tcpstat s;
   1429 
   1430 	s.state = r->idiag_state;
   1431 	s.local.family = s.remote.family = r->idiag_family;
   1432 	s.lport = ntohs(r->id.idiag_sport);
   1433 	s.rport = ntohs(r->id.idiag_dport);
   1434 	if (s.local.family == AF_INET) {
   1435 		s.local.bytelen = s.remote.bytelen = 4;
   1436 	} else {
   1437 		s.local.bytelen = s.remote.bytelen = 16;
   1438 	}
   1439 	memcpy(s.local.data, r->id.idiag_src, s.local.bytelen);
   1440 	memcpy(s.remote.data, r->id.idiag_dst, s.local.bytelen);
   1441 
   1442 	if (f && f->f && run_ssfilter(f->f, &s) == 0)
   1443 		return 0;
   1444 
   1445 	if (netid_width)
   1446 		printf("%-*s ", netid_width, "tcp");
   1447 	if (state_width)
   1448 		printf("%-*s ", state_width, sstate_name[s.state]);
   1449 
   1450 	printf("%-6d %-6d ", r->idiag_rqueue, r->idiag_wqueue);
   1451 
   1452 	formatted_print(&s.local, s.lport);
   1453 	formatted_print(&s.remote, s.rport);
   1454 
   1455 	if (show_options) {
   1456 		if (r->idiag_timer) {
   1457 			if (r->idiag_timer > 4)
   1458 				r->idiag_timer = 5;
   1459 			printf(" timer:(%s,%s,%d)",
   1460 			       tmr_name[r->idiag_timer],
   1461 			       print_ms_timer(r->idiag_expires),
   1462 			       r->idiag_retrans);
   1463 		}
   1464 	}
   1465 	if (show_users) {
   1466 		char ubuf[4096];
   1467 		if (find_users(r->idiag_inode, ubuf, sizeof(ubuf)) > 0)
   1468 			printf(" users:(%s)", ubuf);
   1469 	}
   1470 	if (show_details) {
   1471 		if (r->idiag_uid)
   1472 			printf(" uid:%u", (unsigned)r->idiag_uid);
   1473 		printf(" ino:%u", r->idiag_inode);
   1474 		printf(" sk:");
   1475 		if (r->id.idiag_cookie[1] != 0)
   1476 			printf("%08x", r->id.idiag_cookie[1]);
   1477  		printf("%08x", r->id.idiag_cookie[0]);
   1478 	}
   1479 	if (show_mem || show_tcpinfo) {
   1480 		printf("\n\t");
   1481 		tcp_show_info(nlh, r);
   1482 	}
   1483 
   1484 	printf("\n");
   1485 
   1486 	return 0;
   1487 }
   1488 
   1489 static int tcp_show_netlink(struct filter *f, FILE *dump_fp, int socktype)
   1490 {
   1491 	int fd;
   1492 	struct sockaddr_nl nladdr;
   1493 	struct {
   1494 		struct nlmsghdr nlh;
   1495 		struct inet_diag_req r;
   1496 	} req;
   1497 	char    *bc = NULL;
   1498 	int	bclen;
   1499 	struct msghdr msg;
   1500 	struct rtattr rta;
   1501 	char	buf[8192];
   1502 	struct iovec iov[3];
   1503 
   1504 	if ((fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_INET_DIAG)) < 0)
   1505 		return -1;
   1506 
   1507 	memset(&nladdr, 0, sizeof(nladdr));
   1508 	nladdr.nl_family = AF_NETLINK;
   1509 
   1510 	req.nlh.nlmsg_len = sizeof(req);
   1511 	req.nlh.nlmsg_type = socktype;
   1512 	req.nlh.nlmsg_flags = NLM_F_ROOT|NLM_F_MATCH|NLM_F_REQUEST;
   1513 	req.nlh.nlmsg_pid = 0;
   1514 	req.nlh.nlmsg_seq = 123456;
   1515 	memset(&req.r, 0, sizeof(req.r));
   1516 	req.r.idiag_family = AF_INET;
   1517 	req.r.idiag_states = f->states;
   1518 	if (show_mem) {
   1519 		req.r.idiag_ext |= (1<<(INET_DIAG_MEMINFO-1));
   1520 		req.r.idiag_ext |= (1<<(INET_DIAG_SKMEMINFO-1));
   1521 	}
   1522 
   1523 	if (show_tcpinfo) {
   1524 		req.r.idiag_ext |= (1<<(INET_DIAG_INFO-1));
   1525 		req.r.idiag_ext |= (1<<(INET_DIAG_VEGASINFO-1));
   1526 		req.r.idiag_ext |= (1<<(INET_DIAG_CONG-1));
   1527 	}
   1528 
   1529 	iov[0] = (struct iovec){
   1530 		.iov_base = &req,
   1531 		.iov_len = sizeof(req)
   1532 	};
   1533 	if (f->f) {
   1534 		bclen = ssfilter_bytecompile(f->f, &bc);
   1535 		rta.rta_type = INET_DIAG_REQ_BYTECODE;
   1536 		rta.rta_len = RTA_LENGTH(bclen);
   1537 		iov[1] = (struct iovec){ &rta, sizeof(rta) };
   1538 		iov[2] = (struct iovec){ bc, bclen };
   1539 		req.nlh.nlmsg_len += RTA_LENGTH(bclen);
   1540 	}
   1541 
   1542 	msg = (struct msghdr) {
   1543 		.msg_name = (void*)&nladdr,
   1544 		.msg_namelen = sizeof(nladdr),
   1545 		.msg_iov = iov,
   1546 		.msg_iovlen = f->f ? 3 : 1,
   1547 	};
   1548 
   1549 	if (sendmsg(fd, &msg, 0) < 0) {
   1550 		close(fd);
   1551 		return -1;
   1552 	}
   1553 
   1554 	iov[0] = (struct iovec){
   1555 		.iov_base = buf,
   1556 		.iov_len = sizeof(buf)
   1557 	};
   1558 
   1559 	while (1) {
   1560 		int status;
   1561 		struct nlmsghdr *h;
   1562 
   1563 		msg = (struct msghdr) {
   1564 			(void*)&nladdr, sizeof(nladdr),
   1565 			iov,	1,
   1566 			NULL,	0,
   1567 			0
   1568 		};
   1569 
   1570 		status = recvmsg(fd, &msg, 0);
   1571 
   1572 		if (status < 0) {
   1573 			if (errno == EINTR)
   1574 				continue;
   1575 			perror("OVERRUN");
   1576 			continue;
   1577 		}
   1578 		if (status == 0) {
   1579 			fprintf(stderr, "EOF on netlink\n");
   1580 			close(fd);
   1581 			return 0;
   1582 		}
   1583 
   1584 		if (dump_fp)
   1585 			fwrite(buf, 1, NLMSG_ALIGN(status), dump_fp);
   1586 
   1587 		h = (struct nlmsghdr*)buf;
   1588 		while (NLMSG_OK(h, status)) {
   1589 			int err;
   1590 			struct inet_diag_msg *r = NLMSG_DATA(h);
   1591 
   1592 			if (/*h->nlmsg_pid != rth->local.nl_pid ||*/
   1593 			    h->nlmsg_seq != 123456)
   1594 				goto skip_it;
   1595 
   1596 			if (h->nlmsg_type == NLMSG_DONE) {
   1597 				close(fd);
   1598 				return 0;
   1599 			}
   1600 			if (h->nlmsg_type == NLMSG_ERROR) {
   1601 				struct nlmsgerr *err = (struct nlmsgerr*)NLMSG_DATA(h);
   1602 				if (h->nlmsg_len < NLMSG_LENGTH(sizeof(struct nlmsgerr))) {
   1603 					fprintf(stderr, "ERROR truncated\n");
   1604 				} else {
   1605 					errno = -err->error;
   1606 					if (errno == EOPNOTSUPP) {
   1607 						close(fd);
   1608 						return -1;
   1609 					}
   1610 					perror("TCPDIAG answers");
   1611 				}
   1612 				close(fd);
   1613 				return 0;
   1614 			}
   1615 			if (!dump_fp) {
   1616 				if (!(f->families & (1<<r->idiag_family))) {
   1617 					h = NLMSG_NEXT(h, status);
   1618 					continue;
   1619 				}
   1620 				err = tcp_show_sock(h, NULL);
   1621 				if (err < 0) {
   1622 					close(fd);
   1623 					return err;
   1624 				}
   1625 			}
   1626 
   1627 skip_it:
   1628 			h = NLMSG_NEXT(h, status);
   1629 		}
   1630 		if (msg.msg_flags & MSG_TRUNC) {
   1631 			fprintf(stderr, "Message truncated\n");
   1632 			continue;
   1633 		}
   1634 		if (status) {
   1635 			fprintf(stderr, "!!!Remnant of size %d\n", status);
   1636 			exit(1);
   1637 		}
   1638 	}
   1639 	close(fd);
   1640 	return 0;
   1641 }
   1642 
   1643 static int tcp_show_netlink_file(struct filter *f)
   1644 {
   1645 	FILE	*fp;
   1646 	char	buf[8192];
   1647 
   1648 	if ((fp = fopen(getenv("TCPDIAG_FILE"), "r")) == NULL) {
   1649 		perror("fopen($TCPDIAG_FILE)");
   1650 		return -1;
   1651 	}
   1652 
   1653 	while (1) {
   1654 		int status, err;
   1655 		struct nlmsghdr *h = (struct nlmsghdr*)buf;
   1656 
   1657 		status = fread(buf, 1, sizeof(*h), fp);
   1658 		if (status < 0) {
   1659 			perror("Reading header from $TCPDIAG_FILE");
   1660 			return -1;
   1661 		}
   1662 		if (status != sizeof(*h)) {
   1663 			perror("Unexpected EOF reading $TCPDIAG_FILE");
   1664 			return -1;
   1665 		}
   1666 
   1667 		status = fread(h+1, 1, NLMSG_ALIGN(h->nlmsg_len-sizeof(*h)), fp);
   1668 
   1669 		if (status < 0) {
   1670 			perror("Reading $TCPDIAG_FILE");
   1671 			return -1;
   1672 		}
   1673 		if (status + sizeof(*h) < h->nlmsg_len) {
   1674 			perror("Unexpected EOF reading $TCPDIAG_FILE");
   1675 			return -1;
   1676 		}
   1677 
   1678 		/* The only legal exit point */
   1679 		if (h->nlmsg_type == NLMSG_DONE)
   1680 			return 0;
   1681 
   1682 		if (h->nlmsg_type == NLMSG_ERROR) {
   1683 			struct nlmsgerr *err = (struct nlmsgerr*)NLMSG_DATA(h);
   1684 			if (h->nlmsg_len < NLMSG_LENGTH(sizeof(struct nlmsgerr))) {
   1685 				fprintf(stderr, "ERROR truncated\n");
   1686 			} else {
   1687 				errno = -err->error;
   1688 				perror("TCPDIAG answered");
   1689 			}
   1690 			return -1;
   1691 		}
   1692 
   1693 		err = tcp_show_sock(h, f);
   1694 		if (err < 0)
   1695 			return err;
   1696 	}
   1697 }
   1698 
   1699 static int tcp_show(struct filter *f, int socktype)
   1700 {
   1701 	FILE *fp = NULL;
   1702 	char *buf = NULL;
   1703 	int bufsize = 64*1024;
   1704 
   1705 	dg_proto = TCP_PROTO;
   1706 
   1707 	if (getenv("TCPDIAG_FILE"))
   1708 		return tcp_show_netlink_file(f);
   1709 
   1710 	if (!getenv("PROC_NET_TCP") && !getenv("PROC_ROOT")
   1711 	    && tcp_show_netlink(f, NULL, socktype) == 0)
   1712 		return 0;
   1713 
   1714 	/* Sigh... We have to parse /proc/net/tcp... */
   1715 
   1716 
   1717 	/* Estimate amount of sockets and try to allocate
   1718 	 * huge buffer to read all the table at one read.
   1719 	 * Limit it by 16MB though. The assumption is: as soon as
   1720 	 * kernel was able to hold information about N connections,
   1721 	 * it is able to give us some memory for snapshot.
   1722 	 */
   1723 	if (1) {
   1724 		int guess = slabstat.socks+slabstat.tcp_syns;
   1725 		if (f->states&(1<<SS_TIME_WAIT))
   1726 			guess += slabstat.tcp_tws;
   1727 		if (guess > (16*1024*1024)/128)
   1728 			guess = (16*1024*1024)/128;
   1729 		guess *= 128;
   1730 		if (guess > bufsize)
   1731 			bufsize = guess;
   1732 	}
   1733 	while (bufsize >= 64*1024) {
   1734 		if ((buf = malloc(bufsize)) != NULL)
   1735 			break;
   1736 		bufsize /= 2;
   1737 	}
   1738 	if (buf == NULL) {
   1739 		errno = ENOMEM;
   1740 		return -1;
   1741 	}
   1742 
   1743 	if (f->families & (1<<AF_INET)) {
   1744 		if ((fp = net_tcp_open()) == NULL)
   1745 			goto outerr;
   1746 
   1747 		setbuffer(fp, buf, bufsize);
   1748 		if (generic_record_read(fp, tcp_show_line, f, AF_INET))
   1749 			goto outerr;
   1750 		fclose(fp);
   1751 	}
   1752 
   1753 	if ((f->families & (1<<AF_INET6)) &&
   1754 	    (fp = net_tcp6_open()) != NULL) {
   1755 		setbuffer(fp, buf, bufsize);
   1756 		if (generic_record_read(fp, tcp_show_line, f, AF_INET6))
   1757 			goto outerr;
   1758 		fclose(fp);
   1759 	}
   1760 
   1761 	free(buf);
   1762 	return 0;
   1763 
   1764 outerr:
   1765 	do {
   1766 		int saved_errno = errno;
   1767 		if (buf)
   1768 			free(buf);
   1769 		if (fp)
   1770 			fclose(fp);
   1771 		errno = saved_errno;
   1772 		return -1;
   1773 	} while (0);
   1774 }
   1775 
   1776 
   1777 int dgram_show_line(char *line, const struct filter *f, int family)
   1778 {
   1779 	struct tcpstat s;
   1780 	char *loc, *rem, *data;
   1781 	char opt[256];
   1782 	int n;
   1783 	char *p;
   1784 
   1785 	if ((p = strchr(line, ':')) == NULL)
   1786 		return -1;
   1787 	loc = p+2;
   1788 
   1789 	if ((p = strchr(loc, ':')) == NULL)
   1790 		return -1;
   1791 	p[5] = 0;
   1792 	rem = p+6;
   1793 
   1794 	if ((p = strchr(rem, ':')) == NULL)
   1795 		return -1;
   1796 	p[5] = 0;
   1797 	data = p+6;
   1798 
   1799 	do {
   1800 		int state = (data[1] >= 'A') ? (data[1] - 'A' + 10) : (data[1] - '0');
   1801 
   1802 		if (!(f->states & (1<<state)))
   1803 			return 0;
   1804 	} while (0);
   1805 
   1806 	s.local.family = s.remote.family = family;
   1807 	if (family == AF_INET) {
   1808 		sscanf(loc, "%x:%x", s.local.data, (unsigned*)&s.lport);
   1809 		sscanf(rem, "%x:%x", s.remote.data, (unsigned*)&s.rport);
   1810 		s.local.bytelen = s.remote.bytelen = 4;
   1811 	} else {
   1812 		sscanf(loc, "%08x%08x%08x%08x:%x",
   1813 		       s.local.data,
   1814 		       s.local.data+1,
   1815 		       s.local.data+2,
   1816 		       s.local.data+3,
   1817 		       &s.lport);
   1818 		sscanf(rem, "%08x%08x%08x%08x:%x",
   1819 		       s.remote.data,
   1820 		       s.remote.data+1,
   1821 		       s.remote.data+2,
   1822 		       s.remote.data+3,
   1823 		       &s.rport);
   1824 		s.local.bytelen = s.remote.bytelen = 16;
   1825 	}
   1826 
   1827 	if (f->f && run_ssfilter(f->f, &s) == 0)
   1828 		return 0;
   1829 
   1830 	opt[0] = 0;
   1831 	n = sscanf(data, "%x %x:%x %*x:%*x %*x %d %*d %u %d %llx %[^\n]\n",
   1832 	       &s.state, &s.wq, &s.rq,
   1833 	       &s.uid, &s.ino,
   1834 	       &s.refcnt, &s.sk, opt);
   1835 
   1836 	if (n < 9)
   1837 		opt[0] = 0;
   1838 
   1839 	if (netid_width)
   1840 		printf("%-*s ", netid_width, dg_proto);
   1841 	if (state_width)
   1842 		printf("%-*s ", state_width, sstate_name[s.state]);
   1843 
   1844 	printf("%-6d %-6d ", s.rq, s.wq);
   1845 
   1846 	formatted_print(&s.local, s.lport);
   1847 	formatted_print(&s.remote, s.rport);
   1848 
   1849 	if (show_users) {
   1850 		char ubuf[4096];
   1851 		if (find_users(s.ino, ubuf, sizeof(ubuf)) > 0)
   1852 			printf(" users:(%s)", ubuf);
   1853 	}
   1854 
   1855 	if (show_details) {
   1856 		if (s.uid)
   1857 			printf(" uid=%u", (unsigned)s.uid);
   1858 		printf(" ino=%u", s.ino);
   1859 		printf(" sk=%llx", s.sk);
   1860 		if (opt[0])
   1861 			printf(" opt:\"%s\"", opt);
   1862 	}
   1863 	printf("\n");
   1864 
   1865 	return 0;
   1866 }
   1867 
   1868 
   1869 int udp_show(struct filter *f)
   1870 {
   1871 	FILE *fp = NULL;
   1872 
   1873 	dg_proto = UDP_PROTO;
   1874 
   1875 	if (f->families&(1<<AF_INET)) {
   1876 		if ((fp = net_udp_open()) == NULL)
   1877 			goto outerr;
   1878 		if (generic_record_read(fp, dgram_show_line, f, AF_INET))
   1879 			goto outerr;
   1880 		fclose(fp);
   1881 	}
   1882 
   1883 	if ((f->families&(1<<AF_INET6)) &&
   1884 	    (fp = net_udp6_open()) != NULL) {
   1885 		if (generic_record_read(fp, dgram_show_line, f, AF_INET6))
   1886 			goto outerr;
   1887 		fclose(fp);
   1888 	}
   1889 	return 0;
   1890 
   1891 outerr:
   1892 	do {
   1893 		int saved_errno = errno;
   1894 		if (fp)
   1895 			fclose(fp);
   1896 		errno = saved_errno;
   1897 		return -1;
   1898 	} while (0);
   1899 }
   1900 
   1901 int raw_show(struct filter *f)
   1902 {
   1903 	FILE *fp = NULL;
   1904 
   1905 	dg_proto = RAW_PROTO;
   1906 
   1907 	if (f->families&(1<<AF_INET)) {
   1908 		if ((fp = net_raw_open()) == NULL)
   1909 			goto outerr;
   1910 		if (generic_record_read(fp, dgram_show_line, f, AF_INET))
   1911 			goto outerr;
   1912 		fclose(fp);
   1913 	}
   1914 
   1915 	if ((f->families&(1<<AF_INET6)) &&
   1916 	    (fp = net_raw6_open()) != NULL) {
   1917 		if (generic_record_read(fp, dgram_show_line, f, AF_INET6))
   1918 			goto outerr;
   1919 		fclose(fp);
   1920 	}
   1921 	return 0;
   1922 
   1923 outerr:
   1924 	do {
   1925 		int saved_errno = errno;
   1926 		if (fp)
   1927 			fclose(fp);
   1928 		errno = saved_errno;
   1929 		return -1;
   1930 	} while (0);
   1931 }
   1932 
   1933 
   1934 struct unixstat
   1935 {
   1936 	struct unixstat *next;
   1937 	int ino;
   1938 	int peer;
   1939 	int rq;
   1940 	int wq;
   1941 	int state;
   1942 	int type;
   1943 	char *name;
   1944 };
   1945 
   1946 
   1947 
   1948 int unix_state_map[] = { SS_CLOSE, SS_SYN_SENT,
   1949 			 SS_ESTABLISHED, SS_CLOSING };
   1950 
   1951 
   1952 #define MAX_UNIX_REMEMBER (1024*1024/sizeof(struct unixstat))
   1953 
   1954 void unix_list_free(struct unixstat *list)
   1955 {
   1956 	while (list) {
   1957 		struct unixstat *s = list;
   1958 		list = list->next;
   1959 		if (s->name)
   1960 			free(s->name);
   1961 		free(s);
   1962 	}
   1963 }
   1964 
   1965 void unix_list_print(struct unixstat *list, struct filter *f)
   1966 {
   1967 	struct unixstat *s;
   1968 	char *peer;
   1969 
   1970 	for (s = list; s; s = s->next) {
   1971 		if (!(f->states & (1<<s->state)))
   1972 			continue;
   1973 		if (s->type == SOCK_STREAM && !(f->dbs&(1<<UNIX_ST_DB)))
   1974 			continue;
   1975 		if (s->type == SOCK_DGRAM && !(f->dbs&(1<<UNIX_DG_DB)))
   1976 			continue;
   1977 
   1978 		peer = "*";
   1979 		if (s->peer) {
   1980 			struct unixstat *p;
   1981 			for (p = list; p; p = p->next) {
   1982 				if (s->peer == p->ino)
   1983 					break;
   1984 			}
   1985 			if (!p) {
   1986 				peer = "?";
   1987 			} else {
   1988 				peer = p->name ? : "*";
   1989 			}
   1990 		}
   1991 
   1992 		if (f->f) {
   1993 			struct tcpstat tst;
   1994 			tst.local.family = AF_UNIX;
   1995 			tst.remote.family = AF_UNIX;
   1996 			memcpy(tst.local.data, &s->name, sizeof(s->name));
   1997 			if (strcmp(peer, "*") == 0)
   1998 				memset(tst.remote.data, 0, sizeof(peer));
   1999 			else
   2000 				memcpy(tst.remote.data, &peer, sizeof(peer));
   2001 			if (run_ssfilter(f->f, &tst) == 0)
   2002 				continue;
   2003 		}
   2004 
   2005 		if (netid_width)
   2006 			printf("%-*s ", netid_width,
   2007 			       s->type == SOCK_STREAM ? "u_str" : "u_dgr");
   2008 		if (state_width)
   2009 			printf("%-*s ", state_width, sstate_name[s->state]);
   2010 		printf("%-6d %-6d ", s->rq, s->wq);
   2011 		printf("%*s %-*d %*s %-*d",
   2012 		       addr_width, s->name ? : "*", serv_width, s->ino,
   2013 		       addr_width, peer, serv_width, s->peer);
   2014 		if (show_users) {
   2015 			char ubuf[4096];
   2016 			if (find_users(s->ino, ubuf, sizeof(ubuf)) > 0)
   2017 				printf(" users:(%s)", ubuf);
   2018 		}
   2019 		printf("\n");
   2020 	}
   2021 }
   2022 
   2023 static int unix_show_sock(struct nlmsghdr *nlh, struct filter *f)
   2024 {
   2025 	struct unix_diag_msg *r = NLMSG_DATA(nlh);
   2026 	struct rtattr *tb[UNIX_DIAG_MAX+1];
   2027 	char name[128];
   2028 	int peer_ino;
   2029 	int rqlen;
   2030 
   2031 	parse_rtattr(tb, UNIX_DIAG_MAX, (struct rtattr*)(r+1),
   2032 		     nlh->nlmsg_len - NLMSG_LENGTH(sizeof(*r)));
   2033 
   2034 	if (netid_width)
   2035 		printf("%-*s ", netid_width,
   2036 				r->udiag_type == SOCK_STREAM ? "u_str" : "u_dgr");
   2037 	if (state_width)
   2038 		printf("%-*s ", state_width, sstate_name[r->udiag_state]);
   2039 
   2040 	if (tb[UNIX_DIAG_RQLEN])
   2041 		rqlen = *(int *)RTA_DATA(tb[UNIX_DIAG_RQLEN]);
   2042 	else
   2043 		rqlen = 0;
   2044 
   2045 	printf("%-6d %-6d ", rqlen, 0);
   2046 
   2047 	if (tb[UNIX_DIAG_NAME]) {
   2048 		int len = RTA_PAYLOAD(tb[UNIX_DIAG_NAME]);
   2049 
   2050 		memcpy(name, RTA_DATA(tb[UNIX_DIAG_NAME]), len);
   2051 		name[len] = '\0';
   2052 		if (name[0] == '\0')
   2053 			name[0] = '@';
   2054 	} else
   2055 		sprintf(name, "*");
   2056 
   2057 	if (tb[UNIX_DIAG_PEER])
   2058 		peer_ino = *(int *)RTA_DATA(tb[UNIX_DIAG_PEER]);
   2059 	else
   2060 		peer_ino = 0;
   2061 
   2062 	printf("%*s %-*d %*s %-*d",
   2063 			addr_width, name,
   2064 			serv_width, r->udiag_ino,
   2065 			addr_width, "*", /* FIXME */
   2066 			serv_width, peer_ino);
   2067 
   2068 	if (show_users) {
   2069 		char ubuf[4096];
   2070 		if (find_users(r->udiag_ino, ubuf, sizeof(ubuf)) > 0)
   2071 			printf(" users:(%s)", ubuf);
   2072 	}
   2073 
   2074 	printf("\n");
   2075 
   2076 	return 0;
   2077 }
   2078 
   2079 static int unix_show_netlink(struct filter *f, FILE *dump_fp)
   2080 {
   2081 	int fd;
   2082 	struct {
   2083 		struct nlmsghdr nlh;
   2084 		struct unix_diag_req r;
   2085 	} req;
   2086 	char	buf[8192];
   2087 
   2088 	if ((fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_INET_DIAG)) < 0)
   2089 		return -1;
   2090 
   2091 	memset(&req, 0, sizeof(req));
   2092 	req.nlh.nlmsg_len = sizeof(req);
   2093 	req.nlh.nlmsg_type = SOCK_DIAG_BY_FAMILY;
   2094 	req.nlh.nlmsg_flags = NLM_F_ROOT|NLM_F_MATCH|NLM_F_REQUEST;
   2095 	req.nlh.nlmsg_seq = 123456;
   2096 
   2097 	req.r.sdiag_family = AF_UNIX;
   2098 	req.r.udiag_states = f->states;
   2099 	req.r.udiag_show = UDIAG_SHOW_NAME | UDIAG_SHOW_PEER | UDIAG_SHOW_RQLEN;
   2100 
   2101 	if (send(fd, &req, sizeof(req), 0) < 0) {
   2102 		close(fd);
   2103 		return -1;
   2104 	}
   2105 
   2106 	while (1) {
   2107 		ssize_t status;
   2108 		struct nlmsghdr *h;
   2109 		struct sockaddr_nl nladdr;
   2110 		socklen_t slen = sizeof(nladdr);
   2111 
   2112 		status = recvfrom(fd, buf, sizeof(buf), 0,
   2113 				  (struct sockaddr *) &nladdr, &slen);
   2114 		if (status < 0) {
   2115 			if (errno == EINTR)
   2116 				continue;
   2117 			perror("OVERRUN");
   2118 			continue;
   2119 		}
   2120 		if (status == 0) {
   2121 			fprintf(stderr, "EOF on netlink\n");
   2122 			goto close_it;
   2123 		}
   2124 
   2125 		if (dump_fp)
   2126 			fwrite(buf, 1, NLMSG_ALIGN(status), dump_fp);
   2127 
   2128 		h = (struct nlmsghdr*)buf;
   2129 		while (NLMSG_OK(h, status)) {
   2130 			int err;
   2131 
   2132 			if (/*h->nlmsg_pid != rth->local.nl_pid ||*/
   2133 			    h->nlmsg_seq != 123456)
   2134 				goto skip_it;
   2135 
   2136 			if (h->nlmsg_type == NLMSG_DONE)
   2137 				goto close_it;
   2138 
   2139 			if (h->nlmsg_type == NLMSG_ERROR) {
   2140 				struct nlmsgerr *err = (struct nlmsgerr*)NLMSG_DATA(h);
   2141 				if (h->nlmsg_len < NLMSG_LENGTH(sizeof(struct nlmsgerr))) {
   2142 					fprintf(stderr, "ERROR truncated\n");
   2143 				} else {
   2144 					errno = -err->error;
   2145 					if (errno != ENOENT)
   2146 						fprintf(stderr, "UDIAG answers %d\n", errno);
   2147 				}
   2148 				close(fd);
   2149 				return -1;
   2150 			}
   2151 			if (!dump_fp) {
   2152 				err = unix_show_sock(h, f);
   2153 				if (err < 0) {
   2154 					close(fd);
   2155 					return err;
   2156 				}
   2157 			}
   2158 
   2159 skip_it:
   2160 			h = NLMSG_NEXT(h, status);
   2161 		}
   2162 
   2163 		if (status) {
   2164 			fprintf(stderr, "!!!Remnant of size %zd\n", status);
   2165 			exit(1);
   2166 		}
   2167 	}
   2168 
   2169 close_it:
   2170 	close(fd);
   2171 	return 0;
   2172 }
   2173 
   2174 int unix_show(struct filter *f)
   2175 {
   2176 	FILE *fp;
   2177 	char buf[256];
   2178 	char name[128];
   2179 	int  newformat = 0;
   2180 	int  cnt;
   2181 	struct unixstat *list = NULL;
   2182 
   2183 	if (!getenv("PROC_NET_UNIX") && !getenv("PROC_ROOT")
   2184 	    && unix_show_netlink(f, NULL) == 0)
   2185 		return 0;
   2186 
   2187 	if ((fp = net_unix_open()) == NULL)
   2188 		return -1;
   2189 	fgets(buf, sizeof(buf)-1, fp);
   2190 
   2191 	if (memcmp(buf, "Peer", 4) == 0)
   2192 		newformat = 1;
   2193 	cnt = 0;
   2194 
   2195 	while (fgets(buf, sizeof(buf)-1, fp)) {
   2196 		struct unixstat *u, **insp;
   2197 		int flags;
   2198 
   2199 		if (!(u = malloc(sizeof(*u))))
   2200 			break;
   2201 		u->name = NULL;
   2202 
   2203 		if (sscanf(buf, "%x: %x %x %x %x %x %d %s",
   2204 			   &u->peer, &u->rq, &u->wq, &flags, &u->type,
   2205 			   &u->state, &u->ino, name) < 8)
   2206 			name[0] = 0;
   2207 
   2208 		if (flags&(1<<16)) {
   2209 			u->state = SS_LISTEN;
   2210 		} else {
   2211 			u->state = unix_state_map[u->state-1];
   2212 			if (u->type == SOCK_DGRAM &&
   2213 			    u->state == SS_CLOSE &&
   2214 			    u->peer)
   2215 				u->state = SS_ESTABLISHED;
   2216 		}
   2217 
   2218 		if (!newformat) {
   2219 			u->peer = 0;
   2220 			u->rq = 0;
   2221 			u->wq = 0;
   2222 		}
   2223 
   2224 		insp = &list;
   2225 		while (*insp) {
   2226 			if (u->type < (*insp)->type ||
   2227 			    (u->type == (*insp)->type &&
   2228 			     u->ino < (*insp)->ino))
   2229 				break;
   2230 			insp = &(*insp)->next;
   2231 		}
   2232 		u->next = *insp;
   2233 		*insp = u;
   2234 
   2235 		if (name[0]) {
   2236 			if ((u->name = malloc(strlen(name)+1)) == NULL)
   2237 				break;
   2238 			strcpy(u->name, name);
   2239 		}
   2240 		if (++cnt > MAX_UNIX_REMEMBER) {
   2241 			unix_list_print(list, f);
   2242 			unix_list_free(list);
   2243 			list = NULL;
   2244 			cnt = 0;
   2245 		}
   2246 	}
   2247 	fclose(fp);
   2248 	if (list) {
   2249 		unix_list_print(list, f);
   2250 		unix_list_free(list);
   2251 		list = NULL;
   2252 		cnt = 0;
   2253 	}
   2254 
   2255 	return 0;
   2256 }
   2257 
   2258 
   2259 int packet_show(struct filter *f)
   2260 {
   2261 	FILE *fp;
   2262 	char buf[256];
   2263 	int type;
   2264 	int prot;
   2265 	int iface;
   2266 	int state;
   2267 	int rq;
   2268 	int uid;
   2269 	int ino;
   2270 	unsigned long long sk;
   2271 
   2272 	if (!(f->states & (1<<SS_CLOSE)))
   2273 		return 0;
   2274 
   2275 	if ((fp = net_packet_open()) == NULL)
   2276 		return -1;
   2277 	fgets(buf, sizeof(buf)-1, fp);
   2278 
   2279 	while (fgets(buf, sizeof(buf)-1, fp)) {
   2280 		sscanf(buf, "%llx %*d %d %x %d %d %u %u %u",
   2281 		       &sk,
   2282 		       &type, &prot, &iface, &state,
   2283 		       &rq, &uid, &ino);
   2284 
   2285 		if (type == SOCK_RAW && !(f->dbs&(1<<PACKET_R_DB)))
   2286 			continue;
   2287 		if (type == SOCK_DGRAM && !(f->dbs&(1<<PACKET_DG_DB)))
   2288 			continue;
   2289 		if (f->f) {
   2290 			struct tcpstat tst;
   2291 			tst.local.family = AF_PACKET;
   2292 			tst.remote.family = AF_PACKET;
   2293 			tst.rport = 0;
   2294 			tst.lport = iface;
   2295 			tst.local.data[0] = prot;
   2296 			tst.remote.data[0] = 0;
   2297 			if (run_ssfilter(f->f, &tst) == 0)
   2298 				continue;
   2299 		}
   2300 
   2301 		if (netid_width)
   2302 			printf("%-*s ", netid_width,
   2303 			       type == SOCK_RAW ? "p_raw" : "p_dgr");
   2304 		if (state_width)
   2305 			printf("%-*s ", state_width, "UNCONN");
   2306 		printf("%-6d %-6d ", rq, 0);
   2307 		if (prot == 3) {
   2308 			printf("%*s:", addr_width, "*");
   2309 		} else {
   2310 			char tb[16];
   2311 			printf("%*s:", addr_width,
   2312 			       ll_proto_n2a(htons(prot), tb, sizeof(tb)));
   2313 		}
   2314 		if (iface == 0) {
   2315 			printf("%-*s ", serv_width, "*");
   2316 		} else {
   2317 			printf("%-*s ", serv_width, xll_index_to_name(iface));
   2318 		}
   2319 		printf("%*s*%-*s",
   2320 		       addr_width, "", serv_width, "");
   2321 
   2322 		if (show_users) {
   2323 			char ubuf[4096];
   2324 			if (find_users(ino, ubuf, sizeof(ubuf)) > 0)
   2325 				printf(" users:(%s)", ubuf);
   2326 		}
   2327 		if (show_details) {
   2328 			printf(" ino=%u uid=%u sk=%llx", ino, uid, sk);
   2329 		}
   2330 		printf("\n");
   2331 	}
   2332 
   2333 	return 0;
   2334 }
   2335 
   2336 int netlink_show(struct filter *f)
   2337 {
   2338 	FILE *fp;
   2339 	char buf[256];
   2340 	int prot, pid;
   2341 	unsigned groups;
   2342 	int rq, wq, rc;
   2343 	unsigned long long sk, cb;
   2344 
   2345 	if (!(f->states & (1<<SS_CLOSE)))
   2346 		return 0;
   2347 
   2348 	if ((fp = net_netlink_open()) == NULL)
   2349 		return -1;
   2350 	fgets(buf, sizeof(buf)-1, fp);
   2351 
   2352 	while (fgets(buf, sizeof(buf)-1, fp)) {
   2353 		sscanf(buf, "%llx %d %d %x %d %d %llx %d",
   2354 		       &sk,
   2355 		       &prot, &pid, &groups, &rq, &wq, &cb, &rc);
   2356 
   2357 		if (f->f) {
   2358 			struct tcpstat tst;
   2359 			tst.local.family = AF_NETLINK;
   2360 			tst.remote.family = AF_NETLINK;
   2361 			tst.rport = -1;
   2362 			tst.lport = pid;
   2363 			tst.local.data[0] = prot;
   2364 			tst.remote.data[0] = 0;
   2365 			if (run_ssfilter(f->f, &tst) == 0)
   2366 				continue;
   2367 		}
   2368 
   2369 		if (netid_width)
   2370 			printf("%-*s ", netid_width, "nl");
   2371 		if (state_width)
   2372 			printf("%-*s ", state_width, "UNCONN");
   2373 		printf("%-6d %-6d ", rq, wq);
   2374 		if (resolve_services && prot == 0)
   2375 			printf("%*s:", addr_width, "rtnl");
   2376 		else if (resolve_services && prot == 3)
   2377 			printf("%*s:", addr_width, "fw");
   2378 		else if (resolve_services && prot == 4)
   2379 			printf("%*s:", addr_width, "tcpdiag");
   2380 		else
   2381 			printf("%*d:", addr_width, prot);
   2382 		if (pid == -1) {
   2383 			printf("%-*s ", serv_width, "*");
   2384 		} else if (resolve_services) {
   2385 			int done = 0;
   2386 			if (!pid) {
   2387 				done = 1;
   2388 				printf("%-*s ", serv_width, "kernel");
   2389 			} else if (pid > 0) {
   2390 				char procname[64];
   2391 				FILE *fp;
   2392 				sprintf(procname, "%s/%d/stat",
   2393 					getenv("PROC_ROOT") ? : "/proc", pid);
   2394 				if ((fp = fopen(procname, "r")) != NULL) {
   2395 					if (fscanf(fp, "%*d (%[^)])", procname) == 1) {
   2396 						sprintf(procname+strlen(procname), "/%d", pid);
   2397 						printf("%-*s ", serv_width, procname);
   2398 						done = 1;
   2399 					}
   2400 					fclose(fp);
   2401 				}
   2402 			}
   2403 			if (!done)
   2404 				printf("%-*d ", serv_width, pid);
   2405 		} else {
   2406 			printf("%-*d ", serv_width, pid);
   2407 		}
   2408 		printf("%*s*%-*s",
   2409 		       addr_width, "", serv_width, "");
   2410 
   2411 		if (show_details) {
   2412 			printf(" sk=%llx cb=%llx groups=0x%08x", sk, cb, groups);
   2413 		}
   2414 		printf("\n");
   2415 	}
   2416 
   2417 	return 0;
   2418 }
   2419 
   2420 struct snmpstat
   2421 {
   2422 	int tcp_estab;
   2423 };
   2424 
   2425 int get_snmp_int(char *proto, char *key, int *result)
   2426 {
   2427 	char buf[1024];
   2428 	FILE *fp;
   2429 	int protolen = strlen(proto);
   2430 	int keylen = strlen(key);
   2431 
   2432 	*result = 0;
   2433 
   2434 	if ((fp = net_snmp_open()) == NULL)
   2435 		return -1;
   2436 
   2437 	while (fgets(buf, sizeof(buf), fp) != NULL) {
   2438 		char *p = buf;
   2439 		int  pos = 0;
   2440 		if (memcmp(buf, proto, protolen))
   2441 			continue;
   2442 		while ((p = strchr(p, ' ')) != NULL) {
   2443 			pos++;
   2444 			p++;
   2445 			if (memcmp(p, key, keylen) == 0 &&
   2446 			    (p[keylen] == ' ' || p[keylen] == '\n'))
   2447 				break;
   2448 		}
   2449 		if (fgets(buf, sizeof(buf), fp) == NULL)
   2450 			break;
   2451 		if (memcmp(buf, proto, protolen))
   2452 			break;
   2453 		p = buf;
   2454 		while ((p = strchr(p, ' ')) != NULL) {
   2455 			p++;
   2456 			if (--pos == 0) {
   2457 				sscanf(p, "%d", result);
   2458 				fclose(fp);
   2459 				return 0;
   2460 			}
   2461 		}
   2462 	}
   2463 
   2464 	fclose(fp);
   2465 	errno = ESRCH;
   2466 	return -1;
   2467 }
   2468 
   2469 
   2470 /* Get stats from sockstat */
   2471 
   2472 struct sockstat
   2473 {
   2474 	int socks;
   2475 	int tcp_mem;
   2476 	int tcp_total;
   2477 	int tcp_orphans;
   2478 	int tcp_tws;
   2479 	int tcp4_hashed;
   2480 	int udp4;
   2481 	int raw4;
   2482 	int frag4;
   2483 	int frag4_mem;
   2484 	int tcp6_hashed;
   2485 	int udp6;
   2486 	int raw6;
   2487 	int frag6;
   2488 	int frag6_mem;
   2489 };
   2490 
   2491 static void get_sockstat_line(char *line, struct sockstat *s)
   2492 {
   2493 	char id[256], rem[256];
   2494 
   2495 	if (sscanf(line, "%[^ ] %[^\n]\n", id, rem) != 2)
   2496 		return;
   2497 
   2498 	if (strcmp(id, "sockets:") == 0)
   2499 		sscanf(rem, "%*s%d", &s->socks);
   2500 	else if (strcmp(id, "UDP:") == 0)
   2501 		sscanf(rem, "%*s%d", &s->udp4);
   2502 	else if (strcmp(id, "UDP6:") == 0)
   2503 		sscanf(rem, "%*s%d", &s->udp6);
   2504 	else if (strcmp(id, "RAW:") == 0)
   2505 		sscanf(rem, "%*s%d", &s->raw4);
   2506 	else if (strcmp(id, "RAW6:") == 0)
   2507 		sscanf(rem, "%*s%d", &s->raw6);
   2508 	else if (strcmp(id, "TCP6:") == 0)
   2509 		sscanf(rem, "%*s%d", &s->tcp6_hashed);
   2510 	else if (strcmp(id, "FRAG:") == 0)
   2511 		sscanf(rem, "%*s%d%*s%d", &s->frag4, &s->frag4_mem);
   2512 	else if (strcmp(id, "FRAG6:") == 0)
   2513 		sscanf(rem, "%*s%d%*s%d", &s->frag6, &s->frag6_mem);
   2514 	else if (strcmp(id, "TCP:") == 0)
   2515 		sscanf(rem, "%*s%d%*s%d%*s%d%*s%d%*s%d",
   2516 		       &s->tcp4_hashed,
   2517 		       &s->tcp_orphans, &s->tcp_tws, &s->tcp_total, &s->tcp_mem);
   2518 }
   2519 
   2520 int get_sockstat(struct sockstat *s)
   2521 {
   2522 	char buf[256];
   2523 	FILE *fp;
   2524 
   2525 	memset(s, 0, sizeof(*s));
   2526 
   2527 	if ((fp = net_sockstat_open()) == NULL)
   2528 		return -1;
   2529 	while(fgets(buf, sizeof(buf), fp) != NULL)
   2530 		get_sockstat_line(buf, s);
   2531 	fclose(fp);
   2532 
   2533 	if ((fp = net_sockstat6_open()) == NULL)
   2534 		return 0;
   2535 	while(fgets(buf, sizeof(buf), fp) != NULL)
   2536 		get_sockstat_line(buf, s);
   2537 	fclose(fp);
   2538 
   2539 	return 0;
   2540 }
   2541 
   2542 int print_summary(void)
   2543 {
   2544 	struct sockstat s;
   2545 	struct snmpstat sn;
   2546 
   2547 	if (get_sockstat(&s) < 0)
   2548 		perror("ss: get_sockstat");
   2549 	if (get_snmp_int("Tcp:", "CurrEstab", &sn.tcp_estab) < 0)
   2550 		perror("ss: get_snmpstat");
   2551 
   2552 	printf("Total: %d (kernel %d)\n", s.socks, slabstat.socks);
   2553 
   2554 	printf("TCP:   %d (estab %d, closed %d, orphaned %d, synrecv %d, timewait %d/%d), ports %d\n",
   2555 	       s.tcp_total + slabstat.tcp_syns + s.tcp_tws,
   2556 	       sn.tcp_estab,
   2557 	       s.tcp_total - (s.tcp4_hashed+s.tcp6_hashed-s.tcp_tws),
   2558 	       s.tcp_orphans,
   2559 	       slabstat.tcp_syns,
   2560 	       s.tcp_tws, slabstat.tcp_tws,
   2561 	       slabstat.tcp_ports
   2562 	       );
   2563 
   2564 	printf("\n");
   2565 	printf("Transport Total     IP        IPv6\n");
   2566 	printf("*	  %-9d %-9s %-9s\n", slabstat.socks, "-", "-");
   2567 	printf("RAW	  %-9d %-9d %-9d\n", s.raw4+s.raw6, s.raw4, s.raw6);
   2568 	printf("UDP	  %-9d %-9d %-9d\n", s.udp4+s.udp6, s.udp4, s.udp6);
   2569 	printf("TCP	  %-9d %-9d %-9d\n", s.tcp4_hashed+s.tcp6_hashed, s.tcp4_hashed, s.tcp6_hashed);
   2570 	printf("INET	  %-9d %-9d %-9d\n",
   2571 	       s.raw4+s.udp4+s.tcp4_hashed+
   2572 	       s.raw6+s.udp6+s.tcp6_hashed,
   2573 	       s.raw4+s.udp4+s.tcp4_hashed,
   2574 	       s.raw6+s.udp6+s.tcp6_hashed);
   2575 	printf("FRAG	  %-9d %-9d %-9d\n", s.frag4+s.frag6, s.frag4, s.frag6);
   2576 
   2577 	printf("\n");
   2578 
   2579 	return 0;
   2580 }
   2581 
   2582 static void _usage(FILE *dest)
   2583 {
   2584 	fprintf(dest,
   2585 "Usage: ss [ OPTIONS ]\n"
   2586 "       ss [ OPTIONS ] [ FILTER ]\n"
   2587 "   -h, --help		this message\n"
   2588 "   -V, --version	output version information\n"
   2589 "   -n, --numeric	don't resolve service names\n"
   2590 "   -r, --resolve       resolve host names\n"
   2591 "   -a, --all		display all sockets\n"
   2592 "   -l, --listening	display listening sockets\n"
   2593 "   -o, --options       show timer information\n"
   2594 "   -e, --extended      show detailed socket information\n"
   2595 "   -m, --memory        show socket memory usage\n"
   2596 "   -p, --processes	show process using socket\n"
   2597 "   -i, --info		show internal TCP information\n"
   2598 "   -s, --summary	show socket usage summary\n"
   2599 "\n"
   2600 "   -4, --ipv4          display only IP version 4 sockets\n"
   2601 "   -6, --ipv6          display only IP version 6 sockets\n"
   2602 "   -0, --packet	display PACKET sockets\n"
   2603 "   -t, --tcp		display only TCP sockets\n"
   2604 "   -u, --udp		display only UDP sockets\n"
   2605 "   -d, --dccp		display only DCCP sockets\n"
   2606 "   -w, --raw		display only RAW sockets\n"
   2607 "   -x, --unix		display only Unix domain sockets\n"
   2608 "   -f, --family=FAMILY display sockets of type FAMILY\n"
   2609 "\n"
   2610 "   -A, --query=QUERY, --socket=QUERY\n"
   2611 "       QUERY := {all|inet|tcp|udp|raw|unix|packet|netlink}[,QUERY]\n"
   2612 "\n"
   2613 "   -D, --diag=FILE     Dump raw information about TCP sockets to FILE\n"
   2614 "   -F, --filter=FILE   read filter information from FILE\n"
   2615 "       FILTER := [ state TCP-STATE ] [ EXPRESSION ]\n"
   2616 		);
   2617 }
   2618 
   2619 static void help(void) __attribute__((noreturn));
   2620 static void help(void)
   2621 {
   2622 	_usage(stdout);
   2623 	exit(0);
   2624 }
   2625 
   2626 static void usage(void) __attribute__((noreturn));
   2627 static void usage(void)
   2628 {
   2629 	_usage(stderr);
   2630 	exit(-1);
   2631 }
   2632 
   2633 
   2634 int scan_state(const char *state)
   2635 {
   2636 	int i;
   2637 	if (strcasecmp(state, "close") == 0 ||
   2638 	    strcasecmp(state, "closed") == 0)
   2639 		return (1<<SS_CLOSE);
   2640 	if (strcasecmp(state, "syn-rcv") == 0)
   2641 		return (1<<SS_SYN_RECV);
   2642 	if (strcasecmp(state, "established") == 0)
   2643 		return (1<<SS_ESTABLISHED);
   2644 	if (strcasecmp(state, "all") == 0)
   2645 		return SS_ALL;
   2646 	if (strcasecmp(state, "connected") == 0)
   2647 		return SS_ALL & ~((1<<SS_CLOSE)|(1<<SS_LISTEN));
   2648 	if (strcasecmp(state, "synchronized") == 0)
   2649 		return SS_ALL & ~((1<<SS_CLOSE)|(1<<SS_LISTEN)|(1<<SS_SYN_SENT));
   2650 	if (strcasecmp(state, "bucket") == 0)
   2651 		return (1<<SS_SYN_RECV)|(1<<SS_TIME_WAIT);
   2652 	if (strcasecmp(state, "big") == 0)
   2653 		return SS_ALL & ~((1<<SS_SYN_RECV)|(1<<SS_TIME_WAIT));
   2654 	for (i=0; i<SS_MAX; i++) {
   2655 		if (strcasecmp(state, sstate_namel[i]) == 0)
   2656 			return (1<<i);
   2657 	}
   2658 	return 0;
   2659 }
   2660 
   2661 static const struct option long_opts[] = {
   2662 	{ "numeric", 0, 0, 'n' },
   2663 	{ "resolve", 0, 0, 'r' },
   2664 	{ "options", 0, 0, 'o' },
   2665 	{ "extended", 0, 0, 'e' },
   2666 	{ "memory", 0, 0, 'm' },
   2667 	{ "info", 0, 0, 'i' },
   2668 	{ "processes", 0, 0, 'p' },
   2669 	{ "dccp", 0, 0, 'd' },
   2670 	{ "tcp", 0, 0, 't' },
   2671 	{ "udp", 0, 0, 'u' },
   2672 	{ "raw", 0, 0, 'w' },
   2673 	{ "unix", 0, 0, 'x' },
   2674 	{ "all", 0, 0, 'a' },
   2675 	{ "listening", 0, 0, 'l' },
   2676 	{ "ipv4", 0, 0, '4' },
   2677 	{ "ipv6", 0, 0, '6' },
   2678 	{ "packet", 0, 0, '0' },
   2679 	{ "family", 1, 0, 'f' },
   2680 	{ "socket", 1, 0, 'A' },
   2681 	{ "query", 1, 0, 'A' },
   2682 	{ "summary", 0, 0, 's' },
   2683 	{ "diag", 1, 0, 'D' },
   2684 	{ "filter", 1, 0, 'F' },
   2685 	{ "version", 0, 0, 'V' },
   2686 	{ "help", 0, 0, 'h' },
   2687 	{ 0 }
   2688 
   2689 };
   2690 
   2691 int main(int argc, char *argv[])
   2692 {
   2693 	int do_default = 1;
   2694 	int saw_states = 0;
   2695 	int saw_query = 0;
   2696 	int do_summary = 0;
   2697 	const char *dump_tcpdiag = NULL;
   2698 	FILE *filter_fp = NULL;
   2699 	int ch;
   2700 
   2701 	memset(&current_filter, 0, sizeof(current_filter));
   2702 
   2703 	current_filter.states = default_filter.states;
   2704 
   2705 	while ((ch = getopt_long(argc, argv, "dhaletuwxnro460spf:miA:D:F:vV",
   2706 				 long_opts, NULL)) != EOF) {
   2707 		switch(ch) {
   2708 		case 'n':
   2709 			resolve_services = 0;
   2710 			break;
   2711 		case 'r':
   2712 			resolve_hosts = 1;
   2713 			break;
   2714 		case 'o':
   2715 			show_options = 1;
   2716 			break;
   2717 		case 'e':
   2718 			show_options = 1;
   2719 			show_details++;
   2720 			break;
   2721 		case 'm':
   2722 			show_mem = 1;
   2723 			break;
   2724 		case 'i':
   2725 			show_tcpinfo = 1;
   2726 			break;
   2727 		case 'p':
   2728 			show_users++;
   2729 			user_ent_hash_build();
   2730 			break;
   2731 		case 'd':
   2732 			current_filter.dbs |= (1<<DCCP_DB);
   2733 			do_default = 0;
   2734 			break;
   2735 		case 't':
   2736 			current_filter.dbs |= (1<<TCP_DB);
   2737 			do_default = 0;
   2738 			break;
   2739 		case 'u':
   2740 			current_filter.dbs |= (1<<UDP_DB);
   2741 			do_default = 0;
   2742 			break;
   2743 		case 'w':
   2744 			current_filter.dbs |= (1<<RAW_DB);
   2745 			do_default = 0;
   2746 			break;
   2747 		case 'x':
   2748 			current_filter.dbs |= UNIX_DBM;
   2749 			do_default = 0;
   2750 			break;
   2751 		case 'a':
   2752 			current_filter.states = SS_ALL;
   2753 			break;
   2754 		case 'l':
   2755 			current_filter.states = (1<<SS_LISTEN) | (1<<SS_CLOSE);
   2756 			break;
   2757 		case '4':
   2758 			preferred_family = AF_INET;
   2759 			break;
   2760 		case '6':
   2761 			preferred_family = AF_INET6;
   2762 			break;
   2763 		case '0':
   2764 			preferred_family = AF_PACKET;
   2765 			break;
   2766 		case 'f':
   2767 			if (strcmp(optarg, "inet") == 0)
   2768 				preferred_family = AF_INET;
   2769 			else if (strcmp(optarg, "inet6") == 0)
   2770 				preferred_family = AF_INET6;
   2771 			else if (strcmp(optarg, "link") == 0)
   2772 				preferred_family = AF_PACKET;
   2773 			else if (strcmp(optarg, "unix") == 0)
   2774 				preferred_family = AF_UNIX;
   2775 			else if (strcmp(optarg, "netlink") == 0)
   2776 				preferred_family = AF_NETLINK;
   2777 			else if (strcmp(optarg, "help") == 0)
   2778 				help();
   2779 			else {
   2780 				fprintf(stderr, "ss: \"%s\" is invalid family\n", optarg);
   2781 				usage();
   2782 			}
   2783 			break;
   2784 		case 'A':
   2785 		{
   2786 			char *p, *p1;
   2787 			if (!saw_query) {
   2788 				current_filter.dbs = 0;
   2789 				saw_query = 1;
   2790 				do_default = 0;
   2791 			}
   2792 			p = p1 = optarg;
   2793 			do {
   2794 				if ((p1 = strchr(p, ',')) != NULL)
   2795 					*p1 = 0;
   2796 				if (strcmp(p, "all") == 0) {
   2797 					current_filter.dbs = ALL_DB;
   2798 				} else if (strcmp(p, "inet") == 0) {
   2799 					current_filter.dbs |= (1<<TCP_DB)|(1<<DCCP_DB)|(1<<UDP_DB)|(1<<RAW_DB);
   2800 				} else if (strcmp(p, "udp") == 0) {
   2801 					current_filter.dbs |= (1<<UDP_DB);
   2802 				} else if (strcmp(p, "dccp") == 0) {
   2803 					current_filter.dbs |= (1<<DCCP_DB);
   2804 				} else if (strcmp(p, "tcp") == 0) {
   2805 					current_filter.dbs |= (1<<TCP_DB);
   2806 				} else if (strcmp(p, "raw") == 0) {
   2807 					current_filter.dbs |= (1<<RAW_DB);
   2808 				} else if (strcmp(p, "unix") == 0) {
   2809 					current_filter.dbs |= UNIX_DBM;
   2810 				} else if (strcasecmp(p, "unix_stream") == 0 ||
   2811 					   strcmp(p, "u_str") == 0) {
   2812 					current_filter.dbs |= (1<<UNIX_ST_DB);
   2813 				} else if (strcasecmp(p, "unix_dgram") == 0 ||
   2814 					   strcmp(p, "u_dgr") == 0) {
   2815 					current_filter.dbs |= (1<<UNIX_DG_DB);
   2816 				} else if (strcmp(p, "packet") == 0) {
   2817 					current_filter.dbs |= PACKET_DBM;
   2818 				} else if (strcmp(p, "packet_raw") == 0 ||
   2819 					   strcmp(p, "p_raw") == 0) {
   2820 					current_filter.dbs |= (1<<PACKET_R_DB);
   2821 				} else if (strcmp(p, "packet_dgram") == 0 ||
   2822 					   strcmp(p, "p_dgr") == 0) {
   2823 					current_filter.dbs |= (1<<PACKET_DG_DB);
   2824 				} else if (strcmp(p, "netlink") == 0) {
   2825 					current_filter.dbs |= (1<<NETLINK_DB);
   2826 				} else {
   2827 					fprintf(stderr, "ss: \"%s\" is illegal socket table id\n", p);
   2828 					usage();
   2829 				}
   2830 				p = p1 + 1;
   2831 			} while (p1);
   2832 			break;
   2833 		}
   2834 		case 's':
   2835 			do_summary = 1;
   2836 			break;
   2837 		case 'D':
   2838 			dump_tcpdiag = optarg;
   2839 			break;
   2840 		case 'F':
   2841 			if (filter_fp) {
   2842 				fprintf(stderr, "More than one filter file\n");
   2843 				exit(-1);
   2844 			}
   2845 			if (optarg[0] == '-')
   2846 				filter_fp = stdin;
   2847 			else
   2848 				filter_fp = fopen(optarg, "r");
   2849 			if (!filter_fp) {
   2850 				perror("fopen filter file");
   2851 				exit(-1);
   2852 			}
   2853 			break;
   2854 		case 'v':
   2855 		case 'V':
   2856 			printf("ss utility, iproute2-ss%s\n", SNAPSHOT);
   2857 			exit(0);
   2858 		case 'h':
   2859 		case '?':
   2860 			help();
   2861 		default:
   2862 			usage();
   2863 		}
   2864 	}
   2865 
   2866 	argc -= optind;
   2867 	argv += optind;
   2868 
   2869 	get_slabstat(&slabstat);
   2870 
   2871 	if (do_summary) {
   2872 		print_summary();
   2873 		if (do_default && argc == 0)
   2874 			exit(0);
   2875 	}
   2876 
   2877 	if (do_default)
   2878 		current_filter.dbs = default_filter.dbs;
   2879 
   2880 	if (preferred_family == AF_UNSPEC) {
   2881 		if (!(current_filter.dbs&~UNIX_DBM))
   2882 			preferred_family = AF_UNIX;
   2883 		else if (!(current_filter.dbs&~PACKET_DBM))
   2884 			preferred_family = AF_PACKET;
   2885 		else if (!(current_filter.dbs&~(1<<NETLINK_DB)))
   2886 			preferred_family = AF_NETLINK;
   2887 	}
   2888 
   2889 	if (preferred_family != AF_UNSPEC) {
   2890 		int mask2;
   2891 		if (preferred_family == AF_INET ||
   2892 		    preferred_family == AF_INET6) {
   2893 			mask2= current_filter.dbs;
   2894 		} else if (preferred_family == AF_PACKET) {
   2895 			mask2 = PACKET_DBM;
   2896 		} else if (preferred_family == AF_UNIX) {
   2897 			mask2 = UNIX_DBM;
   2898 		} else if (preferred_family == AF_NETLINK) {
   2899 			mask2 = (1<<NETLINK_DB);
   2900 		} else {
   2901 			mask2 = 0;
   2902 		}
   2903 
   2904 		if (do_default)
   2905 			current_filter.dbs = mask2;
   2906 		else
   2907 			current_filter.dbs &= mask2;
   2908 		current_filter.families = (1<<preferred_family);
   2909 	} else {
   2910 		if (!do_default)
   2911 			current_filter.families = ~0;
   2912 		else
   2913 			current_filter.families = default_filter.families;
   2914 	}
   2915 	if (current_filter.dbs == 0) {
   2916 		fprintf(stderr, "ss: no socket tables to show with such filter.\n");
   2917 		exit(0);
   2918 	}
   2919 	if (current_filter.families == 0) {
   2920 		fprintf(stderr, "ss: no families to show with such filter.\n");
   2921 		exit(0);
   2922 	}
   2923 
   2924 	if (resolve_services && resolve_hosts &&
   2925 	    (current_filter.dbs&(UNIX_DBM|(1<<TCP_DB)|(1<<UDP_DB)|(1<<DCCP_DB))))
   2926 		init_service_resolver();
   2927 
   2928 	/* Now parse filter... */
   2929 	if (argc == 0 && filter_fp) {
   2930 		if (ssfilter_parse(&current_filter.f, 0, NULL, filter_fp))
   2931 			usage();
   2932 	}
   2933 
   2934 	while (argc > 0) {
   2935 		if (strcmp(*argv, "state") == 0) {
   2936 			NEXT_ARG();
   2937 			if (!saw_states)
   2938 				current_filter.states = 0;
   2939 			current_filter.states |= scan_state(*argv);
   2940 			saw_states = 1;
   2941 		} else if (strcmp(*argv, "exclude") == 0 ||
   2942 			   strcmp(*argv, "excl") == 0) {
   2943 			NEXT_ARG();
   2944 			if (!saw_states)
   2945 				current_filter.states = SS_ALL;
   2946 			current_filter.states &= ~scan_state(*argv);
   2947 			saw_states = 1;
   2948 		} else {
   2949 			if (ssfilter_parse(&current_filter.f, argc, argv, filter_fp))
   2950 				usage();
   2951 			break;
   2952 		}
   2953 		argc--; argv++;
   2954 	}
   2955 
   2956 	if (current_filter.states == 0) {
   2957 		fprintf(stderr, "ss: no socket states to show with such filter.\n");
   2958 		exit(0);
   2959 	}
   2960 
   2961 	if (dump_tcpdiag) {
   2962 		FILE *dump_fp = stdout;
   2963 		if (!(current_filter.dbs & (1<<TCP_DB))) {
   2964 			fprintf(stderr, "ss: tcpdiag dump requested and no tcp in filter.\n");
   2965 			exit(0);
   2966 		}
   2967 		if (dump_tcpdiag[0] != '-') {
   2968 			dump_fp = fopen(dump_tcpdiag, "w");
   2969 			if (!dump_tcpdiag) {
   2970 				perror("fopen dump file");
   2971 				exit(-1);
   2972 			}
   2973 		}
   2974 		tcp_show_netlink(&current_filter, dump_fp, TCPDIAG_GETSOCK);
   2975 		fflush(dump_fp);
   2976 		exit(0);
   2977 	}
   2978 
   2979 	netid_width = 0;
   2980 	if (current_filter.dbs&(current_filter.dbs-1))
   2981 		netid_width = 5;
   2982 
   2983 	state_width = 0;
   2984 	if (current_filter.states&(current_filter.states-1))
   2985 		state_width = 10;
   2986 
   2987 	screen_width = 80;
   2988 	if (isatty(STDOUT_FILENO)) {
   2989 		struct winsize w;
   2990 
   2991 		if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &w) != -1) {
   2992 			if (w.ws_col > 0)
   2993 				screen_width = w.ws_col;
   2994 		}
   2995 	}
   2996 
   2997 	addrp_width = screen_width;
   2998 	addrp_width -= netid_width+1;
   2999 	addrp_width -= state_width+1;
   3000 	addrp_width -= 14;
   3001 
   3002 	if (addrp_width&1) {
   3003 		if (netid_width)
   3004 			netid_width++;
   3005 		else if (state_width)
   3006 			state_width++;
   3007 	}
   3008 
   3009 	addrp_width /= 2;
   3010 	addrp_width--;
   3011 
   3012 	serv_width = resolve_services ? 7 : 5;
   3013 
   3014 	if (addrp_width < 15+serv_width+1)
   3015 		addrp_width = 15+serv_width+1;
   3016 
   3017 	addr_width = addrp_width - serv_width - 1;
   3018 
   3019 	if (netid_width)
   3020 		printf("%-*s ", netid_width, "Netid");
   3021 	if (state_width)
   3022 		printf("%-*s ", state_width, "State");
   3023 	printf("%-6s %-6s ", "Recv-Q", "Send-Q");
   3024 
   3025 	printf("%*s:%-*s %*s:%-*s\n",
   3026 	       addr_width, "Local Address", serv_width, "Port",
   3027 	       addr_width, "Peer Address", serv_width, "Port");
   3028 
   3029 	fflush(stdout);
   3030 
   3031 	if (current_filter.dbs & (1<<NETLINK_DB))
   3032 		netlink_show(&current_filter);
   3033 	if (current_filter.dbs & PACKET_DBM)
   3034 		packet_show(&current_filter);
   3035 	if (current_filter.dbs & UNIX_DBM)
   3036 		unix_show(&current_filter);
   3037 	if (current_filter.dbs & (1<<RAW_DB))
   3038 		raw_show(&current_filter);
   3039 	if (current_filter.dbs & (1<<UDP_DB))
   3040 		udp_show(&current_filter);
   3041 	if (current_filter.dbs & (1<<TCP_DB))
   3042 		tcp_show(&current_filter, TCPDIAG_GETSOCK);
   3043 	if (current_filter.dbs & (1<<DCCP_DB))
   3044 		tcp_show(&current_filter, DCCPDIAG_GETSOCK);
   3045 	return 0;
   3046 }
   3047