如何在Python中创建ICMP traceroute

Question

我试图在Python中实现一个基于ICMP的Traceroute。

我发现了一个非常有用的指南（https://blogs.oracle.com/ksplice/entry/learning_by_doing_writing_your），它允许我创建一个基于UDP的Traceroute（下面的代码），所以只需要修改。我已经尝试将发送套接字更改为ICMP，但是没有例外，我无法运行任何内容。

注意-下面的代码工作，但这是一个UDP跟踪路由（发送一个UDP包和接收一个ICMP包），我需要我的程序发送一个ICMP包和接收一个ICMP包。这是因为现在的防火墙比以前更智能，并且在接收到随机端口的UDP数据包后并不总是发送ICMP响应。实际上，对于ICMP套接字，UDP套接字需要更改。

我想这不是最常见的尝试和实现的事情，在网上研究如何做到这一点是有困难的。如果有人能提供一些见解，我们将不胜感激

要记住的要点是traceroutes通过设置TTL工作，因此如果解决方案是使用ICMP库，那么它需要有一个可配置的TTL:-）

#!/usr/bin/python

import socket

def main(dest_name):
    dest_addr = socket.gethostbyname(dest_name)
    port = 33434
    max_hops = 30
    icmp = socket.getprotobyname('icmp')
    udp = socket.getprotobyname('udp')
    ttl = 1
    while True:
        recv_socket = socket.socket(socket.AF_INET, socket.SOCK_RAW, icmp)
        send_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, udp)
        send_socket.setsockopt(socket.SOL_IP, socket.IP_TTL, ttl)
        recv_socket.bind(("", port))
        send_socket.sendto("", (dest_name, port))
        curr_addr = None
        curr_name = None
        try:
            _, curr_addr = recv_socket.recvfrom(512)
            curr_addr = curr_addr[0]
            try:
                curr_name = socket.gethostbyaddr(curr_addr)[0]
            except socket.error:
                curr_name = curr_addr
        except socket.error:
            pass
        finally:
            send_socket.close()
            recv_socket.close()

        if curr_addr is not None:
            curr_host = "%s (%s)" % (curr_name, curr_addr)
        else:
            curr_host = "*"
        print "%d\t%s" % (ttl, curr_host)

        ttl += 1
        if curr_addr == dest_addr or ttl > max_hops:
            break

if __name__ == "__main__":
    main('google.com')

Answer 1

如果您不想自己管理套接字，scapy是一个很好的工具，可以构建您自己的数据包并使用它们进行基本的发送/接收。您可以找到一个如何ping的示例here，从这里（并阅读一些文档）您应该能够非常轻松地构建自己的traceroute。

Answer 2

这是使用Scapy的解决方案-感谢KillianDS。

发送ping

ans,unans=sr(IP(dst="www.google.com",ttl=X)/ICMP())

访问回复

ans.summary(lambda (s,r): r.sprintf("%IP.src%") )