def test_protocol_sniffer(self): samples_per_symbol = 100 center = 0.0942 noise = 0.1 tolerance = 2 modulation_type = "FSK" sample_rate = 1e6 device_name = NetworkSDRInterfacePlugin.NETWORK_SDR_NAME sniffer = ProtocolSniffer(samples_per_symbol=samples_per_symbol, center=center, center_spacing=0.1, noise=noise, tolerance=tolerance, modulation_type=modulation_type, bits_per_symbol=1, device=device_name, backend_handler=BackendHandler(), network_raw_mode=True) port = self.get_free_port() sniffer.rcv_device.set_server_port(port) self.network_sdr_plugin_sender = NetworkSDRInterfacePlugin(raw_mode=True) self.network_sdr_plugin_sender.client_port = port sniffer.sniff() QTest.qWait(10) data = ["101010", "000111", "1111000"] pause = 10 * samples_per_symbol modulator = Modulator("test") modulator.samples_per_symbol = samples_per_symbol modulator.sample_rate = sample_rate modulator.modulation_type = modulation_type modulator.parameters[1] = 20e3 modulator.parameters[0] = 10e3 packages = [] for d in data: packages.append(modulator.modulate(list(map(int, d)), pause)) # verify modulation was correct pa = ProtocolAnalyzer(None) signal = Signal("", "", sample_rate=sample_rate) signal.iq_array = IQArray.concatenate(packages) signal.modulation_type = modulation_type signal.samples_per_symbol = samples_per_symbol signal.tolerance = tolerance signal.noise_threshold = noise signal.center = center pa.signal = signal pa.get_protocol_from_signal() self.assertEqual(pa.plain_bits_str, data) # send data send_data = IQArray.concatenate(packages) self.network_sdr_plugin_sender.send_raw_data(send_data, 1) time.sleep(1) # Send enough pauses to end sniffing self.network_sdr_plugin_sender.send_raw_data(IQArray(None, np.float32, 10 * 2 * samples_per_symbol), 1) time.sleep(1) sniffer.stop() self.assertEqual(sniffer.plain_bits_str, data)
def test_protocol_sniffer(self): bit_len = 100 center = 0.0942 noise = 0.1 tolerance = 2 modulation_type = 1 sample_rate = 1e6 device_name = NetworkSDRInterfacePlugin.NETWORK_SDR_NAME sniffer = ProtocolSniffer(bit_len=bit_len, center=center, noise=noise, tolerance=tolerance, modulation_type=modulation_type, device=device_name, backend_handler=BackendHandler(), network_raw_mode=True) port = self.get_free_port() sniffer.rcv_device.set_server_port(port) self.network_sdr_plugin_sender = NetworkSDRInterfacePlugin(raw_mode=True) self.network_sdr_plugin_sender.client_port = port sniffer.sniff() QTest.qWait(10) data = ["101010", "000111", "1111000"] pause = 10 * bit_len modulator = Modulator("test") modulator.samples_per_bit = bit_len modulator.sample_rate = sample_rate modulator.modulation_type = modulation_type modulator.param_for_one = 20e3 modulator.param_for_zero = 10e3 packages = [] for d in data: packages.append(modulator.modulate(list(map(int, d)), pause)) # verify modulation was correct pa = ProtocolAnalyzer(None) signal = Signal("", "", sample_rate=sample_rate) signal._fulldata = np.concatenate(packages) signal.modulation_type = modulation_type signal.bit_len = bit_len signal.tolerance = tolerance signal.noise_threshold = noise signal.qad_center = center pa.signal = signal pa.get_protocol_from_signal() self.assertEqual(pa.plain_bits_str, data) # send data send_data = np.concatenate(packages) self.network_sdr_plugin_sender.send_raw_data(send_data, 1) time.sleep(1) # Send enough pauses to end sniffing self.network_sdr_plugin_sender.send_raw_data(np.zeros(10 * bit_len, dtype=np.complex64), 1) time.sleep(1) sniffer.stop() self.assertEqual(sniffer.plain_bits_str, data)
class ProtocolSniffDialogController(QDialog): protocol_accepted = pyqtSignal(list) def __init__(self, freq, samp_rate, bw, gain, device, noise, center, bit_length, tolerance, modulation_type_index, parent=None): super().__init__(parent) self.ui = Ui_SniffProtocol() self.ui.setupUi(self) self.setAttribute(Qt.WA_DeleteOnClose) self.ui.spinBoxFreq.setValue(freq) self.ui.spinBoxSampleRate.setValue(samp_rate) self.ui.spinBoxBandwidth.setValue(bw) self.ui.spinBoxGain.setValue(gain) self.ui.spinboxNoise.setValue(noise) self.ui.spinboxCenter.setValue(center) self.ui.spinboxBitLen.setValue(bit_length) self.ui.spinboxErrorTolerance.setValue(tolerance) self.ui.comboxModulation.setCurrentIndex(modulation_type_index) self.sniffer = ProtocolSniffer(bit_length, center, noise, tolerance, modulation_type_index, samp_rate, freq, gain, bw, device) self.sniffer.usrp_ip = self.ui.lineEditIP.text() self.ui.btnStop.setEnabled(False) self.ui.btnClear.setEnabled(False) self.ui.cbDevice.clear() items = [] if constants.SETTINGS.value('usrp_available', type=bool): items.append("USRP") if constants.SETTINGS.value('hackrf_available', type=bool): items.append("HackRF") self.ui.cbDevice.addItems(items) if device in items: self.ui.cbDevice.setCurrentIndex(items.index(device)) self.on_device_edited() ipRange = "(?:[0-1]?[0-9]?[0-9]|2[0-4][0-9]|25[0-5])" ipRegex = QRegExp("^" + ipRange + "\\." + ipRange + "\\." + ipRange + "\\." + ipRange + "$") self.ui.lineEditIP.setValidator(QRegExpValidator(ipRegex)) # Auto Complete like a Boss completer = QCompleter() completer.setModel(QDirModel(completer)) self.ui.lineEditOutputFile.setCompleter(completer) self.create_connects() def create_connects(self): self.ui.btnStart.clicked.connect(self.on_start_clicked) self.ui.btnStop.clicked.connect(self.on_stop_clicked) self.ui.btnClear.clicked.connect(self.on_clear_clicked) self.ui.btnAccept.clicked.connect(self.on_btn_accept_clicked) self.ui.btnClose.clicked.connect(self.close) self.sniffer.rcv_thrd.started.connect(self.on_sniffer_rcv_started) self.sniffer.rcv_thrd.stopped.connect(self.on_sniffer_rcv_stopped) self.sniffer.qt_signals.data_sniffed.connect(self.on_data_sniffed) self.sniffer.qt_signals.sniff_device_errors_changed.connect( self.on_device_errors_changed) self.ui.spinBoxSampleRate.editingFinished.connect( self.on_sample_rate_edited) self.ui.spinBoxGain.editingFinished.connect(self.on_gain_edited) self.ui.spinBoxFreq.editingFinished.connect(self.on_freq_edited) self.ui.spinBoxBandwidth.editingFinished.connect(self.on_bw_edited) self.ui.lineEditIP.editingFinished.connect(self.on_usrp_ip_edited) self.ui.cbDevice.currentIndexChanged.connect(self.on_device_edited) self.ui.spinboxNoise.editingFinished.connect(self.on_noise_edited) self.ui.spinboxCenter.editingFinished.connect(self.on_center_edited) self.ui.spinboxBitLen.editingFinished.connect(self.on_bit_len_edited) self.ui.spinboxErrorTolerance.editingFinished.connect( self.on_tolerance_edited) self.ui.comboxModulation.currentIndexChanged.connect( self.on_modulation_changed) self.ui.comboBoxViewType.currentIndexChanged.connect( self.on_view_type_changed) @property def view_type(self): return self.ui.comboBoxViewType.currentIndex() @property def has_empty_device_list(self): return self.ui.cbDevice.count() == 0 @pyqtSlot() def on_sample_rate_edited(self): self.sniffer.rcv_thrd.sample_rate = self.ui.spinBoxSampleRate.value() @pyqtSlot() def on_freq_edited(self): self.sniffer.rcv_thrd.freq = self.ui.spinBoxFreq.value() @pyqtSlot() def on_bw_edited(self): self.sniffer.rcv_thrd.bandwidth = self.ui.spinBoxBandwidth.value() @pyqtSlot() def on_usrp_ip_edited(self): self.sniffer.usrp_ip = self.ui.lineEditIP.text() @pyqtSlot() def on_gain_edited(self): self.sniffer.rcv_thrd.gain = self.ui.spinBoxGain.value() @pyqtSlot() def on_noise_edited(self): self.sniffer.signal._noise_treshold = self.ui.spinboxNoise.value() @pyqtSlot() def on_center_edited(self): self.sniffer.signal.qad_center = self.ui.spinboxCenter.value() @pyqtSlot() def on_bit_len_edited(self): self.sniffer.signal.bit_len = self.ui.spinboxBitLen.value() @pyqtSlot() def on_tolerance_edited(self): self.sniffer.signal.tolerance = self.ui.spinboxErrorTolerance.value() @pyqtSlot(int) def on_modulation_changed(self, new_index: int): self.sniffer.signal.silent_set_modulation_type(new_index) @pyqtSlot() def on_device_edited(self): dev = self.ui.cbDevice.currentText() self.sniffer.device = dev self.ui.lineEditIP.setVisible(dev == "USRP") self.ui.labelIP.setVisible(dev == "USRP") @pyqtSlot() def on_start_clicked(self): self.ui.spinBoxFreq.editingFinished.emit() self.ui.lineEditIP.editingFinished.emit() self.ui.spinBoxBandwidth.editingFinished.emit() self.ui.spinBoxSampleRate.editingFinished.emit() self.ui.spinboxNoise.editingFinished.emit() self.ui.spinboxCenter.editingFinished.emit() self.ui.spinboxBitLen.editingFinished.emit() self.ui.spinboxErrorTolerance.editingFinished.emit() self.sniffer.sniff() @pyqtSlot() def on_stop_clicked(self): self.sniffer.stop() @pyqtSlot() def on_sniffer_rcv_stopped(self): self.ui.btnStart.setEnabled(True) self.ui.btnStop.setEnabled(False) self.ui.btnClear.setEnabled(True) self.ui.spinBoxSampleRate.setEnabled(True) self.ui.spinBoxFreq.setEnabled(True) self.ui.lineEditIP.setEnabled(True) self.ui.spinBoxBandwidth.setEnabled(True) self.ui.spinBoxGain.setEnabled(True) self.ui.cbDevice.setEnabled(True) self.ui.spinboxNoise.setEnabled(True) self.ui.spinboxCenter.setEnabled(True) self.ui.spinboxBitLen.setEnabled(True) self.ui.spinboxErrorTolerance.setEnabled(True) self.ui.comboxModulation.setEnabled(True) @pyqtSlot() def on_sniffer_rcv_started(self): self.ui.txtEditErrors.clear() self.ui.btnStart.setEnabled(False) self.ui.btnClear.setEnabled(False) self.ui.btnStop.setEnabled(True) self.ui.spinBoxSampleRate.setDisabled(True) self.ui.spinBoxFreq.setDisabled(True) self.ui.spinBoxGain.setDisabled(True) self.ui.spinBoxBandwidth.setDisabled(True) self.ui.spinboxNoise.setDisabled(True) self.ui.spinboxCenter.setDisabled(True) self.ui.spinboxBitLen.setDisabled(True) self.ui.spinboxErrorTolerance.setDisabled(True) self.ui.comboxModulation.setDisabled(True) self.ui.lineEditIP.setDisabled(True) self.ui.cbDevice.setDisabled(True) @pyqtSlot() def on_clear_clicked(self): self.ui.btnClear.setEnabled(False) self.ui.txtEdPreview.clear() self.sniffer.clear() @pyqtSlot(int) def on_data_sniffed(self, from_index: int): self.ui.txtEdPreview.appendPlainText(self.sniffer.plain_to_string( self.view_type, start=from_index)) self.ui.txtEdPreview.verticalScrollBar().setValue( self.ui.txtEdPreview.verticalScrollBar().maximum()) @pyqtSlot(int) def on_view_type_changed(self, new_index: int): self.ui.txtEdPreview.setPlainText( self.sniffer.plain_to_string(new_index)) @pyqtSlot() def on_btn_accept_clicked(self): self.protocol_accepted.emit(self.sniffer.blocks) self.close() @pyqtSlot(str) def on_device_errors_changed(self, txt: str): self.ui.txtEditErrors.append(txt) def closeEvent(self, event: QCloseEvent): self.sniffer.stop() event.accept() @pyqtSlot(str) def on_lineEditOutputFile_textChanged(self, text: str): self.sniffer.sniff_file = text self.ui.btnAccept.setDisabled(bool(self.sniffer.sniff_file))
class ProtocolSniffDialogController(SendRecvDialogController): protocol_accepted = pyqtSignal(list) def __init__(self, freq, samp_rate, bw, gain, device, noise, center, bit_length, tolerance, modulation_type_index, parent=None, testing_mode=False): self.is_rx = True super().__init__(freq, samp_rate, bw, gain, device, parent=parent, testing_mode=testing_mode) self.ui.stackedWidget.setCurrentIndex(2) self.hide_send_ui_items() self.hide_receive_ui_items() self.ui.sliderYscale.hide() self.ui.label_y_scale.hide() self.ui.spinbox_sniff_Noise.setValue(noise) self.ui.spinbox_sniff_Center.setValue(center) self.ui.spinbox_sniff_BitLen.setValue(bit_length) self.ui.spinbox_sniff_ErrorTolerance.setValue(tolerance) self.ui.combox_sniff_Modulation.setCurrentIndex(modulation_type_index) device = self.ui.cbDevice.currentText() self.sniffer = ProtocolSniffer(bit_length, center, noise, tolerance, modulation_type_index, samp_rate, freq, gain, bw, device, testing_mode=testing_mode) self.set_sniff_ui_items_visible(True) self.set_device_ui_items_visible( device != NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) # Auto Complete like a Boss completer = QCompleter() completer.setModel(QDirModel(completer)) self.ui.lineEdit_sniff_OutputFile.setCompleter(completer) self.create_connects() def create_connects(self): super().create_connects() self.ui.btnAccept.clicked.connect(self.on_btn_accept_clicked) self.sniffer.started.connect(self.on_sniffer_rcv_started) self.sniffer.stopped.connect(self.on_sniffer_rcv_stopped) self.sniffer.qt_signals.data_sniffed.connect(self.on_data_sniffed) self.sniffer.qt_signals.sniff_device_errors_changed.connect( self.on_device_errors_changed) self.ui.spinbox_sniff_Noise.editingFinished.connect( self.on_noise_edited) self.ui.spinbox_sniff_Center.editingFinished.connect( self.on_center_edited) self.ui.spinbox_sniff_BitLen.editingFinished.connect( self.on_bit_len_edited) self.ui.spinbox_sniff_ErrorTolerance.editingFinished.connect( self.on_tolerance_edited) self.ui.combox_sniff_Modulation.currentIndexChanged.connect( self.on_modulation_changed) self.ui.comboBox_sniff_viewtype.currentIndexChanged.connect( self.on_view_type_changed) self.ui.lineEdit_sniff_OutputFile.textChanged.connect( self.on_line_edit_output_file_text_changed) def set_device_ui_items_visible(self, visible: bool): super().set_device_ui_items_visible(visible) for item in ("spinbox_sniff_Noise", "combox_sniff_Modulation", "label_sniff_Modulation", "spinbox_sniff_Center", "spinbox_sniff_BitLen", "spinbox_sniff_ErrorTolerance", "label_sniff_Noise", "label_sniff_Center", "label_sniff_BitLength", "label_sniff_Tolerance"): getattr(self.ui, item).setVisible(visible) def closeEvent(self, event: QCloseEvent): if hasattr(self, "sniffer"): self.sniffer.stop() event.accept() @property def device(self): if hasattr(self, "sniffer"): return self.sniffer.rcv_device else: return None @device.setter def device(self, value): if hasattr(self, "sniffer"): self.sniffer.rcv_device = value else: pass @property def view_type(self): return self.ui.comboBox_sniff_viewtype.currentIndex() @pyqtSlot() def on_noise_edited(self): self.sniffer.signal._noise_threshold = self.ui.spinbox_sniff_Noise.value( ) @pyqtSlot() def on_center_edited(self): self.sniffer.signal.qad_center = self.ui.spinbox_sniff_Center.value() @pyqtSlot() def on_bit_len_edited(self): self.sniffer.signal.bit_len = self.ui.spinbox_sniff_BitLen.value() @pyqtSlot() def on_tolerance_edited(self): self.sniffer.signal.tolerance = self.ui.spinbox_sniff_ErrorTolerance.value( ) @pyqtSlot(int) def on_modulation_changed(self, new_index: int): self.sniffer.signal.silent_set_modulation_type(new_index) @pyqtSlot() def on_selected_device_changed(self): dev_name = self.ui.cbDevice.currentText() self.sniffer.device_name = dev_name self.set_device_ui_items_visible( dev_name != NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) self.ui.lineEditIP.setVisible(dev_name == "USRP") self.ui.labelIP.setVisible(dev_name == "USRP") @pyqtSlot() def on_start_clicked(self): self.ui.spinBoxFreq.editingFinished.emit() self.ui.lineEditIP.editingFinished.emit() self.ui.spinBoxBandwidth.editingFinished.emit() self.ui.spinBoxSampleRate.editingFinished.emit() self.ui.spinbox_sniff_Noise.editingFinished.emit() self.ui.spinbox_sniff_Center.editingFinished.emit() self.ui.spinbox_sniff_BitLen.editingFinished.emit() self.ui.spinbox_sniff_ErrorTolerance.editingFinished.emit() self.sniffer.sniff() @pyqtSlot() def on_stop_clicked(self): self.sniffer.stop() @pyqtSlot() def on_sniffer_rcv_stopped(self): self.ui.btnStart.setEnabled(True) self.ui.btnStop.setEnabled(False) self.ui.btnClear.setEnabled(True) self.ui.spinBoxSampleRate.setEnabled(True) self.ui.spinBoxFreq.setEnabled(True) self.ui.lineEditIP.setEnabled(True) self.ui.spinBoxBandwidth.setEnabled(True) self.ui.spinBoxGain.setEnabled(True) self.ui.cbDevice.setEnabled(True) self.ui.spinbox_sniff_Noise.setEnabled(True) self.ui.spinbox_sniff_Center.setEnabled(True) self.ui.spinbox_sniff_BitLen.setEnabled(True) self.ui.spinbox_sniff_ErrorTolerance.setEnabled(True) self.ui.combox_sniff_Modulation.setEnabled(True) @pyqtSlot() def on_sniffer_rcv_started(self): self.ui.txtEditErrors.clear() self.ui.btnStart.setEnabled(False) self.ui.btnClear.setEnabled(False) self.ui.btnStop.setEnabled(True) self.ui.spinBoxSampleRate.setDisabled(True) self.ui.spinBoxFreq.setDisabled(True) self.ui.spinBoxGain.setDisabled(True) self.ui.spinBoxBandwidth.setDisabled(True) self.ui.spinbox_sniff_Noise.setDisabled(True) self.ui.spinbox_sniff_Center.setDisabled(True) self.ui.spinbox_sniff_BitLen.setDisabled(True) self.ui.spinbox_sniff_ErrorTolerance.setDisabled(True) self.ui.combox_sniff_Modulation.setDisabled(True) self.ui.lineEditIP.setDisabled(True) self.ui.cbDevice.setDisabled(True) @pyqtSlot() def on_clear_clicked(self): self.ui.btnClear.setEnabled(False) self.ui.txtEd_sniff_Preview.clear() self.sniffer.clear() @pyqtSlot(int) def on_data_sniffed(self, from_index: int): new_data = self.sniffer.plain_to_string(self.view_type, start=from_index, show_pauses=False) if new_data.strip(): self.ui.txtEd_sniff_Preview.appendPlainText(new_data) self.ui.txtEd_sniff_Preview.verticalScrollBar().setValue( self.ui.txtEd_sniff_Preview.verticalScrollBar().maximum()) @pyqtSlot(int) def on_view_type_changed(self, new_index: int): self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.plain_to_string(new_index, show_pauses=False)) @pyqtSlot() def on_btn_accept_clicked(self): self.protocol_accepted.emit(self.sniffer.messages) self.close() @pyqtSlot(str) def on_device_errors_changed(self, txt: str): self.ui.txtEditErrors.append(txt) @pyqtSlot(str) def on_line_edit_output_file_text_changed(self, text: str): self.sniffer.sniff_file = text self.ui.btnAccept.setDisabled(bool(self.sniffer.sniff_file))
class ProtocolSniffDialogController(SendRecvDialogController): protocol_accepted = pyqtSignal(list) def __init__(self, project_manager, noise, center, bit_length, tolerance, modulation_type_index, encodings, encoding_index=0, parent=None, testing_mode=False): super().__init__(project_manager, is_tx=False, parent=parent, testing_mode=testing_mode) self.set_sniff_ui_items_visible(True) self.graphics_view = self.ui.graphicsView_sniff_Preview self.ui.stackedWidget.setCurrentWidget(self.ui.page_sniff) self.hide_send_ui_items() self.hide_receive_ui_items() self.ui.sliderYscale.hide() self.ui.label_y_scale.hide() self.ui.spinbox_sniff_Noise.setValue(noise) self.ui.spinbox_sniff_Center.setValue(center) self.ui.spinbox_sniff_BitLen.setValue(bit_length) self.ui.spinbox_sniff_ErrorTolerance.setValue(tolerance) self.ui.combox_sniff_Modulation.setCurrentIndex(modulation_type_index) self.sniffer = ProtocolSniffer(bit_length, center, noise, tolerance, modulation_type_index, self.ui.cbDevice.currentText(), self.backend_handler) # set really in on_device_started self.scene_manager = None # type: LiveSceneManager self.init_device() self.set_bandwidth_status() self.graphics_view.setScene(self.scene_manager.scene) self.graphics_view.scene_manager = self.scene_manager # Auto Complete like a Boss completer = QCompleter() completer.setModel(QDirModel(completer)) self.ui.lineEdit_sniff_OutputFile.setCompleter(completer) self.setWindowTitle(self.tr("Sniff Protocol")) self.setWindowIcon(QIcon.fromTheme(":/icons/data/icons/sniffer.svg")) self.encodings = encodings for encoding in self.encodings: self.ui.comboBox_sniff_encoding.addItem(encoding.name) self.create_connects() if encoding_index > -1: self.ui.comboBox_sniff_encoding.setCurrentIndex(encoding_index) self.ui.comboBox_sniff_viewtype.setCurrentIndex( constants.SETTINGS.value('default_view', 0, int)) @property def device(self): if hasattr(self, "sniffer"): return self.sniffer.rcv_device else: return None @device.setter def device(self, value): if hasattr(self, "sniffer"): self.sniffer.rcv_device = value else: pass @property def view_type(self): return self.ui.comboBox_sniff_viewtype.currentIndex() def create_connects(self): super().create_connects() self.ui.btnAccept.clicked.connect(self.on_btn_accept_clicked) self.sniffer.qt_signals.data_sniffed.connect(self.on_data_sniffed) self.sniffer.qt_signals.sniff_device_errors_changed.connect( self.on_device_errors_changed) self.ui.spinbox_sniff_Noise.editingFinished.connect( self.on_noise_edited) self.ui.spinbox_sniff_Center.editingFinished.connect( self.on_center_edited) self.ui.spinbox_sniff_BitLen.editingFinished.connect( self.on_bit_len_edited) self.ui.spinbox_sniff_ErrorTolerance.editingFinished.connect( self.on_tolerance_edited) self.ui.combox_sniff_Modulation.currentIndexChanged.connect( self.on_modulation_changed) self.ui.comboBox_sniff_viewtype.currentIndexChanged.connect( self.on_view_type_changed) self.ui.lineEdit_sniff_OutputFile.editingFinished.connect( self.on_line_edit_output_file_editing_finished) self.ui.comboBox_sniff_encoding.currentIndexChanged.connect( self.on_combobox_sniff_encoding_index_changed) def set_device_ui_items_visibility(self, device_name: str, adjust_gains=True): super().set_device_ui_items_visibility(device_name, adjust_gains) visible = device_name != NetworkSDRInterfacePlugin.NETWORK_SDR_NAME for item in ("spinbox_sniff_Noise", "combox_sniff_Modulation", "label_sniff_Modulation", "graphicsView_sniff_Preview", "spinbox_sniff_Center", "spinbox_sniff_BitLen", "spinbox_sniff_ErrorTolerance", "label_sniff_Noise", "label_sniff_Center", "label_sniff_BitLength", "label_sniff_Tolerance"): getattr(self.ui, item).setVisible(visible) def init_device(self): dev_name = self.ui.cbDevice.currentText() self.sniffer.device_name = dev_name self._create_device_connects() self.scene_manager = SniffSceneManager(np.array([]), parent=self) def emit_editing_finished_signals(self): super().emit_editing_finished_signals() self.ui.spinbox_sniff_Noise.editingFinished.emit() self.ui.spinbox_sniff_Center.editingFinished.emit() self.ui.spinbox_sniff_BitLen.editingFinished.emit() self.ui.spinbox_sniff_ErrorTolerance.editingFinished.emit() def update_view(self): if super().update_view(): self.scene_manager.end = self.device.current_index self.scene_manager.init_scene() self.scene_manager.show_full_scene() self.graphics_view.update() @pyqtSlot() def on_device_started(self): self.scene_manager.data_array = self.device.data.real if hasattr( self.device.data, "real") else None super().on_device_started() self.ui.btnStart.setEnabled(False) self.set_device_ui_items_enabled(False) @pyqtSlot() def on_noise_edited(self): self.sniffer.signal._noise_threshold = self.ui.spinbox_sniff_Noise.value( ) self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot() def on_center_edited(self): self.sniffer.signal.qad_center = self.ui.spinbox_sniff_Center.value() self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot() def on_bit_len_edited(self): self.sniffer.signal.bit_len = self.ui.spinbox_sniff_BitLen.value() self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot() def on_tolerance_edited(self): self.sniffer.signal.tolerance = self.ui.spinbox_sniff_ErrorTolerance.value( ) self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot(int) def on_modulation_changed(self, new_index: int): self.sniffer.signal.silent_set_modulation_type(new_index) self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type)) @pyqtSlot() def on_start_clicked(self): super().on_start_clicked() self.sniffer.sniff() @pyqtSlot() def on_stop_clicked(self): self.sniffer.stop() @pyqtSlot() def on_clear_clicked(self): self.ui.btnClear.setEnabled(False) self.ui.txtEd_sniff_Preview.clear() self.scene_manager.clear_path() self.device.current_index = 0 self.sniffer.clear() @pyqtSlot(int) def on_data_sniffed(self, from_index: int): new_data = self.sniffer.decoded_to_string(self.view_type, start=from_index) if new_data.strip(): self.ui.txtEd_sniff_Preview.appendPlainText(new_data) self.ui.txtEd_sniff_Preview.verticalScrollBar().setValue( self.ui.txtEd_sniff_Preview.verticalScrollBar().maximum()) @pyqtSlot(int) def on_view_type_changed(self, new_index: int): self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(new_index)) @pyqtSlot() def on_btn_accept_clicked(self): self.protocol_accepted.emit(self.sniffer.messages) self.close() @pyqtSlot(str) def on_device_errors_changed(self, txt: str): self.ui.txtEditErrors.append(txt) @pyqtSlot() def on_line_edit_output_file_editing_finished(self): text = self.ui.lineEdit_sniff_OutputFile.text() if text and not text.endswith(".txt"): text += ".txt" self.ui.lineEdit_sniff_OutputFile.setText(text) self.sniffer.sniff_file = text self.ui.btnAccept.setDisabled(bool(self.sniffer.sniff_file)) @pyqtSlot(int) def on_combobox_sniff_encoding_index_changed(self, index: int): if self.sniffer.decoder != self.encodings[index]: self.sniffer.set_decoder_for_messages(self.encodings[index]) self.sniffer.decoder = self.encodings[index] self.ui.txtEd_sniff_Preview.setPlainText( self.sniffer.decoded_to_string(self.view_type))
class ProtocolSniffDialogController(QDialog): protocol_accepted = pyqtSignal(list) def __init__(self, freq, samp_rate, bw, gain, device, noise, center, bit_length, tolerance, modulation_type_index, parent=None): super().__init__(parent) self.ui = Ui_SniffProtocol() self.ui.setupUi(self) self.setAttribute(Qt.WA_DeleteOnClose) self.ui.spinBoxFreq.setValue(freq) self.ui.spinBoxSampleRate.setValue(samp_rate) self.ui.spinBoxBandwidth.setValue(bw) self.ui.spinBoxGain.setValue(gain) self.ui.spinboxNoise.setValue(noise) self.ui.spinboxCenter.setValue(center) self.ui.spinboxBitLen.setValue(bit_length) self.ui.spinboxErrorTolerance.setValue(tolerance) self.ui.comboxModulation.setCurrentIndex(modulation_type_index) self.ui.btnStop.setEnabled(False) self.ui.btnClear.setEnabled(False) self.ui.cbDevice.clear() items = [] bh = BackendHandler() for device_name in bh.DEVICE_NAMES: dev = bh.device_backends[device_name.lower()] if dev.is_enabled and dev.supports_rx: items.append(device_name) pm = PluginManager() if pm.is_plugin_enabled("NetworkSDRInterface"): items.append(NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) self.ui.cbDevice.addItems(items) del bh if device in items: self.ui.cbDevice.setCurrentIndex(items.index(device)) if self.ui.cbDevice.count() == 0: Errors.no_device() self.close() return device = self.ui.cbDevice.currentText() self.sniffer = ProtocolSniffer(bit_length, center, noise, tolerance, modulation_type_index, samp_rate, freq, gain, bw, device) self.set_object_visibility() self.sniffer.usrp_ip = self.ui.lineEditIP.text() self.bw_sr_are_locked = self.ui.btnLockBWSR.isChecked() ipRange = "(?:[0-1]?[0-9]?[0-9]|2[0-4][0-9]|25[0-5])" ipRegex = QRegExp("^" + ipRange + "\\." + ipRange + "\\." + ipRange + "\\." + ipRange + "$") self.ui.lineEditIP.setValidator(QRegExpValidator(ipRegex)) # Auto Complete like a Boss completer = QCompleter() completer.setModel(QDirModel(completer)) self.ui.lineEditOutputFile.setCompleter(completer) self.create_connects() def create_connects(self): self.ui.btnStart.clicked.connect(self.on_start_clicked) self.ui.btnStop.clicked.connect(self.on_stop_clicked) self.ui.btnClear.clicked.connect(self.on_clear_clicked) self.ui.btnAccept.clicked.connect(self.on_btn_accept_clicked) self.ui.btnClose.clicked.connect(self.close) self.sniffer.started.connect(self.on_sniffer_rcv_started) self.sniffer.stopped.connect(self.on_sniffer_rcv_stopped) self.sniffer.qt_signals.data_sniffed.connect(self.on_data_sniffed) self.sniffer.qt_signals.sniff_device_errors_changed.connect( self.on_device_errors_changed) self.ui.spinBoxSampleRate.editingFinished.connect( self.on_sample_rate_edited) self.ui.spinBoxGain.editingFinished.connect(self.on_gain_edited) self.ui.spinBoxFreq.editingFinished.connect(self.on_freq_edited) self.ui.spinBoxBandwidth.editingFinished.connect(self.on_bw_edited) self.ui.lineEditIP.editingFinished.connect(self.on_usrp_ip_edited) self.ui.cbDevice.currentIndexChanged.connect(self.on_device_edited) self.ui.btnLockBWSR.clicked.connect(self.on_btn_lock_bw_sr_clicked) self.ui.spinboxNoise.editingFinished.connect(self.on_noise_edited) self.ui.spinboxCenter.editingFinished.connect(self.on_center_edited) self.ui.spinboxBitLen.editingFinished.connect(self.on_bit_len_edited) self.ui.spinboxErrorTolerance.editingFinished.connect( self.on_tolerance_edited) self.ui.comboxModulation.currentIndexChanged.connect( self.on_modulation_changed) self.ui.comboBoxViewType.currentIndexChanged.connect( self.on_view_type_changed) self.ui.lineEditOutputFile.textChanged.connect( self.on_line_edit_output_file_text_changed) @property def view_type(self): return self.ui.comboBoxViewType.currentIndex() @property def has_empty_device_list(self): return self.ui.cbDevice.count() == 0 def closeEvent(self, event: QCloseEvent): if hasattr(self, "sniffer"): self.sniffer.stop() event.accept() @pyqtSlot() def on_sample_rate_edited(self): self.sniffer.rcv_device.sample_rate = self.ui.spinBoxSampleRate.value() if self.bw_sr_are_locked: self.ui.spinBoxBandwidth.setValue( self.ui.spinBoxSampleRate.value()) self.sniffer.rcv_device.bandwidth = self.ui.spinBoxBandwidth.value( ) @pyqtSlot() def on_freq_edited(self): self.sniffer.rcv_device.frequency = self.ui.spinBoxFreq.value() @pyqtSlot() def on_bw_edited(self): self.sniffer.rcv_device.bandwidth = self.ui.spinBoxBandwidth.value() if self.bw_sr_are_locked: self.ui.spinBoxSampleRate.setValue( self.ui.spinBoxBandwidth.value()) self.sniffer.rcv_device.sample_rate = self.ui.spinBoxSampleRate.value( ) @pyqtSlot() def on_usrp_ip_edited(self): self.sniffer.usrp_ip = self.ui.lineEditIP.text() @pyqtSlot() def on_gain_edited(self): self.sniffer.rcv_device.gain = self.ui.spinBoxGain.value() @pyqtSlot() def on_noise_edited(self): self.sniffer.signal._noise_threshold = self.ui.spinboxNoise.value() @pyqtSlot() def on_center_edited(self): self.sniffer.signal.qad_center = self.ui.spinboxCenter.value() @pyqtSlot() def on_bit_len_edited(self): self.sniffer.signal.bit_len = self.ui.spinboxBitLen.value() @pyqtSlot() def on_tolerance_edited(self): self.sniffer.signal.tolerance = self.ui.spinboxErrorTolerance.value() @pyqtSlot(int) def on_modulation_changed(self, new_index: int): self.sniffer.signal.silent_set_modulation_type(new_index) @pyqtSlot() def on_device_edited(self): dev = self.ui.cbDevice.currentText() self.sniffer.device_name = dev self.set_object_visibility() def set_object_visibility(self): self.ui.lineEditIP.setVisible(self.sniffer.device_name == "USRP") self.ui.labelIP.setVisible(self.sniffer.device_name == "USRP") for object in ("spinBoxFreq", "spinBoxSampleRate", "spinBoxBandwidth", "spinBoxGain", "spinboxNoise", "spinboxCenter", "spinboxBitLen", "spinboxErrorTolerance", "comboxModulation", "btnLockBWSR", "labelFreq", "labelSampleRate", "labelBandWidth", "labelGain", "labelNoise", "labelCenter", "labelBitLength", "labelTolerance", "labelModulation"): getattr( self.ui, object).setHidden(self.sniffer.device_name == NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) @pyqtSlot() def on_start_clicked(self): self.ui.spinBoxFreq.editingFinished.emit() self.ui.lineEditIP.editingFinished.emit() self.ui.spinBoxBandwidth.editingFinished.emit() self.ui.spinBoxSampleRate.editingFinished.emit() self.ui.spinboxNoise.editingFinished.emit() self.ui.spinboxCenter.editingFinished.emit() self.ui.spinboxBitLen.editingFinished.emit() self.ui.spinboxErrorTolerance.editingFinished.emit() self.sniffer.sniff() @pyqtSlot() def on_stop_clicked(self): self.sniffer.stop() @pyqtSlot() def on_sniffer_rcv_stopped(self): self.ui.btnStart.setEnabled(True) self.ui.btnStop.setEnabled(False) self.ui.btnClear.setEnabled(True) self.ui.spinBoxSampleRate.setEnabled(True) self.ui.spinBoxFreq.setEnabled(True) self.ui.lineEditIP.setEnabled(True) self.ui.spinBoxBandwidth.setEnabled(True) self.ui.spinBoxGain.setEnabled(True) self.ui.cbDevice.setEnabled(True) self.ui.spinboxNoise.setEnabled(True) self.ui.spinboxCenter.setEnabled(True) self.ui.spinboxBitLen.setEnabled(True) self.ui.spinboxErrorTolerance.setEnabled(True) self.ui.comboxModulation.setEnabled(True) @pyqtSlot() def on_sniffer_rcv_started(self): self.ui.txtEditErrors.clear() self.ui.btnStart.setEnabled(False) self.ui.btnClear.setEnabled(False) self.ui.btnStop.setEnabled(True) self.ui.spinBoxSampleRate.setDisabled(True) self.ui.spinBoxFreq.setDisabled(True) self.ui.spinBoxGain.setDisabled(True) self.ui.spinBoxBandwidth.setDisabled(True) self.ui.spinboxNoise.setDisabled(True) self.ui.spinboxCenter.setDisabled(True) self.ui.spinboxBitLen.setDisabled(True) self.ui.spinboxErrorTolerance.setDisabled(True) self.ui.comboxModulation.setDisabled(True) self.ui.lineEditIP.setDisabled(True) self.ui.cbDevice.setDisabled(True) @pyqtSlot() def on_btn_lock_bw_sr_clicked(self): self.bw_sr_are_locked = self.ui.btnLockBWSR.isChecked() if self.bw_sr_are_locked: self.ui.btnLockBWSR.setIcon(QIcon(":/icons/data/icons/lock.svg")) else: self.ui.btnLockBWSR.setIcon(QIcon(":/icons/data/icons/unlock.svg")) @pyqtSlot() def on_clear_clicked(self): self.ui.btnClear.setEnabled(False) self.ui.txtEdPreview.clear() self.sniffer.clear() @pyqtSlot(int) def on_data_sniffed(self, from_index: int): self.ui.txtEdPreview.appendPlainText( self.sniffer.plain_to_string(self.view_type, start=from_index, show_pauses=False)) self.ui.txtEdPreview.verticalScrollBar().setValue( self.ui.txtEdPreview.verticalScrollBar().maximum()) @pyqtSlot(int) def on_view_type_changed(self, new_index: int): self.ui.txtEdPreview.setPlainText( self.sniffer.plain_to_string(new_index, show_pauses=False)) @pyqtSlot() def on_btn_accept_clicked(self): self.protocol_accepted.emit(self.sniffer.messages) self.close() @pyqtSlot(str) def on_device_errors_changed(self, txt: str): self.ui.txtEditErrors.append(txt) @pyqtSlot(str) def on_line_edit_output_file_text_changed(self, text: str): self.sniffer.sniff_file = text self.ui.btnAccept.setDisabled(bool(self.sniffer.sniff_file))