commit 769cd3cbbd8521af505f2b740af8468b1a347f2f Author: Arturo Filastò arturo@filasto.net Date: Tue Nov 6 18:48:57 2012 +0100
Refactor DNS Tamper test to use twisted.name.client.queryUDP * Add a "Error: " prefix to utils/log.py * Make some changes to before_i_commit script --- before_i_commit.sh | 10 +++++ nettests/core/dnstamper.py | 87 +++++++++++++++++++++++++++----------------- ooni/reporter.py | 1 + ooni/runner.py | 3 ++ ooni/utils/log.py | 2 +- 5 files changed, 69 insertions(+), 34 deletions(-)
diff --git a/before_i_commit.sh b/before_i_commit.sh index 2816ea5..1bf96a3 100755 --- a/before_i_commit.sh +++ b/before_i_commit.sh @@ -10,6 +10,8 @@ # rm *.yamloo; rm before_i_commit.log #
+rm before_i_commit.log + find . -type f -name "*.py[co]" -delete
./bin/ooniprobe -l before_i_commit.log -o url_lists.yamloo nettests/core/url_list.py -f test_inputs/url_lists_file.txt @@ -24,4 +26,12 @@ find . -type f -name "*.py[co]" -delete
./bin/ooniprobe -l before_i_commit.log -o url_lists.yamloo nettests/core/url_list.py -f test_inputs/url_lists_file.txt
+echo "Below you should not see anything" +echo "---------------------------------" +grep "Error: " before_i_commit.log +echo "---------------------------------" +echo "If you do, it means something is wrong." +echo "Read through the log file and fix it." +echo "If you are having some problems fixing some things that have to do with" +echo "the core of OONI, let's first discuss it on IRC, or open a ticket"
diff --git a/nettests/core/dnstamper.py b/nettests/core/dnstamper.py index da1878c..b25167f 100644 --- a/nettests/core/dnstamper.py +++ b/nettests/core/dnstamper.py @@ -15,12 +15,16 @@ # :authors: Arturo Filastò, Isis Lovecruft # :licence: see LICENSE
-from ooni import nettest -from ooni.utils import log +import pdb + from twisted.internet import defer -from twisted.names import client +from twisted.names import client, dns +from twisted.names.client import Resolver from twisted.names.error import DNSQueryRefusedError
+from ooni import nettest +from ooni.utils import log + class DNSTamperTest(nettest.NetTestCase):
name = "DNS tamper" @@ -32,10 +36,13 @@ class DNSTamperTest(nettest.NetTestCase): inputFile = ['file', 'f', None, 'Input file of list of hostnames to attempt to resolve']
- optParameters = [['controlresolver', 'c', '8.8.8.8', - 'Known good DNS server'], + optParameters = [['controlresolver', 'b', '8.8.8.8', + 'The OONI backend that runs the DNS resolver'], + ['controlresolver-port', 'p', 53, + 'The port of the good DNS resolver'], ['testresolvers', 't', None, - 'file containing list of DNS resolvers to test against']] + 'file containing list of DNS resolvers to test against'] + ]
def setUp(self): self.report['test_lookups'] = {} @@ -63,31 +70,36 @@ class DNSTamperTest(nettest.NetTestCase): self.test_resolvers = [x.strip() for x in fp.readlines()] fp.close()
- def process_a_answers(self, answers, resolver): - log.msg("Processing A answers for %s" % resolver) + def process_a_answers(self, message, resolver_address): + log.msg("Processing A answers for %s" % resolver_address) + log.debug("These are the answers I got %s" % message.answers) + all_a = [] a_a = [] - for answer in answers[0]: + + for answer in message.answers: if answer.type is 1: # A type query r = answer.payload.dottedQuad() - self.report['a_lookups'][resolver] = r + self.report['a_lookups'][resolver_address] = r a_a.append(r) lookup = str(answer.payload) all_a.append(lookup)
- if resolver == 'control': + if resolver_address == 'control': self.report['control_server'] = self.localOptions['controlresolver'] self.report['control_lookup'] = all_a self.control_a_lookups = a_a else: - self.test_a_lookups[resolver] = a_a - self.report['test_lookups'][resolver] = all_a + self.test_a_lookups[resolver_address] = a_a + self.report['test_lookups'][resolver_address] = all_a + log.msg("Done")
def process_ptr_answers(self, answers, resolver): log.msg("Processing PTR answers for %s" % resolver) name = None + for answer in answers[0]: if answer.type is 12: # PTR type @@ -101,9 +113,9 @@ class DNSTamperTest(nettest.NetTestCase): self.report['test_reverse'][resolver] = name
def ptr_lookup_error(self, failure, resolver): - #def ptr_lookup_error(self, *arg, **kw): log.msg("There was an error in PTR lookup %s" % resolver) - log.msg(failure) + log.err(failure) + if resolver == 'control': self.report['control_reverse'] = None else: @@ -111,6 +123,7 @@ class DNSTamperTest(nettest.NetTestCase):
def a_lookup_error(self, failure, resolver): log.msg("There was an error in A lookup %s" % resolver) + log.err(failure)
if failure.type is DNSQueryRefusedError: self.report['tampering'][resolver] = 'connection-refused' @@ -145,34 +158,40 @@ class DNSTamperTest(nettest.NetTestCase): log.msg("Doing the test lookups on %s" % self.input) list_of_ds = [] hostname = self.input + dns_query = [dns.Query(hostname, dns.IN, dns.A)] + dns_server = [(self.localOptions['controlresolver'], + self.localOptions['controlresolver-port'])]
- resolver = [(self.localOptions['controlresolver'], 53)] - res = client.createResolver(servers=resolver, resolvconf='') + resolver = Resolver(servers=dns_server)
- control_r = res.lookupAddress(hostname, timeout=self.lookupTimeout) - control_r.addCallback(self.process_a_answers, 'control') - control_r.addErrback(self.a_lookup_error, 'control') + control_d = resolver.queryUDP(dns_query, timeout=self.lookupTimeout) + control_d.addCallback(self.process_a_answers, 'control') + control_d.addErrback(self.a_lookup_error, 'control')
for test_resolver in self.test_resolvers: log.msg("Going for %s" % test_resolver) - resolver = [(test_resolver, 53)] - res = client.createResolver(servers=resolver, resolvconf='') - #res = self.createResolver(servers=resolver) + dns_server = [(test_resolver, 53)] + + resolver = Resolver(servers=dns_server)
- d = res.lookupAddress(hostname, timeout=self.lookupTimeout) + d = resolver.queryUDP(dns_query, timeout=self.lookupTimeout) d.addCallback(self.process_a_answers, test_resolver) d.addErrback(self.a_lookup_error, test_resolver) + + # This is required to cancel the delayed calls of the + # twisted.names.client resolver list_of_ds.append(d)
- list_of_ds.append(control_r) + list_of_ds.append(control_d) dl = defer.DeferredList(list_of_ds) dl.addCallback(self.do_reverse_lookups) dl.addBoth(self.compare_results) return dl
def reverse_lookup(self, address, resolver): + query = [dns.Query(hostname, dns.IN, dns.PTR)] ptr = '.'.join(address.split('.')[::-1]) + '.in-addr.arpa' - r = resolver.lookupPointer(ptr, self.lookupTimeout) + r = resolver.queryUDP(query, timeout=self.lookupTimeout) return r
def do_reverse_lookups(self, result): @@ -186,12 +205,14 @@ class DNSTamperTest(nettest.NetTestCase): """ log.msg("Doing the reverse lookups %s" % self.input) list_of_ds = [] + dns_server = [(self.localOptions['controlresolver'], + self.localOptions['controlresolver-port'])]
- resolver = [(self.localOptions['controlresolver'], 53)] - res = self.createResolver(servers=resolver) + resolver = Resolver(servers=dns_server)
- test_reverse = self.reverse_lookup(self.control_a_lookups[0], res, + test_reverse = self.reverse_lookup(self.control_a_lookups[0], resolver, timeout=self.lookupTimeout) + test_reverse.addCallback(self.process_ptr_answers, 'control') test_reverse.addErrback(self.ptr_lookup_error, 'control')
@@ -228,17 +249,17 @@ class DNSTamperTest(nettest.NetTestCase): continue
if set(test_a_lookups) & set(self.control_a_lookups): - # Address has not tampered with on DNS server + log.msg("Address has not tampered with on DNS server") self.report['tampering'][test] = False
elif self.control_reverse and set([self.control_reverse]) \ & set([self.report['test_reverse'][test]]): - # Further testing has eliminated false positives + log.msg("Further testing has eliminated false positives") self.report['tampering'][test] = 'reverse-match'
else: - # Reverse DNS on the results returned by returned - # which does not match the expected domainname + log.msg("Reverse DNS on the results returned by returned") + log.msg("which does not match the expected domainname") self.report['tampering'][test] = True
if len(self.test_a_lookups) == len(self.test_resolvers): diff --git a/ooni/reporter.py b/ooni/reporter.py index 8c0b593..ad7956d 100644 --- a/ooni/reporter.py +++ b/ooni/reporter.py @@ -209,6 +209,7 @@ class ReporterFactory(OReporter): 'test_name': options['name'], 'test_version': options['version'], } + self.writeYamlLine(test_details) self._writeln('')
diff --git a/ooni/runner.py b/ooni/runner.py index 97b4003..fa50840 100644 --- a/ooni/runner.py +++ b/ooni/runner.py @@ -19,9 +19,12 @@ from twisted.trial.runner import filenameToModule
from ooni.inputunit import InputUnitFactory from ooni.nettest import InputTestSuite + from ooni.plugoo import tests as oonitests + from ooni.reporter import ReporterFactory from ooni.utils import log, date + from ooni.utils.legacy import LegacyOONITest from ooni.utils.legacy import start_legacy_test, adapt_legacy_test
diff --git a/ooni/utils/log.py b/ooni/utils/log.py index 771bad8..d1f308e 100644 --- a/ooni/utils/log.py +++ b/ooni/utils/log.py @@ -48,7 +48,7 @@ def debug(msg, *arg, **kw): txlog.msg(msg, logLevel=logging.DEBUG, *arg, **kw)
def err(msg, *arg, **kw): - txlog.err(msg, logLevel=logging.ERROR, *arg, **kw) + txlog.err("Error: " + msg, logLevel=logging.ERROR, *arg, **kw)
def exception(*msg): logging.exception(msg)