ping

rasbt · rasbt · commit 38e5d11dd43b · 2014-04-17T13:42:46.000-04:00
diff --git a/useful_scripts/ping.py b/useful_scripts/ping.py
@@ -0,0 +1,39 @@
+# 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'))
