1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
#!/usr/bin/python
#
# Copyright 2017 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Generic netlink interface to TCP metrics."""
from socket import * # pylint: disable=wildcard-import
import struct
import binascii
import cstruct
import genetlink
import net_test
import netlink
### TCP metrics constants. See include/uapi/linux/tcp_metrics.h.
# Family name and version
TCP_METRICS_GENL_NAME = "tcp_metrics"
TCP_METRICS_GENL_VERSION = 1
# Message types.
TCP_METRICS_CMD_GET = 1
TCP_METRICS_CMD_DEL = 2
# Attributes.
TCP_METRICS_ATTR_UNSPEC = 0
TCP_METRICS_ATTR_ADDR_IPV4 = 1
TCP_METRICS_ATTR_ADDR_IPV6 = 2
TCP_METRICS_ATTR_AGE = 3
TCP_METRICS_ATTR_TW_TSVAL = 4
TCP_METRICS_ATTR_TW_TS_STAMP = 5
TCP_METRICS_ATTR_VALS = 6
TCP_METRICS_ATTR_FOPEN_MSS = 7
TCP_METRICS_ATTR_FOPEN_SYN_DROPS = 8
TCP_METRICS_ATTR_FOPEN_SYN_DROP_TS = 9
TCP_METRICS_ATTR_FOPEN_COOKIE = 10
TCP_METRICS_ATTR_SADDR_IPV4 = 11
TCP_METRICS_ATTR_SADDR_IPV6 = 12
TCP_METRICS_ATTR_PAD = 13
class TcpMetrics(genetlink.GenericNetlink):
NL_DEBUG = ["ALL"]
def __init__(self):
super(TcpMetrics, self).__init__()
# Generic netlink family IDs are dynamically assigned. Find ours.
ctrl = genetlink.GenericNetlinkControl()
self.family = ctrl.GetFamily(TCP_METRICS_GENL_NAME)
def _Decode(self, command, msg, nla_type, nla_data, nested):
"""Decodes TCP metrics netlink attributes to human-readable format."""
name = self._GetConstantName(__name__, nla_type, "TCP_METRICS_ATTR_")
if name in ["TCP_METRICS_ATTR_ADDR_IPV4", "TCP_METRICS_ATTR_SADDR_IPV4"]:
data = inet_ntop(AF_INET, nla_data)
elif name in ["TCP_METRICS_ATTR_ADDR_IPV6", "TCP_METRICS_ATTR_SADDR_IPV6"]:
data = inet_ntop(AF_INET6, nla_data)
elif name in ["TCP_METRICS_ATTR_AGE"]:
data = struct.unpack("=Q", nla_data)[0]
elif name in ["TCP_METRICS_ATTR_TW_TSVAL", "TCP_METRICS_ATTR_TW_TS_STAMP"]:
data = struct.unpack("=I", nla_data)[0]
elif name == "TCP_METRICS_ATTR_FOPEN_MSS":
data = struct.unpack("=H", nla_data)[0]
elif name == "TCP_METRICS_ATTR_FOPEN_COOKIE":
data = nla_data
else:
data = binascii.hexlify(nla_data)
return name, data
def MaybeDebugCommand(self, command, unused_flags, data):
if "ALL" not in self.NL_DEBUG and command not in self.NL_DEBUG:
return
parsed = self._ParseNLMsg(data, genetlink.Genlmsghdr)
def _NlAttrSaddr(self, address):
if ":" not in address:
family = AF_INET
nla_type = TCP_METRICS_ATTR_SADDR_IPV4
else:
family = AF_INET6
nla_type = TCP_METRICS_ATTR_SADDR_IPV6
return self._NlAttrIPAddress(nla_type, family, address)
def _NlAttrTcpMetricsAddr(self, address, is_source):
version = net_test.GetAddressVersion(address)
family = net_test.GetAddressFamily(version)
if version == 5:
address = address.replace("::ffff:", "")
nla_name = "TCP_METRICS_ATTR_%s_IPV%d" % (
"SADDR" if is_source else "ADDR", version)
nla_type = globals()[nla_name]
return self._NlAttrIPAddress(nla_type, family, address)
def _NlAttrAddr(self, address):
return self._NlAttrTcpMetricsAddr(address, False)
def _NlAttrSaddr(self, address):
return self._NlAttrTcpMetricsAddr(address, True)
def DumpMetrics(self):
"""Dumps all TCP metrics."""
return self._Dump(self.family, TCP_METRICS_CMD_GET, 1)
def GetMetrics(self, saddr, daddr):
"""Returns TCP metrics for the specified src/dst pair."""
data = self._NlAttrSaddr(saddr) + self._NlAttrAddr(daddr)
self._SendCommand(self.family, TCP_METRICS_CMD_GET, 1, data,
netlink.NLM_F_REQUEST)
hdr, attrs = self._GetMsg(genetlink.Genlmsghdr)
return attrs
def DelMetrics(self, saddr, daddr):
"""Deletes TCP metrics for the specified src/dst pair."""
data = self._NlAttrSaddr(saddr) + self._NlAttrAddr(daddr)
self._SendCommand(self.family, TCP_METRICS_CMD_DEL, 1, data,
netlink.NLM_F_REQUEST)
if __name__ == "__main__":
t = TcpMetrics()
print(t.DumpMetrics())
|
