python 网络协议项目_python网络作业：使用python的socket库实现ICMP协议的traceroute

直接上代码：

#!/usr/bin/python3

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

# 技术支持：https://www.jianshu.com/u/69f40328d4f0

# 技术支持 https://china-testing.github.io/

# https://github.com/china-testing/python-api-tesing/blob/master/practices/traceroute.py

# 项目实战讨论QQ群630011153 144081101

# CreateDate: 2018-11-26

import socket

import io

import struct

import sys

class flushfile(io.FileIO):

def __init__(self, f):

self.f = f

def write(self, x):

self.f.write(x)

self.f.flush()

sys.stdout = flushfile(sys.stdout)

def main(dest_name):

dest_addr = socket.gethostbyname(dest_name)

port = 55285

max_hops = 30

ttl = 1

while True:

rec_socket = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_ICMP)

send_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)

send_socket.setsockopt(socket.SOL_IP, socket.IP_TTL, ttl)

timeout = struct.pack("ll", 2, 0)

rec_socket.setsockopt(socket.SOL_SOCKET, socket.SO_RCVTIMEO, timeout)

rec_socket.bind(("", port))

sys.stdout.write(" %d " % ttl)

send_socket.sendto(bytes("", "utf-8"), (dest_name, port))

curr_addr = None

curr_name = None

finished = False

tries = 1

while not finished and tries > 0:

try:

_, curr_addr = rec_socket.recvfrom(512)

finished = True

curr_addr = curr_addr[0]

try:

curr_name = socket.gethostbyaddr(curr_addr)[0]

except socket.error:

curr_name = curr_addr

except socket.error as err:

tries -= 1

sys.stdout.write("* ")

send_socket.close()

rec_socket.close()

if not finished:

pass

if curr_addr is not None:

curr_host = "%s (%s)" % (curr_name, curr_addr)

else:

curr_host = ""

sys.stdout.write("%s\n" % (curr_host))

ttl += 1

if curr_addr == dest_addr or ttl > max_hops:

break

if __name__ == "__main__":

main("china-testing.github.io")

参考资料

执行

注意要有root或管理员权限：

# python3 traceroute.py china-testing.github.io -t 4

1 *

2 11.223.48.177 (11.223.48.177)

3 11.223.48.10 (11.223.48.10)

4 *

5 106.11.37.58 (106.11.37.58)

6 116.251.105.229 (116.251.105.229)

7 122.224.214.69 (122.224.214.69)

8 *

9 115.239.209.30 (115.239.209.30)

10 *

11 *

...