07python实现traceroute程序[Python基础]

# 这个脚本是实现Linux中traceroute程序的，是探测从我们这个机器到我们要探测的IP地址中间都需要经过那些路由。
# 原理：我们的机器发送UDP高端口的数据包，发送给目的地址，首先设置ttl为1，然后逐次增加，在没有到达我们的目的IP
# 地址的路由，会发送ICMP的超时报文，然后我们从中提取IP地址，因为我们发送的是高端口的报文，到达目的地址的时候，
# 目的地址会发送ICMP的端口不可达报文，这样我们就探测出从我们源端口到目的端口的路由。
from scapy.all import *
import struct,re,random

# 跟我们实现ping程序的想法是一样的，首先构造一个发送一个UDP报文的函数，
# 入参为目的地址，ttl数。
def traceroute_one(dst,ttl_no,dport):
    # 定义发包时间。
    send_time = time.time()
    try:
        # 发送一个包，接收一个包。
        traceroute_one_reply = sr1(IP(dst=dst, ttl=ttl_no) / UDP(dport=dport) / b"hello world", timeout=1,
                                   verbose=False)
        # 判断ICMP包是不是超时回答。
        if traceroute_one_reply.getlayer(ICMP).type == 11 and traceroute_one_reply.getlayer(ICMP).code == 0:
            # 提取源地址
            src_ip = traceroute_one_reply.getlayer(IP).src
            # 定义接收时间。
            recv_time = time.time()
            # 计算时间ms数
            mid_time = (recv_time - send_time) * 1000
            # 返回。
            return 1,src_ip,mid_time
        # 这里接接收的是最后一跳。ICMP应该是端口不可达。
        elif traceroute_one_reply.getlayer(ICMP).type == 3 and traceroute_one_reply.getlayer(ICMP).code == 3:
            # 下边处理是一样的。
            src_ip = traceroute_one_reply.getlayer(IP).src
            recv_time = time.time()
            mid_time = (recv_time - send_time) * 1000
            return 2, src_ip, mid_time
    except Exception as e:
        return None


def traceroute(dst,hops):
    # 目的端口从33434开始算起，入参为目的地址，我们想要查找的路由的条数。
    dport = 33434
    hop = 0
    # 进行循环包。
    while hop         hop += 1
        # 这里需要改变端口。
        dport += hop
        # 发送一个包，获取返回值。
        traceroute_result = traceroute_one(dst,hop,dport)
        # 如果出现了错误，打印*号。
        if traceroute_result == None:
            print("*")
        # 这里代表中间路由，我们进行打印信息。
        elif traceroute_result[0] == 1:
            print("%d %s %4.2fms" % (hop,traceroute_result[1],traceroute_result[2]))
        # 最后一个包，为端口不可达，打印信息后，需要退出循环，因为已经到达目的地址了，虽然可能没有达到我们定义的条数。
        elif traceroute_result[0] == 2:
            print("%d %s %4.2fms" % (hop, traceroute_result[1], traceroute_result[2]))
            break
        time.sleep(1)
if __name__ == "__main__":
    traceroute("180.101.49.12",10)
# 接下来我们就可以使用wirshark进行抓包来看一下。