]> git.proxmox.com Git - mirror_iproute2.git/blobdiff - ip/tcp_metrics.c
libnetlink: add size argument to rtnl_talk
[mirror_iproute2.git] / ip / tcp_metrics.c
index 34e1d8e29d01363c7dd2fc98cc5ec435532c5d2a..bdc503ef831ccee7344c402745093898379e5b58 100644 (file)
@@ -74,7 +74,8 @@ static struct
        int flushp;
        int flushe;
        int cmd;
-       inet_prefix addr;
+       inet_prefix daddr;
+       inet_prefix saddr;
 } f;
 
 static int flush_update(void)
@@ -95,8 +96,8 @@ static int process_msg(const struct sockaddr_nl *who, struct nlmsghdr *n,
        struct rtattr *attrs[TCP_METRICS_ATTR_MAX + 1], *a;
        int len = n->nlmsg_len;
        char abuf[256];
-       inet_prefix addr;
-       int family, i, atype;
+       inet_prefix daddr, saddr;
+       int family, i, atype, stype, dlen = 0, slen = 0;
 
        if (n->nlmsg_type != genl_family)
                return -1;
@@ -114,35 +115,66 @@ static int process_msg(const struct sockaddr_nl *who, struct nlmsghdr *n,
 
        a = attrs[TCP_METRICS_ATTR_ADDR_IPV4];
        if (a) {
-               if (f.addr.family && f.addr.family != AF_INET)
+               if (f.daddr.family && f.daddr.family != AF_INET)
                        return 0;
-               memcpy(&addr.data, RTA_DATA(a), 4);
-               addr.bytelen = 4;
+               memcpy(&daddr.data, RTA_DATA(a), 4);
+               daddr.bytelen = 4;
                family = AF_INET;
                atype = TCP_METRICS_ATTR_ADDR_IPV4;
+               dlen = RTA_PAYLOAD(a);
        } else {
                a = attrs[TCP_METRICS_ATTR_ADDR_IPV6];
                if (a) {
-                       if (f.addr.family && f.addr.family != AF_INET6)
+                       if (f.daddr.family && f.daddr.family != AF_INET6)
                                return 0;
-                       memcpy(&addr.data, RTA_DATA(a), 16);
-                       addr.bytelen = 16;
+                       memcpy(&daddr.data, RTA_DATA(a), 16);
+                       daddr.bytelen = 16;
                        family = AF_INET6;
                        atype = TCP_METRICS_ATTR_ADDR_IPV6;
+                       dlen = RTA_PAYLOAD(a);
                } else
                        return 0;
        }
 
-       if (f.addr.family && f.addr.bitlen >= 0 &&
-           inet_addr_match(&addr, &f.addr, f.addr.bitlen))
+       a = attrs[TCP_METRICS_ATTR_SADDR_IPV4];
+       if (a) {
+               if (f.saddr.family && f.saddr.family != AF_INET)
+                       return 0;
+               memcpy(&saddr.data, RTA_DATA(a), 4);
+               saddr.bytelen = 4;
+               stype = TCP_METRICS_ATTR_SADDR_IPV4;
+               slen = RTA_PAYLOAD(a);
+       } else {
+               a = attrs[TCP_METRICS_ATTR_SADDR_IPV6];
+               if (a) {
+                       if (f.saddr.family && f.saddr.family != AF_INET6)
+                               return 0;
+                       memcpy(&saddr.data, RTA_DATA(a), 16);
+                       saddr.bytelen = 16;
+                       stype = TCP_METRICS_ATTR_SADDR_IPV6;
+                       slen = RTA_PAYLOAD(a);
+               }
+       }
+
+       if (f.daddr.family && f.daddr.bitlen >= 0 &&
+           inet_addr_match(&daddr, &f.daddr, f.daddr.bitlen))
+              return 0;
+       /* Only check for the source-address if the kernel supports it,
+        * meaning slen != 0.
+        */
+       if (slen && f.saddr.family && f.saddr.bitlen >= 0 &&
+           inet_addr_match(&saddr, &f.saddr, f.saddr.bitlen))
                return 0;
 
        if (f.flushb) {
                struct nlmsghdr *fn;
                TCPM_REQUEST(req2, 128, TCP_METRICS_CMD_DEL, NLM_F_REQUEST);
 
-               addattr_l(&req2.n, sizeof(req2), atype, &addr.data,
-                         addr.bytelen);
+               addattr_l(&req2.n, sizeof(req2), atype, &daddr.data,
+                         daddr.bytelen);
+               if (slen)
+                       addattr_l(&req2.n, sizeof(req2), stype, &saddr.data,
+                                 saddr.bytelen);
 
                if (NLMSG_ALIGN(f.flushp) + req2.n.nlmsg_len > f.flushe) {
                        if (flush_update())
@@ -161,12 +193,11 @@ static int process_msg(const struct sockaddr_nl *who, struct nlmsghdr *n,
                fprintf(fp, "Deleted ");
 
        fprintf(fp, "%s",
-               format_host(family, RTA_PAYLOAD(a), &addr.data,
-                           abuf, sizeof(abuf)));
+               format_host(family, dlen, &daddr.data, abuf, sizeof(abuf)));
 
        a = attrs[TCP_METRICS_ATTR_AGE];
        if (a) {
-               __u64 val = rta_getattr_u64(a);
+               unsigned long long val = rta_getattr_u64(a);
 
                fprintf(fp, " age %llu.%03llusec",
                        val / 1000, val % 1000);
@@ -185,35 +216,53 @@ static int process_msg(const struct sockaddr_nl *who, struct nlmsghdr *n,
        a = attrs[TCP_METRICS_ATTR_VALS];
        if (a) {
                struct rtattr *m[TCP_METRIC_MAX + 1 + 1];
+               unsigned long rtt = 0, rttvar = 0;
 
                parse_rtattr_nested(m, TCP_METRIC_MAX + 1, a);
 
                for (i = 0; i < TCP_METRIC_MAX + 1; i++) {
-                       __u32 val;
+                       unsigned long val;
 
                        a = m[i + 1];
                        if (!a)
                                continue;
-                       if (metric_name[i])
-                               fprintf(fp, " %s ", metric_name[i]);
-                       else
-                               fprintf(fp, " metric_%d ", i);
+                       if (i != TCP_METRIC_RTT &&
+                           i != TCP_METRIC_RTT_US &&
+                           i != TCP_METRIC_RTTVAR &&
+                           i != TCP_METRIC_RTTVAR_US) {
+                               if (metric_name[i])
+                                       fprintf(fp, " %s ", metric_name[i]);
+                               else
+                                       fprintf(fp, " metric_%d ", i);
+                       }
                        val = rta_getattr_u32(a);
                        switch (i) {
                        case TCP_METRIC_RTT:
-                               fprintf(fp, "%lluus", (val * 1000ULL) >> 3);
+                               if (!rtt)
+                                       rtt = (val * 1000UL) >> 3;
                                break;
                        case TCP_METRIC_RTTVAR:
-                               fprintf(fp, "%lluus", (val * 1000ULL) >> 2);
+                               if (!rttvar)
+                                       rttvar = (val * 1000UL) >> 2;
+                               break;
+                       case TCP_METRIC_RTT_US:
+                               rtt = val >> 3;
+                               break;
+                       case TCP_METRIC_RTTVAR_US:
+                               rttvar = val >> 2;
                                break;
                        case TCP_METRIC_SSTHRESH:
                        case TCP_METRIC_CWND:
                        case TCP_METRIC_REORDERING:
                        default:
-                               fprintf(fp, "%u", val);
+                               fprintf(fp, "%lu", val);
                                break;
                        }
                }
+               if (rtt)
+                       fprintf(fp, " rtt %luus", rtt);
+               if (rttvar)
+                       fprintf(fp, " rttvar %luus", rttvar);
        }
 
        a = attrs[TCP_METRICS_ATTR_FOPEN_MSS];
@@ -223,7 +272,7 @@ static int process_msg(const struct sockaddr_nl *who, struct nlmsghdr *n,
        a = attrs[TCP_METRICS_ATTR_FOPEN_SYN_DROPS];
        if (a) {
                __u16 syn_loss = rta_getattr_u16(a);
-               __u64 ts;
+               unsigned long long ts;
 
                a = attrs[TCP_METRICS_ATTR_FOPEN_SYN_DROP_TS];
                ts = a ? rta_getattr_u64(a) : 0;
@@ -246,6 +295,12 @@ static int process_msg(const struct sockaddr_nl *who, struct nlmsghdr *n,
                fprintf(fp, " fo_cookie %s", cookie);
        }
 
+       if (slen) {
+               fprintf(fp, " source %s",
+                       format_host(family, slen, &saddr.data, abuf,
+                                   sizeof(abuf)));
+       }
+
        fprintf(fp, "\n");
 
        fflush(fp);
@@ -255,12 +310,14 @@ static int process_msg(const struct sockaddr_nl *who, struct nlmsghdr *n,
 static int tcpm_do_cmd(int cmd, int argc, char **argv)
 {
        TCPM_REQUEST(req, 1024, TCP_METRICS_CMD_GET, NLM_F_REQUEST);
-       int atype = -1;
+       int atype = -1, stype = -1;
        int ack;
 
        memset(&f, 0, sizeof(f));
-       f.addr.bitlen = -1;
-       f.addr.family = preferred_family;
+       f.daddr.bitlen = -1;
+       f.daddr.family = preferred_family;
+       f.saddr.bitlen = -1;
+       f.saddr.family = preferred_family;
 
        switch (preferred_family) {
        case AF_UNSPEC:
@@ -268,36 +325,58 @@ static int tcpm_do_cmd(int cmd, int argc, char **argv)
        case AF_INET6:
                break;
        default:
-               fprintf(stderr, "Unsupported family:%d\n", preferred_family);
+               fprintf(stderr, "Unsupported protocol family: %d\n", preferred_family);
                return -1;
        }
 
        for (; argc > 0; argc--, argv++) {
-               char *who = "address";
-
-               if (strcmp(*argv, "addr") == 0 ||
-                   strcmp(*argv, "address") == 0) {
-                       who = *argv;
+               if (strcmp(*argv, "src") == 0 ||
+                   strcmp(*argv, "source") == 0) {
+                       char *who = *argv;
                        NEXT_ARG();
-               }
-               if (matches(*argv, "help") == 0)
-                       usage();
-               if (f.addr.bitlen >= 0)
-                       duparg2(who, *argv);
-
-               get_prefix(&f.addr, *argv, preferred_family);
-               if (f.addr.bytelen && f.addr.bytelen * 8 == f.addr.bitlen) {
-                       if (f.addr.family == AF_INET)
-                               atype = TCP_METRICS_ATTR_ADDR_IPV4;
-                       else if (f.addr.family == AF_INET6)
-                               atype = TCP_METRICS_ATTR_ADDR_IPV6;
-               }
-               if ((CMD_DEL & cmd) && atype < 0) {
-                       fprintf(stderr, "Error: a specific IP address is expected rather than \"%s\"\n",
-                               *argv);
-                       return -1;
-               }
+                       if (matches(*argv, "help") == 0)
+                               usage();
+                       if (f.saddr.bitlen >= 0)
+                               duparg2(who, *argv);
+
+                       get_prefix(&f.saddr, *argv, preferred_family);
+                       if (f.saddr.bytelen && f.saddr.bytelen * 8 == f.saddr.bitlen) {
+                               if (f.saddr.family == AF_INET)
+                                       stype = TCP_METRICS_ATTR_SADDR_IPV4;
+                               else if (f.saddr.family == AF_INET6)
+                                       stype = TCP_METRICS_ATTR_SADDR_IPV6;
+                       }
 
+                       if (stype < 0) {
+                               fprintf(stderr, "Error: a specific IP address is expected rather than \"%s\"\n",
+                                       *argv);
+                               return -1;
+                       }
+               } else {
+                       char *who = "address";
+                       if (strcmp(*argv, "addr") == 0 ||
+                           strcmp(*argv, "address") == 0) {
+                               who = *argv;
+                               NEXT_ARG();
+                       }
+                       if (matches(*argv, "help") == 0)
+                               usage();
+                       if (f.daddr.bitlen >= 0)
+                               duparg2(who, *argv);
+
+                       get_prefix(&f.daddr, *argv, preferred_family);
+                       if (f.daddr.bytelen && f.daddr.bytelen * 8 == f.daddr.bitlen) {
+                               if (f.daddr.family == AF_INET)
+                                       atype = TCP_METRICS_ATTR_ADDR_IPV4;
+                               else if (f.daddr.family == AF_INET6)
+                                       atype = TCP_METRICS_ATTR_ADDR_IPV6;
+                       }
+                       if ((CMD_DEL & cmd) && atype < 0) {
+                               fprintf(stderr, "Error: a specific IP address is expected rather than \"%s\"\n",
+                                       *argv);
+                               return -1;
+                       }
+               }
                argc--; argv++;
        }
 
@@ -309,8 +388,8 @@ static int tcpm_do_cmd(int cmd, int argc, char **argv)
                cmd = CMD_DEL;
 
        /* flush for all addresses ? Single del without address */
-       if (cmd == CMD_FLUSH && f.addr.bitlen <= 0 &&
-           preferred_family == AF_UNSPEC) {
+       if (cmd == CMD_FLUSH && f.daddr.bitlen <= 0 &&
+           f.saddr.bitlen <= 0 && preferred_family == AF_UNSPEC) {
                cmd = CMD_DEL;
                req.g.cmd = TCP_METRICS_CMD_DEL;
                ack = 1;
@@ -337,8 +416,11 @@ static int tcpm_do_cmd(int cmd, int argc, char **argv)
                if (ack)
                        req.n.nlmsg_flags |= NLM_F_ACK;
                if (atype >= 0)
-                       addattr_l(&req.n, sizeof(req), atype, &f.addr.data,
-                                 f.addr.bytelen);
+                       addattr_l(&req.n, sizeof(req), atype, &f.daddr.data,
+                                 f.daddr.bytelen);
+               if (stype >= 0)
+                       addattr_l(&req.n, sizeof(req), stype, &f.saddr.data,
+                                 f.saddr.bytelen);
        } else {
                req.n.nlmsg_flags |= NLM_F_DUMP;
        }
@@ -385,10 +467,10 @@ static int tcpm_do_cmd(int cmd, int argc, char **argv)
        }
 
        if (ack) {
-               if (rtnl_talk(&grth, &req.n, 0, 0, NULL) < 0)
+               if (rtnl_talk(&grth, &req.n, NULL, 0) < 0)
                        return -2;
        } else if (atype >= 0) {
-               if (rtnl_talk(&grth, &req.n, 0, 0, &req.n) < 0)
+               if (rtnl_talk(&grth, &req.n, &req.n, sizeof(req)) < 0)
                        return -2;
                if (process_msg(NULL, &req.n, stdout) < 0) {
                        fprintf(stderr, "Dump terminated\n");