testdata/pymod_thread.tdir/pymod_thread.py

   1 # -*- coding: utf-8 -*-
   2 '''
   3  ubmodule-msg.py: simple response packet logger
   4
   5  Authors: Zdenek Vasicek (vasicek AT fit.vutbr.cz)
   6           Marek Vavrusa  (xvavru00 AT stud.fit.vutbr.cz)
   7
   8  Copyright (c) 2008. All rights reserved.
   9
  10  This software is open source.
  11
  12  Redistribution and use in source and binary forms, with or without
  13  modification, are permitted provided that the following conditions
  14  are met:
  15
  16  Redistributions of source code must retain the above copyright notice,
  17  this list of conditions and the following disclaimer.
  18
  19  Redistributions in binary form must reproduce the above copyright notice,
  20  this list of conditions and the following disclaimer in the documentation
  21  and/or other materials provided with the distribution.
  22
  23  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  24  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  25  TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  26  PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE
  27  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  28  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  29  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  30  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  31  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  32  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  33  POSSIBILITY OF SUCH DAMAGE.
  34
  35  Modified for unit test by Wouter Wijngaards, NLnet Labs, 2009.
  36 '''
  37 import os
  38
  39 def init(id, cfg):
  40     log_info("pythonmod: init called, module id is %d port: %d script: %s" % (id, cfg.port, mod_env['script']))
  41     return True
  42
  43 def deinit(id):
  44     log_info("pythonmod: deinit called, module id is %d" % id)
  45     return True
  46
  47 def inform_super(id, qstate, superqstate, qdata):
  48     return True
  49
  50 def setTTL(qstate, ttl):
  51     """Sets return_msg TTL and all the RRs TTL"""
  52     if qstate.return_msg:
  53         qstate.return_msg.rep.ttl = ttl
  54         if (qstate.return_msg.rep):
  55             for i in range(0,qstate.return_msg.rep.rrset_count):
  56                 d = qstate.return_msg.rep.rrsets[i].entry.data
  57                 for j in range(0,d.count+d.rrsig_count):
  58                     d.rr_ttl[j] = ttl
  59
  60 def dataHex(data, prefix=""):
  61     res = ""
  62     for i in range(0, int((len(data)+15)/16)):
  63         res += "%s0x%02X | " % (prefix, i*16)
  64         if type(data[0]) == type(1):
  65             d = map(lambda x:int(x), data[i*16:i*16+17])
  66         else:
  67             d = map(lambda x:ord(x), data[i*16:i*16+17])
  68         for ch in d:
  69             res += "%02X " % int(ch)
  70         for i in range(0,17-len(data[i*16:i*16+17])):
  71             res += "   "
  72         res += "| "
  73         for ch in d:
  74             if (ch < 32) or (ch > 127):
  75                 res += ". "
  76             else:
  77                 res += "%c " % ch
  78         res += "\n"
  79     return res
  80
  81 def printReturnMsg(qstate):
  82     print ("Return MSG rep   :: flags: %04X, QDcount: %d, Security:%d, TTL=%d" % (qstate.return_msg.rep.flags, qstate.return_msg.rep.qdcount, qstate.return_msg.rep.security, qstate.return_msg.rep.ttl))
  83     print ("           qinfo :: qname:",qstate.return_msg.qinfo.qname_list, qstate.return_msg.qinfo.qname_str, "type:",qstate.return_msg.qinfo.qtype_str, "class:",qstate.return_msg.qinfo.qclass_str)
  84     if (qstate.return_msg.rep):
  85         print ("RRSets:",qstate.return_msg.rep.rrset_count)
  86         prevkey = None
  87         for i in range(0,qstate.return_msg.rep.rrset_count):
  88             r = qstate.return_msg.rep.rrsets[i]
  89             rk = r.rk
  90             print (i,":",rk.dname_list, rk.dname_str, "flags: %04X" % rk.flags)
  91             print ("type:",rk.type_str,"(%d)" % ntohs(rk.type), "class:",rk.rrset_class_str,"(%d)" % ntohs(rk.rrset_class))
  92
  93             d = r.entry.data
  94             print ("    RRDatas:",d.count+d.rrsig_count)
  95             for j in range(0,d.count+d.rrsig_count):
  96                 print ("    ",j,":","TTL=",d.rr_ttl[j],"RR data:")
  97                 print (dataHex(d.rr_data[j],"         "))
  98
  99
 100 def operate(id, event, qstate, qdata):
 101     log_info("pythonmod: operate called, id: %d, event:%s" % (id, strmodulevent(event)))
 102     #print ("pythonmod: per query data", qdata)
 103
 104     print ("Query:", qstate.qinfo.qname, qstate.qinfo.qname_list, qstate.qinfo.qname_str)
 105     print ("Type:",qstate.qinfo.qtype_str,"(%d)" % qstate.qinfo.qtype)
 106     print ("Class:",qstate.qinfo.qclass_str,"(%d)" % qstate.qinfo.qclass)
 107     print ()
 108
 109     if (event == MODULE_EVENT_NEW or event == MODULE_EVENT_PASS) and (qstate.qinfo.qname_str.endswith("example.com.")):
 110         print (qstate.qinfo.qname_str)
 111
 112         qstate.ext_state[id] = MODULE_FINISHED
 113
 114         # eat time
 115         y = 20
 116         for z in range(2, 10000):
 117                 y = y*2 - z/2
 118                 y = y/2 + z
 119
 120         msg = DNSMessage(qstate.qinfo.qname_str, RR_TYPE_A, RR_CLASS_IN, PKT_QR | PKT_RA | PKT_AA) #, 300)
 121         #msg.authority.append("xxx.seznam.cz. 10 IN A 192.168.1.1")
 122         #msg.additional.append("yyy.seznam.cz. 10 IN A 1.1.1.2.")
 123
 124         # answer can be returned to the client without further checking.
 125
 126         if qstate.qinfo.qtype == RR_TYPE_A:
 127             msg.answer.append("%s 10 IN A 192.168.1.1" % qstate.qinfo.qname_str)
 128         if (qstate.qinfo.qtype == RR_TYPE_SRV) or (qstate.qinfo.qtype == RR_TYPE_ANY):
 129             msg.answer.append("%s 10 IN SRV 0 0 80 neinfo.example.com." % qstate.qinfo.qname_str)
 130         if (qstate.qinfo.qtype == RR_TYPE_TXT) or (qstate.qinfo.qtype == RR_TYPE_ANY):
 131             msg.answer.append("%s 10 IN TXT path=/" % qstate.qinfo.qname_str)
 132
 133         if not msg.set_return_msg(qstate):
 134             qstate.ext_state[id] = MODULE_ERROR
 135             return True
 136
 137         #qstate.return_msg.rep.security = 2 #pokud nebude nasledovat validator, je zapotrebi nastavit security, aby nebyl paket zahozen v mesh_send_reply
 138         printReturnMsg(qstate)
 139
 140         #Authoritative result can't be stored in cache
 141         #if (not storeQueryInCache(qstate, qstate.return_msg.qinfo, qstate.return_msg.rep, 0)):
 142         #    print "Can't store in cache"
 143         #    qstate.ext_state[id] = MODULE_ERROR
 144         #    return False
 145         #print "Store OK"
 146
 147         qstate.return_rcode = RCODE_NOERROR
 148         return True
 149
 150     if event == MODULE_EVENT_NEW:
 151         qstate.ext_state[id] = MODULE_WAIT_MODULE
 152         return True
 153
 154     if event == MODULE_EVENT_MODDONE:
 155         log_info("pythonmod: previous module done")
 156         qstate.ext_state[id] = MODULE_FINISHED
 157         return True
 158
 159     if event == MODULE_EVENT_PASS:
 160         log_info("pythonmod: event_pass")
 161         qstate.ext_state[id] = MODULE_WAIT_MODULE
 162         return True
 163
 164     log_err("pythonmod: BAD event")
 165     qstate.ext_state[id] = MODULE_ERROR
 166     return True
 167
 168 log_info("pythonmod: script loaded.")