class ProtocolSniffDialog(SendRecvDialog):
protocol_accepted = pyqtSignal(list)
def __init__(self,
project_manager,
signal=None,
signals=None,
parent=None,
testing_mode=False):
super().__init__(project_manager,
is_tx=False,
parent=parent,
testing_mode=testing_mode)
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()
signals = [] if signals is None else signals
self.sniff_settings_widget = SniffSettingsWidget(
project_manager=project_manager,
device_name=self.selected_device_name,
signal=signal,
signals=signals,
backend_handler=self.backend_handler)
self.ui.scrollAreaWidgetContents_2.layout().insertWidget(
1, self.sniff_settings_widget)
self.sniff_settings_widget.ui.btn_sniff_use_signal.setAutoDefault(
False)
self.sniffer = self.sniff_settings_widget.sniffer
self.setWindowTitle(self.tr("Sniff Protocol"))
self.setWindowIcon(QIcon.fromTheme(":/icons/icons/sniffer.svg"))
self.ui.txtEd_sniff_Preview.setFont(util.get_monospace_font())
# set really in on_device_started
self.scene_manager = None # type: LiveSceneManager
self.create_connects()
self.device_settings_widget.update_for_new_device(reset_gains=False)
@property
def view_type(self) -> int:
return self.sniff_settings_widget.ui.comboBox_sniff_viewtype.currentIndex(
)
@property
def show_timestamp(self) -> bool:
return self.sniff_settings_widget.ui.checkBox_sniff_Timestamp.isChecked(
)
def closeEvent(self, event: QCloseEvent):
self.sniff_settings_widget.emit_sniff_parameters_changed()
super().closeEvent(event)
def create_connects(self):
super().create_connects()
self.ui.btnAccept.clicked.connect(self.on_btn_accept_clicked)
self.sniff_settings_widget.sniff_parameters_changed.connect(
self.device_parameters_changed.emit)
self.sniff_settings_widget.sniff_setting_edited.connect(
self.on_sniff_setting_edited)
self.sniff_settings_widget.sniff_file_edited.connect(
self.on_sniff_file_edited)
self.sniffer.message_sniffed.connect(self.on_message_sniffed)
self.sniffer.qt_signals.sniff_device_errors_changed.connect(
self.on_device_errors_changed)
def init_device(self):
self.sniffer.device_name = self.selected_device_name
self.device = self.sniffer.rcv_device
self._create_device_connects()
self.scene_manager = SniffSceneManager(np.array([]), parent=self)
def emit_editing_finished_signals(self):
super().emit_editing_finished_signals()
self.sniff_settings_widget.emit_editing_finished_signals()
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_sniff_setting_edited(self):
self.ui.txtEd_sniff_Preview.setPlainText(
self.sniffer.decoded_to_string(
self.view_type, include_timestamps=self.show_timestamp))
@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_message_sniffed(self, index: int):
try:
msg = self.sniffer.messages[index]
except IndexError:
return
new_data = self.sniffer.message_to_string(
msg, self.view_type, include_timestamps=self.show_timestamp)
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()
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_sniff_file_edited(self):
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 ProtocolSniffDialog(SendRecvDialog):
protocol_accepted = pyqtSignal(list)
def __init__(self, project_manager, signal=None, signals=None, parent=None, testing_mode=False):
super().__init__(project_manager, is_tx=False, parent=parent, testing_mode=testing_mode)
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()
signals = [] if signals is None else signals
self.sniff_settings_widget = SniffSettingsWidget(project_manager=project_manager,
device_name=self.selected_device_name,
signal=signal, signals=signals,
backend_handler=self.backend_handler)
self.ui.scrollAreaWidgetContents_2.layout().insertWidget(1, self.sniff_settings_widget)
self.sniff_settings_widget.ui.btn_sniff_use_signal.setAutoDefault(False)
self.sniffer = self.sniff_settings_widget.sniffer
self.setWindowTitle(self.tr("Sniff Protocol"))
self.setWindowIcon(QIcon.fromTheme(":/icons/icons/sniffer.svg"))
self.ui.txtEd_sniff_Preview.setFont(util.get_monospace_font())
# set really in on_device_started
self.scene_manager = None # type: LiveSceneManager
self.create_connects()
self.device_settings_widget.update_for_new_device(overwrite_settings=False)
@property
def view_type(self) -> int:
return self.sniff_settings_widget.ui.comboBox_sniff_viewtype.currentIndex()
@property
def show_timestamp(self) -> bool:
return self.sniff_settings_widget.ui.checkBox_sniff_Timestamp.isChecked()
def closeEvent(self, event: QCloseEvent):
self.sniff_settings_widget.emit_sniff_parameters_changed()
super().closeEvent(event)
def create_connects(self):
super().create_connects()
self.ui.btnAccept.clicked.connect(self.on_btn_accept_clicked)
self.sniff_settings_widget.sniff_parameters_changed.connect(self.device_parameters_changed.emit)
self.sniff_settings_widget.sniff_setting_edited.connect(self.on_sniff_setting_edited)
self.sniff_settings_widget.sniff_file_edited.connect(self.on_sniff_file_edited)
self.sniffer.message_sniffed.connect(self.on_message_sniffed)
self.sniffer.qt_signals.sniff_device_errors_changed.connect(self.on_device_errors_changed)
def init_device(self):
self.sniffer.device_name = self.selected_device_name
self.device = self.sniffer.rcv_device
self._create_device_connects()
self.scene_manager = SniffSceneManager(np.array([]), parent=self)
def emit_editing_finished_signals(self):
super().emit_editing_finished_signals()
self.sniff_settings_widget.emit_editing_finished_signals()
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_sniff_setting_edited(self):
self.ui.txtEd_sniff_Preview.setPlainText(self.sniffer.decoded_to_string(self.view_type,
include_timestamps=self.show_timestamp))
@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.txtEd_sniff_Preview.clear()
self.scene_manager.clear_path()
self.device.current_index = 0
self.sniffer.clear()
@pyqtSlot(int)
def on_message_sniffed(self, index: int):
try:
msg = self.sniffer.messages[index]
except IndexError:
return
new_data = self.sniffer.message_to_string(msg, self.view_type, include_timestamps=self.show_timestamp)
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()
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_sniff_file_edited(self):
self.ui.btnAccept.setDisabled(bool(self.sniffer.sniff_file))