implement sigar_net_connection_walk

This commit is contained in:
Doug MacEachern 2006-07-01 16:41:14 +00:00
parent 475ddf2428
commit bf6984b477
1 changed files with 72 additions and 35 deletions

View File

@ -1695,10 +1695,10 @@ int sigar_net_interface_stat_get(sigar_t *sigar, const char *name,
#include <netinet/tcp_var.h>
#include <netinet/tcp_fsm.h>
static int net_connection_list_get(sigar_t *sigar,
sigar_net_connection_list_t *connlist,
int flags, int proto)
static int net_connection_get(sigar_net_connection_walker_t *walker, int proto)
{
sigar_t *sigar = walker->sigar;
int flags = walker->flags;
int type, istcp = 0;
char *buf;
const char *mibvar;
@ -1764,62 +1764,68 @@ static int net_connection_list_get(sigar_t *sigar,
if ((((flags & SIGAR_NETCONN_SERVER) && so->so_qlimit) ||
((flags & SIGAR_NETCONN_CLIENT) && !so->so_qlimit)))
{
sigar_net_connection_t *conn;
sigar_net_connection_t conn;
SIGAR_NET_CONNLIST_GROW(connlist);
conn = &connlist->data[connlist->number++];
SIGAR_ZERO(&conn);
sigar_inet_ntoa(sigar, inp->inp_laddr.s_addr,
conn->local_address);
conn.local_address);
sigar_inet_ntoa(sigar, inp->inp_faddr.s_addr,
conn->remote_address);
conn->local_port = ntohs(inp->inp_lport);
conn->remote_port = ntohs(inp->inp_fport);
conn->receive_queue = so->so_rcv.sb_cc;
conn->send_queue = so->so_snd.sb_cc;
conn->type = type;
conn.remote_address);
conn.local_port = ntohs(inp->inp_lport);
conn.remote_port = ntohs(inp->inp_fport);
conn.receive_queue = so->so_rcv.sb_cc;
conn.send_queue = so->so_snd.sb_cc;
conn.type = type;
if (!istcp) {
conn->state = SIGAR_TCP_UNKNOWN;
conn.state = SIGAR_TCP_UNKNOWN;
if (walker->add_connection(walker, &conn) != SIGAR_OK) {
break;
}
continue;
}
switch (tp->t_state) {
case TCPS_CLOSED:
conn->state = SIGAR_TCP_CLOSE;
conn.state = SIGAR_TCP_CLOSE;
break;
case TCPS_LISTEN:
conn->state = SIGAR_TCP_LISTEN;
conn.state = SIGAR_TCP_LISTEN;
break;
case TCPS_SYN_SENT:
conn->state = SIGAR_TCP_SYN_SENT;
conn.state = SIGAR_TCP_SYN_SENT;
break;
case TCPS_SYN_RECEIVED:
conn->state = SIGAR_TCP_SYN_RECV;
conn.state = SIGAR_TCP_SYN_RECV;
break;
case TCPS_ESTABLISHED:
conn->state = SIGAR_TCP_ESTABLISHED;
conn.state = SIGAR_TCP_ESTABLISHED;
break;
case TCPS_CLOSE_WAIT:
conn->state = SIGAR_TCP_CLOSE_WAIT;
conn.state = SIGAR_TCP_CLOSE_WAIT;
break;
case TCPS_FIN_WAIT_1:
conn->state = SIGAR_TCP_FIN_WAIT1;
conn.state = SIGAR_TCP_FIN_WAIT1;
break;
case TCPS_CLOSING:
conn->state = SIGAR_TCP_CLOSING;
conn.state = SIGAR_TCP_CLOSING;
break;
case TCPS_LAST_ACK:
conn->state = SIGAR_TCP_LAST_ACK;
conn.state = SIGAR_TCP_LAST_ACK;
break;
case TCPS_FIN_WAIT_2:
conn->state = SIGAR_TCP_FIN_WAIT2;
conn.state = SIGAR_TCP_FIN_WAIT2;
break;
case TCPS_TIME_WAIT:
conn->state = SIGAR_TCP_TIME_WAIT;
conn.state = SIGAR_TCP_TIME_WAIT;
break;
default:
conn->state = SIGAR_TCP_UNKNOWN;
conn.state = SIGAR_TCP_UNKNOWN;
break;
}
if (walker->add_connection(walker, &conn) != SIGAR_OK) {
break;
}
}
@ -1830,24 +1836,19 @@ static int net_connection_list_get(sigar_t *sigar,
return SIGAR_OK;
}
int sigar_net_connection_list_get(sigar_t *sigar,
sigar_net_connection_list_t *connlist,
int flags)
int sigar_net_connection_walk(sigar_net_connection_walker_t *walker)
{
int flags = walker->flags;
int status;
sigar_net_connection_list_create(connlist);
if (flags & SIGAR_NETCONN_TCP) {
status = net_connection_list_get(sigar, connlist,
flags, IPPROTO_TCP);
status = net_connection_get(walker, IPPROTO_TCP);
if (status != SIGAR_OK) {
return status;
}
}
if (flags & SIGAR_NETCONN_UDP) {
status = net_connection_list_get(sigar, connlist,
flags, IPPROTO_UDP);
status = net_connection_get(walker, IPPROTO_UDP);
if (status != SIGAR_OK) {
return status;
}
@ -1856,6 +1857,42 @@ int sigar_net_connection_list_get(sigar_t *sigar,
return SIGAR_OK;
}
static int net_connection_list_walker(sigar_net_connection_walker_t *walker,
sigar_net_connection_t *conn)
{
sigar_net_connection_list_t *connlist =
(sigar_net_connection_list_t *)walker->data;
SIGAR_NET_CONNLIST_GROW(connlist);
memcpy(&connlist->data[connlist->number++],
conn, sizeof(*conn));
return SIGAR_OK; /* continue loop */
}
int sigar_net_connection_list_get(sigar_t *sigar,
sigar_net_connection_list_t *connlist,
int flags)
{
int status;
sigar_net_connection_walker_t walker;
sigar_net_connection_list_create(connlist);
walker.sigar = sigar;
walker.flags = flags;
walker.data = connlist;
walker.add_connection = net_connection_list_walker;
status = sigar_net_connection_walk(&walker);
if (status != SIGAR_OK) {
sigar_net_connection_list_destroy(sigar, connlist);
}
return status;
}
#ifndef DARWIN
#define _KERNEL