fixing previuos broken patch - commited too early...

mami-project · Aug 17, 2018 · 38d4852 · 38d4852
1 parent e2cb9ea
commit 38d4852
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pathspider/chains/dscp.py b/pathspider/chains/dscp.py
@@ -110,11 +110,11 @@ def _dscp_extract(self, rec, ip, rev):
                 rec['dscp_mark_syn_rev' if rev else 'dscp_mark_syn_fwd'] = dscp
                 return True
             if ip.version == 4:
-               ip.pkt_len == ip.hdr_len + (ip.tcp.doff) * 4: # No payload
-                  return True
+                if 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 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'