这两天在测试IPV6与DNS,晚上查了下资料,写了个DNS客户端,其中请求消息编码部分参考网上实例。
刚完成了编码,功能还没有测试。
1、代码:
~~~
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import ConfigParser
import socket
import time
import string
import os
import re
import struct
def dns_codec(hostname):
'''
Function:请求消息编码
Input:hostname:主机名,如www.baidu.com
Output: 编码后的字节流
author: socrates
date:2012-12-14
'''
index = os.urandom(2)
hoststr = ''.join(chr(len(x))+x for x in hostname.split('.'))
data = '%s\x01\x00\x00\x01\x00\x00\x00\x00\x00\x00%s\x00\x00\x01\x00\x01' % (index, hoststr)
data = struct.pack('!H', len(data)) + data
return data
def dns_decode(in_sock):
'''
Function:响应消息解码
Input:in_sock:接收消息的socket
Output:解码后的内容
author: socrates
date:2012-12-14
'''
rfile = in_sock.makefile('rb')
size = struct.unpack('!H', rfile.read(2))[0]
data = rfile.read(size)
iplist = re.findall('\xC0.\x00\x01\x00\x01.{6}(.{4})', data)
return ['.'.join(str(ord(x)) for x in s) for s in iplist]
def dns_sendmsg():
'''
Function:通过socket发送DNS查询消息
Input:None
Output:None
author: socrates
date:2012-12-14
'''
ens_client_config = ConfigParser.ConfigParser()
#读取配置文件
try:
ens_client_config.readfp(open('ens_client_config.ini'))
except ConfigParser.Error:
print 'read ens_client_config.ini failed.'
#获取需要的信息
server_ip_1 = ens_client_config.get("server_info","ip_1")
server_port_1 = ens_client_config.get("server_info","port_1")
sockettype_1 = ens_client_config.get("server_info","sockettype_1")
heartbeat_1 = ens_client_config.get("server_info","heartbeat_1")
msg_1 = ens_client_config.get("server_info","msg_1")
#IP类型
address_family = {True:socket.AF_INET6, False:socket.AF_INET}[':' in server_ip_1]
#传输类型
socket_type = {True:socket.SOCK_STREAM , False:socket.SOCK_DGRAM}['TCP' == sockettype_1.upper()]
try:
sock = socket.socket(address_family, socket_type)
except socket.error, e:
print 'create socket return error. errno = %d, errmsg = %s' % (e.args[0], e.args[1])
#连接服务器并发送消息
try:
#连接服务端
sock.connect((server_ip_1, string.atoi(server_port_1)))
while True:
#发送频率
time.sleep(string.atoi(heartbeat_1))
#发送消息
sock.sendall(dns_codec(msg_1))
#接收并打印消息
print dns_decode(sock)
except socket.error, e:
print 'connect server failed. errno = %d, errmsg = %s' % (e.args[0], e.args[1])
sock.close()
if __name__ == '__main__':
dns_sendmsg()
~~~
2、配置文件:
文件名:ens_client_config.ini
内容:
~~~
[server_info]
ip_1=192.168.1.1
port_1=53
sockettype_1=UDP
heartbeat_1=2
msg_1=www.google.com
~~~
3、早上测试了下UDP协议下的DNS客户端,功能正常可用。
a、wireshark抓包:
请求:
![QQ截图20121216114223](https://box.kancloud.cn/2016-06-08_575793629beb2.jpg)
响应:
![QQ截图20121216114223](https://box.kancloud.cn/2016-06-08_57579362bea0b.jpg)
b、代码运行结果:
![QQ截图20121216114223](https://box.kancloud.cn/2016-06-08_57579362e34af.jpg)
- 前言
- Python:实现文件归档
- Pyhon:按行输出文件内容
- Python:读文件和写文件
- Python:实现一个小算法
- Python:通过命令行发送新浪微博
- Python:通过摄像头实现的监控功能
- Python:通过摄像头抓取图像并自动上传至新浪微博
- Python:简单的摄像头程序实现
- Python:日志模块logging的应用
- Python:操作嵌入式数据库SQLite
- Python:将句子中的单词全部倒排过来,但单词的字母顺序不变
- Python:语音处理,实现在线朗读RFC文档或本地文本文件
- Python:通过计算阶乘来学习lambda和reduce这两个函数的使用
- Python:通过执行100万次打印来比较C和python的性能,以及用C和python结合来解决性能问题的方法
- Python:使用matplotlib绘制图表
- Python:使用pycha快速绘制办公常用图(饼图、垂直直方图、水平直方图、散点图等七种图形)
- Python:使用pycha快速绘制办公常用图二(使用样式定制个性化图表)
- Python:监控键盘输入、鼠标操作,并将捕获到的信息记录到文件中
- Python:通过获取淘宝账号和密码的实验,来看登陆方式选择的重要性
- Python:通过获取淘宝账号和密码的实验,来看登陆方式选择的重要性(二)
- Python:通过远程监控用户输入来获取淘宝账号和密码的实验(一)
- Python:通过远程监控用户输入来获取淘宝账号和密码的实验(二)
- Python:通过自定义系统级快捷键来控制程序运行
- Python:通过自定义系统级快捷键来控制程序开始或停止记录日志(使用小技巧解决一个貌似无解的问题)
- Python:一个多功能的抓图工具开发(附源码)
- Python:程序发布方式简介一(打包为可执行文件EXE)
- Python:新浪微博应用开发简介(认证及授权部分)
- Python:程序最小化到托盘功能实现
- Python:实用抓图工具开发介绍(含需求分析、设计、编码、单元测试、打包、系统测试、发布各环节)
- Python:桌面气泡提示功能实现
- Python:未来三个月的python学习计划
- Python:pygame模块及SDL库简介
- Python:获取新浪微博用户的收听列表和粉丝列表
- Python:pygame游戏编程之旅一(Hello World)
- Python:pygame游戏编程之旅二(自由移动的小球)
- Python:pygame游戏编程之旅三(玩家控制的小球)
- Python:pygame游戏编程之旅四(游戏界面文字处理)
- Python:pygame游戏编程之旅五(游戏界面文字处理详解)
- Python:pygame游戏编程之旅六(游戏中的声音处理)
- Python:pygame游戏编程之旅七(pygame基础知识讲解1)
- Python:编程“八荣八耻”之我见
- Python:脚本的几种执行方式
- wxPython:简单的wxPython程序
- wxPython:简单的wxPython程序的另一种写法
- wxPython:应用程序对象介绍
- wxPython:输出重定向
- wxPython:关闭wxPython程序
- wxPython:Frame类介绍
- wxPython:面板Panel的使用
- wxPython:工具栏、状态栏、菜单实现
- wxPython:消息对话框MessageDialog
- wxPython:文本对话框TextEntryDialog
- wxPython:列表选择框SingleChoiceDialog
- wxPython:事件处理介绍一
- wxPython:事件处理介绍二
- wxPython: 简单的绘图例子
- wxPython:状态栏介绍
- wxPython:菜单介绍
- wxPython:文件对话框wx.FileDialog
- wxPython:颜色选择对话框wx.ColourDialog
- wxPython:布局管理器sizer介绍
- wxPython:启动画面SplashScreen介绍
- wxPython:绘画按钮BitmapButton介绍
- wxPython:进度条Gauge介绍
- Python: 发送新浪微博(使用oauth2)
- Python:读取新浪微博收听列表
- Python:DNS客户端实现