Python:DNS客户端实现

 这两天在测试IPV6与DNS，晚上查了下资料，写了个DNS客户端，其中请求消息编码部分参考网上实例。

  刚完成了编码，功能还没有测试。

1、代码：

[python]  view plain copy print ?

1. #!/usr/bin/env python  
2. # -*- coding: utf-8 -*-  
3.   
4. import ConfigParser  
5. import socket  
6. import time     
7. import string  
8. import os  
9. import re  
10. import struct  
11.   
12. def dns_codec(hostname):  
13.     ''''' 
14.     Function:请求消息编码 
15.     Input：hostname：主机名，如www.baidu.com 
16.     Output: 编码后的字节流 
17.     author: socrates 
18.     date:2012-12-14 
19.     '''  
20.     index = os.urandom(2)  
21.     hoststr = ''.join(chr(len(x))+x for x in hostname.split('.'))  
22.     data = '%s\x01\x00\x00\x01\x00\x00\x00\x00\x00\x00%s\x00\x00\x01\x00\x01' % (index, hoststr)  
23.     data = struct.pack('!H', len(data)) + data  
24.     return data  
25.   
26. def dns_decode(in_sock):  
27.     ''''' 
28.     Function:响应消息解码 
29.     Input：in_sock：接收消息的socket 
30.     Output:解码后的内容 
31.     author: socrates 
32.     date:2012-12-14 
33.     '''  
34.     rfile = in_sock.makefile('rb')  
35.     size = struct.unpack('!H', rfile.read(2))[0]  
36.     data = rfile.read(size)  
37.     iplist = re.findall('\xC0.\x00\x01\x00\x01.{6}(.{4})', data)  
38.     return ['.'.join(str(ord(x)) for x in s) for s in iplist]    
39.       
40. def dns_sendmsg():  
41.     ''''' 
42.     Function:通过socket发送DNS查询消息 
43.     Input：None 
44.     Output:None 
45.     author: socrates 
46.     date:2012-12-14 
47.     '''  
48.     ens_client_config = ConfigParser.ConfigParser()  
49.       
50.     #读取配置文件  
51.     try:  
52.         ens_client_config.readfp(open('ens_client_config.ini'))  
53.     except ConfigParser.Error:  
54.         print 'read ens_client_config.ini failed.'  
55.       
56.     #获取需要的信息  
57.     server_ip_1 = ens_client_config.get("server_info","ip_1")  
58.     server_port_1 = ens_client_config.get("server_info","port_1")  
59.     sockettype_1 = ens_client_config.get("server_info","sockettype_1")  
60.     heartbeat_1 = ens_client_config.get("server_info","heartbeat_1")  
61.     msg_1 = ens_client_config.get("server_info","msg_1")  
62.       
63.     #IP类型  
64.     address_family = {True:socket.AF_INET6, False:socket.AF_INET}[':' in server_ip_1]  
65.     #传输类型  
66.     socket_type = {True:socket.SOCK_STREAM , False:socket.SOCK_DGRAM}['TCP' == sockettype_1.upper()]  
67.       
68.     try:  
69.         sock = socket.socket(address_family, socket_type)    
70.     except socket.error, e:  
71.         print 'create socket return error. errno = %d, errmsg = %s' % (e.args[0], e.args[1])  
72.       
73.     #连接服务器并发送消息          
74.     try:  
75.         #连接服务端  
76.         sock.connect((server_ip_1, string.atoi(server_port_1)))    
77.           
78.         while True:  
79.               
80.             #发送频率  
81.             time.sleep(string.atoi(heartbeat_1))   
82.               
83.             #发送消息   
84.             sock.sendall(dns_codec(msg_1))    
85.               
86.             #接收并打印消息  
87.             print dns_decode(sock)    
88.               
89.     except socket.error, e:    
90.         print 'connect server failed. errno = %d, errmsg = %s' % (e.args[0], e.args[1])  
91.                
92.     sock.close()       
93.       
94.       
95. if __name__ == '__main__':   
96.      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抓包：

请求：

响应：

b、代码运行结果：