HackRF-Treasure-Chest/Software/Universal Radio Hacker/tests/cli/test_cli_parsing.py
2022-09-22 13:46:47 -07:00

188 lines
9.7 KiB
Python

import os
import shutil
import tempfile
from tests.QtTestCase import QtTestCase
from urh.cli import urh_cli
from urh.dev.BackendHandler import Backends
from urh.dev.VirtualDevice import Mode
class TestCLIParsing(QtTestCase):
def setUp(self):
self.parser = urh_cli.create_parser()
def test_build_modulator_from_args(self):
args = self.parser.parse_args("--device HackRF --frequency 433.92e6 --sample-rate 2e6 --raw".split())
self.assertIsNone(urh_cli.build_modulator_from_args(args))
args = self.parser.parse_args("--device HackRF --frequency 433.92e6 --sample-rate 2e6".split())
with self.assertRaises(ValueError):
urh_cli.build_modulator_from_args(args)
args = self.parser.parse_args("--device HackRF --frequency 433.92e6 --sample-rate 2e6 -p0 0".split())
with self.assertRaises(ValueError):
urh_cli.build_modulator_from_args(args)
args = self.parser.parse_args("--device HackRF --frequency 433.92e6 --sample-rate 2e6"
" -pm 0 1 -mo ASK -cf 1337e3 -ca 0.9 -sps 24 -cp 30".split())
modulator = urh_cli.build_modulator_from_args(args)
self.assertEqual(modulator.modulation_type, "ASK")
self.assertEqual(modulator.sample_rate, 2e6)
self.assertEqual(modulator.samples_per_symbol, 24)
self.assertEqual(modulator.parameters[0], 0)
self.assertEqual(modulator.parameters[1], 100)
self.assertEqual(modulator.carrier_freq_hz, 1337e3)
self.assertEqual(modulator.carrier_amplitude, 0.9)
self.assertEqual(modulator.carrier_phase_deg, 30)
args = self.parser.parse_args("--device HackRF --frequency 433.92e6 --sample-rate 2e6"
" -pm 10% 20% -mo ASK -cf 1337e3 -ca 0.9 -sps 24 -cp 30".split())
modulator = urh_cli.build_modulator_from_args(args)
self.assertEqual(modulator.parameters[0], 10)
self.assertEqual(modulator.parameters[1], 20)
args = self.parser.parse_args("--device HackRF --frequency 433.92e6 --sample-rate 2e6"
" -pm 20e3 -20000 -mo FSK -cf 1337e3 -ca 0.9 -sps 24 -cp 30".split())
modulator = urh_cli.build_modulator_from_args(args)
self.assertEqual(modulator.modulation_type, "FSK")
self.assertEqual(modulator.parameters[0], 20e3)
self.assertEqual(modulator.parameters[1], -20e3)
def test_build_backend_handler_from_args(self):
args = self.parser.parse_args("--device USRP --frequency 433.92e6 --sample-rate 2e6".split())
bh = urh_cli.build_backend_handler_from_args(args)
self.assertEqual(bh.device_backends["usrp"].selected_backend, Backends.native)
args = self.parser.parse_args("--device HackRF --frequency 433.92e6 --sample-rate 2e6"
" --device-backend native".split())
bh = urh_cli.build_backend_handler_from_args(args)
self.assertEqual(bh.device_backends["hackrf"].selected_backend, Backends.native)
args = self.parser.parse_args("--device RTL-SDR --frequency 433.92e6 --sample-rate 2e6"
" --device-backend gnuradio".split())
bh = urh_cli.build_backend_handler_from_args(args)
self.assertEqual(bh.device_backends["rtl-sdr"].selected_backend, Backends.grc)
def test_build_device_from_args(self):
args = self.parser.parse_args("--device HackRF --frequency 133.7e6 --sample-rate 2.5e6 -rx "
"-if 24 -bb 30 -g 0 --device-identifier abcde".split())
device = urh_cli.build_device_from_args(args)
self.assertEqual(device.sample_rate, 2.5e6)
self.assertEqual(device.bandwidth, 2.5e6)
self.assertEqual(device.name, "HackRF")
self.assertEqual(device.backend, Backends.native)
self.assertEqual(device.frequency, 133.7e6)
self.assertEqual(device.mode, Mode.receive)
self.assertEqual(device.if_gain, 24)
self.assertEqual(device.gain, 0)
self.assertEqual(device.baseband_gain, 30)
self.assertEqual(device.device_serial, "abcde")
args = self.parser.parse_args("--device RTL-SDR --frequency 133.7e6 --sample-rate 1e6 "
"-rx -db native --device-identifier 42".split())
device = urh_cli.build_device_from_args(args)
self.assertEqual(device.sample_rate, 1e6)
self.assertEqual(device.name, "RTL-SDR")
self.assertEqual(device.backend, Backends.native)
self.assertEqual(device.frequency, 133.7e6)
self.assertEqual(device.mode, Mode.receive)
self.assertEqual(device.device_number, 42)
args = self.parser.parse_args("--device HackRF --frequency 133.7e6 --sample-rate 2.5e6 --bandwidth 5e6 "
"-tx -db native".split())
device = urh_cli.build_device_from_args(args)
self.assertEqual(device.sample_rate, 2.5e6)
self.assertEqual(device.bandwidth, 5e6)
self.assertEqual(device.name, "HackRF")
self.assertEqual(device.backend, Backends.native)
self.assertEqual(device.frequency, 133.7e6)
self.assertEqual(device.mode, Mode.send)
def test_build_protocol_sniffer_from_args(self):
args = self.parser.parse_args("--device HackRF --frequency 50e3 --sample-rate 2.5e6 -rx "
"-if 24 -bb 30 -g 0 --device-identifier abcde "
"-sps 1337 --center 0.5 --noise 0.1234 --tolerance 42 "
"-cs 0.42 -bps 4".split())
sniffer = urh_cli.build_protocol_sniffer_from_args(args)
self.assertEqual(sniffer.rcv_device.frequency, 50e3)
self.assertEqual(sniffer.rcv_device.sample_rate, 2.5e6)
self.assertEqual(sniffer.rcv_device.bandwidth, 2.5e6)
self.assertEqual(sniffer.rcv_device.name, "hackrf")
self.assertEqual(sniffer.rcv_device.backend, Backends.native)
self.assertEqual(sniffer.rcv_device.mode, Mode.receive)
self.assertEqual(sniffer.rcv_device.if_gain, 24)
self.assertEqual(sniffer.rcv_device.gain, 0)
self.assertEqual(sniffer.rcv_device.baseband_gain, 30)
self.assertEqual(sniffer.rcv_device.device_serial, "abcde")
self.assertEqual(sniffer.signal.samples_per_symbol, 1337)
self.assertEqual(sniffer.signal.bits_per_symbol, 4)
self.assertEqual(sniffer.signal.center_spacing, 0.42)
self.assertEqual(sniffer.signal.noise_threshold, 0.1234)
self.assertEqual(sniffer.signal.center, 0.5)
self.assertEqual(sniffer.signal.tolerance, 42)
def test_build_encoding_from_args(self):
args = self.parser.parse_args('--device HackRF --frequency 50e3 --sample-rate 2.5e6 -e "Test,Invert"'.split())
encoding = urh_cli.build_encoding_from_args(args)
self.assertEqual(len(encoding.chain), 2)
def test_read_messages_to_send(self):
args = self.parser.parse_args('--device HackRF --frequency 50e3 --sample-rate 2e6 -rx'.split())
self.assertIsNone(urh_cli.read_messages_to_send(args))
args = self.parser.parse_args('--device HackRF --frequency 50e3 --sample-rate 2e6 -tx'.split())
with self.assertRaises(SystemExit):
urh_cli.read_messages_to_send(args)
args = self.parser.parse_args('--device HackRF --frequency 50e3 --sample-rate 2e6 -tx '
'-file /tmp/test -m 1111'.split())
with self.assertRaises(SystemExit):
urh_cli.read_messages_to_send(args)
test_messages = ["101010/1s", "10000/50ms", "00001111/100.5µs", "111010101/500ns", "1111001", "111110000/2000"]
args = self.parser.parse_args(('--device HackRF --frequency 50e3 --sample-rate 2e6 -tx --pause 1337 '
'-m ' + " ".join(test_messages)).split())
messages = urh_cli.read_messages_to_send(args)
self.assertEqual(len(messages), len(test_messages))
self.assertEqual(messages[0].decoded_bits_str, "101010")
self.assertEqual(messages[0].pause, 2e6)
self.assertEqual(messages[1].decoded_bits_str, "10000")
self.assertEqual(messages[1].pause, 100e3)
self.assertEqual(messages[2].decoded_bits_str, "00001111")
self.assertEqual(messages[2].pause, 201)
self.assertEqual(messages[3].decoded_bits_str, "111010101")
self.assertEqual(messages[3].pause, 1)
self.assertEqual(messages[4].decoded_bits_str, "1111001")
self.assertEqual(messages[4].pause, 1337)
self.assertEqual(messages[5].decoded_bits_str, "111110000")
self.assertEqual(messages[5].pause, 2000)
test_messages = ["aabb/2s"]
filepath = tempfile.mktemp()
with open(filepath, "w") as f:
f.write("\n".join(test_messages))
args = self.parser.parse_args(('--device HackRF --frequency 50e3 --sample-rate 2e6 -tx --pause 1337 --hex '
'-file ' + filepath).split())
messages = urh_cli.read_messages_to_send(args)
self.assertEqual(len(messages), len(test_messages))
self.assertEqual(messages[0].decoded_bits_str, "1010101010111011")
self.assertEqual(messages[0].pause, 4e6)
def test_parse_project_file(self):
f = os.readlink(__file__) if os.path.islink(__file__) else __file__
path = os.path.realpath(os.path.join(f, ".."))
project_file = os.path.realpath(os.path.join(path, "..", "data", "TestProjectForCLI.xml"))
tmp_project_file = os.path.join(tempfile.mkdtemp(), "URHProject.xml")
shutil.copy(project_file, tmp_project_file)
project_params = urh_cli.parse_project_file(tmp_project_file)
self.assertGreater(len(project_params), 0)