close_wait.stp

#!/usr/bin/stap

global sockets

probe tcp_state.tcp_fin
{
	state = sk_state(sk)
	if (state == TCP_ESTABLISHED || state == TCP_SYN_RECV)
		sockets[inet_sk_saddr(sk), inet_sk_sport(sk),
			inet_sk_daddr(sk), inet_sk_dport(sk)] = gettimeofday_us()
}

probe tcp_state.tcp_close_state
{
	if (sk_state(sk) != TCP_CLOSE_WAIT)
		next

	saddr = inet_sk_saddr(sk)
	daddr = inet_sk_daddr(sk)
	sport = inet_sk_sport(sk)
	dport = inet_sk_dport(sk)
	park_time = gettimeofday_us() - sockets[saddr, sport, daddr, dport]
	delete sockets[saddr, sport, daddr, dport]
	printf("%s@%d -> %s@%d: %dus in CLOSE_WAIT state\n",
	       inet_sk_ntop(saddr), sport, inet_sk_ntop(daddr), dport, park_time)
}