如何在Python中检查网络接口的数据传输？

Question

如何在Python中检查网络接口的数据传输？

5

有一个获取给定网络接口IP的套接字方法：

import socket
import fcntl
import struct

def get_ip_address(ifname):
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    return socket.inet_ntoa(fcntl.ioctl(
        s.fileno(),
        0x8915,  # SIOCGIFADDR
        struct.pack('256s', ifname[:15])
    )[20:24])

以下是返回结果：

>>> get_ip_address('lo')
'127.0.0.1'

>>> get_ip_address('eth0')
'38.113.228.130'

是否有类似的方法返回该接口的网络传输量？我知道可以读取/proc/net/dev，但我更喜欢使用套接字方法。

- Matt Sweeney

1

网络传输？字节计数？网络协议？布线类型？ - Marc B

1

一个套接字并不等同于网络接口。如果你想要 TX/RX 字节/数据包，那就是在链路层。你应该能够通过 NETLINK 套接字从内核获取信息（这就是 ip 命令的工作原理），但更简单的方法是解析 /proc/net/dev 并获取相同的信息。 - JimB

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Mike Pennington · Accepted Answer

通过SNMP来轮询以太网接口统计数据是最好的方法...

看起来你正在使用Linux...如果是这样，在安装了snmpd之后，加载你的snmpd时使用以下选项，在/etc/defaults/snmpd中确保SNMPDOPTS那一行像这样：

SNMPDOPTS='-Lsd -Lf /dev/null -u snmp -I -smux,usmConf,iquery,dlmod,diskio,lmSensors,hr_network,snmpEngine,system_mib,at,interface,ifTable,ipAddressTable,ifXTable,ip,cpu,tcpTable,udpTable,ipSystemStatsTable,ip,snmp_mib,tcp,icmp,udp,proc,memory,snmpNotifyTable,inetNetToMediaTable,ipSystemStatsTable,disk -Lsd -p /var/run/snmpd.pid'
你可能还需要将只读社区更改为public ^见注1，并在/etc/snmp/snmpd.conf中设置监听接口（如果不在回环上）...
假设你有一个功能正常的snmpd，此时，你可以使用以下内容轮询你所关心的接口的ifHCInBytes和ifHCOutBytes ^见注2...

poll_bytes.py:

from SNMP import v2Manager
import time

def poll_eth0(manager=None):
    # NOTE: 2nd arg to get_index should be a valid ifName value
    in_bytes = manager.get_index('ifHCInOctets', 'eth0')
    out_bytes = manager.get_index('ifHCOutOctets', 'eth0')
    return (time.time(), int(in_bytes), int(out_bytes))

# Prep an SNMP manager object...
mgr = v2Manager('localhost')
mgr.index('ifName')
stats = list()
# Insert condition below, instead of True...
while True:
    stats.append(poll_eth0(mgr))
    print poll_eth0(mgr)
    time.sleep(5)

SNMP.py:

from subprocess import Popen, PIPE
import re

class v2Manager(object):
    def __init__(self, addr='127.0.0.1', community='public'):
        self.addr      = addr
        self.community = community
        self._index = dict()

    def bulkwalk(self, oid='ifName'):
        cmd = 'snmpbulkwalk -v 2c -Osq -c %s %s %s' % (self.community,
            self.addr, oid)
        po = Popen(cmd, shell=True, stdout=PIPE, executable='/bin/bash')
        output = po.communicate()[0]
        result = dict()
        for line in re.split(r'\r*\n', output):
            if line.strip()=="":
                continue
            idx, value = re.split(r'\s+', line, 1)
            idx = idx.replace(oid+".", '')
            result[idx] = value
        return result

    def bulkwalk_index(self, oid='ifOutOctets'):
        result = dict()
        if not (self._index==dict()):
            vals = self.bulkwalk(oid=oid)
            for key, val in vals.items():
                idx = self._index.get(key, None)
                if not (idx is None):
                    result[idx] = val
                else:
                    raise ValueError, "Could not find '%s' in the index (%s)" % self.index
        else:
            raise ValueError, "Call the index() method before calling bulkwalk_index()"
        return result

    def get_index(self, oid='ifOutOctets', index=''):
        # This method is horribly inefficient... improvement left as exercise for the reader...
        if index:
            return self.bulkwalk_index().get(index, "<unknown>")
        else:
            raise ValueError, "Please include an index to get"

    def index(self, oid='ifName'):
        self._index = self.bulkwalk(oid=oid)

注释:

SNMP v2c使用明文身份验证。如果您担心安全性/有人嗅探您的流量，请更改您的社区并通过源IP地址限制对Linux机器的查询。完美的世界是修改上面的SNMP.py以使用SNMPv3（加密敏感数据）;大多数人只使用非公共社区并通过源IP限制snmp查询。
ifHCInOctets和ifHCOutOctets为接口传输的字节数提供瞬时值。如果您正在寻找数据传输速率，当然会涉及一些额外的数学计算。