|
Packit Service |
3880ab |
/*
|
|
Packit Service |
3880ab |
* tcp_metrics.c "ip tcp_metrics/tcpmetrics"
|
|
Packit Service |
3880ab |
*
|
|
Packit Service |
3880ab |
* This program is free software; you can redistribute it and/or
|
|
Packit Service |
3880ab |
* modify it under the terms of the GNU General Public License
|
|
Packit Service |
3880ab |
* version 2 as published by the Free Software Foundation;
|
|
Packit Service |
3880ab |
*
|
|
Packit Service |
3880ab |
* Authors: Julian Anastasov <ja@ssi.bg>, August 2012
|
|
Packit Service |
3880ab |
*/
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
#include <stdio.h>
|
|
Packit Service |
3880ab |
#include <stdlib.h>
|
|
Packit Service |
3880ab |
#include <string.h>
|
|
Packit Service |
3880ab |
#include <unistd.h>
|
|
Packit Service |
3880ab |
#include <errno.h>
|
|
Packit Service |
3880ab |
#include <sys/types.h>
|
|
Packit Service |
3880ab |
#include <sys/socket.h>
|
|
Packit Service |
3880ab |
#include <arpa/inet.h>
|
|
Packit Service |
3880ab |
#include <sys/ioctl.h>
|
|
Packit Service |
3880ab |
#include <linux/if.h>
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
#include <linux/genetlink.h>
|
|
Packit Service |
3880ab |
#include <linux/tcp_metrics.h>
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
#include "utils.h"
|
|
Packit Service |
3880ab |
#include "ip_common.h"
|
|
Packit Service |
3880ab |
#include "libgenl.h"
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static void usage(void)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
fprintf(stderr,
|
|
Packit Service |
3880ab |
"Usage: ip tcp_metrics/tcpmetrics { COMMAND | help }\n"
|
|
Packit Service |
3880ab |
" ip tcp_metrics { show | flush } SELECTOR\n"
|
|
Packit Service |
3880ab |
" ip tcp_metrics delete [ address ] ADDRESS\n"
|
|
Packit Service |
3880ab |
"SELECTOR := [ [ address ] PREFIX ]\n");
|
|
Packit Service |
3880ab |
exit(-1);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
/* netlink socket */
|
|
Packit Service |
3880ab |
static struct rtnl_handle grth = { .fd = -1 };
|
|
Packit Service |
3880ab |
static int genl_family = -1;
|
|
Packit Service |
3880ab |
static const double usec_per_sec = 1000000.;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
#define TCPM_REQUEST(_req, _bufsiz, _cmd, _flags) \
|
|
Packit Service |
3880ab |
GENL_REQUEST(_req, _bufsiz, genl_family, 0, \
|
|
Packit Service |
3880ab |
TCP_METRICS_GENL_VERSION, _cmd, _flags)
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
#define CMD_LIST 0x0001 /* list, lst, show */
|
|
Packit Service |
3880ab |
#define CMD_DEL 0x0002 /* delete, remove */
|
|
Packit Service |
3880ab |
#define CMD_FLUSH 0x0004 /* flush */
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static const struct {
|
|
Packit Service |
3880ab |
const char *name;
|
|
Packit Service |
3880ab |
int code;
|
|
Packit Service |
3880ab |
} cmds[] = {
|
|
Packit Service |
3880ab |
{ "list", CMD_LIST },
|
|
Packit Service |
3880ab |
{ "lst", CMD_LIST },
|
|
Packit Service |
3880ab |
{ "show", CMD_LIST },
|
|
Packit Service |
3880ab |
{ "delete", CMD_DEL },
|
|
Packit Service |
3880ab |
{ "remove", CMD_DEL },
|
|
Packit Service |
3880ab |
{ "flush", CMD_FLUSH },
|
|
Packit Service |
3880ab |
};
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static const char *metric_name[TCP_METRIC_MAX + 1] = {
|
|
Packit Service |
3880ab |
[TCP_METRIC_RTT] = "rtt",
|
|
Packit Service |
3880ab |
[TCP_METRIC_RTTVAR] = "rttvar",
|
|
Packit Service |
3880ab |
[TCP_METRIC_SSTHRESH] = "ssthresh",
|
|
Packit Service |
3880ab |
[TCP_METRIC_CWND] = "cwnd",
|
|
Packit Service |
3880ab |
[TCP_METRIC_REORDERING] = "reordering",
|
|
Packit Service |
3880ab |
};
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static struct {
|
|
Packit Service |
3880ab |
int flushed;
|
|
Packit Service |
3880ab |
char *flushb;
|
|
Packit Service |
3880ab |
int flushp;
|
|
Packit Service |
3880ab |
int flushe;
|
|
Packit Service |
3880ab |
int cmd;
|
|
Packit Service |
3880ab |
inet_prefix daddr;
|
|
Packit Service |
3880ab |
inet_prefix saddr;
|
|
Packit Service |
3880ab |
} f;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static int flush_update(void)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
if (rtnl_send_check(&grth, f.flushb, f.flushp) < 0) {
|
|
Packit Service |
3880ab |
perror("Failed to send flush request\n");
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
f.flushp = 0;
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static void print_tcp_metrics(struct rtattr *a)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
struct rtattr *m[TCP_METRIC_MAX + 1 + 1];
|
|
Packit Service |
3880ab |
unsigned long rtt = 0, rttvar = 0;
|
|
Packit Service |
3880ab |
int i;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
parse_rtattr_nested(m, TCP_METRIC_MAX + 1, a);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
for (i = 0; i < TCP_METRIC_MAX + 1; i++) {
|
|
Packit Service |
3880ab |
const char *name;
|
|
Packit Service |
3880ab |
__u32 val;
|
|
Packit Service |
3880ab |
SPRINT_BUF(b1);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
a = m[i + 1];
|
|
Packit Service |
3880ab |
if (!a)
|
|
Packit Service |
3880ab |
continue;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
val = rta_getattr_u32(a);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
switch (i) {
|
|
Packit Service |
3880ab |
case TCP_METRIC_RTT:
|
|
Packit Service |
3880ab |
if (!rtt)
|
|
Packit Service |
3880ab |
rtt = (val * 1000UL) >> 3;
|
|
Packit Service |
3880ab |
continue;
|
|
Packit Service |
3880ab |
case TCP_METRIC_RTTVAR:
|
|
Packit Service |
3880ab |
if (!rttvar)
|
|
Packit Service |
3880ab |
rttvar = (val * 1000UL) >> 2;
|
|
Packit Service |
3880ab |
continue;
|
|
Packit Service |
3880ab |
case TCP_METRIC_RTT_US:
|
|
Packit Service |
3880ab |
rtt = val >> 3;
|
|
Packit Service |
3880ab |
continue;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
case TCP_METRIC_RTTVAR_US:
|
|
Packit Service |
3880ab |
rttvar = val >> 2;
|
|
Packit Service |
3880ab |
continue;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
case TCP_METRIC_SSTHRESH:
|
|
Packit Service |
3880ab |
case TCP_METRIC_CWND:
|
|
Packit Service |
3880ab |
case TCP_METRIC_REORDERING:
|
|
Packit Service |
3880ab |
name = metric_name[i];
|
|
Packit Service |
3880ab |
break;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
default:
|
|
Packit Service |
3880ab |
snprintf(b1, sizeof(b1),
|
|
Packit Service |
3880ab |
" metric_%d ", i);
|
|
Packit Service |
3880ab |
name = b1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_uint(PRINT_JSON, name, NULL, val);
|
|
Packit Service |
3880ab |
print_string(PRINT_FP, NULL, " %s ", name);
|
|
Packit Service |
3880ab |
print_uint(PRINT_FP, NULL, "%u", val);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (rtt) {
|
|
Packit Service |
3880ab |
print_float(PRINT_JSON, "rtt", NULL,
|
|
Packit Service |
3880ab |
(double)rtt / usec_per_sec);
|
|
Packit Service |
3880ab |
print_u64(PRINT_FP, NULL,
|
|
Packit Service |
3880ab |
" rtt %luus", rtt);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
if (rttvar) {
|
|
Packit Service |
3880ab |
print_float(PRINT_JSON, "rttvar", NULL,
|
|
Packit Service |
3880ab |
(double) rttvar / usec_per_sec);
|
|
Packit Service |
3880ab |
print_u64(PRINT_FP, NULL,
|
|
Packit Service |
3880ab |
" rttvar %luus", rttvar);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static int process_msg(struct nlmsghdr *n, void *arg)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
FILE *fp = (FILE *) arg;
|
|
Packit Service |
3880ab |
struct genlmsghdr *ghdr;
|
|
Packit Service |
3880ab |
struct rtattr *attrs[TCP_METRICS_ATTR_MAX + 1], *a;
|
|
Packit Service |
3880ab |
const char *h;
|
|
Packit Service |
3880ab |
int len = n->nlmsg_len;
|
|
Packit Service |
3880ab |
inet_prefix daddr, saddr;
|
|
Packit Service |
3880ab |
int atype, stype;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (n->nlmsg_type != genl_family)
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
len -= NLMSG_LENGTH(GENL_HDRLEN);
|
|
Packit Service |
3880ab |
if (len < 0)
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
ghdr = NLMSG_DATA(n);
|
|
Packit Service |
3880ab |
if (ghdr->cmd != TCP_METRICS_CMD_GET)
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
parse_rtattr(attrs, TCP_METRICS_ATTR_MAX, (void *) ghdr + GENL_HDRLEN,
|
|
Packit Service |
3880ab |
len);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (attrs[TCP_METRICS_ATTR_ADDR_IPV4]) {
|
|
Packit Service |
3880ab |
if (f.daddr.family && f.daddr.family != AF_INET)
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_ADDR_IPV4];
|
|
Packit Service |
3880ab |
daddr.family = AF_INET;
|
|
Packit Service |
3880ab |
atype = TCP_METRICS_ATTR_ADDR_IPV4;
|
|
Packit Service |
3880ab |
} else if (attrs[TCP_METRICS_ATTR_ADDR_IPV6]) {
|
|
Packit Service |
3880ab |
if (f.daddr.family && f.daddr.family != AF_INET6)
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_ADDR_IPV6];
|
|
Packit Service |
3880ab |
daddr.family = AF_INET6;
|
|
Packit Service |
3880ab |
atype = TCP_METRICS_ATTR_ADDR_IPV6;
|
|
Packit Service |
3880ab |
} else {
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (get_addr_rta(&daddr, a, daddr.family))
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (f.daddr.family && f.daddr.bitlen >= 0 &&
|
|
Packit Service |
3880ab |
inet_addr_match(&daddr, &f.daddr, f.daddr.bitlen))
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (attrs[TCP_METRICS_ATTR_SADDR_IPV4]) {
|
|
Packit Service |
3880ab |
if (f.saddr.family && f.saddr.family != AF_INET)
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_SADDR_IPV4];
|
|
Packit Service |
3880ab |
saddr.family = AF_INET;
|
|
Packit Service |
3880ab |
stype = TCP_METRICS_ATTR_SADDR_IPV4;
|
|
Packit Service |
3880ab |
} else if (attrs[TCP_METRICS_ATTR_SADDR_IPV6]) {
|
|
Packit Service |
3880ab |
if (f.saddr.family && f.saddr.family != AF_INET6)
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_SADDR_IPV6];
|
|
Packit Service |
3880ab |
saddr.family = AF_INET6;
|
|
Packit Service |
3880ab |
stype = TCP_METRICS_ATTR_SADDR_IPV6;
|
|
Packit Service |
3880ab |
} else {
|
|
Packit Service |
3880ab |
saddr.family = AF_UNSPEC;
|
|
Packit Service |
3880ab |
stype = 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
/* Only get/check for the source-address if the kernel supports it. */
|
|
Packit Service |
3880ab |
if (saddr.family) {
|
|
Packit Service |
3880ab |
if (get_addr_rta(&saddr, a, saddr.family))
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (f.saddr.family && f.saddr.bitlen >= 0 &&
|
|
Packit Service |
3880ab |
inet_addr_match(&saddr, &f.saddr, f.saddr.bitlen))
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (f.flushb) {
|
|
Packit Service |
3880ab |
struct nlmsghdr *fn;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
TCPM_REQUEST(req2, 128, TCP_METRICS_CMD_DEL, NLM_F_REQUEST);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
addattr_l(&req2.n, sizeof(req2), atype, daddr.data,
|
|
Packit Service |
3880ab |
daddr.bytelen);
|
|
Packit Service |
3880ab |
if (saddr.family)
|
|
Packit Service |
3880ab |
addattr_l(&req2.n, sizeof(req2), stype, saddr.data,
|
|
Packit Service |
3880ab |
saddr.bytelen);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (NLMSG_ALIGN(f.flushp) + req2.n.nlmsg_len > f.flushe) {
|
|
Packit Service |
3880ab |
if (flush_update())
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
fn = (struct nlmsghdr *) (f.flushb + NLMSG_ALIGN(f.flushp));
|
|
Packit Service |
3880ab |
memcpy(fn, &req2.n, req2.n.nlmsg_len);
|
|
Packit Service |
3880ab |
fn->nlmsg_seq = ++grth.seq;
|
|
Packit Service |
3880ab |
f.flushp = (((char *) fn) + req2.n.nlmsg_len) - f.flushb;
|
|
Packit Service |
3880ab |
f.flushed++;
|
|
Packit Service |
3880ab |
if (show_stats < 2)
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
open_json_object(NULL);
|
|
Packit Service |
3880ab |
if (f.cmd & (CMD_DEL | CMD_FLUSH))
|
|
Packit Service |
3880ab |
print_bool(PRINT_ANY, "deleted", "Deleted ", true);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
h = format_host(daddr.family, daddr.bytelen, daddr.data);
|
|
Packit Service |
3880ab |
print_color_string(PRINT_ANY,
|
|
Packit Service |
3880ab |
ifa_family_color(daddr.family),
|
|
Packit Service |
3880ab |
"dst", "%s", h);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_AGE];
|
|
Packit Service |
3880ab |
if (a) {
|
|
Packit Service |
3880ab |
__u64 val = rta_getattr_u64(a);
|
|
Packit Service |
3880ab |
double age = val / 1000.;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_float(PRINT_ANY, "age",
|
|
Packit Service |
3880ab |
" age %.03fsec", age);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_TW_TS_STAMP];
|
|
Packit Service |
3880ab |
if (a) {
|
|
Packit Service |
3880ab |
__s32 val = (__s32) rta_getattr_u32(a);
|
|
Packit Service |
3880ab |
__u32 tsval;
|
|
Packit Service |
3880ab |
char tw_ts[64];
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_TW_TSVAL];
|
|
Packit Service |
3880ab |
tsval = a ? rta_getattr_u32(a) : 0;
|
|
Packit Service |
3880ab |
snprintf(tw_ts, sizeof(tw_ts),
|
|
Packit Service |
3880ab |
"%u/%d", tsval, val);
|
|
Packit Service |
3880ab |
print_string(PRINT_ANY, "tw_ts_stamp",
|
|
Packit Service |
3880ab |
" tw_ts %s ago", tw_ts);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (attrs[TCP_METRICS_ATTR_VALS])
|
|
Packit Service |
3880ab |
print_tcp_metrics(attrs[TCP_METRICS_ATTR_VALS]);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_FOPEN_MSS];
|
|
Packit Service |
3880ab |
if (a) {
|
|
Packit Service |
3880ab |
print_uint(PRINT_ANY, "fopen_miss", " fo_mss %u",
|
|
Packit Service |
3880ab |
rta_getattr_u16(a));
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_FOPEN_SYN_DROPS];
|
|
Packit Service |
3880ab |
if (a) {
|
|
Packit Service |
3880ab |
__u16 syn_loss = rta_getattr_u16(a);
|
|
Packit Service |
3880ab |
double ts;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_FOPEN_SYN_DROP_TS];
|
|
Packit Service |
3880ab |
ts = a ? rta_getattr_u64(a) : 0;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_uint(PRINT_ANY, "fopen_syn_drops",
|
|
Packit Service |
3880ab |
" fo_syn_drops %u", syn_loss);
|
|
Packit Service |
3880ab |
print_float(PRINT_ANY, "fopen_syn_drop_ts",
|
|
Packit Service |
3880ab |
"/%.03fusec ago",
|
|
Packit Service |
3880ab |
ts / 1000000.);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
a = attrs[TCP_METRICS_ATTR_FOPEN_COOKIE];
|
|
Packit Service |
3880ab |
if (a) {
|
|
Packit Service |
3880ab |
char cookie[32 + 1];
|
|
Packit Service |
3880ab |
unsigned char *ptr = RTA_DATA(a);
|
|
Packit Service |
3880ab |
int i, max = RTA_PAYLOAD(a);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (max > 16)
|
|
Packit Service |
3880ab |
max = 16;
|
|
Packit Service |
3880ab |
cookie[0] = 0;
|
|
Packit Service |
3880ab |
for (i = 0; i < max; i++)
|
|
Packit Service |
3880ab |
sprintf(cookie + i + i, "%02x", ptr[i]);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_string(PRINT_ANY, "fo_cookie",
|
|
Packit Service |
3880ab |
" fo_cookie %s", cookie);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (saddr.family) {
|
|
Packit Service |
3880ab |
const char *src;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
src = format_host(saddr.family, saddr.bytelen, saddr.data);
|
|
Packit Service |
3880ab |
print_string(PRINT_ANY, "source",
|
|
Packit Service |
3880ab |
" source %s", src);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_string(PRINT_FP, NULL, "\n", "");
|
|
Packit Service |
3880ab |
close_json_object();
|
|
Packit Service |
3880ab |
fflush(fp);
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static int tcpm_do_cmd(int cmd, int argc, char **argv)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
TCPM_REQUEST(req, 1024, TCP_METRICS_CMD_GET, NLM_F_REQUEST);
|
|
Packit Service |
3880ab |
struct nlmsghdr *answer;
|
|
Packit Service |
3880ab |
int atype = -1, stype = -1;
|
|
Packit Service |
3880ab |
int ack;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
memset(&f, 0, sizeof(f));
|
|
Packit Service |
3880ab |
f.daddr.bitlen = -1;
|
|
Packit Service |
3880ab |
f.daddr.family = preferred_family;
|
|
Packit Service |
3880ab |
f.saddr.bitlen = -1;
|
|
Packit Service |
3880ab |
f.saddr.family = preferred_family;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
switch (preferred_family) {
|
|
Packit Service |
3880ab |
case AF_UNSPEC:
|
|
Packit Service |
3880ab |
case AF_INET:
|
|
Packit Service |
3880ab |
case AF_INET6:
|
|
Packit Service |
3880ab |
break;
|
|
Packit Service |
3880ab |
default:
|
|
Packit Service |
3880ab |
fprintf(stderr, "Unsupported protocol family: %d\n", preferred_family);
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
for (; argc > 0; argc--, argv++) {
|
|
Packit Service |
3880ab |
if (strcmp(*argv, "src") == 0 ||
|
|
Packit Service |
3880ab |
strcmp(*argv, "source") == 0) {
|
|
Packit Service |
3880ab |
char *who = *argv;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
NEXT_ARG();
|
|
Packit Service |
3880ab |
if (matches(*argv, "help") == 0)
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
if (f.saddr.bitlen >= 0)
|
|
Packit Service |
3880ab |
duparg2(who, *argv);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
get_prefix(&f.saddr, *argv, preferred_family);
|
|
Packit Service |
3880ab |
if (f.saddr.bytelen && f.saddr.bytelen * 8 == f.saddr.bitlen) {
|
|
Packit Service |
3880ab |
if (f.saddr.family == AF_INET)
|
|
Packit Service |
3880ab |
stype = TCP_METRICS_ATTR_SADDR_IPV4;
|
|
Packit Service |
3880ab |
else if (f.saddr.family == AF_INET6)
|
|
Packit Service |
3880ab |
stype = TCP_METRICS_ATTR_SADDR_IPV6;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (stype < 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Error: a specific IP address is expected rather than \"%s\"\n",
|
|
Packit Service |
3880ab |
*argv);
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
} else {
|
|
Packit Service |
3880ab |
char *who = "address";
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (strcmp(*argv, "addr") == 0 ||
|
|
Packit Service |
3880ab |
strcmp(*argv, "address") == 0) {
|
|
Packit Service |
3880ab |
who = *argv;
|
|
Packit Service |
3880ab |
NEXT_ARG();
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
if (matches(*argv, "help") == 0)
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
if (f.daddr.bitlen >= 0)
|
|
Packit Service |
3880ab |
duparg2(who, *argv);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
get_prefix(&f.daddr, *argv, preferred_family);
|
|
Packit Service |
3880ab |
if (f.daddr.bytelen && f.daddr.bytelen * 8 == f.daddr.bitlen) {
|
|
Packit Service |
3880ab |
if (f.daddr.family == AF_INET)
|
|
Packit Service |
3880ab |
atype = TCP_METRICS_ATTR_ADDR_IPV4;
|
|
Packit Service |
3880ab |
else if (f.daddr.family == AF_INET6)
|
|
Packit Service |
3880ab |
atype = TCP_METRICS_ATTR_ADDR_IPV6;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
if ((CMD_DEL & cmd) && atype < 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Error: a specific IP address is expected rather than \"%s\"\n",
|
|
Packit Service |
3880ab |
*argv);
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
argc--; argv++;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (cmd == CMD_DEL && atype < 0)
|
|
Packit Service |
3880ab |
missarg("address");
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
/* flush for exact address ? Single del */
|
|
Packit Service |
3880ab |
if (cmd == CMD_FLUSH && atype >= 0)
|
|
Packit Service |
3880ab |
cmd = CMD_DEL;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
/* flush for all addresses ? Single del without address */
|
|
Packit Service |
3880ab |
if (cmd == CMD_FLUSH && f.daddr.bitlen <= 0 &&
|
|
Packit Service |
3880ab |
f.saddr.bitlen <= 0 && preferred_family == AF_UNSPEC) {
|
|
Packit Service |
3880ab |
cmd = CMD_DEL;
|
|
Packit Service |
3880ab |
req.g.cmd = TCP_METRICS_CMD_DEL;
|
|
Packit Service |
3880ab |
ack = 1;
|
|
Packit Service |
3880ab |
} else if (cmd == CMD_DEL) {
|
|
Packit Service |
3880ab |
req.g.cmd = TCP_METRICS_CMD_DEL;
|
|
Packit Service |
3880ab |
ack = 1;
|
|
Packit Service |
3880ab |
} else { /* CMD_FLUSH, CMD_LIST */
|
|
Packit Service |
3880ab |
ack = 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (genl_init_handle(&grth, TCP_METRICS_GENL_NAME, &genl_family))
|
|
Packit Service |
3880ab |
exit(1);
|
|
Packit Service |
3880ab |
req.n.nlmsg_type = genl_family;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (!(cmd & CMD_FLUSH) && (atype >= 0 || (cmd & CMD_DEL))) {
|
|
Packit Service |
3880ab |
if (ack)
|
|
Packit Service |
3880ab |
req.n.nlmsg_flags |= NLM_F_ACK;
|
|
Packit Service |
3880ab |
if (atype >= 0)
|
|
Packit Service |
3880ab |
addattr_l(&req.n, sizeof(req), atype, &f.daddr.data,
|
|
Packit Service |
3880ab |
f.daddr.bytelen);
|
|
Packit Service |
3880ab |
if (stype >= 0)
|
|
Packit Service |
3880ab |
addattr_l(&req.n, sizeof(req), stype, &f.saddr.data,
|
|
Packit Service |
3880ab |
f.saddr.bytelen);
|
|
Packit Service |
3880ab |
} else {
|
|
Packit Service |
3880ab |
req.n.nlmsg_flags |= NLM_F_DUMP;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
f.cmd = cmd;
|
|
Packit Service |
3880ab |
if (cmd & CMD_FLUSH) {
|
|
Packit Service |
3880ab |
int round = 0;
|
|
Packit Service |
3880ab |
char flushb[4096-512];
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
f.flushb = flushb;
|
|
Packit Service |
3880ab |
f.flushp = 0;
|
|
Packit Service |
3880ab |
f.flushe = sizeof(flushb);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
for (;;) {
|
|
Packit Service |
3880ab |
req.n.nlmsg_seq = grth.dump = ++grth.seq;
|
|
Packit Service |
3880ab |
if (rtnl_send(&grth, &req, req.n.nlmsg_len) < 0) {
|
|
Packit Service |
3880ab |
perror("Failed to send flush request");
|
|
Packit Service |
3880ab |
exit(1);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
f.flushed = 0;
|
|
Packit Service |
3880ab |
if (rtnl_dump_filter(&grth, process_msg, stdout) < 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Flush terminated\n");
|
|
Packit Service |
3880ab |
exit(1);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
if (f.flushed == 0) {
|
|
Packit Service |
3880ab |
if (round == 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Nothing to flush.\n");
|
|
Packit Service |
3880ab |
} else if (show_stats)
|
|
Packit Service |
3880ab |
printf("*** Flush is complete after %d round%s ***\n",
|
|
Packit Service |
3880ab |
round, round > 1 ? "s" : "");
|
|
Packit Service |
3880ab |
fflush(stdout);
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
round++;
|
|
Packit Service |
3880ab |
if (flush_update() < 0)
|
|
Packit Service |
3880ab |
exit(1);
|
|
Packit Service |
3880ab |
if (show_stats) {
|
|
Packit Service |
3880ab |
printf("\n*** Round %d, deleting %d entries ***\n",
|
|
Packit Service |
3880ab |
round, f.flushed);
|
|
Packit Service |
3880ab |
fflush(stdout);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (ack) {
|
|
Packit Service |
3880ab |
if (rtnl_talk(&grth, &req.n, NULL) < 0)
|
|
Packit Service |
3880ab |
return -2;
|
|
Packit Service |
3880ab |
} else if (atype >= 0) {
|
|
Packit Service |
3880ab |
if (rtnl_talk(&grth, &req.n, &answer) < 0)
|
|
Packit Service |
3880ab |
return -2;
|
|
Packit Service |
3880ab |
if (process_msg(answer, stdout) < 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Dump terminated\n");
|
|
Packit Service |
3880ab |
exit(1);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
free(answer);
|
|
Packit Service |
3880ab |
} else {
|
|
Packit Service |
3880ab |
req.n.nlmsg_seq = grth.dump = ++grth.seq;
|
|
Packit Service |
3880ab |
if (rtnl_send(&grth, &req, req.n.nlmsg_len) < 0) {
|
|
Packit Service |
3880ab |
perror("Failed to send dump request");
|
|
Packit Service |
3880ab |
exit(1);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
new_json_obj(json);
|
|
Packit Service |
3880ab |
if (rtnl_dump_filter(&grth, process_msg, stdout) < 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Dump terminated\n");
|
|
Packit Service |
3880ab |
exit(1);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
delete_json_obj();
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
int do_tcp_metrics(int argc, char **argv)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
int i;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (argc < 1)
|
|
Packit Service |
3880ab |
return tcpm_do_cmd(CMD_LIST, 0, NULL);
|
|
Packit Service |
3880ab |
for (i = 0; i < ARRAY_SIZE(cmds); i++) {
|
|
Packit Service |
3880ab |
if (matches(argv[0], cmds[i].name) == 0)
|
|
Packit Service |
3880ab |
return tcpm_do_cmd(cmds[i].code, argc-1, argv+1);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
if (matches(argv[0], "help") == 0)
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
fprintf(stderr, "Command \"%s\" is unknown, try \"ip tcp_metrics help\".\n",
|
|
Packit Service |
3880ab |
*argv);
|
|
Packit Service |
3880ab |
exit(-1);
|
|
Packit Service |
3880ab |
}
|