python_reference/useful_scripts/ping.py

40 lines
1.1 KiB
Python
Raw Normal View History

2014-04-17 17:42:46 +00:00
# Example run:
# ~/Desktop >sudo python3 ping.py
# 0.017251014709472656
# Original source: https://gist.github.com/pyos/10980172
import time
import random
import select
import socket
def chk(data):
x = sum(a + b * 256 for a, b in zip(data[::2], data[1::2] + b'\x00')) & 0xFFFFFFFF
x = (x >> 16) + (x & 0xFFFF)
x = (x >> 16) + (x & 0xFFFF)
return (~x & 0xFFFF).to_bytes(2, 'little')
def ping(addr, timeout=1):
with socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_ICMP) as conn:
payload = random.randrange(0, 65536).to_bytes(2, 'big') + b'\x01\x00'
packet = b'\x08\x00' + b'\x00\x00' + payload
packet = b'\x08\x00' + chk(packet) + payload
conn.connect((addr, 80))
conn.sendall(packet)
start = time.time()
while select.select([conn], [], [], max(0, start + timeout - time.time()))[0]:
packet = conn.recv(1024)[20:]
unchecked = packet[:2] + b'\0\0' + packet[4:]
if packet == b'\0\0' + chk(unchecked) + payload:
return time.time() - start
if __name__ == '__main__':
print(ping('google.com'))