IPv6 doesn't have a hdr length...

Fix to isssue #234
mami-project · Aug 17, 2018 · e2cb9ea · e2cb9ea
1 parent 1c2122f
commit e2cb9ea
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pathspider/chains/dscp.py b/pathspider/chains/dscp.py
@@ -109,8 +109,12 @@ def _dscp_extract(self, rec, ip, rev):
             if ip.tcp.flags & TCP_SYN == TCP_SYN:
                 rec['dscp_mark_syn_rev' if rev else 'dscp_mark_syn_fwd'] = dscp
                 return True
-            if ip.pkt_len == (ip.hdr_len + ip.tcp.doff) * 4: # No payload
-                return True
+            if ip.version == 4:
+               ip.pkt_len == ip.hdr_len + (ip.tcp.doff) * 4: # No payload
+                  return True
+            elif ip.version == 6:
+               ip.payload_len == (ip.tcp.doff) * 4: # No payload
+                  return True
 
         # If not TCP or TCP non-SYN
         data_key = 'dscp_mark_data_rev' if rev else 'dscp_mark_data_fwd'