python如何实时获取tcpdump输出
作者:文渊 时间:2021-05-05 14:41:36
一、背景
今天有个小需求,要确认客户端有没有往服务端发送udp包,但为了减轻工作量,不想每次到机器上手动执行tcpdump抓包命令。
于是就写了个脚本来释放人力。
二、代码实现
整个脚本我还加了一些其他功能:时间戳、发送端IP提取,数据包分析,数据持久化等。这里都先去掉,仅记录下简单的实时获取tcpdump输出功能。
代码如下:
# -*- coding: utf-8 -*-
# !/usr/bin/env python
# sudo tcpdump -tt -l -nn -c 5 -i enp4s0 udp port 514 or 51414
import subprocess
cmd = ['sudo', 'tcpdump', '-tt', '-l', '-nn', '-c', '5', '-i', 'enp4s0', 'udp', 'port', '514', 'or', '51414']
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE)
while True:
line = proc.stdout.readline()
line = line.strip()
if not line:
print('tcpdump finished...')
break
print(line)
输出如下(实时):
wenyuanblog@localhost:/home/test/script# python tcpdump_udp.py
tcpdump: verbose output suppressed, use -v or -vv for full protocol decode
listening on enp4s0, link-type EN10MB (Ethernet), capture size 262144 bytes
1499774951.124178 IP 192.168.10.210.41974 > 192.168.10.251.514: UDP, length 139
1499774953.125664 IP 192.168.10.210.54995 > 192.168.10.251.51414: UDP, length 139
1499774956.128498 IP 192.168.10.210.56748 > 192.168.10.251.514: UDP, length 139
1499774958.129918 IP 192.168.10.210.53883 > 192.168.10.251.51414: UDP, length 139
1499774961.132921 IP 192.168.10.210.58803 > 192.168.10.251.514: UDP, length 139
5 packets captured
6 packets received by filter
0 packets dropped by kernel
tcpdump finished...
以上代码相当于手动执行了 sudo tcpdump -tt -l -nn -c 5 -i enp4s0 udp port 514 or 51414
这条命令。
注意参数-l很重要(行显)。
三、代码实现(更新)
上面的代码能实现tcpdump的功能,但是有一个问题:没有做超时保护。即当程序执行时间过长时kill该进程(这里使用ctrl+c的方式)。
要实现这个功能有很多种方案,例如定时器+多线程等,这里仅演示一种方案,代码如下:
# -*- coding: utf-8 -*-
# !/usr/bin/env python
# sudo tcpdump -tt -l -nn -c 50 -i enp4s0 udp port 514 or 51414
import subprocess
import signal
import time
import os
import re
import json
class CmdServer:
def __init__(self, cmd, timeout=120):
'''
:param cmd: 执行命令(列表形式)
:param timeout: 任务超时时间(seconds,进程运行超过该时间,kill该进程)
:param taskname: 任务名称(根据该任务名称记录命令输出信息)
'''
self.cmd = cmd
self.timeout = timeout
self.base_path = reduce(lambda x, y: os.path.dirname(x), range(1), os.path.abspath(__file__))
self.output_path = os.path.join(self.base_path, 'data.json')
self.udp_flow_list = []
self.begin_time = int(time.time())
# 执行tcpdump任务
def run(self):
if os.path.exists(self.output_path):
with open(self.output_path, 'r') as f:
self.udp_flow_list = json.load(f)
proc = subprocess.Popen(self.cmd, stdout=subprocess.PIPE)
stdout = ''
while proc.poll() == None:
current_time = int(time.time())
if current_time - self.begin_time >= self.timeout:
print('tcpdump timeout...')
proc.send_signal(signal.SIGINT)
stdout = proc.stdout.read()
if proc.poll() is not None and not stdout:
print('tcpdump finished...')
stdout = proc.stdout.read()
stdout_list = stdout.split('\n')
if stdout_list:
self._merge_data(stdout_list)
self._save_data()
# 数据合并(新增/更新)
def _merge_data(self, stdout_list):
for line in stdout_list:
line = line.strip()
if not line:
continue
timestamp = int(float(line.split('IP')[0].strip())) * 1000
# 源
src_ip_port_list = re.findall(r'IP(.+?)>', line)
if not src_ip_port_list:
continue
src_ip_port_str = src_ip_port_list[0].strip()
src_ip = '.'.join(src_ip_port_str.split('.')[0:4])
# 目的
dst_ip_port_list = re.findall(r'>(.+?):', line)
if not dst_ip_port_list:
continue
dst_ip_port_str = dst_ip_port_list[0].strip()
dst_port = dst_ip_port_str.split('.')[-1]
# 新增/更新latest_timestamp
src_item = filter(lambda x: src_ip == x['src_ip'], self.udp_flow_list)
if src_item:
src_item[0]['dst_port'] = dst_port
src_item[0]['latest_timestamp'] = timestamp
else:
self.udp_flow_list.append(dict(
src_ip=src_ip,
dst_port=dst_port,
latest_timestamp=timestamp
))
# 保存数据
def _save_data(self):
# 写入文件
with open(self.output_path, 'w') as f:
json.dump(self.udp_flow_list, f, encoding="utf-8", ensure_ascii=False)
if __name__ == '__main__':
cmd = ['sudo', 'tcpdump', '-tt', '-l', '-nn', '-c', '5', '-i', 'enp4s0', 'udp', 'port', '514', 'or', '51414']
cmd_server = CmdServer(cmd, 10)
cmd_server.run()
四、总结
比较简单,仅仅是记录下。
来源:https://www.wenyuanblog.com/blogs/python-realtime-tcpdump.html
标签:python,tcpdump
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
利用Python实现Windows下的鼠标键盘模拟的实例代码
2023-06-22 04:37:31
![](https://img.aspxhome.com/file/2023/9/79659_0s.jpg)
python闭包、深浅拷贝、垃圾回收、with语句知识点汇总
2023-09-11 13:20:31
![](https://img.aspxhome.com/file/2023/9/81739_0s.png)
PHP convert_cyr_string()函数讲解
2023-06-05 00:43:42
![](https://img.aspxhome.com/file/2023/4/55394_0s.png)
采用XMLHTTP编写一个天气预报的程序
2007-10-15 12:35:00
Python实现合并两个有序链表的方法示例
2023-04-02 22:20:46
![](https://img.aspxhome.com/file/2023/0/75250_0s.png)
Python中使用subprocess库创建附加进程
2022-01-01 06:30:25
![](https://img.aspxhome.com/file/2023/8/85138_0s.png)
详解python--模拟轮盘抽奖游戏
2023-07-25 15:35:14
![](https://img.aspxhome.com/file/2023/5/71395_0s.png)
Python 正则表达式入门(初级篇)
2021-12-03 23:41:12
基于Python实现一个简易的数据管理系统
2023-08-14 11:49:38
![](https://img.aspxhome.com/file/2023/4/92274_0s.png)
flask使用session保存登录状态及拦截未登录请求代码
2021-02-22 12:00:57
Python探索之pLSA实现代码
2022-01-03 11:30:28
python监控网站运行异常并发送邮件的方法
2021-07-28 06:27:34
详解Python字符串切片
2021-09-10 05:10:43
![](https://img.aspxhome.com/file/2023/5/67425_0s.png)
一文教你彻底解决Python包下载慢问题
2023-04-02 20:29:32
![](https://img.aspxhome.com/file/2023/7/96927_0s.png)
由日文出错的Bug,重新认识 Replace 函数
2009-07-07 22:28:00
python输入错误后删除的方法
2023-07-25 11:55:09
点球小游戏python脚本
2022-07-17 23:28:03
Python Pandas知识点之缺失值处理详解
2023-09-29 20:23:16
![](https://img.aspxhome.com/file/2023/0/63220_0s.png)
深入研究ASP中的Session
2007-09-13 12:37:00
python实现字符串连接的三种方法及其效率、适用场景详解
2023-07-27 19:57:32