Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 12 additions & 17 deletions modules/test/tls/python/src/tls_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -411,19 +411,19 @@ def _security_tls_v1_2_server(self):
LOGGER.info(f'Inspecting Service on port {port}: {service_type}')
tls_1_2_results = self._tls_util.validate_tls_server(
host=self._device_ipv4_addr, port=port, tls_version='1.2')
tls_1_3_results = self._tls_util.validate_tls_server(
host=self._device_ipv4_addr, port=port, tls_version='1.3')
# If TLS 1.2 is not supported don't process the results
if tls_1_2_results[0] is not None:
port_results = self._tls_util.process_tls_server_results(
tls_1_2_results, tls_1_3_results, port=port)
else:
port_results = None
if port_results is not None:
result = port_results[
0] if result is None else result and port_results[0]
details.extend(port_results[1])
if port_results[0]:
if result is None:
result = tls_1_2_results[0]
else:
result = result and tls_1_2_results[0]
status_str = '' if tls_1_2_results[0] else 'not '
details.append(
f'TLS 1.2 {status_str}validated on port {port}:')
if isinstance(tls_1_2_results[1], list):
details.extend(tls_1_2_results[1])
else:
details.append(tls_1_2_results[1])
if tls_1_2_results[0]:
ports_valid.append(port)
else:
ports_invalid.append(port)
Expand All @@ -439,11 +439,6 @@ def _security_tls_v1_2_server(self):
result = 'Feature Not Detected'
description = 'TLS 1.2 certificate could not be validated'
details.append('TLS 1.2 certificate could not be validated.')
# If TLS 1.2 cert is not valid but TLS 1.3 is valid test is Compliant
elif result and not tls_1_2_results[0] and tls_1_3_results[0]:
ports_csv = ','.join(map(str,ports_valid))
description = 'TLS 1.2 certificate invalid and '
description += f'TLS 1.3 certificate valid on ports: {ports_csv}'
elif result:
ports_csv = ','.join(map(str,ports_valid))
description = f'TLS 1.2 certificate valid on ports: {ports_csv}'
Expand Down
62 changes: 22 additions & 40 deletions testing/unit/tls/tls_module_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,11 @@
LOCAL_REPORT_EXT = os.path.join(REPORTS_DIR, 'tls_report_ext_local.html')
LOCAL_REPORT_NO_CERT = os.path.join(REPORTS_DIR,
'tls_report_no_cert_local.html')
CONF_FILE = 'modules/test/' + MODULE + '/conf/module_config.json'
CONF_FILE = (
'/testrun/conf/module_config.json'
if os.path.exists('/testrun/conf/module_config.json')
else 'modules/test/' + MODULE + '/conf/module_config.json'
)

INTERNET_IFACE = 'eth0'

Expand All @@ -69,12 +73,14 @@ def setUpClass(cls):

# Setup the default ipv4 address and the scan results
def setUp(self):
self.tls_module = TLSModule(module=MODULE)
self.tls_module = TLSModule(module=MODULE, conf_file=CONF_FILE)
self.tls_module._device_ipv4_addr = None # pylint: disable=W0212
self.tls_module._scan_results = None # pylint: disable=W0212

def security_tls_v1_2_server_no_ip_test(self):
@patch('tls_module.TLSModule._get_device_ipv4')
def security_tls_v1_2_server_no_ip_test(self, mock_get_device_ipv4):
"""Test _security_tls_v1_2_server when device IP could not be resolved"""
mock_get_device_ipv4.return_value = None

result, description, details = self.tls_module._security_tls_v1_2_server() # pylint: disable=W0212

Expand Down Expand Up @@ -162,11 +168,10 @@ def validate_side_effect(**kwargs):
self.assertEqual(details, ['TLS 1.2 certificate could not be validated.'])

@patch('tls_module.TLSUtil.validate_tls_server')
def security_tls_v1_2_server_compliant_invalid_v1_2_cert_test(self,
def security_tls_v1_2_server_invalid_v1_2_cert_test(self,
mock_validate_tls_server):
"""
Test _security_tls_v1_2_server when TLS 1.2 cert is invalid but
TLS 1.3 cert is valid
Test _security_tls_v1_2_server when TLS 1.2 cert is invalid
"""

self.tls_module._device_ipv4_addr = '10.10.10.14' # pylint: disable=W0212
Expand All @@ -177,30 +182,21 @@ def validate_side_effect(**kwargs):
tls_version = kwargs.get('tls_version')
if tls_version == '1.2':
return (False, ['Certificate has expired'])
elif tls_version == '1.3':
return (True, ['Time range valid',
'Public key valid',
'Signature valid']
)

mock_validate_tls_server.side_effect = validate_side_effect
result, description, details = self.tls_module._security_tls_v1_2_server() # pylint: disable=W0212

# Expects compliant result
self.assertEqual(result, True)
# Expects non-compliant result
self.assertEqual(result, False)

expected_description = (
'TLS 1.2 certificate invalid and TLS 1.3 certificate valid on ports: 443'
'TLS 1.2 certificate invalid on ports: 443'
)
self.assertEqual(description, expected_description )

expected_details = [
'TLS 1.2 not validated on port 443:',
'Certificate has expired',
'TLS 1.3 validated on port 443:',
'Time range valid',
'Public key valid',
'Signature valid'
'Certificate has expired'
]
self.assertEqual(details, expected_details)

Expand Down Expand Up @@ -231,9 +227,7 @@ def validate_side_effect(**kwargs):

expected_details = [
'TLS 1.2 not validated on port 443:',
'Certificate has expired',
'TLS 1.3 not validated on port 443:',
'Device certificate has not been signed'
'Certificate has expired'
]
self.assertEqual(details, expected_details)

Expand Down Expand Up @@ -265,10 +259,6 @@ def validate_side_effect(**kwargs):
'TLS 1.2 validated on port 443:',
'Time range valid',
'Public key valid',
'Signature valid',
'TLS 1.3 validated on port 443:',
'Time range valid',
'Public key valid',
'Signature valid'
]
self.assertEqual(details, expected_details)
Expand Down Expand Up @@ -303,18 +293,10 @@ def validate_side_effect(**kwargs):
'Time range valid',
'Public key valid',
'Signature valid',
'TLS 1.3 validated on port 443:',
'Time range valid',
'Public key valid',
'Signature valid',
'TLS 1.2 validated on port 8443:',
'Time range valid',
'Public key valid',
'Signature valid',
'TLS 1.3 validated on port 8443:',
'Time range valid',
'Public key valid',
'Signature valid',
]
self.assertEqual(details, expected_details)

Expand Down Expand Up @@ -345,10 +327,6 @@ def validate_side_effect(**kwargs):
'Time range valid',
'Public key valid',
'Signature valid',
'TLS 1.3 validated on port 443:',
'Time range valid',
'Public key valid',
'Signature valid',
'HTTP service detected on port 80.'
]
self.assertEqual(details, expected_details)
Expand Down Expand Up @@ -640,7 +618,7 @@ def outbound_connections_report_test(self):
capture_file = os.path.join(CAPTURES_DIR, 'monitor.pcap')
ip_dst = TLS_UTIL.get_all_outbound_connections(
device_mac='70:b3:d5:96:c0:00', capture_files=[capture_file])
tls = TLSModule(module=MODULE)
tls = TLSModule(module=MODULE, conf_file=CONF_FILE)
gen_html = tls.generate_outbound_connection_table(ip_dst)
print(gen_html)

Expand All @@ -651,6 +629,7 @@ def tls_module_report_multi_page_test(self):
monitor_pcap_file = os.path.join(CAPTURES_DIR, 'multi_page_monitor.pcap')
tls_pcap_file = os.path.join(CAPTURES_DIR, 'multi_page_tls.pcap')
tls = TLSModule(module=MODULE,
conf_file=CONF_FILE,
results_dir=OUTPUT_DIR,
startup_capture_file=startup_pcap_file,
monitor_capture_file=monitor_pcap_file,
Expand All @@ -674,6 +653,7 @@ def tls_module_report_test(self):
os.environ['DEVICE_MAC'] = '38:d1:35:01:17:fe'
pcap_file = os.path.join(CAPTURES_DIR, 'tls.pcap')
tls = TLSModule(module=MODULE,
conf_file=CONF_FILE,
results_dir=OUTPUT_DIR,
startup_capture_file=pcap_file,
monitor_capture_file=pcap_file,
Expand All @@ -692,6 +672,7 @@ def tls_module_report_ext_test(self):
os.environ['DEVICE_MAC'] = '28:29:86:27:d6:05'
pcap_file = os.path.join(CAPTURES_DIR, 'tls_ext.pcap')
tls = TLSModule(module=MODULE,
conf_file=CONF_FILE,
results_dir=OUTPUT_DIR,
startup_capture_file=pcap_file,
monitor_capture_file=pcap_file,
Expand All @@ -718,6 +699,7 @@ def tls_module_report_no_cert_test(self):
os.environ['DEVICE_MAC'] = ''
pcap_file = os.path.join(CAPTURES_DIR, 'tls_ext.pcap')
tls = TLSModule(module=MODULE,
conf_file=CONF_FILE,
results_dir=OUTPUT_DIR,
startup_capture_file=pcap_file,
monitor_capture_file=pcap_file,
Expand Down Expand Up @@ -902,7 +884,7 @@ def download_public_cert(self, hostname, port=443):
suite.addTest(TLSModuleTest('security_tls_v1_2_multiple_https_servers_test'))
suite.addTest(TLSModuleTest('security_tls_v1_2_server_http_test'))
suite.addTest(
TLSModuleTest('security_tls_v1_2_server_compliant_invalid_v1_2_cert_test')
TLSModuleTest('security_tls_v1_2_server_invalid_v1_2_cert_test')
)
suite.addTest(
TLSModuleTest(
Expand Down