diff options
Diffstat (limited to 'scripts/external_libs/dpkt-1.8.6.2/dpkt/llc.py')
| -rw-r--r-- | scripts/external_libs/dpkt-1.8.6.2/dpkt/llc.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/scripts/external_libs/dpkt-1.8.6.2/dpkt/llc.py b/scripts/external_libs/dpkt-1.8.6.2/dpkt/llc.py new file mode 100644 index 00000000..ce7429f6 --- /dev/null +++ b/scripts/external_libs/dpkt-1.8.6.2/dpkt/llc.py @@ -0,0 +1,56 @@ +# -*- coding: utf-8 -*- + +import struct +import dpkt +import stp +import ethernet + + +class LLC(dpkt.Packet): + _typesw = {} + + def _unpack_data(self, buf): + if self.type == ethernet.ETH_TYPE_8021Q: + self.tag, self.type = struct.unpack('>HH', buf[:4]) + buf = buf[4:] + elif self.type == ethernet.ETH_TYPE_MPLS or self.type == ethernet.ETH_TYPE_MPLS_MCAST: + # XXX - skip labels + for i in range(24): + if struct.unpack('>I', buf[i:i + 4])[0] & 0x0100: # MPLS_STACK_BOTTOM + break + self.type = ethernet.ETH_TYPE_IP + buf = buf[(i + 1) * 4:] + try: + self.data = self._typesw[self.type](buf) + setattr(self, self.data.__class__.__name__.lower(), self.data) + except (KeyError, dpkt.UnpackError): + self.data = buf + + def unpack(self, buf): + self.data = buf + if self.data.startswith('\xaa\xaa'): + # SNAP + self.type = struct.unpack('>H', self.data[6:8])[0] + self._unpack_data(self.data[8:]) + else: + # non-SNAP + dsap = ord(self.data[0]) + if dsap == 0x06: # SAP_IP + self.data = self.ip = self._typesw[ethernet.ETH_TYPE_IP](self.data[3:]) + elif dsap == 0x10 or dsap == 0xe0: # SAP_NETWARE{1,2} + self.data = self.ipx = self._typesw[ethernet.ETH_TYPE_IPX](self.data[3:]) + elif dsap == 0x42: # SAP_STP + self.data = self.stp = stp.STP(self.data[3:]) + + +def test_llc(): + s = '\xaa\xaa\x03\x00\x00\x00\x08\x00\x45\x00\x00\x28\x07\x27\x40\x00\x80\x06\x1d\x39\x8d\xd4\x37\x3d\x3f\xf5\xd1\x69\xc0\x5f\x01\xbb\xb2\xd6\xef\x23\x38\x2b\x4f\x08\x50\x10\x42\x04\xac\x17\x00\x00' + import ip + llc_pkt = LLC(s) + ip_pkt = ip.IP(llc_pkt.data) + assert (llc_pkt.type == ethernet.ETH_TYPE_IP) + assert (ip_pkt.dst == '\x3f\xf5\xd1\x69') + +if __name__ == '__main__': + test_llc() + print 'Tests Successful...'
\ No newline at end of file |