testdata/pymod.tdir/pymod.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 " % 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("www2.example.com.")):
 110         print (qstate.qinfo.qname_str)
 111
 112         qstate.ext_state[id] = MODULE_FINISHED
 113
 114         msg = DNSMessage(qstate.qinfo.qname_str, RR_TYPE_A, RR_CLASS_IN, PKT_QR | PKT_RA | PKT_AA) #, 300)
 115         #msg.authority.append("xxx.seznam.cz. 10 IN A 192.168.1.1")
 116         #msg.additional.append("yyy.seznam.cz. 10 IN A 1.1.1.2.")
 117
 118         # answer can be returned to the client without further checking.
 119
 120         if qstate.qinfo.qtype == RR_TYPE_A:
 121             msg.answer.append("%s 10 IN A 192.168.1.1" % qstate.qinfo.qname_str)
 122         if (qstate.qinfo.qtype == RR_TYPE_SRV) or (qstate.qinfo.qtype == RR_TYPE_ANY):
 123             msg.answer.append("%s 10 IN SRV 0 0 80 neinfo.example.com." % qstate.qinfo.qname_str)
 124         if (qstate.qinfo.qtype == RR_TYPE_TXT) or (qstate.qinfo.qtype == RR_TYPE_ANY):
 125             msg.answer.append("%s 10 IN TXT path=/" % qstate.qinfo.qname_str)
 126
 127         print(msg.answer)
 128         if not msg.set_return_msg(qstate):
 129             qstate.ext_state[id] = MODULE_ERROR
 130             return True
 131
 132         #qstate.return_msg.rep.security = 2 #pokud nebude nasledovat validator, je zapotrebi nastavit security, aby nebyl paket zahozen v mesh_send_reply
 133         printReturnMsg(qstate)
 134
 135         #Authoritative result can't be stored in cache
 136         #if (not storeQueryInCache(qstate, qstate.return_msg.qinfo, qstate.return_msg.rep, 0)):
 137         #    print "Can't store in cache"
 138         #    qstate.ext_state[id] = MODULE_ERROR
 139         #    return False
 140         #print "Store OK"
 141
 142         qstate.return_rcode = RCODE_NOERROR
 143         return True
 144
 145     if event == MODULE_EVENT_NEW:
 146         qstate.ext_state[id] = MODULE_WAIT_MODULE
 147         return True
 148
 149     if event == MODULE_EVENT_MODDONE:
 150         log_info("pythonmod: previous module done")
 151         qstate.ext_state[id] = MODULE_FINISHED
 152         return True
 153
 154     if event == MODULE_EVENT_PASS:
 155         log_info("pythonmod: event_pass")
 156         qstate.ext_state[id] = MODULE_WAIT_MODULE
 157         return True
 158
 159     log_err("pythonmod: BAD event")
 160     qstate.ext_state[id] = MODULE_ERROR
 161     return True
 162
 163 log_info("pythonmod: script loaded.")