python监控TCP连接数

3212ed

python监控TCP连接数
python监控TCP连接数

先来了解下/proc/net/tcp这个文件，这里记录的是ipv4下所有tcp连接的情况，包括下列数值

sl local_address rem_address st tx_queue rx_queue tr tm->when retrnsmt uid timeout inode

备注：文件中都是用的16进制 我们关注st这列，
状态码对照表
code    状态码    描述
00    ERROR_STATUS   
01    TCP_ESTABLISHED    代表一个打开的连接
02    TCP_SYN_SENT    在发送连接请求后等待匹配的连接请求
03    TCP_SYN_RECV    在收到和发送一个连接请求后等待对方对连接请求的确认
04    TCP_FIN_WAIT1    等待远程TCP连接中断请求，或先前的连接中断请求的确认
05    TCP_FIN_WAIT2    从远程TCP等待连接中断请求
06    TCP_TIME_WAIT    等待足够的时间以确保远程TCP接收到连接中断请求的确认
07    TCP_CLOSE    等待远程TCP对连接中断的确认
08    TCP_CLOSE_WAIT    等待从本地用户发来的连接中断请求
09    TCP_LAST_ACK    等待原来的发向远程TCP的连接中断请求的确认
0A    TCP_LISTEN    侦听来自远方的TCP端口的连接请求
0B    TCP_CLOSING    没有任何连接状态
python代码如下

#!/usr/bin/python
# coding:utf-8
from itertools import dropwhile

sys_st = {
    "00": "ERROR_STATUS",
    "01": "TCP_ESTABLISHED",
    "02": "TCP_SYN_SENT",
    "03": "TCP_SYN_RECV",
    "04": "TCP_FIN_WAIT1",
    "05": "TCP_FIN_WAIT2",
    "06": "TCP_TIME_WAIT",
    "07": "TCP_CLOSE",
    "08": "TCP_CLOSE_WAIT",
    "09": "TCP_LAST_ACK",
    "0A": "TCP_LISTEN",
    "0B": "TCP_CLOSING",
}

tcp_static_dict = {}

with open("/proc/net/tcp") as f:
    for line in dropwhile(lambda line: line.strip().startswith('sl'), f):
        tcp_status_code = line.split()[3]
        if sys_st.has_key(tcp_status_code):
            if tcp_static_dict.get(sys_st[tcp_status_code], None) is None:
                tcp_static_dict[sys_st[tcp_status_code]] = 1
            else:
                tcp_static_dict[sys_st[tcp_status_code]] += 1
print tcp_static_dict