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])