class ProtocolSniffer(ProtocolAnalyzer, QObject): """ This class is used for live sniffing a protocol with certain signal parameters. """ started = pyqtSignal() stopped = pyqtSignal() def __init__(self, bit_len: int, center: float, noise: float, tolerance: int, modulation_type: int, sample_rate: float, freq: float, gain: int, bandwidth: float, device: str, usrp_ip="192.168.10.2"): signal = Signal("", "LiveSignal") signal.bit_len = bit_len signal.qad_center = center signal.noise_threshold = noise signal.tolerance = tolerance signal.silent_set_modulation_type(modulation_type) ProtocolAnalyzer.__init__(self, signal) QObject.__init__(self, None) self.backend_handler = BackendHandler() self.rcv_device = VirtualDevice(self.backend_handler, device, Mode.receive, bandwidth, freq, gain, sample_rate, device_ip=usrp_ip, is_ringbuffer=True) self.rcv_device.index_changed.connect(self.on_rcv_thread_index_changed) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) self.rcv_timer = QTimer() self.rcv_timer.setInterval(1000) self.rcv_timer.timeout.connect(self.on_rcv_timer_timeout) self.rel_symbol_len = self._read_symbol_len() self.data_cache = [] self.conseq_non_data = 0 self.reading_data = False self.store_messages = True self.__sniff_file = "" self.__store_data = True def plain_to_string(self, view: int, show_pauses=True, start=0) -> str: """ :param start: First message to begin with :param show_pauses: Show pauses in output? :param view: 0 - Bits ## 1 - Hex ## 2 - ASCII """ return '\n'.join( msg.view_to_string(view, False, show_pauses) for msg in self.messages[start:]) @property def sniff_file(self): return self.__sniff_file @sniff_file.setter def sniff_file(self, val): self.__sniff_file = val if self.__sniff_file: self.__store_data = False @property def device_name(self): return self.rcv_device.name @device_name.setter def device_name(self, value: str): if value != self.rcv_device.name: self.rcv_device.free_data() self.rcv_device = VirtualDevice(self.backend_handler, value, self.rcv_device.mode, bw=1e6, freq=433.92e6, gain=20, samp_rate=1e6, device_ip="192.168.10.2", is_ringbuffer=True) self.rcv_device.index_changed.connect( self.on_rcv_thread_index_changed) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) @property def usrp_ip(self): return self.rcv_device.ip @usrp_ip.setter def usrp_ip(self, value: str): self.rcv_device.ip = value def sniff(self): self.rcv_device.start() self.rcv_timer.start() @pyqtSlot(int, int) def on_rcv_thread_index_changed(self, old_index, new_index): old_nmsgs = len(self.messages) if self.rcv_device.backend in (Backends.native, Backends.grc): self.__demodulate_data(self.rcv_device.data[old_index:new_index]) elif self.rcv_device.backend == Backends.network: # We receive the bits here for bit_str in self.rcv_device.data: self.messages.append(Message.from_plain_bits_str(bit_str, {})) self.rcv_device.free_data() # do not store received bits twice self.qt_signals.data_sniffed.emit(old_nmsgs) if self.sniff_file and not os.path.isdir(self.sniff_file): # Write Header if not os.path.isfile(self.sniff_file): with open(self.sniff_file, "w") as f: f.write("PROTOCOL:\n\n") with open(self.sniff_file, "a") as myfile: if self.plain_bits_str: myfile.write("\n") myfile.write("\n".join(self.plain_bits_str)) if not self.__store_data: self.messages[:] = [] def __demodulate_data(self, data): """ Demodulates received IQ data and adds demodulated bits to messages :param data: :return: """ signal = self.signal if self.__are_bits_in_data(data): self.reading_data = True elif self.conseq_non_data == 5: self.reading_data = False self.conseq_non_data = 0 else: self.conseq_non_data += 1 if self.reading_data: self.data_cache.append(data) return elif len(self.data_cache) == 0: return signal._fulldata = np.concatenate(self.data_cache) del self.data_cache[:] signal._qad = None bit_len = signal.bit_len ppseq = grab_pulse_lens(signal.qad, signal.qad_center, signal.tolerance, signal.modulation_type) bit_data, pauses, bit_sample_pos = self._ppseq_to_bits( ppseq, bit_len, self.rel_symbol_len) i = 0 first_msg = True for bits, pause in zip(bit_data, pauses): if first_msg or self.messages[-1].pause > 8 * bit_len: # Create new Message middle_bit_pos = bit_sample_pos[i][int(len(bits) / 2)] start, end = middle_bit_pos, middle_bit_pos + bit_len rssi = np.mean(np.abs(signal._fulldata[start:end])) message = Message(bits, pause, bit_len=bit_len, rssi=rssi, message_type=self.default_message_type) self.messages.append(message) first_msg = False else: # Append to last message message = self.messages[-1] nzeros = int(np.round(message.pause / bit_len)) message.plain_bits.extend([False] * nzeros) message.plain_bits.extend(bits) message.pause = pause i += 1 def __are_bits_in_data(self, data): signal = self.signal signal._fulldata = data signal._qad = None bit_len = signal.bit_len ppseq = grab_pulse_lens(signal.qad, signal.qad_center, signal.tolerance, signal.modulation_type) bit_data, pauses, _ = self._ppseq_to_bits(ppseq, bit_len, self.rel_symbol_len) return bool(bit_data) def stop(self): self.rcv_timer.stop() self.rcv_device.stop("Stopping receiving due to user interaction") QApplication.processEvents() time.sleep(0.1) def clear(self): del self.data_cache[:] del self.messages[:] def on_rcv_timer_timeout(self): new_errors = self.rcv_device.read_errors() self.qt_signals.sniff_device_errors_changed.emit(new_errors) if "No devices found for" in new_errors: self.rcv_device.stop("Could not establish connection to USRP") Errors.usrp_ip_not_found() self.stop() elif "FATAL: No supported devices found" in new_errors or \ "HACKRF_ERROR_NOT_FOUND" in new_errors: self.rcv_device.stop("Could not establish connection to HackRF") Errors.hackrf_not_found() self.stop() elif "No module named gnuradio" in new_errors: self.rcv_device.stop("Did not find gnuradio.") Errors.gnuradio_not_installed() self.stop() elif "Address already in use" in new_errors: self.rcv_device.port += 1 self.stop() self.sniff() def __emit_started(self): self.started.emit() def __emit_stopped(self): self.stopped.emit()
class ProtocolSniffer(ProtocolAnalyzer, QObject): """ This class is used for live sniffing a protocol with certain signal parameters. """ started = pyqtSignal() stopped = pyqtSignal() def __init__(self, bit_len: int, center: float, noise: float, tolerance: int, modulation_type: int, device: str, backend_handler: BackendHandler): signal = Signal("", "LiveSignal") signal.bit_len = bit_len signal.qad_center = center signal.noise_threshold = noise signal.tolerance = tolerance signal.silent_set_modulation_type(modulation_type) ProtocolAnalyzer.__init__(self, signal) QObject.__init__(self, None) self.backend_handler = backend_handler self.rcv_device = VirtualDevice(self.backend_handler, device, Mode.receive, is_ringbuffer=False, raw_mode=False) self.rcv_device.index_changed.connect(self.on_rcv_thread_index_changed) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) self.data_cache = [] self.conseq_non_data = 0 self.reading_data = False self.store_messages = True self.__sniff_file = "" self.__store_data = True def decoded_to_string(self, view: int, start=0): return '\n'.join( msg.view_to_string(view, decoded=True, show_pauses=False) for msg in self.messages[start:]) @property def sniff_file(self): return self.__sniff_file @sniff_file.setter def sniff_file(self, val): self.__sniff_file = val if self.__sniff_file: self.__store_data = False @property def device_name(self): return self.rcv_device.name @device_name.setter def device_name(self, value: str): if value != self.rcv_device.name: self.rcv_device.free_data() self.rcv_device = VirtualDevice(self.backend_handler, value, Mode.receive, device_ip="192.168.10.2", is_ringbuffer=False, raw_mode=False) self.rcv_device.index_changed.connect( self.on_rcv_thread_index_changed) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) def sniff(self): self.rcv_device.start() @pyqtSlot(int, int) def on_rcv_thread_index_changed(self, old_index, new_index): old_nmsgs = len(self.messages) if self.rcv_device.backend in (Backends.native, Backends.grc): if old_index == new_index: return self.__demodulate_data(self.rcv_device.data[old_index:new_index]) elif self.rcv_device.backend == Backends.network: # We receive the bits here for bit_str in self.rcv_device.data: msg = Message.from_plain_bits_str(bit_str) msg.decoder = self.decoder self.messages.append(msg) self.rcv_device.free_data() # do not store received bits twice self.qt_signals.data_sniffed.emit(old_nmsgs) if self.sniff_file and not os.path.isdir(self.sniff_file): with open(self.sniff_file, "a") as myfile: myfile.write("\n".join(self.plain_bits_str)) if not self.__store_data: self.messages.clear() def __demodulate_data(self, data): """ Demodulates received IQ data and adds demodulated bits to messages :param data: :return: """ if self.__are_bits_in_data(data): self.reading_data = True elif self.conseq_non_data == 5: self.reading_data = False self.conseq_non_data = 0 else: self.conseq_non_data += 1 if self.reading_data: self.data_cache.append(data) return elif len(self.data_cache) == 0: return self.signal._fulldata = np.concatenate(self.data_cache) self.data_cache.clear() self.signal._qad = None bit_len = self.signal.bit_len ppseq = grab_pulse_lens(self.signal.qad, self.signal.qad_center, self.signal.tolerance, self.signal.modulation_type) bit_data, pauses, bit_sample_pos = self._ppseq_to_bits(ppseq, bit_len) i = 0 first_msg = True for bits, pause in zip(bit_data, pauses): if first_msg or self.messages[-1].pause > 8 * bit_len: # Create new Message middle_bit_pos = bit_sample_pos[i][int(len(bits) / 2)] start, end = middle_bit_pos, middle_bit_pos + bit_len rssi = np.mean(np.abs(self.signal._fulldata[start:end])) message = Message(bits, pause, bit_len=bit_len, rssi=rssi, message_type=self.default_message_type, decoder=self.decoder) self.messages.append(message) first_msg = False else: # Append to last message message = self.messages[-1] nzeros = int(np.round(message.pause / bit_len)) message.plain_bits.extend([False] * nzeros) message.plain_bits.extend(bits) message.pause = pause i += 1 def __are_bits_in_data(self, data): self.signal._fulldata = data self.signal._qad = None bit_len = self.signal.bit_len ppseq = grab_pulse_lens(self.signal.qad, self.signal.qad_center, self.signal.tolerance, self.signal.modulation_type) bit_data, pauses, _ = self._ppseq_to_bits(ppseq, bit_len) return bool(bit_data) def stop(self): self.rcv_device.stop("Stopping receiving due to user interaction") def clear(self): self.data_cache.clear() self.messages.clear() def __emit_started(self): self.started.emit() def __emit_stopped(self): self.stopped.emit()
class ProtocolSniffer(ProtocolAnalyzer, QObject): """ This class is used for live sniffing a protocol with certain signal parameters. """ started = pyqtSignal() stopped = pyqtSignal() message_sniffed = pyqtSignal() def __init__(self, bit_len: int, center: float, noise: float, tolerance: int, modulation_type: int, device: str, backend_handler: BackendHandler, network_raw_mode=False): signal = Signal("", "LiveSignal") signal.bit_len = bit_len signal.qad_center = center signal.noise_threshold = noise signal.tolerance = tolerance signal.silent_set_modulation_type(modulation_type) ProtocolAnalyzer.__init__(self, signal) QObject.__init__(self, None) self.network_raw_mode = network_raw_mode self.backend_handler = backend_handler self.rcv_device = VirtualDevice(self.backend_handler, device, Mode.receive, resume_on_full_receive_buffer=True, raw_mode=network_raw_mode) self.rcv_device.emit_data_received_signal = True self.rcv_device.data_received.connect(self.on_data_received) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) self.real_time = real_time self.data_cache = [] self.reading_data = False self.adaptive_noise = False self.pause_length = 0 self.store_messages = True self.__sniff_file = "" self.__store_data = True def decoded_to_string(self, view: int, start=0, include_timestamps=True): result = [] for msg in self.messages[start:]: result.append(self.message_to_string(msg, view, include_timestamps)) return "\n".join(result) def message_to_string(self, message: Message, view: int, include_timestamps=True): msg_str_data = [] if include_timestamps: msg_date = datetime.fromtimestamp(message.timestamp) msg_str_data.append(msg_date.strftime("[%Y-%m-%d %H:%M:%S.%f]")) msg_str_data.append( message.view_to_string(view, decoded=True, show_pauses=False)) return " ".join(msg_str_data) @property def sniff_file(self): return self.__sniff_file @sniff_file.setter def sniff_file(self, val): self.__sniff_file = val if self.__sniff_file: self.__store_data = False @property def device_name(self): return self.rcv_device.name @device_name.setter def device_name(self, value: str): if value != self.rcv_device.name: self.rcv_device.free_data() self.rcv_device = VirtualDevice(self.backend_handler, value, Mode.receive, device_ip="192.168.10.2", resume_on_full_receive_buffer=True, raw_mode=self.network_raw_mode) self.rcv_device.emit_data_received_signal = True self.rcv_device.data_received.connect(self.on_data_received) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) def sniff(self): self.rcv_device.start() @pyqtSlot(np.ndarray) def on_data_received(self, data: np.ndarray): if self.rcv_device.is_raw_mode: self.__demodulate_data(data) elif self.rcv_device.backend == Backends.network: # We receive the bits here for bit_str in self.rcv_device.data: msg = Message.from_plain_bits_str(bit_str) msg.decoder = self.decoder self.messages.append(msg) self.message_sniffed.emit() self.rcv_device.free_data() # do not store received bits twice if self.sniff_file and not os.path.isdir(self.sniff_file): plain_bits_str = self.plain_bits_str if plain_bits_str: with open(self.sniff_file, "a") as f: f.write("\n".join(plain_bits_str) + "\n") if not self.__store_data: self.messages.clear() def __demodulate_data(self, data): """ Demodulates received IQ data and adds demodulated bits to messages :param data: :return: """ rssi_squared = np.mean(data.real**2 + data.imag**2) is_above_noise = rssi_squared > self.signal.noise_threshold**2 if self.adaptive_noise and not is_above_noise: self.signal.noise_threshold = 0.9 * self.signal.noise_threshold + 0.1 * np.max( np.abs(data)) if is_above_noise: self.data_cache.append(data) self.pause_length = 0 return else: self.pause_length += len(data) if self.pause_length < 10 * self.signal.bit_len: self.data_cache.append(data) return if len(self.data_cache) == 0: return # clear cache and start a new message self.signal._fulldata = np.concatenate(self.data_cache) self.data_cache.clear() self.signal._qad = None bit_len = self.signal.bit_len ppseq = grab_pulse_lens(self.signal.qad, self.signal.qad_center, self.signal.tolerance, self.signal.modulation_type, self.signal.bit_len) bit_data, pauses, bit_sample_pos = self._ppseq_to_bits( ppseq, bit_len, write_bit_sample_pos=False) for bits, pause in zip(bit_data, pauses): message = Message(bits, pause, bit_len=bit_len, message_type=self.default_message_type, decoder=self.decoder) self.messages.append(message) self.message_sniffed.emit() def stop(self): self.rcv_device.stop("Stopping receiving due to user interaction") def clear(self): self.data_cache.clear() self.messages.clear() def __emit_started(self): self.started.emit() def __emit_stopped(self): if hasattr(self, "stopped"): self.stopped.emit()
class ProtocolSniffer(ProtocolAnalyzer, QObject): """ This class is used for live sniffing a protocol with certain signal parameters. """ started = pyqtSignal() stopped = pyqtSignal() message_sniffed = pyqtSignal(int) BUFFER_SIZE_MB = 100 def __init__(self, samples_per_symbol: int, center: float, center_spacing: float, noise: float, tolerance: int, modulation_type: str, bits_per_symbol: int, device: str, backend_handler: BackendHandler, network_raw_mode=False): signal = Signal("", "LiveSignal") signal.samples_per_symbol = samples_per_symbol signal.center = center signal.center_spacing = center_spacing signal.noise_threshold = noise signal.tolerance = tolerance signal.silent_set_modulation_type(modulation_type) signal.bits_per_symbol = bits_per_symbol ProtocolAnalyzer.__init__(self, signal) QObject.__init__(self, None) self.network_raw_mode = network_raw_mode self.backend_handler = backend_handler self.rcv_device = VirtualDevice(self.backend_handler, device, Mode.receive, resume_on_full_receive_buffer=True, raw_mode=network_raw_mode) signal.iq_array = IQArray(None, self.rcv_device.data_type, 0) self.sniff_thread = Thread(target=self.check_for_data, daemon=True) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) self.__buffer = IQArray(None, np.float32, 0) self.__init_buffer() self.__current_buffer_index = 0 self.reading_data = False self.adaptive_noise = False self.automatic_center = False self.pause_length = 0 self.is_running = False self.store_messages = True self.__sniff_file = "" self.__store_data = True def __add_to_buffer(self, data: np.ndarray): n = len(data) if n + self.__current_buffer_index > len(self.__buffer): n = len(self.__buffer) - self.__current_buffer_index - 1 logger.warning("Buffer of protocol sniffer is full") self.__buffer[self.__current_buffer_index:self.__current_buffer_index + n] = data[:n] self.__current_buffer_index += n def __clear_buffer(self): self.__current_buffer_index = 0 def __buffer_is_full(self): return self.__current_buffer_index >= len(self.__buffer) - 2 def __init_buffer(self): self.__buffer = IQArray(None, self.rcv_device.data_type, int(self.BUFFER_SIZE_MB * 1000 * 1000 / 8)) self.__current_buffer_index = 0 def decoded_to_string(self, view: int, start=0, include_timestamps=True): result = [] for msg in self.messages[start:]: result.append(self.message_to_string(msg, view, include_timestamps)) return "\n".join(result) def message_to_string(self, message: Message, view: int, include_timestamps=True): msg_str_data = [] if include_timestamps: msg_date = datetime.fromtimestamp(message.timestamp) msg_str_data.append(msg_date.strftime("[%Y-%m-%d %H:%M:%S.%f]")) msg_str_data.append( message.view_to_string(view, decoded=True, show_pauses=False)) return " ".join(msg_str_data) @property def sniff_file(self): return self.__sniff_file @sniff_file.setter def sniff_file(self, val): self.__sniff_file = val if self.__sniff_file: self.__store_data = False @property def device_name(self): return self.rcv_device.name @device_name.setter def device_name(self, value: str): if value != self.rcv_device.name: self.rcv_device.free_data() self.rcv_device = VirtualDevice(self.backend_handler, value, Mode.receive, device_ip="192.168.10.2", resume_on_full_receive_buffer=True, raw_mode=self.network_raw_mode) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) self.signal.iq_array = IQArray(None, self.rcv_device.data_type, 0) self.__init_buffer() def sniff(self): self.is_running = True self.rcv_device.start() self.sniff_thread = Thread(target=self.check_for_data, daemon=True) self.sniff_thread.start() def check_for_data(self): old_index = 0 while self.is_running: time.sleep(0.01) if self.rcv_device.is_raw_mode: if old_index <= self.rcv_device.current_index: data = self.rcv_device.data[old_index:self.rcv_device. current_index] else: data = np.concatenate( (self.rcv_device.data[old_index:], self.rcv_device.data[:self.rcv_device.current_index])) old_index = self.rcv_device.current_index self.__demodulate_data(data) elif self.rcv_device.backend == Backends.network: # We receive the bits here for bit_str in self.rcv_device.data: msg = Message.from_plain_bits_str(bit_str) msg.decoder = self.decoder self.messages.append(msg) self.message_sniffed.emit(len(self.messages) - 1) self.rcv_device.free_data() # do not store received bits twice if self.sniff_file and not os.path.isdir(self.sniff_file): plain_bits_str = self.plain_bits_str if plain_bits_str: with open(self.sniff_file, "a") as f: f.write("\n".join(plain_bits_str) + "\n") if not self.__store_data: self.messages.clear() def __demodulate_data(self, data): """ Demodulates received IQ data and adds demodulated bits to messages :param data: :return: """ if len(data) == 0: return power_spectrum = data.real**2.0 + data.imag**2.0 is_above_noise = np.sqrt( np.mean(power_spectrum)) > self.signal.noise_threshold if self.adaptive_noise and not is_above_noise: self.signal.noise_threshold = 0.9 * self.signal.noise_threshold + 0.1 * np.sqrt( np.max(power_spectrum)) if is_above_noise: self.__add_to_buffer(data) self.pause_length = 0 if not self.__buffer_is_full(): return else: self.pause_length += len(data) if self.pause_length < 10 * self.signal.samples_per_symbol: self.__add_to_buffer(data) if not self.__buffer_is_full(): return if self.__current_buffer_index == 0: return # clear cache and start a new message self.signal.iq_array = IQArray( self.__buffer[0:self.__current_buffer_index]) self.__clear_buffer() self.signal._qad = None samples_per_symbol = self.signal.samples_per_symbol if self.automatic_center: self.signal.center = AutoInterpretation.detect_center( self.signal.qad, max_size=150 * samples_per_symbol) ppseq = grab_pulse_lens(self.signal.qad, self.signal.center, self.signal.tolerance, self.signal.modulation_type, self.signal.samples_per_symbol, self.signal.bits_per_symbol, self.signal.center_spacing) bit_data, pauses, bit_sample_pos = self._ppseq_to_bits( ppseq, samples_per_symbol, self.signal.bits_per_symbol, write_bit_sample_pos=False) for bits, pause in zip(bit_data, pauses): message = Message(bits, pause, samples_per_symbol=samples_per_symbol, message_type=self.default_message_type, decoder=self.decoder) self.messages.append(message) self.message_sniffed.emit(len(self.messages) - 1) def stop(self): self.is_running = False self.rcv_device.stop("Stopping receiving due to user interaction") if self.sniff_thread.is_alive(): self.sniff_thread.join(0.1) if self.sniff_thread.is_alive(): logger.error("Sniff thread is still alive") def clear(self): self.__clear_buffer() self.messages.clear() def __emit_started(self): self.started.emit() def __emit_stopped(self): if hasattr(self, "stopped"): self.stopped.emit()
class ProtocolSniffer(ProtocolAnalyzer, QObject): """ This class is used for live sniffing a protocol with certain signal parameters. """ started = pyqtSignal() stopped = pyqtSignal() def __init__(self, bit_len: int, center: float, noise: float, tolerance: int, modulation_type: int, device: str, backend_handler: BackendHandler): signal = Signal("", "LiveSignal") signal.bit_len = bit_len signal.qad_center = center signal.noise_threshold = noise signal.tolerance = tolerance signal.silent_set_modulation_type(modulation_type) ProtocolAnalyzer.__init__(self, signal) QObject.__init__(self, None) self.backend_handler = backend_handler self.rcv_device = VirtualDevice(self.backend_handler, device, Mode.receive, resume_on_full_receive_buffer=True, raw_mode=False) self.rcv_device.index_changed.connect(self.on_rcv_thread_index_changed) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) self.data_cache = [] self.conseq_non_data = 0 self.reading_data = False self.store_messages = True self.__sniff_file = "" self.__store_data = True def decoded_to_string(self, view: int, start=0): return '\n'.join(msg.view_to_string(view, decoded=True, show_pauses=False) for msg in self.messages[start:]) @property def sniff_file(self): return self.__sniff_file @sniff_file.setter def sniff_file(self, val): self.__sniff_file = val if self.__sniff_file: self.__store_data = False @property def device_name(self): return self.rcv_device.name @device_name.setter def device_name(self, value: str): if value != self.rcv_device.name: self.rcv_device.free_data() self.rcv_device = VirtualDevice(self.backend_handler, value, Mode.receive, device_ip="192.168.10.2", resume_on_full_receive_buffer=True, raw_mode=False) self.rcv_device.index_changed.connect(self.on_rcv_thread_index_changed) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) def sniff(self): self.rcv_device.start() @pyqtSlot(int, int) def on_rcv_thread_index_changed(self, old_index, new_index): old_nmsgs = len(self.messages) if self.rcv_device.backend in (Backends.native, Backends.grc): if old_index == new_index: return self.__demodulate_data(self.rcv_device.data[old_index:new_index]) elif self.rcv_device.backend == Backends.network: # We receive the bits here for bit_str in self.rcv_device.data: msg = Message.from_plain_bits_str(bit_str) msg.decoder = self.decoder self.messages.append(msg) self.rcv_device.free_data() # do not store received bits twice self.qt_signals.data_sniffed.emit(old_nmsgs) if self.sniff_file and not os.path.isdir(self.sniff_file): with open(self.sniff_file, "a") as myfile: myfile.write("\n".join(self.plain_bits_str)) if not self.__store_data: self.messages.clear() def __demodulate_data(self, data): """ Demodulates received IQ data and adds demodulated bits to messages :param data: :return: """ if self.__are_bits_in_data(data): self.reading_data = True elif self.conseq_non_data == 5: self.reading_data = False self.conseq_non_data = 0 else: self.conseq_non_data += 1 if self.reading_data: self.data_cache.append(data) return elif len(self.data_cache) == 0: return self.signal._fulldata = np.concatenate(self.data_cache) self.data_cache.clear() self.signal._qad = None bit_len = self.signal.bit_len ppseq = grab_pulse_lens(self.signal.qad, self.signal.qad_center, self.signal.tolerance, self.signal.modulation_type) bit_data, pauses, bit_sample_pos = self._ppseq_to_bits(ppseq, bit_len) i = 0 first_msg = True for bits, pause in zip(bit_data, pauses): if first_msg or self.messages[-1].pause > 8 * bit_len: # Create new Message middle_bit_pos = bit_sample_pos[i][int(len(bits) / 2)] start, end = middle_bit_pos, middle_bit_pos + bit_len rssi = np.mean(np.abs(self.signal._fulldata[start:end])) message = Message(bits, pause, bit_len=bit_len, rssi=rssi, message_type=self.default_message_type, decoder=self.decoder) self.messages.append(message) first_msg = False else: # Append to last message message = self.messages[-1] nzeros = int(np.round(message.pause / bit_len)) message.plain_bits.extend([False] * nzeros) message.plain_bits.extend(bits) message.pause = pause i += 1 def __are_bits_in_data(self, data): self.signal._fulldata = data self.signal._qad = None bit_len = self.signal.bit_len ppseq = grab_pulse_lens(self.signal.qad, self.signal.qad_center, self.signal.tolerance, self.signal.modulation_type) bit_data, pauses, _ = self._ppseq_to_bits(ppseq, bit_len) return bool(bit_data) def stop(self): self.rcv_device.stop("Stopping receiving due to user interaction") def clear(self): self.data_cache.clear() self.messages.clear() def __emit_started(self): self.started.emit() def __emit_stopped(self): self.stopped.emit()
class ProtocolSniffer(ProtocolAnalyzer, QObject): """ This class is used for live sniffing a protocol with certain signal parameters. """ started = pyqtSignal() stopped = pyqtSignal() message_sniffed = pyqtSignal(int) BUFFER_SIZE_MB = 100 def __init__(self, bit_len: int, center: float, noise: float, tolerance: int, modulation_type: int, device: str, backend_handler: BackendHandler, network_raw_mode=False): signal = Signal("", "LiveSignal") signal.bit_len = bit_len signal.qad_center = center signal.noise_threshold = noise signal.tolerance = tolerance signal.silent_set_modulation_type(modulation_type) ProtocolAnalyzer.__init__(self, signal) QObject.__init__(self, None) self.network_raw_mode = network_raw_mode self.backend_handler = backend_handler self.rcv_device = VirtualDevice(self.backend_handler, device, Mode.receive, resume_on_full_receive_buffer=True, raw_mode=network_raw_mode) self.sniff_thread = Thread(target=self.check_for_data, daemon=True) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) self.__buffer = np.zeros(int(self.BUFFER_SIZE_MB * 1000 * 1000 / 8), dtype=np.complex64) self.__current_buffer_index = 0 self.reading_data = False self.adaptive_noise = False self.automatic_center = False self.pause_length = 0 self.is_running = False self.store_messages = True self.__sniff_file = "" self.__store_data = True def __add_to_buffer(self, data: np.ndarray): n = len(data) if n + self.__current_buffer_index > len(self.__buffer): n = len(self.__buffer) - self.__current_buffer_index - 1 logger.warning("Buffer of protocol sniffer is full") self.__buffer[self.__current_buffer_index:self.__current_buffer_index + n] = data[:n] self.__current_buffer_index += n def __clear_buffer(self): self.__current_buffer_index = 0 def __buffer_is_full(self): return self.__current_buffer_index >= len(self.__buffer) - 2 def decoded_to_string(self, view: int, start=0, include_timestamps=True): result = [] for msg in self.messages[start:]: result.append(self.message_to_string(msg, view, include_timestamps)) return "\n".join(result) def message_to_string(self, message: Message, view: int, include_timestamps=True): msg_str_data = [] if include_timestamps: msg_date = datetime.fromtimestamp(message.timestamp) msg_str_data.append(msg_date.strftime("[%Y-%m-%d %H:%M:%S.%f]")) msg_str_data.append(message.view_to_string(view, decoded=True, show_pauses=False)) return " ".join(msg_str_data) @property def sniff_file(self): return self.__sniff_file @sniff_file.setter def sniff_file(self, val): self.__sniff_file = val if self.__sniff_file: self.__store_data = False @property def device_name(self): return self.rcv_device.name @device_name.setter def device_name(self, value: str): if value != self.rcv_device.name: self.rcv_device.free_data() self.rcv_device = VirtualDevice(self.backend_handler, value, Mode.receive, device_ip="192.168.10.2", resume_on_full_receive_buffer=True, raw_mode=self.network_raw_mode) self.rcv_device.started.connect(self.__emit_started) self.rcv_device.stopped.connect(self.__emit_stopped) def sniff(self): self.is_running = True self.rcv_device.start() self.sniff_thread = Thread(target=self.check_for_data, daemon=True) self.sniff_thread.start() def check_for_data(self): old_index = 0 while self.is_running: time.sleep(0.01) if self.rcv_device.is_raw_mode: if old_index <= self.rcv_device.current_index: data = self.rcv_device.data[old_index:self.rcv_device.current_index] else: data = np.concatenate((self.rcv_device.data[old_index:], self.rcv_device.data[:self.rcv_device.current_index])) old_index = self.rcv_device.current_index self.__demodulate_data(data) elif self.rcv_device.backend == Backends.network: # We receive the bits here for bit_str in self.rcv_device.data: msg = Message.from_plain_bits_str(bit_str) msg.decoder = self.decoder self.messages.append(msg) self.message_sniffed.emit(len(self.messages) - 1) self.rcv_device.free_data() # do not store received bits twice if self.sniff_file and not os.path.isdir(self.sniff_file): plain_bits_str = self.plain_bits_str if plain_bits_str: with open(self.sniff_file, "a") as f: f.write("\n".join(plain_bits_str) + "\n") if not self.__store_data: self.messages.clear() def __demodulate_data(self, data): """ Demodulates received IQ data and adds demodulated bits to messages :param data: :return: """ if len(data) == 0: return power_spectrum = data.real ** 2 + data.imag ** 2 is_above_noise = np.sqrt(np.mean(power_spectrum)) > self.signal.noise_threshold if self.adaptive_noise and not is_above_noise: self.signal.noise_threshold = 0.9 * self.signal.noise_threshold + 0.1 * np.sqrt(np.max(power_spectrum)) if is_above_noise: self.__add_to_buffer(data) self.pause_length = 0 if not self.__buffer_is_full(): return else: self.pause_length += len(data) if self.pause_length < 10 * self.signal.bit_len: self.__add_to_buffer(data) if not self.__buffer_is_full(): return if self.__current_buffer_index == 0: return # clear cache and start a new message self.signal._fulldata = self.__buffer[0:self.__current_buffer_index] self.__clear_buffer() self.signal._qad = None bit_len = self.signal.bit_len if self.automatic_center: self.signal.qad_center = AutoInterpretation.detect_center(self.signal.qad, max_size=150*self.signal.bit_len) ppseq = grab_pulse_lens(self.signal.qad, self.signal.qad_center, self.signal.tolerance, self.signal.modulation_type, self.signal.bit_len) bit_data, pauses, bit_sample_pos = self._ppseq_to_bits(ppseq, bit_len, write_bit_sample_pos=False) for bits, pause in zip(bit_data, pauses): message = Message(bits, pause, bit_len=bit_len, message_type=self.default_message_type, decoder=self.decoder) self.messages.append(message) self.message_sniffed.emit(len(self.messages) - 1) def stop(self): self.is_running = False self.rcv_device.stop("Stopping receiving due to user interaction") if self.sniff_thread.is_alive(): self.sniff_thread.join(0.1) if self.sniff_thread.is_alive(): logger.error("Sniff thread is still alive") def clear(self): self.__clear_buffer() self.messages.clear() def __emit_started(self): self.started.emit() def __emit_stopped(self): if hasattr(self, "stopped"): self.stopped.emit()
class SendRecvDialogController(QDialog): files_recorded = pyqtSignal(list) recording_parameters = pyqtSignal(str, str, str, str, str) def __init__(self, freq, samp_rate, bw, gain, device: str, mode: Mode, modulated_data=None, parent=None): super().__init__(parent) self.ui = Ui_SendRecvDialog() self.ui.setupUi(self) self.setAttribute(Qt.WA_DeleteOnClose) self.graphics_view = self.ui.graphicsViewSend if mode == Mode.send else self.ui.graphicsViewReceive self.backend_handler = BackendHandler() if mode == Mode.spectrum: self.update_interval = 1 else: self.update_interval = 50 if mode == Mode.send and modulated_data is None: raise ValueError("I need modulated data to send!") if mode == Mode.receive or mode == Mode.spectrum: self.ui.spinBoxNRepeat.hide() self.ui.labelNRepeat.hide() self.ui.lblCurrentRepeatValue.hide() self.ui.lblRepeatText.hide() self.ui.lSamplesSentText.hide() self.ui.progressBar.hide() self.ui.stackedWidget.setCurrentIndex(0) else: self.ui.stackedWidget.setCurrentIndex(1) if mode == Mode.send or mode == Mode.spectrum: self.ui.lSamplesCaptured.hide() self.ui.lSamplesCapturedText.hide() self.ui.lSignalSize.hide() self.ui.lSignalSizeText.hide() self.ui.lTime.hide() self.ui.lTimeText.hide() self.ui.btnSave.hide() if mode == Mode.spectrum: self.setWindowTitle("Spectrum analyzer") if mode == Mode.send: self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.setWindowTitle("Send signal") self.ui.btnStart.setToolTip("Send data") self.ui.btnStop.setToolTip("Stop sending") self.ui.progressBar.setMaximum(len(modulated_data)) self.device_is_sending = False self.ui.btnStop.setEnabled(False) self.ui.btnClear.setEnabled(False) self.ui.btnSave.setEnabled(False) self.start = 0 self.already_saved = True self.bw_sr_are_locked = constants.SETTINGS.value("lock_bandwidth_sample_rate", True, bool) self.ui.spinBoxFreq.setValue(freq) self.ui.spinBoxSampleRate.setValue(samp_rate) self.ui.spinBoxBandwidth.setValue(bw) self.ui.spinBoxGain.setValue(gain) self.ui.spinBoxNRepeat.setValue(constants.SETTINGS.value('num_sending_repeats', 1, type=int)) self.ui.cbDevice.clear() items = [] for device_name in self.backend_handler.DEVICE_NAMES: dev = self.backend_handler.device_backends[device_name.lower()] if mode == Mode.send and dev.is_enabled and dev.supports_tx: items.append(device_name) elif mode in (Mode.receive, Mode.spectrum) and dev.is_enabled and dev.supports_rx: items.append(device_name) if mode == Mode.send and PluginManager().is_plugin_enabled("NetworkSDRInterface"): items.append(NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) self.ui.cbDevice.addItems(items) if device in items: self.ui.cbDevice.setCurrentIndex(items.index(device)) dev_name = self.ui.cbDevice.currentText() nrep = self.ui.spinBoxNRepeat.value() self.device = VirtualDevice(self.backend_handler, dev_name, mode, bw, freq, gain, samp_rate, samples_to_send=modulated_data, device_ip=self.ui.lineEditIP.text(), sending_repeats=nrep, parent=self) self.ui.lineEditIP.setVisible(dev_name == "USRP") self.ui.labelIP.setVisible(dev_name == "USRP") self.recorded_files = [] self.timer = QTimer(self) if mode == Mode.receive: self.scene_manager = LiveSceneManager(np.array([]), parent=self) # set really in on_device_started elif mode == Mode.send: signal = Signal.from_samples(modulated_data, "Modulated Preview", samp_rate) self.scene_manager = SignalSceneManager(signal, parent=self) self.send_indicator = self.scene_manager.scene.addRect(0, -2, 0, 4, QPen(QColor(Qt.transparent), Qt.FlatCap), QBrush(constants.SEND_INDICATOR_COLOR)) self.send_indicator.stackBefore(self.scene_manager.scene.selection_area) self.scene_manager.init_scene() self.graphics_view.set_signal(signal) self.graphics_view.sample_rate = samp_rate else: self.scene_manager = FFTSceneManager(parent=self, graphic_view=self.graphics_view) self.graphics_view.setScene(self.scene_manager.scene) self.graphics_view.scene_manager = self.scene_manager 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)) self.create_connects() self.ui.btnLockBWSR.setChecked(self.bw_sr_are_locked) self.on_btn_lock_bw_sr_clicked() @property def mode(self): return self.device.mode @property def has_empty_device_list(self): return self.ui.cbDevice.count() == 0 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.btnSave.clicked.connect(self.on_save_clicked) self.__create_device_connects() self.timer.timeout.connect(self.update_view) self.ui.spinBoxSampleRate.editingFinished.connect(self.on_sample_rate_changed) self.ui.spinBoxGain.editingFinished.connect(self.on_gain_changed) self.ui.spinBoxFreq.editingFinished.connect(self.on_freq_changed) self.ui.spinBoxBandwidth.editingFinished.connect(self.on_bw_changed) self.ui.lineEditIP.editingFinished.connect(self.on_usrp_ip_changed) self.ui.cbDevice.currentIndexChanged.connect(self.on_selected_device_changed) self.ui.spinBoxNRepeat.editingFinished.connect(self.on_num_repeats_changed) self.ui.sliderYscale.valueChanged.connect(self.on_slideyscale_value_changed) if hasattr(self.graphics_view, "freq_clicked"): self.graphics_view.freq_clicked.connect(self.on_graphics_view_freq_clicked) if hasattr(self.graphics_view, "save_as_clicked"): self.graphics_view.save_as_clicked.connect(self.on_graphics_view_save_as_clicked) if hasattr(self.scene_manager, "signal"): self.scene_manager.signal.data_edited.connect(self.on_signal_data_edited) self.ui.btnLockBWSR.clicked.connect(self.on_btn_lock_bw_sr_clicked) def __create_device_connects(self): self.device.stopped.connect(self.on_device_stopped) self.device.started.connect(self.on_device_started) self.device.sender_needs_restart.connect(self.__restart_device_thread) def __update_send_indicator(self, width: int): y, h = self.ui.graphicsViewSend.view_rect().y(), self.ui.graphicsViewSend.view_rect().height() self.send_indicator.setRect(0, y - h, width, 2 * h + abs(y)) @pyqtSlot() def on_sample_rate_changed(self): self.device.sample_rate = self.ui.spinBoxSampleRate.value() if self.bw_sr_are_locked: self.ui.spinBoxBandwidth.setValue(self.ui.spinBoxSampleRate.value()) self.device.bandwidth = self.ui.spinBoxBandwidth.value() @pyqtSlot() def on_freq_changed(self): self.device.frequency = self.ui.spinBoxFreq.value() if self.mode == Mode.spectrum: self.scene_manager.scene.center_freq = self.ui.spinBoxFreq.value() self.scene_manager.clear_path() self.scene_manager.clear_peak() @pyqtSlot() def on_bw_changed(self): self.device.bandwidth = self.ui.spinBoxBandwidth.value() if self.bw_sr_are_locked: self.ui.spinBoxSampleRate.setValue(self.ui.spinBoxBandwidth.value()) self.device.sample_rate = self.ui.spinBoxSampleRate.value() @pyqtSlot() def on_usrp_ip_changed(self): self.device.ip = self.ui.lineEditIP.text() @pyqtSlot() def on_gain_changed(self): self.device.gain = self.ui.spinBoxGain.value() @pyqtSlot() def on_selected_device_changed(self): dev_name = self.ui.cbDevice.currentText() if dev_name == NetworkSDRInterfacePlugin.NETWORK_SDR_NAME: self.ui.cbDevice.blockSignals(True) self.ui.cbDevice.setCurrentText(self.device.name) self.ui.cbDevice.blockSignals(False) Errors.network_sdr_send_is_elsewhere() return nrep = self.ui.spinBoxNRepeat.value() sts = self.device.samples_to_send self.device.free_data() # gc.collect() # Cant do GC here, because the SencRecvDialog itself would be deleted (see https://github.com/jopohl/urh/issues/83) self.device = VirtualDevice(self.backend_handler, dev_name, self.device.mode, self.device.bandwidth, self.device.frequency, self.device.gain, self.device.sample_rate, sts, self.device.ip, nrep, self) self.__create_device_connects() if hasattr(self.scene_manager, "plot_data"): del self.scene_manager.plot_data if self.mode == Mode.receive: self.scene_manager = LiveSceneManager(np.array([]), parent=self) self.graphics_view.scene_manager = self.scene_manager self.graphics_view.setScene(self.scene_manager.scene) self.ui.lineEditIP.setVisible(dev_name == "USRP") self.ui.labelIP.setVisible(dev_name == "USRP") @pyqtSlot() def on_start_clicked(self): if self.mode == Mode.send: if self.ui.progressBar.value() >= self.ui.progressBar.maximum() - 1: self.on_clear_clicked() self.ui.spinBoxFreq.editingFinished.emit() self.ui.lineEditIP.editingFinished.emit() self.ui.spinBoxBandwidth.editingFinished.emit() self.ui.spinBoxSampleRate.editingFinished.emit() self.ui.spinBoxNRepeat.editingFinished.emit() if self.mode == Mode.send and self.device_is_sending: self.device.stop("Sending paused by user") else: self.device.start() @pyqtSlot() def on_num_repeats_changed(self): if not self.ui.spinBoxNRepeat.isVisible(): return self.device.num_sending_repeats = self.ui.spinBoxNRepeat.value() @pyqtSlot() def on_stop_clicked(self): self.device.stop("Stopped receiving: Stop button clicked") if self.mode == Mode.send: self.on_clear_clicked() @pyqtSlot() def on_device_stopped(self): self.graphics_view.capturing_data = False if self.mode == Mode.send: self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-start")) self.ui.btnStart.setToolTip("Start sending") self.device_is_sending = False else: self.ui.btnStart.setEnabled(True) self.ui.btnStop.setEnabled(False) self.ui.btnClear.setEnabled(True) self.ui.btnSave.setEnabled(self.device.current_index > 0) 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.spinBoxNRepeat.setEnabled(True) self.timer.stop() self.scene_manager.set_text("") self.update_view() @pyqtSlot() def on_device_started(self): if self.mode == Mode.receive: self.scene_manager.plot_data = self.device.data.real if self.device.data is not None else None self.ui.txtEditErrors.clear() self.scene_manager.set_text("Waiting for device..") self.graphics_view.capturing_data = True self.ui.btnSave.setEnabled(False) if self.mode == Mode.send: self.ui.btnStart.setIcon(QIcon.fromTheme("media-playback-pause")) self.ui.btnStart.setToolTip("Pause sending") self.device_is_sending = True else: self.ui.btnStart.setEnabled(False) self.ui.btnClear.setEnabled(self.mode == Mode.spectrum) self.ui.spinBoxNRepeat.setEnabled(False) self.ui.btnStop.setEnabled(True) if self.mode != Mode.spectrum: self.ui.spinBoxSampleRate.setDisabled(True) self.ui.spinBoxFreq.setDisabled(True) self.ui.spinBoxGain.setDisabled(True) self.ui.spinBoxBandwidth.setDisabled(True) self.ui.lineEditIP.setDisabled(True) self.ui.cbDevice.setDisabled(True) self.timer.start(self.update_interval) self.already_saved = False def update_view(self): txt = self.ui.txtEditErrors.toPlainText() new_errors = self.device.read_errors() if "No devices found for" in new_errors: self.device.stop_on_error("Could not establish connection to USRP") Errors.usrp_ip_not_found() self.on_clear_clicked() elif "FATAL: No supported devices found" in new_errors or \ "HACKRF_ERROR_NOT_FOUND" in new_errors or \ "HACKRF_ERROR_LIBUSB" in new_errors: self.device.stop_on_error("Could not establish connection to HackRF") Errors.hackrf_not_found() self.on_clear_clicked() elif "No module named gnuradio" in new_errors: self.device.stop_on_error("Did not find gnuradio.") Errors.gnuradio_not_installed() self.on_clear_clicked() elif "Address already in use" in new_errors: self.__restart_device_thread() if len(new_errors) > 1: self.ui.txtEditErrors.setPlainText(txt + new_errors) self.ui.progressBar.setValue(self.device.current_index) self.ui.lSamplesCaptured.setText("{0:n}".format(self.device.current_index)) self.ui.lSignalSize.setText("{0:n}".format((8 * self.device.current_index) / (1024 ** 2))) self.ui.lTime.setText(locale.format_string("%.2f", self.device.current_index / self.device.sample_rate)) if not self.device.sending_finished: self.ui.lblCurrentRepeatValue.setText(str(self.device.current_iteration + 1)) else: self.ui.lblCurrentRepeatValue.setText("Done") if self.device.current_index == 0: return if self.mode == Mode.receive: self.scene_manager.end = self.device.current_index elif self.mode == Mode.spectrum: x, y = self.device.spectrum if x is None or y is None: return self.scene_manager.scene.frequencies = x self.scene_manager.plot_data = y elif self.mode == Mode.send: self.__update_send_indicator(self.device.current_index) return self.scene_manager.init_scene() self.scene_manager.show_full_scene() self.graphics_view.update() def __restart_device_thread(self): self.device.stop("Restarting with new port") QApplication.processEvents() self.device.port = random.randint(1024, 65536) logger.info("Retry with port " + str(self.device.port)) self.device.start() QApplication.processEvents() @pyqtSlot() def on_clear_clicked(self): if self.mode == Mode.send: self.__update_send_indicator(0) else: self.scene_manager.clear_path() if self.mode in (Mode.send, Mode.receive): self.ui.txtEditErrors.clear() self.device.current_index = 0 self.device.current_iteration = 0 self.ui.lSamplesCaptured.setText("0") self.ui.lSignalSize.setText("0") self.ui.lTime.setText("0") self.ui.lblCurrentRepeatValue.setText("-") self.scene_manager.set_text("") self.ui.progressBar.setValue(0) self.ui.btnClear.setEnabled(False) self.ui.btnSave.setEnabled(False) elif self.mode == Mode.spectrum: self.scene_manager.clear_peak() @pyqtSlot() def on_save_clicked(self): data = self.device.data[:self.device.current_index] dev = self.device big_val = Formatter.big_value_with_suffix initial_name = "{0} {1}Hz {2}Sps {3}Hz.complex".format(dev.name, big_val(dev.frequency), big_val(dev.sample_rate), big_val(dev.bandwidth)).replace( Formatter.local_decimal_seperator(), "_").replace("_000", "") filename = FileOperator.save_data_dialog(initial_name, data, parent=self) self.already_saved = True if filename is not None and filename not in self.recorded_files: self.recorded_files.append(filename) def closeEvent(self, event: QCloseEvent): if self.device.backend == Backends.network: event.accept() return self.device.stop("Dialog closed. Killing recording process.") if self.mode == Mode.receive and not self.already_saved and self.device.current_index > 0: reply = QMessageBox.question(self, self.tr("Save data?"), self.tr("Do you want to save the data you have captured so far?"), QMessageBox.Yes | QMessageBox.No | QMessageBox.Abort) if reply == QMessageBox.Yes: self.on_save_clicked() elif reply == QMessageBox.Abort: event.ignore() return time.sleep(0.1) if self.device.backend != Backends.none: self.device.cleanup() self.files_recorded.emit(self.recorded_files) self.recording_parameters.emit(str(self.device.frequency), str(self.device.sample_rate), str(self.device.bandwidth), str(self.device.gain), str(self.device.name)) event.accept() @pyqtSlot() def on_btn_lock_bw_sr_clicked(self): self.bw_sr_are_locked = self.ui.btnLockBWSR.isChecked() constants.SETTINGS.setValue("lock_bandwidth_sample_rate", self.bw_sr_are_locked) 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(int) def on_slideyscale_value_changed(self, new_value: int): # Scale Up = Top Half, Scale Down = Lower Half middle = int((self.ui.sliderYscale.maximum() + 1 - self.ui.sliderYscale.minimum()) / 2) scale_up = new_value >= middle current_factor = self.graphics_view.sceneRect().height() / self.graphics_view.view_rect().height() scale_factor = (new_value + 1 - middle) / current_factor if scale_up else current_factor / new_value if scale_factor > 0: self.graphics_view.scale(1, scale_factor) @pyqtSlot(float) def on_graphics_view_freq_clicked(self, freq: float): self.ui.spinBoxFreq.setValue(freq) self.ui.spinBoxFreq.editingFinished.emit() @pyqtSlot() def on_signal_data_edited(self): signal = self.scene_manager.signal self.ui.progressBar.setMaximum(signal.num_samples) self.device.samples_to_send = signal.data self.scene_manager.init_scene() self.ui.graphicsViewSend.redraw_view() @pyqtSlot() def on_graphics_view_save_as_clicked(self): filename = FileOperator.get_save_file_name("signal.complex", parent=self) if filename: try: self.scene_manager.signal.save_as(filename) except Exception as e: QMessageBox.critical(self, self.tr("Error saving signal"), e.args[0])