从零开始，用Python编写你的网络监控系统

大家好，我是涛哥，本文内容来自 涛哥聊Python ，转载请标原创。

更多Python学习内容：http://ipengtao.com

在现代网络管理中，实时监控网络流量和状态是保证网络正常运行的关键。使用Python编写网络监控工具可以帮助管理员及时发现和解决网络问题。本文将详细介绍如何使用Python编写网络监控工具，包括基本概念、常用库及其应用场景，并提供相应的示例代码。

网络监控的基本概念

网络监控是指通过技术手段对网络设备、服务器、应用程序和网络流量进行实时监控，及时发现和解决潜在问题，保证网络的正常运行。

网络监控主要包括以下几个方面：

1. 网络流量监控：监控网络中的数据包流量，分析流量特征，识别异常流量。
2. 设备状态监控：监控网络设备（如路由器、交换机）的运行状态，及时发现设备故障。
3. 服务可用性监控：监控网络服务（如HTTP、FTP、DNS）的可用性，确保服务正常运行。

Python 网络监控常用库

1. psutil：用于系统和进程监控，提供了获取CPU、内存、磁盘、网络等信息的接口。
2. scapy：强大的网络数据包处理库，用于捕获、解析和生成网络数据包。
3. socket：Python标准库，用于网络通信和服务监控。
4. requests：用于发送HTTP请求，监控Web服务的可用性。

使用 psutil 进行网络监控

psutil 是一个跨平台库，用于检索系统和进程的信息，包括CPU、内存、磁盘、网络等。

安装 psutil

pip install psutil

示例：监控网络接口流量

import psutil
import time

def get_network_info():
    net_io = psutil.net_io_counters()
    return net_io.bytes_sent, net_io.bytes_recv

def monitor_network(interval=1):
    print("监控网络接口流量...")
    while True:
        bytes_sent, bytes_recv = get_network_info()
        print(f"发送字节: {bytes_sent}, 接收字节: {bytes_recv}")
        time.sleep(interval)

# 使用示例
monitor_network(2)

在这个示例中，使用 psutil.net_io_counters() 获取网络接口的流量信息，包括发送和接收的字节数，并每隔2秒打印一次。

使用 scapy 进行网络数据包捕获

scapy 是一个强大的网络数据包处理库，可以用于捕获、解析和生成网络数据包。

安装 scapy

pip install scapy

示例：捕获并解析网络数据包

from scapy.all import sniff, IP

def packet_callback(packet):
    if IP in packet:
        ip_src = packet[IP].src
        ip_dst = packet[IP].dst
        print(f"IP包: {ip_src} -> {ip_dst}")

def start_sniffing():
    print("开始捕获网络数据包...")
    sniff(prn=packet_callback, filter="ip", store=0)

# 使用示例
start_sniffing()

在这个示例中，使用 scapy.sniff() 捕获网络数据包，并在回调函数中解析和打印IP包的源地址和目的地址。

使用 socket 进行服务监控

socket 是Python的标准库，用于网络通信和服务监控。

示例：监控HTTP服务的可用性

import socket

def check_http_service(host, port=80, timeout=5):
    try:
        sock = socket.create_connection((host, port), timeout)
        print(f"HTTP服务 {host}:{port} 可用")
        sock.close()
    except socket.error as e:
        print(f"HTTP服务 {host}:{port} 不可用: {e}")

# 使用示例
check_http_service('www.example.com')

在这个示例中，使用 socket.create_connection() 尝试连接指定的HTTP服务，并根据连接结果判断服务是否可用。

使用 requests 监控Web服务

requests 是一个简单易用的HTTP库，用于发送HTTP请求，监控Web服务的可用性。

安装 requests

pip install requests

示例：监控Web服务的响应时间

import requests
import time

def check_web_service(url, interval=10):
    while True:
        try:
            response = requests.get(url, timeout=5)
            response_time = response.elapsed.total_seconds()
            print(f"服务 {url} 响应时间: {response_time} 秒")
        except requests.RequestException as e:
            print(f"服务 {url} 不可用: {e}")
        time.sleep(interval)

# 使用示例
check_web_service('http://www.example.com')

在这个示例中，使用 requests.get() 发送HTTP请求，并根据响应时间判断服务的可用性。

综合示例：实现一个简单的网络监控工具

下面是一个综合示例，结合 psutil、scapy、socket 和 requests 实现一个简单的网络监控工具。

import psutil
import time
import threading
from scapy.all import sniff, IP
import socket
import requests

# 监控网络接口流量
def monitor_network(interval=1):
    def get_network_info():
        net_io = psutil.net_io_counters()
        return net_io.bytes_sent, net_io.bytes_recv

    while True:
        bytes_sent, bytes_recv = get_network_info()
        print(f"网络流量 - 发送字节: {bytes_sent}, 接收字节: {bytes_recv}")
        time.sleep(interval)

# 捕获并解析网络数据包
def start_sniffing():
    def packet_callback(packet):
        if IP in packet:
            ip_src = packet[IP].src
            ip_dst = packet[IP].dst
            print(f"捕获IP包: {ip_src} -> {ip_dst}")

    print("开始捕获网络数据包...")
    sniff(prn=packet_callback, filter="ip", store=0)

# 监控HTTP服务的可用性
def check_http_service(host, port=80, interval=10):
    def check_service():
        try:
            sock = socket.create_connection((host, port), 5)
            print(f"HTTP服务 {host}:{port} 可用")
            sock.close()
        except socket.error as e:
            print(f"HTTP服务 {host}:{port} 不可用: {e}")

    while True:
        check_service()
        time.sleep(interval)

# 监控Web服务的响应时间
def check_web_service(url, interval=10):
    def check_service():
        try:
            response = requests.get(url, timeout=5)
            response_time = response.elapsed.total_seconds()
            print(f"Web服务 {url} 响应时间: {response_time} 秒")
        except requests.RequestException as e:
            print(f"Web服务 {url} 不可用: {e}")

    while True:
        check_service()
        time.sleep(interval)

# 启动监控工具
if __name__ == "__main__":
    # 启动网络流量监控
    threading.Thread(target=monitor_network, args=(2,)).start()

    # 启动网络数据包捕获
    threading.Thread(target=start_sniffing).start()

    # 启动HTTP服务监控
    threading.Thread(target=check_http_service, args=('www.example.com', 80, 10)).start()

    # 启动Web服务监控
    threading.Thread(target=check_web_service, args=('http://www.example.com', 10)).start()

在这个综合示例中，使用多线程并行运行网络流量监控、网络数据包捕获、HTTP服务监控和Web服务监控，实现了一个简单的网络监控工具。

总结

本文详细介绍了如何使用Python编写网络监控工具，涵盖网络流量监控、数据包捕获、服务可用性监控等方面。通过使用psutil、scapy、socket和requests等库，开发者可以实现多种网络监控功能，实时监控网络状态，及时发现和解决网络问题。文中提供了丰富的示例代码，展示了如何实现网络接口流量监控、网络数据包捕获、HTTP服务监控和Web服务监控。希望本文对大家理解和应用Python进行网络监控有所帮助。