从网上找的 ping 源代码

wonder365

标准ping程序的python实现。代码不错，注释的也不错，里边还有icmp的报文结构。
http://www.gnist.org/~lars/code/ping/ping.py

rhinux

沙发...对我来说值得一读

rhinux

方便大家阅读我把代码复制过来了

1. #!/usr/bin/env python
   
2. # -*- coding: iso-8859-1 -*-
   
3. """ping.py
   
4. 
   
5. ping.py uses the ICMP protocol's mandatory ECHO_REQUEST
   
6. datagram to elicit an ICMP ECHO_RESPONSE from a
   
7. host or gateway.
   
8. 
   
9. Copyright (C) 2004 - Lars Strand <lars strand at gnist org>;
   
10. 
    
11. This program is free software; you can redistribute it and/or
    
12. modify it under the terms of the GNU General Public License
    
13. as published by the Free Software Foundation; either version 2
    
14. of the License, or (at your option) any later version.
    
15. 
    
16. This program is distributed in the hope that it will be useful,
    
17. but WITHOUT ANY WARRANTY; without even the implied warranty of
    
18. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    
19. GNU General Public License for more details.
    
20. 
    
21. You should have received a copy of the GNU General Public License
    
22. along with this program; if not, write to the Free Software
    
23. Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
    
24. 
    
25. Must be running as root, or write a suid-wrapper. Since newer *nix
    
26. variants, the kernel ignores the set[ug]id flags on #! scripts for
    
27. security reasons
    
28. 
    
29. RFC792, echo/reply message:
    
30. 
    
31.   0                   1                   2                   3
    
32.   0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
    
33. +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
    
34. |     Type      |     Code      |          Checksum             |
    
35. +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
    
36. |           Identifier          |        Sequence Number        |
    
37. +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
    
38. |     Data ...
    
39. +-+-+-+-+-
    
40. 
    
41. 
    
42. TODO:
    
43. - do not create socket inside 'while' (but if not: ipv6 won't work)
    
44. - add support for broadcast/multicast
    
45. - add support for own payload string
    
46. 
    
47. CHANGELOG:
    
48. DONE -->; bugfix from Filip Van Raemdonck mechanix debian org
    
49. DONE -->; add more support for modules (raise instead of sys.exit)
    
50. DONE -->; locale func names
    
51. DONE -->; package def
    
52. DONE -->; some code cleanup
    
53. 
    
54. """
    
55. 
    
56. import sys
    
57. import os
    
58. import struct
    
59. import array
    
60. import time
    
61. import select
    
62. import binascii
    
63. import math
    
64. import getopt
    
65. import string
    
66. import socket
    
67. 
    
68. # total size of data (payload)
    
69. ICMP_DATA_STR = 56  
    
70. 
    
71. # initial values of header variables
    
72. ICMP_TYPE = 8
    
73. ICMP_TYPE_IP6 = 128
    
74. ICMP_CODE = 0
    
75. ICMP_CHECKSUM = 0
    
76. ICMP_ID = 0
    
77. ICMP_SEQ_NR = 0
    
78. 
    
79. # Package definitions.
    
80. __program__   = 'ping'
    
81. __version__   = '0.5a'
    
82. __date__      = '2004/15/12'
    
83. __author__    = 'Lars Strand <lars at unik no>;'
    
84. __licence__   = 'GPL'
    
85. __copyright__ = 'Copyright (C) 2004 Lars Strand'
    
86. 
    
87. def _construct(id, size, ipv6):
    
88.     """Constructs a ICMP echo packet of variable size
    
89.     """
    
90. 
    
91.     # size must be big enough to contain time sent
    
92.     if size < int(struct.calcsize("d")):
    
93.         _error("packetsize to small, must be at least %d" % int(struct.calcsize("d")))
    
94.    
    
95.     # construct header
    
96.     if ipv6:
    
97.         header = struct.pack('BbHHh', ICMP_TYPE_IP6, ICMP_CODE, ICMP_CHECKSUM, \
    
98.                              ICMP_ID, ICMP_SEQ_NR+id)
    
99.     else:
    
100.         header = struct.pack('bbHHh', ICMP_TYPE, ICMP_CODE, ICMP_CHECKSUM, \
     
101.                              ICMP_ID, ICMP_SEQ_NR+id)
     
102. 
     
103.     # if size big enough, embed this payload
     
104.     load = "-- IF YOU ARE READING THIS YOU ARE A NERD! --"
     
105.    
     
106.     # space for time
     
107.     size -= struct.calcsize("d")
     
108. 
     
109.     # construct payload based on size, may be omitted :)
     
110.     rest = ""
     
111.     if size >; len(load):
     
112.         rest = load
     
113.         size -= len(load)
     
114. 
     
115.     # pad the rest of payload
     
116.     rest += size * "X"
     
117. 
     
118.     # pack
     
119.     data = struct.pack("d", time.time()) + rest
     
120.     packet = header + data          # ping packet without checksum
     
121.     checksum = _in_cksum(packet)    # make checksum
     
122. 
     
123.     # construct header with correct checksum
     
124.     if ipv6:
     
125.         header = struct.pack('BbHHh', ICMP_TYPE_IP6, ICMP_CODE, checksum, \
     
126.                              ICMP_ID, ICMP_SEQ_NR+id)
     
127.     else:
     
128.         header = struct.pack('bbHHh', ICMP_TYPE, ICMP_CODE, checksum, ICMP_ID, \
     
129.                              ICMP_SEQ_NR+id)
     
130. 
     
131.     # ping packet *with* checksum
     
132.     packet = header + data
     
133. 
     
134.     # a perfectly formatted ICMP echo packet
     
135.     return packet
     
136. 
     
137. def _in_cksum(packet):
     
138.     """THE RFC792 states: 'The 16 bit one's complement of
     
139.     the one's complement sum of all 16 bit words in the header.'
     
140. 
     
141.     Generates a checksum of a (ICMP) packet. Based on in_chksum found
     
142.     in ping.c on FreeBSD.
     
143.     """
     
144. 
     
145.     # add byte if not dividable by 2
     
146.     if len(packet) & 1:              
     
147.         packet = packet + '\0'
     
148. 
     
149.     # split into 16-bit word and insert into a binary array
     
150.     words = array.array('h', packet)
     
151.     sum = 0
     
152. 
     
153.     # perform ones complement arithmetic on 16-bit words
     
154.     for word in words:
     
155.         sum += (word & 0xffff)
     
156. 
     
157.     hi = sum >;>; 16
     
158.     lo = sum & 0xffff
     
159.     sum = hi + lo
     
160.     sum = sum + (sum >;>; 16)
     
161.    
     
162.     return (~sum) & 0xffff # return ones complement
     
163. 
     
164. def pingNode(alive=0, timeout=1.0, ipv6=0, number=sys.maxint, node=None, \
     
165.              flood=0, size=ICMP_DATA_STR):
     
166.     """Pings a node based on input given to the function.
     
167.     """
     
168. 
     
169.     # if no node, exit
     
170.     if not node:
     
171.         _error("")
     
172. 
     
173.     # if not a valid host, exit
     
174.     if ipv6:
     
175.         if socket.has_ipv6:
     
176.             try:
     
177.                 info, port = socket.getaddrinfo(node, None)
     
178.                 host = info[4][0]
     
179.                 # do not print ipv6 twice if ipv6 address given as node
     
180.                 if host == node:
     
181.                     noPrintIPv6adr = 1
     
182.             except:
     
183.                 _error("cannot resolve %s: Unknow host" % node)
     
184.         else:
     
185.             _error("No support for IPv6 on this plattform")
     
186.     else:    # IPv4
     
187.         try:
     
188.             host = socket.gethostbyname(node)
     
189.         except:
     
190.             _error("cannot resolve %s: Unknow host" % node)
     
191. 
     
192.     # trying to ping a network?
     
193.     if not ipv6:
     
194.         if int(string.split(host, ".")[-1]) == 0:
     
195.             _error("no support for network ping")
     
196. 
     
197.     # do some sanity check
     
198.     if number == 0:
     
199.         _error("invalid count of packets to transmit: '%s'" % str(a))
     
200.     if alive:
     
201.         number = 1
     
202. 
     
203.     # Send the ping(s)
     
204.     start = 1; mint = 999; maxt = 0.0; avg = 0.0
     
205.     lost = 0; tsum = 0.0; tsumsq = 0.0
     
206. 
     
207.     # tell the user what we do
     
208.     if not alive:
     
209.         if ipv6:
     
210.             # do not print the ipv6 twice if ip adress given as node
     
211.             # (it can be to long in term window)
     
212.             if noPrintIPv6adr == 1:
     
213.                 # add 40 (header) + 8 (icmp header) + payload
     
214.                 print "PING %s : %d data bytes (40+8+%d)" % (str(node), \
     
215.                                                              40+8+size, size)
     
216.             else:
     
217.                 # add 40 (header) + 8 (icmp header) + payload
     
218.                 print "PING %s (%s): %d data bytes (40+8+%d)" % (str(node), \
     
219.                                                                  str(host), 40+8+size, size)
     
220.         else:
     
221.             # add 20 (header) + 8 (icmp header) + payload
     
222.             print "PING %s (%s): %d data bytes (20+8+%d)" % (str(node), str(host), \
     
223.                                                              20+8+size, size)
     
224.         
     
225.     # trap ctrl-d and ctrl-c
     
226.     try:
     
227.         
     
228.         # send the number of ping packets as given
     
229.         while start <= number:
     
230.             lost += 1 # in case user hit ctrl-c
     
231.             
     
232.             # create the IPv6/IPv4 socket
     
233.             if ipv6:
     
234.                 # can not create a raw socket if not root or setuid to root
     
235.                 try:
     
236.                     pingSocket = socket.socket(socket.AF_INET6, socket.SOCK_RAW, \
     
237.                                                socket.getprotobyname("ipv6-icmp"))
     
238.                 except socket.error, e:
     
239.                     print "socket error: %s" % e
     
240.                     _error("You must be root (uses raw sockets)" % os.path.basename(sys.argv[0]))
     
241.                     
     
242.             # IPv4
     
243.             else:
     
244.                 # can not create a raw socket if not root or setuid to root
     
245.                 try:
     
246.                     pingSocket = socket.socket(socket.AF_INET, socket.SOCK_RAW, \
     
247.                                                socket.getprotobyname("icmp"))
     
248.                 except socket.error, e:
     
249.                     print "socket error: %s" % e
     
250.                     _error("You must be root (%s uses raw sockets)" % os.path.basename(sys.argv[0]))
     
251.                
     
252.             packet = _construct(start, size, ipv6) # make a ping packet
     
253. 
     
254.             # send the ping
     
255.             try:
     
256.                 pingSocket.sendto(packet,(node,1))
     
257.             except socket.error, e:
     
258.                 _error("socket error: %s" % e)
     
259. 
     
260.             # reset values
     
261.             pong = ""; iwtd = []
     
262. 
     
263.             # wait until there is data in the socket
     
264.             while 1:
     
265.                 # input, output, exceptional conditions
     
266.                 iwtd, owtd, ewtd = select.select([pingSocket], [], [], timeout)
     
267.                 break # no data and timout occurred
     
268. 
     
269.             # data on socket - this means we have an answer
     
270.             if iwtd:  # ok, data on socket
     
271.                 endtime = time.time()  # time packet received
     
272.                 # read data (we only need the header)
     
273.                 pong, address = pingSocket.recvfrom(size+48)
     
274.                 lost -= 1 # in case user hit ctrl-c
     
275. 
     
276.                 # examine packet
     
277.                 # fetch TTL from IP header
     
278.                 if ipv6:
     
279.                     # since IPv6 header and any extension header are never passed
     
280.                     # to a raw socket, we can *not* get hoplimit field..
     
281.                     # I hoped that a socket option would help, but it's not
     
282.                     # supported:
     
283.                     #   pingSocket.setsockopt(IPPROTO_IPV6, IPV6_RECVHOPLIMIT, 1)
     
284.                     # so we can't fetch hoplimit..
     
285. 
     
286.                     # fetch hoplimit
     
287.                     #rawPongHop = struct.unpack("c", pong[7])[0]
     
288. 
     
289.                     # fetch pong header
     
290.                     pongHeader = pong[0:8]
     
291.                     pongType, pongCode, pongChksum, pongID, pongSeqnr = \
     
292.                               struct.unpack("bbHHh", pongHeader)
     
293. 
     
294.                     # fetch starttime from pong
     
295.                     starttime = struct.unpack("d", pong[8:16])[0]
     
296. 
     
297.                 # IPv4
     
298.                 else:
     
299.                     # time to live
     
300.                     rawPongHop = struct.unpack("s", pong[8])[0]
     
301. 
     
302.                     # convert TTL from 8 bit to 16 bit integer
     
303.                     pongHop = int(binascii.hexlify(str(rawPongHop)), 16)
     
304. 
     
305.                     # fetch pong header
     
306.                     pongHeader = pong[20:28]
     
307.                     pongType, pongCode, pongChksum, pongID, pongSeqnr = \
     
308.                               struct.unpack("bbHHh", pongHeader)
     
309. 
     
310.                     # fetch starttime from pong
     
311.                     starttime = struct.unpack("d", pong[28:36])[0]
     
312. 
     
313.                 # valid ping packet received?
     
314.                 if not pongSeqnr == start:
     
315.                     pong = None
     
316. 
     
317.             # NO data on socket - timeout waiting for answer
     
318.             if not pong:
     
319.                 if alive:
     
320.                     print "no reply from %s (%s)" % (str(node), str(host))
     
321.                 else:
     
322.                     print "ping timeout: %s (icmp_seq=%d) " % (host, start)
     
323. 
     
324.                 # do not wait if just sending one packet
     
325.                 if number != 1 and start < number:
     
326.                     time.sleep(flood ^ 1)
     
327.                 start += 1
     
328.                 continue  # lost a packet - try again
     
329. 
     
330.             triptime  = endtime - starttime # compute RRT
     
331.             tsum     += triptime            # triptime for all packets (stddev)
     
332.             tsumsq   += triptime * triptime # triptime^2  for all packets (stddev)
     
333. 
     
334.             # compute statistic
     
335.             maxt = max ((triptime, maxt))
     
336.             mint = min ((triptime, mint))
     
337. 
     
338.             if alive:
     
339.                 print str(node) + " (" + str(host) +") is alive"
     
340.             else:
     
341.                 if ipv6:
     
342.                     # size + 8 = payload + header
     
343.                     print "%d bytes from %s: icmp_seq=%d time=%.5f ms" % \
     
344.                           (size+8, host, pongSeqnr, triptime*1000)
     
345.                 else:
     
346.                     print "%d bytes from %s: icmp_seq=%d ttl=%s time=%.5f ms" % \
     
347.                           (size+8, host, pongSeqnr, pongHop, triptime*1000)
     
348. 
     
349.             # do not wait if just sending one packet
     
350.             if number != 1 and start < number:
     
351.                 # if flood = 1; do not sleep - just ping               
     
352.                 time.sleep(flood ^ 1) # wait before send new packet
     
353. 
     
354.             # the last thing to do is update the counter - else the value
     
355.             # (can) get wrong when computing summary at the end (if user
     
356.             # hit ctrl-c when pinging)
     
357.             start += 1
     
358.             # end ping send/recv while
     
359. 
     
360.     # if user ctrl-d or ctrl-c
     
361.     except (EOFError, KeyboardInterrupt):
     
362.         # if user disrupts ping, it is most likly done before
     
363.         # the counter get updates - if do not update it here, the
     
364.         # summary get all wrong.
     
365.         start += 1
     
366.         pass
     
367. 
     
368.     # compute and print som stats
     
369.     # stddev computation based on ping.c from FreeBSD
     
370.     if start != 0 or lost >; 0:  # do not print stats if 0 packet sent
     
371.         start -= 1              # since while is '<='
     
372.         avg = tsum / start      # avg round trip
     
373.         vari = tsumsq / start - avg * avg
     
374.         # %-packet lost
     
375.         if start == lost:
     
376.             plost = 100
     
377.         else:
     
378.             plost = (lost/start)*100
     
379. 
     
380.         if not alive:
     
381.             print "\n--- %s ping statistics ---" % node
     
382.             print "%d packets transmitted, %d packets received, %d%% packet loss" % \
     
383.                   (start, start-lost, plost)
     
384.             # don't display summary if 100% packet-loss
     
385.             if plost != 100:
     
386.                 print "round-trip min/avg/max/stddev = %.3f/%.3f/%.3f/%.3f ms" % \
     
387.                       (mint*1000, (tsum/start)*1000, maxt*1000, math.sqrt(vari)*1000)
     
388. 
     
389.     pingSocket.close()
     
390.    
     
391. def _error(err):
     
392.     """Exit if running standalone, else raise an exception
     
393.     """
     
394. 
     
395.     if __name__ == '__main__':
     
396.         print "%s: %s" % (os.path.basename(sys.argv[0]), str(err))
     
397.         print "Try `%s --help' for more information." % os.path.basename(sys.argv[0])
     
398.         sys.exit(1)
     
399.     else:
     
400.         raise Exception, str(err)
     
401.    
     
402. def _usage():
     
403.     """Print usage if run as a standalone program
     
404.     """
     
405.     print """usage: %s [OPTIONS] HOST
     
406. Send ICMP ECHO_REQUEST packets to network hosts.
     
407. 
     
408. Mandatory arguments to long options are mandatory for short options too.
     
409.   -c, --count=N    Stop after sending (and receiving) 'N' ECHO_RESPONSE
     
410.                    packets.
     
411.   -s, --size=S     Specify the number of data bytes to be sent. The default
     
412.                    is 56, which translates into 64 ICMP data bytes when
     
413.                    combined with the 8 bytes of ICMP header data.
     
414.   -f, --flood      Flood ping. Outputs packets as fast as they come back. Use
     
415.                    with caution!
     
416.   -6, --ipv6       Ping using IPv6.
     
417.   -t, --timeout=s  Specify a timeout, in seconds, before a ping packet is
     
418.                    considered 'lost'.
     
419.   -h, --help       Display this help and exit
     
420. 
     
421. Report bugs to lars [at] gnist org""" % os.path.basename(sys.argv[0])
     
422. 
     
423. 
     
424. if __name__ == '__main__':
     
425.     """Main loop
     
426.     """
     
427. 
     
428.     # version control
     
429.     version = string.split(string.split(sys.version)[0][:3], ".")
     
430.     if map(int, version) < [2, 3]:
     
431.         _error("You need Python ver 2.3 or higher to run!")
     
432. 
     
433.     try:
     
434.         # opts = arguments recognized,
     
435.         # args = arguments NOT recognized (leftovers)
     
436.         opts, args = getopt.getopt(sys.argv[1:-1], "hat:6c:fs:", \
     
437.                                    ["help", "alive", "timeout=", "ipv6", \
     
438.                                     "count=", "flood", "packetsize="])
     
439.     except getopt.GetoptError:
     
440.         # print help information and exit:
     
441.         _error("illegal option(s) -- " + str(sys.argv[1:]))
     
442. 
     
443.     # test whether any host given
     
444.     if len(sys.argv) >;= 2:
     
445.         node = sys.argv[-1:][0]   # host to be pinged
     
446.         if node[0] == '-' or node == '-h' or node == '--help' :  
     
447.             _usage()
     
448.     else:
     
449.         _error("No arguments given")
     
450. 
     
451.     if args:
     
452.         _error("illegal option -- %s" % str(args))
     
453.         
     
454.     # default variables
     
455.     alive = 0; timeout = 1.0; ipv6 = 0; count = sys.maxint;
     
456.     flood = 0; size = ICMP_DATA_STR
     
457. 
     
458.     # run through arguments and set variables
     
459.     for o, a in opts:
     
460.         if o == "-h" or o == "--help":    # display help and exit
     
461.             _usage()
     
462.             sys.exit(0)
     
463.         if o == "-t" or o == "--timeout": # timeout before "lost"
     
464.             try:
     
465.                 timeout = float(a)
     
466.             except:
     
467.                 _error("invalid timout: '%s'" % str(a))
     
468.         if o == "-6" or o == "--ipv6":    # ping ipv6
     
469.             ipv6 = 1
     
470.         if o == "-c" or o == "--count":   # how many pings?
     
471.             try:
     
472.                 count = int(a)
     
473.             except:
     
474.                 _error("invalid count of packets to transmit: '%s'" % str(a))
     
475.         if o == "-f" or o == "--flood":   # no delay between ping send
     
476.             flood = 1
     
477.         if o == "-s" or o == "--packetsize":  # set the ping payload size
     
478.             try:
     
479.                 size = int(a)
     
480.             except:
     
481.                 _error("invalid packet size: '%s'" % str(a))
     
482.         # just send one packet and say "it's alive"
     
483.         if o == "-a" or o == "--alive":   
     
484.             alive = 1
     
485. 
     
486.     # here we send
     
487.     pingNode(alive=alive, timeout=timeout, ipv6=ipv6, number=count, \
     
488.              node=node, flood=flood, size=size)
     
489.     # if we made it this far, do a clean exit
     
490.     sys.exit(0)
     
491. 
     
492. ### end

复制代码

rc_hz

很好

plumlee

看不太懂

穿越时光

墙啊！！顶

nnggss

现在还看不明白,不过很快会明白的

kinnsei

good

qwent

好像用python写的啊.

但是不太明.

yoursmile

8页纸，我打印出来慢慢研究~
谢谢楼主