1 files changed, 243 insertions, 0 deletions
diff --git a/external_libs/python/dpkt-1.8.6.2/dpkt/diameter.py b/external_libs/python/dpkt-1.8.6.2/dpkt/diameter.py
new file mode 100644
index 00000000..a6994276
--- /dev/null
+++ b/external_libs/python/dpkt-1.8.6.2/dpkt/diameter.py
@@ -0,0 +1,243 @@
+# $Id: diameter.py 23 2006-11-08 15:45:33Z dugsong $
+# -*- coding: utf-8 -*-
+"""Diameter."""
+
+import struct
+import dpkt
+from decorators import deprecated
+
+# Diameter Base Protocol - RFC 3588
+# http://tools.ietf.org/html/rfc3588
+
+# Request/Answer Command Codes
+ABORT_SESSION = 274
+ACCOUTING = 271
+CAPABILITIES_EXCHANGE = 257
+DEVICE_WATCHDOG = 280
+DISCONNECT_PEER = 282
+RE_AUTH = 258
+SESSION_TERMINATION = 275
+
+
+class Diameter(dpkt.Packet):
+    __hdr__ = (
+        ('v', 'B', 1),
+        ('len', '3s', 0),
+        ('flags', 'B', 0),
+        ('cmd', '3s', 0),
+        ('app_id', 'I', 0),
+        ('hop_id', 'I', 0),
+        ('end_id', 'I', 0)
+    )
+
+    @property
+    def request_flag(self):
+        return (self.flags >> 7) & 0x1
+
+    @request_flag.setter
+    def request_flag(self, r):
+        self.flags = (self.flags & ~0x80) | ((r & 0x1) << 7)
+
+    @property
+    def proxiable_flag(self):
+        return (self.flags >> 6) & 0x1
+
+    @proxiable_flag.setter
+    def proxiable_flag(self, p):
+        self.flags = (self.flags & ~0x40) | ((p & 0x1) << 6)
+
+    @property
+    def error_flag(self):
+        return (self.flags >> 5) & 0x1
+
+    @error_flag.setter
+    def error_flag(self, e):
+        self.flags = (self.flags & ~0x20) | ((e & 0x1) << 5)
+
+    @property
+    def retransmit_flag(self):
+        return (self.flags >> 4) & 0x1
+
+    @retransmit_flag.setter
+    def retransmit_flag(self, t):
+        self.flags = (self.flags & ~0x10) | ((t & 0x1) << 4)
+
+    # Deprecated methods, will be removed in the future
+    # ======================================================
+    @deprecated
+    def _get_r(self): return self.request_flag
+
+    @deprecated
+    def _set_r(self, r): self.request_flag = r
+
+    @deprecated
+    def _get_p(self): return self.proxiable_flag
+
+    @deprecated
+    def _set_p(self, p): self.proxiable_flag = p
+
+    @deprecated
+    def _get_e(self): return self.error_flag
+
+    @deprecated
+    def _set_e(self, e): self.error_flag = e
+
+    @deprecated
+    def _get_t(self): return self.request_flag
+
+    @deprecated
+    def _set_t(self, t): self.request_flag = t
+
+    # ======================================================
+
+    def unpack(self, buf):
+        dpkt.Packet.unpack(self, buf)
+        self.cmd = (ord(self.cmd[0]) << 16) | (ord(self.cmd[1]) << 8) | (ord(self.cmd[2]))
+        self.len = (ord(self.len[0]) << 16) | (ord(self.len[1]) << 8) | (ord(self.len[2]))
+        self.data = self.data[:self.len - self.__hdr_len__]
+
+        l = []
+        while self.data:
+            avp = AVP(self.data)
+            l.append(avp)
+            self.data = self.data[len(avp):]
+        self.data = self.avps = l
+
+    def pack_hdr(self):
+        self.len = chr((self.len >> 16) & 0xff) + chr((self.len >> 8) & 0xff) + chr(self.len & 0xff)
+        self.cmd = chr((self.cmd >> 16) & 0xff) + chr((self.cmd >> 8) & 0xff) + chr(self.cmd & 0xff)
+        return dpkt.Packet.pack_hdr(self)
+
+    def __len__(self):
+        return self.__hdr_len__ + sum(map(len, self.data))
+
+    def __str__(self):
+        return self.pack_hdr() + ''.join(map(str, self.data))
+
+
+class AVP(dpkt.Packet):
+    __hdr__ = (
+        ('code', 'I', 0),
+        ('flags', 'B', 0),
+        ('len', '3s', 0),
+    )
+
+    @property
+    def vendor_flag(self):
+        return (self.flags >> 7) & 0x1
+
+    @vendor_flag.setter
+    def vendor_flag(self, v):
+        self.flags = (self.flags & ~0x80) | ((v & 0x1) << 7)
+
+    @property
+    def mandatory_flag(self):
+        return (self.flags >> 6) & 0x1
+
+    @mandatory_flag.setter
+    def mandatory_flag(self, m):
+        self.flags = (self.flags & ~0x40) | ((m & 0x1) << 6)
+
+    @property
+    def protected_flag(self):
+        return (self.flags >> 5) & 0x1
+
+    @protected_flag.setter
+    def protected_flag(self, p):
+        self.flags = (self.flags & ~0x20) | ((p & 0x1) << 5)
+
+    # Deprecated methods, will be removed in the future
+    # ======================================================
+    @deprecated
+    def _get_v(self):
+        return self.vendor_flag
+
+    @deprecated
+    def _set_v(self, v):
+        self.vendor_flag = v
+
+    @deprecated
+    def _get_m(self):
+        return self.mandatory_flag
+
+    @deprecated
+    def _set_m(self, m):
+        self.mandatory_flag = m
+
+    @deprecated
+    def _get_p(self):
+        return self.protected_flag
+
+    @deprecated
+    def _set_p(self, p):
+        self.protected_flag = p
+
+    # ======================================================
+
+    def unpack(self, buf):
+        dpkt.Packet.unpack(self, buf)
+        self.len = (ord(self.len[0]) << 16) | (ord(self.len[1]) << 8) | (ord(self.len[2]))
+
+        if self.vendor_flag:
+            self.vendor = struct.unpack('>I', self.data[:4])[0]
+            self.data = self.data[4:self.len - self.__hdr_len__]
+        else:
+            self.data = self.data[:self.len - self.__hdr_len__]
+
+    def pack_hdr(self):
+        self.len = chr((self.len >> 16) & 0xff) + chr((self.len >> 8) & 0xff) + chr(self.len & 0xff)
+        data = dpkt.Packet.pack_hdr(self)
+        if self.vendor_flag:
+            data += struct.pack('>I', self.vendor)
+        return data
+
+    def __len__(self):
+        length = self.__hdr_len__ + sum(map(len, self.data))
+        if self.vendor_flag:
+            length += 4
+        return length
+
+
+__s = '\x01\x00\x00\x28\x80\x00\x01\x18\x00\x00\x00\x00\x00\x00\x41\xc8\x00\x00\x00\x0c\x00\x00\x01\x08\x40\x00\x00\x0c\x68\x30\x30\x32\x00\x00\x01\x28\x40\x00\x00\x08'
+__t = '\x01\x00\x00\x2c\x80\x00\x01\x18\x00\x00\x00\x00\x00\x00\x41\xc8\x00\x00\x00\x0c\x00\x00\x01\x08\xc0\x00\x00\x10\xde\xad\xbe\xef\x68\x30\x30\x32\x00\x00\x01\x28\x40\x00\x00\x08'
+
+
+def test_pack():
+    d = Diameter(__s)
+    assert (__s == str(d))
+    d = Diameter(__t)
+    assert (__t == str(d))
+
+
+def test_unpack():
+    d = Diameter(__s)
+    assert (d.len == 40)
+    # assert (d.cmd == DEVICE_WATCHDOG_REQUEST)
+    assert (d.request_flag == 1)
+    assert (d.error_flag == 0)
+    assert (len(d.avps) == 2)
+
+    avp = d.avps[0]
+    # assert (avp.code == ORIGIN_HOST)
+    assert (avp.mandatory_flag == 1)
+    assert (avp.vendor_flag == 0)
+    assert (avp.len == 12)
+    assert (len(avp) == 12)
+    assert (avp.data == '\x68\x30\x30\x32')
+
+    # also test the optional vendor id support
+    d = Diameter(__t)
+    assert (d.len == 44)
+    avp = d.avps[0]
+    assert (avp.vendor_flag == 1)
+    assert (avp.len == 16)
+    assert (len(avp) == 16)
+    assert (avp.vendor == 3735928559)
+    assert (avp.data == '\x68\x30\x30\x32')
+
+
+if __name__ == '__main__':
+    test_pack()
+    test_unpack()
+    print 'Tests Successful...'
+