Python简单DNS服务实现代码

#-*-coding:gb2312-*-
#
# 功能: DNS服务
# 作者: Ease
# 日期: 2009-10-23
# QQ: 12114666
#说明: 部份源码来自网络. 这里仅是在本机作一个中转.使用时将其它电脑DNS设为本机IP即可.
#    扩展功能,你可以过滤某些域名的DNS请求, 请求过的地址,可以存入数据库中缓存
#    对多机同时请求未作过多测试,负载大时需要对程序作多线等处理.
#    可以通过此代码来扩展二级动态域名,类似于花生壳.

import socket
class DNSQuery:
def __init__(self, data):
    self.data=data
    self.dominio=''
   
    tipo = (ord(data[2]) >> 3) & 15   # Opcode bits
    if tipo == 0:                     # Standard query
      ini=12
      lon=ord(data[ini])              #获取域名DNS请求
      while lon != 0:
        self.dominio+=data[ini+1:ini+lon+1]+'.'
        ini+=lon+1
        lon=ord(data[ini])
#返回包含IP的DNS数据包
def respuesta(self, ip):
    packet=''
    if self.dominio:
      packet+=self.data[:2] + "x81x80"
      packet+=self.data[4:6] + self.data[4:6] + 'x00x00x00x00'   # Questions and Answers Counts
      packet+=self.data[12:]                                         # Original Domain Name Question
      packet+='xc0x0c'                                             # Pointer to domain name
      packet+='x00x01x00x01x00x00x00x3cx00x04'             # Response type, ttl and resource data length -> 4 bytes
      packet+=str.join('',map(lambda x: chr(int(x)), ip.split('.'))) # 4bytes of IP
    return packet
#向服务器请求IP
def request(self):
    self.ip=''
    tmp=''
    if self.dominio:
       try: s=socket.getaddrinfo(self.dominio,None)
       except: tmp=''
       tmp=s[0][4][0]
    return tmp
if __name__ == '__main__':
udps = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
udps.bind(('',53))

try:
    while 1:
      data, addr = udps.recvfrom(1024)
      p=DNSQuery(data)
      ip=p.request()
      udps.sendto(p.respuesta(ip), addr)
      print 'Respuesta: %s -> %s' % (p.dominio, ip)
except KeyboardInterrupt:
      print 'Finalizando'
      udps.close()