python实现一个简单的ping工具方法


Posted in Python onJanuary 31, 2019

继上一篇计算checksum校验和,本章通过socket套接字,struct字节打包成二进制,select返回套接字的文件描述符的结合,实现一个简单的ping工具。

#!/usr/bin/python3.6.4
#!coding:utf-8
__author__ = 'Rosefinch'
__date__ = '2018/5/31 22:27'

import time
import struct
import socket
import select
import sys

def chesksum(data):
  """
  校验
  """
  n = len(data)
  m = n % 2
  sum = 0 
  for i in range(0, n - m ,2):
    sum += (data[i]) + ((data[i+1]) << 8)#传入data以每两个字节(十六进制)通过ord转十进制,第一字节在低位,第二个字节在高位
  if m:
    sum += (data[-1])
  #将高于16位与低16位相加
  sum = (sum >> 16) + (sum & 0xffff)
  sum += (sum >> 16) #如果还有高于16位,将继续与低16位相加
  answer = ~sum & 0xffff
  #主机字节序转网络字节序列(参考小端序转大端序)
  answer = answer >> 8 | (answer << 8 & 0xff00)
  return answer 

  '''
  连接套接字,并将数据发送到套接字
  '''
def raw_socket(dst_addr,imcp_packet):
  rawsocket = socket.socket(socket.AF_INET,socket.SOCK_RAW,socket.getprotobyname("icmp"))
  send_request_ping_time = time.time()
  #send data to the socket
  rawsocket.sendto(imcp_packet,(dst_addr,80))
  return send_request_ping_time,rawsocket,dst_addr

  '''
  request ping
  '''
def request_ping(data_type,data_code,data_checksum,data_ID,data_Sequence,payload_body):
  #把字节打包成二进制数据
  imcp_packet = struct.pack('>BBHHH32s',data_type,data_code,data_checksum,data_ID,data_Sequence,payload_body)
  icmp_chesksum = chesksum(imcp_packet)#获取校验和
  imcp_packet = struct.pack('>BBHHH32s',data_type,data_code,icmp_chesksum,data_ID,data_Sequence,payload_body)
  return imcp_packet
  '''
  reply ping
  '''
def reply_ping(send_request_ping_time,rawsocket,data_Sequence,timeout = 2):
  while True:
    started_select = time.time()
    what_ready = select.select([rawsocket], [], [], timeout)
    wait_for_time = (time.time() - started_select)
    if what_ready[0] == []: # Timeout
      return -1
    time_received = time.time()
    received_packet, addr = rawsocket.recvfrom(1024)
    icmpHeader = received_packet[20:28]
    type, code, checksum, packet_id, sequence = struct.unpack(
      ">BBHHH", icmpHeader
    )
    if type == 0 and sequence == data_Sequence:
      return time_received - send_request_ping_time
    timeout = timeout - wait_for_time
    if timeout <= 0:
      return -1

  '''
  实现 ping 主机/ip
  '''
def ping(host):
  data_type = 8 # ICMP Echo Request
  data_code = 0 # must be zero
  data_checksum = 0 # "...with value 0 substituted for this field..."
  data_ID = 0 #Identifier
  data_Sequence = 1 #Sequence number
  payload_body = b'abcdefghijklmnopqrstuvwabcdefghi' #data
  dst_addr = socket.gethostbyname(host)#将主机名转ipv4地址格式,返回以ipv4地址格式的字符串,如果主机名称是ipv4地址,则它将保持不变
  print("正在 Ping {0} [{1}] 具有 32 字节的数据:".format(host,dst_addr))
  for i in range(0,4):
    icmp_packet = request_ping(data_type,data_code,data_checksum,data_ID,data_Sequence + i,payload_body)
    send_request_ping_time,rawsocket,addr = raw_socket(dst_addr,icmp_packet)
    times = reply_ping(send_request_ping_time,rawsocket,data_Sequence + i)
    if times > 0:
      print("来自 {0} 的回复: 字节=32 时间={1}ms".format(addr,int(times*1000)))
      time.sleep(0.7)
    else:
      print("请求超时。")

if __name__ == "__main__":
  if len(sys.argv) < 2:
    sys.exit('Usage: ping.py <host>')
  ping(sys.argv[1])
>python .\Ping_tool.py www.csdn.net
正在 Ping www.csdn.net [47.95.164.112] 具有 32 字节的数据:
来自 47.95.164.112 的回复: 字节=32 时间=39ms
来自 47.95.164.112 的回复: 字节=32 时间=40ms
来自 47.95.164.112 的回复: 字节=32 时间=40ms
来自 47.95.164.112 的回复: 字节=32 时间=39ms

引用:

https://docs.python.org/3/library/socket.html

https://docs.python.org/3/library/select.html#select.select

https://docs.python.org/3/library/struct.html

以上这篇python实现一个简单的ping工具方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python标准库urllib2的一些使用细节总结
Mar 16 Python
Python松散正则表达式用法分析
Apr 29 Python
全面了解python字符串和字典
Jul 07 Python
Window10+Python3.5安装opencv的教程推荐
Apr 02 Python
selenium+python 去除启动的黑色cmd窗口方法
May 22 Python
Python中存取文件的4种不同操作
Jul 02 Python
Tensorflow 实现修改张量特定元素的值方法
Jul 30 Python
一行代码让 Python 的运行速度提高100倍
Oct 08 Python
python 比较2张图片的相似度的方法示例
Dec 18 Python
python为QT程序添加图标的方法详解
Mar 09 Python
PySide2出现“ImportError: DLL load failed: 找不到指定的模块”的问题及解决方法
Jun 10 Python
pytorch常用数据类型所占字节数对照表一览
May 17 Python
Python获取网段内ping通IP的方法
Jan 31 #Python
Python实现删除排序数组中重复项的两种方法示例
Jan 31 #Python
python重试装饰器的简单实现方法
Jan 31 #Python
Python实现合并两个有序链表的方法示例
Jan 31 #Python
Django 日志配置按日期滚动的方法
Jan 31 #Python
Python类的继承用法示例
Jan 31 #Python
判断python对象是否可调用的三种方式及其区别详解
Jan 31 #Python
You might like
PHP 判断常量,变量和函数是否存在
2009/04/26 PHP
解析func_num_args与func_get_args函数的使用
2013/06/24 PHP
浅谈ThinkPHP的URL重写
2014/11/25 PHP
php有道翻译api调用方法实例
2014/12/22 PHP
WordPress特定文章对搜索引擎隐藏或只允许搜索引擎查看
2015/12/31 PHP
jquery绑定原理 简单解析与实现代码分享
2011/09/06 Javascript
如何将一个String和多个String值进行比较思路分析
2013/04/22 Javascript
js 高效去除数组重复元素示例代码
2013/12/19 Javascript
精彩的Bootstrap案例分享 重点在注释!(选项卡、栅格布局)
2016/07/01 Javascript
微信小程序 实现tabs选项卡效果实例代码
2016/10/31 Javascript
jquery easyui validatebox remote的使用详解
2016/11/09 Javascript
解决vue中post方式提交数据后台无法接收的问题
2018/08/11 Javascript
利用es6 new.target来对模拟抽象类的方法
2019/05/10 Javascript
小程序组件之自定义顶部导航实例
2019/06/12 Javascript
js笔试题-接收get请求参数
2019/06/15 Javascript
jQuery 实现DOM元素拖拽交换位置的实例代码
2020/07/14 jQuery
Nuxt的动态路由和参数校验操作
2020/11/09 Javascript
手写Vue源码之数据劫持示例详解
2021/01/04 Vue.js
Python文件操作中进行字符串替换的方法(保存到新文件/当前文件)
2019/06/28 Python
在 Python 中接管键盘中断信号的实现方法
2020/02/04 Python
如何通过python实现全排列
2020/02/11 Python
python3检查字典传入函数键是否齐全的实例
2020/06/05 Python
pycharm导入源码的具体步骤
2020/08/04 Python
Python 中如何写注释
2020/08/28 Python
python如何提升爬虫效率
2020/09/27 Python
详解使用scrapy进行模拟登陆三种方式
2021/02/21 Python
CSS3弹性盒模型开发笔记(二)
2016/04/26 HTML / CSS
90后毕业生的求职信范文
2013/09/21 职场文书
公司授权委托书
2014/04/04 职场文书
重大事项社会稳定风险评估方案
2014/06/15 职场文书
工程索赔意向书
2014/08/30 职场文书
出生医学证明书
2014/09/15 职场文书
大学生团员个人总结
2015/02/14 职场文书
行政申诉状范文
2015/05/20 职场文书
2016新年问候语大全
2015/11/11 职场文书
VS2019连接MySQL数据库的过程及常见问题总结
2021/11/27 MySQL