Skip to content

Commit

Permalink
tlsfuzzer/extract.py: Add support for {r,s} tuples in sigs file
Browse files Browse the repository at this point in the history
  • Loading branch information
@GeorgePantelakis
GeorgePantelakis committed May 6, 2024
1 parent 8dbcb98 commit c33dffd
Show file tree
Hide file tree
Showing 6 changed files with 165 additions and 49 deletions.
4 changes: 4 additions & 0 deletions tests/measurements_test_files/data_r_and_s.bin
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
6-0��
���"6�P�$���Y`�u׌� ��0I��I�ߗʎ>�=�[E�N�F�
:�I��ߜ>�k�͆�?E�-�G,�[���|��Ѹ�BK�ҡ?u ��˽2`���9�I�e1�3�r�{��o뿦Q�;K�k�v[Kq_��z$�b�|;$Ip%�G#q#���Sq�����'��{K�g%,�V�{wp�A�Q��Jb8�p�.B_��ϡX��M�o8��*t4�O.<���������#;�2�#l� +���a �58����g���xĆ�+
-:�g5��1�OV+S��ez��ͻ��bi
Expand Down
5 changes: 5 additions & 0 deletions tests/measurements_test_files/priv_key_r_and_s.pem
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
-----BEGIN PRIVATE KEY-----
MIGHAgEBMBMGByqGSM49AgEGCCqGSM49AwEHBG0wawIBAQQgmBZ3w97A7vU3uUW3R05Xrrtuz25s
QRMXx05VOo410f+hRANCAAQwiv85bw4LWmyCH3+CXxbnIx5VkhWOeN07XXG0gSiIaFLe36c5aDnN
vndtWj9hvWxgXK4UIZ3G0MNrtv9pDNW2
-----END PRIVATE KEY-----
Binary file added tests/measurements_test_files/sigs_r_and_s.bin
View file
Binary file not shown.
Binary file added tests/measurements_test_files/times_r_and_s.bin
View file
Binary file not shown.
120 changes: 101 additions & 19 deletions tests/test_tlsfuzzer_extract.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
from os.path import join, dirname, abspath
import hashlib
from random import choice
import ecdsa

from tlsfuzzer.utils.log import Log

Expand Down Expand Up @@ -266,7 +267,7 @@ def test_command_line(self, mock_parse, mock_write, mock_write_pkt,
no_quickack=False, delay=None, carriage_return=None,
data=None, data_size=None, sigs=None, priv_key=None,
key_type=None, frequency=None, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_measurements.assert_not_called()

@mock.patch(
Expand Down Expand Up @@ -303,7 +304,7 @@ def test_delay_and_CR(self, mock_parse, mock_write, mock_write_pkt,
no_quickack=False, delay=3.5, carriage_return='\n',
data=None, data_size=None, sigs=None, priv_key=None,
key_type=None, frequency=None, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_measurements.assert_not_called()

@mock.patch(
Expand Down Expand Up @@ -339,7 +340,7 @@ def test_no_quickack(self, mock_parse, mock_write, mock_write_pkt,
no_quickack=True, delay=None, carriage_return=None,
data=None, data_size=None, sigs=None, priv_key=None,
key_type=None, frequency=None, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_measurements.assert_not_called()

@mock.patch(
Expand Down Expand Up @@ -372,7 +373,7 @@ def test_raw_times(self, mock_parse, mock_write, mock_write_pkt, mock_log,
no_quickack=False, delay=None, carriage_return=None,
data=None, data_size=None, sigs=None, priv_key=None,
key_type=None, frequency=None, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_measurements.assert_not_called()

@mock.patch(
Expand Down Expand Up @@ -406,7 +407,7 @@ def test_raw_binary_times(self, mock_parse, mock_write, mock_write_pkt,
no_quickack=False, delay=None, carriage_return=None,
data=None, data_size=None, sigs=None, priv_key=None,
key_type=None, frequency=None, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_measurements.assert_not_called()

@mock.patch('tlsfuzzer.extract.Log')
Expand Down Expand Up @@ -510,7 +511,7 @@ def test_raw_times_with_column_name(self, mock_parse, mock_write,
no_quickack=False, delay=None, carriage_return=None,
data=None, data_size=None, sigs=None, priv_key=None,
key_type=None, frequency=None, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_measurements.assert_not_called()

@mock.patch('__main__.__builtins__.print')
Expand Down Expand Up @@ -582,7 +583,7 @@ def test_rsa_keys_options(self, mock_parse, mock_write, mock_process,
data=None, data_size=None, sigs=None,
priv_key=None, key_type=None, frequency=None,
hash_func=hashlib.sha256, workers=None, verbose=False,
rsa_keys=priv_key)
rsa_keys=priv_key, r_and_s=False)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
Expand Down Expand Up @@ -623,7 +624,7 @@ def test_ecdsa_signs_options(self, mock_parse, mock_process, mock_write,
data=raw_data, data_size=data_size, sigs=raw_sigs,
priv_key=priv_key, key_type="ecdsa", frequency=None,
hash_func=hashlib.sha256, workers=None, verbose=False,
rsa_keys=None)
rsa_keys=None, r_and_s=False)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
Expand Down Expand Up @@ -664,7 +665,7 @@ def test_verbose_option(self, mock_parse, mock_process, mock_write,
data=raw_data, data_size=data_size, sigs=raw_sigs,
priv_key=priv_key, key_type="ecdsa", frequency=None,
hash_func=hashlib.sha256, workers=None, verbose=True,
rsa_keys=None)
rsa_keys=None, r_and_s=False)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
Expand Down Expand Up @@ -706,7 +707,7 @@ def test_frequency_option(self, mock_parse, mock_process, mock_write,
data=raw_data, data_size=data_size, sigs=raw_sigs,
priv_key=priv_key, key_type="ecdsa",
frequency=frequency * 1e6, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
Expand Down Expand Up @@ -748,7 +749,7 @@ def test_hash_func_option(self, mock_parse, mock_process, mock_write,
data=raw_data, data_size=data_size, sigs=raw_sigs,
priv_key=priv_key, key_type="ecdsa",
frequency=None, hash_func=hashlib.sha384,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
Expand Down Expand Up @@ -789,7 +790,7 @@ def test_prehashed_option(self, mock_parse, mock_process, mock_write,
data=raw_data, data_size=data_size, sigs=raw_sigs,
priv_key=priv_key, key_type="ecdsa",
frequency=None, hash_func=None,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
Expand Down Expand Up @@ -831,7 +832,8 @@ def test_workers_option(self, mock_parse, mock_process, mock_write,
data=raw_data, data_size=data_size, sigs=raw_sigs,
priv_key=priv_key, key_type="ecdsa",
frequency=None, hash_func=hashlib.sha256,
workers=workers, verbose=False, rsa_keys=None)
workers=workers, verbose=False, rsa_keys=None,
r_and_s=False)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
Expand Down Expand Up @@ -872,7 +874,7 @@ def test_skip_invert_option(self, mock_parse, mock_process, mock_write,
data=raw_data, data_size=data_size, sigs=raw_sigs,
priv_key=priv_key, key_type="ecdsa",
frequency=None, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None)
workers=None, verbose=False, rsa_keys=None, r_and_s=False)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
Expand All @@ -882,6 +884,47 @@ def test_skip_invert_option(self, mock_parse, mock_process, mock_write,
for mode in files_passes_in_process.values():
self.assertNotIn("invert", mode)

@mock.patch('tlsfuzzer.extract.Log')
@mock.patch('tlsfuzzer.extract.Extract._write_pkts')
@mock.patch('tlsfuzzer.extract.Extract._write_csv')
@mock.patch(
'tlsfuzzer.extract.Extract.process_and_create_multiple_csv_files'
)
@mock.patch('tlsfuzzer.extract.Extract.parse')
def test_skip_invert_option(self, mock_parse, mock_process, mock_write,
mock_write_pkt, mock_log):
output = "/tmp"
raw_data = "/tmp/data"
data_size = 32
raw_sigs = "/tmp/sigs"
raw_times = "/tmp/times"
priv_key = "/tmp/key"
args = ["extract.py",
"-o", output,
"--raw-data", raw_data,
"--data-size", data_size,
"--raw-sigs", raw_sigs,
"--raw-times", raw_times,
"--priv-key-ecdsa", priv_key,
"--r-and-s"]
mock_init = mock.Mock()
mock_init.return_value = None
with mock.patch('tlsfuzzer.extract.Extract.__init__', mock_init):
with mock.patch("sys.argv", args):
main()
mock_init.assert_called_once_with(
mock.ANY, None, output, None, None,
raw_times, None, binary=None, endian="little",
no_quickack=False, delay=None, carriage_return=None,
data=raw_data, data_size=data_size, sigs=raw_sigs,
priv_key=priv_key, key_type="ecdsa",
frequency=None, hash_func=hashlib.sha256,
workers=None, verbose=False, rsa_keys=None, r_and_s=True)
mock_write.assert_not_called()
mock_write_pkt.assert_not_called()
mock_log.assert_not_called()
mock_process.assert_called_once()

def test_specify_to_private_keys(self):
args = [
"extract.py", "-o", "/tmp", "--raw-data", "/tmp/data",
Expand Down Expand Up @@ -1383,15 +1426,16 @@ def setUp(self):
self.times_used_write_on_hamming = 0
self.k_time_map = []

out_dir = join(dirname(abspath(__file__)), "measurements_test_files")
common_dir = "measurements_test_files"
out_dir = join(dirname(abspath(__file__)), common_dir)
raw_times = join(dirname(abspath(__file__)),
"measurements_test_files", "times.bin")
common_dir, "times.bin")
raw_sigs = join(dirname(abspath(__file__)),
"measurements_test_files", "sigs.bin")
common_dir, "sigs.bin")
raw_data = join(dirname(abspath(__file__)),
"measurements_test_files", "data.bin")
common_dir, "data.bin")
priv_key = join(dirname(abspath(__file__)),
"measurements_test_files", "priv_key.pem")
common_dir, "priv_key.pem")

self.extract = Extract(
output=out_dir, raw_times=raw_times, binary=8,
Expand Down Expand Up @@ -1455,6 +1499,7 @@ def test_measurement_creation_with_verbose_and_frequency(
):
self.extract.frequency = 1
self.extract.verbose = True
self.k_time_map = []

mock_file.side_effect = self.file_emulator
self.times_used_write = 0
Expand All @@ -1471,6 +1516,43 @@ def test_measurement_creation_with_verbose_and_frequency(
self.times_used_write = 0
self.extract.frequency = None
self.extract.verbose = False
self.k_time_map = []

@mock.patch('builtins.print')
def test_measurement_creation_r_and_s(self, mock_print):
self.k_time_map = []
common_dir = "measurements_test_files"
out_dir = join(dirname(abspath(__file__)), common_dir)
raw_times = join(dirname(abspath(__file__)),
common_dir, "times_r_and_s.bin")
raw_sigs = join(dirname(abspath(__file__)),
common_dir, "sigs_r_and_s.bin")
raw_data = join(dirname(abspath(__file__)),
common_dir, "data_r_and_s.bin")
priv_key = join(dirname(abspath(__file__)),
common_dir, "priv_key_r_and_s.pem")

extract = Extract(
output=out_dir, raw_times=raw_times, binary=8,
sigs=raw_sigs, data=raw_data, data_size=32, priv_key=priv_key,
key_type="ecdsa", hash_func=None, r_and_s=True
)

self.times_used_write = 0

with mock.patch('__main__.__builtins__.open') as mock_file:
mock_file.side_effect = self.file_emulator
extract.process_measurements_and_create_csv_file(
extract.ecdsa_iter(), extract.ecdsa_max_value()
)

self.assertGreater(
self.times_used_write, 0,
"At least one measurement should have been written."
)

self.times_used_write = 0
self.k_time_map = []

@mock.patch('__main__.__builtins__.open')
def test_measurement_creation_with_k_size_invert(
Expand Down
Loading

0 comments on commit c33dffd

Please sign in to comment.