1 files changed, 55 insertions, 0 deletions
diff --git a/scripts/external_libs/dpkt-1.8.6/dpkt/llc.py b/scripts/external_libs/dpkt-1.8.6/dpkt/llc.py
new file mode 100644
index 00000000..28246131
--- /dev/null
+++ b/scripts/external_libs/dpkt-1.8.6/dpkt/llc.py
@@ -0,0 +1,55 @@
+import struct
+import dpkt, stp, 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:])
+
+if __name__ == '__main__':
+    import unittest
+    
+    class LLCTestCase(unittest.TestCase):
+
+        def test_llc(self):
+            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)
+            self.failUnless(llc_pkt.type == ethernet.ETH_TYPE_IP)
+            self.failUnless(ip_pkt.dst == '\x3f\xf5\xd1\x69')
+
+    unittest.main()