traffic-taffy 0.9.2__tar.gz → 0.9.4__tar.gz

{traffic_taffy-0.9.2 → traffic_taffy-0.9.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: traffic-taffy
-Version: 0.9.2
+Version: 0.9.4
 Summary: A tool for doing differential analysis of pcap files
 Project-URL: Homepage, https://traffic-taffy.github.io/
 Author-email: Wes Hardaker <opensource@hardakers.net>
@@ -56,6 +56,12 @@ might be coming that will cause merge conflicts.
 
 # Copyright and License
 
+Traffic-taffy was created by [Wes Hardaker], a computer scientist at
+[USC/ISI], with support from the Comcast Innovation Fund.
+
+[Wes Hardaker]: https://ant.isi.edu/~hardaker/
+[USC/ISI]: https://www.isi.edu/
+
 This project is copyrighted by the University of Southern California,
 Information Sciences institute.  It is released under the Apache 2.0
 license.

{traffic_taffy-0.9.2 → traffic_taffy-0.9.4}/README.md

@@ -28,6 +28,12 @@ might be coming that will cause merge conflicts.
 
 # Copyright and License
 
+Traffic-taffy was created by [Wes Hardaker], a computer scientist at
+[USC/ISI], with support from the Comcast Innovation Fund.
+
+[Wes Hardaker]: https://ant.isi.edu/~hardaker/
+[USC/ISI]: https://www.isi.edu/
+
 This project is copyrighted by the University of Southern California,
 Information Sciences institute.  It is released under the Apache 2.0
 license.

traffic_taffy-0.9.4/traffic_taffy/__init__.py

@@ -0,0 +1 @@
+__VERSION__ = "0.9.4"

{traffic_taffy-0.9.2 → traffic_taffy-0.9.4}/traffic_taffy/algorithms/statistical.py

@@ -64,7 +64,10 @@ class ComparisonStatistical(ComparisonSlicesAlgorithm):
                     left_count = 0
                     right_count = right_side[key][subkey]
                     left_percentage = 0.0
-                    right_percentage = right_side[key][subkey] / right_side_total
+                    if right_side_total == 0:
+                        right_percentage = 1.0
+                    else:
+                        right_percentage = right_side[key][subkey] / right_side_total
                     new_right_count += 1  # this value wasn't in the left
 
                     report[key][subkey] = CompareSlicesReport(
@@ -78,12 +81,12 @@ class ComparisonStatistical(ComparisonSlicesAlgorithm):
                     )
 
             if right_side_total == 0:
-                right_percent = 100
+                right_percent = 1.0
             else:
                 right_percent = new_right_count / right_side_total
 
             if left_side_total == 0:
-                left_percent = 100
+                left_percent = 1.0
             else:
                 left_percent = new_left_count / left_side_total
 

{traffic_taffy-0.9.2 → traffic_taffy-0.9.4}/traffic_taffy/dissector_engine/dpkt.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from logging import debug
+from logging import debug, error
 from traffic_taffy.dissector_engine import DissectionEngine
 from traffic_taffy.dissection import Dissection, PCAPDissectorLevel
 from pcap_parallel import PCAPParallel
@@ -20,6 +20,7 @@ class DissectionEngineDpkt(DissectionEngine):
     def __init__(self, *args: list, **kwargs: dict):
         """Create a dissection engine for quickly parsing and counting packets."""
         super().__init__(*args, **kwargs)
+        self.data_link_type = None
 
     def load_data(self) -> None:
         """Load the specified PCAP into memory."""
@@ -29,6 +30,9 @@ class DissectionEngineDpkt(DissectionEngine):
         else:
             # it's an open handle already
             pcap = dpkt.pcap.Reader(self.pcap_file)
+
+        self.data_link_type = pcap.datalink()
+
         if self.pcap_filter:
             pcap.setfilter(self.pcap_filter)
         pcap.dispatch(self.maximum_count, self.callback)
@@ -144,14 +148,30 @@ class DissectionEngineDpkt(DissectionEngine):
             level = level.value
 
         if level >= PCAPDissectorLevel.THROUGH_IP.value:
-            eth = dpkt.ethernet.Ethernet(packet)
-            # these names are designed to match scapy names
-            self.incr("Ethernet_dst", eth.dst)
-            self.incr("Ethernet_src", eth.src)
-            self.incr("Ethernet_type", eth.type)
-
-            if isinstance(eth.data, dpkt.ip.IP):
-                ip = eth.data
+            if self.data_link_type == 1:
+                # Ethernet based encapsulation
+                eth = dpkt.ethernet.Ethernet(packet)
+                # these names are designed to match scapy names
+                self.incr("Ethernet_dst", eth.dst)
+                self.incr("Ethernet_src", eth.src)
+                self.incr("Ethernet_type", eth.type)
+                data = eth.data
+            elif self.data_link_type == 101:
+                # Raw IP encapsulation
+                if packet[0] == 0x45:
+                    data = dpkt.ip.IP(packet)
+                elif packet[0] == 0x60:
+                    data = dpkt.ip6.IP6(packet)
+                else:
+                    error("Unknown IP version in data")
+                    raise ValueError("unknown IP version")
+            else:
+                error(f"unknown link type: {self.data_link_type}")
+                raise ValueError("unknown link type")
+
+            # TODO(hardaker): add ip6.IP6 support
+            if isinstance(data, dpkt.ip.IP):
+                ip = data
                 udp = None
                 tcp = None
 

traffic_taffy-0.9.4/traffic_taffy/report.py

@@ -0,0 +1,12 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class Report:
+    delta_percentage: float
+    delta_absolute: int
+    total: int
+    left_count: int
+    right_count: int
+    left_percentage: float
+    right_percentage: float

traffic_taffy-0.9.4/traffic_taffy/tests/test_dpkt_engine.py

@@ -0,0 +1,15 @@
+import os
+from traffic_taffy.dissection import PCAPDissectorLevel
+from traffic_taffy.dissector_engine.dpkt import DissectionEngineDpkt
+
+def test_dpkt_engine():
+    test_pcap = "dns.pcap"
+    test_pcap = "port53-2023-30-31_20.pcap"
+    test_pcap = "airplane-wireless.pcap"
+    if not os.path.exists(test_pcap):
+        return
+
+    engine = DissectionEngineDpkt(test_pcap,
+                                  dissector_level = PCAPDissectorLevel.COMMON_LAYERS)
+    dissection = engine.load()
+

{traffic_taffy-0.9.2 → traffic_taffy-0.9.4}/traffic_taffy/tools/config.py

@@ -33,48 +33,48 @@ except ModuleNotFoundError:
     logging.debug("psl module not loadable")
 
 
+def taffy_config_parse_args() -> Namespace:
+    """Parse the command line arguments."""
+
+    config: TaffyConfig = TaffyConfig()
+    config.config_option_names = ["-y", "--config"]
+    config[TT_CFG.LOG_LEVEL] = "info"
+
+    config.read_configfile_from_arguments(sys.argv)
+
+    parser = ArgumentParser(
+        formatter_class=RichHelpFormatter,
+        description=__doc__,
+        epilog="Example Usage: taffy-config > defaults.yml",
+    )
+
+    parser.add_argument(
+        "-y",
+        "--config",
+        default=None,
+        type=str,
+        help="Configuration file (YAML) to load.",
+    )
+
+    parser.add_argument(
+        "--log-level",
+        "--ll",
+        default="info",
+        help="Define the logging verbosity level (debug, info, warning, error, fotal, critical).",
+    )
+
+    args = parser.parse_args()
+    log_level = args.log_level.upper()
+    logging.basicConfig(level=log_level, format="%(levelname)-10s:\t%(message)s")
+
+    config.load_namespace(args)
+    return config
+
+
 def main() -> None:
     """Dissect a pcap file and report contents."""
 
-    def parse_args() -> Namespace:
-        """Parse the command line arguments."""
-
-        config: TaffyConfig = TaffyConfig()
-        config.config_option_names = ["-y", "--config"]
-        config[TT_CFG.LOG_LEVEL] = "info"
-
-        config.read_configfile_from_arguments(sys.argv)
-
-        parser = ArgumentParser(
-            formatter_class=RichHelpFormatter,
-            description=__doc__,
-            epilog="Example Usage: taffy-config > defaults.yml",
-        )
-
-        parser.add_argument(
-            "-y",
-            "--config",
-            default=None,
-            type=str,
-            help="Configuration file (YAML) to load.",
-        )
-
-        parser.add_argument(
-            "--log-level",
-            "--ll",
-            default="info",
-            help="Define the logging verbosity level (debug, info, warning, error, fotal, critical).",
-        )
-
-        args = parser.parse_args()
-        log_level = args.log_level.upper()
-        logging.basicConfig(level=log_level, format="%(levelname)-10s:\t%(message)s")
-
-        config.load_namespace(args)
-        return config
-
-    config = parse_args()
-    config.as_namespace()
+    config = taffy_config_parse_args()
 
     print(yaml.dump(dict(config)))
 

traffic_taffy-0.9.2/traffic_taffy/__init__.py

@@ -1 +0,0 @@
-__VERSION__ = "0.9.2"