def setup_env(scenario, child_env, config, config_name_list, j2template_list):
""" Set up test environment and config """
# Clear test directory
del_files(TMPDIR, False)
# Set up libfaketime
os.environ["FAKETIME_NO_CACHE"] = "1"
os.environ["FAKETIME_TIMESTAMP_FILE"] = '%s/.time' % TMPDIR
child_env["FAKETIME_NO_CACHE"] = "1"
child_env["FAKETIME_TIMESTAMP_FILE"] = '%s/.time' % TMPDIR
write_timestamp_file(child_env["FAKETIME_TIMESTAMP_FILE"],
int(time.time()))
# Set up child process env()
child_env["SOCKET_WRAPPER_DEFAULT_IFACE"] = "%i" % CHILD_IFACE
child_env["SOCKET_WRAPPER_DIR"] = TMPDIR
no_minimize = "true"
trust_anchor_str = ""
stub_addr = ""
features = {}
feature_list_delimiter = DEFAULT_FEATURE_LIST_DELIM
feature_pair_delimiter = DEFAULT_FEATURE_PAIR_DELIM
selfaddr = testserver.get_local_addr_str(socket.AF_INET, DEFAULT_IFACE)
for k, v in config:
# Enable selectively for some tests
if k == 'query-minimization' and str2bool(v):
no_minimize = "false"
elif k == 'trust-anchor':
trust_anchor_str = v.strip('"\'')
elif k == 'val-override-timestamp':
override_timestamp_str = v.strip('"\'')
write_timestamp_file(child_env["FAKETIME_TIMESTAMP_FILE"],
int(override_timestamp_str))
elif k == 'val-override-date':
override_date_str = v.strip('"\'')
ovr_yr = override_date_str[0:4]
ovr_mnt = override_date_str[4:6]
ovr_day = override_date_str[6:8]
ovr_hr = override_date_str[8:10]
ovr_min = override_date_str[10:12]
ovr_sec = override_date_str[12:]
override_date_str_arg = '{0} {1} {2} {3} {4} {5}'.format(
ovr_yr, ovr_mnt, ovr_day, ovr_hr, ovr_min, ovr_sec)
override_date = time.strptime(override_date_str_arg,
"%Y %m %d %H %M %S")
override_date_timestamp = calendar.timegm(override_date)
write_timestamp_file(child_env["FAKETIME_TIMESTAMP_FILE"],
override_date_timestamp)
elif k == 'stub-addr':
stub_addr = v.strip('"\'')
elif k == 'features':
feature_list = v.split(feature_list_delimiter)
try:
for f_item in feature_list:
if f_item.find(feature_pair_delimiter) != -1:
f_key, f_value = [
x.strip()
for x in f_item.split(feature_pair_delimiter, 1)
]
else:
f_key = f_item.strip()
f_value = ""
features[f_key] = f_value
except Exception as e:
raise Exception(
"can't parse features (%s) in config section (%s)" %
(v, str(e)))
elif k == 'feature-list':
try:
f_key, f_value = [
x.strip() for x in v.split(feature_pair_delimiter, 1)
]
if f_key not in features:
features[f_key] = []
f_value = f_value.replace("{{INSTALL_DIR}}", INSTALLDIR)
features[f_key].append(f_value)
except Exception as e:
raise Exception(
"can't parse feature-list (%s) in config section (%s)" %
(v, str(e)))
elif k == 'force-ipv6' and v.upper() == 'TRUE':
scenario.force_ipv6 = True
self_sockfamily = socket.AF_INET
if scenario.force_ipv6 == True:
self_sockfamily = socket.AF_INET6
if stub_addr != "":
selfaddr = stub_addr
else:
selfaddr = testserver.get_local_addr_str(self_sockfamily,
DEFAULT_IFACE)
childaddr = testserver.get_local_addr_str(self_sockfamily, CHILD_IFACE)
# Prebind to sockets to create necessary files
# @TODO: this is probably a workaround for socket_wrapper bug
if 'NOPRELOAD' not in os.environ:
for sock_type in (socket.SOCK_STREAM, socket.SOCK_DGRAM):
sock = socket.socket(self_sockfamily, sock_type)
sock.setsockopt(self_sockfamily, socket.SO_REUSEADDR, 1)
sock.bind((childaddr, 53))
if sock_type == socket.SOCK_STREAM:
sock.listen(5)
# Generate configuration files
j2template_loader = jinja2.FileSystemLoader(
searchpath=os.path.dirname(os.path.abspath(__file__)))
j2template_env = jinja2.Environment(loader=j2template_loader)
j2template_ctx = {
"ROOT_ADDR": selfaddr,
"SELF_ADDR": childaddr,
"NO_MINIMIZE": no_minimize,
"TRUST_ANCHOR": trust_anchor_str,
"WORKING_DIR": TMPDIR,
"INSTALL_DIR": INSTALLDIR,
"FEATURES": features
}
for template_name, config_name in itertools.izip(j2template_list,
config_name_list):
j2template = j2template_env.get_template(template_name)
cfg_rendered = j2template.render(j2template_ctx)
f = open(os.path.join(TMPDIR, config_name), 'w')
f.write(cfg_rendered)
f.close()
def setup_env(scenario, child_env, config, config_name_list, j2template_list):
""" Set up test environment and config """
# Clear test directory
del_files(TMPDIR, False)
# Set up libfaketime
os.environ["FAKETIME_NO_CACHE"] = "1"
os.environ["FAKETIME_TIMESTAMP_FILE"] = '%s/.time' % TMPDIR
child_env["FAKETIME_NO_CACHE"] = "1"
child_env["FAKETIME_TIMESTAMP_FILE"] = '%s/.time' % TMPDIR
write_timestamp_file(child_env["FAKETIME_TIMESTAMP_FILE"], int (time.time()))
# Set up child process env()
child_env["SOCKET_WRAPPER_DEFAULT_IFACE"] = "%i" % CHILD_IFACE
child_env["SOCKET_WRAPPER_DIR"] = TMPDIR
no_minimize = "true"
trust_anchor_str = ""
stub_addr = ""
features = {}
feature_list_delimiter = DEFAULT_FEATURE_LIST_DELIM
feature_pair_delimiter = DEFAULT_FEATURE_PAIR_DELIM
selfaddr = testserver.get_local_addr_str(socket.AF_INET, DEFAULT_IFACE)
for k,v in config:
# Enable selectively for some tests
if k == 'query-minimization' and str2bool(v):
no_minimize = "false"
elif k == 'trust-anchor':
trust_anchor_str = v.strip('"\'')
elif k == 'val-override-timestamp':
override_timestamp_str = v.strip('"\'')
write_timestamp_file(child_env["FAKETIME_TIMESTAMP_FILE"], int(override_timestamp_str))
elif k == 'val-override-date':
override_date_str = v.strip('"\'')
ovr_yr = override_date_str[0:4]
ovr_mnt = override_date_str[4:6]
ovr_day = override_date_str[6:8]
ovr_hr = override_date_str[8:10]
ovr_min = override_date_str[10:12]
ovr_sec = override_date_str[12:]
override_date_str_arg = '{0} {1} {2} {3} {4} {5}'.format(ovr_yr,ovr_mnt,ovr_day,ovr_hr,ovr_min,ovr_sec)
override_date = time.strptime(override_date_str_arg,"%Y %m %d %H %M %S")
override_date_timestamp = calendar.timegm(override_date)
write_timestamp_file(child_env["FAKETIME_TIMESTAMP_FILE"], override_date_timestamp)
elif k == 'stub-addr':
stub_addr = v.strip('"\'')
elif k == 'features':
feature_list = v.split(feature_list_delimiter)
try :
for f_item in feature_list:
if f_item.find(feature_pair_delimiter) != -1:
f_key, f_value = [x.strip() for x in f_item.split(feature_pair_delimiter,1)]
else:
f_key = f_item.strip()
f_value = ""
features[f_key] = f_value
except Exception as e:
raise Exception ("can't parse features (%s) in config section (%s)" % (v,str(e)));
elif k == 'feature-list':
try :
f_key, f_value = [x.strip() for x in v.split(feature_pair_delimiter,1)]
if f_key not in features:
features[f_key] = []
f_value = f_value.replace("{{INSTALL_DIR}}",INSTALLDIR)
features[f_key].append(f_value)
except Exception as e:
raise Exception ("can't parse feature-list (%s) in config section (%s)" % (v,str(e)));
elif k == 'force-ipv6' and v.upper() == 'TRUE':
scenario.force_ipv6 = True
self_sockfamily = socket.AF_INET
if scenario.force_ipv6 == True:
self_sockfamily = socket.AF_INET6
if stub_addr != "":
selfaddr = stub_addr
else:
selfaddr = testserver.get_local_addr_str(self_sockfamily, DEFAULT_IFACE)
childaddr = testserver.get_local_addr_str(self_sockfamily, CHILD_IFACE)
# Prebind to sockets to create necessary files
# @TODO: this is probably a workaround for socket_wrapper bug
if 'NOPRELOAD' not in os.environ:
for sock_type in (socket.SOCK_STREAM, socket.SOCK_DGRAM):
sock = socket.socket(self_sockfamily, sock_type)
sock.setsockopt(self_sockfamily, socket.SO_REUSEADDR, 1)
sock.bind((childaddr, 53))
if sock_type == socket.SOCK_STREAM:
sock.listen(5)
# Generate configuration files
j2template_loader = jinja2.FileSystemLoader(searchpath=os.path.dirname(os.path.abspath(__file__)))
j2template_env = jinja2.Environment(loader=j2template_loader)
j2template_ctx = {
"ROOT_ADDR" : selfaddr,
"SELF_ADDR" : childaddr,
"NO_MINIMIZE" : no_minimize,
"TRUST_ANCHOR" : trust_anchor_str,
"WORKING_DIR" : TMPDIR,
"INSTALL_DIR" : INSTALLDIR,
"FEATURES" : features
}
for template_name, config_name in itertools.izip(j2template_list,config_name_list):
j2template = j2template_env.get_template(template_name)
cfg_rendered = j2template.render(j2template_ctx)
f = open(os.path.join(TMPDIR,config_name), 'w')
f.write(cfg_rendered)
f.close()
# Wait until the server accepts TCP clients
sockfamily = socket.AF_INET
if case.force_ipv6 == True:
sockfamily = socket.AF_INET6
sock = socket.socket(sockfamily, socket.SOCK_STREAM)
while True:
time.sleep(0.1)
if daemon_proc.poll() != None:
server.stop()
print(open('%s/server.log' % TMPDIR).read())
raise Exception('process died "%s", logs in "%s"' %
(os.path.basename(binary_name), TMPDIR))
try:
sock.connect(
(testserver.get_local_addr_str(sockfamily, CHILD_IFACE), 53))
except:
continue
break
sock.close()
# Bind to test servers
for r in case.ranges:
family = socket.AF_INET6 if ':' in r.address else socket.AF_INET
server.start_srv((r.address, 53), family)
# Bind addresses in ad-hoc REPLYs
for s in case.steps:
if s.type == 'REPLY':
reply = s.data[0].message
for rr in itertools.chain(reply.answer, reply.additional,
reply.question, reply.authority):
def play_object(path, binary_name, config_name, j2template, binary_additional_pars):
""" Play scenario from a file object. """
# Parse scenario
case, config = scenario.parse_file(fileinput.input(path))
# Setup daemon environment
daemon_env = os.environ.copy()
setup_env(case, daemon_env, config, config_name, j2template)
server = testserver.TestServer(case, config, DEFAULT_IFACE)
server.start()
# Start binary
daemon_proc = None
daemon_log = open('%s/server.log' % TMPDIR, 'w')
daemon_args = [binary_name] + binary_additional_pars
try :
daemon_proc = subprocess.Popen(daemon_args, stdout=daemon_log, stderr=daemon_log,
cwd=TMPDIR, preexec_fn=os.setsid, env=daemon_env)
except Exception as e:
server.stop()
raise Exception("Can't start '%s': %s" % (daemon_args, str(e)))
# Wait until the server accepts TCP clients
sockfamily = socket.AF_INET
if case.force_ipv6 == True:
sockfamily = socket.AF_INET6
sock = socket.socket(sockfamily, socket.SOCK_STREAM)
while True:
time.sleep(0.1)
if daemon_proc.poll() != None:
server.stop()
print(open('%s/server.log' % TMPDIR).read())
raise Exception('process died "%s", logs in "%s"' % (os.path.basename(binary_name), TMPDIR))
try:
sock.connect((testserver.get_local_addr_str(sockfamily, CHILD_IFACE), 53))
except: continue
break
sock.close()
# Bind to test servers
for r in case.ranges:
family = socket.AF_INET6 if ':' in r.address else socket.AF_INET
server.start_srv((r.address, 53), family)
# Bind addresses in ad-hoc REPLYs
for s in case.steps:
if s.type == 'REPLY':
reply = s.data[0].message
for rr in itertools.chain(reply.answer,reply.additional,reply.question,reply.authority):
for rd in rr:
if rd.rdtype == dns.rdatatype.A:
server.start_srv((rd.address, 53), socket.AF_INET)
elif rd.rdtype == dns.rdatatype.AAAA:
server.start_srv((rd.address, 53), socket.AF_INET6)
# Play test scenario
try:
server.play(CHILD_IFACE)
if VERBOSE:
print(open('%s/server.log' % TMPDIR).read())
except:
print(open('%s/server.log' % TMPDIR).read())
raise
finally:
server.stop()
daemon_proc.terminate()
daemon_proc.wait()
# Do not clear files if the server crashed (for analysis)
del_files(TMPDIR, OWN_TMPDIR)
