def update_requestdomain(domain):
# Update Database here
global request_batch
global processed_batches
global DUT_IP
global contacted_hosts
global current_dns_records
if str(domain) == str(request_batch.get_requesturl()):
print "UDP packet doubling detected!: %s" % (domain)
return
print "currently processing: %s" % (domain)
if request_batch._requestURL != "": # check wheter the requestbatch has been touched before
processed_batches.append(request_batch)
# sets the extracted DNS records to the data structure
for conn in request_batch._active_connections:
for dns in current_dns_records:
if conn._dst_IP in dns:
conn._DNS = dns
current_dns_records = []
request_batch = RequestBatch()
request_batch.set_requesturl(str(domain))
contacted_hosts = []
if "| MOBILE |" in str(domain):
DUT_IP = "10.0.0.23"
if "| DESKTOP |" in str(domain):
DUT_IP = "10.0.0.42"
def read_from_sql(sql_statement, batches):
conn = sqlite3.connect(opts.file)
cursor = conn.cursor()
cursor.execute(sql_statement)
for entry in cursor.fetchall():
batch = RequestBatch()
batch.set_filename(entry[1])
batch.set_requesturl(entry[2])
batch.set_getrequests(entry[3])
batch.set_dnsrequests(entry[4])
batch.set_downstreamvolume(entry[5])
batch.set_upstreamvolume(entry[6])
batch.set_nr_of_host_contacts(entry[7])
batch.set_connection_count(entry[8])
batch.set_nr_of_web_bugs(entry[9])
batches.append(batch)
conn.close()
