直接上代码:
#!/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-26import socketimport ioimport structimport sysclass 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("%sn" % (curr_host))
ttl += 1
if curr_addr == dest_addr or ttl > max_hops: breakif __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 * ...
作者:python作业AI毕业设计
链接:https://www.jianshu.com/p/a4efc40983e1



