python图像跟踪代码_Python—路由追踪（并生成追踪图片）

需要先安装两个包

[root@localhost ~]# yum install graphviz // 为了使用dot命令

[root@localhost ~]# yum install ImageMagick // 为了使用/usr/bin/convert命令

python脚本代码如下

#!/usr/bin/evn python

#-*-coding:utf-8 -*-

import os,sys,time,subprocess

import warnings,logging

from scapy.all import traceroute

from scapy.as_resolvers import AS_resolver_radb

warnings.filterwarnings("ignore", category=DeprecationWarning)

logging.getLogger("scapy.runtine").setLevel(logging.ERROR)

domains = raw_input('please input the domain or IP what you want: ')

target = domains.split(" ")

dport = [80, 21]

if len(target) >= 1 and target[0] != '':

# 启动路由跟踪

res, unans = traceroute(target, dport=dport, retry=-2)

# traceroute 生成的信息绘制成svg

# res.graph(target="> graph.svg")

res.graph(target="> graph.svg", ASres=AS_resolver_radb(), type="svg")

time.sleep(2)

# svg 转格式为 png

subprocess.Popen("/usr/bin/convert graph.svg graph.png", shell=True)

else:

print("IP/domain number of errors, exit")

注意：traceroute程序都需要root来运行。traceroute函数可以传入以列表格式的多个域名/IP，和多个端口(这里有80和21)。