home *** CD-ROM | disk | FTP | other *** search
- # Source Generated with Decompyle++
- # File: in.pyc (Python 2.6)
-
- import socket
- import struct
- import dns.ipv4 as dns
- import dns.rdata as dns
- _proto_tcp = socket.getprotobyname('tcp')
- _proto_udp = socket.getprotobyname('udp')
-
- class WKS(dns.rdata.Rdata):
- __slots__ = [
- 'address',
- 'protocol',
- 'bitmap']
-
- def __init__(self, rdclass, rdtype, address, protocol, bitmap):
- super(WKS, self).__init__(rdclass, rdtype)
- self.address = address
- self.protocol = protocol
- self.bitmap = bitmap
-
-
- def to_text(self, origin = None, relativize = True, **kw):
- bits = []
- for i in xrange(0, len(self.bitmap)):
- byte = ord(self.bitmap[i])
- for j in xrange(0, 8):
- if byte & 128 >> j:
- bits.append(str(i * 8 + j))
- continue
-
-
- text = ' '.join(bits)
- return '%s %d %s' % (self.address, self.protocol, text)
-
-
- def from_text(cls, rdclass, rdtype, tok, origin = None, relativize = True):
- address = tok.get_string()
- protocol = tok.get_string()
- if protocol.isdigit():
- protocol = int(protocol)
- else:
- protocol = socket.getprotobyname(protocol)
- bitmap = []
- while None:
- (ttype, value) = tok.get()
- if ttype == dns.tokenizer.EOL or ttype == dns.tokenizer.EOF:
- break
-
- if value.isdigit():
- serv = int(value)
- elif protocol != _proto_udp and protocol != _proto_tcp:
- raise NotImplementedError, 'protocol must be TCP or UDP'
-
- if protocol == _proto_udp:
- protocol_text = 'udp'
- else:
- protocol_text = 'tcp'
- serv = socket.getservbyname(value, protocol_text)
- i = serv // 8
- l = len(bitmap)
- if l < i + 1:
- for j in xrange(l, i + 1):
- bitmap.append('\x00')
-
-
- bitmap[i] = chr(ord(bitmap[i]) | 128 >> serv % 8)
- continue
- bitmap = dns.rdata._truncate_bitmap(bitmap)
- return cls(rdclass, rdtype, address, protocol, bitmap)
-
- from_text = classmethod(from_text)
-
- def to_wire(self, file, compress = None, origin = None):
- file.write(dns.ipv4.inet_aton(self.address))
- protocol = struct.pack('!B', self.protocol)
- file.write(protocol)
- file.write(self.bitmap)
-
-
- def from_wire(cls, rdclass, rdtype, wire, current, rdlen, origin = None):
- address = dns.ipv4.inet_ntoa(wire[current:current + 4])
- (protocol,) = struct.unpack('!B', wire[current + 4:current + 5])
- current += 5
- rdlen -= 5
- bitmap = wire[current:current + rdlen]
- return cls(rdclass, rdtype, address, protocol, bitmap)
-
- from_wire = classmethod(from_wire)
-
- def _cmp(self, other):
- sa = dns.ipv4.inet_aton(self.address)
- oa = dns.ipv4.inet_aton(other.address)
- v = cmp(sa, oa)
- if v == 0:
- sp = struct.pack('!B', self.protocol)
- op = struct.pack('!B', other.protocol)
- v = cmp(sp, op)
- if v == 0:
- v = cmp(self.bitmap, other.bitmap)
-
-
- return v
-
-
-