jketterl/openwebrx
1
0
Fork 0
mirror of https://github.com/jketterl/openwebrx.git synced 2026-01-09 10:10:01 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

backport pickle detection from @luarvique

Browse source
This commit is contained in:
Jakob Ketterl 2023-03-18 00:51:58 +01:00
parent f0e6839816
commit 949437c662
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
owrx/dsp.py
View file

@ -650,13 +650,17 @@ class DspManager(SdrSourceEventClient, ClientDemodulatorSecondaryDspEventClient)
def _unpickle(self, callback):
def unpickler(data):
b = data.tobytes()
# If we know it's not pickled, let us not unpickle
if len(b) < 2 or b[0] != 0x80 or not 3 <= b[1] <= pickle.HIGHEST_PROTOCOL:
callback(b.decode("ascii"))
return
io = BytesIO(b)
try:
while True:
callback(pickle.load(io))
except EOFError:
pass
# TODO: this is not ideal. is there a way to know beforehand if the data will be pickled?
except pickle.UnpicklingError:
callback(b.decode("ascii"))