osmith
/
wireshark
forked from osmocom/wireshark
1
0
Fork 0
Code Pull Requests Releases Wiki Activity

test: convert suite_text2pcap to use fixtures 

Inline all capture file names and use fixtures instead of the global
config object. This makes dependencies more explicit.

Change-Id: I37a6eda73822735b5a6957b44bce53bb5ecd1aa0
Reviewed-on: https://code.wireshark.org/review/30631
Petri-Dish: Peter Wu <peter@lekensteyn.nl>
Tested-by: Petri Dish Buildbot
Reviewed-by: Peter Wu <peter@lekensteyn.nl>
This commit is contained in:
Peter Wu 2018-11-14 17:54:00 +01:00
parent c82c60136d
commit bd84c71668
2 changed files with 214 additions and 201 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/fixtures_ws.py
View File

@ -141,13 +141,18 @@ def conf_path(home_path):
@fixtures.fixture
def base_env(home_path):
def base_env(home_path, request):
"""A modified environment to ensure reproducible tests. Tests can modify
this environment as they see fit."""
env = os.environ.copy()
env['TZ'] = 'UTC'
home_env = 'APPDATA' if sys.platform.startswith('win32') else 'HOME'
env[home_env] = home_path
# Remove this if test instances no longer inherit from SubprocessTestCase?
assert isinstance(request.instance, subprocesstest.SubprocessTestCase)
# Inject the test environment as default if it was not overridden.
request.instance.injected_test_env = env
return env

408
test/suite_text2pcap.py
View File

@ -9,36 +9,11 @@
#
'''Text2pcap tests'''
import config
import os.path
import pprint
import re
import subprocesstest
import unittest
import json
# XXX We should probably generate these automatically in config.py.
c1222_std_example8_pcap = os.path.join(config.capture_dir, 'c1222_std_example8.pcap')
dhcp_nanosecond_pcap = os.path.join(config.capture_dir, 'dhcp-nanosecond.pcap')
dhcp_nanosecond_pcapng = os.path.join(config.capture_dir, 'dhcp-nanosecond.pcapng')
dhcp_pcap = os.path.join(config.capture_dir, 'dhcp.pcap')
dhcp_pcapng = os.path.join(config.capture_dir, 'dhcp.pcapng')
dhe1_pcapng_gz = os.path.join(config.capture_dir, 'dhe1.pcapng.gz')
dmgr_pcapng = os.path.join(config.capture_dir, 'dmgr.pcapng')
dns_icmp_pcapng_gz = os.path.join(config.capture_dir, 'dns+icmp.pcapng.gz')
dns_port_pcap = os.path.join(config.capture_dir, 'dns_port.pcap')
dvb_ci_uv1_0000_pcap = os.path.join(config.capture_dir, 'dvb-ci_UV1_0000.pcap')
empty_pcap = os.path.join(config.capture_dir, 'empty.pcap')
ikev1_certs_pcap = os.path.join(config.capture_dir, 'ikev1-certs.pcap')
packet_h2_14_headers_pcapng = os.path.join(config.capture_dir, 'packet-h2-14_headers.pcapng')
rsa_p_lt_q_pcap = os.path.join(config.capture_dir, 'rsa-p-lt-q.pcap')
rsasnakeoil2_pcap = os.path.join(config.capture_dir, 'rsasnakeoil2.pcap')
sample_control4_2012_03_24_pcap = os.path.join(config.capture_dir, 'sample_control4_2012-03-24.pcap')
segmented_fpm_pcap = os.path.join(config.capture_dir, 'segmented_fpm.pcap')
sip_pcapng = os.path.join(config.capture_dir, 'sip.pcapng')
snakeoil_dtls_pcap = os.path.join(config.capture_dir, 'snakeoil-dtls.pcap')
wpa_induction_pcap_gz = os.path.join(config.capture_dir, 'wpa-Induction.pcap.gz')
wpa_eap_tls_pcap_gz = os.path.join(config.capture_dir, 'wpa-eap-tls.pcap.gz')
import fixtures
testin_txt = 'testin.txt'
testout_pcap = 'testout.pcap'
@ -109,166 +84,185 @@ def compare_capinfos_info(self, cii1, cii2, filename1, filename2):
self.diffOutput(cii1_pp, cii2_pp, filename1, filename2)
self.fail('text2pcap output file differs from input file.')
def check_text2pcap(self, cap_file, file_type, expected_packets=None, expected_datasize=None):
# Perform the following actions
# - Get information for the input pcap file with capinfos
# - Generate an ASCII hexdump with TShark
# - Convert the ASCII hexdump back to pcap using text2pcap
# - Get information for the output pcap file with capinfos
# - Check that file type, encapsulation type, number of packets and data size
# in the output file are the same as in the input file
@fixtures.fixture
def check_text2pcap(cmd_tshark, cmd_text2pcap, capture_file):
def check_text2pcap_real(self, cap_filename, file_type, expected_packets=None, expected_datasize=None):
# Perform the following actions
# - Get information for the input pcap file with capinfos
# - Generate an ASCII hexdump with TShark
# - Convert the ASCII hexdump back to pcap using text2pcap
# - Get information for the output pcap file with capinfos
# - Check that file type, encapsulation type, number of packets and data size
# in the output file are the same as in the input file
pre_cap_info = check_capinfos_info(self, cap_file)
# Due to limitations of "tshark -x", the output might contain more than one
# data source which is subsequently interpreted as additional frame data.
# See https://bugs.wireshark.org/bugzilla/show_bug.cgi?id=14639
if expected_packets is not None:
self.assertNotEqual(pre_cap_info['packets'], expected_packets)
pre_cap_info['packets'] = expected_packets
if expected_datasize is not None:
self.assertNotEqual(pre_cap_info['datasize'], expected_datasize)
pre_cap_info['datasize'] = expected_datasize
self.assertTrue(pre_cap_info['encapsulation'] in encap_to_link_type)
cap_file = capture_file(cap_filename)
pre_cap_info = check_capinfos_info(self, cap_file)
# Due to limitations of "tshark -x", the output might contain more than one
# data source which is subsequently interpreted as additional frame data.
# See https://bugs.wireshark.org/bugzilla/show_bug.cgi?id=14639
if expected_packets is not None:
self.assertNotEqual(pre_cap_info['packets'], expected_packets)
pre_cap_info['packets'] = expected_packets
if expected_datasize is not None:
self.assertNotEqual(pre_cap_info['datasize'], expected_datasize)
pre_cap_info['datasize'] = expected_datasize
self.assertTrue(pre_cap_info['encapsulation'] in encap_to_link_type)
self.assertTrue(file_type in file_type_to_testout, 'Invalid file type')
self.assertTrue(file_type in file_type_to_testout, 'Invalid file type')
# text2pcap_generate_input()
# $TSHARK -o 'gui.column.format:"Time","%t"' -tad -P -x -r $1 > testin.txt
testin_file = self.filename_from_id(testin_txt)
cf_path = os.path.join(config.capture_dir, cap_file)
tshark_cmd = '{cmd} -r {cf} -o gui.column.format:"Time","%t" -t ad -P -x > {of}'.format(
cmd = config.cmd_tshark,
cf = cf_path,
of = testin_file,
)
self.assertRun(tshark_cmd, shell=True, env=config.baseEnv())
# text2pcap_generate_input()
# $TSHARK -o 'gui.column.format:"Time","%t"' -tad -P -x -r $1 > testin.txt
testin_file = self.filename_from_id(testin_txt)
tshark_cmd = '{cmd} -r {cf} -o gui.column.format:"Time","%t" -t ad -P -x > {of}'.format(
cmd = cmd_tshark,
cf = cap_file,
of = testin_file,
)
self.assertRun(tshark_cmd, shell=True)
testout_fname = file_type_to_testout[file_type]
testout_file = self.filename_from_id(testout_fname)
if 'pcapng' in pre_cap_info['filetype'] or 'nanosecond libpcap' in pre_cap_info['filetype']:
pcapng_flag = '-n'
else:
pcapng_flag = ''
text2pcap_cmd = '{cmd} {ns} -d -l {linktype} -t "%Y-%m-%d %H:%M:%S." {in_f} {out_f}'.format(
cmd = config.cmd_text2pcap,
ns = pcapng_flag,
linktype = encap_to_link_type[pre_cap_info['encapsulation']],
in_f = testin_file,
out_f = testout_file,
)
self.assertRun(text2pcap_cmd, shell=True)
self.assertTrue(self.grepOutput('potential packet'), "text2pcap didn't complete")
self.assertFalse(self.grepOutput('Inconsistent offset'), 'text2pcap detected inconsistent offset')
testout_fname = file_type_to_testout[file_type]
testout_file = self.filename_from_id(testout_fname)
if 'pcapng' in pre_cap_info['filetype'] or 'nanosecond libpcap' in pre_cap_info['filetype']:
pcapng_flag = '-n'
else:
pcapng_flag = ''
text2pcap_cmd = '{cmd} {ns} -d -l {linktype} -t "%Y-%m-%d %H:%M:%S." {in_f} {out_f}'.format(
cmd = cmd_text2pcap,
ns = pcapng_flag,
linktype = encap_to_link_type[pre_cap_info['encapsulation']],
in_f = testin_file,
out_f = testout_file,
)
self.assertRun(text2pcap_cmd, shell=True)
self.assertTrue(self.grepOutput('potential packet'), "text2pcap didn't complete")
self.assertFalse(self.grepOutput('Inconsistent offset'), 'text2pcap detected inconsistent offset')
post_cap_info = check_capinfos_info(self, testout_file)
compare_capinfos_info(self, pre_cap_info, post_cap_info, cap_file, testout_fname)
post_cap_info = check_capinfos_info(self, testout_file)
compare_capinfos_info(self, pre_cap_info, post_cap_info, cap_file, testout_fname)
return check_text2pcap_real
@fixtures.mark_usefixtures('base_env')
@fixtures.uses_fixtures
class case_text2pcap_pcap(subprocesstest.SubprocessTestCase):
def test_text2pcap_empty_pcap(self):
def test_text2pcap_empty_pcap(self, check_text2pcap):
'''Test text2pcap with empty.pcap.'''
check_text2pcap(self, empty_pcap, 'pcap')
check_text2pcap(self, 'empty.pcap', 'pcap')
def test_text2pcap_dhcp_pcap(self):
def test_text2pcap_dhcp_pcap(self, check_text2pcap):
'''Test text2pcap with dhcp.pcap.'''
check_text2pcap(self, dhcp_pcap, 'pcap')
check_text2pcap(self, 'dhcp.pcap', 'pcap')
def test_text2pcap_dhcp_nanosecond_pcap(self):
def test_text2pcap_dhcp_nanosecond_pcap(self, check_text2pcap):
'''Test text2pcap with dhcp-nanosecond.pcap.'''
check_text2pcap(self, dhcp_nanosecond_pcap, 'pcap')
check_text2pcap(self, 'dhcp-nanosecond.pcap', 'pcap')
def test_text2pcap_segmented_fpm_pcap(self):
def test_text2pcap_segmented_fpm_pcap(self, check_text2pcap):
'''Test text2pcap with segmented_fpm.pcap.'''
check_text2pcap(self, segmented_fpm_pcap, 'pcap')
check_text2pcap(self, 'segmented_fpm.pcap', 'pcap')
def test_text2pcap_c1222_std_example8_pcap(self):
def test_text2pcap_c1222_std_example8_pcap(self, check_text2pcap):
'''Test text2pcap with c1222_std_example8.pcap.'''
check_text2pcap(self, c1222_std_example8_pcap, 'pcap')
check_text2pcap(self, 'c1222_std_example8.pcap', 'pcap')
def test_text2pcap_dns_port_pcap(self):
def test_text2pcap_dns_port_pcap(self, check_text2pcap):
'''Test text2pcap with dns_port.pcap.'''
check_text2pcap(self, dns_port_pcap, 'pcap')
check_text2pcap(self, 'dns_port.pcap', 'pcap')
def test_text2pcap_dvb_ci_uv1_0000_pcap(self):
def test_text2pcap_dvb_ci_uv1_0000_pcap(self, check_text2pcap):
'''Test text2pcap with dvb-ci_UV1_0000.pcap.'''
check_text2pcap(self, dvb_ci_uv1_0000_pcap, 'pcap')
check_text2pcap(self, 'dvb-ci_UV1_0000.pcap', 'pcap')
def test_text2pcap_ikev1_certs_pcap(self):
def test_text2pcap_ikev1_certs_pcap(self, check_text2pcap):
'''Test text2pcap with ikev1-certs.pcap.'''
check_text2pcap(self, ikev1_certs_pcap, 'pcap')
check_text2pcap(self, 'ikev1-certs.pcap', 'pcap')
def test_text2pcap_rsa_p_lt_q_pcap(self):
def test_text2pcap_rsa_p_lt_q_pcap(self, check_text2pcap):
'''Test text2pcap with rsa-p-lt-q.pcap.'''
check_text2pcap(self, rsa_p_lt_q_pcap, 'pcap')
check_text2pcap(self, 'rsa-p-lt-q.pcap', 'pcap')
def test_text2pcap_rsasnakeoil2_pcap(self):
def test_text2pcap_rsasnakeoil2_pcap(self, check_text2pcap):
'''Test text2pcap with rsasnakeoil2.pcap.'''
check_text2pcap(self, rsasnakeoil2_pcap, 'pcap')
check_text2pcap(self, 'rsasnakeoil2.pcap', 'pcap')
def test_text2pcap_sample_control4_2012_03_24_pcap(self):
def test_text2pcap_sample_control4_2012_03_24_pcap(self, check_text2pcap):
'''Test text2pcap with sample_control4_2012-03-24.pcap.'''
# tshark currently output decrypted ZigBee packets and
# as a result the number of packets and data size are different
check_text2pcap(self, sample_control4_2012_03_24_pcap, 'pcap', 239, 10103)
check_text2pcap(self, 'sample_control4_2012-03-24.pcap', 'pcap', 239, 10103)
def test_text2pcap_snakeoil_dtls_pcap(self):
def test_text2pcap_snakeoil_dtls_pcap(self, check_text2pcap):
'''Test text2pcap with snakeoil-dtls.pcap.'''
check_text2pcap(self, snakeoil_dtls_pcap, 'pcap')
check_text2pcap(self, 'snakeoil-dtls.pcap', 'pcap')
def test_text2pcap_wpa_eap_tls_pcap_gz(self):
def test_text2pcap_wpa_eap_tls_pcap_gz(self, check_text2pcap):
'''Test text2pcap with wpa-eap-tls.pcap.gz.'''
# tshark reassembles some packets and because of this
# the number of packets and data size are different
check_text2pcap(self, wpa_eap_tls_pcap_gz, 'pcap', 88, 38872)
check_text2pcap(self, 'wpa-eap-tls.pcap.gz', 'pcap', 88, 38872)
def test_text2pcap_wpa_induction_pcap(self):
def test_text2pcap_wpa_induction_pcap(self, check_text2pcap):
'''Test text2pcap with wpa-Induction.pcap.gz.'''
check_text2pcap(self, wpa_induction_pcap_gz, 'pcap')
check_text2pcap(self, 'wpa-Induction.pcap.gz', 'pcap')
@fixtures.mark_usefixtures('base_env')
@fixtures.uses_fixtures
class case_text2pcap_pcapng(subprocesstest.SubprocessTestCase):
def test_text2pcap_dhcp_pcapng(self):
def test_text2pcap_dhcp_pcapng(self, check_text2pcap):
'''Test text2pcap with dhcp.pcapng.'''
check_text2pcap(self, dhcp_pcapng, 'pcapng')
check_text2pcap(self, 'dhcp.pcapng', 'pcapng')
def test_text2pcap_dhcp_nanosecond_pcapng(self):
def test_text2pcap_dhcp_nanosecond_pcapng(self, check_text2pcap):
'''Test text2pcap with dhcp-nanosecond.pcapng.'''
check_text2pcap(self, dhcp_nanosecond_pcapng, 'pcapng')
check_text2pcap(self, 'dhcp-nanosecond.pcapng', 'pcapng')
def test_text2pcap_dhe1_pcapng_gz(self):
def test_text2pcap_dhe1_pcapng_gz(self, check_text2pcap):
'''Test text2pcap with dhe1.pcapng.gz.'''
check_text2pcap(self, dhe1_pcapng_gz, 'pcapng')
check_text2pcap(self, 'dhe1.pcapng.gz', 'pcapng')
def test_text2pcap_dmgr_pcapng(self):
def test_text2pcap_dmgr_pcapng(self, check_text2pcap):
'''Test text2pcap with dmgr.pcapng.'''
check_text2pcap(self, dmgr_pcapng, 'pcapng')
check_text2pcap(self, 'dmgr.pcapng', 'pcapng')
def test_text2pcap_dns_icmp_pcapng_gz(self):
def test_text2pcap_dns_icmp_pcapng_gz(self, check_text2pcap):
'''Test text2pcap with dns+icmp.pcapng.gz.'''
# Different data size
# Most probably the problem is that input file timestamp precision is in microseconds
# File timestamp precision: microseconds (6)
check_text2pcap(self, dns_icmp_pcapng_gz, 'pcapng', None, 3202)
check_text2pcap(self, 'dns+icmp.pcapng.gz', 'pcapng', None, 3202)
def test_text2pcap_packet_h2_14_headers_pcapng(self):
def test_text2pcap_packet_h2_14_headers_pcapng(self, check_text2pcap):
'''Test text2pcap with packet-h2-14_headers.pcapng.'''
check_text2pcap(self, packet_h2_14_headers_pcapng, 'pcapng')
check_text2pcap(self, 'packet-h2-14_headers.pcapng', 'pcapng')
def test_text2pcap_sip_pcapng(self):
def test_text2pcap_sip_pcapng(self, check_text2pcap):
'''Test text2pcap with sip.pcapng.'''
check_text2pcap(self, sip_pcapng, 'pcapng')
check_text2pcap(self, 'sip.pcapng', 'pcapng')
@fixtures.fixture
def check_rawip(run_text2pcap_capinfos_tshark, request):
def check_rawip_real(pdata, packets, datasize):
self = request.instance
self.assertEqual({'encapsulation': 'Raw IPv4', 'packets': packets,
'datasize': datasize, 'expert': ''},
run_text2pcap_capinfos_tshark(pdata, ("-l228",)))
return check_rawip_real
@fixtures.mark_usefixtures('base_env')
@fixtures.uses_fixtures
class case_text2pcap_parsing(subprocesstest.SubprocessTestCase):
def test_text2pcap_eol_hash(self):
def test_text2pcap_eol_hash(self, cmd_text2pcap, capture_file):
'''Test text2pcap hash sign at the end-of-line.'''
txt_fname = 'text2pcap_hash_eol.txt'
txt_file = os.path.join(config.capture_dir, txt_fname)
testout_file = self.filename_from_id(testout_pcap)
self.assertRun((config.cmd_text2pcap,
self.assertRun((cmd_text2pcap,
'-n',
'-d',
'-t', '%Y-%m-%d %H:%M:%S.',
txt_file,
capture_file(txt_fname),
testout_file,
))
self.assertFalse(self.grepOutput('Inconsistent offset'), 'text2pcap failed to parse the hash sign at the end of the line')
@ -277,12 +271,7 @@ class case_text2pcap_parsing(subprocesstest.SubprocessTestCase):
post_cmp_info = check_capinfos_info(self, testout_file)
compare_capinfos_info(self, pre_cmp_info, post_cmp_info, txt_fname, testout_pcap)
def check_rawip(self, pdata, packets, datasize):
self.assertEqual({'encapsulation': 'Raw IPv4', 'packets': packets,
'datasize': datasize, 'expert': ''},
run_text2pcap_capinfos_tshark(self, pdata, ("-l228",)))
def test_text2pcap_doc_no_line_limit(self):
def test_text2pcap_doc_no_line_limit(self, check_rawip):
'''
Verify: There is no limit on the width or number of bytes per line and
Bytes/hex numbers can be uppercase or lowercase.
@ -291,9 +280,9 @@ class case_text2pcap_parsing(subprocesstest.SubprocessTestCase):
"000A 7C C9 7F 00 00 01" \
" 7f 00 00 01 ff 98 00 13 00 0d b5 48 66 69 72 73\n" \
"0020 74\n"
self.check_rawip(pdata, 1, 33)
check_rawip(pdata, 1, 33)
def test_text2pcap_doc_ignore_text(self):
def test_text2pcap_doc_ignore_text(self, check_rawip):
'''
Verify: the text dump at the end of the line is ignored. Any hex numbers
in this text are also ignored. Any lines of text between the bytestring
@ -310,9 +299,9 @@ class case_text2pcap_parsing(subprocesstest.SubprocessTestCase):
" \t# 0020 12 34 56<-- comment, ignore this!\n" \
"0020 6e 64\n" \
"12 34 56 78 90 # ignore this due to missing offset!\n"
self.check_rawip(pdata, 2, 67)
check_rawip(pdata, 2, 67)
def test_text2pcap_doc_leading_text_ignored(self):
def test_text2pcap_doc_leading_text_ignored(self, check_rawip):
'''
Verify: Any test before the offset is ignored, including email
forwarding characters '>'. An offset is a hex number longer than two
@ -324,128 +313,136 @@ class case_text2pcap_parsing(subprocesstest.SubprocessTestCase):
"> >> 000 45 00 00 22 00 01 00 00 40 11 7c c8 7f 00 00 01\n" \
"> >> 010 7f 00 00 01 ff 99 00 13 00 0e bc e9 73 65 63 6f\n" \
"> >> 020 6e 64\n"
self.check_rawip(pdata, 2, 67)
check_rawip(pdata, 2, 67)
def test_text2pcap_doc_require_offset(self):
def test_text2pcap_doc_require_offset(self, check_rawip):
'''Any line which has only bytes without a leading offset is ignored.'''
pdata = "45 00 00 21 00 01 00 00 40 11 7c c9 7f 00 00 01\n" \
"7f 00 00 01 ff 98 00 13 00 0d b5 48 66 69 72 73\n"
self.check_rawip(pdata, 0, 0)
check_rawip(pdata, 0, 0)
def test_text2pcap_eol_missing(self):
def test_text2pcap_eol_missing(self, check_rawip):
'''Verify that the last LF can be missing.'''
pdata = "0000 45 00 00 21 00 01 00 00 40 11 7c c9 7f 00 00 01\n" \
"0010 7f 00 00 01 ff 98 00 13 00 0d b5 48 66 69 72 73\n" \
"0020 74"
self.check_rawip(pdata, 1, 33)
check_rawip(pdata, 1, 33)
def run_text2pcap_content(test, content, args):
testin_file = test.filename_from_id(testin_txt)
testout_file = test.filename_from_id(testout_pcap)
@fixtures.fixture
def run_text2pcap_capinfos_tshark(cmd_text2pcap, cmd_tshark, request):
def run_text2pcap_capinfos_tshark_real(content, args):
test = request.instance
testin_file = test.filename_from_id(testin_txt)
testout_file = test.filename_from_id(testout_pcap)
fin = open(testin_file, "w")
fin.write(content)
fin.close()
with open(testin_file, "w") as f:
f.write(content)
test.assertRun((cmd_text2pcap,) + args + (testin_file, testout_file))
test.assertRun((config.cmd_text2pcap,) + args + (testin_file, testout_file))
return testout_file
capinfo = get_capinfos_cmp_info(check_capinfos_info(test, testout_file))
def run_text2pcap_capinfos_tshark(test, content, args):
testout_file = run_text2pcap_content(test, content, args)
test.assertRun((cmd_tshark, '-q', '-z', 'expert,warn',
'-o', 'udp.check_checksum: TRUE',
'-o', 'tcp.check_checksum: TRUE',
'-o', 'sctp.checksum:TRUE',
'-r', testout_file))
capinfo['expert'] = test.processes[-1].stdout_str
return capinfo
return run_text2pcap_capinfos_tshark_real
capinfo = get_capinfos_cmp_info(check_capinfos_info(test, testout_file))
test.assertRun((config.cmd_tshark, '-q', '-z', 'expert,warn',
'-o', 'udp.check_checksum: TRUE',
'-o', 'tcp.check_checksum: TRUE',
'-o', 'sctp.checksum:TRUE',
'-r', testout_file))
capinfo['expert'] = test.processes[-1].stdout_str
return capinfo;
@fixtures.mark_usefixtures('base_env')
@fixtures.uses_fixtures
class case_text2pcap_headers(subprocesstest.SubprocessTestCase):
'''Test TCP, UDP or SCTP header without -4 or -6 option'''
maxDiff = None
def run_text2pcap(self, content, args):
return run_text2pcap_capinfos_tshark(self, content, args);
def test_text2pcap_tcp(self):
def test_text2pcap_tcp(self, run_text2pcap_capinfos_tshark):
'''Test TCP over IPv4'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 60, 'expert': ''},
self.run_text2pcap("0000: ff ff ff ff\n", ("-T", "1234,1234")))
run_text2pcap_capinfos_tshark(
"0000: ff ff ff ff\n", ("-T", "1234,1234")))
def test_text2pcap_udp(self):
def test_text2pcap_udp(self, run_text2pcap_capinfos_tshark):
'''Test UDP over IPv4'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 60, 'expert': ''},
self.run_text2pcap("0000: ff ff ff ff\n", ("-u", "1234,1234")))
run_text2pcap_capinfos_tshark(
"0000: ff ff ff ff\n", ("-u", "1234,1234")))
def test_text2pcap_sctp(self):
def test_text2pcap_sctp(self, run_text2pcap_capinfos_tshark):
'''Test SCTP over IPv4'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 70, 'expert': ''},
self.run_text2pcap(
run_text2pcap_capinfos_tshark(
"0000 00 03 00 18 00 00 00 00 00 00 00 00 00 00 00 03\n" +
"0010 01 00 03 03 00 00 00 08\n",
("-s", "2905,2905,3")))
def test_text2pcap_sctp_data(self):
def test_text2pcap_sctp_data(self, run_text2pcap_capinfos_tshark):
'''Test SCTP DATA over IPv4'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 70, 'expert': ''},
self.run_text2pcap("0000: 01 00 03 03 00 00 00 08\n",
run_text2pcap_capinfos_tshark(
"0000: 01 00 03 03 00 00 00 08\n",
("-S", "2905,2905,3")))
@fixtures.fixture
def run_text2pcap_ipv4(run_text2pcap_capinfos_tshark):
def run_text2pcap_ipv4_real(content, args):
return run_text2pcap_capinfos_tshark(content,
("-4", "127.0.0.1,127.0.0.1") + args)
return run_text2pcap_ipv4_real
@fixtures.mark_usefixtures('base_env')
@fixtures.uses_fixtures
class case_text2pcap_ipv4(subprocesstest.SubprocessTestCase):
'''Test TCP, UDP or SCTP header with -4 option'''
maxDiff = None
def run_text2pcap_ipv4(self, content, args):
return run_text2pcap_capinfos_tshark(self, content,
("-4", "127.0.0.1,127.0.0.1") + args)
def test_text2pcap_ipv4_tcp(self):
def test_text2pcap_ipv4_tcp(self, run_text2pcap_ipv4):
'''Test TCP over IPv4'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 60, 'expert': ''},
self.run_text2pcap_ipv4("0000: ff ff ff ff\n", ("-T", "1234,1234")))
run_text2pcap_ipv4("0000: ff ff ff ff\n", ("-T", "1234,1234")))
def test_text2pcap_ipv4_udp(self):
def test_text2pcap_ipv4_udp(self, run_text2pcap_ipv4):
'''Test UDP over IPv4'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 60, 'expert': ''},
self.run_text2pcap_ipv4("0000: ff ff ff ff\n", ("-u", "1234,1234")))
run_text2pcap_ipv4("0000: ff ff ff ff\n", ("-u", "1234,1234")))
def test_text2pcap_ipv4_sctp(self):
def test_text2pcap_ipv4_sctp(self, run_text2pcap_ipv4):
'''Test SCTP over IPv4'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 70, 'expert': ''},
self.run_text2pcap_ipv4(
run_text2pcap_ipv4(
"0000 00 03 00 18 00 00 00 00 00 00 00 00 00 00 00 03\n" +
"0010 01 00 03 03 00 00 00 08\n",
("-s", "2905,2905,3")))
def test_text2pcap_ipv4_sctp_data(self):
def test_text2pcap_ipv4_sctp_data(self, run_text2pcap_ipv4):
'''Test SCTP DATA over IPv4'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 70, 'expert': ''},
self.run_text2pcap_ipv4("0000: 01 00 03 03 00 00 00 08\n",
run_text2pcap_ipv4("0000: 01 00 03 03 00 00 00 08\n",
("-S", "2905,2905,3")))
class case_text2pcap_ipv6(subprocesstest.SubprocessTestCase):
'''Test TCP, UDP or SCTP header with -6 option'''
maxDiff = None
def run_text2pcap_ipv6(self, content, text2pcap_args, tshark_args = ()):
@fixtures.fixture
def run_text2pcap_ipv6(cmd_tshark, run_text2pcap_capinfos_tshark, request):
self = request.instance
def run_text2pcap_ipv6_real(content, text2pcap_args, tshark_args = ()):
#Run the common text2pcap tests
result = run_text2pcap_capinfos_tshark(self, content,
result = run_text2pcap_capinfos_tshark(content,
("-6", "::1,::1") + text2pcap_args)
#Decode the output pcap in JSON format
self.assertRun((config.cmd_tshark, '-T', 'json',
self.assertRun((cmd_tshark, '-T', 'json',
'-r', self.filename_from_id(testout_pcap)) + tshark_args)
data = json.loads(self.processes[-1].stdout_str)
@ -454,80 +451,91 @@ class case_text2pcap_ipv6(subprocesstest.SubprocessTestCase):
result['ipv6'] = {
'plen': ipv6.get('ipv6.plen', None),
'plen_tree': ipv6.get('ipv6.plen_tree', None)}
return result;
return result
return run_text2pcap_ipv6_real
def test_text2pcap_ipv6_tcp(self):
@fixtures.mark_usefixtures('base_env')
@fixtures.uses_fixtures
class case_text2pcap_ipv6(subprocesstest.SubprocessTestCase):
'''Test TCP, UDP or SCTP header with -6 option'''
maxDiff = None
def test_text2pcap_ipv6_tcp(self, run_text2pcap_ipv6):
'''Test TCP over IPv6'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 78, 'expert': '',
'ipv6': {'plen': '24', 'plen_tree': None}},
self.run_text2pcap_ipv6("0000: ff ff ff ff\n", ("-T", "1234,1234")))
run_text2pcap_ipv6("0000: ff ff ff ff\n", ("-T", "1234,1234")))
def test_text2pcap_ipv6_udp(self):
def test_text2pcap_ipv6_udp(self, run_text2pcap_ipv6):
'''Test UDP over IPv6'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 66, 'expert': '',
'ipv6': {'plen': '12', 'plen_tree': None}},
self.run_text2pcap_ipv6("0000: ff ff ff ff\n", ("-u", "1234,1234")))
run_text2pcap_ipv6("0000: ff ff ff ff\n", ("-u", "1234,1234")))
def test_text2pcap_ipv6_sctp(self):
def test_text2pcap_ipv6_sctp(self, run_text2pcap_ipv6):
'''Test SCTP over IPv6'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 90, 'expert': '',
'ipv6': {'plen': '36', 'plen_tree': None}},
self.run_text2pcap_ipv6(
run_text2pcap_ipv6(
"0000 00 03 00 18 00 00 00 00 00 00 00 00 00 00 00 03\n" +
"0010 01 00 03 03 00 00 00 08\n",
("-s", "2905,2905,3")))
def test_text2pcap_ipv6_sctp_data(self):
def test_text2pcap_ipv6_sctp_data(self, run_text2pcap_ipv6):
'''Test SCTP DATA over IPv6'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 90, 'expert': '',
'ipv6': {'plen': '36', 'plen_tree': None}},
self.run_text2pcap_ipv6("0000: 01 00 03 03 00 00 00 08\n",
run_text2pcap_ipv6("0000: 01 00 03 03 00 00 00 08\n",
("-S", "2905,2905,3")))
@fixtures.mark_usefixtures('base_env')
@fixtures.uses_fixtures
class case_text2pcap_i_proto(subprocesstest.SubprocessTestCase):
'''Test -i <proto> for IPv4 and IPv6'''
maxDiff = None
def test_text2pcap_i_icmp(self):
def test_text2pcap_i_icmp(self, run_text2pcap_capinfos_tshark):
'''Test -i <proto> without -4 or -6'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 98, 'expert': ''},
run_text2pcap_capinfos_tshark(self,
run_text2pcap_capinfos_tshark(
"0000 08 00 bb b3 d7 3b 00 00 51 a7 d6 7d 00 04 51 e4\n" +
"0010 08 09 0a 0b 0c 0d 0e 0f 10 11 12 13 14 15 16 17\n" +
"0020 18 19 1a 1b 1c 1d 1e 1f 20 21 22 23 24 25 26 27\n" +
"0030 28 29 2a 2b 2c 2d 2e 2f 30 31 32 33 34 35 36 37\n",
("-i", "1")))
def test_text2pcap_i_icmp_ipv4(self):
def test_text2pcap_i_icmp_ipv4(self, run_text2pcap_capinfos_tshark):
'''Test -i <proto> with IPv4 (-4) header'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 98, 'expert': ''},
run_text2pcap_capinfos_tshark(self,
run_text2pcap_capinfos_tshark(
"0000 08 00 bb b3 d7 3b 00 00 51 a7 d6 7d 00 04 51 e4\n" +
"0010 08 09 0a 0b 0c 0d 0e 0f 10 11 12 13 14 15 16 17\n" +
"0020 18 19 1a 1b 1c 1d 1e 1f 20 21 22 23 24 25 26 27\n" +
"0030 28 29 2a 2b 2c 2d 2e 2f 30 31 32 33 34 35 36 37\n",
("-i", "1", "-4", "127.0.0.1,127.0.0.1")))
def test_text2pcap_i_icmpv6_ipv6(self):
def test_text2pcap_i_icmpv6_ipv6(self, run_text2pcap_capinfos_tshark):
'''Test -i <proto> with IPv6 (-6) header'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 86, 'expert': ''},
run_text2pcap_capinfos_tshark(self,
run_text2pcap_capinfos_tshark(
"0000 87 00 f2 62 00 00 00 00 fe 80 00 00 00 00 00 00\n" +
"0010 00 00 00 00 00 00 00 02 01 01 52 54 00 12 34 56\n",
("-i", "58", "-6", "::1,::1")))
def test_text2pcap_i_sctp_ipv6(self):
def test_text2pcap_i_sctp_ipv6(self, run_text2pcap_capinfos_tshark):
'''Test -i <proto> with IPv6 (-6) header'''
self.assertEqual({'encapsulation': 'Ethernet', 'packets': 1,
'datasize': 90, 'expert': ''},
run_text2pcap_capinfos_tshark(self,
run_text2pcap_capinfos_tshark(
"0000 0b 59 0b 59 00 00 00 00 26 98 58 51 00 03 00 18\n" +
"0010 00 00 00 00 00 00 00 00 00 00 00 03 01 00 03 03\n" +
"0020 00 00 00 08\n",