4

我想在我的电脑上监控上传和下载速度。一个名为 conky 的程序已经在 conky conf 中使用了以下内容:

Connection quality: $alignr ${wireless_link_qual_perc wlan0}%
${downspeedgraph wlan0}
DLS:${downspeed wlan0} kb/s $alignr total: ${totaldown wlan0}

它在我浏览时几乎实时地向我显示速度。我希望能够使用 python 访问相同的信息。

4

4 回答 4

13

您可以根据设备的 rx_bytes 和 tx_bytes 自己计算速度,并在一段时间内轮询这些值

这是我使用 Python 3 共同破解的一个非常简单的解决方案

#!/usr/bin/python3

import time

def get_bytes(t, iface='wlan0'):
    with open('/sys/class/net/' + iface + '/statistics/' + t + '_bytes', 'r') as f:
        data = f.read();
    return int(data)

if __name__ == '__main__':
    (tx_prev, rx_prev) = (0, 0)

    while(True):
        tx = get_bytes('tx')
        rx = get_bytes('rx')

        if tx_prev > 0:
            tx_speed = tx - tx_prev
            print('TX: ', tx_speed, 'bps')

        if rx_prev > 0:
            rx_speed = rx - rx_prev
            print('RX: ', rx_speed, 'bps')

        time.sleep(1)

        tx_prev = tx
        rx_prev = rx
于 2014-11-10T21:13:50.597 回答
5

我会研究 Python 的psutil 模块

这是一个简短的片段,它打印出自您启动机器以来发送的字节数:

import psutil
iostat = psutil.net_io_counters(pernic=False)
print iostat[0] #upload only

您可以轻松扩展它以以恒定间隔获取值并区分这两个值以确定在该时间段内发送和/或接收的字节数。

于 2014-11-12T01:59:10.273 回答
2

为了获得特定于接口的统计信息,已经提出的方法可以正常工作。

我将尝试为您的第二个请求提出解决方案:

知道哪个程序正在使用该带宽也很有帮助,但到目前为止我还没有看到任何可以做到这一点的东西。

如前所述,nethogs打印进程特定的统计信息。据我所知,没有简单的方法可以访问这些值,/proc因此我将解释 nethogs 如何实现这一点。

考虑一个具有 pid PID 的进程,nethogs 首先检索该进程打开的所有套接字的列表,列出 /proc/PID/fd 的内容:

➜  ~ [1] at 23:59:31 [Sat 15] $ ls -la /proc/21841/fd
total 0
dr-x------ 2 marco marco  0 Nov 15 23:41 .
dr-xr-xr-x 8 marco marco  0 Nov 15 23:41 ..
lrwx------ 1 marco marco 64 Nov 15 23:42 0 -> /dev/pts/15
l-wx------ 1 marco marco 64 Nov 15 23:42 1 -> /dev/null
lrwx------ 1 marco marco 64 Nov 15 23:41 2 -> /dev/pts/15
lrwx------ 1 marco marco 64 Nov 15 23:42 4 -> socket:[177472]

这里我们只有一个套接字,177472 是 inode 号。我们会在这里找到所有类型的套接字:TCPv4、TCPv6、UDP、netlink。在这种情况下,我将只考虑 TCPv4。

一旦收集了所有的 inode 编号,就会为每个套接字分配一个唯一标识符,即(IP_SRC, PORT_SRC, IP_DEST, PORT_DEST). 当然,与 PID 的配对也会被存储。(IP_SRC, PORT_SRC, IP_DEST, PORT_DEST)可以读取元组来检索/proc/net/tcp(对于 TCPv4)。在这种情况下:

➜  ~ [1] at 0:06:05 [Sun 16] $ cat /proc/net/tcp | grep 177472
  sl  local_address rem_address   st tx_queue rx_queue tr tm->when retrnsmt   uid  timeout inode
  38: 1D00A8C0:1F90 0400A8C0:A093 01 00000000:00000000 00:00000000 00000000  1000        0 177472 1 f6fae080 21 4 0 10 5 

地址表示为 IP:PORT,IP 表示为 4 字节的 LE 编号。然后,您可以构建一个key->value结构,其中键是(IP_SRC, PORT_SRC, IP_DEST, PORT_DEST)PID,值是 PID。

此时,nethogs 使用 libpcap 捕获所有网络流量。当它检测到 TCP 数据包时,它会尝试将元组(IP_SRC_PACKET, PORT_SRC_PACKET, IP_DEST_PACKET, PORT_DEST_PACKET)与表内的所有连接进行匹配。当然它必须尝试交换 SRC 和 DEST,数据包可能是传入 (DL) 或传出 (UL)。如果它处理了一个连接,它会检索该连接所属进程的 PID,并将 TCP 有效负载的大小添加到 TX 或 RX 计数器。通过捕获每个数据包时更新的字节数,可以轻松计算每个进程的传输速度。

理论上,这可以在 python 中使用 pypcap 实现,尽管它需要一些工作。我试图实现一些东西,但它非常缓慢,而且需要更多的工作才能使用。我只监控一个 PID,一个连接,没有更新连接表,但是超过 3MB/s 我的脚本无法跟上网络流量。

如您所见,这并非微不足道。解析现有工具的输出可能会带来更好的解决方案,并且可能会为您节省大量工作。

于 2014-11-15T23:54:59.533 回答
1

你可以做一些狡猾的事情,比如调用conky -i 1并解析输出:

import subprocess
conky=subprocess.check_output("conky -i 1", shell=True)
lines=conky.splitlines()
print lines[11].split()[1::3]

导致:

['1234B', '5678B']

我的配置看起来像:

${scroll 16 $nodename - $sysname $kernel on $machine | }
Uptime: $uptime
Frequency (in MHz): $freq
Frequency (in GHz): $freq_g
RAM Usage: $mem/$memmax - $memperc% ${membar 4}
Swap Usage: $swap/$swapmax - $swapperc% ${swapbar 4}
CPU Usage: $cpu% ${cpubar 4}
Processes: $processes  Running: $running_processes
File systems:
 / ${fs_used /}/${fs_size /} ${fs_bar 6 /}
Networking:
Up: ${upspeed eth0}  - Down: ${downspeed eth0}
Name              PID   CPU%   MEM%
 ${top name 1} ${top pid 1} ${top cpu 1} ${top mem 1}
 ${top name 2} ${top pid 2} ${top cpu 2} ${top mem 2}
 ${top name 3} ${top pid 3} ${top cpu 3} ${top mem 3}
 ${top name 4} ${top pid 4} ${top cpu 4} ${top mem 4}
于 2014-11-13T01:54:18.913 回答