[tor-commits] [ooni-probe/master] Add a Parasitic Traceroute NetTest

[art at torproject.org]

commit 142cd3a29155dadc3ed2e7bc0def7a63632f0ebb
Author: aagbsn <aagbsn at extc.org>
Date:   Sat Feb 1 20:43:38 2014 +0000

    Add a Parasitic Traceroute NetTest
---
 ooni/nettests/manipulation/parasitictraceroute.py |   48 +++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/ooni/nettests/manipulation/parasitictraceroute.py b/ooni/nettests/manipulation/parasitictraceroute.py
new file mode 100644
index 0000000..7820721
--- /dev/null
+++ b/ooni/nettests/manipulation/parasitictraceroute.py
@@ -0,0 +1,48 @@
+from twisted.python import usage
+from twisted.internet import defer, reactor
+from ooni.templates import scapyt
+from ooni.utils import log
+from ooni.utils.txscapy import ParasiticTraceroute
+from ooni.settings import config
+
+from scapy.all import TCPerror, IPerror
+
+class ParasiticTracerouteTest(scapyt.BaseScapyTest):
+    name = "Parasitic Traceroute Test"
+    description = "Injects duplicate TCP packets with varying TTL values by sniffing traffic"
+    version = '0.1'
+
+    samplePeriod = 40
+
+    def setUp(self):
+        self.report['parasitic_traceroute'] = {}
+
+    def test_parasitic_traceroute(self):
+        self.pt = ParasiticTraceroute()
+        config.scapyFactory.registerProtocol(self.pt)
+        d = defer.Deferred()
+        reactor.callLater(self.samplePeriod, d.callback, self.pt)
+        return d
+
+    def postProcessor(self, *args, **kwargs):
+        self.pt.stopListening()
+        self.report['received_packets'] = self.pt.received_packets
+
+        for packet in self.pt.received_packets:
+            k = (packet[IPerror].id, packet[TCPerror].sport, packet[TCPerror].dport, packet[TCPerror].seq)
+            if k in self.pt.matched_packets:
+                ttl = self.pt.matched_packets[k]['ttl']
+            else:
+                ttl = 'Unknown'
+            hop = (ttl, packet.src)
+            path = 'hops_%s' % packet[IPerror].dst
+            if path in self.report['parasitic_traceroute']:
+               self.report['parasitic_traceroute'][path].append(hop)
+            else:
+               self.report['parasitic_traceroute'][path] = [hop]
+        for p in self.report['parasitic_traceroute'].keys():
+            self.report['parasitic_traceroute'][p].sort(key=lambda x: x[0])
+                
+        self.report['sent_packets'] = self.pt.sent_packets
+        return self.report
+