From 68fb4030ee50a25b912a7be701622490312421ca Mon Sep 17 00:00:00 2001 From: Toni Uhlig Date: Thu, 13 Aug 2020 10:52:10 +0200 Subject: flow-info.py, flow-undetected-to-pcap.py: throw socket error runtime exception if disconnect received Signed-off-by: Toni Uhlig --- examples/py-flow-info/flow-info.py | 2 +- examples/py-flow-undetected-to-pcap/flow-undetected-to-pcap.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/py-flow-info/flow-info.py b/examples/py-flow-info/flow-info.py index 07fbba651..96f2b0346 100755 --- a/examples/py-flow-info/flow-info.py +++ b/examples/py-flow-info/flow-info.py @@ -23,7 +23,7 @@ class nDPIsrvdSocket: def receive(self): recvd = self.sock.recv(NETWORK_BUFFER_MAX_SIZE - len(self.buffer)) - if recvd == '': + if len(recvd) == 0: raise RuntimeError('socket connection broken') self.buffer += recvd diff --git a/examples/py-flow-undetected-to-pcap/flow-undetected-to-pcap.py b/examples/py-flow-undetected-to-pcap/flow-undetected-to-pcap.py index b3a4f84a6..e97b66cf1 100755 --- a/examples/py-flow-undetected-to-pcap/flow-undetected-to-pcap.py +++ b/examples/py-flow-undetected-to-pcap/flow-undetected-to-pcap.py @@ -27,7 +27,7 @@ class nDPIsrvdSocket: def receive(self): recvd = self.sock.recv(NETWORK_BUFFER_MAX_SIZE - len(self.buffer)) - if recvd == '': + if len(recvd) == 0: raise RuntimeError('socket connection broken') self.buffer += recvd -- cgit v1.2.3