1
#!/usr/bin/env python3
2
"""
3
iperf3自动化网络监控脚本
4
"""
5

6
import subprocess
7
import json
8
import time
9
from datetime import datetime
10
import csv
11

12
def run_iperf_test(server_ip, duration=10, reverse=False):
13
    """执行iperf3测试并返回结果"""
14
    cmd = ['iperf3', '-c', server_ip, '-t', str(duration), '-J']
15
    if reverse:
16
        cmd.append('-R')
17

18
    try:
19
        result = subprocess.run(cmd, capture_output=True, text=True, timeout=duration+5)
20
        if result.returncode == 0:
21
            return json.loads(result.stdout)
22
        else:
23
            print(f"测试失败: {result.stderr}")
24
            return None
25
    except Exception as e:
26
        print(f"执行错误: {e}")
27
        return None
28

29
def monitor_network_quality(server_ip, interval=300, duration=60):
30
    """持续监控网络质量"""
31
    with open('network_monitor.csv', 'w', newline='') as csvfile:
32
        writer = csv.writer(csvfile)
33
        writer.writerow(['时间戳', '下载带宽(Mbps)', '上传带宽(Mbps)',
34
                        '下载重传数', '上传重传数', '下载抖动(ms)', '上传抖动(ms)'])
35

36
        while True:
37
            timestamp = datetime.now().isoformat()
38

39
            # 下载测试
40
            down_result = run_iperf_test(server_ip, duration)
41
            # 上传测试
42
            up_result = run_iperf_test(server_ip, duration, reverse=True)
43

44
            if down_result and up_result:
45
                down_stats = down_result['end']['sum_received']
46
                up_stats = up_result['end']['sum_received']
47

48
                writer.writerow([
49
                    timestamp,
50
                    down_stats['bits_per_second'] / 1e6,
51
                    up_stats['bits_per_second'] / 1e6,
52
                    down_result['end']['sum_sent']['retransmits'],
53
                    up_result['end']['sum_sent']['retransmits'],
54
                    down_stats.get('jitter_ms', 0),
55
                    up_stats.get('jitter_ms', 0)
56
                ])
57
                csvfile.flush()
58

59
                print(f"[{timestamp}] "
60
                      f"下载: {down_stats['bits_per_second']/1e6:.2f} Mbps, "
61
                      f"上传: {up_stats['bits_per_second']/1e6:.2f} Mbps")
62

63
            time.sleep(interval)
64

65
if __name__ == "__main__":
66
    # 使用示例
67
    monitor_network_quality("192.168.1.100", interval=600, duration=30)